前言

對(duì)于干活而言靴寂，針對(duì)Category的運(yùn)用牛柒，只要知道怎么用，有些需要注意的地方能夠注意到（比如，Category里定義的方法與主類(lèi)相同時(shí)歇攻，這個(gè)方法調(diào)用時(shí)實(shí)際上是Category里的）就可以了购公。另外呻率，對(duì)于一些地方的運(yùn)用很澄，可以知道如何給Category加屬性基本上也就夠了。

但實(shí)際上他托，有的時(shí)候掖肋，還是會(huì)碰到有人問(wèn)你一句Category是怎么實(shí)現(xiàn)的。從我個(gè)人的理解赏参，其實(shí)就是希望你從其實(shí)現(xiàn)原理上來(lái)回答那幾個(gè)需要注意的地方志笼，也就是當(dāng)Category中定義了一個(gè)與主類(lèi)相同的方法時(shí)，會(huì)調(diào)用Category的把篓，而不是主類(lèi)的纫溃。

當(dāng)然了，也可能會(huì)碰到再有人問(wèn)你如果就要調(diào)用主類(lèi)的呢韧掩？

總之皇耗，林林總總就是這樣的。你能做多少事，跟你能說(shuō)多少事郎楼，這之間的差距就是你資（zhuang）深（bi）的程度了。當(dāng)然了窒悔，如果能從原理的角度上呜袁，加深對(duì)于Category的認(rèn)識(shí)也是很好的事情，所以简珠，讓我們開(kāi)始資（zhuang）深（bi）吧～

Category的實(shí)現(xiàn)

先讓我們來(lái)看看Category的數(shù)據(jù)結(jié)構(gòu)

typedef 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;
} category_t;

上面這個(gè)就是Category的結(jié)構(gòu)體阶界，我們可以看到，Category從結(jié)構(gòu)上來(lái)講聋庵，可以存儲(chǔ)實(shí)例方法(instanceMethods)膘融、類(lèi)方法(classMethods)、委托(protocols)以及屬性(instanceProperties).

但在這里我們要看到祭玉，結(jié)構(gòu)體中雖然有屬性氧映，但并沒(méi)有實(shí)例變量（ivar）的存儲(chǔ)，這就回答了我們一個(gè)問(wèn)題——為什么Category加屬性需要自己使用objc_getAssociatedObject的方式來(lái)實(shí)現(xiàn)脱货，實(shí)際上就是因?yàn)閏ategory里不能存儲(chǔ)實(shí)例變量岛都。

比如我們可以在Category中寫(xiě)一個(gè)屬性，具體如下：

@interface NSObject (IndieBandName)
@property (nonatomic, copy) NSString *indieBandName;
@end

通常來(lái)講振峻，我們這么寫(xiě)是不會(huì)有問(wèn)題的臼疫，編譯也過(guò)得去。但只要你一使用扣孟，就會(huì)報(bào)錯(cuò)：錯(cuò)誤的內(nèi)容大概是告訴你找不到setIndieBandName方法這類(lèi)的烫堤。所以，我們還要實(shí)現(xiàn)它的getter,setter方法

- (void)setIndieBandName:(NSString *)indieBandName
{
    _indieBandName = [indieBandName copy];
}

寫(xiě)到這里發(fā)現(xiàn)問(wèn)題了吧凤价，_indieBandName這個(gè)實(shí)例變量在Category中根本就沒(méi)生成鸽斟！結(jié)合category的結(jié)構(gòu)體也可以理解，因?yàn)閏ategory_t里料仗，也沒(méi)地方保存湾盗！所以，我們才使用objc_getAssociatedObject來(lái)保存對(duì)象立轧，用來(lái)充當(dāng)getter／setter方法中負(fù)責(zé)返回和保存的實(shí)例變量格粪。

好了讓我們繼續(xù)下去，我們知道了Category的結(jié)構(gòu)體后氛改，就要看看Category里定義的方法是怎么加載的了帐萎。
看過(guò)OC對(duì)象內(nèi)存布局的人都知道，OC的類(lèi)對(duì)象和元類(lèi)對(duì)象中有methodList來(lái)分別保存實(shí)例方法和類(lèi)方法胜卤。那么Category的方法到底是怎么保存在類(lèi)對(duì)象里的呢疆导？這就要從Category的加載聊起。

