目录
01. format属性声明
GNU 通过 attribute 扩展的 format 属性,用来指定变参函数的参数格式检查。
用法如下:
__attribute__(( format (archetype, string-index, first-to-check)))
void LOG(const char *fmt, ...) __attribute__((format(printf,1,2)));
我们经常实现一些自己的打印调试函数。这些打印函数往往是变参函数,那编译器编译程序时,怎么知道我们的参数格式是否正确呢?因为我们实现的是变参函数,参数的个数和格式都不确定。所以编译器表示压力很大,不知道该如何处理。
attribute 的format属性这时候就自带 BGM,隆重出场了。如上面的示例代码,我们定义一个 LOG 变参函数,用来实现打印功能。那编译器编译程序时,如何检查我们参数的格式是否正确呢?其实很简单,通过给 LOG 函数添加 attribute((format(printf,1,2))) 这个属性声明,就是告诉编译器:你知道printf函数不?你怎么对这个函数参数格式检查的,就按同样的方法,对 LOG 函数进行检查。
属性 format(printf,1,2) 有三个参数。第一个参数 printf 是告诉编译器,按照 printf 函数的检查标准来检查;第2个参数表示在 LOG 函数所有的参数列表中,格式字符串的位置索引;第3个参数是告诉编译器要检查的参数的起始位置。
LOG("I am tom\n");
LOG("I am tom, I have %d houses!\n",0);
LOG("I am tom, I have %d houses! %d cars\n",0,0);
上面代码,是我们的 LOG 函数使用示例。变参函数,其参数个数跟 printf 函数一样,是不固定的。那么编译器如何检查我们的打印格式是否正确呢?很简单,我们只需要将格式字符串的位置告诉编译器就可以了,比如在第2行代码中:
LOG("I am tom, I have %d houses!\n",0);
在这个 LOG 函数中有2个参数,第一个是格式字符串,第2个是要打印的一个常量值0,用来匹配格式字符串中的格式符。
什么是格式字符串呢?顾名思义,如果一个字符串中含有格式符,那这个字符串就是格式字符串。比如这个格式字符串:“I am tom, I have %d houses!\n”,里面含有格式符%,我们也可以叫它占位符。打印的时候,后面变参的值会代替这个占位符,在屏幕上显示出来。
我们通过 format(printf,1,2) 属性声明,告诉编译器:LOG 函数的参数,格式字符串的位置在所有参数列表中的索引是1,即第一个参数;要编译器帮忙检查的参数,在所有的参数列表里索引是2。知道了 LOG 参数列表中格式字符串的位置和要检查的参数位置,编译器就会按照检查 printf 的格式打印一样,对 LOG 函数进行参数检查。
如果我们的 LOG 函数定义为下面形式:
void LOG(int num, char *fmt, ...) __attribute__((format(printf,2,3)));
在这个函数定义中,多了一个参数 num,格式字符串在参数列表中的位置发生了变化(在所有的参数列表中,索引为2),要检查的第一个变参的位置也发生了变化(索引为3),那我们使用 format 属性声明时,就要写成 format(printf,2,3) 的形式了。
以上就是 format 属性的使用方法。
02. 变参函数的设计思路
变参函数,顾名思义,跟 printf 函数一样:参数的个数、类型都不固定。我们在函数体内因为预先不知道传进来的参数类型和个数,所以实现起来会稍微麻烦一点。首先要解析传进来的实参,保存起来,然后才能接着像普通函数一样,对实参进行处理。
我们接下来,就定义一个变参函数,实现的功能很简单,即打印传进来的实参值。
程序示例
#include <stdio.h>
void fun(int count, ...)
{
int i = 0;
int *args = NULL;args = &count + 1;
for (i = 0; i < count; i++)
{
printf("args: %d %p\n", *args, args);
args++;
}
}int main(void)
{fun(5, 1, 2, 3, 4, 5);
return 0;
}
测试结果
# 根据平台不同,可能结果不同
deng@itcast:~/tmp$ ./a.out
args: 832 0x7ffc05619808
args: 832 0x7ffc05619804
args: 832 0x7ffc05619800
args: 21940 0x7ffc056197fc
args: 975176187 0x7ffc056197f8
变参函数的参数存储其实跟 main 函数的参数存储很像,由一个连续的参数列表组成,列表里存放的是每个参数的地址。在上面的函数中,有一个固定的参数 count,这个固定参数的存储地址后面,就是一系列参数的指针。在 fun函数中,先获取 count 参数地址,然后使用 &count + 1 就可以获取下一个参数的指针地址,使用指针变量 args 保存这个地址,并依次访问下一个地址,就可以直接打印传进来的各个实参值了。
上面的程序使用一个 int * 的指针变量依次去访问实参列表。我们接下来把程序改进一下,使用 char * 类型的指针来实现这个功能,使之兼容更多的参数类型。
程序示例
#include <stdio.h>
void fun(int count, ...)
{
int i = 0;
char *args = NULL;args = (void*)&count + 4;
for (i = 0; i < count; i++)
{
printf("args: %d %p\n", *(int*)args, args);
args += 4;
}
}int main(void)
{fun(5, 1, 2, 3, 4, 5);
return 0;
}
03. 变参函数宏
对于变参函数,编译器或计算机系统一般会提供一些宏给程序员使用,用来解析函数的参数。这样程序员就不用自己解析参数了,直接使用封装好的宏即可。编译器提供的宏有:
va_list:定义在编译器头文件中 typedef char* va_list;。
va_start(args,fmt):根据参数 fmt 的地址,获取 fmt 后面参数的地址,并保存在 args 指针变量中。
va_end(args):释放 args 指针,将其赋值为 NULL。有了这些宏,我们的工作就简化了很多。我们就不用撸起袖子,自己解析了。
程序示例
#include <stdio.h>
#include <stdarg.h>void fun(int count, ...)
{
va_list args;va_start(args, count);
for (int i = 0; i < count; i++)
{
printf("*args = %d\n", va_arg(args, int));
}va_end(args);
}int main(void)
{fun(5, 1, 2, 3, 4, 5);
return 0;
}
执行结果
deng@itcast:~/tmp$ gcc test.c
deng@itcast:~/tmp$ ./a.out
*args = 1
*args = 2
*args = 3
*args = 4
*args = 5
我们使用编译器提供的三个宏,省去了解析参数的麻烦。但打印的时候,我们还必须自己实现。在 V4.0 版本中,我们继续改进,使用 vprintf 函数实现我们的打印功能。vprintf 函数的声明在 stdio.h 头文件中。
# if !(__USE_FORTIFY_LEVEL > 0 && defined __fortify_function)
/* Write formatted output to stdout from argument list ARG. */
__STDIO_INLINE int
vprintf (const char *__restrict __fmt, __gnuc_va_list __arg)
{
return vfprintf (stdout, __fmt, __arg);
}
# endif
printf 函数有2个参数,一个是格式字符串指针,一个是变参列表。在下面的程序里,我们可以将,使用 va_start 解析后的变参列表,直接传递给 vprintf 函数,实现打印功能。
程序示例
#include <stdio.h>
#include <stdarg.h>void fun(char *fmt, ...)
{
va_list args;va_start(args, fmt);
vprintf(fmt, args);
va_end(args);
}int main(void)
{
int n = 88;fun("hello world %d\n", n);
return 0;
}
执行结果
deng@itcast:~/tmp$ ./a.out
hello world 88
上一个示例程序基本上实现了跟 printf() 函数相同的功能:支持变参,支持多种格式的数据打印。接下来,我们还需要对其添加 format 属性声明,让编译器在编译时,像检查 printf 一样,检查 fun() 函数的参数格式。
程序示例
#include <stdio.h>
#include <stdarg.h>void __attribute__((format(printf,1,2))) fun(char *fmt, ...)
{
va_list args;va_start(args, fmt);
vprintf(fmt, args);
va_end(args);
}int main(void)
{
int n = 88;fun("hello world %d\n", n);
return 0;
}
执行结果
deng@itcast:~/tmp$ ./a.out
hello world 88
04. 应用示例
在调试一个模块,或者一个系统,有好多个文件。如果你在每个文件里添加 printf 打印,调试完成后再删掉,是不是很麻烦?我们自己实现的打印函数,通过一个宏开关,就可以直接关掉或打开,比较方便。比如下面的代码。
输出日志信息程序
#include <stdio.h>
#include <stdarg.h>#define DEBUG
void __attribute__((format(printf,1,2))) LOG(char *fmt, ...)
{
#ifdef DEBUG
va_list args;va_start(args, fmt);
vprintf(fmt, args);
va_end(args);
#endif
}int main(void)
{
int n = 88;LOG("hello world %d\n", n);
return 0;
}
执行结果
deng@itcast:~/tmp$ ./a.out
hello world 88
deng@itcast:~/tmp$
当我们定义一个 DEBUG 宏时,LOG 函数实现普通的打印功能;当这个 DEBUG 宏没有定义,LOG 函数就是个空函数。通过这个宏,我们就实现了打印函数的开关功能,在实际调试中比较实用,非常方便。在 Linux 内核的各个模块中,你会经常看到大量的自定义打印函数或宏,如 pr_debug、pr_info 等。
除此之外,你可以通过宏,设置一些打印等级。比如可以分为 ERROR、WARNNING、INFO、LOG 等级,根据你设置的打印等级,模块打印的 log 信息也会不一样。