Runtime 04 - 应用（动态创建类、交换方法）

动态创建类

需要创建的类结构如下

// 人的抽象模型
@interface Person : NSObject

@property (nonatomic, copy) NSString *name;
@property (nonatomic, assign) int age;

- (void)sayHello;
- (void)run;

@end

@implementation Person

- (void)sayHello {
    NSLog(@"Hi，我是%@，我今年%d了！", self.name, self.age);
}

- (void)run {
    NSLog(@"我跑啊跑啊跑啊跑。。。");
}

@end

// 歌唱家
@protocol Singer

- (void)sing;

@end

// 歌唱明星
@interface Star : Person <Singer>
{
    @public
    NSString *_company;
}
@end

@implementation Star

- (void)sing {
    NSLog(@"la la la so mi ~");
}

@end

动态创建类的示例

先定义一些全局函数，后面用作 Person 类的属性方法：

// 从一个 Class 对象中获取实例变量
#define GetIvar(class, name) class_getInstanceVariable(class, name)

// Person 类 name 属性的 getter
NSString* name(id self, SEL _cmd) {
    return object_getIvar(self, GetIvar([self class], "_name"));
}

// Person 类 name 属性的 setter
void setName(id self, SEL _cmd, NSString *name) {
    return object_setIvar(self, GetIvar([self class], "_name"), name);
}

// Person 类 age 属性的 getter
int age(id self, SEL _cmd) {
    return [object_getIvar(self, GetIvar([self class], "_age")) intValue];
}

// Person 类 age 属性的 setter
void setAge(id self, SEL _cmd, int age) {
    return object_setIvar(self, GetIvar([self class], "_age"), @(age));
}

用 Runtime 动态创建一个 Person 类

void allocatePersonClass() {
    // 创建 Person 类
    Class class = objc_allocateClassPair([NSObject class], "Person", 0);

    // 为 name 属性添加成员变量
    class_addIvar(class, "_name", sizeof(NSString*), 1, @encode(NSString));
    // 为 age 属性添加成员变量
    class_addIvar(class, "_age", sizeof(int), 2, @encode(int));

    // 为 name 属性添加 getter、setter
    class_addMethod(class, sel_registerName("name"), (IMP)name, "@16@0:8");
    class_addMethod(class, sel_registerName("setName:"), (IMP)setName, "v24@0:8@16");

    // 为 age 属性添加 getter、setter
    class_addMethod(class, sel_registerName("age"), (IMP)age, "i16@0:8");
    class_addMethod(class, sel_registerName("setAge:"), (IMP)setAge, "v20@0:8i16");
    
    // 为 Person 类添加 name 属性
    objc_property_attribute_t nameAttrs[] = {
        {"T", "@\"NSString\""}, // 类型：NSString
        {"N", ""},              // 原子性：nonatomic
        {"C", ""},              // 内存管理：copy
        {"V", "_name"}          // 变量名：_name
    };
    class_addProperty(class, "name", nameAttrs, 4);

    // 为 Person 类添加 age 属性
    const objc_property_attribute_t ageAttrs[] = {
        {"T", "i"},             // 类型：int
        {"N", ""},              // 原子性：nonatomic
        {"V", "_age"}           // 变量名：_age
    }; // 内存管理：默认 assign
    class_addProperty(class, "age", ageAttrs, 3);

    // 为 Person 类添加一个 sayHello 方法
    class_addMethod(class, sel_registerName("sayHello"), (IMP)sayHello, "v16@0:8");

    // 通过 Block 的方式为 Person 类添加一个 run 方法
    class_addMethod(class, sel_registerName("run"), imp_implementationWithBlock(^(id self, SEL _cmd) {
        NSLog(@"我跑啊跑啊跑啊跑。。。");
    }), "v16@0:8");

    // 注册 Person 类
    objc_registerClassPair(class);
}

用 Runtime 动态创建一个 Singer 协议

void allocateSingerProtocol() {
    // 创建 Singer
    Protocol *protocol = objc_allocateProtocol("Singer");
    // 为 Singer 协议添加一个 sing 实例方法
    protocol_addMethodDescription(protocol, sel_registerName("sing"), "v16@0:8", YES, YES);
    // 注册 Singer 协议
    objc_registerProtocol(protocol);
}

用 Runtime 动态创建一个 Star 类

void allocateStarClass() {
    // 创建 Star 类，继承自 Person 类
    Class class = objc_allocateClassPair(objc_getClass("Person"), "Star", 0);
    // 为 Star 类添加 Singer 协议
    class_addProtocol(class, objc_getProtocol("Singer"));

    // 为 Star 类添加一个 _company 成员变量
    class_addIvar(class, "_company", sizeof(NSString*), 3, @encode(NSString));

    // 为 Start 类实现 Singer 协议中的 sing 方法
    class_addMethod(class, sel_registerName("sing"), imp_implementationWithBlock(^(id self, SEL _cmd) {
        NSLog(@"la la la so mi ~");
    }), "v16@0:8");

    // 注册 Star 类
    objc_registerClassPair(class);
}

