C语言中do...while(0)技巧

在linux内核代码中,经常看到e(0)的宏,e(0)有很多作用,具体技巧有哪些呢?以下仅供参考!

C语言中do...while(0)技巧

1、避免goto语句:

通常,如果一个函数开始要分配一些资源,然后如果在中途遇到错误则要退出函数,当然,退出前要释放资源,我们的代码可能如下:

#defien N 10

bool Execute()

{

// 分配资源

int *p = (int *)malloc(N * sizeof(int));

bool bOk = true;

// 执行并进行错误处理

bOk = func1();

if(!bOk)

{

free(p);

p = NULL;

return false;

}

bOk = func2();

if(!bOk)

{

free(p);

p = NULL;

return false;

}

bOk = func3();

if(!bOk)

{

free(p);

p = NULL;

return false;

}

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

// 执行成功,释放资源并返回

free(p);

p = NULL;

return true;

}

这里最大的问题是代码冗余,每增加一个操作,就要做相应的错误处理,非常不灵活,于是想到了一下的goto:

#defien N 10

bool Execute()

{

// 分配资源

int *p = (int *)malloc(N * sizeof(int));

bool bOk = true;

// 执行并进行错误处理

bOk = func1();

if(!bOk) goto errorhandle;

bOk = func2();

if(!bOk) goto errorhandle;

bOk = func3();

if(!bOk) goto errorhandle;

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

// 执行成功,释放资源并返回

free(p);

p = NULL;

return true;

errorhandle:

free(p);

p = NULL;

return false;

}

代码冗余是解决了,但是引入了C语言中比较微妙的goto语句,虽然正确的使用goto语句可以大大提高程序的灵活性与简洁性,但是会使我们的程序捉摸不定,为了既避免使用goto语句,又能消除代码冗余,可以考虑使用下面的` e(0):

#defien N 10

bool Execute()

{

//分配资源

int *p = (int *)malloc(N * sizeof(int));

bool bOK = true;

do {

//执行并进行错误处理

bOK = fun1();

if(!bOK) break;

bOK = fun2();

if(!bOK) break;

bOK = fun3();

if(!bOK) break;

//.........

} while(0);

//释放资源

free(p);

p = NULL;

return bOK;

}

2、避免空声明在编译时出现警告:

在linux内核源代码中,经常看到如下宏以避免在编译时出现警告:

#define FOO do { } while(0)

3、编写符合习惯的代码块:

你可能经常会使用如下的宏:

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

然而在某些情况下将会失效,下面的代码使用...

if (x > y)

exch(x,y); // 分支 1

else

do_something(); // 分支 2

但是将被解释为一个分支的if语句:

if (x > y) {

int tmp;

tmp = x;

x = y;

y = tmp;

}

; // 空语句

else // ERROR!!!

do_something();

错误出在“;”直接位于代码块的后面,解决的办法是将代码嵌入e(0),于是得到下面的代码:

if (x > y)

do {

int tmp;

tmp = x;

x = y;

y = tmp;

} while(0);

else

do_something();

于是上面的宏可以修改为:

#define exch(x,y) do {

int tmp;

tmp = x;

x = y;

y = tmp;

} while(0)

4、在条件语句中使用复杂的宏:

假如一个宏包含类似如下几行代码:

#define FOO(x)

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

do_something_useful(x);

现在想像一下下面的代码:

if (blah == 2)

FOO(blah);

这将解释为:

if (blah == 2)

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

do_something_useful(blah);;

我们就会发现,if语句只作用于printf(), do_something_useful() 没按照愿意一起执行,即没有像你预期的那样被包含在if代码中,于是可以使用如下的代码块:

if (blah == 2)

do {

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

do_something_useful(blah);

} while (0);

这样上面的宏就可以改为:

#define FOO(x) do {

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

do_something_useful(blah);

} while (0)