大学IT网 - 最懂大学生的IT学习网站! QQ资料交流群:367606806
当前位置:大学IT网 > C技巧 > C语言中do...while(0)用法小结

C语言中do...while(0)用法小结(1)

关键词:小结用法语言  阅读(1177) 赞(10)

[摘要]本文是对C语言中do...while(0)用法小结的讲解,与大家分享。

在linux内核代码中,经常看到do...while(0)的宏,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语句,又能消除代码冗余,可以考虑使用下面的 do...while(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内核源代码中,经常看到如下宏以避免在编译时出现警告:

«上一页12下一页»


相关评论