这是我在 C 中所做的,用于包装任何函数。我希望它能帮助/启发你。主要限制是您只能包装用户定义的函数,因为您必须修改函数的签名。
我的解决方案使用 va_arg 结构。
#include <stdarg.h>
void * super_wrapper( void * (*fn)(), void * ret, int n, ...)
{
va_list = my_list;
va_start(my_list, n);
ret = (*fn)(ret, n, my_list);
va_end(my_list);
return ret
}
包装器的签名采用指向您要执行的函数的指针,一个指向该函数以检索您的返回值的指针,您调用该函数的参数数量,然后是您想要的函数的参数称呼。
va_list 是参数列表。你用 va_start 初始化它。va_start 的第一个参数是要初始化的 va_list。第二个参数是当前函数的最后一个已知参数(在本例中为“n”)。
现在,您将放置一个参数 va_list,而不是用户定义函数中的省略号。
我有两个例子:
double sum (int n, ...)
{
int i;
double res = 0;
double tmp;
va_list = my_list;
va_start(my_list, n);
for(i=0; i<n; i++)
{
tmp = va_arg(my_list, double);
res += tmp;
}
va_end(my_list);
return res;
}
void my_print(int n, ...)
{
int i;
va_list my_list;
va_start(my_list, n);
char * tmp;
for(i=0; i<n; i++)
{
tmp = va_arg(my_list, char *);
printf("%s ", tmp);
}
printf("\n");
va_end(my_list);
}
我设法使用相同的包装器来调用这两个不同的函数,并对签名和函数体进行了一些修改。问题是在参数和返回中,我只给出指针。所以我必须取消引用指针。
void * sum (void * ret, int n, va_list my_list)
{
int i;
double res = 0;
double *tmp;
for(i=0; i<n; i++)
{
tmp = va_arg(my_list, double);
res += *tmp;
}
ret = &res;
return ret;
}
void* my_print(int n, ...)
{
int i;
char ** tmp;
for(i=0; i<n; i++)
{
tmp = va_arg(my_list, char **);
printf("%s ", *tmp);
}
printf("\n");
return ret;
}
现在我可以在我的 main 中使用相同的包装函数来调用这两个函数:
void main()
{
double two = 2.0;
double three = 3.0;
double fifteen = 15.0;
char *I, *am, *a, *great, *wrapper;
I = malloc(sizeof(char) * 2);
am = malloc(sizeof(char) * 3);
a = malloc(sizeof(char)*2);
great = malloc(sizeof(char) * 6;
wrapper = malloc(sizeof(char) * 8);
I = strcpy(I, "I\0");
am = strcpy(am, "am\0");
a = strcpy(a,"a\0");
great = strcpy(great, "great\0");
wrapper = strcpy(wrapper, "wrapper\0");
void * ret;
printf("Sum = %f\n", *( (double*)super_wrapper(sum, ret, 3, &two, &three, &fifteen) ) );
super_wrapper(my_print, ret, 5, &I, &am, &a, &great, &wrapper);
}