1 files changed, 119 insertions, 0 deletions
diff --git a/csgo-loader/csgo-module/Networking/TCPClient.cpp b/csgo-loader/csgo-module/Networking/TCPClient.cpp
new file mode 100644
index 0000000..e22bfdf
--- /dev/null
+++ b/csgo-loader/csgo-module/Networking/TCPClient.cpp
@@ -0,0 +1,119 @@
+#include <Networking/TCPClient.hpp>
+
+namespace Networking
+{
+	// We will only receive up to 256 bytes per cycle.
+	constexpr int BufferSize = 256;
+
+	void TCPClient::SendRawBytes(ByteArray &Bytes)
+	{
+		// Send data.
+		send(m_Socket, (char *)Bytes.data(), (int)Bytes.size(), 0);
+	}
+
+	ByteArray TCPClient::ReceiveRawBytes()
+	{
+		ByteArray	ReceivedBytes;
+		uint8_t		RecvBuffer[BufferSize];
+
+		// Attempt to receive a packet.
+		while(true)
+		{
+			int32_t Received = recv(m_Socket, (char*)RecvBuffer, BufferSize, 0);
+
+			// No more bytes left to receive.
+			if(Received < 0)
+				break;
+
+			// Emplace all received bytes.
+			for(int n = 0; n < Received; ++n)
+			{
+				ReceivedBytes.push_back(RecvBuffer[n]);
+			}
+
+			// No more bytes left to receive.
+			if(Received < BufferSize)
+				break;
+		}
+
+		return ReceivedBytes;
+	}
+
+	void TCPClient::SendBytes(ByteArray &Bytes)
+	{
+		// Encrypt outgoing data.
+		ByteArray EncryptionKey;
+		EncryptionKey.insert(
+			EncryptionKey.begin(),
+			m_EncryptionKey,
+			m_EncryptionKey + sizeof m_EncryptionKey
+		);
+
+		Wrapper::Encryption Encryption; Encryption.Start(EncryptionKey);
+
+		ByteArray Encrypted = Encryption.Encrypt(Bytes);
+
+		SendRawBytes(Encrypted);
+	}
+
+	ByteArray TCPClient::ReceiveBytes()
+	{
+		// Decrypt incoming data.
+		ByteArray ReceivedBytes = ReceiveRawBytes();
+
+		ByteArray EncryptionKey;
+		EncryptionKey.insert(
+			EncryptionKey.begin(),
+			m_EncryptionKey,
+			m_EncryptionKey + sizeof m_EncryptionKey
+		);
+
+		Wrapper::Encryption Encryption; Encryption.Start(EncryptionKey);
+
+		ByteArray Decrypted = Encryption.Decrypt(ReceivedBytes);
+
+		return Decrypted;
+	}
+
+	bool TCPClient::Start(uint32_t ServerAddress, uint16_t ServerPort)
+	{
+		const int32_t version = 0x101;
+
+		// Initialise WinSocks.
+		if(WSAStartup(version, &m_WinSocks))
+			return false;
+
+		// Create an IPv4 socket.
+		m_Socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+		if(m_Socket == INVALID_SOCKET)
+			return false;
+
+		// Set up client context.
+		m_Context.sin_addr.s_addr = ServerAddress;
+		m_Context.sin_family      = AF_INET;
+		m_Context.sin_port        = htons(ServerPort);
+
+		// Attempt connection.
+		if(connect(m_Socket, (sockaddr *)&m_Context, sizeof m_Context))
+			return false;
+
+		// Initialise encryption wrapper.
+		ByteArray EncryptionKey = ReceiveRawBytes();
+		
+		if(EncryptionKey.empty())
+			return false;
+
+		std::memcpy(m_EncryptionKey, EncryptionKey.data(), EncryptionKey.size());
+
+		return true;
+	}
+
+	void TCPClient::Kill()
+	{
+		if(m_Socket)
+			closesocket(m_Socket);
+
+		WSACleanup();
+	}
+}
+\ No newline at end of file