[英]Redirect Input and Output of Powershell.exe to Pipes in C++
我試圖在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),我得到如下給出的輸出:
indows Power S hell C版權所有(C)2 0 1 4 M icrosoft C orporation。 版權所有 。
PSC:\\ W indows>
然后應用程序掛起! 它不接受任何輸入,或者在此之后給出任何輸出!
你把字符串傳遞給錯誤的地方:
CreateProcess(TEXT("C:\\\\Windows\\\\System32\\\\cmd.exe")
實際上第一個參數應該是NULL: CreateProcess(NULL, TEXT("C:\\\\Windows\\\\System32\\\\WindowsPowerShell\\\\v1.0\\\\powershell.exe")
你的主要困惑點似乎是寬字符或字節字符。 在經典ASCII字符串中,每個字符都是一個字節。 現代系統使用Unicode,兩種最流行的風格是UTF-8 (在unix上很流行)和UTF-16 ,大多數Windows API都使用它。 最常見的Windows(總是?)使用little-endian變種,其中第一個字節是低8位,第二個字節是高8位。 在unicode中,前127個代碼點向后兼容ASCII的前127個字符,因此ASCII中的字母“W”為0x57
,而UTF-16中的字母“W”為0x57 0x00
。
您正在將ReadFile與printf混合使用。 ReadFile使用顯式長度來讀取緩沖區和字節,因此它可以很好地將UTF-16作為二進制數據傳輸。 但是,printf來自一個舊的傳統的ASCII字符串,它以NUL字節終止。 所以從printf的角度來看,你給它一個長度為1的字符串,因為第二個字節是0x00
。
看看有關printf的寬字符的這個問題 ,看看你應該做些什么不同的事情。
默認情況下,PowerShell將UTF-16寫入其控制台,其中舊的cmd.exe仍在使用ASCII字符串。 事實證明,PowerShell根本不使用它的輸入句柄,除非你傳遞選項-Command -
。 但是,使用該選項,它會切換回ASCII字符串以進行輸出和輸入。 因此,您真正需要做的就是通過該命令行選項,事情應該像Cmd.exe一樣開始工作。
我正在為此創建一個perl模塊,而不是C ++,但您可能會發現我的源代碼很有幫助。
順便說一句,我對此頁面上的其他錯誤信息感到不安:
在Windows中,管道句柄,控制台句柄和文件句柄各自具有不同的行為,而不是“所有管道”。 可以說它們都是句柄,並且您可以讀取/寫入每個句柄並將它們用於程序的stdin / stdout / stderr。
while(1) { if (!condition) break; ... }
while(1) { if (!condition) break; ... }
絕對在功能上等同於while(condition) { ... }
並且除了樣式之外沒有理由避免它。 如果你的情況不適合單行表達,那么使用while(1)
是完全合理的。
您不應該將CreateProcess的第一個參數設置為NULL,因為它不明確地告訴Windows您要執行哪個程序。 如果你在第二個參數中傳遞它,那么你需要確保它被正確引用,因為其中包含空格的路徑可能運行與預期不同的程序,甚至成為安全漏洞。 您不必使用第一個參數,但如果你能做到這一點。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.