内核中的do+++while(0)

内核中 do...while(0) 的应用

1. do...while(0)消除goto语句。
通常，如果在一个函数中开始要分配一些资源，然后在中途执行过程中如果遇到错误则退出函数，当然，退出前先释放资源，我们的代码可能是这样：
version 1

bool Execute()
{
  // 分配资源
  int *p = new int;
  bool bOk(true);

  // 执行并进行错误处理
  bOk = func1();
  if(!bOk) 
  {
  delete p;  
  p = NULL;
  return false;
  }

  bOk = func2();
  if(!bOk) 
  {
  delete p;  
  p = NULL;
  return false;
  }

  bOk = func3();
  if(!bOk) 
  {
  delete p;  
  p = NULL;
  return false;
  }

  // ..........

  // 执行成功，释放资源并返回
  delete p;  
  p = NULL;
  return true;
   
}

这里一个最大的问题就是代码的冗余，而且我每增加一个操作，就需要做相应的错误处理，非常不灵活。于是我们想到了goto:
version 2


bool Execute()
{
  // 分配资源
  int *p = new int;
  bool bOk(true);

  // 执行并进行错误处理
  bOk = func1();
  if(!bOk) goto errorhandle;

  bOk = func2();
  if(!bOk) goto errorhandle;

  bOk = func3();
  if(!bOk) goto errorhandle;

  // ..........

  // 执行成功，释放资源并返回
  delete p;  
  p = NULL;
  return true;

errorhandle:
  delete p;  
  p = NULL;
  return false;
   
}

代码冗余是消除了，但是我们引入了C++中身份比较微妙的goto语句，虽然正确的使用goto可以大大提高程序的灵活性与简洁性，但太灵活的东西往往是很危险的，它会让我们的程序捉摸不定，那么怎么才能避免使用goto语句，又能消除代码冗余呢，请看do...while(0)循环：
version3


bool Execute()
{
  // 分配资源
  int *p = new int;

  bool bOk(true);
  do
  {
  // 执行并进行错误处理
  bOk = func1();
  if(!bOk) break;

  bOk = func2();
  if(!bOk) break;

  bOk = func3();
  if(!bOk) break;

  // ..........

  }while(0);

  // 释放资源
  delete p;  
  p = NULL;
  return bOk;
   
}

为什么在内核中碰到很多 #defines ... do{ ... } while(0)?

　　有以下几点原因:

　　·空语句在编译时候会出现警告，所以有必要用#define FOO do { } while(0).

　　·这样做是为了能够在里面定义局部变量

　　·这样做是为了能够在条件语句中使用复杂的宏定义. 例如下面这段代码:

　　#define FOO(x) \

　　printf("arg is %s\n", x); \

　　do_something_useful(x);

　　如果这样用:

　　if (blah == 2)

　　FOO(blah);

　　将会被展开为:

　　if (blah == 2)

　　printf("arg is %s\n", blah);

　　do_something_useful(blah);;

　　这样,if条件之包含了printf()语句，而 do_something_useful()调用不能按期望那样工作。而使用 do { ... } while(0)定义后，就会展开成以下语句:

　　if (blah == 2)

　　do {

　　printf("arg is %s\n", blah);

　　do_something_useful(blah);

　　} while (0);

　　这是所期望的状况.

　　·如果你希望定义一个包含多行语句和一些局部变量的时候. 一般的定义方式只能这样:

　　#define exch(x,y) { int tmp; tmp=x; x=y; y=tmp; }

　　然而在某些情况下，这样并不能正常工作. 下面是包含两个分支的if语句:

　　if (x > y)

　　exch(x,y);          // Branch 1

　　else

　　do_something();     // Branch 2

　　但这样却只能展开成单分支的if语句，如下:

　　if (x > y) {                // 单分支if

　　int tmp;

　　tmp = x;

　　x = y;

　　y = tmp;

　　}

　　;                           // 空语句

　　else                        // 错误!!! "parse error before else"

　　do_something();

　　问题是由于在语句块后直接加入分号（;）引起的. 解决办法是将语句块放入 do 和 while (0)中间.

#define each(x,y) \

do{                          \

int tmp;                   \

tmp = x;                  \

x = y;                       \

y = tmp;                  \

} while(0)           

注意 while(0) 后面 没有加 " ; " !!!!!

这样就得到了一条单语句, 而不是被编译器判断为语句块.现在的if语句如下:

　　if (x > y)

　　do {

　　int tmp;

　　tmp = x;

　　x = y;

　　y = tmp;

　　} while(0);

　　else

　　do_something();