do{...} while(0)的四大使用場景介紹

這里只針對單核CPU，linux內(nèi)核和其他一些開源的代碼中，經(jīng)常會遇到這樣的代碼：

do{
 ...
}while(0)

這樣的代碼一看就不是一個循環(huán)，do..while表面上在這里一點意義都沒有，那么為什么要這么用呢？

實際上，do{...}while(0)的作用遠大于美化你的代碼。查了些資料，總結(jié)起來這樣寫主要有以下幾點好處：

1、輔助定義復雜的宏，避免引用的時候出錯： 舉例來說，假設你需要定義這樣一個宏：

#define DOSOMETHING()
               foo1();
               foo2();

這個宏的本意是，當調(diào)用DOSOMETHING()時，函數(shù)foo1()和foo2()都會被調(diào)用。但是如果你在調(diào)用的時候這么寫：

if(a > 0)
    DOSOMETHING();

因為宏在預處理的時候會直接被展開，你實際上寫的代碼是這個樣子的：

if(a > 0)
    foo1();
foo2();

這就出現(xiàn)了問題，因為無論a是否大于0，foo2()都會被執(zhí)行，導致程序出錯。

那么僅僅使用{}將foo1()和foo2()包起來行么？

我們在寫代碼的時候都習慣在語句右面加上分號，如果在宏中使用{}，代碼里就相當于這樣寫了:“ { ... } ; ”，展開后就是這個樣子 :

if(a > 0)
{
    foo1();
    foo2();
};

這樣甚至不會編譯通過。所以，很多人才采用了do{...}while(0);

 #define DOSOMETHING() 
         do{ 
           foo1();
           foo2();
         }while(0)
 
 ...
 
 if(a > 0)
    DOSOMETHING();


...

這樣，宏被展開后，才會保留初始的語義。GCC提供了Statement-Expressions用以替代do{...}while(0); 所以你也可以這樣定義宏:

#define DOSOMETHING() ({
        foo1(); 
        foo2(); 
})

2、避免使用goto對程序流進行統(tǒng)一的控制： 有些函數(shù)中，在函數(shù)return之前我們經(jīng)常會進行一些收尾的工作，比如free掉一塊函數(shù)開始malloc的內(nèi)存，goto一直都是一個比較簡便的方法：

 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)化(bug菌PS:其實在Linux里面goto用得非常廣泛的，可以參考下(點擊藍色字體):goto關(guān)鍵字你不知道的"那些事"(C語言提升))，而且有可能使得代碼難懂，所以很多人都不倡導使用，那這個時候就可以用do{}while(0)來進行統(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之后，就能夠達到同樣的效果。

3、避免空宏引起的warning

內(nèi)核中由于不同架構(gòu)的限制，很多時候會用到空宏，在編譯的時候，空宏會給出warning，為了避免這樣的warning，就可以使用do{}while(0)來定義空宏:

#defineEMPTYMICROdo{}while(0)

4、定義一個單獨的函數(shù)塊來實現(xiàn)復雜的操作： 當你的功能很復雜，變量很多你又不愿意增加一個函數(shù)的時候，使用do{}while(0);，將你的代碼寫在里面，里面可以定義變量而不用考慮變量名會同函數(shù)之前或者之后的重復。

審核編輯：劉清