Перенаправить ввод-вывод процесса в Windows-сокет - PullRequest
3 голосов
/ 14 февраля 2011

Я новичок в winsock, я попытался написать серверный сокет, который принимает новое соединение, затем он вызывает внешний исполняемый файл. Как мы можем перенаправить stdin и stdout внешнего исполняемого файла в клиентский сокет, который был принят. Я гуглил и нашел код ниже, но он не работает. Новый процесс был успешно создан, но клиент не смог получить никаких данных от нового процесса. Я использую Windows 7 и Visual Studio 2008 Express Edition. Любые подсказки и комментарии приветствуются. Большое спасибо!

сервер

#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 ]

11 голосов
/ 20 апреля 2011

На самом деле, вы можете перенаправить ввод-вывод в сокеты. Просто убедитесь, что вы открываете сокет с помощью WSASocket вместо socket () и НЕ указываете WSA_FLAG_OVERLAPPED.

Причина этого немного запутана. Любые «Стандартные дескрипторы», которые вы предоставляете CreateProcess для перенаправления ввода / вывода, должны быть не перекрыты (то есть не поддерживать перекрывающиеся операции ввода / вывода). Сокеты в Windows открываются перекрытыми при создании с помощью socket () и не перекрываются при создании, как описано выше с помощью WSASocket.

Также убедитесь, что для bInheritHandles установлено значение TRUE в StartupInfo

4 голосов
/ 15 февраля 2011

Сокеты нельзя использовать напрямую для перенаправления.При запуске внешнего процесса используйте CreatePipe() для создания анонимных каналов для перенаправленных дескрипторов STDIN / OUT / ERR процесса, затем используйте ReadFile() и WriteFile() (или эквиваленты) с send() и recv(), чтобы вручнуюпрокси данные между сокетом и процессом самостоятельно через каналы.Другими словами, когда данные поступают в сокет, считайте данные из сокета и запишите их в канал STDIN.Когда процесс выведет данные, прочитайте их из каналов STDOUT / ERR и запишите их в сокет.

2 голосов
/ 14 февраля 2011

Windows рассматривает почти все как HANDLE.Сокеты, которые не являются объектами ядра, являются исключением, и их нельзя использовать для перенаправления.Вам нужно будет использовать канал, и если вам нужно отправить данные в / из сокета, вам понадобится вспомогательный процесс для копирования данных между каналом и сокетом.

Посмотрите на netcat Исходный код для версии win32 (если вы можете его найти), он в значительной степени выполняет переадресацию с помощью сокетов <-> stdin и stdout.

1 голос
/ 09 мая 2016

Неправильно говорить, что дескрипторы сокетов нельзя использовать для перенаправленного ввода-вывода в дочерние процессы.Мы делаем это все время для CGI в коде нашего веб-сервера.Мы не используем именованные или безымянные каналы или промежуточные процессы.

a_mole является правильным.Вам просто нужны неперекрывающиеся сокеты.

Вы можете использовать WSASocket или, если вы уже создали сокет с сокетом (или не можете использовать WSASocket в вашей целевой ОС, например, до Windows 7 SP1), вы можете использоватьsetsockopt для установки SO_OPENTYPE текущего потока равным SO_SYNCHRONOUS_NONALERT

...