5

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

服务器

#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;
}
4

4 回答 4

15

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

这样做的原因有点牵涉。您为 I/O 重定向提供给 CreateProcess 的任何“标准句柄”都必须是非重叠的(即,不支持重叠的 I/O)。Windows 上的套接字在使用 socket() 创建时以重叠方式打开,如果使用 WSASocket 如上所述创建,则非重叠方式。

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

于 2011-04-20T04:43:04.683 回答
4

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

启动外部进程时,CreatePipe()改为使用为进程的重定向 STDIN/OUT/ERR 句柄创建匿名管道,然后使用ReadFile()WriteFile()(或等效项)send()recv()手动通过管道代理套接字和进程之间的数据。

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

于 2011-02-14T23:01:26.453 回答
2

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

看看netcatwin32版本的源代码(如果你能找到它),它几乎完全正确地完成了socket <-> stdin和stdout转发。

于 2011-02-14T14:56:04.317 回答
2

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

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

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

于 2016-05-09T10:58:40.150 回答