-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.cpp
83 lines (72 loc) · 2.73 KB
/
server.cpp
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
#include <sys/socket.h>
#include <netinet/in.h>
#include "server.h"
#include <string>
#include <cstring>
#include <unistd.h>
#include <arpa/inet.h>
#include <thread>
#include "parser.h"
#include "handlers.h"
Server::Server() {
sock_fd_ = socket(AF_INET, SOCK_STREAM, 0);
if (sock_fd_ == -1) {
std::cerr << "Error creating socket\n";
return;
}
memset(&serverAddr_, 0, sizeof(serverAddr_));
serverAddr_.sin_family = AF_INET;
serverAddr_.sin_addr.s_addr = INADDR_ANY;
serverAddr_.sin_port = htons(4222);
if (bind(sock_fd_, (struct sockaddr*)&serverAddr_, sizeof(serverAddr_)) == -1) {
std::cerr << "Error binding socket\n";
return;
}
listen(sock_fd_, 10);
}
void Server::Start() {
while (true) {
int client_socket_fd = accept(sock_fd_, NULL, NULL);
if (client_socket_fd == -1) {
std::cerr << "Error accepting connection\n";
return;
}
std::thread client_thread(&Server::HandleClient, this, client_socket_fd);
client_thread.detach();
}
}
void Server::HandleClient(int client_socket_fd) {
struct sockaddr_in clientAddr;
socklen_t clientAddrLen = sizeof(clientAddr);
// Print client IP
char clientIP[INET_ADDRSTRLEN];
if (getpeername(client_socket_fd, (struct sockaddr*)&clientAddr, &clientAddrLen) == 0) {
inet_ntop(AF_INET, &(clientAddr.sin_addr), clientIP, INET_ADDRSTRLEN);
std::cout << "Client connected from IP: " << clientIP << std::endl;
} else {
std::cerr << "Error getting client IP\n";
}
std::string infoMessage = "INFO {\"host\":\"0.0.0.0\",\"port\":4222,\"client_ip\":\"" + std::string(clientIP) + "\"}\r\n";
send(client_socket_fd, infoMessage.c_str(), infoMessage.size(), 0);
// Keep reading from client until they disconnect
while (true) {
Command* command = new Command();
char * buffer = new char[1024];
int bytesReceived = recv(client_socket_fd, buffer, 1024, 0);
if (bytesReceived <= 0) {
// Client has closed the connection
std::cout << "Client disconnected\n";
// TODO: Remove the client from the subscribers list
break;
}
command->input = std::string(buffer);
auto command_handler = CommandParser::parse(command);
if (command_handler) {
command_handler->Handle(client_socket_fd, subscribers_);
// std::cout << "Thread number: " << std::this_thread::get_id() << std::endl;
// std::cout << "Subscribers size: " << subscribers_.size() << std::endl;
send(client_socket_fd, command_handler->GetMessage().c_str(), command_handler->GetMessage().size(), 0);
}
}
close(client_socket_fd);
}