Windows:如何使用 CreateProcess 停止缓冲重定向的 Stdout

Question

我正在使用管道从命令行可执行文件获取重定向的标准输出输出。不幸的是，在该过程完成之前我没有得到任何输出。可执行文件在运行时输出进度状态，这就是我想要解析的内容。

BOOL RunCmd( char  *pCmd, 
             char  *pWorkingDir, 
             int    nWaitSecs, 
             BOOL   fRegImport, 
             DWORD *pdwExitCode )
{
  BOOL                fSuccess = TRUE;
  STARTUPINFO         si;
  PROCESS_INFORMATION pi;
  SECURITY_ATTRIBUTES sFileSecurity;
  ZeroMemory( &sFileSecurity, sizeof( sFileSecurity ) );
  sFileSecurity.nLength        = sizeof( sFileSecurity );
  sFileSecurity.bInheritHandle = TRUE;

  HANDLE hReadPipe  = NULL;
  HANDLE hWritePipe = NULL;

  fSuccess = CreatePipe( &hReadPipe, &hWritePipe, &sFileSecurity, 0 );
  SetHandleInformation( hReadPipe, HANDLE_FLAG_INHERIT, 0 ); 

  ZeroMemory( &si, sizeof(si) );
  ZeroMemory( &pi, sizeof(pi) );
  si.cb          = sizeof( si );
  si.dwFlags     = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
  si.hStdOutput  = hWritePipe;
  si.hStdError   = hWritePipe;
  si.wShowWindow = SW_HIDE;

  int rc;

  // Start the child process. 
  rc = CreateProcess( NULL, // No module name (use command line). 
                      pCmd, // Command line. 
                      NULL,             // Process handle not inheritable. 
                      NULL,             // Thread handle not inheritable. 
                      TRUE,
                      CREATE_NO_WINDOW,
                      NULL,             // Use parent's environment block. 
                      pWorkingDir,      // Working folder
                      &si,              // Pointer to STARTUPINFO structure.
                      &pi );            // Pointer to PROCESS_INFORMATION structure.


  if( ! rc )
    return FALSE;

  // Wait until child process exits.
  DWORD dwWaitResult;
  DWORD dwTimeStart = ::GetTickCount();
  DWORD dwTimeNow;

  #define BUFSIZE 4096  

  DWORD dwRead = 0; 
  DWORD dwAvail;
  CHAR  chBuf[ BUFSIZE ];
  BOOL  bSuccess = TRUE;

  for( ;; )
  {
    dwTimeNow    = ::GetTickCount();
    dwWaitResult = ::WaitForSingleObject( pi.hProcess, ONE_SECOND );
    dwRead       = 0;

    for( dwAvail = 0; PeekNamedPipe( hReadPipe, 0, 0, 0, &dwAvail, 0 ) && dwAvail; dwAvail = 0 )
    {
      dwRead = 0;
      ReadFile( hReadPipe, chBuf, min( BUFSIZE, dwAvail ), &dwRead, NULL );

      if( dwRead > 0 )
      {
        FILE *op = fopen( "c:\\REDIR.OUT", "a" );

        if( op )
        {
          fwrite( chBuf, 1, dwRead, op );
          fclose( op );
        }
      }
    }

    if( dwWaitResult == WAIT_OBJECT_0 )
    {
      DWORD dwExitCode;
      GetExitCodeProcess( pi.hProcess, &dwExitCode );

      if( pdwExitCode )
        (*pdwExitCode) = dwExitCode;

      break;
    }

    if( dwWaitResult == WAIT_TIMEOUT )
    {
      if( dwTimeNow - dwTimeStart < (DWORD)( ONE_SECOND * nWaitSecs ) )
        continue;
      else
      {
        fSuccess = FALSE;
        break;
      }
    }

    fSuccess = FALSE;
    break;
  }

  CloseHandle( pi.hProcess );
  CloseHandle( pi.hThread );
  CloseHandle( hReadPipe );
  CloseHandle( hWritePipe );

  return fSuccess;
}

每秒调用一次 PeekNamedPipe()，并且 dwAvail 每次都为零，直到该过程完成。

如何更快地获得流程的输出？从控制台运行该进程时，我会看到进度输出。该进程将在其输出中使用“\r”在同一行的开头显示百分比。

Answer 1

注意:我的回答仅涉及使用 MSVC 编译的可执行文件。

缓冲策略在 Microsoft C 运行时 (CRT) 库内进行编码。详情可了解here 。本文建议使用控制台句柄并操作控制台缓冲区来接收无缓冲的输出。

但是，Microsoft C 运行时中有一个未记录的功能，可以使用 STARTUPINFO 的 lpReserved2 和 cbReserved2 字段直接从其父进程继承带有一些内部标志的文件句柄 结构。您可以在 Microsoft Visual Studio 提供的 crt 源代码中找到详细信息。或者在 GitHub 上搜索类似 posfhnd 的内容。

我们可以利用这个未记录的功能来提供管道句柄并指定FOPEN | FDEV 标志给子进程，以欺骗子进程，将该管道句柄视为 FILE_TYPE_CHAR 句柄。

我有一个工作Python3 script演示此方法。