Networking Initial

Storage/applications/CMakeLists.txt

@@ -0,0 +1,11 @@
+
+cmake_minimum_required(VERSION 3.2)
+
+project(Applications)
+
+### ---------------------- Applications  ---------------------  ###
+add_executable(Server ./Server.cpp)
+add_executable(Client ./Client.cpp)
+
+include_directories(Client ./../../Externals/asio/asio/include)
+include_directories(Server ./../../Externals/asio/asio/include)

Storage/applications/Client.cpp

@@ -1,45 +1,81 @@
-#include <iostream>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
-#include <unistd.h>
-#include <cstring>
+#include "asio.hpp"
 
-constexpr int PORT = 8080;
-constexpr int BUFFER_SIZE = 1024;
+#include <iostream>
+#include <string>
+#include <mutex>
+
+// #include <unistd.h>
+// #include <cstring>
+
+constexpr int PORT = 3333;
 const char* SERVER_IP = "127.0.0.1";
 
+std::mutex mutex;
+
+void* readServerBroadcast(void* clientSocketPtr) {
+	auto socket = ((asio::ip::tcp::socket*)clientSocketPtr);
+
+	while (true) {
+
+		short messageSize;
+		asio::read(*socket, asio::buffer(&messageSize, 2));
+
+		mutex.lock();
+
+		auto message = new char[messageSize + 1];
+		message[messageSize] = 0;
+
+		asio::read(*socket, asio::buffer(message, messageSize));
+
+		std::cerr << "Broadcast : " << message << std::endl;
+
+		delete[] message;
+
+		mutex.unlock();
+	}
+
+	return nullptr;
+}
+
 int main() {
-	// Create socket
-	int clientSocket = socket(AF_INET, SOCK_STREAM, 0);
-	if (clientSocket == -1) {
-		std::cerr << "Failed to create socket" << std::endl;
+	asio::io_context ioContext;
+
+	// Create a TCP socket
+	asio::ip::tcp::socket socket(ioContext);
+
+	// Connect to a server
+	asio::ip::tcp::endpoint endpoint(asio::ip::address::from_string(SERVER_IP), PORT);
+	socket.connect(endpoint);
+
+	// Create a new thread to handle the client
+	pthread_t threadId;
+	if (pthread_create(&threadId, nullptr, readServerBroadcast, (void*) &socket) != 0) {
+		std::cerr << "Failed to create thread for client" << std::endl;
 		return 1;
 	}
 
-	// Connect to server
-	sockaddr_in serverAddress{};
-	serverAddress.sin_family = AF_INET;
-	serverAddress.sin_port = htons(PORT);
-	if (inet_pton(AF_INET, SERVER_IP, &serverAddress.sin_addr) <= 0) {
-		std::cerr << "Invalid address or address not supported" << std::endl;
-		return 1;
-	}
+	// Detach the thread so it can run independently
+	pthread_detach(threadId);
 
-	if (connect(clientSocket, reinterpret_cast<sockaddr*>(&serverAddress), sizeof(serverAddress)) == -1) {
-		std::cerr << "Failed to connect to server" << std::endl;
-		return 1;
-	}
+	while (true) {
+		std::string message;
+		std::cout << " >> ";
+		std::cin >> message;
 
-	// Send message to server
-	const char* message = "Hello, server!";
-	if (write(clientSocket, message, strlen(message)) == -1) {
-		std::cerr << "Failed to write to socket" << std::endl;
-		return 1;
+		mutex.lock();
+
+		// Send a message to the server
+		auto messageSize = (short) message.size();
+		asio::write(socket, asio::buffer(&messageSize, 2));
+
+		// Send a message to the server
+		asio::write(socket, asio::buffer(message + "\n"));
+
+		mutex.unlock();
 	}
 
 	// Close socket
-	close(clientSocket);
+	// close(clientSocket);
 
 	return 0;
 }

Storage/applications/Server.cpp

@@ -1,54 +1,56 @@
+#include <asio.hpp>
+
+#include <cstring>
 #include <iostream>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <unistd.h>
+#include <list>
+#include <mutex>
 #include <pthread.h>
 
-
 class Server {
+
+	struct SharedData {
+		std::list<asio::ip::tcp::socket*> clients;
+		std::mutex mutex;
+	};
+
+	SharedData mSharedData;
+
+	//int serverSocket;
+	int port;
+
 public:
-	Server(int port) : serverSocket(-1), port(port) {}
+	Server(int port) : port(port) {}
+	~Server() { assert(0); }
 
 	bool start() {
 		// Create socket
-		serverSocket = socket(AF_INET, SOCK_STREAM, 0);
-		if (serverSocket == -1) {
-			std::cerr << "Failed to create socket" << std::endl;
-			return false;
-		}
+		asio::io_context io_context;
+		asio::ip::tcp::acceptor serverSocket(io_context);
 
 		// Bind socket to port
-		sockaddr_in serverAddress{};
-		serverAddress.sin_family = AF_INET;
-		serverAddress.sin_port = htons(port);
-		serverAddress.sin_addr.s_addr = INADDR_ANY;
-		if (bind(serverSocket, reinterpret_cast<sockaddr*>(&serverAddress), sizeof(serverAddress)) == -1) {
-			std::cerr << "Failed to bind socket" << std::endl;
-			return false;
-		}
+		serverSocket.open(asio::ip::tcp::v4());
+		// serverSocket.set_option(asio::ip::tcp::acceptor::reuse_address(true));
+		serverSocket.bind(asio::ip::tcp::endpoint(asio::ip::tcp::v4(), port));
 
 		// Listen for connections
-		if (listen(serverSocket, 5) == -1) {
-			std::cerr << "Failed to listen on socket" << std::endl;
-			return false;
-		}
+		serverSocket.listen();
 
 		std::cout << "Server listening on port " << port << std::endl;
 
 		// Start accepting clients
 		while (true) {
 			// Accept client connection
-			sockaddr_in clientAddress{};
-			socklen_t clientAddressSize = sizeof(clientAddress);
-			int clientSocket = accept(serverSocket, reinterpret_cast<sockaddr*>(&clientAddress), &clientAddressSize);
-			if (clientSocket == -1) {
-				std::cerr << "Failed to accept client connection" << std::endl;
-				return false;
-			}
+			auto clientSocket = new asio::ip::tcp::socket(io_context);
+			serverSocket.accept(*clientSocket);
+
+			// add client
+			mSharedData.mutex.lock();
+			mSharedData.clients.push_back(clientSocket);
+			mSharedData.mutex.unlock();
 
 			// Create a new thread to handle the client
 			pthread_t threadId;
-			if (pthread_create(&threadId, nullptr, handleClient, &clientSocket) != 0) {
+			if (pthread_create(&threadId, nullptr, handleClient, this) != 0) {
 				std::cerr << "Failed to create thread for client" << std::endl;
 				return false;
 			}
@@ -58,40 +60,81 @@ public:
 		}
 
 		// Close server socket
-		close(serverSocket);
+		serverSocket.close();
 
 		return true;
 	}
 
-private:
-	int serverSocket;
-	int port;
 
-	static void* handleClient(void* clientSocketPtr) {
-		int clientSocket = *(reinterpret_cast<int*>(clientSocketPtr));
-		constexpr int BUFFER_SIZE = 1024;
+	static void* handleClient(void* in) {
+		auto self = (Server*) in;
 
-		// Receive and print client message
-		char buffer[BUFFER_SIZE];
-		ssize_t bytesRead = read(clientSocket, buffer, BUFFER_SIZE - 1);
-		if (bytesRead == -1) {
-			std::cerr << "Failed to read from socket" << std::endl;
-			return nullptr;
+		// read shared data - current client id
+		self->mSharedData.mutex.lock();
+		auto clientSocket = self->mSharedData.clients.back();
+		self->mSharedData.mutex.unlock();
+
+	MESSAGE:
+		// wait for a message request1
+		short messageSize;
+		{
+			auto bytesRead = asio::read(*clientSocket, asio::buffer(&messageSize, 2));
+			if (bytesRead == -1) {
+				std::cerr << "Failed to read from socket" << std::endl;
+				(*clientSocket).close();
+				return nullptr;
+			}
 		}
 
-		buffer[bytesRead] = '\0';
-		std::cout << "Received message from client: " << buffer << std::endl;
+		self->mSharedData.mutex.lock();
+
+		// Receive client message
+		auto message = new char[messageSize + 1];
+		message[messageSize] = '\0';
+		{
+			auto bytesRead = asio::read(*clientSocket, asio::buffer(message, messageSize));
+			if (bytesRead == -1) {
+				std::cerr << "Failed to read from socket" << std::endl;
+				memcpy(message, "Cant wanna say something but i cant read", 100);
+			}
+		}
+
+		// Broadcast to all clients
+		for (auto client: self->mSharedData.clients) {
+			auto bytesWritten = asio::write(*client, asio::buffer(&messageSize, 2));
+			if (bytesWritten == -1) {
+				std::cerr << "Failed to write to socket" << std::endl;
+			}
+
+			bytesWritten = write(*client, asio::buffer(message, messageSize));
+			if (bytesWritten == -1) {
+				std::cerr << "Failed to write to socket" << std::endl;
+			}
+		}
+
+		auto exit = memcmp(message, "exit", strlen("exit")) == 0;
 
 		// Close socket
-		close(clientSocket);
+		delete[] message;
 
-		return nullptr;
+
+		if (exit) {
+			(*clientSocket).close();
+			self->mSharedData.clients.remove(clientSocket);
+		}
+
+		self->mSharedData.mutex.unlock();
+
+		if (exit) {
+			return nullptr;
+		}
+		goto MESSAGE;
 	}
 };
 
 int main() {
-	Server server(8080);
+	Server server(3333);
 	server.start();
 
 	return 0;
-}
+}