将进程的 IO 重定向到 Windows 套接字答案

【问题标题】：Redirect IO of process to Windows socket将进程的 IO 重定向到 Windows 套接字
【发布时间】：2011-02-14 14:10:46
【问题描述】：

我是winsock的新手，我尝试编写一个接受新连接的服务器套接字，然后它调用一个外部可执行文件。我们如何将外部可执行文件的标准输入和标准输出重定向到已接受的客户端套接字。我用谷歌搜索并找到了下面的代码，但它不起作用。新进程创建成功，但客户端无法从新进程接收任何数据。我正在使用 Windows 7 和 Visual Studio 2008 Express 版本。任何帮助和 cmets 表示赞赏。非常感谢！

服务器

#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>

#pragma comment(lib, "Ws2_32.lib")

#define DEFAULT_PORT "27015"
#define DEFAULT_BUFLEN 512

int _tmain(int argc, _TCHAR* argv[])
{

        WSADATA wsaData;
        int iResult;

        // Initialize Winsock
        iResult = WSAStartup(MAKEWORD(2,2), &wsaData);

        struct addrinfo *result = NULL, *ptr = NULL, hints;

        ZeroMemory(&hints, sizeof (hints));
        hints.ai_family = AF_INET;
        hints.ai_socktype = SOCK_STREAM;
        hints.ai_protocol = IPPROTO_TCP;
        hints.ai_flags = AI_PASSIVE;

        // Resolve the local address and port to be used by the server
        iResult = getaddrinfo(NULL, DEFAULT_PORT, &hints, &result);

        SOCKET ListenSocket = INVALID_SOCKET;

        ListenSocket = socket(result->ai_family, result->ai_socktype, result->ai_protocol);

         // Setup the TCP listening socket
        iResult = bind( ListenSocket, result->ai_addr, (int)result->ai_addrlen);

        freeaddrinfo(result);

        if ( listen( ListenSocket, SOMAXCONN ) == SOCKET_ERROR ) {
            printf( "Listen failed with error: %ld\n", WSAGetLastError() );
            closesocket(ListenSocket);
            WSACleanup();
            return 1;
        }

        SOCKET ClientSocket;

        ClientSocket = INVALID_SOCKET;

        // Accept a client socket
        ClientSocket = accept(ListenSocket, NULL, NULL);
        if (ClientSocket == INVALID_SOCKET) {
            printf("accept failed: %d\n", WSAGetLastError());
            closesocket(ListenSocket);
            WSACleanup();
            return 1;
        }

        STARTUPINFO si;
        memset( &si, 0, sizeof( si ) );
        si.cb = sizeof( si );
        si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
        si.wShowWindow = SW_HIDE;

        si.hStdInput = (HANDLE)ClientSocket;
        si.hStdOutput = (HANDLE)ClientSocket;
        si.hStdError = (HANDLE)ClientSocket;

        PROCESS_INFORMATION pi;

        TCHAR cmd[] = TEXT("C:\\Users\\dell\\Desktop\\hello.exe");

        if (CreateProcess(NULL, cmd, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi)) {
            printf("create process successfully\n");
            DWORD i = WaitForSingleObject( pi.hProcess, INFINITE );
            printf("%8x\n", i);
        }


        CloseHandle( pi.hProcess );
        CloseHandle( pi.hThread );
        closesocket( ClientSocket );
        WSACleanup();

}

客户

#define WIN32_LEAN_AND_MEAN

#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdlib.h>
#include <stdio.h>


// Need to link with Ws2_32.lib, Mswsock.lib, and Advapi32.lib
#pragma comment (lib, "Ws2_32.lib")
#pragma comment (lib, "Mswsock.lib")
#pragma comment (lib, "AdvApi32.lib")


#define DEFAULT_BUFLEN 512
#define DEFAULT_PORT "27015"


