C指针的这些使用技巧,掌握后立刻提升一个Level

开发 前端
今天主要是介绍指针在应用程序的编程中,经常使用的技巧。

 

一、前言

半个月前写的那篇关于指针最底层原理的文章,得到了很多朋友的认可(链接: C语言指针-从底层原理到花式技巧,用图文和代码帮你讲解透彻),特别是对刚学习C语言的小伙伴来说,很容易就从根本上理解指针到底是什么、怎么用,这也让我坚信一句话;用心写出的文章,一定会被读者感受到!在写这篇文章的时候,我列了一个提纲,写到后面的时候,发现已经超过一万字了,但是提纲上还有最后一个主题没有写。如果继续写下去,文章体积就太大了,于是就留下了一个尾巴。

今天,我就把这个尾巴给补上去:主要是介绍指针在应用程序的编程中,经常使用的技巧。如果之前的那篇文章勉强算是“道”层面的话,那这篇文章就属于“术”的层面。主要通过 8 个示例程序来展示在 C 语言应用程序中,关于指针使用的常见套路,希望能给你带来收获。

记得我在校园里学习C语言的时候,南师大的黄凤良老师花了大半节课的时间给我们解释指针,现在最清楚地记得老师说过的一句话就是:指针就是地址,地址就是指针!

二、八个示例

1. 开胃菜:修改主调函数中的数据

  1. // 交换 2 个 int 型数据 
  2. void demo1_swap_data(int *a, int *b) 
  3.     int tmp = *a; 
  4.     *a = *b; 
  5.     *b = tmp; 
  6.  
  7. void demo1() 
  8.     int i = 1; 
  9.     int j = 2; 
  10.     printf("before: i = %d, j = %d \n", i, j); 
  11.     demo1_swap_data(&i, &j); 
  12.     printf("after:  i = %d, j = %d \n", i, j); 

这个代码不用解释了,大家一看就明白。如果再过多解释的话,好像在侮辱智商。

2. 在被调用函数中,分配系统资源

