【发布时间】:2016-10-21 13:25:26
【问题描述】:
我正在尝试在 C++ 中执行 powershell 命令并通过管道获取其输出。
我的程序非常适合 cmd.exe。但是,当我尝试用 powershell.exe 做同样的事情时,我只会得到“W”作为输出。
我已经注释了下面代码中需要修改才能执行powershell.exe的行 以下是适用于 cmd.exe 的代码:
HANDLE stdinRd, stdinWr, stdoutRd, stdoutWr;
DWORD readFromCmd();
DWORD writeToCmd(CString command);
int main(int argc,char* argv[])
{
SECURITY_ATTRIBUTES sa={sizeof(SECURITY_ATTRIBUTES), NULL, true};
if(!CreatePipe(&stdinRd, &stdinWr, &sa, 1000000) || !CreatePipe(&stdoutRd,&stdoutWr, &sa, 1000000))
{
printf("CreatePipe()");
}
STARTUPINFO si;
PROCESS_INFORMATION pi;
GetStartupInfo(&si);
si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
si.wShowWindow = SW_HIDE;
si.hStdOutput = stdoutWr;
si.hStdError = stdoutWr;
si.hStdInput = stdinRd;
// If powershell.exe is invoked, it does not work, however works for cmd.exe
//if(!CreateProcess(TEXT("C:\\Windows\\System32\\WindowsPowerShell\\v1.0\\powershell.exe"), NULL, NULL, NULL, TRUE,0, NULL, TEXT("C:\\Windows"), &si, &pi))
if(!CreateProcess(TEXT("C:\\Windows\\System32\\cmd.exe"), NULL, NULL, NULL, TRUE,0, NULL, TEXT("C:\\Windows"), &si, &pi))
{
printf("CreateProcess()");
printf("CreateProcess() failed in initiatecmd(CString,int) method",0);
return -1;
}
writeToCmd(L"dir");
Sleep(1000);
readFromCmd();
getchar();
TerminateProcess(pi.hProcess,0);
CloseHandle(pi.hProcess);
return 0;
}
DWORD writeToCmd(CString command)
{
DWORD ret;
DWORD numberofbyteswritten;
command.AppendChar('\n');
LPSTR command_ANSI;
int size_needed = WideCharToMultiByte(CP_UTF8,0,command.GetString(),-1,NULL,0,NULL,NULL);
command_ANSI = (LPSTR) calloc(1, ( size_needed + 1 )* sizeof(char));
WideCharToMultiByte(CP_UTF8,0,command.GetString(),-1,command_ANSI,size_needed,NULL,NULL);
ret = WriteFile(stdinWr, command_ANSI, size_needed-1, &numberofbyteswritten, NULL);
if(ret==0)
{
printf("WriteFile()");
printf("WriteFile() method failed in writeToCmd(CString) method",0);
return 0;
}
CStringA temp;
temp.Format("%d",numberofbyteswritten);
temp += " bytes (Command:";
temp+=command;
temp+=") are successfully written to cmd";
printf("%s",temp);
return 1;
}
DWORD readFromCmd()
{
CString output_jsonstring;
DWORD ret;
DWORD dwRead;
while(1)
{
DWORD totalbytesavailable;
if(PeekNamedPipe(stdoutRd, NULL, 0, NULL, &totalbytesavailable, 0) == 0)
{
printf("PeekNamedPipe()");
printf("PeekNamedPipe() method failed in responseHandler() method",0);
return 0;
}
if(totalbytesavailable != 0)
{
char output_cmd[1000000];
if(ReadFile(stdoutRd, output_cmd, min(1000000,totalbytesavailable), &dwRead, NULL)==0)
{
printf("ReadFile()");
printf("ReadFile() method failed in responseHandler() method",0);
return 0;
}
int min = min(1000000,totalbytesavailable);
output_cmd[min]='\0';
printf("\n%s",output_cmd);
}
if(totalbytesavailable == 0)
break;
Sleep(100);
}
return 1;
}
如果 CreateProcess() 用于 powershell,它的工作方式不同,但我只得到 W 作为输出。
这是什么原因? 和 如何解决这个问题?
编辑 1:如果我在循环中逐字符显示 output_cmd 作为 output_cmd[i] 其中 i = 0 到 strlen(output_cmd),我会得到如下输出:
i n d o w s P o w e r S h e l l C o p y r i g h t ( C ) 2 0 1 4 M i c r o s o f t A l l r i g h t s r e s e r v e d 。
P S C : \ W i n d o w s >
然后应用程序挂起!它不接受任何输入,之后也不给出任何输出!
【问题讨论】:
-
我可以想象 Powershell 返回对象数组而不是纯字符串的区别。使用此命令
dir | out-string得到什么输出? -
btw : “重定向到管道” 没有任何意义,因为每个 Windows 进程的每个输入/输出都已经被管道化了,这就是我的过程/O 可以工作——甚至 Linux 也可以这样工作。您可以更改连接的管道、链接甚至合并它们,但除了管道 I/O 之外不会有其他任何东西。大多数时候,使用未命名管道 - 如果您喜欢,您可以切换到命名管道 ...
-
while(1)也是糟糕的代码,会让你的应用程序在某个时候崩溃。 SO上相当多的人会告诉你其他情况——但也有不少人是业余程序员……只是一个友好的建议。不要使用无限循环。永远。 -
@specializt:谢谢!我会接受这个建议。但是现在,你能帮我解决我面临的问题吗?
-
以字符串形式读写输入输出有问题吗?
标签: c++ powershell winapi pipe named-pipes