int _tmain(int argc, _TCHAR* argv[])
{
        WSADATA wsaData;
        SOCKET ConnectSocket = INVALID_SOCKET;
        struct addrinfo *result = NULL,
                        *ptr = NULL,
                        hints;
        char recvbuf[DEFAULT_BUFLEN];
        int iResult;
        int recvbuflen = DEFAULT_BUFLEN;

        // Initialize Winsock
        iResult = WSAStartup(MAKEWORD(2,2), &wsaData);

        ZeroMemory( &hints, sizeof(hints) );
        hints.ai_family = AF_UNSPEC;
        hints.ai_socktype = SOCK_STREAM;
        hints.ai_protocol = IPPROTO_TCP;

        // Resolve the server address and port
        iResult = getaddrinfo("localhost", DEFAULT_PORT, &hints, &result);

        // Attempt to connect to an address until one succeeds
        for(ptr=result; ptr != NULL ;ptr=ptr->ai_next) {
            // Create a SOCKET for connecting to server
            ConnectSocket = socket(ptr->ai_family, ptr->ai_socktype, 
                ptr->ai_protocol);
            if (ConnectSocket == INVALID_SOCKET) {
                printf("socket failed with error: %ld\n", WSAGetLastError());
                WSACleanup();
                return 1;
            }

            // Connect to server.
            iResult = connect( ConnectSocket, ptr->ai_addr, (int)ptr->ai_addrlen);
            if (iResult == SOCKET_ERROR) {
                closesocket(ConnectSocket);
                ConnectSocket = INVALID_SOCKET;
                continue;
            }
            break;
        }

        freeaddrinfo(result);

        if (ConnectSocket == INVALID_SOCKET) {
            printf("Unable to connect to server!\n");
            WSACleanup();
            return 1;
        }

        // Receive until the peer closes the connection
        do {

            iResult = recv(ConnectSocket, recvbuf, recvbuflen, 0);
            if ( iResult > 0 )
                printf("Bytes received: %d\n", iResult);
            else if ( iResult == 0 )
                printf("Connection closed\n");
            else
                printf("recv failed with error: %d\n", WSAGetLastError());

        } while( iResult > 0 );

        // cleanup
        closesocket(ConnectSocket);
        WSACleanup();

        return 0;
}

外部程序 hello.cpp

int _tmain(int argc, _TCHAR* argv[])
{
    printf("hello world!\n");       
    return 0;
}

【问题讨论】：

标签： c++ windows winsock winsock2

【解决方案1】：

实际上，您可以将 IO 重定向到套接字。只需确保使用 WSASocket 而不是 socket() 打开套接字，并且不要指定 WSA_FLAG_OVERLAPPED。

其中的原因有点复杂。您为 I/O 重定向提供给 CreateProcess 的任何“标准句柄”都必须是非重叠的（即不支持重叠的 I/O）。 Windows 上的套接字在使用 socket() 创建时会重叠打开，如果使用 WSASocket 创建如上所述，则不会重叠。

还要确保在 StartupInfo 中将 bInheritHandles 设置为 TRUE

【讨论】：

谢谢，我刚刚用 socket = WSASocket (AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, 0) 替换了我的套接字调用（我已经在做 WSAStartup 等），它就可以工作了 :)
这应该是正确的答案。无需使用中间管道。此答案中提供的信息对于适应通用 CreateProcess() 命令行实用程序很有用：github.com/cubiclesoft/createprocess-windows

【解决方案2】：

重叠的套接字（例如用socket()创建的）不能直接用于I/O重定向。

当启动外部进程时，使用CreatePipe() 代替为进程的重定向STDIN/OUT/ERR 句柄创建匿名管道，然后使用ReadFile() 和WriteFile()（或等效项）与send() 和@ 987654326@通过管道手动代理socket和进程之间的数据。

换句话说，当数据到达套接字时，从套接字中读取数据并将其写入 STDIN 管道。当进程输出数据时，从 STDOUT/ERR 管道中读取数据并将其写入套接字。

【讨论】：

【解决方案3】：

Windows 将几乎所有内容都视为HANDLE。不是内核对象的套接字是一个例外，它们不能用于重定向。您需要使用管道，如果您需要向/从套接字发送数据，则需要一个辅助进程来在管道和套接字之间复制数据。

看看netcat win32 版本的源代码（如果你能找到的话），它几乎完全完成了套接字标准输入和标准输出转发。

【讨论】：

【解决方案4】：

说套接字句柄不能用于将 IO 重定向到子进程是不正确的。我们一直在我们的网络服务器代码中为 CGI 做这件事。我们不使用命名或未命名的管道或中间进程。

a_mole 是正确的。您只需要非重叠的套接字。

您可以使用 WSASocket，或者如果您已经使用套接字创建了套接字（或者不能在您的目标操作系统上使用 WSASocket，例如 Pre Windows 7 SP1），您可以使用 setsockopt 将当前线程的 SO_OPENTYPE 设置为 SO_SYNCHRONOUS_NONALERT

【讨论】：