单元测试中如何稳定重定向与恢复stderr？

我在单元测试中需要重定向stderr来验证相关逻辑，写了一段重定向和恢复的代码，但第二次调用redirect()时dup(STDERR_FILENO)返回-1，触发断言失败导致崩溃。以下是我的代码：

#include <stdio.h>
#include <unistd.h>
#include <assert.h>

int old_std;
void redirect()
{
    old_std = dup(STDERR_FILENO);
    assert(old_std != -1);    // 第二次调用redirect时断言失败
    freopen("test.txt", "w", stderr);
    setbuf(stderr, NULL);
}

void revert()
{
    fclose(stderr);
    stderr = fdopen(old_std, "w");
    fflush(stderr);
    setbuf(stderr, NULL);
}

int main(int argc, char **argv)
{
    redirect();
    // 测试逻辑
    revert();
    redirect();
    // 测试逻辑
    revert();
    return 0;
}

恳请提供单元测试中稳定重定向与恢复stderr的方法。

问题根源 #

你的代码核心错误在于混淆了stdio流（stderr）和文件描述符（STDERR_FILENO，固定为2）：

1. 第一次调用freopen时，会关闭stderr流绑定的原文件描述符（即STDERR_FILENO=2），再将test.txt的fd绑定到stderr流。此时fd2已被关闭，第二次调用dup(STDERR_FILENO)本质是dup一个已关闭的fd，必然返回-1触发断言。
2. 直接修改全局stderr变量的做法存在未定义行为，标准库对stderr的维护有隐含规则，手动赋值可能导致流状态混乱。

推荐解决方案：文件描述符级重定向（可靠） #

直接操作STDERR_FILENO文件描述符，通过dup/dup2实现重定向与恢复，既能覆盖所有向stderr的输出（包括直接用write的代码），又不会破坏标准库流状态：

#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <assert.h>
#include <string.h>

static int original_stderr_fd = -1;

void redirect_stderr(const char *path) {
    assert(original_stderr_fd == -1); // 防止重复重定向
    // 保存原始stderr的文件描述符
    original_stderr_fd = dup(STDERR_FILENO);
    assert(original_stderr_fd != -1);

    // 打开目标文件（覆盖写入，不存在则创建）
    int new_fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0644);
    assert(new_fd != -1);

    // 将STDERR_FILENO（2）重定向到新文件
    dup2(new_fd, STDERR_FILENO);
    close(new_fd); // dup2后无需保留原fd，关闭释放资源

    // 关闭stdio缓冲，确保输出实时写入文件
    setbuf(stderr, NULL);
}

void revert_stderr() {
    assert(original_stderr_fd != -1); // 确保已执行过重定向
    // 将STDERR_FILENO恢复为原始fd
    dup2(original_stderr_fd, STDERR_FILENO);
    close(original_stderr_fd);
    original_stderr_fd = -1; // 重置标记，允许下次重定向

    setbuf(stderr, NULL);
}

int main(int argc, char **argv) {
    redirect_stderr("test.txt");
    fprintf(stderr, "第一次重定向输出\n");
    revert_stderr();

    fprintf(stderr, "恢复后的标准错误输出\n");

    redirect_stderr("test2.txt");
    fprintf(stderr, "第二次重定向输出\n");
    revert_stderr();

    return 0;
}

备选方案：stdio流级重定向（仅覆盖stdio函数输出） #

如果你的测试场景只需要捕获fprintf等stdio函数的输出，可以直接保存并替换stderr指针（但无法捕获直接调用write(2, ...)的输出）：

#include <stdio.h>
#include <unistd.h>
#include <assert.h>

static FILE *original_stderr = NULL;

void redirect_stderr(const char *path) {
    assert(original_stderr == NULL);
    // 保存原始stderr流指针
    original_stderr = stderr;
    // 打开新文件并绑定到stderr
    stderr = fopen(path, "w");
    assert(stderr != NULL);
    setbuf(stderr, NULL);
}

void revert_stderr() {
    assert(original_stderr != NULL);
    // 关闭当前重定向的流
    fclose(stderr);
    // 恢复原始stderr流
    stderr = original_stderr;
    original_stderr = NULL;
    setbuf(stderr, NULL);
}

int main(int argc, char **argv) {
    redirect_stderr("test.txt");
    fprintf(stderr, "第一次重定向输出\n");
    revert_stderr();

    fprintf(stderr, "恢复后的输出\n");

    redirect_stderr("test2.txt");
    fprintf(stderr, "第二次重定向输出\n");
    revert_stderr();

    return 0;
}

内容的提问来源于stack exchange，提问作者Wei Li