摘要:雙向鏈表Doubly Linked List是鴻蒙輕內核最重要的數據結構之一,在各個模塊有着非常廣泛的使用。
在學習OpenHarmony鴻蒙輕內核源代碼的時候,常常會遇到一些數據結構的使用。如果沒有掌握它們的用法,會導致閱讀源代碼時很費解、很吃力。本文會給讀者介紹源碼中重要的數據結構,雙向循環鏈表Doubly Linked List。在講解時,會結合數據結構相關繪圖,培養讀者們的數據結構的平面想象能力,幫助更好的學習和理解這些數據結構的用法。
本文中所涉及的源碼,以OpenHarmony LiteOS-M內核為例,均可以在開源站點https://gitee.com/openharmony/kernel_liteos_m 獲取。
1 雙向循環鏈表
雙向鏈表LOS_DL_LIST的源代碼在utils\los_list.h雙向鏈表頭文件中,包含LOS_DL_LIST結構體定義、inline內聯函數LOS_ListXXX,還有相關的函數宏定義LOS_DL_LIST_XXXX。雙向鏈表頭文件可以網頁訪問utils/los_list.h,也可以檢出到本地閱讀。
1.1 雙向鏈表結構體
雙向鏈表節點結構體LOS_DL_LIST定義如下。其結構非常簡單、通用、抽象,只包含前驅、后繼兩個節點,負責承上啟下的雙向鏈表作用。雙向鏈表不包含任何業務數據信息,一般不會單獨使用。通常,雙向鏈表節點和業務數據信息作為結構體成員,一起組成業務結構體來使用,使用示例稍后會有講述。
typedef struct LOS_DL_LIST { struct LOS_DL_LIST *pstPrev; /** 指向當前鏈表節點的前驅節點的指針 */ struct LOS_DL_LIST *pstNext; /** 指向當前鏈表節點的后繼節點的指針 */ } LOS_DL_LIST;
從雙向鏈表中的任意一個節點開始,都可以很方便地訪問它的前驅節點和后繼節點,這種環狀數據結構形式使得雙向鏈表在查找、插入、刪除等操作上非常方便。業務場景使用雙向鏈表時,可以定義一個LOS_DL_LIST類型的全局變量作為雙向循環鏈表Head頭結點,業務結構體的鏈表成員節點依次掛載在頭結點上。還有些業務結構體的雙向鏈表節點作為Head頭節點,依次掛載其他業務結構體的鏈表成員節點。從Head節點可以依次遍歷下一個節點,Head節點的前驅節點就是Tail尾節點。
下面通過鴻蒙輕內核代碼中互斥鎖結構體LosMuxCB定義,來了解如何使用雙向鏈表結構體:
typedef struct { UINT8 muxStat; /**< 互斥鎖狀態 */ UINT16 muxCount; /**< 互斥鎖當前被持有的次數 */ UINT32 muxID; /**< 互斥鎖編號ID */ LOS_DL_LIST muxList; /**< 互斥鎖的雙向鏈表 */ LosTaskCB *owner; /**< 當前持有鎖的任務TCB */ UINT16 priority; /**< 持有互斥鎖的任務優先級 */ } LosMuxCB;
互斥鎖結構體中包括雙向鏈表LOS_DL_LIST muxList成員變量和其他包含互斥鎖業務信息的成員變量,這里通過雙向鏈表把各個互斥鎖鏈接起來,掛載在頭結點LOS_DL_LIST g_unusedMuxList;通過其他業務成員變量承載業務數據,鏈表和其他業務成員關系如下圖所示:
2 初始化雙向鏈表
2.1 LOS_ListInit(LOS_DL_LIST *list)
LOS_DL_LIST的兩個成員pstPrev和pstNext, 是LOS_DL_LIST結構體類型的指針。需要為雙向鏈表節點申請長度為sizeof(LOS_DL_LIST)的一段內存空間。為鏈表節點申請到內存后,可以調用初始化LOS_ListInit(LOS_DL_LIST *list)方法,把這個節點鏈接為環狀的雙向鏈表。初始化鏈表時,只有一個鏈表節點,這個節點的前驅和后繼節點都是自身。鏈表節點初始化為鏈表,如圖所示:
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListInit(LOS_DL_LIST *list) { list->pstNext = list; list->pstPrev = list; }
2.2 LOS_DL_LIST_HEAD(LOS_DL_LIST list)
除了LOS_ListInit(),還提供了一個相同功能的函數式宏LOS_DL_LIST_HEAD,通過直接定義一個雙向鏈表節點,實現將該節點初始化為雙向鏈表。區別於LOS_ListInit(),在調用函數式宏前,不需要動態申請內存空間。
#define LOS_DL_LIST_HEAD(list) LOS_DL_LIST list = { &(list), &(list) }
3 判斷空鏈表
3.1 LOS_ListEmpty(LOS_DL_LIST *list)
該內聯函數用於判斷鏈表是否為空。如果雙向鏈表的前驅/后繼節點均為自身,只有一個鏈節點,沒有掛載業務結構體的鏈表節點,稱該鏈表為空鏈表。
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC_INLINE BOOL LOS_ListEmpty(LOS_DL_LIST *node) { return (BOOL)(node->pstNext == node); }
4 插入雙向鏈表節點
雙向鏈表提供三種鏈表節點插入方法,在指定鏈表節點后面插入LOS_ListAdd、尾部插入LOS_ListTailInsert、頭部插入LOS_ListHeadInsert。在頭部插入的節點,從頭部開始遍歷時第一個遍歷到,從尾部插入的節點,最后一個遍歷到。
4.1 LOS_ListAdd(LOS_DL_LIST *list, LOS_DL_LIST *node)
該內聯函數往鏈表節點*list所在的雙向鏈表中插入一個鏈表節點*node,插入位置在鏈表節點*list的后面。如圖所示,在插入過程中,會將*node的后繼節點設置為list->pstNext,*node的前驅節點為*list,並將list->pstNext的前驅節點從*list修改為*node,*list的后繼節點從list->pstNext修改為*node。
圖示:
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListAdd(LOS_DL_LIST *list, LOS_DL_LIST *node) { node->pstNext = list->pstNext; node->pstPrev = list; list->pstNext->pstPrev = node; list->pstNext = node; }
4.2 LOS_ListTailInsert(LOS_DL_LIST *list, LOS_DL_LIST *node)
該內聯函數往鏈表節點*list所在的雙向鏈表中插入一個鏈表節點*node,插入位置在鏈表節點*list的前面,list->pstPrev節點的后面。
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListTailInsert(LOS_DL_LIST *list, LOS_DL_LIST *node) { LOS_ListAdd(list->pstPrev, node); }
4.3 LOS_ListHeadInsert(LOS_DL_LIST *list, LOS_DL_LIST *node)
該內聯函數和LOS_ListAdd()實現同樣的功能,往鏈表節點*list所在的雙向鏈表中插入一個鏈表節點*node,插入位置在鏈表節點*list的后面。
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListHeadInsert(LOS_DL_LIST *list, LOS_DL_LIST *node)
{
LOS_ListAdd(list, node);
}
5 刪除雙向鏈表節點
雙向鏈表提供兩種鏈表節點的刪除方法,刪除指定節點LOS_ListDelete()、刪除並初始化為一個新鏈表LOS_ListDelInit()。
5.1 LOS_ListDelete(LOS_DL_LIST *node)
該內聯函數將鏈表節點*node從所在的雙向鏈表中刪除。節點刪除后,可能需要主動釋放節點所占用的內存。如圖所示,刪除節點過程中,會將*node的后繼節點的前驅改為*node的前驅節點,*node的前驅節點的后繼改為*node的后繼節點,並把*node節點的前驅、后繼節點設置為null,這樣*node節點就脫離了該雙向鏈表。
圖示:
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListDelete(LOS_DL_LIST *node) { node->pstNext->pstPrev = node->pstPrev; node->pstPrev->pstNext = node->pstNext; node->pstNext = NULL; node->pstPrev = NULL; }
5.2 LOS_ListDelInit(LOS_DL_LIST *list)
該內聯函數將鏈表節點*list從所在的雙向鏈表中刪除, 並把刪除后的節點重新初始化為一個新的雙向鏈表。
和LOS_ListDelete()類似,該函數也會將*list的后繼節點的前驅改為*list的前驅,*list的前驅節點的后繼改為*list的后繼,但不同的是,因為要重新初始化為新雙向鏈表,所以這個函數並不會把*list的前驅、后繼節點設置為null,而是把這個刪除的節點重新初始化為以*list為頭節點的新雙向鏈表。
源碼如下:
LITE_OS_SEC_ALW_INLINE STATIC INLINE VOID LOS_ListDelInit(LOS_DL_LIST *list) { list->pstNext->pstPrev = list->pstPrev; list->pstPrev->pstNext = list->pstNext; LOS_ListInit(list); }
6 獲取雙向鏈表節點
雙向鏈表還提供獲取鏈表節點、獲取包含鏈表的結構體地址的操作。
6.1 LOS_DL_LIST_LAST(object)
獲取指定鏈表節點的前驅節點。
源碼如下:
#define LOS_DL_LIST_LAST(object) ((object)->pstPrev)
6.2 LOS_DL_LIST_FIRST(object)
獲取指定鏈表節點的后繼節點。
源碼如下:
#define LOS_DL_LIST_FIRST(object) ((object)->pstNext)
7 遍歷雙向循環鏈表節點
雙向循環鏈表提供兩種遍歷雙向鏈表的方法,LOS_DL_LIST_FOR_EACH和LOS_DL_LIST_FOR_EACH_SAFE。
7.1 LOS_DL_LIST_FOR_EACH(item, list)
該宏定義LOS_DL_LIST_FOR_EACH遍歷雙向鏈表,將每次循環獲取的鏈表節點保存在第一個入參中,第二個入參是要遍歷的雙向鏈表的起始節點。這個宏是個for循環條件,在每次循環中,獲取下一個鏈表節點保存到入參item。業務代碼寫在宏后面的代碼塊{}內。
源碼如下:
#define LOS_DL_LIST_FOR_EACH(item, list) \ for ((item) = (list)->pstNext; (item) != (list); (item) = (item)->pstNext)
我們以實例演示如何使用LOS_DL_LIST_FOR_EACH。在kernel\src\los_task.c文件中,UINT32 OsPriqueueSize(UINT32 priority)函數的片段如下:
STATIC UINT32 OsPriqueueSize(UINT32 priority) { UINT32 itemCnt = 0; LOS_DL_LIST *curPQNode = (LOS_DL_LIST *)NULL; ⑴ LOS_DL_LIST_FOR_EACH(curPQNode, &g_losPriorityQueueList[priority]) { ++itemCnt; } return itemCnt; }
其中⑴處代碼,g_losPriorityQueueList[priority]是要循環遍歷的雙向鏈表,curPQNode指向遍歷過程中的鏈表節點。
7.2 LOS_DL_LIST_FOR_EACH_SAFE(item, next, list)
該宏定義LOS_DL_LIST_FOR_EACH_SAFE和LOS_DL_LIST_FOR_EACH的唯一區別就是多了一個入參next, 這個參數表示遍歷到的雙向鏈表節點的下一個節點。該宏用於安全刪除,如果刪除遍歷到的item, 不影響繼續遍歷。
源碼如下:
#define LOS_DL_LIST_FOR_EACH_SAFE(item, next, list) \ for ((item) = (list)->pstNext, (next) = (item)->pstNext; (item) != (list); \ (item) = (next), (next) = (item)->pstNext)
8 獲取鏈表節點所在結構體
8.1 LOS_OFF_SET_OF(type, member)
根據結構體類型名稱type和其中的成員變量名稱member,獲取member成員變量相對於結構體type的內存地址偏移量。在鏈表的應用場景上,業務結構體包含雙向鏈表作為成員,當知道雙向鏈表成員變量的內存地址和相對於業務結構體的偏移時,就可以進一步獲取業務結構體的內存地址,具體見下面LOS_DL_LIST_ENTRY的宏實現。
源碼如下:
#define LOS_OFF_SET_OF(type, member) ((UINTPTR)&((type *)0)->member)
8.2 LOS_DL_LIST_ENTRY(item, type, member)
函數宏中的三個參數分別為:業務結構體類型名稱type,作為結構體成員的雙向鏈表成員變量名稱member,作為結構體成員的雙向鏈表節點指針item。通過調用該宏函數LOS_DL_LIST_ENTRY即可以獲取雙向鏈表節點所在的業務結構體的內存地址。
源碼如下:
基於雙向鏈表節點的內存地址,和雙向鏈表成員變量在結構體中的地址偏移量,可以計算出結構體的內存地址。
#define LOS_DL_LIST_ENTRY(item, type, member) \ ((type *)(VOID *)((CHAR *)(item) - LOS_OFF_SET_OF(type, member)))
9 遍歷包含雙向鏈表的結構體
雙向鏈表提供三個宏定義來遍歷包含雙向鏈表成員的結構體,LOS_DL_LIST_FOR_EACH_ENTRY、LOS_DL_LIST_FOR_EACH_ENTRY_SAFE和LOS_DL_LIST_FOR_EACH_ENTRY_HOOK。
9.1 LOS_DL_LIST_FOR_EACH_ENTRY(item, list, type, member)
該宏定義LOS_DL_LIST_FOR_EACH_ENTRY通過遍歷雙向鏈表,在每次循環中獲取包含該雙向鏈表成員的結構體變量並保存在第一個入參中。第二個入參是要遍歷的雙向鏈表的起始節點,第三個入參是要獲取的結構體類型名稱,第四個入參是該結構體中的雙向鏈表成員變量的名稱。這個宏是個for循環條件,業務代碼寫在宏后面的代碼塊{}內。
源碼如下:
for循環的初始化語句item = LOS_DL_LIST_ENTRY((list)->pstNext, type, member)表示獲取包含雙向鏈表第一個有效節點的結構體,並保存到指針變量item中。條件測試語句&(item)->member != (list)表示當雙向鏈表遍歷一圈到自身節點時,停止循環。循環更新語句item = LOS_DL_LIST_ENTRY((item)->member.pstNext, type, member))中,使用(item)->member.pstNext遍歷到下一個鏈表節點,然后根據這個節點獲取對應的下一個結構體的指針變量item,直至遍歷完畢。
#define LOS_DL_LIST_FOR_EACH_ENTRY(item, list, type, member) \ for (item = LOS_DL_LIST_ENTRY((list)->pstNext, type, member); \ &(item)->member != (list); \ item = LOS_DL_LIST_ENTRY((item)->member.pstNext, type, member))
9.2 LOS_DL_LIST_FOR_EACH_ENTRY_SAFE(item, next, list, type, member)
該宏定義和LOS_DL_LIST_FOR_EACH_ENTRY的唯一區別就是多了一個入參next, 這個參數表示遍歷到的結構體的下一個結構體。該宏用於安全刪除,如果刪除遍歷到的item,不影響繼續遍歷。
源碼如下:
#define LOS_DL_LIST_FOR_EACH_ENTRY_SAFE(item, next, list, type, member) \ for (item = LOS_DL_LIST_ENTRY((list)->pstNext, type, member), \ next = LOS_DL_LIST_ENTRY((item)->member->pstNext, type, member); \ &(item)->member != (list); \ item = next, next = LOS_DL_LIST_ENTRY((item)->member.pstNext, type, member))
9.3 LOS_DL_LIST_FOR_EACH_ENTRY_HOOK(item, list, type, member, hook)
該宏定義和LOS_DL_LIST_FOR_EACH_ENTRY的區別就是多了一個入參hook,hook表示鈎子函數。在每次遍歷循環中,會調用該鈎子函數,實現用戶任務的定制。
源碼如下:
#define LOS_DL_LIST_FOR_EACH_ENTRY_HOOK(item, list, type, member, hook) \ for (item = LOS_DL_LIST_ENTRY((list)->pstNext, type, member), hook; \ &(item)->member != (list); \ item = LOS_DL_LIST_ENTRY((item)->member.pstNext, type, member), hook)
本文分享自華為雲社區《鴻蒙輕內核M核源碼分析系列二 數據結構-雙向循環鏈表》,原文作者:zhushy 。