欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

do{...}while(0)的巧妙用处

程序员文章站 2024-03-22 21:13:40
...

最近读pjsip源码的时候看到do{...}while(0)的这种用法:

do{...}while(0)的巧妙用处

于是查了下这种用法的作用,发现挺有意思的。

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

假设要定义一个宏:

#define F() f1(); f2();

这个宏的意思是,当调用F()时,f1()和f2()都会被调用。但是在调用的时候如果这么写:

if(expr)
    F();

 而宏在预处理的时候会直接被展开为:

if(expr)
    f1();f2();

这就导致无论expr是否为真,f2()一定被执行,因为if语句只能控制下一句f1(),这并不是我们的预期。

一种解决办法是用{}将f1();f2{};包起来:即

#define F() { f1(); f2(); }

但这种情况下,我们发现语句被展开为:

if(expr)
    {f1();f2();};

 最后的分号是我们习惯添加的,显然,这是个语法错误。有些编译器并不能通过。

这时候我们发现do{...}while(0)似乎是一种很好的解决办法:

#define F() \
        do{ \
          f1();\
          f2();\
        }while(0)\

可以达到语句被执行且仅执行一次,还能像普通函数调用一样在后面加分号。

pjsip C源码中很多的这种用法:

#if defined(PJ_ENABLE_EXTRA_CHECK) && PJ_ENABLE_EXTRA_CHECK != 0
#   define PJ_ASSERT_ON_FAIL(expr,exec_on_fail)    \
	    do { \
		pj_assert(expr); \
		if (!(expr)) exec_on_fail; \
	    } while (0)
#else
#   define PJ_ASSERT_ON_FAIL(expr,exec_on_fail)    pj_assert(expr)
#endif

2.避免使用goto对程序流进行统一的控制:

一些代码中想达到goto这种简单的代码流控制效果,例如:有些函数中,在函数return之前我们经常会进行一些收尾的工作,比如free掉一块函数开始malloc的内存,goto一直都是一个比较简便的方法。

int f()
{
    somestruct* ptr = malloc(...);
    ...;
    if(error)
    {
        goto END;
    }
    ...;
    if(error)
    {
        goto END;
    }
    ...;
 
END:
    free(ptr);
    return 0;
}

但是goto不符合软件工程的结构化,而且有可能使得代码难懂,所以不倡导使用,这时可以用do{}while(0)来进行统一的管理:

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

    do{
        ...;
        if(error)
        {
            break;
        }
        ...;
        if(error)
        {
            break;
        }
        ...;
    }while(0);
 
    free(ptr);
    return 0;
}

 这里将函数主体部分使用do{...}while(0)包含起来,使用break来代替goto,后续的清理工作在while之后,现在既能达到同样的效果,而且代码的可读性、可维护性都要比上面的goto代码好的多了。

3、避免空宏引起的警告

内核中由于不同架构的限制,很多时候会用到空宏。在编译的时候,这些空宏会给出warning,为了避免这样的warning,可以使用do{...}while(0)来定义空宏:
#define EMPTYMICRO do{}while(0)
这种情况不太常见,因为有很多编译器,已经支持空宏。

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

复杂的函数,变量很多,若不想要增加新的函数,可以用do{...}while(0),将代码写在里面,里面可以定义变量而不用考虑变量名会同函数之前或者之后的重复。
但是不建议这样做,尽量声明不同的变量名,以便于后续开发人员阅读。

int i;
unsigned j;
int func()
{
    int j = fi();
    unsigned j = fj();
    ...;
    do{
        int i;
        unsigned j;
        ...;
    }while(0);    
}