我們?cè)贑語(yǔ)言編程中會(huì)遇到一些參數(shù)個(gè)數(shù)可變的函數(shù),例如printf()這個(gè)函數(shù),這里將介紹可變函數(shù)的寫法以及原理.
* 1. 可變參數(shù)的宏
一般在調(diào)試打印Debug 信息的時(shí)候, 需要可變參數(shù)的宏. 從C99開(kāi)始可以使編譯器標(biāo)準(zhǔn)支持可變參數(shù)宏(variadic macros), 另外GCC 也支持可變參數(shù)宏, 但是兩種在細(xì)節(jié)上可能存在區(qū)別.
1. __VA_ARGS__
__VA_ARGS__ 將"..." 傳遞給宏.如
#define debug(format, ...) fprintf(stderr, fmt, __VA_ARGS__)
在GCC中也支持這類表示, 但是在G++ 中不支持這個(gè)表示.
2. GCC 的復(fù)雜宏
GCC使用一種不同的語(yǔ)法從而可以使你可以給可變參數(shù)一個(gè)名字,如同其它參數(shù)一樣。
#define debug(format, args...) fprintf (stderr, format, args)
這和上面舉的那個(gè)定義的宏例子是完全一樣的,但是這么寫可讀性更強(qiáng)并且更容易進(jìn)行描述。
3. ##__VA_ARGS__
上面兩個(gè)定義的宏, 如果出現(xiàn)debug("A Message") 的時(shí)候, 由于宏展開(kāi)后有個(gè)多余的逗號(hào), 所以將導(dǎo)致編譯錯(cuò)誤. 為了解決這個(gè)問(wèn)題,CPP使用一個(gè)特殊的‘##’操作。
#define debug(format, ...) fprintf (stderr, format, ## __VA_ARGS__)
這里,如果可變參數(shù)被忽略或?yàn)榭眨?#8216;##’操作將使預(yù)處理器(preprocessor)去除掉它前面的那個(gè)逗號(hào)。如果你在宏調(diào)用時(shí),確實(shí)提供了一些可變參數(shù),GNU CPP也會(huì)工作正常,它會(huì)把這些可變參數(shù)放到逗號(hào)的后面。
4. 其他方法
一種流行的技巧是用一個(gè)單獨(dú)的用括弧括起來(lái)的的 "參數(shù)" 定義和調(diào)用宏, 參數(shù)在宏擴(kuò)展的時(shí)候成為類似 printf() 那樣的函數(shù)的整個(gè)參數(shù)列表。
#define DEBUG(args) (printf("DEBUG: "), printf(args))
* 2. 可變參數(shù)的函數(shù)
寫可變參數(shù)的C函數(shù)要在程序中用到以下這些宏:
void va_start( va_list arg_ptr, prev_param )
type va_arg( va_list arg_ptr, type )
void va_end( va_list arg_ptr )
va在這里是variable-argument(可變參數(shù))的意思,這些宏定義在stdarg.h中.下面我們寫一個(gè)簡(jiǎn)單的可變參數(shù)的函數(shù),該函數(shù)至少有一個(gè)整數(shù)參數(shù),第二個(gè)參數(shù)也是整數(shù),是可選的.函數(shù)只是打印這兩個(gè)參數(shù)的值.
void simple_va_fun(int i, ...)
{
va_list arg_ptr;
int j=0;
va_start(arg_ptr, i);
j=va_arg(arg_ptr, int);
va_end(arg_ptr);
printf("%d %d\n", i, j);
return;
}
在程序中可以這樣調(diào)用:
simple_va_fun(100);
simple_va_fun(100,200);
從這個(gè)函數(shù)的實(shí)現(xiàn)可以看到,使用可變參數(shù)應(yīng)該有以下步驟:
1)首先在函數(shù)里定義一個(gè)va_list型的變量,這里是arg_ptr,這個(gè)變量是指向參數(shù)的指針.
2)然后用va_start宏初始化變量arg_ptr,這個(gè)宏的第二個(gè)參數(shù)是第一個(gè)可變參數(shù)的前一個(gè)參數(shù),是一個(gè)固定的參數(shù).
3)然后用va_arg返回可變的參數(shù),并賦值給整數(shù)j. va_arg的第二個(gè)參數(shù)是你要返回的參數(shù)的類型,這里是int型.
4)最后用va_end宏結(jié)束可變參數(shù)的獲取.然后你就可以在函數(shù)里使用第二個(gè)參數(shù)了.如果函數(shù)有多個(gè)可變參數(shù)的,依次調(diào)用va_arg獲取各個(gè)參數(shù).
如果我們用下面三種方法調(diào)用的話,都是合法的,但結(jié)果卻不一樣:
1)simple_va_fun(100);
結(jié)果是:100 -123456789(會(huì)變的值)
2)simple_va_fun(100,200);
結(jié)果是:100 200
3)simple_va_fun(100,200,300);
結(jié)果是:100 200
我們看到第一種調(diào)用有錯(cuò)誤,第二種調(diào)用正確,第三種調(diào)用盡管結(jié)果正確,但和我們函數(shù)最初的設(shè)計(jì)有沖突.下面一節(jié)我們探討出現(xiàn)這些結(jié)果的原因和可變參數(shù)在編譯器中是如何處理的.
* 3. 可變參數(shù)函數(shù)原理
va_start,va_arg,va_end是在stdarg.h中被定義成宏的,由于硬件平臺(tái)的不同,編譯器的不同,所以定義的宏也有所不同,下面以VC++中stdarg.h里x86平臺(tái)的宏定義摘錄如下:
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,t) ( *(t *)((ap += _INTSIZEOF(t)) - _INTSIZEOF(t)) )
#define va_end(ap) ( ap = (va_list)0 )
定義_INTSIZEOF(n)主要是為了內(nèi)存對(duì)齊,C語(yǔ)言的函數(shù)是從右向左壓入堆棧的(設(shè)數(shù)據(jù)進(jìn)棧方向?yàn)閺母叩刂废虻偷刂钒l(fā)展,即首先壓入的數(shù)據(jù)在高地址). 下圖是函數(shù)的參數(shù)在堆棧中的分布位置:
低地址 |-----------------------------|<-- &v
|第n-1個(gè)參數(shù)(最后一個(gè)固定參數(shù))|
|-----------------------------|<--va_start后ap指向
|第n個(gè)參數(shù)(第一個(gè)可變參數(shù)) |
|-----------------------------|
|....... |
|-----------------------------|
|函數(shù)返回地址 |
高地址 |-----------------------------|
1. va_list 被定義為char *
2. va_start 將地址ap定義為 &v+_INTSIZEOF(v),而&v是固定參數(shù)在堆棧的地址,所以va_start(ap, v)以后,ap指向第一個(gè)可變參數(shù)在堆棧的地址
3. va_arg 取得類型t的可變參數(shù)值,以int型為例,va_arg取int型的返回值:
j= ( *(int*)((ap += _INTSIZEOF(int))-_INTSIZEOF(int)) );
4. va_end 使ap不再指向堆棧,而是跟NULL一樣.這樣編譯器不會(huì)為va_end產(chǎn)生代碼.
在不同的操作系統(tǒng)和硬件平臺(tái)的定義有些不同,但原理卻是相似的.
* 4. 小結(jié)
對(duì)于可變參數(shù)的函數(shù),因?yàn)関a_start, va_arg, va_end等定義成宏,所以它顯得很愚蠢,可變參數(shù)的類型和個(gè)數(shù)需要在該函數(shù)中由程序代碼控制;另外,編譯器對(duì)可變參數(shù)的函數(shù)的原型檢查不夠嚴(yán)格,對(duì)編程查錯(cuò)不利.
所以我們寫一個(gè)可變函數(shù)的C函數(shù)時(shí),有利也有弊,所以在不必要的場(chǎng)合,無(wú)需用到可變參數(shù).如果在C++里,我們應(yīng)該利用C++的多態(tài)性來(lái)實(shí)現(xiàn)可變參數(shù)的功能,盡量避免用C語(yǔ)言的方式來(lái)實(shí)現(xiàn).
* 5. 附一些代碼
#define debug(format, ...) fprintf(stderr, fmt, __VA_ARGS__)
#define debug(format, args...) fprintf (stderr, format, args)
#define debug(format, ...) fprintf (stderr, format, ## __VA_ARGS__)
// 使用va... 實(shí)現(xiàn)
void debug(const char *fmt, ...)
{
int nBuf;
char szBuffer[1024];
va_list args;
va_start(args, fmt);
nBuf = vsprintf(szBuffer, fmt, args) ;
assert(nBuf >= 0);
printf("QDOGC ERROR:%s\n",szBuffer);
va_end(args);
}