繁体   English   中英

将Powershell.exe的输入和输出重定向到C ++中的管道

[英]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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM