首页 > 文章列表 > 深入解析Linux内核中的container_of宏

深入解析Linux内核中的container_of宏

linux Linux系统 Linux命令 Shell脚本 linux教程 良许 linux入门 linux学习
390 2024-02-06

在之前的文章中,我们讨论了在移植和使用Linux内核的通用链表时,如何避免重复造轮子,并提供了完整的代码实现。今天,我们将深入分析内核是如何求解结构体成员变量的地址的,尤其是在计算结构体偏移时巧妙地使用了container_of宏定义。

  • 1. 结构体在内存中是如何存储的
  • 2. container_of宏
  • 3. typeof
  • 4. (((type *)0)->member)
  • 5. const typeof(((type * )0) ->member)*__mptr = (ptr);
  • 6. offsetof(type, member))
  • 7. (type * )((char * )__mptr – offsetof(type, member))
  • 8. 举例

1. 结构体在内存中是如何存储的

int main()
{

 Student stu;
 stu.id = 123456;
 strcpy(stu.name,"feizhufeifei");
 stu.math = 90;
 stu.PE = 80;
 printf("Student:%prn",&stu);
 printf("stu.ID:%prn",&stu.ID);
 printf("stu.name:%prn",&stu.name);
 printf("stu.math:%prn",&stu.math);
 return 0;
}

打印结果如下:

//结构体的地址
Student:0xffffcbb0
//结构体第一个成员的地址
stu.ID:0xffffcbb0  //偏移地址 +0
stu.name:0xffffcbb4//偏移地址 +4
stu.math:0xffffcbd4//偏移地址 +24

我们可以看到,结构体的地址和结构体第一个成员的地址是相同的。这也就是我们之前在拒绝造轮子!如何移植并使用Linux内核的通用链表(附完整代码实现)中提到的为什么在结构体中要把 struct list_head放在首位。

不太理解的再看下这两个例子:

struct A { int a; char b; int c; char d; };a 偏移为 0 , b 偏移为 4 , c 偏移为 8 (大于 4 + 1 的 4 的最小整数倍), d 偏移为 12 。A 对齐为 4 ,大小为 16 。

struct B { int a; char b; char c; long d; };a 偏移为 0 , b 偏移为 4 , c 偏移为 5 , d 偏移为 8 。B 对齐为 8 , 大小为 16 。

Linux内核中container_of宏的详细解释

我们可以看到,结构体中成员变量在内存中存储的其实是偏移地址。也就是说结构体A的地址+成员变量的偏移地址 = 结构体成员变量的起始地址

因此,我们也可以根据结构体变量的起始地址和成员变量的偏移地址来反推出结构体A的地址。

2. container_of宏

#define offsetof(TYPE, MEMBER) ((size_t) &((TYPE*)0)->MEMBER)
#define container_of(ptr, type, member) ({          
        const typeof(((type *)0)->member)*__mptr = (ptr);    
    (type *)((char *)__mptr - offsetof(type, member)); })

首先看下三个参数, ptr是成员变量的指针, type是指结构体的类型, member是成员变量的名字。

container_of宏的作用是通过结构体内某个成员变量的地址和该变量名,以及结构体类型,找到该结构体变量的地址。这里使用的是一个利用编译器技术的小技巧,即先求得结构成员在结构中的偏移量,然后根据成员变量的地址反过来得出主结构变量的地址。下面具体分析下各个部分。

3. typeof

首先看下typeof,是用于返回一个变量的类型,这是GCC编译器的一个扩展功能,也就是说typeof是编译器相关的。既不是C语言规范的所要求,也不是某个标准的一部分。

typeof

int main()
{
 int a = 5;
 //这里定义一个和a类型相同的变量b
 typeof(a) b  = 6;
 printf("%d,%drn",a,b);//5 6
 return 0;
}

4. (((type *)0)->member)

((TYPE *)0) 将0转换为type类型的结构体指针,换句话说就是让编译器认为这个结构体是开始于程序段起始位置0,开始于0地址的话,我们得到的成员变量的地址就直接等于成员变量的偏移地址了。

(((type *)0)->member) 引用结构体中MEMBER成员。

typedef struct student{
 int id;
 char name[30];
 int math;
}Student;
int main()
{
 //这里时把结构体强制转换成0地址,然后打印name的地址。
 printf("%drn",&((Student *)0)->name);//4
 return 0;
}

5. const typeof(((type * )0) ->member)*__mptr = (ptr);

这句代码意思是用typeof()获取结构体里member成员属性的类型,然后定义一个该类型的临时指针变量__mptr,并将ptr所指向的member的地址赋给__mptr;

为什么不直接使用 ptr 而要多此一举呢?我想可能是为了避免对 ptr 及prt 指向的内容造成破坏。

6. offsetof(type, member))

((size_t) &((TYPE*)0)->MEMBER)

size_t是标准C库中定义的,在32位架构中被普遍定义为:

typedef unsigned int size_t;

而在64位架构中被定义为:

typedef unsigned long size_t;

可以从定义中看到,size_t是一个非负数,所以size_t通常用来计数(因为计数不需要负数区):

for(size_t i=0;i

为了使程序有很好的移植性,因此内核使用size_t,而不是int,unsigned。((size_t) &((TYPE*)0)->MEMBER) 结合之前的解释,我们可以知道这句话的意思就是求出MEMBER相对于0地址的一个偏移值。

7. (type * )((char * )__mptr – offsetof(type, member))

这句话的意思就是,把 __mptr 转换成 char* 类型。因为 offsetof 得到的偏移量是以字节为单位。两者相减得到结构体的起始位置, 再强制转换成 type 类型。

8. 举例

#define offsetof(TYPE, MEMBER) ((size_t) &((TYPE *)0)->MEMBER)
#define container_of(ptr, type, member) ({ 
        const typeof( ((type *)0)->member ) *__mptr = (ptr); 
        (type *)( (char *)__mptr - offsetof(type,member) );})
        
typedef struct student
{
 int id;
 char name[30];
 int math;
}Student;

int main()
{
    Student stu;
        Student *sptr = NULL;
  stu.id = 123456;
  strcpy(stu.name,"zhongyi");
  stu.math = 90;
        sptr = container_of(&stu.id,Student,id);
        printf("sptr=%pn",sptr);
        sptr = container_of(&stu.name,Student,name);
        printf("sptr=%pn",sptr);
        sptr = container_of(&stu.math,Student,id);
        printf("sptr=%pn",sptr);
        return 0; 
}

运行结果如下:

sptr=0xffffcb90
sptr=0xffffcb90
sptr=0xffffcbb4

宏展开可能会看的更清楚一些

int main()
{
    Student stu;
        Student *sptr = NULL;
  stu.id = 123456;
  strcpy(stu.name,"zhongyi");
  stu.math = 90;
  //展开替换
        sptr = ({ const unsigned char  *__mptr = (&stu.id); (Student *)( (char *)__mptr - ((size_t) &((Student *)0)->id) );});
        printf("sptr=%pn",sptr);
        //展开替换
        sptr = ({ const unsigned char  *__mptr = (&stu.name); (Student *)( (char *)__mptr - ((size_t) &((Student *)0)->name) );});
        printf("sptr=%pn",sptr);
        //展开替换
        sptr = ({ const unsigned int *__mptr = (&stu.math); (Student *)( (char *)__mptr - ((size_t) &((Student *)0)->math) );});
        printf("sptr=%pn",sptr);
        return 0; 
}