问题描述
我是WIN32 API的新手,可以使用一些帮助来确定代码中缺少的内容。 我的目标是运行批处理脚本(或PowerShell脚本),等待完成,然后在完成时获取stdout / stderr。
有关msdn的this和this文章,内容涉及创建进程并将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;