用 Runtime API 查看 Person 类的结构

• 查看成员变量：

  void lookIvars(Class class) {
      unsigned count;
      Ivar *ivars = class_copyIvarList(class, &count); // 成员变量列表
  
      for (int i = 0; i < count; i++) {
          Ivar ivar = ivars[i]; // 成员变量
          const char *name = ivar_getName(ivar); // 变量名
          const char *typeEncoding = ivar_getTypeEncoding(ivar); // 类型编码
          ptrdiff_t offset = ivar_getOffset(ivar); // 偏移量
          printf("%s %s %zd\n", name, typeEncoding, offset);
      }
  
      free(ivars);
  }
  

• 查看属性：

  void lookProperties(Class class) {
      unsigned int count;
      objc_property_t *properties = class_copyPropertyList(class, &count); // 属性列表
  
      for (int i = 0; i < count; i++) {
          objc_property_t property = properties[i]; // 属性
          const char *name = property_getName(property); // 属性名
          const char *attrs_desc = property_getAttributes(property); // 属性描述
          printf("%s %s\n", name, attrs_desc);
  
          unsigned attrsCount;
          objc_property_attribute_t *attrs = property_copyAttributeList(property, &attrsCount); // 描述属性的属性列表
  
          for (int j = 0; j < attrsCount; j++) {
              objc_property_attribute_t attr = attrs[j]; // 描述属性的属性
              printf("%s %s\n", attr.name, attr.value); // name：名称，value：值
          }
  
          printf("\n");
      }
  
      free(properties);
  }
  

• 查看方法：

  void lookMethods(Class class) {
      unsigned int count;
      Method *methods = class_copyMethodList(class, &count); // 方法列表
  
      for (int i = 0; i < count; i++) {
          Method method = methods[i]; // 方法
          const char *name = sel_getName(method_getName(method)); // 方法名
          const char *typeEncoding = method_getTypeEncoding(method); // 类型编码
          printf("%s %s\n", name, typeEncoding);
      }
  
      free(methods);
  }
  

allocatePersonClass();
Class class = objc_getClass("Person");

printf("\n---- 变量 ----\n");
lookIvars(class);

printf("\n---- 属性 ----\n");
lookProperties(class);

printf("\n---- 方法 ----\n");
lookMethods(class);

通过打印结果可以看到动态创建的类结构正常。

使用 Star 创建对象，并访问属性、成员变量、方法

allocatePersonClass();
allocateSingerProtocol();
allocateStarClass();

Class class = objc_getClass("Star");
id star = [[class alloc] init];

[star setValue:@"锅盖" forKey:@"name"];
NSString *name = [star valueForKey:@"name"];

[star setValue:@22 forKey:@"age"];
int age = [[star valueForKey:@"age"] intValue];

object_setIvar(star, GetIvar(class, "_company"), @22);
NSString *company = object_getIvar(star, GetIvar(class, "_company"));

NSLog(@"Star：%@, %d, %@", name, age, company);

[star performSelector:NSSelectorFromString(@"sayHello")];
[star performSelector:NSSelectorFromString(@"run")];
[star performSelector:NSSelectorFromString(@"sing")];

通过打印结果可以看到 Star 可以正常使用。

交换方法实现

需求是为所有 ViewController 的生命周期方法进行埋点，可以通过 Runtime 已少量的代码实现，并且不需要在每个 ViewController 的生命周期方法中增加代码。

@interface UIViewController (Hook)

@end

@implementation UIViewController (Hook)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^ {
        // hook：钩子函数
        Method method1 = class_getInstanceMethod(self, @selector(viewDidLoad));
        Method method2 = class_getInstanceMethod(self, @selector(hook_viewDidLoad));
        method_exchangeImplementations(method1, method2);
    });
}

- (void)hook_viewDidLoad {
    // do someting...
    [self hook_viewDidLoad];
}

@end

objc_property_attribute_t

用来描述一个属性的类型、读写性、内存管理、原子性、变量名。

typedef struct {
     const char * _Nonnull name;
     const char * _Nonnull value;
 } objc_property_attribute_t;

• 类型：

  • name：T
  • value：
    • 基本数据类型：i - int，f - float，d - double，q - NSInteger，Q - NSUInteger
    • id：@
    • 对象类型：@"ClassName"，例如 @"NSString"
    • Block：@?

• 读写性：

  • name：readwrite - 无（默认），readonly - R
  • value：""

• 内存管理：

  • name：assign - 无（基本数据类型时为默认），strong - &（对象类型时为默认），weak - W，copy - C
  • value：""

• 原子性：

  • name：atomic - 无（默认），nonatomic - N
  • value：""

• 变量名：

  • name：V
  • value：变量名

使用 property_getAttributes() 函数获取的是一个属性所有 objc_property_attribute_t 的字符串拼接：

• 类型、读写性、内存管理、原子性、变量名。
• (attr1.name + attr1.value), (attr2.name + attr2.value)