C 宏的乐趣

Question

假设我在 C 中有一个名为 FOO 的函数宏。还有两个宏称​​为BAR1 和BAR2，它们基本上是同一个宏的两种风格。我想编写一个宏BAR，使其在函数中扩展为BAR1，在使用BAR 之前调用FOO，否则扩展为BAR2。比如：

void func1(void)
{
    FOO();
    ...
    BAR();
}

相当于

void func1(void)
{
    FOO();
    ...
    BAR1();
}

这个函数：

void func2(void)
{
    BAR();
}

相当于

void func2(void)
{
    BAR2();
}

我想避免在运行时引入全局变量或进行额外检查。这甚至可能吗？

Answer 1

简短回答：不。 C 预编译器对函数限制一无所知，因此即使您设法根据需要修改了 BAR 宏，也不会限制为当前函数。

现在，如果您愿意，可以在 BAR 宏中添加一些检查。并且可以编写这些检查以在编译时解决，因此不会产生运行时开销。

例如：

extern char _sentinel_[2];

#define FOO() char _sentinel_;

#define BAR() if (sizeof(_sentinel_) == 1) BAR1() else BAR2()

诀窍在于查找变量_sentinel_ 将解析全局变量或局部变量，具体取决于FOO() 的使用。由于if 中的条件是编译器常量，编译器会优化掉另一个分支。

Answer 2

我尝试使用gotos 的破解失败了，因为当不使用FOO() 时，BAR() 的跳转标签丢失。但是，不要害怕，我想出了一个更糟糕的破解方法。

您可以使用#includes 代替FOO() 和BAR() 的宏。这将允许您完全控制代码的扩展方式。

/* FOO file */
#define BAR_IS_BAR2
/* whatever code FOO needs to do */

/* BAR file */
#ifdef BAR_IS_BAR2
    BAR2();
    #undef BAR_IS_BAR2
#else
    BAR1();        
#endif

/*...in you source code...*/
void func1 () {
    #include "FOO"
    /*...*/
    #include "BAR"
}

void func2 () {
    #include "BAR"
}