FirstCS/win_clnt.cpp at master · vkuksa/FirstCS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#include "win_clnt.h"


Client::Client(): port(7300)
{
	memset(buff, 0, sizeof(buff));

	WSADATA wsadata;

	if (WSAStartup(0x0202, &wsadata))
	{
		//if error with startup
		std::cout << "Error with WSAStartup " << WSAGetLastError();
	}
}

int Client::connection_to_port(sockaddr_in & dest_addr)
{
	if ((client_socket = socket(AF_INET, SOCK_STREAM, 0)) < 0)	//AF_INET - internet socket SOCK_STREAM - socket for TCP
	{
		//if error with socket
		std::cout << "Error with socket " << WSAGetLastError();
		WSACleanup();	//Deinit of Winsock
	}

	HOSTENT *hst;

	if (inet_addr(serv_addr) != INADDR_NONE)		//gettin host addr
		dest_addr.sin_addr.s_addr = inet_addr(serv_addr);
	else
		if (hst = gethostbyname(serv_addr))
			((unsigned long*)&dest_addr.sin_addr)[0] =
			((unsigned long **)hst->h_addr_list)[0][0];
		else
		{
			//if err with Server addr
			std::cout << "Invalid address " << serv_addr << std::endl;
			closesocket(client_socket);
			WSACleanup();
		}

	if (connect(client_socket, (sockaddr*)&dest_addr, sizeof(dest_addr)))
	{
		//if err with connect
		std::cout << "COnnecting error " << WSAGetLastError() << std::endl;
	}
	return 0;
}

bool Client::change_port(sockaddr_in & dest_addr)
{
	if (closesocket(client_socket) == 0)
	{
		dest_addr.sin_port = htons(port);
		Sleep(100);				//We should w8 before server will be able to accept clients
		connection_to_port(dest_addr);
		return true;
	}
	return false;
}

int Client::get_port()
{
	return port;
}

void Client::starting_response()
{
	if ((recv(client_socket, &buff[0], sizeof(buff), 0)) != SOCKET_ERROR)
	{
		std::cout << buff;
		const char* response = "Successfully connected";
		send(client_socket, response, sizeof(buff), 0);
	}
}

void Client::dialog_with_server(sockaddr_in & dest_addr)
{
	while ((recv(client_socket, &buff[0], sizeof(buff), 0)) != SOCKET_ERROR)
	{
		if (strncmp(buff, "#1port",6) == 0)
		{
			port = atoi(buff + 6);
			if (change_port(dest_addr) != false)
				send(client_socket, "success", strlen("success"), 0);
			else
				send(client_socket, "failure", strlen("success"), 0);
		}
		else
		{
			std::cout << "\nServer: " << buff;

			std::cout << "\nClient: ";
			fgets(buff, 1024, stdin);
			if (strncmp(buff, "quit", 4) == 0)
			{
				std::cout << "Exit";
				closesocket(client_socket);
				WSACleanup();
				exit(0);
			}
			if (strncmp(buff, "NewPort-", 8) == 0)
			{
				port = atoi(buff + 8);
				char portmsg[11] = "#1port";
				memcpy(portmsg + 6, buff + 8, 4);
				send(client_socket, &portmsg[0], sizeof(portmsg), 0);
				change_port(dest_addr);
			}
			else
				send(client_socket, &buff[0], sizeof(buff), 0);
		}
	}
}

Client::~Client()
{
	closesocket(client_socket);
	WSACleanup();
}