c++ - 从 C++ 中的另一个程序重定向标准输出

Question

我正在编写单元测试，因此无法更改我正在测试的文件中的代码。我正在测试的代码在 cout 中有消息，我试图将这些消息重定向到文件中以检查以确保程序输出正确的消息。有没有人有办法在另一个不会导致延迟的程序中重定向标准输出？我已经尝试过 freopen() ，这导致我的程序由于某种原因挂起。

Answer 1

你可以用它创建一个filebufthen replacecout的streambuf：

{
  std::filebuf f;
  f.open("output.txt", std::ios::out);
  std::streambuf* o = std::cout.rdbuf(&f);
  std::cout << "hello" << std::endl;  // endl will flush the stream
  std::cout.rdbuf(o);
}

您需要再次恢复cout的原始流缓冲区（或将其设置为空指针），否则在刷新和销毁全局流时它可能会崩溃，因为它filebuf已经超出范围。

Answer 2

您可以使用“open()”和“dup2()”。您可以使用下面提供的助手。如何使用它们的示例：

void
code_to_test ()
{
    std::cout << "Here we go" << std::endl;
    std::cerr << "Danger danger" << std::endl;
}

run_test(code_to_test);

run_test助手调用重定向，并运行测试代码。

template <typename TEST> void
run_test (TEST t, bool append = false) {
    flush_output();
    Redirect o(1, "/tmp/test_stdout", append);
    Redirect e(2, "/tmp/test_stderr", append);
    t();
    flush_output();
}

flush_output助手刷新流。

void flush_output () {
    fflush(stdout);
    fflush(stderr);
    std::cout.flush();
    std::cerr.flush();
}

该类Redirect影响构造函数中的重定向。它在析构函数中恢复原始描述符。

class Redirect
{
    int m_what;
    int m_old_what;
public:
    Redirect (int what, std::string where, bool append = false)
        : m_what(what), m_old_what(dup(what)) {
        int flags = O_CREAT|O_WRONLY;
        if (append) flags |= O_APPEND;
        int f = open(where.c_str(), flags, 0660);
        dup2(f, m_what);
        close(f);
    }
    ~Redirect () {
        dup2(m_old_what, m_what);
        close(m_old_what);
    }
};