c++ - 无需等待缓冲区的 Win32 ReadFile 输出

Question

我使用 CreateProcess() 向 cmd.exe 运行命令。命令本身有无穷无尽的输出，所以我使用了一个我从 this answer 修改的函数将部分输出转换为字符串。

#define BUFSIZE 4096
HANDLE g_hChildStd_OUT_Rd = NULL;

std::string ReadFromPipe(PROCESS_INFORMATION piProcInfo) {
    DWORD dwRead; 
    CHAR chBuf[BUFSIZE];
    ReadFile( g_hChildStd_OUT_Rd, chBuf, BUFSIZE, &dwRead, NULL);
    std::string s(chBuf, dwRead);
    return s;
}

但是这段代码会产生一些问题。

首先，每次我调用它时，它可能会在等待输出缓冲到 4096 字节时卡住程序。

其次，它将始终只获取输出队列中接下来的 4096 个字节。 (即使当前输出大很多)

我想要的是调用该函数并获取同时输出的所有数据，并且还能够设置要获取的最小字节数(而不是缓冲区)。如果最小字节数尚不可用，我希望它完全跳过 ReadFile() 并只返回 false。 (而不是卡住应用程序)

这可能吗？

Answer 1

Arkadiy 的评论将我引向了正确的方向。我在 PeekNamedPipe()

的帮助下让它工作

我当前的代码:

#include <string>
#include <iostream>
#include <windows.h> 
#include <stdio.h>

#pragma warning( disable : 4800 ) // hide bool warning
#define BUFSIZE 1024
HANDLE g_hChildStd_OUT_Rd = NULL;

std::string ReadFromPipe(PROCESS_INFORMATION piProcInfo) {
    CHAR chBuf[BUFSIZE];
    DWORD dwRead;
    DWORD avail;
    bool bSuccess = FALSE;
    bool tSuccess = FALSE;
    std::string out = "";
    tSuccess = PeekNamedPipe( g_hChildStd_OUT_Rd, NULL, 0, NULL, &avail, NULL );
    if (tSuccess && avail >= BUFSIZE) {
        while (avail >= BUFSIZE) {
            bSuccess=ReadFile( g_hChildStd_OUT_Rd, chBuf, BUFSIZE, &dwRead, NULL);
            if( ! bSuccess || dwRead == 0 ) break; 
            std::string s(chBuf, dwRead);
            out += s;
            avail = avail - BUFSIZE;
        }
        return out;
    } else {
        return "[false]";
    }
}

调用此函数时，将返回自上次调用以来的输出。 BUFSIZE 现在充当最小缓冲区，如果输出小于最小缓冲区，该函数将返回 [false](作为字符串，而不是 bool)。