iOS底层原理 - 探寻Category本质

面试题引发的思考:

Q: Category的作用是什么?

  • 在不修改原来类的基础上,为一个类扩展方法。

Q: Category的实现原理?

  • Category编译之后的底层结构是struct category_t,里面存储着分类的对象方法、类方法、属性、协议信息;
  • 在程序运行的时候,Runtime会将Category的数据,合并到类信息中(类对象、元类对象中)。

1. Category介绍

// TODO: -----------------  Person类  -----------------
@interface Person : NSObject
- (void)study;
+ (void)exam;
@end

@implementation Person
- (void)study {
    NSLog(@"Person - study");
}
+ (void)exam {
    NSLog(@"Person - exam");
}
@end

// TODO: -----------------  Person (Student)类  -----------------
@interface Person (Student) <NSCopying, NSCoding>
- (void)study;
+ (void)exam;
- (void)study1;
+ (void)exam1;
@property (nonatomic, copy) NSString *name;
@property (nonatomic, assign) int age;
@end

@implementation Person (Student)
- (void)study {
    NSLog(@"Person (Student) - study");
}
+ (void)exam {
    NSLog(@"Person (Student) - exam");
}
- (void)study1 {
    NSLog(@"Person (Student) - study1");
}
+ (void)exam1 {
    NSLog(@"Person (Student) - exam1");
}
@end

// TODO: -----------------  Person (Teacher)类  -----------------
@interface Person (Teacher)
- (void)study;
@end

@implementation Person (Teacher)
- (void)study {
    NSLog(@"Person (Teacher) - study");
}
@end

// TODO: ----------------- ViewController类  -----------------
- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    Person *person = [[Person alloc] init];
    [person study];
    [person study1];
    [Person exam];
    [Person exam1];
}

根据iOS底层原理 - OC对象的本质(二)可知:

Person的instance对象调用对象方法study时,通过instance对象的isa找到class对象,最后找到对象方法的实现进行调用。

Q: 那又是怎样调用Person分类的对象方法study1呢?

是通过instance对象的isa找到一个category-class对象,然后找到对象方法的实现进行调用吗?

实际上是:

  • 一个instance对象,只有一个class对象。

  • class对象会把本身以及分类的对象方法放到class对象中,即study方法、study1方法都在Person的class对象中。
    即当instance调用对象方法studystudy1时,通过instance的isa找到class,最后找到对象方法的实现进行调用。

  • 同理meta-class对象会把本身以及分类的类方法放到meta-class对象中。即exam方法、exam1方法都在Person的meta-class对象中。
    即当Person调用类方法examexam1时,通过class的isa找到meta-class,最后找到类方法的实现进行调用。


2. Category底层结构:

OC源码objc-runtime-new.h文件可知:

category_t结构体

OC源码可知:

  • Category本质是category_t结构体;
  • category_t结构体存储着对象方法、类方法、协议、属性;
  • category_t结构体没有成员变量的定义,说明分类是不允许添加成员变量;
  • 分类中添加的属性不会生成成员变量,只会生成get方法、set方法的声明,需要我们自己去实现。

(1) 对以上结论进行实例分析:

通过命令行语句xcrun -sdk iphoneos clang -arch arm64 -rewrite-objc Preson+Student.m将获得C++文件,文件中可以找到category_t结构体:

category_t结构体

category_t结构体与OC源码中一样。

通过category_t可以找到_OBJC_$_CATEGORY_Person_$_Student

_OBJC_$_CATEGORY_Person_$_Student

_OBJC_$_CATEGORY_Person_$_Student结构体中的参数与category_t结构体中的参数一一对应,分别对应类名、_class_t结构体、对象方法、类方法、协议、属性。

接着通过_OBJC_$_CATEGORY_INSTANCE_METHODS_Person_$_Student找到instance_methods实现:

instance_methods实现

里面有两个方法studystudy1,这是我们在分类中定义的对象方法。

接着通过_OBJC_$_CATEGORY_CLASS_METHODS_Person_$_Student找到class_methods实现:

class_methods实现

里面有两个方法examexam1,这是我们在分类中定义的类方法。

接着通过_OBJC_CATEGORY_PROTOCOLS_$_Person_$_Student找到protocols实现:

