cn-lab2-server-mt

// cn-lab2-server-mt.cpp: определяет точку входа для консольного приложения.
//

#include "stdafx.h"
#include <winsock2.h>
#include <iostream>
#include <sstream>
#include <thread>
#include <vector>
#include <regex>
#include <map>

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

using namespace std;

vector<thread> requests;
map<sockaddr_in, int> clients;

bool operator< (const sockaddr_in addr_left, const sockaddr_in addr_right) {
    string left = inet_ntoa(addr_left.sin_addr) + ':' + addr_left.sin_port;
    string right = inet_ntoa(addr_right.sin_addr) + ':' + addr_right.sin_port;
    return left < right;
}

int request_process(const int request_id, const SOCKET sock, const sockaddr_in client_addr, const int client_addr_size, string request) {
    regex triangle_pattern(" *\\d+ +\\d+ +\\d+ *");
    if (!regex_match(request, triangle_pattern)) {
        sendto(sock, "Invalid input", sizeof("Invalid input"), 0, (sockaddr *)&client_addr, client_addr_size);
        cout << "Answer " << request_id << " sent" << endl;
        return 0;
    }
    int a, b, c;
    stringstream ss;
    ss << request;
    ss >> a >> b >> c;
    string triangle_exist = "false";
    if ((a < b + c) && (b < a + c) && (c < a + b))
        triangle_exist = "true";
    sendto(sock, triangle_exist.c_str(), triangle_exist.length(), 0, (sockaddr *)&client_addr, client_addr_size);
    cout << "Answer " << request_id << " sent" << endl;
    return 0;
}

int port_listen(const SOCKET &sock) {
    for (int request_id = 1, client_id = 1;;) {
        sockaddr_in client_addr;
        int client_addr_size = sizeof(client_addr);
        char recv_buff[1024];
        int len = recvfrom(sock, &recv_buff[0], sizeof(recv_buff) - 1, 0, (sockaddr *)&client_addr, &client_addr_size);
        if (len < 0)
            return 1;
        recv_buff[len] = '\0';
        string request = recv_buff;
        if (request == "&new_client") {
            clients.insert(pair<sockaddr_in, int>(client_addr, client_id));
            cout << "New client #" << client_id << " was opened" << endl;
            client_id++;
            continue;
        }
        if (request == "&client_exit") {
            cout << "Client #" << clients.at(client_addr) << " was closed" << endl;
            clients.erase(client_addr);
            continue;
        }
        cout << "Request " << request_id << " received" << endl;
        requests.push_back(thread(request_process, request_id, sock, client_addr, client_addr_size, request));
        request_id++;
    }
    return 0;
}

int main()
{
    cout << "cn-lab2 var10 server start" << endl;
    WSAData wsadata;
    WSAStartup(0x0202, &wsadata);
    cout << "WinSock2 initialized" << endl;
    SOCKET sock;
    sock = socket(AF_INET, SOCK_DGRAM, 0);
    if (sock == INVALID_SOCKET) {
        cout << "Create socket error " << WSAGetLastError() << endl;
        WSACleanup();
        return 1;
    }
    cout << "Socket created successfully" << endl;
    sockaddr_in local_addr;
    local_addr.sin_family = AF_INET;
    local_addr.sin_addr.S_un.S_addr = 0;
    local_addr.sin_port = htons(5002);
    if (::bind(sock, (sockaddr *)&local_addr, sizeof(local_addr))) {
        cout << "Error bind " << WSAGetLastError() << endl;
        closesocket(sock);
        WSACleanup();
        return 2;
    }
    cout << "Bound successful" << endl;
    thread listen(port_listen, sock);
    string s;
    do {
        cin >> s;
    } while (s != "stop");
    for (auto client : clients)
        sendto(sock, "&server_stop", sizeof("&server_stop"), 0, (sockaddr *)&client.first, sizeof(client.first));
    closesocket(sock);
    if (listen.joinable())
        listen.join();
    for (int i = 0; i < requests.size(); i++) {
        if (requests[i].joinable())
            requests[i].join();
    }
    WSACleanup();
    cout << "cn-lab2 var10 server stopped" << endl;
    return 0;
}