Category的加載

我們知道葛躏，Objective-C的運(yùn)行是依賴(lài)OC的runtime的澈段，而OC的runtime和其他系統(tǒng)庫(kù)一樣悠菜，是OS X和iOS通過(guò)dyld動(dòng)態(tài)加載的。具體的入口函數(shù)大概是這個(gè)樣子：

void _objc_init(void)
{
    static bool initialized = false;
    if (initialized) return;
    initialized = true;

    // fixme defer initialization until an objc-using image is found?
    environ_init();
    tls_init();
    lock_init();
    exception_init();

    // Register for unmap first, in case some +load unmaps something
    _dyld_register_func_for_remove_image(&unmap_image);
    dyld_register_image_state_change_handler(dyld_image_state_bound,
                                         1/*batch*/, &map_images);
    dyld_register_image_state_change_handler(
        dyld_image_state_dependents_initialized, 0/*not batch*/, &load_images);
}

寫(xiě)這些東西就是為了說(shuō)明category被附加到類(lèi)上面是在map_images的時(shí)候發(fā)生的败富，我們截取在_read_images方法的結(jié)尾悔醋，它大概是這個(gè)樣子的：

for (EACH_HEADER) {
    category_t **catlist =
        _getObjc2CategoryList(hi, &count);
    for (i = 0; i < count; i++) {
        category_t *cat = catlist[i];
        class_t *cls = remapClass(cat->cls);

        if (!cls) {
            // Category's target class is missing (probably weak-linked).
            // Disavow any knowledge of this category.
            catlist[i] = NULL;
            if (PrintConnecting) {
                _objc_inform("CLASS: IGNORING category \?\?\?(%s) %p with "
                             "missing weak-linked target class",
                             cat->name, cat);
            }
            continue;
        }

        // Process this category. 
        // First, register the category with its target class. 
        // Then, rebuild the class's method lists (etc) if 
        // the class is realized. 
        BOOL classExists = NO;
        if (cat->instanceMethods ||  cat->protocols 
            ||  cat->instanceProperties)
        {
            addUnattachedCategoryForClass(cat, cls, hi);
            if (isRealized(cls)) {
                remethodizeClass(cls);
                classExists = YES;
            }
            if (PrintConnecting) {
                _objc_inform("CLASS: found category -%s(%s) %s",
                             getName(cls), cat->name,
                             classExists ? "on existing class" : "");
            }
        }

        if (cat->classMethods  ||  cat->protocols 
            /* ||  cat->classProperties */)
        {
            addUnattachedCategoryForClass(cat, cls->isa, hi);
            if (isRealized(cls->isa)) {
                remethodizeClass(cls->isa);
            }
            if (PrintConnecting) {
                _objc_inform("CLASS: found category +%s(%s)",
                             getName(cls), cat->name);
            }
        }
    }
}

從這個(gè)代碼中，我們可以看到兽叮，首先芬骄，系統(tǒng)獲取了Category的列表（catlist），之后就開(kāi)始遍歷鹦聪，遍歷時(shí)拿到每一個(gè)Category(category_t *cat = catlist[i])账阻，同時(shí)獲取Category所附著的類(lèi)（class_t *cls = remapClass(cat->cls)）。然后就開(kāi)始干活了泽本，具體步驟如下：

1. 如果cat結(jié)構(gòu)體中淘太，instanceMethods，protocols观挎，instanceProperties三個(gè)之中琴儿，只要有值，就會(huì)通過(guò)addUnattachedCategoryForClass(cat,cls,hi)方法嘁捷，把Category和類(lèi)先關(guān)聯(lián)起來(lái)造成，但這時(shí)還沒(méi)有針對(duì)Category的方法做操作。
2. 如果cat結(jié)構(gòu)體中雄嚣，classMethods晒屎，protocols兩個(gè)當(dāng)中，只要有值缓升，就睡通過(guò)addUnattachedCategoryForClass(cat,cls->isa,hi)方法鼓鲁，將Category和元類(lèi)對(duì)象關(guān)聯(lián)起來(lái)。

PS. cls->isa指向的就是元類(lèi)對(duì)象港谊，這塊大家應(yīng)該知道哈～ 另外骇吭，我們發(fā)現(xiàn)上面與元類(lèi)對(duì)象關(guān)聯(lián)的代碼中有一小段注釋掉的cat->classProperties。仔細(xì)想想歧寺，哎呦我靠燥狰，看來(lái)當(dāng)初蘋(píng)果還想弄個(gè)類(lèi)屬性出來(lái)哈～

至此，我們已經(jīng)將Category有關(guān)實(shí)例方法斜筐、委托以及實(shí)例屬性關(guān)聯(lián)到類(lèi)對(duì)象上龙致，而類(lèi)方法等關(guān)聯(lián)到元類(lèi)對(duì)象上，后面就是要重新構(gòu)建類(lèi)對(duì)象和元類(lèi)對(duì)象的methodList了顷链。

實(shí)際上目代，真正處理Category中方法的實(shí)現(xiàn)，是一個(gè)叫做remethodizeClass的靜態(tài)方法，它大概是長(zhǎng)這個(gè)樣子：

static void remethodizeClass(class_t *cls)
{
    category_list *cats;
    BOOL isMeta;

    rwlock_assert_writing(&runtimeLock);

    isMeta = isMetaClass(cls);

    // Re-methodizing: check for more categories
    if ((cats = unattachedCategoriesForClass(cls))) {
        chained_property_list *newproperties;
        const protocol_list_t **newprotos;

        if (PrintConnecting) {
            _objc_inform("CLASS: attaching categories to class '%s' %s",
                     getName(cls), isMeta ? "(meta)" : "");
        }

        // Update methods, properties, protocols

        BOOL vtableAffected = NO;
        attachCategoryMethods(cls, cats, &vtableAffected);

        newproperties = buildPropertyList(NULL, cats, isMeta);
        if (newproperties) {
            newproperties->next = cls->data()->properties;
            cls->data()->properties = newproperties;
        }

        newprotos = buildProtocolList(cats, NULL, cls->data()->protocols);
        if (cls->data()->protocols  &&  cls->data()->protocols != newprotos) {
            _free_internal(cls->data()->protocols);
        }
        cls->data()->protocols = newprotos;

        _free_internal(cats);

        // Update method caches and vtables
        flushCaches(cls);
        if (vtableAffected) flushVtables(cls);
    }
}

這段代碼中榛了，我們看到聲明了新的屬性列表(newproperties)和委托列表(newprotos)在讶，之后，又調(diào)用了attachCategoryMethods方法忽冻，實(shí)際上真朗，這個(gè)方法就是將指定的類(lèi)對(duì)象中的methodList與Category中methodList進(jìn)行整合的地方。

不管怎么樣僧诚，attach后，newproperties和newprotos都獲得了新的數(shù)據(jù)蝗碎，并把這些數(shù)據(jù)賦值給了cls湖笨。整體的融合也就完成了。

好了蹦骑，看來(lái)attachCategoryMethods方法是其中的核心慈省，讓我們來(lái)看看它都干了啥

static void 
attachCategoryMethods(class_t *cls, category_list *cats,
                  BOOL *inoutVtablesAffected)
{
    if (!cats) return;
    if (PrintReplacedMethods) printReplacements(cls, cats);

    BOOL isMeta = isMetaClass(cls);
    method_list_t **mlists = (method_list_t **) 
                              _malloc_internal(cats->count * sizeof(*mlists));

    // Count backwards through cats to get newest categories first
    int mcount = 0;
    int i = cats->count;
    BOOL fromBundle = NO;
    while (i--) {
        method_list_t *mlist = cat_method_list(cats->list[i].cat, isMeta);
        if (mlist) {
            mlists[mcount++] = mlist;
            fromBundle |= cats->list[i].fromBundle;
        }
    }

    attachMethodLists(cls, mlists, mcount, NO, fromBundle, inoutVtablesAffected);

    _free_internal(mlists);

}

這段代碼主要的目的就是把所有的Category都拿出來(lái)遍歷一次，然后用新的方法列表mlists來(lái)填裝眠菇，填裝完成后边败，就把它連同類(lèi)對(duì)象扔給了attachMethodLists方法了。

好吧捎废，我們需要看一下attachMethodLists又都干了啥...... GFY

