如果stdin被另一个进程的管道替换,Std::getline将中断

std::getline breaks if stdin is replaced by pipe of another process

本文关键字:Std getline 中断 替换 管道 stdin 另一个 进程 如果      更新时间:2023-10-16

我有一个简单的示例程序,它从std::cin读取并写入std::cout。如果在cmd.exe或visual studio调试器中运行,它可以正常工作。代码(server.cpp):

#include <iostream>
#include <istream>
#include <ostream>
#include <string>
int main(int argc, char* argv[])
{
    std::string input;
    while (std::getline(std::cin, input))
    {
        if (input == "dog")
        {
            std::cout << "cat" << std::endl;
        }
        else if (input == "white")
        {
            std::cout << "black" << std::endl;
        }
        else if (input == "quit")
        {
            std::cout << "exiting" << std::endl;
            return 0;
        }
        else if (input != "")
        {
            std::cout << "unknown" << std::endl;
        }
    }
    std::cout << "error" << std::endl;
}

现在我想在另一个进程中运行它,这个进程写入标准输入并从标准输出读取。我创建了两个管道,并使用CreateProcess启动一个进程,其中一个管道的读句柄作为StdInput句柄,另一个管道的写句柄作为stoutput句柄。代码(client.cpp):

#include <Windows.h>
#include <cassert>
#include <iostream>
#include <ostream>
#include <string>
namespace
{
    class Server
    {
    public:
        Server() :
            m_pi()
        {
            SECURITY_ATTRIBUTES sa = {sizeof(SECURITY_ATTRIBUTES)};
            sa.bInheritHandle = TRUE;
            assert(CreatePipe(&m_ro, &m_wo, &sa, 0));
            assert(SetHandleInformation(m_ro, HANDLE_FLAG_INHERIT, 0));
            assert(CreatePipe(&m_ri, &m_wi, &sa, 0));
            assert(SetHandleInformation(m_ri, HANDLE_FLAG_INHERIT, 0));
            STARTUPINFO si = {sizeof(STARTUPINFO)};
            si.dwFlags |= STARTF_USESTDHANDLES;
            si.hStdInput = m_ri;
            si.hStdOutput = m_wo;
            assert(CreateProcess(L"..\Debug\server.exe", 0, 0, 0, 1, 0, 0, 0, &si, &m_pi));
        }
        ~Server()
        {
            execute("quitn");
            assert(WaitForSingleObject(m_pi.hProcess, INFINITE) != WAIT_FAILED);
            assert(CloseHandle(m_pi.hThread));
            assert(CloseHandle(m_pi.hProcess));
            assert(CloseHandle(m_wi));
            assert(CloseHandle(m_ri));
            assert(CloseHandle(m_wo));
            assert(CloseHandle(m_ro));
        }
        std::string execute(std::string const& cmd)
        {
            DWORD num_bytes;
            assert(WriteFile(m_wi, cmd.c_str(), (DWORD)cmd.size(), &num_bytes, 0));
            std::string output;
            DWORD n = 0;
            while (n == 0)
            {
                Sleep(0);
                assert(PeekNamedPipe(m_ro, 0, 0, 0, &n, 0));
                if (n > 0)
                {
                    output.resize(n);
                    assert(ReadFile(m_ro, &output[0], n, &num_bytes, 0));
                }
            }
            return output;
        }
    private:
        HANDLE m_wo, m_ro, m_wi, m_ri;
        PROCESS_INFORMATION m_pi;
    };
    Server g_server;
}
int main(int argc, char* argv[])
{
    std::cout << g_server.execute("whiten") << std::endl;
    std::cout << g_server.execute("foobarn") << std::endl;
    std::cout << g_server.execute("dogn") << std::endl;
}

问题是客户端只收到消息"error",所以服务器的std::cin似乎被破坏了。

我的问题是,我做错了什么?

您正在禁用子进程将用于从stdin读取的句柄的继承-子进程需要继承该句柄。而不是:

SetHandleInformation(m_ri, HANDLE_FLAG_INHERIT, 0);

尝试以下操作来禁用服务器进程用于写入子进程stdin的句柄的继承:

SetHandleInformation(m_wi, HANDLE_FLAG_INHERIT, 0);