嵌入式C語言自我修養 12:有一種宏,叫可變參數宏

12.1 什麼是可變參數宏

在上面的教程中,我們學會了變參函數的定義和使用,基本套路就是使用 va_list、va_start、va_end 等宏,去解析那些可變參數列表我們找到這些參數的存儲地址後,就可以對這些參數進行處理了:要麼自己動手,自己處理;要麼繼續調用其它函來處理。

void print_num(int count, ...)
{
    va_list args;
    va_start(args,count);
    for(int i = 0; i < count; i++)
    {
        printf("*args: %d\n",*(int *)args);
        args += 4; 
    }
}
void __attribute__((format(printf,2,3))) LOG(int k,char *fmt,...)
{
    va_list args;
    va_start(args,fmt);
    vprintf(fmt,args);
    va_end(args);
}

GNU C 覺得這樣不過癮,再來個猛錘:乾脆宏定義也支持變參吧!

這一節我們要學習一下可變參數宏的定義和使用。其實,C99 標準已經支持了這個特性,但是其它的編譯器不太給力,對 C99 標準的支持不是很好,只有 GNU C 支持這個功能,所以有時候我們也把這個可變參數宏看作是 GNU C 的一個語法擴展。 上面的 LOG 函數,如果我們想使用一個變參宏實現,就可以直接這樣定義。

#define LOG(fmt, ...) printf(fmt, __VA_ARGS__)

#define DEBUG(...) printf(__VA_ARGS__)

int main(void)
{
    LOG("Hello! I'm %s\n","Wanglitao");
    DEBUG("Hello! I'm %s\n","Wanglitao");
    return 0;
}

變參宏的實現形式其實跟變參函數差不多:用 ... 表示變參列表,變參列表由不確定的參數組成,各個參數之間用逗號隔開。可變參數宏使用 C99 標準新增加的一個 VA_ARGS__ 預定義標識符來表示前面的變參列表,而不是像變參函數一樣,使用 va_list、va_start、va_end 這些宏去解析變參列表。預處理器在將宏展開時,會用變參列表替換掉宏定義中的所有 VA_ARGS__ 標識符。

使用宏定義實現一個變參打印功能,你會發現,它的實現甚至比變參函數還方便!內核中的很多打印宏,經常使用可變參數宏來實現,宏定義一般爲下面這個格式。

#define LOG(fmt, ...) printf(fmt, __VA_ARGS__)

在這個宏定義中,有一個固定參數,通常爲一個格式字符串,後面的變參用來打印各種格式的數據,跟前面的格式字符串相匹配。這種定義方式有一個漏洞,即當變參爲空時,宏展開時就會產生一個語法錯誤。

#define LOG(fmt,...) printf(fmt,__VA_ARGS__)
int main(void)
{
    LOG("hello\n");
    return 0;
}

上面這個程序編譯時就會通不過,產生一個語法錯誤。這是因爲,我們只給 LOG 宏傳遞了一個參數,而變參爲空。當宏展開後,就變成了下面這個樣子。

printf("hello\n", );

宏展開後,在第一個字符串參數的後面還有一個逗號,所以就產生了一個語法錯誤。我們需要繼續對這個宏進行改進,使用宏連接符 ##,來避免這個語法錯誤。

12.2 繼續改進我們的宏

我們接下來,使用宏連接符 ## 來改進上面的宏。

宏連接符 ## 的主要作用就是連接兩個字符串,我們在宏定義中可以使用 ## 來連接兩個字符。預處理器在預處理階段對宏展開時,會將 ## 兩邊的字符合並,並刪除 ## 這兩個字符。

#define A(x) a##x
int main(void)
{
    int A(1) = 2; //int a1 = 2;
    int A() = 3;  //int a=3;
    printf("%d %d\n",a1,a);
    return 0;   
}

如上面的程序,我們定義一個宏。

#define A(x) a##x

這個宏的功能就是連接字符 a 和 x。在程序中,A(1) 展開後就是 a1,A( ) 展開後就是 a。我們使用 printf( ) 函數可以直接打印變量 a1、a 的值,因爲宏展開後,就相當於使用 int 關鍵字定義了兩個整型變量 a1 和 a。上面的程序可以編譯通過,運行結果如下。

2  3

知道了宏連接符 ## 的使用方法,我們接下來就可以就對 LOG 宏做一些修改。

#define LOG(fmt,...) printf(fmt, ##__VA_ARGS__)
int main(void)
{
    LOG("hello\n");
    return 0;
}

我們在標識符 __VA_ARGS__ 前面加上宏連接符 ##,這樣做的好處是,當變參列表非空時,## 的作用是連接 fmt,和變參列表,各個參數之間用逗號隔開,宏可以正常使用;當變參列表爲空時,## 還有一個特殊的用處,它會將固定參數 fmt 後面的逗號刪除掉,這樣宏也就可以正常使用了。

12.3 可變參數宏的另一種寫法

