C里面的变长参数
C里面的变长参数
原文来自个人博客(求访问/关注/收藏): https://bbing.com.cn/ cnblog个人博客不定期转载
stdarg.h
这里用到的是stdarg.h这个库, 可以在C语言里面实现可变长参数.
当然C++会简单得多, C++11之后的模板原生支持可变长参数.
几个函数va_list、va_start、va_arg、va_end，定义在stdarg.h
内存结构
先需要理解C/C++函数入参的顺序.
按照以下的demo, 将其翻译成汇编代码.
#include <iostream>
using namespace std;
int sum(const int &a, const int &b, const int &c)
{
int d = 0;
d = a + b + c;
return d;
}
int main()
{
int s = sum(1, 2, 3);
cout << s << endl;
return 1;
}
首先是main函数, 主体部分的汇编
mov DWORD PTR [rbp-0x10],0x3
mov DWORD PTR [rbp-0xc],0x2
mov DWORD PTR [rbp-0x8],0x1
lea rdx,[rbp-0x10]
lea rcx,[rbp-0xc]
lea rax,[rbp-0x8]
mov rsi,rcx
mov rdi,rax
call 401172 <sum(int const&, int const&, int const&)>
可以看到, main函数调用了sum函数, 首先搜获取三个参数, 1, 2, 3; 获取顺序是从右往左的. 先获取了3再是2再是1.
之后是一些操作将这三个参数从内存放到寄存器(Why?), 然后调用sum函数.
sum函数的汇编代码如下
mov QWORD PTR [rbp-0x18],rdi
mov QWORD PTR [rbp-0x20],rsi
mov QWORD PTR [rbp-0x28],rdx
mov DWORD PTR [rbp-0x4],0x0
mov rax,QWORD PTR [rbp-0x18]
mov edx,DWORD PTR [rax]
mov rax,QWORD PTR [rbp-0x20]
mov eax,DWORD PTR [rax]
add edx,eax
mov rax,QWORD PTR [rbp-0x28]
mov eax,DWORD PTR [rax]
add eax,edx
mov DWORD PTR [rbp-0x4],eax
mov eax,DWORD PTR [rbp-0x4]
首先是从寄存器取值, 放到内存, 然后进入函数, 执行函数内部的操作, 最后将计算结果从内存放到寄存器. 这里注意一下型参的顺序.
rdi, rsi, rdx对应的内存分别是a, b, c.
所以, 可以对上面的demo, 可以知道其内存分布是
- 对函数本体:
从低地址到高地址, 型参按照从左往右的顺序, 函数体按照从上往下的顺序执行;
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Aa7wKQB7-1611797623781)(https://s3.ax1x.com/2021/01/27/szNKCF.png "函数本体")]
- 对函数调用:
从低地址到高地址, 实参按照从右往左的顺序, 函数体按照从上往下的顺序执行;
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-62CNYxtK-1611797623782)(https://s3.ax1x.com/2021/01/27/szNM34.png "函数调用")]
内存对齐
源码头文件中，注意一下这个宏，内存对齐作用 看这里：
#define __va_rounded_size(TYPE) \
(((sizeof (TYPE) + sizeof (int) - 1) / sizeof (int)) * sizeof (int))
- TYPE size >= 4，偏移量=(sizeof(TYPE) / 4) * 4
- TYPE size < 4, 偏移量=4
所以是按4Byte，32位对齐。
va_list
typedef char *va_list;
仅是一个指针, 这是一个适用于 va_start()、va_arg() 和 va_end() 这三个宏存储信息的类型。
va_start
将AP指向第一个参数的下一个参数的地址.
#ifndef __sparc__
#define va_start(AP, LASTARG) \
(AP = ((char *) &(LASTARG) + __va_rounded_size (LASTARG)))
#else
#define va_start(AP, LASTARG) \
(__builtin_saveregs (), \
AP = ((char *) &(LASTARG) + __va_rounded_size (LASTARG)))
#endif
va_arg
AP指向下一个参数, 同时返回上一个参数的内容.
#define va_arg(AP, TYPE) \
(AP += __va_rounded_size (TYPE), \
*((TYPE *) (AP - __va_rounded_size (TYPE))))
va_end
将AP指针置空, 做保护用.
#define va_end(AP)
//有些代码中定义为
#define va_end(ap) ( ap = (va_list)0 )
用例
int sum(int count, ...)
{
va_list vl;
int sum = 0;
va_start(vl, count);
for (int i = 0; i < count; ++i)
{
sum += va_arg(vl, int);
}
va_end(vl);
return sum;
}
结合开头讲述的内存分布就不难理解, va_list是一个指针, 型参都是在连续内存中的.
va_start(vl, count)的时候, 指向了count的下一个指针(count地址, 加上count的size).
va_arg(vl, int)的时候, 先是将vl指向下一个地址, 然后再返回上一个地址的值.
