美文网首页
iOS分类(Catagory)底层源码分析

iOS分类(Catagory)底层源码分析

作者: pipiOU | 来源:发表于2019-11-25 23:56 被阅读0次

    说明:本文涉及到runtime源码(objc4-756.2)中的objc-runtime-new.h/objc-runtime-new.mm文件.

    首先来看category数据结构:

    //category数据结构
    struct category_t {
        const char *name;//分类名称
        classref_t cls;//分类所属的宿主类
        struct method_list_t *instanceMethods;//实例方法列表
        struct method_list_t *classMethods;//类方法列表
        struct protocol_list_t *protocols;//协议方法列表
        struct property_list_t *instanceProperties;//属性列表
        // Fields below this point are not always present on disk.
        struct property_list_t *_classProperties;
    
        method_list_t *methodsForMeta(bool isMeta) {
            if (isMeta) return classMethods;
            else return instanceMethods;
        }
    
        property_list_t *propertiesForMeta(bool isMeta, struct header_info *hi);
    };
    

    从category_t数据结构中不难看出,我们可以为分类当中添加实例方法类方法协议以及属性,在这里只以添加实例方法为例.

    分类加载调用栈:

    _objc_init(runtime初始化) —> map_2_images(镜像相关处理 )—>map_images _nolock(镜像相关处理)—> _read_images(读取镜像)—> remethodizeClass

    上面是对加载分类的一个简单流程说明,分类的实现逻辑都在remethodizeClass中,因此我们最主要的是看remethodizeClass实现.

    /***********************************************************************
    * remethodizeClass
    * Attach outstanding categories to an existing class.
    * Fixes up cls's method list, protocol list, and property list.
    * Updates method caches for cls and its subclasses.
    * Locking: runtimeLock must be held by the caller
     将未完成整合的分类附加到现有的类中
     整理类的方法列表,协议列表,属性列表
     更新类及其子类的方法缓存
     Locking:runtimeLock必须由调用者持有
    **********************************************************************/
    static void remethodizeClass(Class cls)
    {
        category_list *cats;//分类列表
        bool isMeta;//是否为元类
        
        runtimeLock.assertLocked();
        
        /*
        分析分类当中实例方法添加的逻辑
        所以假设isMeta = NO
        */
        isMeta = cls->isMetaClass();
    
        // Re-methodizing: check for more categories
        //unattachedCategoriesForClass函数获取cls中未完成整合的所有分类
        if ((cats = unattachedCategoriesForClass(cls, false/*not realizing*/))) {
            if (PrintConnecting) {
                _objc_inform("CLASS: attaching categories to class '%s' %s", 
                             cls->nameForLogging(), isMeta ? "(meta)" : "");
            }
            //将分类cats拼接到cls上
            attachCategories(cls, cats, true /*flush caches*/);        
            free(cats);
        }
    }
    

    attachCategories函数内部实现:

    #pragma mark - attachCategories函数内部实现
    // Attach method lists and properties and protocols from categories to a class.
    // Assumes the categories in cats are all loaded and sorted by load order, 
    // oldest categories first.
    //附加分类的方法、协议、属性到类中
    //假设cats中的类别都已加载并按加载顺序排序,
    static void attachCategories(Class cls, category_list *cats, bool flush_caches)
    {
        if (!cats) return;
        if (PrintReplacedMethods) printReplacements(cls, cats);
    
        /*
         分析分类当中实例方法添加的逻辑
         so,假设isMeta = NO
         */
        bool isMeta = cls->isMetaClass();
    
        /*
         声明3个局部变量 分别是方法列表、属性列表、协议列表 都是二维数组
         如method_list_t二维数组结构
         [[method_t,method_t,...],[method_t],[method_t,method_t,method_t],...]
         */
        // fixme rearrange to remove these intermediate allocations
        method_list_t **mlists = (method_list_t **)
            malloc(cats->count * sizeof(*mlists));
        property_list_t **proplists = (property_list_t **)
            malloc(cats->count * sizeof(*proplists));
        protocol_list_t **protolists = (protocol_list_t **)
            malloc(cats->count * sizeof(*protolists));
    
        // Count backwards through cats to get newest categories first
        int mcount = 0;
        int propcount = 0;
        int protocount = 0;
        int i = cats->count;//宿主类分类的总数
        bool fromBundle = NO;
        while (i--) {//这里是倒叙遍历,最先访问最后编译的分类
            //获取一个分类
            auto& entry = cats->list[i];
            //获取该分类的方法列表
            method_list_t *mlist = entry.cat->methodsForMeta(isMeta);
            if (mlist) {
                //最后编译的分类最先添加到分类数组中
                mlists[mcount++] = mlist;
                fromBundle |= entry.hi->isBundle();
            }
            //属性列表添加规则  同方法列表添加规则
            property_list_t *proplist = 
                entry.cat->propertiesForMeta(isMeta, entry.hi);
            if (proplist) {
                proplists[propcount++] = proplist;
            }
            //协议列表添加规则 同方法列表添加规则
            protocol_list_t *protolist = entry.cat->protocols;
            if (protolist) {
                protolists[protocount++] = protolist;
            }
        }
        //获取宿主类当中的rw数据,其中包含宿主类的方法列表信息
        auto rw = cls->data();
        //主要是针对 分类中有关于内存管理相关方法情况下,一些特殊处理
        prepareMethodLists(cls, mlists, mcount, NO, fromBundle);
        /*
           rw代表类
           methods代表类的方法列表
           attachLists 方法含义是 将含有mcount个元素的mlists拼接到rw的methods上
        */
        //到这里分类的方法才真正的添加到了宿主类中  因此说分类是运行时决议
        rw->methods.attachLists(mlists, mcount);
        free(mlists);
        if (flush_caches  &&  mcount > 0) flushCaches(cls);
    
        rw->properties.attachLists(proplists, propcount);
        free(proplists);
    
        rw->protocols.attachLists(protolists, protocount);
        free(protolists);
    }
         /*
          假如有3个分类,分类A中有两个方法,分类B中有一个方法,分类C中有3个方法
          那么最终mlists就得到下面的结果
          [[method_t,method_t],  [method_t], [method_t,method_t,method_t]]
           -------------------   ----------  -----------------------------
          分类A中的方法列表           B                   C
         */
        
    

    从上面的内部实现可以看出:

    1.Category是运行时决议,它在运行时才把分类方法添加到对应宿主类的方法列表中
    2.假如在分类有多个的情况下,如果每个分类都有一个同名的分类方法,那么最终最后编译的分类方法会生效(前面的会被”覆盖“)
    因为最后编译的分类方法会位于数组列表的前面位置,在消息发送或者消息函数查找的过程中,会根据选择器的名字来查找,一旦查找到了方法实现就是返回

    附加列表函数具体实现:

        //附加列表函数具体实现
        /*
         addedLists传递过来的二维数组
         [[method_t,method_t],   [method_t],   [method_t,method_t,method_t]]
         --------------------    ----------    -----------------------------
         分类A中的方法列表(A)         B                     C
        
         addedCount = 3
         */
        //addedLists:要附加的分类列表  addedCount:二维数组元素个数
        void attachLists(List* const * addedLists, uint32_t addedCount) {
            if (addedCount == 0) return;//判空
    
            if (hasArray()) {
                // many lists -> many lists
                //列表中原有元素总数  假设oldCount = 2
                uint32_t oldCount = array()->count;
                //拼接之后的元素总数  5
                uint32_t newCount = oldCount + addedCount;
               //根据新总数重新分配内存
                setArray((array_t *)realloc(array(), array_t::byteSize(newCount)));
               //重新设置元素总数  5
                array()->count = newCount;
                /*
                 内存移动
                 [[],[],[],[原有的第一个元素],[原有的第二个元素]]
                 */
                memmove(array()->lists + addedCount, array()->lists, 
                        oldCount * sizeof(array()->lists[0]));
                /*
                 内存拷贝
                 [
                    A  --->  [addedLists中的第一个元素],
                    B  --->  [addedLists中的第二个元素],
                    C  --->  [addedLists中的第三个元素],
                    [原有的第一个元素],
                    [原有的第二个元素]
                 ]
                 */
                //memcpy函数把addedLists中的元素拷贝到lists当中
                memcpy(array()->lists, addedLists, 
                       addedCount * sizeof(array()->lists[0]));
            }//后两个分支是 关于列表当中存储的分类具体采取的是list还是array_t结构的区别,不影响分析分类实现逻辑
            else if (!list  &&  addedCount == 1) {
                // 0 lists -> 1 list
                list = addedLists[0];
            } 
            else {
                // 1 list -> many lists
                List* oldList = list;
                uint32_t oldCount = oldList ? 1 : 0;
                uint32_t newCount = oldCount + addedCount;
                setArray((array_t *)malloc(array_t::byteSize(newCount)));
                array()->count = newCount;
                if (oldList) array()->lists[addedCount] = oldList;
                memcpy(array()->lists, addedLists, 
                       addedCount * sizeof(array()->lists[0]));
            }
        }
    
    

    从上面的内部实现可以看出:

    方法同名情况下,分类方法会”覆盖“宿主类的方法的原因:
    宿主类方法仍然存在,但是分类方法位于方法列表数组前面的位置了,在消息函数查找的过程中,根据方法的名字来查找,一旦查找到方法实现就会返回,也即分类方法会被优先实现.

    相关文章

      网友评论

          本文标题:iOS分类(Catagory)底层源码分析

          本文链接:https://www.haomeiwen.com/subject/uysswctx.html