自定义 vargs 可变参数实现

c语言的va_list可以传入多个参数, 而它的实现也是很简单, 很容易理解, 就是3个宏

va_list实现

typedef char * va_list;

va_list是一个char *字符指针.

va_start实现

#define SIZEOF(n) ((sizeof(n) + sizeof(int) - 1) & ~(sizeof(int) - 1))

#define va_start(va, fmt) (va = (va_list)&fmt + SIZEOF(fmt))

SIZEOF的宏定义是为了保证变量在32位上是以4字节进行对齐, 64位上是以8字节对齐的, 很容易能看出来这个对齐的字节大小正好是指针的大小. 至于为什么是指针的大小这就要说一下关于函数参数参数的方式了.

函数传入的参数一般都是从右往左的压栈, 第一个参数被压在最上面, 每个参数 压入的都是指针大小的字节, 所以访问下一个参数, 需要在上一个参数上面加上指针的大小就行了.

所以va_start函数的功能也就不难明白了, 在当前参数前面加上一个指针的大小就能访问下一个参数. 同样, 根据上面的传入参数的解释也能推断怎么访问下一个数据了.

va_arg实现

#define va_arg(va, type) (*(type *)((va += SIZEOF(type)) - SIZEOF(type)))

不容易理解的应该是(va += SIZEOF(type)) - SIZEOF(type)这里, 为什么相加之后又剪掉, 这里主要是让指针先指向下一个内存单元, 实现一个++的操作, 然后减去一个单元的大小回到现在访问的内存单元. 虽然返回的是当前单元的内存地址, 但是我们的指针已经指向了下一个单元的位置了. 这里其实很巧妙的进行了迭代 , 我们不再手动去指向下一个内存地址, va_arg函数自动完成了这一步. 这就是很巧妙的操作.

va_end实现

#define va_end(va) (va = (va_list)0)

这个就直接是将指针指向NULL就行了, 安全.