您的位置:首页 > 编程语言 > C语言/C++

#define do{}while(0)妙用

2015-05-05 11:56 609 查看
大家都知道,do…while(condition)可以表示循环,但你有没有遇到在一些宏定义中可以不用循环的地方,也用到了 do…while.比如:

#define DELETE_POINTER(p) \

do \

{ \

if(NULL != p) \

delete p; \

p = NULL; \

}while(0)

这时,do…while(0)的功能就不仅仅是循环了,这是do..while(0)的一种巧妙用法。它有以下几种功能:

1,空的宏定义避免warning:

#define foo() do{}while(0)


2,存在一个独立的block,可以用来进行变量定义,进行比较复杂的实现。

3,如果出现在判断语句过后的宏,这样可以保证作为一个整体来是实现:

#define foo(x) \
action1(); \
action2();
在以下情况下:
if(NULL == pPointer)
foo();
就会出现action1和action2不会同时被执行的情况,而这显然不是程序设计的目的。


4,以上的第3种情况用单独的{}也可以实现,但是为什么一定要一个do{}while(0)呢,看以下代码:

#define switch(x,y) {int tmp; tmp="x";x=y;y=tmp;}
if(x>y)
switch(x,y);
else       //error, parse error before else
otheraction();


在把宏引入代码中,会多出一个分号,从而会报错。

linux内核和其他一些开源的代码中,经常会遇到这样的代码:

do{
...
}while(0)


这样的代码一看就不是一个循环,do..while表面上在这里一点意义都没有,那么为什么要这么用呢?

实际上,do{…}while(0)的作用远大于美化你的代码。查了些资料,总结起来这样写主要有以下几点好处:

1、辅助定义复杂的宏,避免引用的时候出错:

举例来说,假设你需要定义这样一个宏:

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


这个宏的本意是,当调用DOSOMETHING()时,函数foo1()和foo2()都会被调用。但是如果你在调用的时候这么写:

if([a]>0)
DOSOMETHING();


因为宏在预处理的时候会直接被展开,你实际上写的代码是这个样子的:

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


这就出现了问题,因为无论a是否大于0,foo2()都会被执行,导致程序出错

那么仅仅使用{}将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对程序流进行统一的控制:

有些函数中,在函数return之前我们经常会进行一些收尾的工作,比如free掉一块函数开始malloc的内存,goto一直都是一个比较简便的方法:

int foo()
{
somestruct* ptr = malloc(...);

dosomething...;
if(error)
{
goto END;
}

dosomething...;
if(error)
{
goto END;
}
dosomething...;

END:
free(ptr);
return 0;

}


由于goto不符合软件工程的结构化,而且有可能使得代码难懂,所以很多人都不倡导使用,那这个时候就可以用do{}while(0)来进行统一的管理:

int foo()
{

somestruct* ptr = malloc(...);

do{
dosomething...;
if(error)
{
break;
}

dosomething...;
if(error)
{
break;
}
dosomething...;
}while(0);

free(ptr);
return 0;

}


这里将函数主体使用do()while(0)包含起来,使用break来代替goto,后续的处理工作在while之后,就能够达到同样的效果。

3、避免空宏引起的warning

内核中由于不同架构的限制,很多时候会用到空宏,在编译的时候,空宏会给出warning,为了避免这样的warning,就可以使用do{}while(0)来定义空宏:

#define EMPTYMICRO
do{}while(0)


4、定义一个单独的函数块来实现复杂的操作:

当你的功能很复杂,变量很多你又不愿意增加一个函数的时候,使用do{}while(0);,将你的代码写在里面,里面可以定义变量而不用考虑变量名会同函数之前或者之后的重复。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  C-C++