我希望根据一些输入调用函数。每个函数都有不同数量的参数。换句话说,
if (strcmp(str, "funcA") == 0) funcA(a, b, c);
else if (strcmp(str, "funcB") == 0) funcB(d);
else if (strcmp(str, "funcC") == 0) funcC(f, g);
这有点笨重,难以维护。理想情况下,这些是可变函数(例如,printf风格)并且可以使用varargs。但他们不是。所以利用cdecl调用约定,我通过一个充满参数的结构填充堆栈。我想知道是否有更好的方法来做到这一点。请注意,这仅限于内部(例如,简单工具,单元测试等),不会用于任何可能遭受恶意攻击的生产代码。
示例:
#include <stdio.h>
typedef struct __params
{
unsigned char* a;
unsigned char* b;
unsigned char* c;
} params;
int funcA(int a, int b)
{
printf("a = %d, b = %d\n", a, b);
return a;
}
int funcB(int a, int b, const char* c)
{
printf("a = %d, b = %d, c = %s\n", a, b, c);
return b;
}
int funcC(int* a)
{
printf("a = %d\n", *a);
*a *= 2;
return 0;
}
typedef int (*f)(params);
int main(int argc, char**argv)
{
int val;
int tmp;
params myParams;
f myFuncA = (f)funcA;
f myFuncB = (f)funcB;
f myFuncC = (f)funcC;
myParams.a = (unsigned char*)100;
myParams.b = (unsigned char*)200;
val = myFuncA(myParams);
printf("val = %d\n", val);
myParams.c = (unsigned char*)"This is a test";
val = myFuncB(myParams);
printf("val = %d\n", val);
tmp = 300;
myParams.a = (unsigned char*)&tmp;
val = myFuncC(myParams);
printf("a = %d, val = %d\n", tmp, val);
return 0;
}
输出:
gcc -o func func.c
./func
a = 100, b = 200
val = 100
a = 100, b = 200, c = This is a test
val = 200
a = 300
a = 600, val = 0
答案 0 :(得分:0)
通过定义可变参数包装函数,我不会试图用语言来折磨这个突然出现的情况,而是以更简单的方式做到这一点。您可以使用宏来执行此操作:
#define WRAP_VARIADIC_2_ARG(FNAME, RETTYPE, ARGTYPE1, ARGTYPE2) \
(RETTYPE) wrapped_##FNAME(...) { \
va_list args; \
va_start(args, 2); \
(ARGTYPE1) arg1 = va_arg(args, (ARGTYPE1)); \
(ARGTYPE2) arg2 = va_arg(args, (ARGTYPE2)); \
va_end(args); \
return FNAME(arg1, arg2); \
}
其他arg计数的类似宏。然后,你打电话:
WRAP_VARIADIC_2_ARG(funcA, int, int, int)
WRAP_VARIADIC_3_ARG(funcB, int, int, int, const char*)
WRAP_VARIADIC_1_ARG(funcC, int, int*)
这将定义一组具有以下签名的函数,您可以在调度函数中使用这些函数:
int wrapped_funcA(...)
int wrapped_funcB(...)
int wrapped_funcC(...)
这应该是直截了当的。