Question

我正在尝试包装open函数。我不知道如何将可选的第三个参数传递给真实的open。据我所知，无法验证va_list，因此if (mode)在下面的示例中不正确。有没有办法用适当数量的参数调用open？

#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdarg.h>

extern "C" 
{

int shouldWrap = 0;

int willCallRealOpen(const char * path, int flags, va_list args);
int __real_open(const char * path, int flags, ...);

int __wrap_open(const char * path, int flags, ...) {
    if(shouldWrap != 0){
        printf("Fake called\n");
        return 0;
    }
    else {
        printf("Real called\n");
        va_list args;
            va_start(args, flags);
            int res = willCallRealOpen(path, flags, args);
        va_end(args);
        return res;
    }
}

int willCallRealOpen(const char * path, int flags, va_list args) {
    mode_t mode = va_arg(args, mode_t);
    if (mode) {
        printf("3 args\n");
        return __real_open(path, flags, mode);
    }
    else {
        printf("2 args\n");
        return __real_open(path, flags);
    }
}   

}

int main() {
    //int fd = open("temp.txt", O_CREAT | O_WRONLY, S_IRUSR);
    int fd = open("temp.txt", O_CREAT | O_WRONLY);
}

Answer 1

open的手册页说：

模式指定在新文件出现时使用的权限创建。在O_CREAT或O_TMPFILE时必须提供此参数在flags中指定;如果既未指定O_CREAT也未指定O_TMPFILE，然后忽略模式。

所以我认为你应该这样做：

int willCallRealOpen(const char * path, int flags, va_list args) {
  if (flags & (O_CREAT | O_TMPFILE))
  {
    mode_t mode = va_arg(args, mode_t);
    printf("3 args\n");
    return __real_open(path, flags, mode);
  }
  else {
    printf("2 args\n");
    return __real_open(path, flags);
  }
}

包装可变函数

1 个答案: