可变参数
可变参数顾名思义,就是函数参数列表的参数数量是可变的,我们常见的使用可变参数的函数有 scanf 和 printf
可变参数的使用
我们先看一个带有可变参数的函数:
void func(int num, ...)
{
;
}
第一个参数 num 还是一个固定参数,后面的 ... 就表示可变参数。通常来说,我们需要在传参时传一个固定参数,来标识可变参数的个数,这对我们使用可变参数的宏是很重要的。
下面是使用可变参数的几个相关宏:
typedef char * va_list;
#define _INTSIZEOF(n) ( (sizeof(n)+sizeof(int)-1) & ~(sizeof(int)-1) )
#define va_start(ap,v) ( ap = (va_list)&v + _INTSIZEOF(v) )
#define va_arg(ap, type) ( *(type *)((ap += _INTSIZEOF(type)) - _INTSIZEOF(type)) )
#define va_end(ap) ( ap = (va_list)0 )
_INTSIZEOF(n)
_INTSIZEOF 宏考虑到某些系统需要内存地址对齐。从宏名看应按照sizeof(int) 即栈粒度对齐,参数在内存中的地址均为sizeof(int)=4 的倍数。
例如,若1≤sizeof(n)≤4 ,则_INTSIZEOF(n)=4 ;若5≤sizeof(n)≤8 ,则_INTSIZEOF(n)=8 。
va_start(ap,v)
va_start 初始化可变参数列表 ap ,v 是可变参数的前一个参数,(va_list)&v + _INTSIZEOF(v) 即取出 v 的地址,再跳过该地址,到达下一个地址,即参数列表的第一个地址
va_arg(ap, type)
返回 ap 当前指向的变量,ap 指向下一个变量。首先 ap += _INTSIZEOF(type) ,即 ap 指向下一个变量,然后返回更新后 ap - _INTSIZEOF(type) 的结果,即原 ap 指向的变量
va_end(ap)
va_end 宏使 ap 不再指向有效的内存地址。该宏的某些实现定义为((void*)0),编译时不会为其产生代码,调用与否并无区别。但某些实现中 va_end 宏用于在函数返回前完成一些必要的清理工作:如 va_start 宏可能以某种方式修改栈,导致返回操作无法完成,va_end 宏可将有关修改复原;又如 va_start 宏可能为参数列表动态分配内存以便于遍历,va_end 宏可释放此内存。因此,从使用 va_start 宏的函数中退出之前,必须调用一次 va_end 宏。
使用技巧
变参宏无法智能识别可变参数的数目和类型,因此实现变参函数时需自行判断可变参数的数目和类型。所以我们就要想一些办法,比如
- 显式提供变参数目或设定遍历结束条件
- 显式提供变参类型枚举值,或在固定参数中包含足够的类型信息(如 printf 函数通过分析 format 字符串即可确定各变参类型)
- 主调函数和被调函数可约定变参的数目和类型
- …
示例
函数通过传入固定参数指定可变参数的个数,获取所有参数相加的结果:
int sum(int num, ...)
{
va_list ap;
va_start(ap, num);
int sum = 0;
for (int i = 0; i < num; ++i)
{
sum += va_arg(ap, int);
}
va_end(ap);
return sum;
}
int main()
{
int ret = sum(3, 2, 1, 2);
printf("%d\n", ret);
return 0;
}
|