宏因为其各种副作用而备受诟病,就像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
至此,使用宏完美地解决了这个看似无解的问题。