1、前言
今天在看代碼時,遇到offsetof和container_of兩個宏,覺得很有意思,功能很強大。offsetof是用來判斷結構體中成員的偏移位置,container_of宏用來根據成員的地址來獲取結構體的地址。兩個宏設計的很巧妙,值得學習。linux內核中有著兩個宏的定義,並在鏈表結構中得到應用。不得不提一下linux內核中的鏈表,設計的如此之妙,只需要兩個指針就搞定了。後續認真研究一下這個鏈表結構。
2、offsetof宏
使用offsetof宏需要包含stddef.h頭文件,實例可以參考:http://www.cplusplus.com/reference/cstddef/offsetof/。
offsetof宏的定義如下:
#define offsetof(type, member) (size_t)&(((type*)0)->member)
巧妙之處在於將地址0強制轉換為type類型的指針,從而定位到member在結構體中偏移位置。編譯器認為0是一個有效的地址,從而認為0是type指針的起始地址。
3、container_of宏
使用container_of宏需要包含linux/kernel.h頭文件,container_of宏的定義如下所示:
#define container_of(ptr, type, member) ({ \ const typeof( ((type *)0)->member ) *__mptr = (ptr); \ (type *)( (char *)__mptr - offsetof(type,member) );})
container_of宏分為兩部分,
第一部分:const typeof( ((type *)0)->member ) *__mptr = (ptr);
通過typeof定義一個member指針類型的指針變量__mptr,(即__mptr是指向member類型的指針),並將__mptr賦值為ptr。
第二部分: (type *)( (char *)__mptr - offsetof(type,member) ),通過offsetof宏計算出member在type中的偏移,然後用member的實際地址__mptr減去偏移,得到type的起始地址,即指向type類型的指針。
第一部分的目的是為了將統一轉換為member類型指針。
4、測試程序
1 #include <stdio.h> 2 #include <stdlib.h> 3 4 #define NAME_STR_LEN 32 5 6 #define offsetof(type, member) (size_t)&(((type*)0)->member) 7 8 #define container_of(ptr, type, member) ({ \ 9 const typeof( ((type *)0)->member ) *__mptr = (ptr); \ 10 (type *)( (char *)__mptr - offsetof(type,member) );}) 11 12 typedef struct student_info 13 { 14 int id; 15 char name[NAME_STR_LEN]; 16 int age; 17 }student_info; 18 19 20 int main() 21 { 22 size_t off_set = 0; 23 off_set = offsetof(student_info, id); 24 printf("id offset: %u\n",off_set); 25 off_set = offsetof(student_info, name); 26 printf("name offset: %u\n",off_set); 27 off_set = offsetof(student_info, age); 28 printf("age offset: %u\n",off_set); 29 student_info *stu = (student_info *)malloc(sizeof(student_info)); 30 stu->age = 10; 31 student_info *ptr = container_of(&(stu->age), student_info, age); 32 printf("age:%d\n", ptr->age); 33 printf("stu address:%p\n", stu); 34 printf("ptr address:%p\n", ptr); 35 return 0; 36
測試結果: