使用C语言的宏来解决一个问题

宏因为其各种副作用而备受诟病,就像goto一样,有些产品谈宏色变。但是语言特性本身并无善恶之分,其善恶在于使用者。有时候使用某些语言特性,能够非常好的解决一些编程中的难题。本文就使用宏来解决一个比较棘手的问题。
有一个函数,原型如下:

// a.h
void TEST(unsigned int a, unsigned int b, unsigned int c);

在UT测试中,要对这个函数打桩,打桩成了一个宏,实现如下:

// b.h
#define TEST(a, b, c)  a = b + c

在main函数中如下调用:

// mian.c
int main()
{
    int a = 5, b = 3, c = 8;
    TEST(a, b, c);
    return 0;
}

在编译时,会根据是否是UT工程决定使用a.h还是b.h。

当使用a.h时会有一个告警,因为TEST函数的定义形参要求是unsigned int,而实参是int。因为使用TEST的地方很多,且情况比较复杂,只能使用类型强转清除这个告警。
改成如下形式,UT工程编译不过(原因是宏展开后,等式的左值不能有强转)

// mian.c
int main()
{
    int a = 5, b = 3, c = 8;
    TEST((unsigned int)a, (unsigned int)b, (unsigned int)c);
    return 0;
}

怎么办?
好像没有什么好办法,试着将TEST重定义,如下:

#ifdef _UT_
#define MY_TEST(a, b, c)  TEST(a, b, c)
#else
#define MY_TEST(a, b, c)  TEST((unsigned int)a, (unsigned int)b, (unsigned int)c)
#endif

int main()
{
    int a = 5, b = 3, c = 8;
    MY_TEST(a, b, c);
    return 0;
}

问题看似解决,但是伴随的是成千上万出的TEST调用要换成MY_TEST的调用,并且后续开发人员必须要遵守使用MY_TEST的规则,否则就会出错。

有没有更好的办法,使得影响降至最小?仔细想下,UT工程并不关心数据类型的转换,如果能够将类型转换那一部分((unsigned int))在宏展开时替换掉就能解决问题。
如果能够让(unsigned int)前面拼接一个符号组成一个宏函数,然后定义这个宏函数的实现为空即可,如下:

#define DUMMY(...)
#define TEST(a, b, c)  DUMMY#a = DUMMY#b + DUMMY#c

对于TEST((unsigned int)a, (unsigned int)b, (unsigned int)c);期望展开后是DUMMY(unsigned int)a = DUMMY(unsigned int)b + DUMMY(unsigned int)c,DUMMY(unsigned int)为空,则替换后为a = b + c,完美。

但是,上面的做法是错的,编译不过,因为使用#展开后的形式是DUMMY"(unsigned int)a" = DUMMY"(unsigned int)b" + DUMMY"(unsigned int)c"
正确的写法是去掉#,但一定要在DUMMY和宏参数间加上空格,如下:

#define DUMMY(...)
#define TEST(a, b, c)  DUMMY a = DUMMY b + DUMMY c

至此,使用宏完美地解决了这个看似无解的问题。

你可能感兴趣的:(使用C语言的宏来解决一个问题)