Linux 內核中的數據結構:基數樹(radix tree)


轉自:https://www.cnblogs.com/wuchanming/p/3824990.html


 

 

基數(radix)樹


Linux基數樹(radix tree)是將指針與long整數鍵值相關聯的機制,它存儲有效率,並且可快速查詢,用於指針與整數值的映射(如:IDR機制)、內存管理等。
IDR(ID Radix)機制是將對象的身份鑒別號整數值ID與對象指針建立關聯表,完成從ID與指針之間的相互轉換。IDR機制使用radix樹狀結構作為由id進行索引獲取指針的稀疏數組,通過使用位圖可以快速分配新的ID,IDR機制避免了使用固定尺寸的數組存放指針。IDR機制的API函數在lib/idr.c中實現,這里不加分析。
Linux radix樹最廣泛的用途是用於內存管理,結構address_space通過radix樹跟蹤綁定到地址映射上的核心頁,該radix樹允許內存管理代碼快速查找標識為dirty或writeback的頁。Linux radix樹的API函數在lib/radix-tree.c中實現。

 

(1)radix樹概述

 

radix樹是通用的字典類型數據結構,radix樹又稱為PAT位樹(Patricia Trie or crit bit tree)。Linux內核使用了數據類型unsigned long的固定長度輸入的版本。每級代表了輸入空間固定位數。
radix tree是一種多叉搜索樹,樹的葉子結點是實際的數據條目。每個結點有一個固定的、2^n指針指向子結點(每個指針稱為槽slot),並有一個指針指向父結點。

Linux內核利用radix樹在文件內偏移快速定位文件緩存頁,圖4是一個radix樹樣例,該radix樹的分叉為4(22),樹高為4,樹的每個葉子結點用來快速定位8位文件內偏移,可以定位4x4x4x4=256頁,如:圖中虛線對應的兩個葉子結點的路徑組成值0x00000010和0x11111010,指向文件內相應偏移所對應的緩存頁。 


圖4 一個四叉radix樹


Linux radix樹每個結點有64個slot,與數據類型long的位數相同,圖1顯示了一個有3級結點的radix樹,每個數據條目(item)可用3個6位的鍵值(key)進行索引,鍵值從左到右分別代表第1~3層結點位置。沒有孩子的結點在圖中不出現。因此,radix樹為稀疏樹提供了有效的存儲,代替固定尺寸數組提供了鍵值到指針的快速查找。 


圖1 一個3級結點的radix樹及其鍵值表示

 

(2)radix樹slot數

 

Linux內核根用戶配置將樹的slot數定義為4或6,即每個結點有16或64個slot,如圖2所示,當樹高為1時,64個slot對應64個頁,當樹高為2時,對應64*64個頁。 


圖2 高為1和2、slot數為64的radix樹

 

Linux內核radix樹的slot數定義如下(在lib/radix-tree.c中):


#ifdef __KERNEL__
/*值為6時,表示每個結點有2^6=64個slot,值為4時,表示有2^4=16個slot*/
#define RADIX_TREE_MAP_SHIFT (CONFIG_BASE_SMALL ? 4 : 6)
#else
#define RADIX_TREE_MAP_SHIFT 3 /* 用於更有強度的測試 */
#endif
/*表示1個葉子結點可映射的頁數,如:1<<6=64,表示可映射64個slot映射64頁*/
#define RADIX_TREE_MAP_SIZE (1UL << RADIX_TREE_MAP_SHIFT)
#define RADIX_TREE_MAP_MASK (RADIX_TREE_MAP_SIZE-1)
/*定義slot數占用的long類型長度個數,每個slot用位圖1位進行標記,如:64個slot時,值為2*/
#define RADIX_TREE_TAG_LONGS \
((RADIX_TREE_MAP_SIZE + BITS_PER_LONG - 1) / BITS_PER_LONG)

 

(3)結點結構

 

樹的根結點結構radix_tree_root列出如下(在include/linux/radix-tree.h中):
#define RADIX_TREE_MAX_TAGS 2 /*每個slot需要的最大標簽位數*/

