cn-lab3-client

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

#include "stdafx.h"
#include <iostream>
#include <winsock2.h>
#include <string>
#include <thread>
#include <condition_variable>

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

using namespace std;

mutex m;
condition_variable cvar;
bool cvar_backup;

int input(SOCKET main_sock) {
    string s;
    while (true) {
        getline(cin, s);
        if (s == "exit") {
            send(main_sock, "&close", sizeof("&close"), 0);
            cvar.notify_one();
            cvar_backup = true;
            return 1;
        }
        send(main_sock, s.c_str(), s.length(), 0);
    }
    return 0;
}

int receive(SOCKET main_sock) {
    while (true) {
        char recv_buff[1024];
        int len = recv(main_sock, &recv_buff[0], sizeof(recv_buff), 0);
        if (len == -1)
            return 1;
        recv_buff[len] = '\0';
        string ans = recv_buff;
        if (ans == "&stop") {
            cout << "Server was stopped. Type \"exit\" to exit." << endl;
            return 2;
        }
        cout << ans << endl;
    }
    return 0;
}

int _tmain(int argc, _TCHAR* argv[])
{
    WSAData wsadata;
    WSAStartup(0x0202, &wsadata);
    SOCKET main_sock;
    main_sock = socket(AF_INET, SOCK_STREAM, 0);
    sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(5001);
    connect(main_sock, (sockaddr *)&server_addr, sizeof(server_addr));
    thread listen(receive, main_sock);
    thread input(input, main_sock);
    unique_lock<mutex> locked(m);
    cvar.wait(locked, [] {return cvar_backup; });
    closesocket(main_sock);
    if (input.joinable())
        input.join();
    if (listen.joinable())
        listen.join();
    WSACleanup();
    return 0;
}