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

最近读pjsip源码的时候看到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);    
}
  • 类似的还有

  • for(;;) 
    {
        ...
        ...
        break;
    }

     

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值