C MACRO 将函数名称与预定义的名称列表进行比较

Question

我有一些功能要挂在 Windows 上（大约 200 个），我想为所有这些功能添加一些通用功能。根据我给我的应用程序的标志，挂钩或不挂钩一些预定义的功能。

我的想法是这样的：

#define SHOULD_CHECK_STOP_MONITOR           (1 == Configuration.dwShouldMonitor)
#define MY_CHECK_MACRO(f) (if(SHOULD_CHECK_STOP_MONITOR && /* What do I do here? */))

void MyHookedFunctionName_A(...)
{
    if(!MY_CHECK_MACRO(__FUNCTION__))
        goto lbl_Cleanup;

    /* rest of code logic */

lbl_Cleanup:
    return ...;
}

void MyHookedFunctionName_B(...)
{
    if(!MY_CHECK_MACRO(__FUNCTION__))
        goto lbl_Cleanup;

    /* rest of code logic */

lbl_Cleanup:
    return ...;
}

我想要实现的结果是一个预编译的逻辑，不是在动态运行时执行，而是在编译时执行。如果1 == Configuration.dwShouldMonitor 配置为True，它将检查其余逻辑，其中我将有某种预定义的函数名称列表["MyHookedFunctionName_A", "MyHookedFunctionName_C", "MyHookedFunctionName_D"]，并且宏会知道是返回False 还是@ 987654326@ 基于函数名称。所以在这个例子中，MyHookedFunctionName_B 会执行，而MyHookedFunctionName_A 不会因为编译时的预定义列表。

这样的事情可能吗？对于基于函数名称的“复杂”逻辑并将其与预定义的字符串函数名称进行比较？

编辑 2019.09.15： 我在这里添加我希望预编译逻辑之后的结果

void MyHookedFunctionName_A(...)
{
    if(!(1 == Configuration.dwShouldMonitor && False))
        goto lbl_Cleanup;
...
}

void MyHookedFunctionName_B(...)
{
    if(!(1 == Configuration.dwShouldMonitor && True))
        goto lbl_Cleanup;
...
}

True 或 False 是在编译时根据函数字符串名称生成的

Answer 1

许多编译器应该能够完全优化您的“运行时”检查。

这里有一个简单的程序来说明这一点。

#include <string.h>
#include <stdio.h>

static const char* func_list[] = { "foo", "bar" };

static inline int is_in_list(const char* f) {
    for (int i = 0; i < sizeof(func_list)/sizeof(func_list[0]); ++i)
        if (strcmp(f, func_list[i]) == 0)
            return 1;
    return 0;
}

#define HOOKME if (SHOULD_HOOK && is_in_list(__func__)) return;

void foo() {
    HOOKME;
    printf("foo\n");
}
void bar() {
    HOOKME;
    printf("bar\n");
}
void baz() {
    HOOKME;
    printf("baz\n");
}

int main()
{
    foo();
    bar();
    baz();
}

用 -DSHOULD_HOOK=1 和 -O2 编译它，一个好的编译器可能会让 foo 和 bar 消失。

foo:
  ret
bar:
  ret
.LC0:
  .string "baz"
baz:
  mov edi, OFFSET FLAT:.LC0
  jmp puts
main:
  sub rsp, 8
  xor eax, eax
  call baz
  xor eax, eax
  add rsp, 8
  ret

并非所有编译器都能做到这一点。明智地选择你的工具。 Demo.