代码的目的是:在被调用函数中,从堆区分配 size 个字节的空间,返回给主调函数中的 pData 指针。

  1. void demo2_malloc_heap_error(char *buf, int size
  2.     buf = (char *)malloc(size); 
  3.     printf("buf = 0x%x \n", buf); 
  4.  
  5. void demo2_malloc_heap_ok(char **buf, int size
  6.     *buf = (char *)malloc(size); 
  7.     printf("*buf = 0x%x \n", *buf); 
  8.  
  9. void demo2() 
  10.     int size = 1024; 
  11.     char *pData = NULL
  12.  
  13.     // 错误用法 
  14.     demo2_malloc_heap_error(pData, size); 
  15.     printf("&pData = 0x%x, pData = 0x%x \n", &pData, pData); 
  16.  
  17.     // 正确用法 
  18.     demo2_malloc_heap_ok(&pData, size); 
  19.     printf("&pData = 0x%x, pData = 0x%x \n", &pData, pData); 
  20.     free(pData); 

2.1 错误用法

刚进入被调用函数 demo2_malloc_heap_error 的时候,形参 buff 是一个 char* 型指针,它的值等于 pData 变量的值,也就是说 buff 与 pData 的值相同(都为 NULL),内存模型如图:


在被调用函数中执行 malloc 语句之后,从堆区申请得到的地址空间赋值给 buf,就是说它就指向了这个新的地址空间,而 pData 里仍然是NULL,内存模型如下:


从图中可以看到,pData 的内存中一直是 NULL,没有指向任何堆空间。另外,由于形参 buf 是放在函数的栈区的,从被调函数中返回的时候,堆区这块申请的空间就被泄漏了。

2.2 正确用法

刚进入被调用函数 demo2_malloc_heap_error 的时候,形参 buf 是一个 char* 型的二级指针,就是说 buf 里的值是另一个指针变量的地址,在这个示例中 buf 里的值就是 pData 这个指针变量的地址,内存模型如下:


在被调用函数中执行 malloc 语句之后,从堆区申请得到的地址空间赋值给 *buf,因为 buf = &pData,所以 *buf 就相当于是 pData,那么从堆区申请得到的地址空间就赋值 pData 变量,内存模型如下:


从被调函数中返回之后,pData 就正确的得到了一块堆空间,别忘了使用之后要主动释放。

3. 传递函数指针

从上篇文章中我们知道,函数名本身就代表一个地址,在这个地址中存储着函数体中定义的一连串指令码,只要给这个地址后面加上一个调用符(小括号),就进入这个函数中执行。在实际程序中,函数名常常作为函数参数来进行传递。

熟悉C++的小伙伴都知道,在标准库中对容器类型的数据进行各种算法操作时,可以传入用户自己的提供的算法函数(如果不传入函数,标准库就使用默认的)。

下面是一个示例代码,对一个 int 行的数组进行排序,排序函数 demo3_handle_data 的最后一个参数是一个函数指针,因此需要传入一个具体的排序算法函数。示例中有 2 个候选函数可以使用:

  • 降序排列: demo3_algorithm_decend;
  • 升序排列: demo3_algorithm_ascend;
  1. typedef int BOOL; 
  2. #define FALSE 0 
  3. #define TRUE  1 
  4.  
  5. BOOL demo3_algorithm_decend(int a, int b) 
  6.     return a > b; 
  7.  
  8. BOOL demo3_algorithm_ascend(int a, int b) 
  9.     return a < b; 
  10.  
  11. typedef BOOL (*Func)(intint); 
  12. void demo3_handle_data(int *data, int size, Func pf) 
  13.     for (int i = 0; i < size - 1; ++i) 
  14.     { 
  15.         for (int j = 0; j < size - 1 - i; ++j) 
  16.         { 
  17.             // 调用传入的排序函数 
  18.             if (pf(data[j], data[j+1])) 
  19.             { 
  20.                 int tmp = data[j]; 
  21.                 data[j] = data[j + 1]; 
  22.                 data[j + 1] = tmp; 
  23.             } 
  24.         } 
  25.     } 
  26.  
  27. void demo3() 
  28.     int a[5] = {5, 1, 9, 2, 6}; 
  29.     int size = sizeof(a)/sizeof(int); 
  30.     // 调用排序函数,需要传递排序算法函数 
  31.     //demo3_handle_data(a, size, demo3_algorithm_decend); // 降序排列 
  32.     demo3_handle_data(a, size, demo3_algorithm_ascend);   // 升序排列 
  33.     for (int i = 0; i < size; ++i) 
  34.         printf("%d ", a[i]); 
  35.     printf("\n"); 

这个就不用画图了,函数指针 pf 就指向了传入的那个函数地址,在排序的时候
直接调用就可以了。

4. 指向结构体的指针

在嵌入式开发中,指向结构体的指针使用特别广泛,这里以智能家居中的一条控制指令来举例。在一个智能家居系统中,存在各种各样的设备(插座、电灯、电动窗帘等),每个设备的控制指令都是不一样的,因此可以在每个设备的控制指令结构体中的最前面,放置所有指令都需要的、通用的成员变量,这些变量可以称为指令头(指令头中包含一个代表命令类型的枚举变量)。

当处理一条控制指令时,先用一个通用命令(指令头)的指针来接收指令,然后根据命令类型枚举变量来区分,把控制指令强制转换成具体的那个设备的数据结构,这样就可以获取到控制指令中特定的控制数据了。

本质上,与 Java/C++ 中的接口、基类的概念类似。

  1. // 指令类型枚举 
  2. typedef enum _CMD_TYPE_ { 
  3.     CMD_TYPE_CONTROL_SWITCH = 1, 
  4.     CMD_TYPE_CONTROL_LAMP, 
  5. } CMD_TYPE; 
  6.  
  7. // 通用的指令数据结构(指令头) 
  8. typedef struct _CmdBase_ { 
  9.     CMD_TYPE cmdType; // 指令类型 
  10.     int deviceId;     // 设备 Id 
  11. } CmdBase; 
  12.  
  13. typedef struct _CmdControlSwitch_ { 
  14.     // 前 2 个参数是指令头 
  15.     CMD_TYPE cmdType;    
  16.     int deviceId; 
  17.      
  18.     // 下面都有这个指令私有的数据 
  19.     int slot;  // 排插上的哪个插口 
  20.     int state; // 0:断开, 1:接通 
  21. } CmdControlSwitch; 
  22.  
  23. typedef struct _CmdControlLamp_ { 
  24.     // 前 2 个参数是指令头 
  25.     CMD_TYPE cmdType; 
  26.     int deviceId; 
  27.      
  28.     // 下面都有这个指令私有的数据 
  29.     int color;      // 颜色 
  30.     int brightness; // 亮度 
  31. } CmdControlLamp; 
  32.  
  33. // 参数是指令头指针 
  34. void demo4_control_device(CmdBase *pcmd) 
  35.     // 根据指令头中的命令类型,把指令强制转换成具体设备的指令 
  36.     if (CMD_TYPE_CONTROL_SWITCH == pcmd->cmdType) 
  37.     { 
  38.         // 类型强制转换 
  39.         CmdControlSwitch *cmd = pcmd; 
  40.         printf("control switch. slot = %d, state = %d \n", cmd->slot, cmd->state); 
  41.     } 
  42.     else if (CMD_TYPE_CONTROL_LAMP == pcmd->cmdType) 
  43.     { 
  44.         // 类型强制转换 
  45.         CmdControlLamp * cmd = pcmd; 
  46.         printf("control lamp.   color = 0x%x, brightness = %d \n", cmd->color, cmd->brightness); 
  47.     } 
  48.  
  49. void demo4() 
  50.     // 指令1:控制一个开关 
  51.     CmdControlSwitch cmd1 = {CMD_TYPE_CONTROL_SWITCH, 1, 3, 0}; 
  52.     demo4_control_device(&cmd1); 
  53.  
  54.     // 指令2:控制一个灯泡 
  55.     CmdControlLamp cmd2 = {CMD_TYPE_CONTROL_LAMP, 2, 0x112233, 90}; 
  56.     demo4_control_device(&cmd2); 

5. 函数指针数组

这个示例在上篇文章中演示过,为了完整性,这里再贴一下。

  1. int add(int a, int b) { return a + b; } 
  2. int sub(int a, int b) { return a - b; } 
  3. int mul(int a, int b) { return a * b; } 
  4. int divide(int a, int b) { return a / b; } 
  5.  
  6. void demo5() 
  7.     int a = 4, b = 2; 
  8.     int (*p[4])(intint); 
  9.     p[0] = add
  10.     p[1] = sub; 
  11.     p[2] = mul; 
  12.     p[3] = divide; 
  13.     printf("%d + %d = %d \n", a, b, p[0](a, b)); 
  14.     printf("%d - %d = %d \n", a, b, p[1](a, b)); 
  15.     printf("%d * %d = %d \n", a, b, p[2](a, b)); 
  16.     printf("%d / %d = %d \n", a, b, p[3](a, b)); 

6. 在结构体中使用柔性数组

先不解释概念,我们先来看一个代码示例:

  1. // 一个结构体,成员变量 data 是指针 
  2. typedef struct _ArraryMemberStruct_NotGood_ { 
  3.     int num; 
  4.     char *data; 
  5. } ArraryMemberStruct_NotGood; 
  6.  
  7. void demo6_not_good() 
  8.     // 打印结构体的内存大小 
  9.     int size = sizeof(ArraryMemberStruct_NotGood); 
  10.     printf("size = %d \n"size); 
  11.  
  12.     // 分配一个结构体指针 
  13.     ArraryMemberStruct_NotGood *ams = (ArraryMemberStruct_NotGood *)malloc(size); 
  14.     ams->num = 1; 
  15.  
  16.     // 为结构体中的 data 指针分配空间 
  17.     ams->data = (char *)malloc(1024); 
  18.     strcpy(ams->data, "hello"); 
  19.     printf("ams->data = %s \n", ams->data); 
  20.  
  21.     // 打印结构体指针、成员变量的地址 
  22.     printf("ams = 0x%x \n", ams); 
  23.     printf("ams->num  = 0x%x \n", &ams->num); 
  24.     printf("ams->data = 0x%x \n", ams->data); 
  25.  
  26.     // 释放空间 
  27.     free(ams->data); 
  28.     free(ams); 

在我的电脑上,打印结果如下:


可以看到:该结构体一共有 8 个字节(int 型占 4 个字节,指针型占 4 个字节)。

结构体中的 data 成员是一个指针变量,需要单独为它申请一块空间才可以使用。而且在结构体使用之后,需要先释放 data,然后释放结构体指针 ams,顺序不能错。这样使用起来,是不是有点麻烦?

于是,C99 标准就定义了一个语法:flexible array member(柔性数组),直接上代码(下面的代码如果编译时遇到警告,请检查下编译器对这个语法的支持):

  1. // 一个结构体,成员变量是未指明大小的数组 
  2. typedef struct _ArraryMemberStruct_Good_ { 
  3.     int num; 
  4.     char data[]; 
  5. } ArraryMemberStruct_Good; 
  6.  
  7. void demo6_good() 
  8.     // 打印结构体的大小 
  9.     int size = sizeof(ArraryMemberStruct_Good); 
  10.     printf("size = %d \n"size); 
  11.  
  12.     // 为结构体指针分配空间 
  13.     ArraryMemberStruct_Good *ams = (ArraryMemberStruct_Good *)malloc(size + 1024); 
  14.  
  15.     strcpy(ams->data, "hello"); 
  16.     printf("ams->data = %s \n", ams->data); 
  17.  
  18.     // 打印结构体指针、成员变量的地址 
  19.     printf("ams = 0x%x \n", ams); 
  20.     printf("ams->num  = 0x%x \n", &ams->num); 
  21.     printf("ams->data = 0x%x \n", ams->data); 
  22.  
  23.     // 释放空间 
  24.     free(ams); 

打印结果如下:


与第一个例子中有下面几个不同点:

  1. 结构体的大小变成了 4;
  2. 为结构体指针分配空间时,除了结构体本身的大小外,还申请了 data 需要的空间大小;
  3. 不需要为 data 单独分配空间了;
  4. 释放空间时,直接释放结构体指针即可;

是不是用起来简单多了?!这就是柔性数组的好处。

  • 从语法上来说,柔性数组就是指结构体中最后一个元素个数未知的数组,也可以理解为长度为 0,那么就可以让这个结构体称为可变长的。
  • 前面说过,数组名就代表一个地址,是一个不变的地址常量。在结构体中,数组名仅仅是一个符号而已,只代表一个偏移量,不会占用具体的空间。

另外,柔性数组可以是任意类型。这里示例大家多多体会,在很多通讯类的处理场景中,常常见到这种用法。

7. 通过指针来获取结构体中成员变量的偏移量

这个标题读起来似乎有点拗口,拆分一下:在一个结构体变量中,可以利用指针操作的技巧,获取某个成员变量的地址、距离结构体变量的开始地址、之间的偏移量。

在 Linux 内核代码中你可以看到很多地方都利用了这个技巧,代码如下:

  1. #define offsetof(TYPE, MEMBER) ((size_t) &(((TYPE*)0)->MEMBER)) 
  2.  
  3. typedef struct _OffsetStruct_ { 
  4.     int a; 
  5.     int b; 
  6.     int c; 
  7. } OffsetStruct; 
  8.  
  9. void demo7() 
  10.     OffsetStruct os; 
  11.     // 打印结构体变量、成员变量的地址 
  12.     printf("&os = 0x%x \n", &os); 
  13.     printf("&os->a = 0x%x \n", &os.a); 
  14.     printf("&os->b = 0x%x \n", &os.b); 
  15.     printf("&os->c = 0x%x \n", &os.c); 
  16.     printf("===== \n"); 
  17.     // 打印成员变量地址,与结构体变量开始地址,之间的偏移量 
  18.     printf("offset: a = %d \n", (char *)&os.a - (char *)&os); 
  19.     printf("offset: b = %d \n", (char *)&os.b - (char *)&os); 
  20.     printf("offset: c = %d \n", (char *)&os.c - (char *)&os); 
  21.     printf("===== \n"); 
  22.     // 通过指针的强制类型转换来获取偏移量 
  23.     printf("offset: a = %d \n", (size_t) &((OffsetStruct*)0)->a); 
  24.     printf("offset: b = %d \n", (size_t) &((OffsetStruct*)0)->b); 
  25.     printf("offset: c = %d \n", (size_t) &((OffsetStruct*)0)->c); 
  26.     printf("===== \n"); 
  27.     // 利用宏定义来得到成员变量的偏移量 
  28.     printf("offset: a = %d \n", offsetof(OffsetStruct, a)); 
  29.     printf("offset: b = %d \n", offsetof(OffsetStruct, b)); 
  30.     printf("offset: c = %d \n", offsetof(OffsetStruct, c)); 

先来看打印结果:

 

前面 4 行的打印信息不需要解释了,直接看下面这个内存模型即可理解。


下面这个语句也不需要多解释,就是把两个地址的值进行相减,得到距离结构体变量开始地址的偏移量,注意:需要把地址强转成 char* 型之后,才可以相减。

  1. printf("offset: a = %d \n", (char *)&os.a - (char *)&os); 

下面这条语句需要好好理解:

  1. printf("offset: a = %d \n", (size_t) &((OffsetStruct*)0)->a); 

数字 0 看成是一个地址,也就是一个指针。上篇文章解释过,指针就代表内存中的一块空间,至于你把这块空间里的数据看作是什么,这个随便你,你只要告诉编译器,编译器就按照你的意思去操作这些数据。

现在我们把 0 这个地址里的数据看成是一个 OffsetStruct 结构体变量(通过强制转换来告诉编译器),这样就得到了一个 OffsetStruct 结构体指针(下图中绿色横线),然后得到该指针变量中的成员变量 a(蓝色横线),再然后通过取地址符 & 得到 a 的地址(橙色横线),最后把这个地址强转成 size_t 类型(红色横线)。

因为这个结构体指针变量是从 0 地址开始的,因此,成员变量 a 的地址就是 a 距离结构体变量开始地址的偏移量。

上面的描述过程,如果感觉拗口,请结合下面这张图再读几遍:


上面这张图如果能看懂的话,那么最后一种通过宏定义获取偏移量的打印语句也就明白了,无非就是把代码抽象成宏定义了,方便调用:

  1. #define offsetof(TYPE, MEMBER) ((size_t) &(((TYPE*)0)->MEMBER)) 
  2.  
  3. printf("offset: a = %d \n", offsetof(OffsetStruct, a)); 

可能有小伙伴提出:获取这个偏移量有什么用啊?那就请接着看下面的示例 8。

8. 通过结构体中成员变量的指针,来获取该结构体的指针

标题同样比较拗口,直接结合代码来看:

  1. typedef struct _OffsetStruct_ { 
  2.     int a; 
  3.     int b; 
  4.     int c; 
  5. } OffsetStruct; 

假设有一个 OffsetStruct 结构体变量 os,我们只知道 os 中成员变量 c 的地址(指针),那么我们想得到变量 os 的地址(指针),应该怎么做?这就是标题所描述的目的。

下面代码中的宏定义 container_of 同样是来自于 Linux 内核中的(大家平常没事时多挖掘,可以发现很多好东西)。

  1. #define container_of(ptr, type, member) ({ \ 
  2.      const typeof( ((type *)0)->member ) *__mptr = (ptr); \ 
  3.      (type *)( (char *)__mptr - offsetof(type,member) );}) 
  4.  
  5. void demo8() 
  6.     // 下面 3 行仅仅是演示 typeof 关键字的用法 
  7.     int n = 1; 
  8.     typeof(n) m = 2;  // 定义相同类型的变量m 
  9.     printf("n = %d, m = %d \n", n, m);  
  10.  
  11.     // 定义结构体变量,并初始化 
  12.     OffsetStruct os = {1, 2, 3}; 
  13.      
  14.     // 打印结构体变量的地址、成员变量的值(方便后面验证) 
  15.     printf("&os = 0x%x \n", &os); 
  16.     printf("os.a = %d, os.b = %d, os.c = %d \n", os.a, os.b, os.c); 
  17.  
  18.     printf("===== \n"); 
  19.      
  20.     // 假设只知道某个成员变量的地址 
  21.     int *pc = &os.c; 
  22.     OffsetStruct *p = NULL
  23.      
  24.     // 根据成员变量的地址,得到结构体变量的地址 
  25.     p = container_of(pc, OffsetStruct, c); 
  26.      
  27.     // 打印指针的地址、成员变量的值 
  28.     printf("p = 0x%x \n", p); 
  29.     printf("p->a = %d, p->b = %d, p->c = %d \n", p->a, p->b, p->c); 

