C 语言精髓之变参函数

我们以 printf 这个 very 熟悉的函数为例，来分析一下变参函数。先看下 printf 函数的定义：

int printf(const char *fmt, ...)
{
    int i;
    int len;
    /* va_list 即 char * */
    va_list args;

    va_start(args, fmt);

    /* 内部使用了 va_arg() */
    len = vsprintf(g_PCOutBuf,fmt,args);

    va_end(args);

    for (i = 0; i < strlen(g_PCOutBuf); i++)
    {
        putc(g_PCOutBuf[i]);
    }

    return len;
}

什么是变参函数？

可变参数函数的原型声明为 type VAFunction(type arg1, type arg2, … )；

参数可以分为两部分：个数确定的固定参数和个数可变的可选参数。函数至少需要一个固定参数，固定参数的声明和普通函数一样；可选参数由于个数不确定，声明时用 "..." 表示。固定参数和可选参数共同构成一个函数的参数列表。

printf 函数涉及了以下几个重要的宏：

typedef char *   va_list;

/*
 * Storage alignment properties
 */

#define _AUPBND (sizeof (acpi_native_int) - 1) //acpi_native_int 为 4 字节（根据机器字长而定）

#define _ADNBND (sizeof (acpi_native_int) - 1) 

/*
 * Variable argument list macro definitions
 */

#define _bnd(X, bnd) (((sizeof (X)) + (bnd)) & (~(bnd)))
#define va_arg(ap, T) (*(T *)(((ap) += (_bnd (T, _AUPBND))) - (_bnd (T,_ADNBND))))
#define va_end(ap) (void) 0
#define va_start(ap, A) (void) ((ap) = (((char *) &(A)) + (_bnd (A,_AUPBND))))

分析以下三个宏的作用

1) va_start 宏

作用： 根据 A 取得可变参数表的首指针并赋值给 ap。

原理： 根据最后一个固定参数 A 的地址 + 第一个变参对 A 的偏移地址，然后赋值给 ap，这样 ap 就是可变参数表的首地址(函数传递的参数会从右向左依次入栈，并且 ARM 的栈为降栈，所以参数 A 的地址最低)。

2) va_arg 宏

作用： 指取出当前 ap 所指的可变参数并将 ap 指针指向下一可变参数。

3) va_end 宏

作用： 结束可变参数的获取，与 va_start 对应使用。

堆栈中，各个函数的分布情况是倒序的，即最后一个参数在列表中地址最高部分，第一个参数在列表地址的最低部分。参数在堆栈中的分布情况如下:

    **************************
       最后一个参数
       倒数第二个参数
       ...
       第一个参数
       函数返回地址
       函数的局部变量
    **************************

得到可变参数个数的三种办法:
1) 函数的第一个参数，指定后续的参数个数，如 func(int num,...)；
2) 根据隐含参数，判断参数个数,如 printf 系列的，通过字符串中 % 的个数判断；
3) 特殊情况下(如参数都是不大于 0xFFFF 的 int)，可以一直向低处访问堆栈，直到返回地址。

而 _bnd(X, bnd) 宏就是以 4 字节对齐的变量 X 的大小。

自己实现一个简单的 printf 函数

#include <stdio.h>
#include <string.h>
#include <stdarg.h>
#include <stdlib.h>

void print(char *fmt, ...)
{
	va_list ptr;
	va_list temp_ptr = NULL;
	/* 用于存储最终转换的结果 */
	char buf[100] = {0};
	/* 用于存储临时转换的结果 */
	char temp_buf[50] = {0};
	unsigned char index = 0;
	unsigned char len, arg_len;

	len = strlen(fmt);

    /* 得到可变参数的首地址 */
	va_start(ptr, fmt);

	for(; *fmt; fmt++){
		if(*fmt == '%'){
			switch(*++fmt){
				case 'd':
				case 'D':
					sprintf(temp_buf, "%d", va_arg(ptr, int));
					temp_ptr = temp_buf;
					break;
				case 's':
				case 'S':
				    /* 取出当前变量，并将指针指向下一个可变参数 */
					temp_ptr = va_arg(ptr, char*);
					break;
			}
			arg_len = strlen(temp_ptr);
			strcat(buf+index, temp_ptr);
			index += arg_len;
		}else{
			buf[index] = *fmt;
			index++;
		}
	}

    /* 结束取参 */
	va_end(ptr);

    /* 输出最终转换结果 */
	puts(buf);
}

int main()
{
	print("My name is %s and my height is %d cm.", "Lance#", 178);

	return 0;
}

程序运行结果：

My name is Lance# and my height is 178 cm.