【问题标题】:Redirect Input and Output of Powershell.exe to Pipes in C++将 Powershell.exe 的输入和输出重定向到 C++ 中的管道
【发布时间】: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


【解决方案1】:

你把字符串传到了错误的地方:

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

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

【讨论】:

    【解决方案2】:

    您的主要困惑似乎是宽字符或字节字符。在经典的 ASCII 字符串中,每个字符是一个字节。现代系统使用 Unicode,最流行的两种风格是 UTF-8(在 unix 上流行)和 UTF-16,大多数 Windows API 都使用它们。 Windows 最常(总是?)使用 little-endian 变体,其中第一个字节是低 8 位,第二个字节是高 8 位。在 unicode 中,前 127 个代码点向后兼容 ASCII 的前 127 个字符,因此 ASCII 中的字母“W”是 0x57,在 UTF-16 中是 0x57 0x00

    您正在将 ReadFile 与 printf 混合使用。 ReadFile 对缓冲区和读取的字节使用明确的长度,因此它可以愉快地将 UTF-16 作为二进制数据传输。但是,printf 来自以 NUL 字节结尾的 ASCII 字符串的古老传统。所以从 printf 的角度来看,你给它一个长度为 1 的字符串,因为第二个字节是 0x00

    看看这个question about wide characters with printf,看看你应该做些什么不同的事情。

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

    我正在为此开发一个 perl 模块,而不是 C++,但您可能会发现 my source code 很有帮助。

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

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

    • while(1) { if (!condition) break; ... } 在功能上绝对等同于 while(condition) { ... } 除了风格之外,没有理由避免它。如果您的情况不能很好地用单行表达式表达,那么使用while(1) 是完全合理的。

    • 您不应该将 CreateProcess 的第一个参数设置为 NULL,因为它明确地告诉窗口您打算执行哪个程序。如果您在第二个参数中传递它,那么您需要确保正确引用它,因为其中包含空格的路径可能会运行与预期不同的程序,甚至会成为安全错误。您没有必须使用第一个参数,但如果可以的话,请使用。

    【讨论】:

      猜你喜欢
      • 2016-04-26
      • 1970-01-01
      • 2012-05-28
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-05-08
      • 2021-10-16
      • 2013-07-16
      相关资源
      最近更新 更多