Win32 GUI C(++)应用程序将stdout和stderr都重定向到磁盘上的同一文件

时间:2019-07-08 17:20:15

标签: c++ c windows stdout dup2

我正在创建Windows服务,该服务不能具有关联的控制台。因此,我想将stdout和stderr重定向到(相同的)文件。这是我到目前为止发现的:

  1. 在C ++中重定向coutcerr可以通过changing the buffers完成,但这不会像puts或Windows I / O句柄那样影响CI / O。
  2. 因此,我们可以使用freopen重新打开stdout或stderr作为here之类的文件,但是我们不能两次指定相同的文件。
  3. 要同时使用相同的文件,我们可以使用dup2之类的here将stderr重定向到stdout。

到目前为止,一切顺利,当我们使用/SUBSYSTEM:CONSOLE(项目属性→链接器→系统)运行此代码时,一切正常:

#include <Windows.h>
#include <io.h>
#include <fcntl.h>
#include <cstdio>
#include <iostream>

void doit()
{
    FILE *stream;
    if (_wfreopen_s(&stream, L"log.log", L"w", stdout)) __debugbreak();
    // Also works as service when uncommenting this line: if (_wfreopen_s(&stream, L"log2.log", L"w", stderr)) __debugbreak();
    if (_dup2(_fileno(stdout), _fileno(stderr)))
    {
        const auto err /*EBADF if service; hover over in debugger*/ = errno;
        __debugbreak();
    }

    // Seemingly can be left out for console applications
    if (!SetStdHandle(STD_OUTPUT_HANDLE, reinterpret_cast<HANDLE>(_get_osfhandle(_fileno(stdout))))) __debugbreak();
    if (!SetStdHandle(STD_ERROR_HANDLE, reinterpret_cast<HANDLE>(_get_osfhandle(_fileno(stderr))))) __debugbreak();

    if (_setmode(_fileno(stdout), _O_WTEXT) == -1) __debugbreak();
    if (_setmode(_fileno(stderr), _O_WTEXT) == -1) __debugbreak();

    std::wcout << L"1☺a" << std::endl;
    std::wcerr << L"1☺b" << std::endl;

    _putws(L"2☺a");
    fflush(stdout);
    fputws(L"2☺b\n", stderr);
    fflush(stderr);

    const std::wstring a3(L"3☺a\n"), b3(L"3☺b\n");
    if (!WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), a3.c_str(), a3.size() * sizeof(wchar_t), nullptr, nullptr))
        __debugbreak();
    if (!WriteFile(GetStdHandle(STD_ERROR_HANDLE), b3.c_str(), b3.size() * sizeof(wchar_t), nullptr, nullptr))
        __debugbreak();
}

int        main() { doit(); }
int WINAPI wWinMain(HINSTANCE, HINSTANCE, PWSTR, int) { return doit(), 0; }

这很好地将以下文本写入log.log

1☺a
1☺b
2☺a
2☺b
3☺a
3☺b

当然我们需要表情符号,因此我们需要某种unicode。在这种情况下,我们使用宽字符,这意味着我们需要使用setmode,否则所有内容都会混乱)

但是现在回到最初的问题:这是作为不带控制台的服务来完成的,或者等效但更易于调试的GUI应用程序(/SUBSYSTEM:WINDOWS)。 问题在于,在这种情况下 dup2失败,因为fileno(stderr)不是有效的文件描述符,因为该应用最初没有关联的流。如前所述,herefileno(stderr) == -2

请注意,当我们第一次使用freopen将stderr作为另一个文件打开时,一切正常,但是我们创建了一个虚拟的空文件。

所以现在我的问题是:在最初没有流的应用程序中将stdout和stderr重定向到同一文件的最佳方法是什么?

回顾一下:问题在于when stdout or stderr is not associated with an output stream, fileno returns -2,所以我们不能将其传递给dup2

(我不想更改用于实际打印的代码,因为这可能意味着外部函数产生的某些输出将不会被重定向。)

1 个答案:

答案 0 :(得分:1)

这里是一个程序示例,该程序创建一个要写入的文件,然后使用CreateProcess并将stdoutstderr的处理过程设置为已创建文件的HANDLE 。这个示例只是以一个虚拟参数开始,使其向stdoutstderr写很多东西,这些东西将被写入output.txt

// RedirectStd.cpp

#include <iostream>
#include <string_view>
#include <vector>

#include <Windows.h>

struct SecAttrs_t : public SECURITY_ATTRIBUTES {
    SecAttrs_t() : SECURITY_ATTRIBUTES{ 0 } {
        nLength = sizeof(SECURITY_ATTRIBUTES);
        bInheritHandle = TRUE;
    }
    operator SECURITY_ATTRIBUTES* () { return this;  }
};

struct StartupInfo_t : public STARTUPINFO {
    StartupInfo_t(HANDLE output) : STARTUPINFO{ 0 } {
        cb = sizeof(STARTUPINFO);
        dwFlags = STARTF_USESTDHANDLES;
        hStdOutput = output;
        hStdError = output;
    }
    operator STARTUPINFO* () { return this; }
};

int cppmain(const std::string_view program, std::vector<std::string_view> args) {
    if (args.size() == 0) {
        // no arguments, create a file and start a new process
        SecAttrs_t sa;

        HANDLE hFile = CreateFile(L"output.txt",
            GENERIC_WRITE,
            FILE_SHARE_READ,
            sa, // lpSecurityAttributes
            CREATE_ALWAYS, // dwCreationDisposition
            FILE_ATTRIBUTE_NORMAL, // dwFlagsAndAttributes
            NULL // dwFlagsAndAttributesparameter
        );
        if (hFile == INVALID_HANDLE_VALUE) return 1;

        StartupInfo_t su(hFile); // set output handles to hFile
        PROCESS_INFORMATION pi;

        std::wstring commandline = L"RedirectStd.exe dummy";
        BOOL bCreated = CreateProcess(
            NULL,
            commandline.data(),
            NULL, // lpProcessAttributes
            NULL, // lpThreadAttributes
            TRUE, // bInheritHandles
            0, // dwCreationFlags
            NULL, // lpEnvironment
            NULL, // lpCurrentDirectory
            su, // lpStartupInfo
            &pi
        );
        if (bCreated == 0) return 2;
        CloseHandle(pi.hThread); // no need for this

        WaitForSingleObject(pi.hProcess, INFINITE); // wait for the process to finish         
        CloseHandle(pi.hProcess);
        CloseHandle(hFile);
    }
    else {
        // called with an argument, output stuff to stdout and stderr 
        for (int i = 0; i < 1024; ++i) {
            std::cout << "stdout\n";
            std::cerr << "stderr\n";
        }
    }

    return 0;
}

int main(int argc, char* argv[]) {
    return cppmain(argv[0], { argv + 1, argv + argc });
}