add __strong ivar at runtime under ARC

2019-08-12 01:39发布

问题:

A normal ivar declared in @interface is __strong default.

@interface XLPerson : NSObject {
    NSString *name; // __strong default
}
@end

Now, I create above class at runtime:

Class XLPerson = objc_allocateClassPair([NSObject class], "XLPerson", 0);
size_t size = sizeof(NSString*);
class_addIvar(XLPerson, "name", size, log2(align), @encode(NSString*)));
objc_registerClass(XLPerson);

However, the ivar named "name" isn't a __strong ivar.
While I using object_setIvar(), the Ivar can't hold the newValue (it will be deallocated at the end of Autorelease Pool).

id person = [[XLPerson alloc] init];
Ivar ivar = class_getInstanceVariable(XLPerson, "name");

@autoreleasepool {
   object_setIvar(person, ivar, [NSString stringWithFormat@"Stack%@", @"Overflow"]);
   // @"StackOverflow" will be deallocated. 
}
NSLog(@"%@", object_getIvar(person, ivar));
// BAD_ACCESS  *** -[CFString retain]: message sent to deallocated instance 0x1004002f0

Then I find two functions class_setIvarLayout and class_setWeakIvarLayout, but there is not any useful information in Objective-C Runtime Reference.

So, how can I add a __strong Ivar into my class created at runtime?

回答1:

You need to set ivar layouts into class, in your case (a single strong ivar):

class_setIvarLayout(class, (const uint8_t *)"\x01");

then tell runtime your class is managed under ARC:

static void fixup_class_arc(Class class) {
    struct {
        Class isa;
        Class superclass;
        struct {
            void *_buckets;
            uint32_t _mask;
            uint32_t _occupied;
        } cache;
        uintptr_t bits;
    } *objcClass = (__bridge typeof(objcClass))class;

#if !__LP64__
#define FAST_DATA_MASK 0xfffffffcUL
#else
#define FAST_DATA_MASK 0x00007ffffffffff8UL
#endif

    struct {
        uint32_t flags;
        uint32_t version;
        struct {
            uint32_t flags;
        } *ro;
    } *objcRWClass = (typeof(objcRWClass))(objcClass->bits & FAST_DATA_MASK);

#define RO_IS_ARR 1<<7

    objcRWClass->ro->flags |= RO_IS_ARR;
}

Call it after your class's registering, this is very tricky and maybe unstable, I would prefer properties rather than ivars.

My blog about this in Chinese, link here