先看打印结果:


首先要清楚宏定义中参数的类型:

  1. ptr: 成员变量的指针;
  2. type: 结构体类型;
  3. member:成员变量的名称;

这里的重点就是理解宏定义 container_of,结合下面这张图,把宏定义拆开来进行描述:


宏定义中的第 1 条语句分析:

  1. 绿色横线:把数字 0 看成是一个指针,强转成结构体 type 类型;
  2. 蓝色横线:获取该结构体指针中的成员变量 member;
  3. 橙色横线:利用 typeof 关键字,获取该 member 的类型,然后定义这个类型的一个指针变量 __mptr;
  4. 红色横线:把宏参数 ptr 赋值给 __mptr 变量;

宏定义中的第 2 条语句分析:

  1. 绿色横线:利用 demo7 中的 offset 宏定义,得到成员变量 member 距离结构体变量开始地址的偏移量,而这个成员变量指针刚才已经知道了,就是 __mptr;
  2. 蓝色横线:把 __mptr 这个地址,减去它自己距离结构体变量开始地址的偏移量,就得到了该结构体变量的开始地址;
  3. 橙色横线:最后把这个指针(此时是 char* 型),强转成结构体 type 类型的指针;

三、总结

上面这 8 个关于指针的用法掌握之后,再去处理子字符、数组、链表等数据,基本上就是熟练度和工作量的问题了。希望大家都能用好指针这个神器,提高程序程序执行效率。

 

责任编辑:姜华 来源: IOT物联网小镇
相关推荐

2021-09-09 17:05:36

C++智能指针语言

2024-08-13 08:22:04

缓存机制C#内存缓存工具

2010-01-28 13:57:19

C++指针基础

2022-05-30 00:04:16

开源Github技巧

2021-06-16 17:46:55

函数指针结构

2017-07-19 16:34:44

笔记本拆解技巧

2018-04-10 14:36:18

数据库MySQL优化技巧

2017-03-07 14:26:19

Eclipse技巧效率

2024-08-06 12:35:42

C#代码重构

2023-11-28 12:19:49

C++函数指针

2016-12-13 17:02:49

androidjava移动应用开发

2024-04-12 08:28:38

优化查询语句PostgreSQL索引

2020-03-23 10:51:40

面试技巧技术

2021-05-11 12:30:21

PyTorch代码Python

2024-09-12 17:39:27

2021-01-12 11:44:48

java垃圾回收

2023-02-17 08:14:29

C语言C技巧内存

2023-12-31 16:23:39

KubernetesPod容器

2023-11-27 18:01:17

MySQL技巧

2022-10-08 00:03:00

Debug技巧调试
点赞
收藏

51CTO技术栈公众号