1 files changed, 87 insertions, 0 deletions
diff --git a/csgo-loader/csgo-server/Networking/TCPServer.hpp b/csgo-loader/csgo-server/Networking/TCPServer.hpp
new file mode 100644
index 0000000..adb6e7c
--- /dev/null
+++ b/csgo-loader/csgo-server/Networking/TCPServer.hpp
@@ -0,0 +1,87 @@
+#pragma once
+
+// For encryption wrappers.
+#include <Security/Encryption.hpp>
+
+// WinSocks
+#include <winsock2.h>
+#pragma comment(lib, "ws2_32.lib")
+
+// std::function
+#include <functional>
+
+// std::min
+#include <algorithm>
+
+// std::thread
+#include <thread>
+
+namespace Networking {
+	// Base connection class, used to handle multiple connections in a thread-based model.
+	class TCPConnection {
+		SOCKET				m_Socket;
+		Wrapper::Encryption m_Encryption;
+		const char			*m_IpAddress;
+	public:
+		// Initialiser for TCPConnection class.
+		TCPConnection(SOCKET Connection, const char *IpAddress, Wrapper::Encryption &RSA) :
+			m_Encryption(RSA), m_Socket(Connection), m_IpAddress(IpAddress) {
+			printf("[=>] %s connected!\n", IpAddress);
+		}
+
+		// Release the connection once it goes out of scope.
+		void Close();
+
+		// Wrappers for sending/receiving data.
+		void SendRawBytes(ByteArray &Bytes);
+		ByteArray ReceiveRawBytes();
+
+		void SendBytes(ByteArray &Bytes);
+		ByteArray ReceiveBytes();
+
+		// Overload for getting the socket, in case we need to expose it.
+		SOCKET operator()() {
+			return m_Socket;
+		}
+
+		// Expose the encryption key for the connection.
+		ByteArray GetEncryptionKey() {
+			return m_Encryption.GetKey();
+		}
+	};
+
+	// Basic TCP server. Supports custom connection handling (pass a lambda to the handler list).
+	using ConnectionHandler = std::function<void(TCPConnection &)>;
+
+	class TCPServer {
+		WSADATA		m_WinSocks;
+		SOCKET		m_Socket;
+		sockaddr_in m_Context;
+
+		// Connection handlers, will be called sequentially upon connection.
+		ConnectionHandler m_ConnectionHandler;
+
+	public:
+		// Default constructor, nothing needed for now.
+		TCPServer() = default;
+		
+		// Handle destruction of server once it goes out of scope.
+		~TCPServer() {
+			// If we have a socket, close it.
+			if(m_Socket)
+				closesocket(m_Socket);
+
+			// Close WSA context.
+			WSACleanup();
+		}
+
+		// Handle the creation and handling of TCPServer connections.
+		bool Start(uint16_t ServerPort);
+		void AcceptConnection();
+
+		// Overload for adding connection handlers, C# style support for events.
+		void operator+=(std::function<void(TCPConnection &)> Function) {
+			m_ConnectionHandler = Function;
+		}
+	};
+}
+\ No newline at end of file