Endil

cn-lab4-client

Dec 11th, 2015
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.30 KB | None | 0 0
  1. // cn-lab4-client.cpp: определяет точку входа для консольного приложения.
  2. //
  3.  
  4. #include "stdafx.h"
  5. #define _WINSOCK_DEPRECATED_NO_WARNINGS
  6. #include <condition_variable>
  7. #include <iostream>
  8. #include <regex>
  9. #include <string>
  10. #include <thread>
  11. #include <winsock2.h>
  12.  
  13. #pragma comment(lib, "ws2_32.lib")
  14.  
  15. using namespace std;
  16.  
  17. mutex m;
  18. condition_variable cvar;
  19. bool cvar_backup;
  20.  
  21. int user_input(const SOCKET main_sock) {
  22.     string s;
  23.     while (true) {
  24.         getline(cin, s);
  25.         if (s == "exit") {
  26.             send(main_sock, "&close", sizeof("&close"), 0);
  27.             cvar.notify_one();
  28.             cvar_backup = true;
  29.             return 1;
  30.         }
  31.         send(main_sock, s.c_str(), s.length(), 0);
  32.     }
  33.     return 0;
  34. }
  35.  
  36. int answer_display(string ans) {
  37.     for (; ans != "|||";) {
  38.         regex pattern(R"(\|{0,3}(.+?)\^{3}(.+?)\^{3}(.+?)\^{3}(.+?)\^{3}(.+?)\^{3}(.+?)\^{3}(.+)\r?)");
  39.         smatch match_results;
  40.         regex_match(ans, match_results, pattern);
  41.         cout << match_results[2] << " \"" << match_results[3] << "\" - " << match_results[5] << ", " << match_results[4] << endl;
  42.         ans = match_results.str(7);
  43.     }
  44.     return 0;
  45. }
  46.  
  47. int network_receiver(const SOCKET main_sock) {
  48.     thread *answer_display_thread = nullptr;
  49.     while (true) {
  50.         char recv_buff[1024];
  51.         int len = recv(main_sock, &recv_buff[0], sizeof(recv_buff), 0);
  52.         if (len == -1)
  53.             break;
  54.         recv_buff[len] = 0;
  55.         string ans = recv_buff;    
  56.         if (ans == "&stop") {
  57.             cout << "Server was stopped. Type \"exit\" to exit." << endl;
  58.             break;
  59.         }
  60.         if (answer_display_thread != nullptr)
  61.             answer_display_thread->join();
  62.         answer_display_thread = new thread(answer_display, ans);
  63.     }
  64.     return 0;
  65. }
  66.  
  67. int main() {
  68.     WSAData wsadata;
  69.     WSAStartup(0x0202, &wsadata);
  70.     SOCKET main_sock;
  71.     main_sock = socket(AF_INET, SOCK_STREAM, 0);
  72.     sockaddr_in server_addr;
  73.     server_addr.sin_family = AF_INET;
  74.     server_addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
  75.     server_addr.sin_port = htons(5004);
  76.     connect(main_sock, (sockaddr *)&server_addr, sizeof(server_addr));
  77.     thread listen(network_receiver, main_sock);
  78.     thread input(user_input, main_sock);
  79.     unique_lock<mutex> locked(m);
  80.     cvar.wait(locked, [] {return cvar_backup; });
  81.     closesocket(main_sock);
  82.     if (input.joinable())
  83.         input.join();
  84.     if (listen.joinable())
  85.         listen.join();
  86.     WSACleanup();
  87.     return 0;
  88. }
Advertisement
Add Comment
Please, Sign In to add comment