/*根結點的標簽存放在gfp_mask中,通過__GFP_BITS_SHIFT移位得到 */
struct radix_tree_root {
    unsigned int height; /* 從葉子向上計算的樹高度 */
    gfp_t gfp_mask;
    /*間接指針指向結點而非數據條目,通過設置root->rnode的低位表示是否是間指針*/
    struct radix_tree_node *rnode;
#ifdef CONFIG_RADIX_TREE_CONCURRENT
    struct radix_tree_context *context;
    struct task_struct *owner;
#endif
};


結構radix_tree_context列出如下:
struct radix_tree_context {
    struct radix_tree_root *root;
#ifdef CONFIG_RADIX_TREE_CONCURRENT
    spinlock_t *locked;
#endif
}

 

樹的結點結構radix_tree_node列出如下(在lib/radix-tree.c中):

struct radix_tree_node {
    unsigned int height; /* 從葉子向上計算的樹高度 */
    unsigned int count; /*非葉子結點含有一個count域,表示出現在該結點的孩子的數量*/

    struct rcu_head rcu_head;
    void *slots[RADIX_TREE_MAP_SIZE]; /*結點的slot指針*/
   /* 結點標簽數組=每個slot需要的最大標簽位數*slot數所需的long類型變量數 */
   unsigned long tags[RADIX_TREE_MAX_TAGS][RADIX_TREE_TAG_LONGS];
};


Linux內核radix樹高度是可變的,它依賴於存儲的索引鍵值。radix樹查詢必須知道樹高度,以便知道結點的slot是指向葉子結點的指針還是結點的其他層。Linux在結點結構中有成員height存入樹高度信息。
Linux radix樹的獨特特征是標簽擴展。圖2顯示了有2個標簽,分別表示打開和關閉。這些標簽是結點結構的成員tags,該標簽用一個位圖實現,在加鎖的情況下設置和清除。每個標簽基本上是一個在radix樹頂層上的位圖序號。標簽與群查詢一起用來在給定的范圍內查找給定標簽集的頁。
標簽在每結點位圖中維護,在每個給定的級別上,可以判定下一個級別是否至少有一個標簽集。

 
圖2 帶有2位圖標簽指示的8位radix樹


在結點結構radix_tree_node中,tags域是一個二維數組,每個slot用2位標識,這是一個典型的用空間換時間的應用。tags域用於記錄該結點下面的子結點有沒有相應的標志位。目前RADIX_TREE_MAX_TAGS為2,表示只記錄兩個標志,其中tags[0]為PAGE_CACHE_DIRTY,tags[1]為PAGE_CACHE_WRITEBACK。如果當前節點的tags[0]值為1,那么它的子樹節點就存在PAGE_CACHE_DIRTY節點,否則這個子樹分枝就不存在着這樣的節點,就不必再查找這個子樹了。比如在查找PG_dirty的頁面時,就不需要遍歷整個樹,而可以跳過那些tags[0]為0值的子樹,這樣就提高了查找效率。
“加標簽查詢”是返回有指定標簽集radix樹條目的查詢,可以查詢設置了標簽的條目。加標簽查詢可以並行執行,但它們需要通過設置或清除標簽從操作上互斥。


(4) 並行操作的優化


Linux radix樹並行操作包括並行查詢和並行修改,其中,並行修改在標准內核中未完沒有實現,需要通過打補丁獲得該功能。並行操作說明如下:

 

  • RCU並發查詢

 

通過使用RCU,RCU Radix樹可以進行完全並發的查詢操作。RCU從根本上要求原子操作地移動指針從數據結構的一個版本到新的版本,保持舊版本直到系統經過靜止狀態。在靜止狀態點,舊版本數據結構已沒有用戶,因此可以被安全釋放。
RCU radix樹的修改操作之間還需要串行化,但是查詢不再需要與修改操作串行化。

 

  • 並發修改

 

RCU可使RCU radix樹查詢完全並行化,但修改操作成了“瓶頸”。這可通過將全樹的鎖破碎成較小的鎖進行改善,再明顯的方法是對結點進行加鎖而非對整個樹加鎖。
radix樹修改操作可分為單向和雙向操作。單向操作僅執行從根節點和葉子結點的單方向指針移動,它包括插入、更新和設置標簽操作。雙向操作較復雜,它需要在指針移到葉子后又回移,它包括刪除和清除標簽操作。
梯級加鎖(Ladder Locking)和鎖耦合(Lock-Coupling)技術常用於數據庫方面,允許單向遍歷結點加鎖的樹(雙向可能產生死鎖)。如果所有的修改者從樹頂到樹底進行修改,並且修改的結點持有鎖,那么,向下遍歷時對孩子加鎖,在孩子被鎖住時再釋放該結點鎖。在這種情況下並發操作是可能的,因為只要根結點解鎖,另一個操作就可以自上向下進行。如果兩操作的路徑沒有相同操作結點,后一個操作可能在前一個操作完成之前完成。最壞的情況是流水線操作,但這還是比串行化操作好很多。


