今天看 linux 内核代码时,发现了结构体中只有一个成员的数据结构,事实上,在 linux 内核中,这种结构相当常见。举例如下:
struct rb_root{ struct rb_node *rb_node;};
![c463a2e05c8161e2c352e9b34c48d8b1.png](https://i-blog.csdnimg.cn/blog_migrate/8961239b701b69c90cbdb26b6225cbf1.jpeg)
只有一个成员的结构体
一开始,我觉得这样无非就是为了方便突出 rb_root 的特异性(虽然它也是一个节点,但是 root 节点),或者方便以后扩展。但今天我的脑海里却灵光一闪,发现采用这种数据结构,其实更方便的点在于:能够避免使用多级指针。
![38d8bce0243b62078023f31beb1827a0.png](https://i-blog.csdnimg.cn/blog_migrate/3da7a14c4e0a7f462101a70edf3edf24.jpeg)
在我之前的文章里讨论过,C语言使用指针做参数常常能够提高程序紧凑型,和运行效率。事实上,struct rb_root 数据结构是用于定义红黑树根节点的,红黑树的根节点可能会在某次着色变换中改变,这时如果仅仅使用 struct rb_node 来定义根节点指针,负责“改变”根节点的函数就需要二级指针做参数了。
需要使用二级指针的例子
这样说可能有些懵,我们还是来看实例吧。假设在某个函数中,需要 malloc 一块内存:
char *p = (char*)malloc(10);
使用完毕后,我们应该将其释放,而且应该让 p 指向 NULL,原因我之前的文章解释过。
if(p){ free(p); p = NULL;}
![3378f909dd40064cd402e29b967539a0.png](https://i-blog.csdnimg.cn/blog_migrate/868d17f7b6110ef79137b02ee53b1b48.jpeg)
如果需要频繁的释放 p,每次这么写代码就太繁琐了,那能否定义一个函数完成这两个步骤呢?当然是可以的,请看以下C语言代码:
void my_free(char* p){ if(p){ free(p); p = NULL; }}// p != NULLmy_free(p);// p != NULL
my_free() 函数的确能够释放 p,但是却无法完成将 p 指向 NULL 的操作。原因我的之前的这篇文章《一文彻底弄懂函数的形参和实参》已经解释过。若想实现我们的计划,需要借助二级指针才可以,请看如下C语言代码:
void my_free(char** p){ if(*p){ free(*p); *p = NULL; }}// p != NULLmy_free(&p);// p == NULL
![44f3a62009a0ba0f3cd3b33da4bb778e.png](https://i-blog.csdnimg.cn/blog_migrate/2df672c1e02259888d48264be4ef02f2.jpeg)
使用二级指针的确能够解决问题,但是它也让代码变得不那么直观了,使用起来也不是特别方便,所以可以借助文章一开头提到的那种结合结构体的数据结构。
![1e1398eba77b3535a9c699100a99a064.png](https://i-blog.csdnimg.cn/blog_migrate/b69fa9e1ac5d1f34409ca247fd3828d1.jpeg)
避免使用二级指针
先来定义数据结构如下,请看C语言代码:
struct PTR{ char* data;};
之后我们 malloc 和 free 时都使用 PTR 的成员 data,就能避免使用二级指针了,请看下面的C语言代码:
void my_free(struct PTR* ptr){ if(ptr){ if(ptr->data){ free(ptr->data); ptr->data = NULL; } }}int main(){ struct PTR ptr; ptr.data = (char*)malloc(10); if(NULL==ptr.data) exit(110); printf("before free, data: %p