Простой неблокирующий многопоточный tcp сервер - PullRequest
Новая
       37

Простой неблокирующий многопоточный tcp сервер

0 голосов
/ 17 июня 2019

Я изучаю C ++, и в эти выходные я начал играть с сокетами и потоками. Bellow - это простой многопоточный сервер, который я делаю на основе нескольких уроков.

Проблема, с которой я сталкиваюсь, заключается в том, что при соединении с двумя клиентами telnet на сервере появляются только нажатия клавиш при первом подключении. Любое нажатие клавиши, отправленное со второго соединения telnet, появляется внезапно после закрытия первого соединения telnet. Может ли кто-нибудь объяснить мне, что я здесь не так сделал? 

#include <iostream>
#include <string>
#include <thread>
#include <winsock2.h>
#include <ws2tcpip.h>

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

void clientSocketHandler(SOCKET clientSocket, std::string client_ip) {

    char buf[4096];

    std::thread::id thread_id = std::this_thread::get_id();
    std::cout << thread_id << " - " << client_ip << ": connected" << std::endl;

    while (true)
    {

        ZeroMemory(buf, 4096);

        int bytesReceived = recv(clientSocket, buf, 4096, 0);

        if (bytesReceived == 0)
        {

            std::cout << thread_id << " - " << client_ip << ": disconnected" << std::endl;

            break;

        }

        if (bytesReceived > 0) 
        {

            std::cout << thread_id << " - " << client_ip << ": " << std::string(buf, 0, bytesReceived) << std::endl;

            //send(clientSocket, buf, bytesReceived + 1, 0);

        }

    }

    std::cout << thread_id << " - " << client_ip << ": closing client socket & exiting thread..." << std::endl;

    closesocket(clientSocket);

}

void waitForConnections(SOCKET serverSocket) {

    sockaddr_in hint;

    hint.sin_family = AF_INET;
    hint.sin_port = htons(1337);
    hint.sin_addr.S_un.S_addr = INADDR_ANY;

    bind(serverSocket, (sockaddr*)&hint, sizeof(hint));
    listen(serverSocket, SOMAXCONN);

    while (true) {

        sockaddr_in client;

        int clientSize = sizeof(client);

        SOCKET clientSocket = accept(serverSocket, (sockaddr*)&client, &clientSize);

        if (clientSocket != INVALID_SOCKET) 
        {

            char host[NI_MAXHOST];      // Client's remote name

            ZeroMemory(host, NI_MAXHOST); // same as memset(host, 0, NI_MAXHOST);

            std::string client_ip = inet_ntop(AF_INET, &client.sin_addr, host, NI_MAXHOST);
            std::thread t(clientSocketHandler, clientSocket, client_ip);

            t.join();

        }

        Sleep(100);

    }

}

int main()
{
    // Initialze winsock
    WSADATA wsData;
    WORD ver = MAKEWORD(2, 2);

    int wsOk = WSAStartup(ver, &wsData);

    if (wsOk != 0)
    {
        std::cerr << "Can't Initialize winsock! Quitting..." << std::endl;

        return 1;
    }

    // Create a socket
    SOCKET serverSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if (serverSocket == INVALID_SOCKET)
    {
        WSACleanup();
        std::cerr << "Can't create a socket! Quitting..." << std::endl;

        return 1;
    }

    // If serverSocketMode = 0, blocking is enabled; 
    // If serverSocketMode != 0, non-blocking mode is enabled.
    u_long serverSocketMode = 1;

    if (ioctlsocket(serverSocket, FIONBIO, &serverSocketMode) != NO_ERROR) 
    {
        WSACleanup();
        std::cerr << "Can't set socket to non-blocking mode! Quitting..." << std::endl;

        return 1;
    }

    // Disables the Nagle algorithm for send coalescing.
    // This socket option is included for backward 
    // compatibility with Windows Sockets 1.1
    BOOL flag = TRUE;

    if (setsockopt(serverSocket, IPPROTO_TCP, TCP_NODELAY, (const char *)&flag, sizeof(flag)) != NO_ERROR)
    {
        WSACleanup();
        std::cerr << "Can't set socket NO_DELAY option! Quitting..." << std::endl;

        return 1;
    }

    // Start listening for connections
    waitForConnections(serverSocket);

    // Cleanup winsock
    WSACleanup();

    system("pause");

    return 0;

}

