Linux 驱动程序中 container

众所周知，linux内核的主要开发语言是C，但是现在内核的框架使用了非常多的面向对象的思想，这就面临了一个用C语言来实现面向对象编程的问题，今天我们就来讲讲其中一个例子。

利用结构体中元素指针获取结构体指针

Kobject是linux设备驱动模型的基础，也是设备模型中抽象的一部分。

linux内核为了兼容各种形形色色的设备，就需要对各种设备的共性进行抽象，抽象出一个基类，其余的设备只需要继承此基类就可以了。

而此基类就是kobject(暂且把它看成是一个类)，但是C语言没有面向对象语法。

在C++中这样的操作非常简单，继承基类就可以了，而在C语言中需要将基类的结构体指针嵌入到派生的类中，那么为什么将基类指针嵌入就可以得到派生类的指针呢？

这个实现是一个宏：container_of。

container_of

container_of 被定义于 /linux/kernel.h 中。

为了阅读和理解的方便，我们在这里先假设基类为Base,而派生类为Derived，显然，派生类Derived中是包含基类Base的定义的，即已知类为Base，未知类为Derived，我们要求解Derived的地址，一个具体的对应就是通过kobject地址求解包含kobject的父结构体的地址，由此我们来逐一分析这个宏的实现。

首先，它的原型是这样的：

#define container_of(ptr, type, member) ({                        \
void *__mptr = (void *)(ptr);                                    \
BUILD_BUG_ON_MSG(!__same_type(*(ptr), ((type *)0)->member) &&    \
         !__same_type(*(ptr), void),                            \
         "pointer type mismatch in container_of()");            \
((type *)(__mptr - offsetof(type, member))); })

参数：

ptr
ptr是已知基类Base的地址
type
type是派生类的的类型
member
基类Base在派生类Derived中定义的成员名字

上面的原型可能不同的 Linux 内核版本中定义不同。

void __mptr = (void )(ptr);

这一条使用一个临时变量 __mptr 赋值，仅仅是为了避免对ptr的误操作导致修改了ptr的值，这可不是我们的初衷。

思考一下，为什么在这里要使用void类型，保持原类型可不可以？

BUILD_BUG_ON_MSG(…);

这一条包含两个操作，首先是使用 __same_type 宏，其实这个宏名就已经解释了这个宏的作用，即判断传入的两个类型是否一致：

# define __same_type(a, b) __builtin_types_compatible_p(typeof(a), typeof(b))

__builtin_types_compatible_p是一个內建功能函数，typeof是gcc关键字，这里的作用是判断传入的两个参数类型是否匹配。

然后是 BUILD_BUG_ON_MSG 宏：

#define BUILD_BUG_ON_MSG(cond, msg) compiletime_assert(!(cond), msg)

这个宏检测 cond(condition)，如果条件不为零，输出 msg。

这一部分总体的作用就是检测 container_of(ptr, type, member)中，ptr 的类型是否和 member 类型一致，如果不一致，则报错。

((type *)(__mptr - offsetof(type, member)))

这一部分就是核心所在。

首先，使用offsetof找到基类Base在派生类中的偏移地址，原型是这样的：

//SofTool's Note: 可以简单理解为 取 TYPE 结构体中 MEMBER 成员变量相对于本结构体开始位置(0嘛 呵呵)的相对偏移量。
#define offsetof(TYPE, MEMBER)    ((size_t)&((TYPE *)0)->MEMBER)

&((TYPE *)0)->MEMBER：这一个表达式将0地址强制转换成TYPE类型(TYPE是用户传入的派生类Derived的类型)，然后访问member成员，用&获取member成员地址。

对C语言不那么熟悉的小伙伴会非常疑惑：不是说0地址是非法地址么，为什么这里可以用？0地址是非法地址没错，但是仅仅是限于不能进行访问操作。

为什么经过强制转换的0地址可以访问member成员，显然0地址处并不存在TYPE的实例？这个问题其实就是属于指针以及指针类型的经典难题了，这里简要解释一下：
首先我们要弄清楚指针的类型以及指针指向数据的类型。
指针：是变量，指针的值为一个地址，指针本身的类型是int型；
指针指向数据的类型则有多种，常见的int，char等等；
当我们对一个指针变量进行加减运算时，事实上是以指针指向数据类型的size进行操作；
比如：
ptr指向一个int类型，ptr值为0x1000，ptr++是将指针的值+4(32位)，即0x1004；
若ptr指向一个char型，ptr++就是+1,ptr值为0x1001；
而我们对目标类型进行强制转换，比如ptr本来是指向int型，我们这样操作：((char)prt)++,这样ptr就临时成为了char型指针，ptr++的操作时ptr的值只+1，即0x1001.
即强制转换提供一种临时的指向数据类型的变换，在强制转换过程中目标地址保持为被转换类型。
对于结构体而言，结构体指针可以直接通过->来访问某个成员。
而我们对0地址进行强制转换，就可以看成是0地址处放置了一个目标结构体的实例，然后通过->成员来访问，当然这里并没有访问，如果访问肯定会出错，这里只是取出了member的相对于结构体地址的偏移地址。

刚刚只是介绍了offsetof(type, member)，然后我们结合来看：

((type *)(\_\_mptr - offsetof(type, member)))

__mptr - offsetof(type, member)：即基类ptr在结构体中的地址减去基类ptr在结构体中的偏移地址，就得到了派生类结构体的地址(这一部分涉及到结构体的存储方式)。

Softool’s Note:

基类ptr在结构体中的地址
指ptr在派生类中位置相对于内存起始的物理地址;
基类ptr在结构体中的偏移地址
指ptr在派生类中位置相对于当前派生类开始的偏移量;

刚刚我们提到了一个问题：
为什么在__mptr要使用void类型，保持原类型可不可以？如果你理解了我上面提到的指针部分，这个问题就很好解释了。
如果__mptr为void类型，__mptr - offsetof(type, member)这个表达式就是__mptr的值减去offsetof(type, member)的值，得到__mptr。
如果保持原类型，__mptr - offsetof(type, member)这个表达式成了__mptr的值减去 offsetof(type, member)sizeof(type),得到__mptr,显然，我们要的是第一种。

（参考上面的指针类型简析的第四行）

小结

container_of(ptr, type, member)

ptr
为基类在结构体中的地址
type
为派生类类型
member
为基类在派生类中定义时的名字，其实就是为了在宏中直接以member名来访问

总的来说，就是我们可以通过结构体内元素指针的值获取结构体的值。其实container_of的用法并不局限于kobject系统中，而是适用于所有基类派生类的模式。

示例

上面讲了那么多概念，还是得用实例来说话，下面是博主写的一个小demo：

#include <stdio.h>

struct Base{                    //定义一个Base类
    int var;
    char *string;
};

struct Derived{                 //定义一个派生类
    int var;
    struct Base base;           //派生类中包含了struct Base类
};

#define offsetof(TYPE, MEMBER)    ((size_t)&((TYPE *)0)->MEMBER)       //offset_of宏
#define container_of(ptr, type, member) ({                \            //阉割版container_of，省去了类型检查
    void *__mptr = (void *)(ptr);                     \
    ((type *)(__mptr - offsetof(type, member))); })

struct Base *base_p;                  
struct Derived test_derived;       

struct Derived *get;
int main()
{
   base_p = &test_derived.base;                                     //赋值给指针
   printf("Derived addr = %x\n",(unsigned int)&test_derived);
   printf("=================================\n");
   get = container_of(base_p,struct Derived,base);                 //使用base_p指针获取派生类的首地址
   printf("get Derived addr = %x\n",(unsigned int)get);
   return 0;
}

输出：

Derived addr = 601050
=================================
get Derived addr = 601050

作者：返回主页牧野星辰
https://www.cnblogs.com/downey-blog/p/10485477.html

Linux 驱动程序中 container_of 宏解析

利用结构体中元素指针获取结构体指针

container_of

void *__mptr = (void *)(ptr);

BUILD_BUG_ON_MSG(…);

小结

示例

void __mptr = (void )(ptr);