讀讀objc源碼(二):weak類型指針的實現

weak指針這部分代碼寫的很好啊,結構清晰,接口定義到關鍵位置,讀取來很舒服。

整體結構

weak指針管理結構.png
  • SideTable包含了引用計數表和weak指針表,大概就是內存管理的總表,SideTable有多張,對象根據內存地址會關聯(lián)上某一張
  • weak_table_t 包含了所有具有weak指針的對象的weak指針信息
  • weak_entry_t 對應某一個對象,一個對象可能有多個weak指針,它們作為一個整體存放在這里
  • weak_entry_t包含兩部分,一個是對象的內存地址,這個相當于key/id的作用,用來識別是對應哪個對象的;另一部分就是指向這個對象的所有weak指針。

整體的邏輯就是:
使用hash表把對象和所有指向它的weak指針關聯(lián)起來,等這個對象dealloc的時候,把這些weak指針拿出來,全部設置成nil。

SideTable

weak指針原理-sideTable.png

We cannot use a C++ static initializer to initialize SideTables because
libc calls us before our C++ initializers run.

代碼注釋里有句話,所以這就是為什么用靜態(tài)內存+指針強轉來構建SideTable的原因吧,要足夠早。

SideTable是用StripedMap包裝了的,StripedMap的作用,看它的讀取方法:

T& operator[] (const void *p) { 
        return array[indexForPointer(p)].value; 
    }

它重載了中括號[],從array里把值取出來,關鍵就是indexForPointer這個函數,它完成從指針到索引的轉換:

    static unsigned int indexForPointer(const void *p) {
        uintptr_t addr = reinterpret_cast<uintptr_t>(p);
        return ((addr >> 4) ^ (addr >> 9)) % StripeCount;
    }

所以它其實是一個hash函數,根據指針的值,也就是指向內存的地址,轉化成落在[0, StripeCount]范圍內的一個unsigned int值。

整體來看,對一個對象,獲取它的SideTable,就是把這個對象的地址轉化成了一個[0, StripeCount]范圍內的索引,在拿到這個索引的SideTable。

weak_table_t和weak_entry_t單看結構沒什么特別的,在使用的時候再看。

weak指針的使用

3中情況:

  1. weakA = weakB
  2. weakA = strongB
  3. strongA = weakB

情況1和2都是調用了id objc_storeWeak(id *location, id newObj),情況3走的是id objc_loadWeakRetained(id *location),而objc_loadWeakRetained實際就是把weak對象retain了一下,屬于另外的問題了。

還有一種情況,定義一個weak指針的時候:__weak TFBook *weakBook = nil;,這個也是走了id objc_storeWeak(id *location, id newObj)

所以objc_storeWeak是核心的核心。

怎么看調用什么方法?猥瑣一點,搞個while循環(huán),在里面寫想查看的方法,然后用instrument工具里的Time Profiler看占掉cpu 100%的那個就是了!

objc_storeWeak

template <bool HaveOld, bool HaveNew, bool CrashIfDeallocating>
static id 
storeWeak(id *location, objc_object *newObj)
  • HaveOld 是否有就對象,weakA = weakB,如果weakA之前是nil,那HaveOld就是false.
  • HaveNew 是否新對象
  • 這個操作處在deallocing調用過程中是否奔潰
  • location是指向weak指針的指針,因為要修改weak指針
  • newObj新對象

它的作用就是解除舊對象關系,和新對象建立聯(lián)系。

