与 CreateNamedPipe 重叠的 CreateProcess 标准输出

Question

我正在尝试读取从 CreateProcess 开始的外部进程 (pgdump) 的标准输出。我使用匿名管道进行了此操作，但随后输出被阻止，并且不像我通过命令行执行它时那样（缺少最终输出）。我已经阅读了很多帖子，发现我需要 CreateNamedPipes 和 WaitForSingleObject 但我似乎无法让它工作。这是我使用匿名管道的工作代码，但被阻止了，我错过了输出的结尾

#include <QDebug>
#include <QString>

#include <windows.h>
#include <sstream>
#include <iostream>
#include <random>

int main()
{   
    #define BUFFERSIZE 256

    std::string program = "\"C:\\Program Files (x86)\\PostgreSQL\\10\\bin\\pg_dump.exe\"" +
            std::string( " --dbname=postgresql://postgresUser:PostGresql13@127.0.0.1:5432/employee -j1 -Fd -b -v -f "
                         "C:\\development\\myproject\\Build\\debug\\1-export.psql");

    HANDLE hReadStdOut = NULL;
    HANDLE hWriteStdOut = NULL;


    SECURITY_ATTRIBUTES saAttr;
    PROCESS_INFORMATION piProcInfo;
    STARTUPINFO siStartInfo;

    ZeroMemory( &saAttr, sizeof( SECURITY_ATTRIBUTES ));
    ZeroMemory( &piProcInfo, sizeof( PROCESS_INFORMATION ));
    ZeroMemory( &siStartInfo, sizeof( STARTUPINFO ));

    saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
    saAttr.bInheritHandle = TRUE;
    saAttr.lpSecurityDescriptor = NULL;

    OVERLAPPED sOverlap;


    if( !CreatePipe(&hReadStdOut, &hWriteStdOut, &saAttr, 0) )
    {
        std::ostringstream os;
        os << GetLastError();

        qDebug() << "create pipe error : " << QString::fromStdString( os.str());
    }


    TCHAR* szCmdline = new TCHAR[ program.size() + 1];
    szCmdline[ program.size()] = 0;
    std::copy( program.begin(), program.end(), szCmdline );

    siStartInfo.cb = sizeof(STARTUPINFO);
    siStartInfo.hStdError = hWriteStdOut;
    siStartInfo.hStdOutput = hWriteStdOut;
    siStartInfo.hStdInput = NULL;
    siStartInfo.dwFlags |= STARTF_USESTDHANDLES;

    BOOL bSuccess = CreateProcess( NULL, szCmdline, NULL, NULL, TRUE, CREATE_NO_WINDOW, NULL, NULL, &siStartInfo,&piProcInfo );

    if ( ! bSuccess )
    {
        std::ostringstream os;
        os << GetLastError();

        qDebug() << "create process error : " << QString::fromStdString( os.str());
    }
    else
    {
        CloseHandle( hWriteStdOut );

        DWORD err;
        DWORD nBytesRead;
        char buf[BUFFERSIZE + 1];

        int i(1);

        for(;;)
        {
            std::cout << "iteration " << std::to_string( i ) << std::endl;

            if( !ReadFile( hReadStdOut, buf, sizeof( buf), &nBytesRead, NULL) || !nBytesRead )
            {}

            if( GetLastError() == ERROR_SUCCESS )
            {
            }

            std::cout.flush();
            buf[nBytesRead] = '\0';
            std::string string_ = buf;
            std::cout << string_ << std::endl;

            std::size_t found = string_.find("contents of");

            if( !nBytesRead )
                break;

            i++;
        }

        CloseHandle(piProcInfo.hProcess);
        CloseHandle(piProcInfo.hThread);
    }

    return 0;
}

Answer 1

例如，如果你的 pg_dump.exe 只写一行到标准输出。有几种方法：

1. std::cout << "Hello World!\n";
2. printf("Hello World!\n");
3. std::cout << "Hello World!\n"<< std::endl;
4. WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), "Hello World!\n", 14, &dwWritten, NULL);

如果 pg_dump.exe exit 在写入该行后，ReadFile 不会被阻塞并返回“Hello World!”行。

但是，如果 pg_dump.exe 不退出 后写入该行并继续其他工作。前两种写入方式（1.、2.）将导致ReadFile 被阻止。但是，如果您使用第三种或第四种方式，ReadFile 将阻塞并返回“Hello World！”行。

这里的区别是前两种写入方式（std::cout << "Hello World!\n"; 和printf("Hello World!\n");）在写入结束时没有刷新操作，但第三种或第四种方式有。 std::endl 和 WriteFile 刷新输出缓冲区。

总结：

刷新输出以将其写入底层流（可能是文件、终端或管道）。在以下条件下刷新标准输出：

1. 程序正常结束时。
2. 最后使用std::endl。
3. 使用WriteFile。

您可以检查是否是您的情况。