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.

秒客网