c - 打印到缓冲区或文件重用代码

Question

我想定义两个函数，fdump 和 sdump，分别使用 fprintf 和 sprintf 将结构转储到文件或缓冲区。

有没有办法在不重复两个函数中的代码的情况下定义它们？一种解决方案是定义 sdump，然后基于它定义 fdump，即:

void fdump(FILE* f, struct mystruct* param) {
  char buffer[MAX];
  sdump(buffer, MAX, param);
  fprint(f, "%s", buffer);
}

但该解决方案会浪费和中间缓冲区。虽然也许 fprintf 做同样的事情。其他解决方案可能是通过预处理宏，但它看起来很复杂。请问，有什么想法吗？

提前致谢

Answer 1

您可以使用 fmemopen 为您提供一个指向一 block 内存的文件句柄，然后只编写一个接受文件句柄的函数版本:

#include <stdio.h>

void foo(FILE *fh) {
  fprintf(fh, "test\n");
}

int main() {
  foo(stderr);
  char str[100];
  FILE *mem = fmemopen(str, sizeof str, "w");
  foo(mem);
  fclose(mem);
  fprintf(stdout, "%s", str);
  return 0;
}