重定向之dup和dup2函数

在C语言中使用文件描述符实现重定向输出

在UNIX-like系统中，重定向输出到文件是编程中常见的需求。在C语言中，我们可以通过文件描述符来实现这一目标。文件描述符是内核为了高效管理已打开的文件或其他I/O资源（如套接字、管道等），而提供的一个整数值。标准输入、输出和错误输出分别对应文件描述符0、1和2。

一、不使用dup和dup2函数的方法

这种方法首先关闭标准输出（文件描述符为1），然后打开目标文件。由于系统会为每个进程打开的文件分配最小的未使用文件描述符，因此打开文件后，标准输出默认会被重定向到新打开的文件。

#define FNAME "/tmp/out"

int main()
{
    int fd;
    close(1); // 关闭标准输出
    fd = open(FNAME, O_WRONLY | O_CREAT | O_TRUNC, 0600); // 打开文件并创建
    if(fd < 0)
    {
        perror("open()");
        exit(1);
    }
    puts("hello"); // 输出到文件
    close(fd); // 关闭文件描述符
    return 0;
}

使用cat命令查看文件内容，会发现有hello字样。

二、dup和dup2函数

dup和dup2函数都可以复制一个现有的文件描述符。

#include <unistd.h>
int dup(int fd);
int dup2(int fd, int fd2);

dup函数返回一个新的文件描述符，这个描述符是当前可用文件描述符中的最小值。关闭标准输出后，调用dup函数，返回的描述符通常是1，即新打开的文件。

int fd;
close(1);
fd = open(FNAME, O_WRONLY | O_CREAT | O_TRUNC, 0600);
if(fd < 0)
{
    perror("open()");
    exit(1);
}
dup(fd); // 复制文件描述符，使新文件描述符指向同一文件
close(fd); // 关闭原始文件描述符
puts("hello!"); // 输出到文件

三、使用dup2的问题

使用dup2函数可以避免dup函数存在的问题，即先关闭原文件描述符再复制，可能会因为描述符被其他进程占用而导致错误。dup2是原子操作，可以同时关闭和复制文件描述符。

dup2(fd, 1); // 将新文件描述符复制到1，即标准输出
if(fd != 1)
    close(fd); // 如果原始文件描述符不是1，则关闭它
puts("hello!"); // 输出到文件

四、完整的示例

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

int main(){
    // 先备份现场
    int outfd = dup(1);
    // 打开Word文件，给他写权限，如果没有，则创建文件。
    int fd = open("world", O_WRONLY|O_CREAT, 0666);
    // 先做重定向，将标准输出重定向到fd对应的文件
    dup2(fd,1);
    // printf调用write的时候需要一个刷新机制，否则就写不到Word文件中，而是依然存储在缓冲区
    // 因此需要用fflush进行刷新。
    printf("hello world\n");

    // 需要来一次刷新下
    fflush(stdout);

    // 需要恢复文件描述符1，使其重新对应标准输出
    dup2(outfd,1);

    printf("hello world\n");
    close(fd);
    return 0;
}

在这个示例中，我们首先备份了标准输出的文件描述符，然后打开了一个新的文件用于重定向。使用dup2函数将标准输出重定向到新文件，然后输出内容。为了确保内容确实写入文件，我们使用了fflush函数来刷新输出缓冲区。最后，我们使用dup2函数将标准输出恢复到原来的文件描述符，并再次输出内容，这次内容会输出到屏幕上。

注意：在使用printf时，如果不指定文件描述符，它默认会输出到标准输出（文件描述符1）。在这个示例中，我们通过dup2函数指定了新的输出目标。

通过以上方法，我们可以在C语言中方便地实现输出重定向，这对于日志记录、文件打印等应用场景非常有用。