for (uint32_t m = 0; 
     (scanForCustomRR || scanForCustomAWZ)  &&  m < mlist->count;
     m++)
{
     SEL sel = method_list_nth(mlist, m)->name;
     if (scanForCustomRR  &&  isRRSelector(sel)) {
         cls->setHasCustomRR();
         scanForCustomRR = false;
     } 
     else if (scanForCustomAWZ  &&  isAWZSelector(sel)) {
         cls->setHasCustomAWZ();
         scanForCustomAWZ = false;
     }
}

 // Fill method list array
newLists[newCount++] = mlist;

.
.
.

// Copy old methods to the method list array
for (i = 0; i < oldCount; i++) {
    newLists[newCount++] = oldLists[i];
}

整體來(lái)說(shuō)笑窜，就是又有了一個(gè)新的數(shù)組，這數(shù)組首先是把所有的Category的方法都保存了下來(lái)登疗，之后排截，又把所謂的老方法列表追加到了數(shù)組中，這里的oldList實(shí)際上就是你類(lèi)對(duì)象中原有的方法列表辐益。換句話說(shuō)断傲，在重組的新方法列表中，Category方法都是排在前面的智政，而來(lái)方法都追加到了后面认罩，這時(shí)，如果發(fā)生Category和主類(lèi)中都有相同的方法续捂，那么Category的方法在前面而主類(lèi)的方法在后面垦垂。

到這里，我們又能明確一件事情了疾忍，那就是為什么當(dāng)Category和主類(lèi)擁有相同方法時(shí)乔外，實(shí)際被執(zhí)行的是Category的方法。因?yàn)樗谧钋懊媛铮?/strong> 同時(shí)一罩，我們也能夠發(fā)現(xiàn)杨幼，主類(lèi)的方法并沒(méi)有被覆蓋，而只是排在了后面。因此差购，如果我們想的話四瘫，只需要獲取當(dāng)前對(duì)象的方法列表，倒序找到第一個(gè)或者正向找到最后一個(gè)同名方法欲逃，那就是主類(lèi)的方法了找蜜。

static void 
attachCategoryMethods(class_t *cls, category_list *cats,
                  BOOL *inoutVtablesAffected)
{
    if (!cats) return;
    if (PrintReplacedMethods) printReplacements(cls, cats);

    BOOL isMeta = isMetaClass(cls);
    method_list_t **mlists = (method_list_t **) 
                              _malloc_internal(cats->count * sizeof(*mlists));

    // Count backwards through cats to get newest categories first
    int mcount = 0;
    int i = cats->count;
    BOOL fromBundle = NO;
    while (i--) {
        method_list_t *mlist = cat_method_list(cats->list[i].cat, isMeta);
        if (mlist) {
            mlists[mcount++] = mlist;
            fromBundle |= cats->list[i].fromBundle;
        }
    }

    attachMethodLists(cls, mlists, mcount, NO, fromBundle, inoutVtablesAffected);

    _free_internal(mlists);

}

Class currentClass = [MyClass class];
MyClass *my = [[MyClass alloc] init];

if (currentClass) {
    unsigned int methodCount;
    Method *methodList = class_copyMethodList(currentClass, &methodCount);
    IMP lastImp = NULL;
    SEL lastSel = NULL;
    for (NSInteger i = 0; i < methodCount; i++) {
        Method method = methodList[i];
        NSString *methodName = [NSString stringWithCString:sel_getName(method_getName(method)) 
                                    encoding:NSUTF8StringEncoding];
        if ([@"printName" isEqualToString:methodName]) {
            lastImp = method_getImplementation(method);
            lastSel = method_getName(method);
        }
    }
    typedef void (*fn)(id,SEL);

    if (lastImp != NULL) {
        fn f = (fn)lastImp;
        f(my,lastSel);
    }
    free(methodList);
}

void *objc_destructInstance(id obj) 
{
    if (obj) {
        Class isa_gen = _object_getClass(obj);
        class_t *isa = newcls(isa_gen);

        // Read all of the flags at once for performance.
        bool cxx = hasCxxStructors(isa);
        bool assoc = !UseGC && _class_instancesHaveAssociatedObjects(isa_gen);

        // This order is important.
        if (cxx) object_cxxDestruct(obj);
        if (assoc) _object_remove_assocations(obj);

        if (!UseGC) objc_clear_deallocating(obj);
    }

    return obj;
}