當我們定義一個變參宏時,除了使用預定義標識符 __VA_ARGS__ 表示變參列表外,還可以使用下面這種寫法。

#define LOG(fmt,args...) printf(fmt, args)

使用預定義標識符 VA_ARGS 來定義一個變參宏,是 C99 標準規定的寫法。而上面這種格式是 GNU C 擴展的一個新寫法。我們不再使用 VA_ARGS,而是直接使用 args... 來表示一個變參列表,然後在後面的宏定義中,直接使用 args 代表變參列表就可以了。

跟上面一樣,爲了避免變參列表爲空時的語法錯誤,我們也需要添加一個連接符##。

#define LOG(fmt,args...) printf(fmt,##args)
int main(void)
{
    LOG("hello\n");
    return 0;
}

使用這種方式,你會發現這種寫法比使用 __VA_ARGS__ 看起來更加直觀和方便。

12.4 內核中的可變參數宏

可變參數宏在內核中主要用於日誌打印。一些驅動模塊或子系統有時候會定義自己的打印宏,可以支持打印開關、打印格式、優先級控制等。如在 printk.h 頭文件中,我們可以看到 pr_debug 宏的定義。

#if defined(CONFIG_DYNAMIC_DEBUG)
#define pr_debug(fmt, ...) \
    dynamic_pr_debug(fmt, ##__VA_ARGS__)
#elif defined(DEBUG)
#define pr_debug(fmt, ...) \
    printk(KERN_DEBUG pr_fmt(fmt), ##__VA_ARGS__)
#else
#define pr_debug(fmt, ...) \
    no_printk(KERN_DEBUG pr_fmt(fmt), ##__VA_ARGS__)
#endif

#define dynamic_pr_debug(fmt, ...)                \
do {                                \
    DEFINE_DYNAMIC_DEBUG_METADATA(descriptor, fmt); \
    if (unlikely(descriptor.flags       \
            & _DPRINTK_FLAGS_PRINT))    \
        __dynamic_pr_debug(&descriptor, pr_fmt(fmt),    \
                   ##__VA_ARGS__);      \
} while (0)

static inline __printf(1, 2)
int no_printk(const char *fmt, ...)
{
    return 0;
}

#define __printf(a, b)    \   
__attribute__((format(printf, a, b)))

看到這個宏定義,不得不佩服宏的作者。一個小小的宏,綜合運用各種技巧和知識點,把 C 語言發揮到極致!

這個宏定義了三個版本。如果我們在編譯內核時有動態調試選項,那麼這個宏就定義爲 dynamicprdebug。如果沒有配置動態調試選項,那我們還可以通過 DEBUG 這個宏,來控制這個宏的打開和關閉。

no_printk() 作爲一個內聯函數,定義在 printk.h 頭文件中,而且通過 format 屬性聲明,指示編譯器按照 printf 標準去做參數格式檢查。

最有意思的是 dynamicprdebug 宏,宏定義採用 do{ ... }while(0) 結構。這看起來貌似有點多餘,有它沒它,我們的宏都可以工作。反正都是執行一次,爲什麼要用這種看似“畫蛇添足”的循環結構呢?道理很簡單,這樣定義就是爲了防止宏在條件、選擇等分支結構的語句中展開後,產生宏歧義。

比如我們定義一個宏,由兩條打印語句構成。

#define DEBUG() \
 printf("hello ");printf("else\n")

int main(void)
{
    if(1)
        printf("hello if\n");
    else
        DEBUG();
    return 0;
}

程序運行結果如下。

hello if
else

理論情況下,else 分支是執行不到的。但通過運行結果可以看到,程序也執行了 else 分支的一部分代碼。這是因爲我們定義的宏由多條語句組成,直接展開後,就變成了下面這樣。

int main(void)
    {
        if(1)
            printf("hello if\n");
        else
            printf("hello ");
            printf("else\n");
        return 0;
    }

多條語句在宏調用處直接展開,就破壞了程序原來的 if-else 分支結構,導致程序邏輯發生變化,所以你纔會看到 else 分支的非正常打印。而採用 do{ ... }while(0) 這種結構,可以將我們宏定義中的複合語句包起來,宏展開後,就是一個代碼塊,就避免了這種邏輯錯誤。

一個小小的宏,暗藏各個知識點,綜合使用各種技巧,仔細分析下來,就能學到很多知識。大家在以後的工作和學習中,可能會接觸到各種各樣、形形×××的宏,只要我們有牢固的 C 語言基礎,熟悉 GNU C 的常用擴展語法,再遇到這樣類似的宏,我們都可以慢慢去分析了。不用怕,只有自己真正分析過,纔算真正掌握,才能轉化爲自己的知識和能力,才能領略它的精妙之處。

本教程根據 C語言嵌入式Linux高級編程視頻教程 第05期 改編,電子版書籍可加入QQ羣:475504428 下載,更多嵌入式視頻教程,可關注:
微信公衆號:宅學部落(armlinuxfun)
51CTO學院-王利濤老師:http://edu.51cto.com/sd/d344f

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章