雙向操作包括刪除和清除標簽操作,分別說明如下:

 

1)清除標簽

 

在radix樹中清除一個標簽包括向下遍歷樹、查找定位條目和清除條目標簽的操作。只要孩子結點沒有打標簽的條目,就可以向上遍歷結點清除標簽。結束條件是:如果遍歷遇到一個結點,在清除一個標簽后,它還有一個或多個條目帶有標簽集,就可以結束向上遍歷。為了與向下遍歷期間有同樣的結束點,將終止條件改為:向上遍歷將在有比清除標簽數更多標簽的結點處結束。這樣,不論何時遇到這樣的結點,將作為上遍歷樹的結束點。


2)刪除元素


刪除元素在刪除無用結點時還需要刪除該條目的所有標簽。它的終止條件需要滿足這兩個方面。向上回退遍歷樹時需要滿足下面的條件:當遇到一個非空結點且沒有無用的標簽時應終止向上回退遍歷樹。
在向下遍歷樹時鑒別此點的條件是:當遇到有超過2個孩子的結點、並且每個標簽來說結點有多於一個標簽條目被清除時,結束向上遍歷。該條件用來鑒別向上回退遍歷的終止點。

 

(5)radix樹API說明

 

  • 聲明和初始化radix樹

聲明和初始化radix樹的方法列出如下:


#include <linux/radix-tree.h>
/* gfp_mask表示如何執行內存分配,如果操作(如:插入)以原子性上下文中執行,其值為GFP_ATOMIC*/
RADIX_TREE(name, gfp_mask); /* 聲明和初始化名為name的樹*/

struct radix_tree_root my_tree;
INIT_RADIX_TREE(my_tree, gfp_mask);

 

  • 插入條目


插入條目的函數定義列出如下:

int radix_tree_insert(struct radix_tree_root *root, unsigned long index, void *item)

 

函數radix_tree_insert插入條目item到樹root中,如果插入條目中內存分配錯誤,將返回錯誤-ENOMEM。該函數不能覆蓋寫正存在的條目。如果索引鍵值index已存在於樹中,返回錯誤-EEXIST。插入操作成功是,返回0。
對於插入條目操作失敗將引起嚴重問題的場合,下面的一對函數可避免插入操作失敗:

int radix_tree_preload(gfp_t gfp_mask);
void radix_tree_preload_end(void);

 

函數radix_tree_preload嘗試用給定的gfp_mask分配足夠的內存,保證下一個插入操作不會失敗。在調用插入操作函數之前調用此函數,分配的結構將存放在每CPU變量中。函數radix_tree_preload操作成功后,將完畢內核搶占。因此,在插入操作完成之后,用戶應調用函數radix_tree_preload_end打開內核搶占。

 

  • 刪除條目


刪除條目的函數定義列出如下:

void *radix_tree_delete(struct radix_tree_root *root, unsigned long index)

 

函數radix_tree_delete刪除與索引鍵值index相關的條目,如果刪除條目在樹中,返回該條目的指針,否則返回NULL。

 

  • 查詢操作

 

用於查詢操作的函數定義列出如下:

/*在樹中查找指定鍵值的條目,查找成功,返回該條目的指針,否則,返回NULL*/
void *radix_tree_lookup(struct radix_tree_root *root, unsigned long index);
/*返回指向slot的指針,該slot含有指向查找到條目的指針*/
void **radix_tree_lookup_slot(struct radix_tree_root *root, unsigned long index);
/*查詢返回max_items條目在results中。查詢時鍵值索引從first_index開始*/
radix_tree_gang_lookup(struct radix_tree_root *root, void **results, unsigned long first_index, unsigned int max_items);

 

  • 標簽操作


與標簽操作相關的函數說明列出如下:

