(peakflys注:本篇文章轉(zhuǎn)自 http://blogread.cn/it/article/5907?f=hot3)
linux內(nèi)核和其他一些開源的代碼中,經(jīng)常會(huì)遇到這樣的代碼:
這樣的代碼一看就不是一個(gè)循環(huán),do..while表面上在這里一點(diǎn)意義都沒有,那么為什么要這么用呢?
實(shí)際上,do{...}while(0)的作用遠(yuǎn)大于美化你的代碼。查了些資料,總結(jié)起來這樣寫主要有以下幾點(diǎn)好處:
1、輔助定義復(fù)雜的宏,避免引用的時(shí)候出錯(cuò):
舉例來說,假設(shè)你需要定義這樣一個(gè)宏:
#define DOSOMETHING()\\ foo1();\\ foo2(); |
這個(gè)宏的本意是,當(dāng)調(diào)用DOSOMETHING()時(shí),函數(shù)foo1()和foo2()都會(huì)被調(diào)用。但是如果你在調(diào)用的時(shí)候這么寫:
因?yàn)楹暝陬A(yù)處理的時(shí)候會(huì)直接被展開,你實(shí)際上寫的代碼是這個(gè)樣子的:
這就出現(xiàn)了問題,因?yàn)闊o論a是否大于0,foo2()都會(huì)被執(zhí)行,導(dǎo)致程序出錯(cuò)。
那么僅僅使用{}將foo1()和foo2()包起來行么?
我們在寫代碼的時(shí)候都習(xí)慣在語句右面加上分號,如果在宏中使用{},代碼里就相當(dāng)于這樣寫了:“{...};”,展開后就是這個(gè)樣子:
if(a>0) { foo1(); foo2(); }; |
這樣甚至不會(huì)編譯通過。所以,很多人才采用了do{...}while(0);
#define DOSOMETHING() \\ do{ \\ foo1();\\ foo2();\\ }while(0)\\ ... if(a>0) DOSOMETHING(); ... |
這樣,宏被展開后,才會(huì)保留初始的語義。GCC提供了Statement-Expressions用以替代do{...}while(0); 所以你也可以這樣定義宏:
#define DOSOMETHING() ({\\ foo1(); \\ foo2(); \\ }) |
2、避免使用goto對程序流進(jìn)行統(tǒng)一的控制:
有些函數(shù)中,在函數(shù)return之前我們經(jīng)常會(huì)進(jìn)行一些收尾的工作,比如free掉一塊函數(shù)開始malloc的內(nèi)存,goto一直都是一個(gè)比較簡便的方法:
int foo() { somestruct* ptr = malloc(...); dosomething...; if(error) { goto END; } dosomething...; if(error) { goto END; } dosomething...; END: free(ptr); return 0; } |
由于goto不符合軟件工程的結(jié)構(gòu)化,而且有可能使得代碼難懂,所以很多人都不倡導(dǎo)使用,那這個(gè)時(shí)候就可以用do{}while(0)來進(jìn)行統(tǒng)一的管理:
int foo() { somestruct* ptr = malloc(...); do{ dosomething...; if(error) { break; } dosomething...; if(error) { break; } dosomething...; }while(0); free(ptr); return 0; } |
這里將函數(shù)主體使用do()while(0)包含起來,使用break來代替goto,后續(xù)的處理工作在while之后,就能夠達(dá)到同樣的效果。
3、避免空宏引起的warning
內(nèi)核中由于不同架構(gòu)的限制,很多時(shí)候會(huì)用到空宏,在編譯的時(shí)候,空宏會(huì)給出warning,為了避免這樣的warning,就可以使用do{}while(0)來定義空宏:
#define EMPTYMICRO do{}while(0) |
4、定義一個(gè)單獨(dú)的函數(shù)塊來實(shí)現(xiàn)復(fù)雜的操作:
當(dāng)你的功能很復(fù)雜,變量很多你又不愿意增加一個(gè)函數(shù)的時(shí)候,使用do{}while(0);,將你的代碼寫在里面,里面可以定義變量而不用考慮變量名會(huì)同函數(shù)之前或者之后的重復(fù)。