如何将标准输出重定向到文件然后恢复标准输出？

Question

这是我的代码，我无法让它工作。

int pfd = open("file", O_WRONLY, 0777);
int saved = dup(1);
close(1);
dup(pfd);
close(pfd);
printf("This goes into file\n");

// restore it back
dup2(saved, 1);
close(saved);
printf("this goes to stdout");

我对我的代码添加了一些修改。

Answer 1

您需要检查函数调用的返回值。对于大多数功能，您应该检查错误情况。这样做可能会暴露一个问题，如果您希望 open() 在最初不存在的情况下创建请求的文件，那么您需要添加 O_CREAT 标志。

但这不是您的主要问题 - 您正在处理缓冲问题。第一个printf() 的输出缓冲在内存中，因此即使文件描述符1 在调用printf() 时引用您的文件，您写入的数据也不会立即刷新到目标文件。然后，您恢复原始的stdout 文件句柄，因此当实际刷新数据时，它们会转到（恢复的）原始标准输出。在切换 stdout 之前通过 fflush()ing 解决这个问题：

int pfd = open("file", O_WRONLY | O_CREAT, 0777);
int saved = dup(1);

close(1);
dup(pfd);
close(pfd);
printf("This goes into file\n");
fflush(stdout);  // <-- THIS

// restore it back
dup2(saved, 1);
close(saved);
printf("this goes to stdout");

还请注意，dup2() 将文件描述符复制到 特定 文件描述符编号更干净、更安全。您在恢复时执行此操作，但您也应该在初始重定向时执行此操作。