我编写了一个小程序,它应该通过命令重定向cmd.exe stdout,并使用管道来获取输入。
使用ipconfig
之类的命令时,程序运行得非常好,但是当我尝试将tasklist
作为命令传递时,程序就会卡住并且不会得到任何输出。 / p>
以下是代码:
#include <Windows.h>
#include <tchar.h>
#include <iostream>
#define BUFFSIZE 100000
int wmain(int argc, wchar_t* argv[])
{
HANDLE hReadPipe, hWritePipe;
SECURITY_ATTRIBUTES saAttr;
saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
saAttr.bInheritHandle = TRUE;
saAttr.lpSecurityDescriptor = NULL;
if (CreatePipe(&hReadPipe, &hWritePipe, &saAttr, 0) != 0)
{
STARTUPINFO structStartUpInfo;
ZeroMemory(&structStartUpInfo, sizeof(STARTUPINFO));
structStartUpInfo.cb = sizeof(STARTUPINFO);
structStartUpInfo.wShowWindow = SW_HIDE;
structStartUpInfo.hStdInput = hReadPipe;
structStartUpInfo.hStdOutput = hWritePipe;
structStartUpInfo.dwFlags = STARTF_USESTDHANDLES;
PROCESS_INFORMATION structProcInf;
ZeroMemory(&structProcInf, sizeof(PROCESS_INFORMATION));
if (CreateProcess(NULL, L"tasklist", NULL, NULL, TRUE, NORM AL_PRIORITY_CLASS, NULL, NULL, &structStartUpInfo, &structProcInf) != 0)
{
WaitForSingleObject(structProcInf.hProcess, INFINITE);
CHAR output[BUFFSIZE];
DWORD wdByteToRead = BUFFSIZE;
DWORD dwByteread;
ReadFile(hReadPipe, output, wdByteToRead, &dwByteread, NULL);
std::string final = output;
final = final.substr(0, dwByteread);
std::cout << final.c_str();
}
}
return 0;
我该如何解决这个问题?
答案 0 :(得分:1)
管道具有最大缓冲区大小。 tasklist
可能在没有阻塞的情况下写入缓冲区大小,但在此之后写入时会阻塞。
通过查看代码,子进程需要读取自己的输入以避免阻塞。
你需要调用CreatePipe
两次,然后保持一对的读管和另一对的写管(自己的过程),并提供前一对的写管和读管后一对对子进程的影响。
即便如此,管道的缓冲区大小也是最大的。为避免子进程被阻塞,您必须继续读取父进程中的管道。
编辑:如果您不打算写入子进程,您也可以关闭父进程的写入端或将hStdInput
设置为零。
此外,您应该使用另一个管道的写入端或将其设置为零来初始化hStdError
。
基于(其中一个)Microsoft C运行时的程序在零时忽略标准句柄,但其他程序可能不会,因此有3对管道可能是最安全的方法。但是,你必须从该管道读取,以避免子进程在写入时阻塞,但不太可能。