/*將鍵值index對應的條目設置標簽tag,返回值為設置標簽的條目*/
void *radix_tree_tag_set(struct radix_tree_root *root, unsigned long index, unsigned int tag);
/*從鍵值index對應的條目清除標簽tag,返回值為清除標簽的條目*/
void *radix_tree_tag_clear(struct radix_tree_root *root, unsigned long index, unsigned int tag);
/*檢查鍵值index對應的條目是否為標簽tag,如果鍵值不存在,返回0,如果鍵值存在,但標簽未設置,返回-1;如果鍵值存在,且標簽已設置,返回1*/
int radix_tree_tag_get(struct radix_tree_root *root, unsigned long index, unsigned int tag);
/*從first_index起查詢樹root中標簽值為tag的條目,在results中返回*/
unsigned int radix_tree_gang_lookup_tag(struct radix_tree_root *root, void **results, unsigned long first_index, unsigned int max_items, unsigned int tag);
/*如果樹root中有任何條目使用tag標簽,返回鍵值*/
int radix_tree_tagged(struct radix_tree_root *root, unsigned int tag);

 

(6)並行操作使用radix樹API的方法

 

  • 查詢獲取slot操作

 

查詢操作支持RCU無阻塞並行讀操作,因此,需要遵循RCU的用法加RCU讀鎖,還需要將rcu_dereference()用於獲得的slot,在寫(或更新)操作時,需要給新的slot使用rcu_assign_pointer()。查詢操作的使用方法列出如下:

struct page **slot, *page;
rcu_read_lock();
slot = radix_tree_lookup_slot(&mapping->page_tree, index);
page = rcu_dereference(*slot);
rcu_read_unlock();

 

  • 查詢修改slot操作

 

Linux內核的radix樹需要打補丁才支持並發修改。查詢僅有一個全局狀態:RCU靜止狀態,並發修改需要跟蹤持有什么鎖。鎖狀態對於操作來說必須是外部的,因此,我們需要實例化一個本地上下文跟蹤這些鎖。查詢修改slot的方法列出如下:

struct page **slot;
DEFINE_RADIX_TREE_CONTEXT(ctx,&mapping->page_tree);
radix_tree_lock(&ctx); /*鎖住了根結點*/
/* ctx.tree代替&mapping->page_tree作為根,可以傳遞上下文
slot = radix_tree_lookup_slot(tx.tree, index);
rcu_assign_pointer(*slot, new_page);
radix_tree_unlock(&ctx);

 

radix樹API函數radix_tree_lookup_slot含有鎖從樹頂向下移動機制,鎖移動的代碼部分列出如下:

void **radix_tree_lookup_slot(struct radix_tree *root, unsigned long index)
{
    ...
    RADIX_TREE_CONTEXT(context, root); /*提供上下文和實際的root指針*、
    ...
    do {
        ...
        /* 從樹頂向下移動鎖*/
        radix_ladder_lock(context, node);
        ...
    } while (height > 0);
    ...
}

 

(7)radix樹API的實現

 

樹的操作通常包括查找、插入、刪除和樹調整,下面分別說明radix樹這些操作的實現。

 

  • 查找單個條目

 

radix樹通過索引鍵值進行查詢,如圖1所示,它按路徑組成索引鍵值,圖中3級結點對應3段6位表示的索引值,圖中key對應的葉子slot的索引鍵值為“2,63,1”。通過葉子slot的指針slot[1]就可得到所存儲的數據item。因此,查詢迭代時,需要將索引鍵值“2,63,1”移去高2層“2,63”,得到葉子slot的指針索引鍵值“1”。

 
圖1 一個3級結點的radix樹及其鍵值表示

 

函數radix_tree_lookup執行查找操作,查找方法是:從葉子到樹頂,通過數組索引鍵值值查看數組元素的方法,一層層地查找slot。其列出如下(在lib/radix-tree.c中):