1 Ответ

0 голосов
/ 17 июня 2019

Это должно работать.Я удалил бессмысленные вещи, такие как установка сокета в неблокирующее состояние и отключение алгоритма Nagle.Последнее должно быть сделано только для вещей, которые нуждаются в малой миллисекундной интерактивности.

Но существенное изменение, которое должно решить вашу проблему, меняется с join на detach.Использование join заставляет вашу программу ждать завершения потока, прежде чем продолжить.Использование detach говорит: «Этот поток будет работать в фоновом режиме, и мне наплевать на его дальнейшую судьбу».

Если вы не используете один из двух, иобъект ::std::thread уничтожен, система выдает исключение, потому что вы уничтожаете единственное средство получения информации о том, завершился ли поток с какой-либо ошибкой, сообщив, что либо вас это не волнуетинформация или явный запрос.

У меня нет Windows, поэтому я не могу проверить ее: 

#include <iostream>
#include <string>
#include <thread>
#include <winsock2.h>
#include <ws2tcpip.h>

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

void clientSocketHandler(SOCKET clientSocket, std::string client_ip)
{
   char buf[4096];

   std::thread::id thread_id = std::this_thread::get_id();
   std::cout << thread_id << " - " << client_ip << ": connected" << std::endl;

   while (true)
   {

      ZeroMemory(buf, 4096);

      int bytesReceived = recv(clientSocket, buf, 4096, 0);

      if (bytesReceived == 0)
      {

         std::cout << thread_id << " - " << client_ip << ": disconnected" << std::endl;

         break;

      }

      if (bytesReceived > 0)
      {

         std::cout << thread_id << " - " << client_ip << ": " << std::string(buf, 0, bytesReceived) << std::endl;

         //send(clientSocket, buf, bytesReceived + 1, 0);

      }

   }

   std::cout << thread_id << " - " << client_ip << ": closing client socket & exiting thread..." << std::endl;

   closesocket(clientSocket);

}

void waitForConnections(SOCKET serverSocket)
{

   sockaddr_in hint;

   hint.sin_family = AF_INET;
   hint.sin_port = htons(1337);
   hint.sin_addr.S_un.S_addr = INADDR_ANY;

   bind(serverSocket, (sockaddr*)&hint, sizeof(hint));
   listen(serverSocket, SOMAXCONN);

   while (true) {

      sockaddr_in client;

      int clientSize = sizeof(client);

      SOCKET clientSocket = accept(serverSocket, (sockaddr*)&client, &clientSize);

      if (clientSocket != INVALID_SOCKET)
      {

         char host[NI_MAXHOST];      // Client's remote name

         ZeroMemory(host, NI_MAXHOST); // same as memset(host, 0, NI_MAXHOST);

         std::string client_ip = inet_ntop(AF_INET, &client.sin_addr, host, NI_MAXHOST);
         std::thread t(clientSocketHandler, clientSocket, client_ip);

         t.detach();

      }

      Sleep(100);

   }

}

int main()
{
    // Initialze winsock
    WSADATA wsData;
    WORD ver = MAKEWORD(2, 2);

    int wsOk = WSAStartup(ver, &wsData);

    if (wsOk != 0)
    {
        std::cerr << "Can't Initialize winsock! Quitting..." << std::endl;

        return 1;
    }

    // Create a socket
    SOCKET serverSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if (serverSocket == INVALID_SOCKET)
    {
        WSACleanup();
        std::cerr << "Can't create a socket! Quitting..." << std::endl;

        return 1;
    }

    // Start listening for connections
    waitForConnections(serverSocket);

    // Cleanup winsock
    WSACleanup();

    system("pause");

    return 0;

}
...