/* * Copyright (c) 2008, 2009, 2010, 2011, 2012 * Jonathan Schleifer <js@webkeks.org> * * All rights reserved. * * This file is part of ObjFW. It may be distributed under the terms of the * Q Public License 1.0, which can be found in the file LICENSE.QPL included in * the packaging of this file. * * Alternatively, it may be distributed under the terms of the GNU General * Public License, either version 2 or 3, which can be found in the file * LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this * file. */ /* * This file replaces NSAutoreleasePool with OFAutoreleasePool when it is * linked. This is done so there is no conflict because OFBlocks are used * (blocks are OFBlocks as soon as ObjFW is linked). An application expecting * an NSBlock, but getting an OFBlock because ObjFW is linked, would fail to * autorelease the block otherwise, as the block would be in an * OFAutoreleasePool. By replacing NSAutoreleasePool with OFAutoreleasePool, * the application will still properly free the autoreleased block. */ #include "config.h" #import <objc/runtime.h> #import "OFAutoreleasePool.h" static id alloc(Class self, SEL _cmd) { return [OFAutoreleasePool alloc]; } static void addObject(Class self, SEL _cmd, id obj) { [OFAutoreleasePool addObject: obj]; } static id autorelease(id self, SEL _cmd) { [OFAutoreleasePool addObject: self]; return self; } static void __attribute__((constructor)) init(void) { Class NSAutoreleasePool = objc_getClass("NSAutoreleasePool"); Class NSObject = objc_getClass("NSObject"); Method allocMethod; Method addObjectMethod; Method autoreleaseMethod; if (NSAutoreleasePool == Nil || NSObject == Nil) return; allocMethod = class_getClassMethod(NSAutoreleasePool, @selector(alloc)); addObjectMethod = class_getClassMethod(NSAutoreleasePool, @selector(addObject:)); autoreleaseMethod = class_getInstanceMethod(NSObject, @selector(autorelease)); if (allocMethod == NULL || addObjectMethod == NULL || autoreleaseMethod == NULL) return; class_replaceMethod(object_getClass(NSAutoreleasePool), @selector(alloc), (IMP)alloc, method_getTypeEncoding(allocMethod)); class_replaceMethod(object_getClass(NSAutoreleasePool), @selector(addObject:), (IMP)addObject, method_getTypeEncoding(addObjectMethod)); class_replaceMethod(NSObject, @selector(autorelease), (IMP)autorelease, method_getTypeEncoding(autoreleaseMethod)); }