refactor and add feature

src/server.cpp

@@ -0,0 +1,153 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <iostream>
+#include <fstream>
+#include <unistd.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <typeinfo>
+#include <ctime>
+#include "colors.h"
+
+#define LOGFILE "udpchatserver.log"
+
+//структура одного клиента на сервере
+struct client_t {
+    sockaddr_in addr;
+    char username[1024]; //пример данных которые нужно менять при отправке пакета от клиента на наш сервер
+};
+
+const int num_of_clients = 64; // количество клиентов
+client_t clients[num_of_clients];
+
+bool HaveClient(sockaddr_in addr)
+{
+    for(int i = 0; i < num_of_clients; i++)
+        if(clients[i].addr.sin_addr.s_addr == addr.sin_addr.s_addr && clients[i].addr.sin_port == addr.sin_port)
+            return true;
+    
+    return false; 
+}
+
+int main(int argc, char** argv) {
+
+	if (argc != 3) {
+		printf("Usage: %s ip port\n", argv[0]);
+		exit(0);
+	}
+	char* ip = argv[1];
+	int port = atoi(argv[2]);
+
+	int sockfd = -1;
+	struct sockaddr_in server_addr;
+	socklen_t addr_size;
+	int n;
+
+	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
+	if (sockfd < 0) {
+		perror("[-] socket error");
+		exit(1);
+	}
+
+	memset(&server_addr, '\0', sizeof(server_addr));
+	server_addr.sin_family = AF_INET;
+	server_addr.sin_port = htons(port);
+	server_addr.sin_addr.s_addr = inet_addr(ip);
+
+	n = bind(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));
+
+	if (n < 0) {
+		perror("[-] bind error");
+		exit(1);
+	}
+
+	sockaddr_in client_addr;
+	unsigned int clientlength = sizeof(client_addr);
+
+	for (int i = 0; i < num_of_clients; i++) {
+		std::cout << clients[i].addr.sin_port << clients[i].addr.sin_addr.s_addr << ";";
+	}
+
+
+	char buffer[1024];
+	fd_set  rset;
+
+	FD_ZERO(&rset);
+	int maxfdp1 = sockfd + 1;
+	int nready;
+
+	while (1) {
+		FD_SET(sockfd, &rset);
+		if ((select(maxfdp1, &rset, NULL, NULL, NULL)) < 0) {
+			if (errno == EINTR)
+				continue;
+			else
+				perror("select error");
+		}
+
+		if (FD_ISSET(sockfd, &rset)) {
+			bzero(buffer, 1024);
+			clientlength = sizeof(client_addr);
+			int bytesin = recvfrom(sockfd, buffer, 1024, 0, (sockaddr *) &client_addr, &clientlength);
+			if (bytesin < 0) {
+				std::cout << "error receving from client\n";
+				continue;
+			}
+			if (!HaveClient(client_addr)) {
+				for(int i = 0; i < num_of_clients; i++) {
+					if (clients[i].addr.sin_port == 0 && clients[i].addr.sin_addr.s_addr == 0) {
+						clients[i].addr = client_addr;
+						strcpy(clients[i].username, buffer);
+						break;
+					}
+				}
+			}else {
+				for (int i = 0; i < num_of_clients; i++) {
+					if (clients[i].addr.sin_port != 0 && clients[i].addr.sin_addr.s_addr != 0) {
+						if (!(clients[i].addr.sin_port == client_addr.sin_port && clients[i].addr.sin_addr.s_addr == client_addr.sin_addr.s_addr)) {
+							if (strcmp(buffer, "")) {
+								std::ofstream logfile(LOGFILE, std::ios::app);
+								char res[1024];
+								char logres[1024];
+								char time[128];
+								memset(&logres, '\0', 1024);
+								memset(&res, '\0', 1024);
+								memset(&time, '\0', 128);
+								std::time_t t = std::time(nullptr);
+	    						std::tm* now = std::localtime(&t);
+	    						strftime(time, sizeof(time), "%X", now);
+
+	    						for (int i = 0; i < num_of_clients; i++) {
+									if (clients[i].addr.sin_addr.s_addr == client_addr.sin_addr.s_addr && clients[i].addr.sin_port == client_addr.sin_port) {
+										strcpy(res, GREEN);
+										strcat(res, clients[i].username);
+										strcat(logres, clients[i].username);
+									}
+								}
+								strcat(res, BLUE " [ ");
+								strcat(res, time);
+								strcat(res, " ] " RESET ": " YELLOW);
+								strcat(res, buffer);
+								strcat(res, RESET);
+								sendto(sockfd, res, 1024, 0, (sockaddr *) &clients[i].addr, sizeof(clients[i].addr));
+								printf("[ i ] %s\n", res);
+								strcat(logres, " [ ");
+								strcat(logres, time);
+								strcat(logres, " ] : ");
+								strcat(logres, buffer);
+								logfile << "[ i ] " << logres << "\n";
+								logfile.close();
+								// fprintf(logfile, "[ i ] %s\n", res);
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+
+    return 0;
+}