void *radix_tree_lookup(struct radix_tree_root *root, unsigned long index)
{
    unsigned int height, shift;
    struct radix_tree_node *node, **slot;

    node = rcu_dereference(root->rnode); /*獲取根結點*/
    if (node == NULL)
        return NULL;

 

    /*間接指針指向結點而非數據條目,通過設置root->rnode的低位表示是否是間指針。對於間接指針來說,樹高度值root->height大於0,但是RCU查找需要測試間接指針,因為root->height 值不可靠。這種問題僅的RCU下需要考慮*/ 
    if (!radix_tree_is_indirect_ptr(node)) { /*非間接指針,說明只有根結點*/
        if (index > 0)
            return NULL;
        return node;
    }

 

    /*獲取真正結點指針,因為根結點指針的第0位設置為1表示為間接指針。當使用結點指針時,必須將第0位設置為0,因為地址以字對齊*/
    node = radix_tree_indirect_to_ptr(node);

    height = node->height;
    if (index > radix_tree_maxindex(height)) /*索引鍵值不能超過最大索引值*/
        return NULL;

 

    /*每層索引偏移值為RADIX_TREE_MAP_SHIFT,葉子索引值偏移基數為(樹高-1)*每層索引偏移值*/ 
    shift = (height-1) * RADIX_TREE_MAP_SHIFT;

    do { /*從葉子到樹頂,通過樹路徑組成的索引查找指定索引鍵值的slot*/
        slot = (struct radix_tree_node **)(node->slots + ((index>>shift) & RADIX_TREE_MAP_MASK)); /*如:slots +1*/
        node = rcu_dereference(*slot);
        if (node == NULL)
            return NULL;

        shift -= RADIX_TREE_MAP_SHIFT; /*向上移一層,再迭代查找*/
        height--;
    } while (height > 0);

   return node;
}

 

  • 查找多個條目

 

函數radix_tree_gang_lookup執行多個索引鍵值的查找,其列出如下:

unsigned int radix_tree_gang_lookup(struct radix_tree_root *root, void **results, unsigned long first_index, unsigned int max_items)
{
    unsigned long max_index;
    struct radix_tree_node *node;
    unsigned long cur_index = first_index;
    unsigned int ret;

    node = rcu_dereference(root->rnode);
    if (!node)
        return 0;

    if (!radix_tree_is_indirect_ptr(node)) { /*如果為非間接指針,表示只有根節點*/
        if (first_index > 0)
            return 0;
        results[0] = node;
        return 1;
    }
    node = radix_tree_indirect_to_ptr(node); /*清除用於間接指針標識的第0位*/

    max_index = radix_tree_maxindex(node->height); /*獲取樹的最大索引鍵值*/

    ret = 0;
    while (ret < max_items) { /* max_items為查找的最大條目數*/
        unsigned int nr_found;
        unsigned long next_index; /* 下一個搜索的索引鍵值*/

        if (cur_index > max_index) /*已查詢完所需查詢的索引鍵值*/
            break;
        nr_found = __lookup(node, results + ret, cur_index,
        max_items - ret, &next_index);
        ret += nr_found;
        if (next_index == 0)
            break;
        cur_index = next_index;
    }

    return ret;
}

 

static unsigned int
__lookup(struct radix_tree_node *slot, void **results, unsigned long index, unsigned int max_items, unsigned long *next_index)
{
    unsigned int nr_found = 0;
    unsigned int shift, height;
    unsigned long i;

    height = slot->height;
    if (height == 0)
        goto out;
    /*所有葉子slot的索引鍵值基數偏移*/
    shift = (height-1) * RADIX_TREE_MAP_SHIFT; 

    /*從底層向樹頂層,
    for ( ; height > 1; height--) { /*從葉子向樹頂查找*/
        i = (index >> shift) & RADIX_TREE_MAP_MASK;
        for (;;) { /*遍歷每一層的各個路徑,由樹頂到當前層一條路徑組成索引鍵值*/
            /*如果slot不為空,那么它掛有子結點,跳出本循環,進入子結點層進行本循環*/
            if (slot->slots[i] != NULL)
                break;
           /*如果slot為空,就跳過slot對應的所有索引鍵值*/ 
           /*清除索引號低位.將索引號與該層slot的起始索引號對齊*/
           index &= ~((1UL << shift) - 1);
           /*跳過一個slot的索引鍵值數*/
          index += 1UL << shift;
          if (index == 0)
              goto out; /* 32-bit wraparound */
          i++; /*找到多個slot*/
          if (i == RADIX_TREE_MAP_SIZE)
              goto out;
       }

       shift -= RADIX_TREE_MAP_SHIFT; /*向上移一層,基數偏移減少*/
       slot = rcu_dereference(slot->slots[i]);
       if (slot == NULL)
           goto out;

    }

    /* 返回找到的多個slot*/
    for (i = index & RADIX_TREE_MAP_MASK; i < RADIX_TREE_MAP_SIZE; i++) {
        struct radix_tree_node *node;
        index++;
        node = slot->slots[i];
        if (node) {
            results[nr_found++] = rcu_dereference(node);
           if (nr_found == max_items)
               goto out;
        }
    }
    out:
    *next_index = index;
    return nr_found;
}

 

  • 插入條目


