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

5

我正在尝试在C++中执行powershell命令,并通过管道获取其输出。

我的程序对于cmd.exe完美地工作。但是,当我尝试使用powershell.exe做同样的事情时,我只得到"W"作为输出。

下面是可以用于cmd.exe的代码,我已经在下面的代码中注释了需要修改以执行powershell.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,其中i = 0到strlen(output_cmd),我得到如下输出:
Windows PowerShell Copyright (C) 2014 Microsoft Corporation. All rights reserved. PS C:\Windows> 然后应用程序挂起!之后它不会输入任何内容或给出任何输出!

我想区别在于Powershell返回一个对象数组而不是一个普通字符串。使用这个命令“dir | out-string”会得到什么输出? - Lieven Keersmaekers
顺便提一下:“重定向到管道”没有任何意义,因为每个Windows进程的每个输入/输出已经被管道化了,这就是进程I/O的工作方式——即使Linux也是如此。您可以更改附加的管道、链接甚至合并它们,但除了管道I/O之外,不会有其他东西。大多数情况下,使用未命名的管道——如果您愿意,可以切换到命名管道。 - specializt
同时,while(1)糟糕的代码一定会在某个时候导致应用程序崩溃。在这里,有很多人会告诉你相反的意见 - 但也有很多人只是业余程序员...这只是一个友好的建议。永远不要使用无限循环。 - specializt
@specializt:谢谢!我会采纳你的建议。但现在,你能帮我解决我面临的问题吗? - K K
读取和写入字符串作为输入和输出是否存在问题? - K K
显示剩余6条评论
2个回答

0

你把字符串传到了错误的位置:

CreateProcess(TEXT("C:\\Windows\\System32\\cmd.exe")

实际上第一个参数应该是NULL:CreateProcess(NULL, TEXT("C:\\Windows\\System32\\WindowsPowerShell\\v1.0\\powershell.exe")


0
你似乎最困惑的是宽字符或字节字符。在经典的ASCII字符串中,每个字符占用一个字节。现代系统使用Unicode,其中最流行的两种编码方式是UTF-8(在Unix上很受欢迎)和UTF-16(大多数Windows API使用)。Windows通常(总是?)使用小端序列,其中第一个字节是低8位,第二个字节是高8位。在Unicode中,前127个代码点与ASCII的前127个字符向后兼容,因此ASCII中的字母“W”为0x57,在UTF-16中为0x57 0x00
你正在混合使用ReadFile和printf。ReadFile对缓冲区和读取的字节数使用显式长度,因此它可以愉快地将UTF-16作为二进制数据传输。但是,printf来自于以NUL字节终止的ASCII字符串的旧传统。因此,从printf的角度来看,你正在给它一个长度为1的字符串,因为第二个字节是0x00

看一下这个有关printf中宽字符的问题,看看你应该怎么做。

默认情况下,PowerShell将UTF-16写入其控制台,而旧的cmd.exe仍在使用ASCII字符串。事实证明,除非您传递选项-Command -,否则PowerShell根本不使用其输入句柄。然而,通过该选项,它会切换回用于输出和输入的ASCII字符串。因此,您真正需要做的就是传递该命令行选项,然后事情应该开始像Cmd.exe一样工作。

我正在为此编写一个Perl模块,而不是C ++,但您可能会发现我的源代码有所帮助。

顺便说一句,我对此页面上的其他错误信息感到不安:

  • 在Windows中,管道句柄、控制台句柄和文件句柄具有不同的行为,它们并不是“所有管道”。可以说它们都是句柄,你可以读/写每一个句柄,并将它们用于程序的stdin/stdout/stderr。

  • while(1) { if (!condition) break; ... } while(condition) { ... } 完全等效,除了风格之外没有避免使用它的理由。如果你的条件不适合一行表达式,使用while(1)是完全合理的。

  • 不应该将CreateProcess的第一个参数设置为NULL,因为这会明确告诉Windows你打算执行哪个程序。如果你将它传递到第二个参数中,则需要确保它被正确引用,因为路径中带有空格的程序可能会运行不同于预期的程序,甚至成为安全漏洞。你不必使用第一个参数,但如果可以,请使用它。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接