protocols实现

里面有两个协议NSCopyingNSCoding,这是我们在分类中遵守的协议。

接着通过_OBJC_$_PROP_LIST_Person_$_Student找到properties实现:

properties实现

里面有两个参数nameage,这是我们在分类中定义的属性。

由以上分析可知:

分类源码中将我们定义的对象方法、类方法、协议、属性等都存放在catagory_t结构体中。


(2) catagory_t存储方式

现在通过Runtime源码查看catagory_t存储的对象方法、类方法、属性、协议等是如何存储在类对象中的。

首先找到Runtime初始化函数_objc_init

Runtime初始化函数

然后通过&map_images读取模块找到map_images函数:

map_images函数

然后通过map_images_nolock找到_read_images函数:

map_images_nolock函数

然后通过_read_images找到Discover categories分类相关代码:

Discover categories

Discover categories分类相关代码可知:

搜索分类信息,也就是说这段代码的功能是用来查找是否有分类。

通过_getObjc2CategoryList函数获取到分类列表以后,再进行遍历,获得其中的方法、协议、属性等;

然后通过remethodizeClass函数重新组织类方法,我们进入remethodizeClass函数内部:

remethodizeClass函数

remethodizeClass函数相关代码可知:

attachCategories函数传入了对象cls和分类数组cats,对两者进行相关操作。我们进入attachCategories函数内部:

attachCategories函数

由前文分析可知:

分类信息存储在category_t结构体中,而一个类可以有多个分类,则多个分类就保存在categroy_list中。

根据以上源码对对象方法的操作进行分析:

  • 首先通过malloc分配内存空间来存储方法数组;
  • 然后通过遍历取出分类中对象的方法列表;
  • 得到class_rw_t结构体rw,并通过attachList函数将所有分类的对象方法附加到类对象的方法列表中。

由以上代码可知: attachCategories函数对类方法、属性、协议的处理与对象方法同理。

接下来进入attachList函数内部:

attachList函数

其中有两个函数:

  • memmove内存移动:
    void *memmove(void *__dst, const void *__src, size_t __len);
    __src的内存移动__len块内存到__dst

  • memcpy内存拷贝:
    void *memcpy(void *__dst, const void *__src, size_t __n);
    __src的内存拷贝__n块内存到__dst

根据以上代码分析:

  • addedLists是所有分类的方法列表、属性列表、协议列表,addedCount是分类的个数;
  • array()->lists是类的方法列表、属性列表、协议列表,array()->count是原来的个数;
  • memmove函数:将array()->lists的内存移动oldCount * sizeof(array()->lists[0])块内存到array()->lists + addedCount中;
  • memcpy函数:将addedLists的内存复制addedCount * sizeof(array()->lists[0])块内存到array()->lists中。

(3) 根据最初的代码分析:

代码包括:Person类,分类Person+Teacher和分类Person+Student,
array()->lists是类的方法列表,array()->count1,结构显示如下:

方法列表初始结构

通过realloc函数重新分配内存,addedLists包括分类Person+Teacher和分类Person+Student的方法列表,addedCount2,结构显示如下:

重新分配内存之后的结构
  • 通过memmove函数将array()->lists的内存移动oldCount * sizeof(array()->lists[0])块内存到array()->lists + addedCount
    即将类的方法列表的内存移动1块内存到原地址+2的内存中。

  • 通过memcpy函数将addedLists的内存复制addedCount * sizeof(array()->lists[0])块内存到array()->lists
    即将两个分类的方法列表的内存移动2块内存到原地址的内存中。

结构显示如下:

memmove、memcpy之后的结构

由以上分析可知:

  • 分类的方法列表追加到类的方法列表前面,保证了分类方法的优先调用;
  • 分类重写类的方法,不是覆盖,而是优先调用,类的方法存在最后的内存中;
  • 多个分类和类有同样的方法时,调用方法是按照分类的编译顺序逆序排列,示例如下:
方法调用顺序

同理可知:

分类的属性、协议的处理与分类的方法同理。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,904评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,581评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,527评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,463评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,546评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,572评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,582评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,330评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,776评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,087评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,257评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,923评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,571评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,192评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,436评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,145评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352

推荐阅读更多精彩内容