函數radix_tree_insert找到索引鍵值對應的結點,將item加到該結點的slot指針上。其列出如下:

int radix_tree_insert(struct radix_tree_root *root, unsigned long index, void *item)
{
    struct radix_tree_node *node = NULL, *slot;
    unsigned int height, shift;
    int offset;
    int error;

    BUG_ON(radix_tree_is_indirect_ptr(item));

    /* 如果樹的高度不夠,就擴展樹。函數radix_tree_maxindex計算樹高容納的最大索引*/
    if (index > radix_tree_maxindex(root->height)) {
        error = radix_tree_extend(root, index);
        if (error)
            return error;
    }

    slot = radix_tree_indirect_to_ptr(root->rnode);

    height = root->height;
    shift = (height-1) * RADIX_TREE_MAP_SHIFT; /*計算偏移基數*/

    offset = 0; 
    while (height > 0) {
        if (slot == NULL) { /*如果slot為空,則需要加入孩子結點*/
            /* 分配slot */
            if (!(slot = radix_tree_node_alloc(root)))
                return -ENOMEM;
            slot->height = height;
            if (node) {/*添加slot*/
                rcu_assign_pointer(node->slots[offset], slot);
             node->count++;
         } else
             rcu_assign_pointer(root->rnode,radix_tree_ptr_to_indirect(slot));
        }

        /* 進入上一層*/
        offset = (index >> shift) & RADIX_TREE_MAP_MASK;
        node = slot;
        slot = node->slots[offset];
        shift -= RADIX_TREE_MAP_SHIFT;
        height--;
    }
    /*如果index對應的slot已有映射頁面,返回-EEXIST*/
    if (slot != NULL)
        return -EEXIST;

    if (node) {
        node->count++; /*增加子結點的計數*/
        rcu_assign_pointer(node->slots[offset], item);
        BUG_ON(tag_get(node, 0, offset));
        BUG_ON(tag_get(node, 1, offset));
    } else { /*為頂層結點*/
        rcu_assign_pointer(root->rnode, item);
        BUG_ON(root_tag_get(root, 0));
        BUG_ON(root_tag_get(root, 1));
    }

    return 0;
}

 

  • 擴展樹的高度


如果當前樹高度不足以存放index,就需要擴展樹,擴展方法是在舊樹頂上加新的根結點,並將原根結點的tag信息移到新根結點的第1個slot。函數radix_tree_extend 列出如下:
static int radix_tree_extend(struct radix_tree_root *root, unsigned long index)
{
    struct radix_tree_node *node;
    unsigned int height;
    int tag;

    /* 計算擴展的深度*/
    height = root->height + 1;
    /*如果index超過樹高所容納的最大索引值,樹高遞增*/
    while (index > radix_tree_maxindex(height)) 
        height++;

    /*到這里已計算好合適的高度height*/ 
    if (root->rnode == NULL) { /*只有根結點,設置好樹高度就可返回*/
        root->height = height;
        goto out;
    }

    /*將當前樹擴展到高度height*/
    do {
        unsigned int newheight;
        if (!(node = radix_tree_node_alloc(root))) /*分配一個結點*/
            return -ENOMEM;

        /* 增加樹高,在樹頂點之上增加一個結點*/
        node->slots[0] = radix_tree_indirect_to_ptr(root->rnode);

        /*傳播tag信息到新根結點*/
        /*以前的根結點現成為新頂結點的第1個插槽。 如果以前的根結點打上了tag,就將新增結點的第1個插槽對應的子節點打上相應的tag*/
        for (tag = 0; tag < RADIX_TREE_MAX_TAGS; tag++) {
            if (root_tag_get(root, tag))
               tag_set(node, tag, 0);
        }

        newheight = root->height+1;
        node->height = newheight;
        node->count = 1;
        node = radix_tree_ptr_to_indirect(node);
        rcu_assign_pointer(root->rnode, node);
        root->height = newheight;
    } while (height > root->height);
out:
    return 0;
}

 

  • 刪除條目


