tcp 서버 클라이언트

글쓴이: 익명 사용자 / 작성시간: 화, 2021/09/14 - 5:49오후
#pragma once #pragma comment(lib, "ws2_32.lib") #define _WINSOCK_DEPRECATED_NO_WARNINGS #define PORT 8080 #define PACKET_SIZE 1024 #include <iostream> #include <winsock2.h> #include <windows.h> #include <thread> //thread #include <cstdlib> //strcpy_s #include <string> //getline #include <windows.h> bool flag = 1; void recvData(SOCKET& s) { char buffer[PACKET_SIZE]; while (TRUE) { ZeroMemory(buffer, sizeof buffer); recv(s, buffer, sizeof buffer, 0); if (strcmp(buffer, "[exit]") == 0) { // 서버를 정상종료, 클라이언트까지 모두 종료 send(s, buffer, sizeof buffer, 0); std::cout << "[서버종료]\n"; break; } if (strcmp(buffer, "[wait]") == 0) { // 서버만 종료, 클라이언트는 대기상태 std::cout << "[서버가 접속할때까지 대기]\n"; flag = 0; break; } std::cout << "Server : " << buffer << std::endl; } } int main() { char IP[PACKET_SIZE] = { 0 }, name[PACKET_SIZE] = { 0 }, message[PACKET_SIZE] = { 0 }; std::cout << "이름을 입력하세요 : "; std::cin >> name; std::cout << "아이피를 입력하세요 : "; // 로컬주소 127.0.0.1 std::cin >> IP; while (TRUE) { std::cout << "접속대기중.."; WSADATA wsa; WSAStartup(MAKEWORD(2, 2), &wsa); SOCKET server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); SOCKADDR_IN addr = { 0 }; addr.sin_addr.s_addr = inet_addr(IP); addr.sin_port = PORT; addr.sin_family = AF_INET; while (connect(server, (SOCKADDR*)&addr, sizeof addr)); flag = 1; std::cout << "서버와 연결되었습니다.\n"; send(server, name, sizeof name, 0); std::thread(recvData, std::ref(server)).detach(); while (flag) { ZeroMemory(message, sizeof message); std::cout << "메세지를 입력하세요: "; std::cin >> message; send(server, message, sizeof message, 0); } } }
#pragma once #pragma comment(lib, "ws2_32.lib") //링커 #define PORT 8080 #define PACKET_SIZE 1024 #include <iostream> // c++ stadard input output #include <winsock2.h> // winsock #include <windows.h> // ZeroMemory #include <vector> //vector #include <utility> //pair #include <thread> //thread #include <cstdlib> class CLIENT { public: SOCKET client; SOCKADDR_IN clientaddr = { 0 }; int clientsize = sizeof clientaddr; int number = -1; CLIENT() { } }; typedef std::pair<CLIENT, std::string> pii; std::vector<pii> Client; void recvData(SOCKET s, int num) { char buffer[PACKET_SIZE] = { 0 }; recv(s, buffer, sizeof buffer, 0); Client[num].second = buffer; while (TRUE) { ZeroMemory(buffer, sizeof buffer); recv(s, buffer, sizeof buffer, 0); if (strcmp(buffer, "[exit]") == 0) { send(s, buffer, sizeof buffer, 0); std::cout << "[서버종료]\n"; break; } if (strcmp(buffer, "[wait]") == 0) { send(s, buffer, sizeof buffer, 0); std::cout << "[서버대기]\n"; } std::cout << Client[num].second << " : " << buffer << std::endl; } } void ACCEPT(SOCKET &s) { int cnt = 0; while (TRUE) { Client.push_back(pii(CLIENT(), "")); Client[cnt].first.client = accept(s, (SOCKADDR*)&Client[cnt].first.clientaddr, &Client[cnt].first.clientsize); Client[cnt].first.number = cnt; std::thread(recvData, Client[cnt].first.client, cnt).detach(); cnt += 1; } } int main() { WSADATA wsa; WSAStartup(MAKEWORD(2,2), &wsa); SOCKET server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); SOCKADDR_IN addr = { 0 }; addr.sin_addr.s_addr = htonl(INADDR_ANY); addr.sin_port = PORT; addr.sin_family = AF_INET; bind(server, (SOCKADDR*)&addr, sizeof addr); listen(server, SOMAXCONN); std::thread(ACCEPT, std::ref(server)).detach(); char name[PACKET_SIZE], message[PACKET_SIZE]; while (TRUE) { ZeroMemory(message, sizeof message); std::cin >> message; for (int i = 0; i < Client.size(); i++) send(Client[i].first.client, message, sizeof message, 0); } }
tcp 서버와 클라이언트 간단한 채팅 코드인데
서버가 [wait]을 입력하면 클라이언트가 대기하면서 서버가 접속할때까지 기다리도록 하고싶습니다.
서버가 다시 켜지면 자동으로 클라이언트와 접속되도록 하고싶은데
쓰레드를 이용해서 메시지를 받는것과 보내는 것을 나눠두었고 while문에서 계속 메시지를 입력하도록 해야하는데 다시 연결이 되려면 메시지를 입력하는 while문에서 벗어나기 위해 flag를 사용하였습니다. 하지만 이미 cin으로 메시지를 받기위해 기다리고있기 때문에 채팅에서 한번은 메시지를 입력해야지 대기상태로 넘어갈 수 있습니다. 이를 해결하기 위한 방법이 없을까요?
Forums:
댓글 달기