storeWeak函數.png
weak_unregister_no_lock:
.....
if ((entry = weak_entry_for_referent(weak_table, referent))) {
        remove_referrer(entry, referrer);
.....
if (empty) {
            weak_entry_remove(weak_table, entry);
        }
.....

取出entry,移除referrer,referrer是weak指針的引用,這里的weak_table是舊表,舊表里移除weak指針,就是解除了久對象和weak指針的關系。

如果這個empty空了,就從table里去掉。

  • weak_entry_for_referent

    size_t begin = hash_pointer(referent) & weak_table->mask;
    ...
    while (weak_table->weak_entries[index].referent != referent) {
          index = (index+1) & weak_table->mask;
          if (index == begin) bad_weak_table(weak_table->weak_entries);
          hash_displacement++;
          if (hash_displacement > weak_table->max_hash_displacement) {
              return nil;
          }
      }
    

    從weak_table_t里面取出entry,用了hash表的邏輯:

    • hash_pointer也是使用指針地址,映射到一個索引。&weak_table->mask這個操作是?這個mask實際值是表的size-1,而size是2的n次方方式擴張的,所以mask的形式就1111 1111 1111這種,索引和mask位與之后的值必定就落在了[0, size]范圍內。簡潔高效,牛逼!
    • index都取到了,為什么還要while循環(huán)?因為hash函數也會重合的,如果index1的位置已經有人占了,又來一個人要占index1怎么辦?往后挪,直到找到一個空位置。所以hash函數得到的index和實際位置有那么一點的偏差。
    • hash_displacement是在存入數據的時候記錄了最大的偏差值,有這個做把控,偏移超過了這個值肯定是沒有了。
  • remove_referrer

if (! entry->out_of_line()) {
       for (size_t i = 0; i < WEAK_INLINE_COUNT; i++) {
           if (entry->inline_referrers[i] == old_referrer) {
               entry->inline_referrers[i] = nil;
               return;
           }
       }
.....
size_t begin = w_hash_pointer(old_referrer) & (entry->mask);
....
while (entry->referrers[index] != old_referrer) {
       index = (index+1) & entry->mask;
       if (index == begin) bad_weak_table(entry);
       hash_displacement++;
       if (hash_displacement > entry->max_hash_displacement) {
          .....
           objc_weak_error();
           return;
       }
   }

weak_entry_t有個奇怪的地方就是里面有個union:

union {
        struct {
            weak_referrer_t *referrers;
            uintptr_t        out_of_line_ness : 2;
            uintptr_t        num_refs : PTR_MINUS_2;
            uintptr_t        mask;
            uintptr_t        max_hash_displacement;
        };
        struct {
            // out_of_line_ness field is low bits of inline_referrers[1]
            weak_referrer_t  inline_referrers[WEAK_INLINE_COUNT];
        };
    };

這兩個東西都是用來存儲指向這個對象的所有weak指針的,但是是不同時期使用的,到weak指針在4(WEAK_INLINE_COUNT)個以內的時候,用數組inline_referrers,超過用weak_referrer_t,這個還是hash表。

我的理解是這是為了性能考慮。一般情況,就一兩個weak指針會指向同一個對象,用數組管理,存取快。但是也得允許N多weak指針指向同一個對象,WEAK_INLINE_COUNT不可能無限大。感受到了一點空時間、分階段處理的思想。

weak_referrer_t的存取跟上面weak_table_t一樣。

out_of_line是用來判斷是否超過數組個數的,就是它用來做兩種方案的切換:

    bool out_of_line() {
        return (out_of_line_ness == REFERRERS_OUT_OF_LINE);
    }

out_of_line_ness是否被設置了REFERRERS_OUT_OF_LINE這個標識。這個標識的值實際是2。注釋里有一段話:

// out_of_line_ness field overlaps with the low two bits of inline_referrers[1].
// inline_referrers[1] is a DisguisedPtr of a pointer-aligned address.
// The low two bits of a pointer-aligned DisguisedPtr will always be 0b00
// (disguised nil or 0x80..00) or 0b11 (any other address).
// Therefore out_of_line_ness == 0b10 is used to mark the out-of-line state.

因為union的關系,out_of_line_ness的內存位置對應的就是數組inline_referrers里第二個(weak_referrer_t和weak_referrer_t *都是8個字節(jié))。根據這段注釋,weak_referrer_t的數據的二進制結尾要么是00要么是11,不會是10,所以用10來做標識。

如果只使用inline_referrers,那么out_of_line_ness讀取出來就要么是00要么是11,所以如果讀出來是10,也就是十進制2,就是使用hash表的referrers。

我沒搞懂的是為什么weak_referrer_t的結尾不會是10。

weak_register_no_lock

這個函數和weak_unregister_no_lock幾乎就是反操作了:

weak_entry_t *entry;
    if ((entry = weak_entry_for_referent(weak_table, referent))) {
        append_referrer(entry, referrer);
    } 
    else {
        weak_entry_t new_entry(referent, referrer);
        weak_grow_maybe(weak_table);
        weak_entry_insert(weak_table, &new_entry);
    }
  • weak_grow_maybe+weak_entry_insert對應weak_entry_remove
  • append_referrer對應remove_referrer

總結

  • 使用hash表把對象和所有指向它的weak指針關聯(lián)起來,等這個對象dealloc的時候,把這些weak指針拿出來,全部設置成nil。
  • 3層表:side table+weak table--->weak entry---> referrers + inline_referrers
  • hash表的使用邏輯
  • referrers和inline_referrers的切換
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,646評論 6 533
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 98,595評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,560評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,035評論 1 314
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,814評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,224評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,301評論 3 442
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,444評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體,經...
    沈念sama閱讀 48,988評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,804評論 3 355
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,998評論 1 370
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,544評論 5 360
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,237評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,665評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,927評論 1 287
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,706評論 3 393
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,993評論 2 374

推薦閱讀更多精彩內容