很少有人知道weak表其實(shí)是一個(gè)hash（哈希）表诬辈，Key是所指對象的地址，Value是weak指針的地址數(shù)組菩佑。更多人的人只是知道weak是弱引用自晰，所引用對象的計(jì)數(shù)器不會加一，并在引用對象被釋放的時(shí)候自動(dòng)被設(shè)置為nil稍坯。通常用于解決循環(huán)引用問題酬荞。但現(xiàn)在單知道這些已經(jīng)不足以應(yīng)對面試了，好多公司會問weak的原理瞧哟。weak的原理是什么呢混巧？下面就分析一下weak的工作原理（只是自己對這個(gè)問題好奇，學(xué)習(xí)過程中的筆記勤揩，希望對讀者也有所幫助）咧党。

weak 實(shí)現(xiàn)原理的概括

Runtime維護(hù)了一個(gè)weak表，用于存儲指向某個(gè)對象的所有weak指針陨亡。weak表其實(shí)是一個(gè)hash（哈希）表傍衡，Key是所指對象的地址深员，Value是weak指針的地址（這個(gè)地址的值是所指對象指針的地址）數(shù)組。

weak 的實(shí)現(xiàn)原理可以概括一下三步：

1蛙埂、初始化時(shí)：runtime會調(diào)用objc_initWeak函數(shù)倦畅，初始化一個(gè)新的weak指針指向?qū)ο蟮牡刂贰?br> 2、添加引用時(shí)：objc_initWeak函數(shù)會調(diào)用 objc_storeWeak() 函數(shù)绣的， objc_storeWeak() 的作用是更新指針指向叠赐，創(chuàng)建對應(yīng)的弱引用表。
3屡江、釋放時(shí)芭概，調(diào)用clearDeallocating函數(shù)。clearDeallocating函數(shù)首先根據(jù)對象地址獲取所有weak指針地址的數(shù)組惩嘉，然后遍歷這個(gè)數(shù)組把其中的數(shù)據(jù)設(shè)為nil罢洲，最后把這個(gè)entry從weak表中刪除，最后清理對象的記錄宏怔。

下面將開始詳細(xì)介紹每一步：

1奏路、初始化時(shí)：runtime會調(diào)用objc_initWeak函數(shù)，objc_initWeak函數(shù)會初始化一個(gè)新的weak指針指向?qū)ο蟮牡刂贰?/h2>

示例代碼：

{
    NSObject *obj = [[NSObject alloc] init];
    id __weak obj1 = obj;
}

當(dāng)我們初始化一個(gè)weak變量時(shí)臊诊，runtime會調(diào)用 NSObject.mm 中的objc_initWeak函數(shù)鸽粉。這個(gè)函數(shù)在Clang中的聲明如下：

id objc_initWeak(id *object, id value);

而對于 objc_initWeak() 方法的實(shí)現(xiàn)

id objc_initWeak(id *location, id newObj) {

    if (!newObj) {
        *location = nil;
        return nil;
    }

    return storeWeakfalse, true, true>
    (location, (objc_object*)newObj);
}

可以看出，這個(gè)函數(shù)僅僅是一個(gè)深層函數(shù)的調(diào)用入口抓艳，而一般的入口函數(shù)中触机，都會做一些簡單的判斷（例如 objc_msgSend 中的緩存判斷），這里判斷了其指針指向的類對象是否有效玷或，無效直接釋放儡首，不再往深層調(diào)用函數(shù)。否則偏友，object將被注冊為一個(gè)指向value的__weak對象蔬胯。而這事應(yīng)該是objc_storeWeak函數(shù)干的。

注意：objc_initWeak函數(shù)有一個(gè)前提條件：就是object必須是一個(gè)沒有被注冊為__weak對象的有效指針位他。而value則可以是null氛濒，或者指向一個(gè)有效的對象。

2鹅髓、添加引用時(shí)：objc_initWeak函數(shù)會調(diào)用 objc_storeWeak() 函數(shù)舞竿， objc_storeWeak() 的作用是更新指針指向，創(chuàng)建對應(yīng)的弱引用表窿冯。

objc_storeWeak的函數(shù)聲明如下：

id objc_storeWeak(id *location, id value);