函數radix_tree_delete刪除index對應的條目,並返回刪除條目的地址。其列出如下:
void *radix_tree_delete(struct radix_tree_root *root, unsigned long index)
{
    /*數組path用於保存路徑上的結點及索引偏移值*/
    struct radix_tree_path path[RADIX_TREE_MAX_PATH + 1], *pathp = path;
    struct radix_tree_node *slot = NULL;
    struct radix_tree_node *to_free;
    unsigned int height, shift;
    int tag;
    int offset;

    height = root->height;
    if (index > radix_tree_maxindex(height)) /*index不能超過樹的最大索引值*/
        goto out;

    slot = root->rnode;
    if (height == 0) { /*只有根結點*/
        root_tag_clear_all(root);
        root->rnode = NULL;
        goto out;
    }
    slot = radix_tree_indirect_to_ptr(slot);

    shift = (height - 1) * RADIX_TREE_MAP_SHIFT;
    pathp->node = NULL;
    /*獲取刪除條目的路徑*/
    /*將index從根到葉子的路徑所經過的結點及相應索引偏移值存放在數組pathp中*/
    do {
        if (slot == NULL)
            goto out;

        pathp++;
        offset = (index >> shift) & RADIX_TREE_MAP_MASK;
        pathp->offset = offset;
        pathp->node = slot;
        slot = slot->slots[offset];
        shift -= RADIX_TREE_MAP_SHIFT;
        height--;
    } while (height > 0);

    if (slot == NULL)
        goto out;

    /*清除與刪除條目相關的所有標簽*/
    for (tag = 0; tag < RADIX_TREE_MAX_TAGS; tag++) {
        if (tag_get(pathp->node, tag, pathp->offset))
            radix_tree_tag_clear(root, index, tag);
    }

    to_free = NULL;
    /*釋放不再需要的結點 */
    while (pathp->node) { /*刪除
        pathp->node->slots[pathp->offset] = NULL; /*清除slot*/
        pathp->node->count--; /*孩子數量減1*/
        /*調用RCU機制的函數call_rcu在最后一個引用結束時延遲釋放結點to_free */
        if (to_free)
            radix_tree_node_free(to_free);

        if (pathp->node->count) { /*還有孩子存在*/
           if (pathp->node == radix_tree_indirect_to_ptr(root->rnode)) /*為根結點的孩子*/
                radix_tree_shrink(root); /*樹縮小*/
           goto out;
        }

        /*釋放有0個slot的結點 */
        to_free = pathp->node;
        pathp--;

    }
    /*運行到這里,說明是根結點*/
    root_tag_clear_all(root);
    root->height = 0;
    root->rnode = NULL;
    if (to_free)
        radix_tree_node_free(to_free);

out:
    return slot;
}

 

  • 縮小樹的高度


函數radix_tree_shrink縮小樹的高度到最小。其列出如下:

static inline void radix_tree_shrink(struct radix_tree_root *root)
{
    /* 嘗試縮小樹的高度*/
    while (root->height > 0) {
        struct radix_tree_node *to_free = root->rnode;
        void *newptr;

        BUG_ON(!radix_tree_is_indirect_ptr(to_free));
        to_free = radix_tree_indirect_to_ptr(to_free);

        /*候選結點多於一個孩子或孩子不在最左邊slot,不能縮小樹的高度,跳出循環*/
        if (to_free->count != 1)
            break;
        if (!to_free->slots[0])
            break;

        /*不需要調用rcu_assign_pointer(),因為僅從樹的一部分到另一部分移動結點。如果釋放舊指針的引用to_free->slots[0]是安全的,那么釋放新指針的引用root->rnode也是安全的*/ 
       newptr = to_free->slots[0];
       if (root->height > 1)
           newptr = radix_tree_ptr_to_indirect(newptr);
       root->rnode = newptr;
       root->height--; 
       /* 僅釋放0結點*/
       tag_clear(to_free, 0, 0);
       tag_clear(to_free, 1, 0);
       to_free->slots[0] = NULL;
       to_free->count = 0;
       radix_tree_node_free(to_free);
    }
}

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM