具有2个重定向管道的CreateProcess返回退出代码1

问题描述

我是WIN32 API的新手,可以使用一些帮助来确定代码中缺少的内容。 我的目标是运行批处理脚本(或PowerShell脚本),等待完成,然后在完成时获取stdout / stderr。

有关msdn的thisthis文章内容涉及创建进程并将stdout和stderr重定向到句柄。 我混搭在一起,并给出了以下代码

int execute_commnad(std::string& command,std::string& output,int& timeout_seconds) {
    STARTUPINFO si;
    PROCESS_informatION pi;
    DWORD return_code = NULL;

    ZeroMemory(&pi,sizeof(pi));

    ZeroMemory(&si,sizeof(si));
    si.cb = sizeof(si);
    si.hStdOutput = child_stdout_handle;
    si.hStdError = child_stdout_handle;
    si.dwFlags |= STARTF_USESTDHANDLES;

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

    // Connect Child stdout to parent_stdout_handle
    if (!CreatePipe(&parent_stdout_handle,&child_stdout_handle,&saAttr,0))
    {
        output = "Could not create stdout pipe. Error: " + std::to_string(GetLastError());
        close_handles();
        return -1;
    }

    if (! SetHandleinformation(parent_stdout_handle,HANDLE_FLAG_INHERIT,0))
    {
        output = "Could not set handle information for stdout pipe. Error: " + std::to_string(GetLastError());
        close_handles();
        return -1;
    }

    // Create Process
    if (!CreateProcess(NULL,LPSTR(command.c_str()),NULL,TRUE,&si,&pi)) {
        output = "Failed to create process. Error: " + std::to_string(GetLastError());
        close_handles();
        return -1;
    }
    CloseHandle(child_stdout_handle);


    // Wait for process to finish or timeout
    if (timeout_seconds == NULL) {
        WaitForSingleObject(pi.hProcess,INFINITE);
    }
    else {
        WaitForSingleObject(pi.hProcess,DWORD(timeout_seconds * 1000));
    }
    
    if (!GetExitCodeProcess(pi.hProcess,&return_code)) {
        output = "Failed to fetch exit code";
        close_handles();
        return -1;
    }
    
    if (STILL_ACTIVE == return_code) {
        TerminateProcess(sub_process,return_code);
        output = "Command did not finish within defined timeout threshold";
    }
    else if (STATUS_PENDING == return_code) {
        output = "process is pending";
    }
    else {
        DWORD dwRead;
        CHAR chBuf[4096];
        BOOL bSuccess = FALSE;

        for (;;) {
            bSuccess = ReadFile(parent_stdout_handle,chBuf,4096,&dwRead,NULL);
            if (! bSuccess || dwRead == 0) break;
            output = *reinterpret_cast<std::string*>(parent_stdout_handle);
        }
    }
    
    close_handles();
    return 0;    
};


void close_handles(void) {
    CloseHandle(parent_stdout_handle);
    CloseHandle(child_stdout_handle);
    CloseHandle(sub_process);
};

在执行上述功能之前,我使用包含回声的std::ofstream创建.bat文件,并将文件的FQN保存到名为file_name的变量中,然后运行以下代码

int rc = 0;
std::string cmd = "cmd.exe /c " + file_name + " " + script_params;
rc = execute_commnad(cmd,this->output,timeout_in_seconds);

调试时,这些是我在GetExitCodeProcess之后得到的值:

  • cmd:cmd.exe / c“ O:\\ Programming \\ cpp \ ... \\ script_file.bat” 1 2 3
  • 返回码:1

我尝试用FQN替换cmd.exe,但结果相同,尝试运行像cmd.exe /c "echo hello"这样简单的内容,但仍返回return_code 1。

最近两天我一直在努力寻找原因,但无济于事。

除了无效的代码(在性能方面)外,还有人建议可能是什么问题吗?

谢谢!

解决方法

在将句柄传递到child_stdout_handle之前,请确保初始化CreateProcess

此外,您需要打印在chBuf中读取的ReadFile的值,而不是parent_stdout_handle。将文件句柄转换为string是没有意义的,您可以尝试:

    for (;;) {
        bSuccess = ReadFile(parent_stdout_handle,chBuf,4096,&dwRead,NULL);
        if (!bSuccess || dwRead == 0) break;
        chBuf[dwRead] = '\0';
        output += chBuf;
    }
    cout << output << endl;