objc_storeWeak() 的具體實(shí)現(xiàn)如下：

template bool HaveOld, bool HaveNew, bool CrashIfDeallocating>
static id storeWeak(id *location, objc_object *newObj) {

    Class previouslyInitializedClass = nil;
    id oldObj;

    SideTable *oldTable;
    SideTable *newTable;

retry:
    if (HaveOld) {

        oldObj = *location;
        oldTable = &SideTables()[oldObj];
    } else {
        oldTable = nil;
    }
    if (HaveNew) {

        newTable = &SideTables()[newObj];
    } else {
        newTable = nil;
    }

    SideTable::lockTwoHaveOld, HaveNew>(oldTable, newTable);

    if (HaveOld  &&  *location != oldObj) {
        SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);
        goto retry;
    }

    if (HaveNew  &&  newObj) {

        Class cls = newObj->getIsa();

        if (cls != previouslyInitializedClass  &&
            !((objc_class *)cls)->isInitialized()) {

            SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);

            _class_initialize(_class_getNonMetaClass(cls, (id)newObj));

            previouslyInitializedClass = cls;

            goto retry;
        }
    }

    if (HaveOld) {
        weak_unregister_no_lock(&oldTable->weak_table, oldObj, location);
    }

    if (HaveNew) {
        newObj = (objc_object *)weak_register_no_lock(&newTable->weak_table,
                                                      (id)newObj, location,
                                                      CrashIfDeallocating);

        if (newObj  &&  !newObj->isTaggedPointer()) {

            newObj->setWeaklyReferenced_nolock();
        }

        *location = (id)newObj;
    }
    else {

    }
    SideTable::unlockTwoHaveOld, HaveNew>(oldTable, newTable);
    return (id)newObj;
}

撇開源碼中各種鎖操作骗奖，來看看這段代碼都做了些什么。

1)、SideTable

SideTable 這個(gè)結(jié)構(gòu)體执桌，我給他起名引用計(jì)數(shù)和弱引用依賴表鄙皇，因?yàn)樗饕糜诠芾韺ο蟮囊糜?jì)數(shù)和 weak 表。在 NSObject.mm 中聲明其數(shù)據(jù)結(jié)構(gòu)：

struct SideTable {

    spinlock_t slock;

    RefcountMap refcnts;

    weak_table_t weak_table;

}
對于 slock 和 refcnts 兩個(gè)成員不用多說鼻吮，第一個(gè)是為了防止競爭選擇的自旋鎖育苟，第二個(gè)是協(xié)助對象的 isa 指針的 extra_rc 共同引用計(jì)數(shù)的變量（對于對象結(jié)果较鼓，在今后的文中提到）椎木。這里主要看 weak 全局 hash 表的結(jié)構(gòu)與作用。

2)博烂、weak表

weak表是一個(gè)弱引用表香椎，實(shí)現(xiàn)為一個(gè)weak_table_t結(jié)構(gòu)體，存儲了某個(gè)對象相關(guān)的的所有的弱引用信息禽篱。其定義如下(具體定義在objc-weak.h中)：

   struct weak_table_t {

    weak_entry_t *weak_entries;

    size_t    num_entries;

    uintptr_t mask;

    uintptr_t max_hash_displacement;
};

這是一個(gè)全局弱引用hash表畜伐。使用不定類型對象的地址作為 key ，用 weak_entry_t 類型結(jié)構(gòu)體對象作為 value 躺率。其中的 weak_entries 成員玛界，從字面意思上看，即為弱引用表入口悼吱。其實(shí)現(xiàn)也是這樣的慎框。

其中weak_entry_t是存儲在弱引用表中的一個(gè)內(nèi)部結(jié)構(gòu)體，它負(fù)責(zé)維護(hù)和存儲指向一個(gè)對象的所有弱引用hash表后添。其定義如下：

typedef objc_object ** weak_referrer_t;
struct weak_entry_t {
    DisguisedPtrobjc_object> referent;
    union {
        struct {
            weak_referrer_t *referrers;
            uintptr_t        out_of_line : 1;
            uintptr_t        num_refs : PTR_MINUS_1;
            uintptr_t        mask;
            uintptr_t        max_hash_displacement;
        };
        struct {

            weak_referrer_t  inline_referrers[WEAK_INLINE_COUNT];
        };
    }
}

在 weak_entry_t 的結(jié)構(gòu)中笨枯，DisguisedPtr referent 是對泛型對象的指針做了一個(gè)封裝，通過這個(gè)泛型類來解決內(nèi)存泄漏的問題遇西。從注釋中寫 out_of_line 成員為最低有效位馅精，當(dāng)其為0的時(shí)候， weak_referrer_t 成員將擴(kuò)展為多行靜態(tài) hash table粱檀。其實(shí)其中的 weak_referrer_t 是二維 objc_object 的別名洲敢，通過一個(gè)二維指針地址偏移，用下標(biāo)作為 hash 的 key茄蚯，做成了一個(gè)弱引用散列压彭。
那么在有效位未生效的時(shí)候，out_of_line 第队、 num_refs哮塞、 mask 、 max_hash_displacement 有什么作用凳谦？以下是筆者自身的猜測：

out_of_line：最低有效位忆畅，也是標(biāo)志位。當(dāng)標(biāo)志位 0 時(shí)，增加引用表指針緯度家凯。
num_refs：引用數(shù)值缓醋。這里記錄弱引用表中引用有效數(shù)字，因?yàn)槿跻帽硎褂玫氖庆o態(tài) hash 結(jié)構(gòu)绊诲，所以需要使用變量來記錄數(shù)目送粱。
mask：計(jì)數(shù)輔助量。
max_hash_displacement：hash 元素上限閥值掂之。
其實(shí) out_of_line 的值通常情況下是等于零的抗俄，所以弱引用表總是一個(gè) objc_objective 指針二維數(shù)組。一維 objc_objective 指針可構(gòu)成一張弱引用散列表世舰，通過第三緯度實(shí)現(xiàn)了多張散列表动雹，并且表數(shù)量為 WEAK_INLINE_COUNT 。

總結(jié)一下 StripedMap[] ： StripedMap 是一個(gè)模板類跟压，在這個(gè)類中有一個(gè) array 成員胰蝠，用來存儲 PaddedT 對象，并且其中對于 [] 符的重載定義中震蒋，會返回這個(gè) PaddedT 的 value 成員茸塞，這個(gè) value 就是我們傳入的 T 泛型成員，也就是 SideTable 對象查剖。在 array 的下標(biāo)中钾虐，這里使用了 indexForPointer 方法通過位運(yùn)算計(jì)算下標(biāo)，實(shí)現(xiàn)了靜態(tài)的 Hash Table梗搅。而在 weak_table 中禾唁，其成員 weak_entry 會將傳入對象的地址加以封裝起來，并且其中也有訪問全局弱引用表的入口无切。

31B622B5-77ED-4D50-8CF9-0803785117BC.png

舊對象解除注冊操作 weak_unregister_no_lock

該方法主要作用是將舊對象在 weak_table 中接觸 weak 指針的對應(yīng)綁定荡短。根據(jù)函數(shù)名，稱之為解除注冊操作哆键。從源碼中掘托，可以知道其功能就是從 weak_table 中接觸 weak 指針的綁定。而其中的遍歷查詢籍嘹，就是針對于 weak_entry 中的多張弱引用散列表闪盔。

新對象添加注冊操作 weak_register_no_lock

這一步與上一步相反，通過 weak_register_no_lock 函數(shù)把心的對象進(jìn)行注冊操作辱士，完成與對應(yīng)的弱引用表進(jìn)行綁定操作泪掀。

初始化弱引用對象流程一覽

弱引用的初始化，從上文的分析中可以看出颂碘，主要的操作部分就在弱引用表的取鍵异赫、查詢散列、創(chuàng)建弱引用表等操作，可以總結(jié)出如下的流程圖：

C6029C92-5145-4334-9C25-3CDBA50F142B.png

這個(gè)圖中省略了很多情況的判斷塔拳，但是當(dāng)聲明一個(gè) __weak 會調(diào)用上圖中的這些方法鼠证。當(dāng)然， storeWeak 方法不僅僅用在 __weak 的聲明中靠抑，在 class 內(nèi)部的操作中也會常常通過該方法來對 weak 對象進(jìn)行操作量九。

3、釋放時(shí)颂碧，調(diào)用clearDeallocating函數(shù)荠列。clearDeallocating函數(shù)首先根據(jù)對象地址獲取所有weak指針地址的數(shù)組，然后遍歷這個(gè)數(shù)組把其中的數(shù)據(jù)設(shè)為nil稚伍，最后把這個(gè)entry從weak表中刪除弯予，最后清理對象的記錄。

當(dāng)weak引用指向的對象被釋放時(shí)个曙，又是如何去處理weak指針的呢？當(dāng)釋放對象時(shí)受楼，其基本流程如下：

1垦搬、調(diào)用objc_release
2、因?yàn)閷ο蟮囊糜?jì)數(shù)為0艳汽，所以執(zhí)行dealloc
3猴贰、在dealloc中，調(diào)用了_objc_rootDealloc函數(shù)
4河狐、在_objc_rootDealloc中米绕，調(diào)用了object_dispose函數(shù)
5、調(diào)用objc_destructInstance
6馋艺、最后調(diào)用objc_clear_deallocating

重點(diǎn)看對象被釋放時(shí)調(diào)用的objc_clear_deallocating函數(shù)栅干。該函數(shù)實(shí)現(xiàn)如下：

void  objc_clear_deallocating(id obj) 
{
    assert(obj);
    assert(!UseGC);
    if (obj->isTaggedPointer()) return;
   obj->clearDeallocating();
}

也就是調(diào)用了clearDeallocating，繼續(xù)追蹤可以發(fā)現(xiàn)捐祠，它最終是使用了迭代器來取weak表的value碱鳞，然后調(diào)用weak_clear_no_lock,然后查找對應(yīng)的value，將該weak指針置空踱蛀，weak_clear_no_lock函數(shù)的實(shí)現(xiàn)如下：

void weak_clear_no_lock(weak_table_t *weak_table, id referent_id)
{
    objc_object *referent = (objc_object *)referent_id;
    weak_entry_t *entry = weak_entry_for_referent(weak_table, referent);
    if (entry == nil) {

        return;
    }

    weak_referrer_t *referrers;
    size_t count;

    if (entry->out_of_line) {
        referrers = entry->referrers;
        count = TABLE_SIZE(entry);
    }
    else {
        referrers = entry->inline_referrers;
        count = WEAK_INLINE_COUNT;
    }

    for (size_t i = 0; i < count; ++i) {
        objc_object **referrer = referrers[i];
        if (referrer) {
            if (*referrer == referent) {
                *referrer = nil;
            }
            else if (*referrer) {
                _objc_inform("__weak variable at %p holds %p instead of %p. "
                             "This is probably incorrect use of "
                             "objc_storeWeak() and objc_loadWeak(). "
                             "Break on objc_weak_error to debug.\n",
                             referrer, (void*)*referrer, (void*)referent);
                objc_weak_error();
            }
        }
    }
    weak_entry_remove(weak_table, entry);
}

objc_clear_deallocating該函數(shù)的動(dòng)作如下：

1窿给、從weak表中獲取廢棄對象的地址為鍵值的記錄
2、將包含在記錄中的所有附有 weak修飾符變量的地址率拒，賦值為nil
3崩泡、將weak表中該記錄刪除
4、從引用計(jì)數(shù)表中刪除廢棄對象的地址為鍵值的記錄

看了objc-weak.mm的源碼就明白了：其實(shí)Weak表是一個(gè)hash（哈希）表猬膨，然后里面的key是指向?qū)ο蟮牡刂方亲玻琕alue是Weak指針的地址的數(shù)組。

補(bǔ)充：.m和.mm的區(qū)別

.m：源代碼文件，這個(gè)典型的源代碼文件擴(kuò)展名靴寂，可以包含OC和C代碼磷蜀。
.mm：源代碼文件，帶有這種擴(kuò)展名的源代碼文件百炬，除了可以包含OC和C代碼之外褐隆，還可以包含C++代碼。僅在你的OC代碼中確實(shí)需要使用C++類或者特性的時(shí)候才用這種擴(kuò)展名剖踊。

參考資料：
weak 弱引用的實(shí)現(xiàn)方式
weak的生命周期：具體實(shí)現(xiàn)方法

本文為轉(zhuǎn)載文章庶弃，原文出處