iOS认识load方法

Posted 2020-09-22 多两口

----- 意识到load方法是因为最近学习Method Swilzzing时发现与其它的系统方法不同。 当时创建了一个UIViewController的catagory并重写了load方法。 

在这篇文章中指出：

+ load 作为 Objective-C 中的一个方法，与其它方法有很大的不同。只是一个在整个文件被加载到运行时，在 main 函数调用之前被 ObjC 运行时调用的钩子方法。其中关键字有这么几个：

• 文件刚加载

• main 函数之前

• 钩子方法

全文回答了两个问题这里做个学习总结

Q：load 方法是如何被调用的？

A：当 Objective-C 运行时初始化的时候，会通过 dyld_register_image_state_change_handler 在每次有新的镜像加入运行时的时候，进行回调。执行 load_images 将所有包含 load 方法的文件加入列表 loadable_classes ，然后从这个列表中找到对应的 load 方法的实现，调用 load 方法。

//  在系统内核做好程序准备工作之后，交由 dyld(是the dynamic link editor 的缩写，它是苹果的动态链接器) 负责余下的工作。 

//  1、首先在整个运行时初始化时 _objc_init 注册的回调
dyld_register_image_state_change_handler(dyld_image_state_dependents_initialized, 0/*not batch*/, &load_images);

//  2、每当有新的镜像加载到 runtime 时，都会执行 load_images 方法进行回调，并传入最新镜像的信息列表 infoList
const char * load_images(enum dyld_image_states state, uint32_t infoCount, const struct dyld_image_info infoList[]) {
    bool found;
    found = false;
    for (uint32_t i = 0; i < infoCount; i++) {
        if (hasLoadMethods((const headerType *)infoList[i].imageLoadAddress)) {
        //  3、如果在扫描镜像的过程中发现了 + load 符号 
            found = true;
            break;
        }
    }
    if (!found) return nil;
    recursive_mutex_locker_t lock(loadMethodLock);
    {
        rwlock_writer_t lock2(runtimeLock);
        //  4、会进入 load_images_nolock 来查找 load 方法
        found = load_images_nolock(state, infoCount, infoList);
    }
    if (found) {
        //  7、在将镜像加载到运行时、对 load 方法的准备就绪之后，执行 call_load_methods 开始调用 load 方法 
        call_load_methods();
    }
    return nil;
}

//  4、调用 load_image_nolock 
bool load_images_nolock(enum dyld_image_states state,uint32_t infoCount, const struct dyld_image_info infoList[] {
    bool found = NO;
    uint32_t i;
    i = infoCount;
    while (i--) {
        const headerType *mhdr = (headerType*)infoList[i].imageLoadAddress;
        if (!hasLoadMethods(mhdr)) continue;
        //  5、调用 prepare_load_methods 对 load 方法的调用进行准备
        prepare_load_methods(mhdr);
        found = YES;
    }
    return found;
}

//  5、调用 prepare_load_methods（将需要调用 load 方法的类添加到一个列表中）
void prepare_load_methods(const headerType *mhdr) {
    size_t count, i;
    runtimeLock.assertWriting();
    /* 通过 _getObjc2NonlazyClassList 获取所有的类的列表 */
    classref_t *classlist = _getObjc2NonlazyClassList(mhdr, &count);
    for (i = 0; i < count; i++) {
        //  6、调用 schedule_class_load 递归地安排当前类和没有调用 load 的父类进入列表
        /* 通过 remapClass 获取类对应的指针 */ 
        schedule_class_load(remapClass(classlist[i])); 
    }
    /* 同理处理分类 */
    category_t **categorylist = _getObjc2NonlazyCategoryList(mhdr, &count);
    for (i = 0; i < count; i++) {
        category_t *cat = categorylist[i];
        Class cls = remapClass(cat->cls);
        if (!cls) continue;  // category for ignored weak-linked class
        realizeClass(cls);
        assert(cls->ISA()->isRealized());
        add_category_to_loadable_list(cat);
    }
}

//  6、调用 schedule_class_load (红色代码保证了父类在子类前面调用 load 方法)
static void schedule_class_load(Class cls) {
    if (!cls) return;
    assert(cls->isRealized());
    if (cls->data()->flags & RW_LOADED) return;
    /* 先把父类加入待加载的列表 */
    schedule_class_load(cls->superclass);
    /* 再执行 add_class_to_loadabel_list 将当前类也加入进去 */
    add_class_to_loadable_list(cls);
    cls->setInfo(RW_LOADED); 
}

//  7、调用 call_load_methods
void call_load_methods(void) {
    ...
    do {
        while (loadable_classes_used > 0) {
            call_class_loads();
        }
        more_categories = call_category_loads();
    } while (loadable_classes_used > 0  ||  more_categories);
    ...
}

//  其中call_class_loads 会从一个待加载的类列表 loadable_classes 中寻找对应的类，然后找到 @selector(load) 的实现并执行。
static void call_class_loads(void) {
    int i;
    struct loadable_class *classes = loadable_classes;
    int used = loadable_classes_used;
    loadable_classes = nil;
    loadable_classes_allocated = 0;
    loadable_classes_used = 0;
    for (i = 0; i < used; i++) {
        Class cls = classes[i].cls;
        load_method_t load_method = (load_method_t)classes[i].method;
        if (!cls) continue;
        (*load_method)(cls, SEL_load); // 真正调用 +[XXObject load] 方法
    }
    if (classes) free(classes);
}

（镜像是）

Q：load 方法会有先走原类后走category这种调用顺序？

A：