From 77b52da44b263df4884be2f35f885d8edccbb6fa Mon Sep 17 00:00:00 2001
From: boris <wzn@moneybot.cc>
Date: Wed, 19 Dec 2018 00:13:24 +1300
Subject: added new loader project :)

merry christmas
---
 csgo-loader/csgo-client/Networking/TCPClient.cpp | 102 +++++++++++++++++++++++
 csgo-loader/csgo-client/Networking/TCPClient.hpp |  39 +++++++++
 2 files changed, 141 insertions(+)
 create mode 100644 csgo-loader/csgo-client/Networking/TCPClient.cpp
 create mode 100644 csgo-loader/csgo-client/Networking/TCPClient.hpp

(limited to 'csgo-loader/csgo-client/Networking')

diff --git a/csgo-loader/csgo-client/Networking/TCPClient.cpp b/csgo-loader/csgo-client/Networking/TCPClient.cpp
new file mode 100644
index 0000000..3bdea21
--- /dev/null
+++ b/csgo-loader/csgo-client/Networking/TCPClient.cpp
@@ -0,0 +1,102 @@
+#include <Networking/TCPClient.hpp>
+#include <UserExperience/UserInterface.hpp>
+
+namespace Networking {
+	// We will only receive up to 256 bytes per cycle.
+	constexpr int BufferSize = 256;
+
+	void TCPClient::SendRawBytes(ByteArray &Bytes) {
+		// Send data.
+		int32_t Result = send(m_Socket, (char *)Bytes.data(), (int)Bytes.size(), 0);
+
+		if(Result == -1)
+			INFO_ASSERT("[000F:00002B00] Server closed the connection suddenly.");
+	}
+
+	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 Encrypted = m_Encryption.Encrypt(Bytes);
+
+		SendRawBytes(Encrypted);
+	}
+
+	ByteArray TCPClient::ReceiveBytes() {
+		ByteArray ReceivedBytes = ReceiveRawBytes();
+
+		// Decrypt incoming data.
+		ByteArray Decrypted = m_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;
+
+		// Allow the socket to time-out.
+		timeval timeout;
+		timeout.tv_sec = 5;
+
+		if(setsockopt(m_Socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof timeout) == INVALID_SOCKET)
+			return false;
+
+		if(setsockopt(m_Socket, SOL_SOCKET, SO_SNDTIMEO, (char *)&timeout, sizeof timeout) == INVALID_SOCKET)
+			return false;
+
+		// Initialise encryption wrapper.
+		ByteArray EncryptionKey = ReceiveRawBytes();
+		m_Encryption.Start(EncryptionKey);
+
+		return true;
+	}
+
+	void TCPClient::Kill() {
+		if(m_Socket)
+			closesocket(m_Socket);
+		
+		WSACleanup();
+	}
+}
\ No newline at end of file
diff --git a/csgo-loader/csgo-client/Networking/TCPClient.hpp b/csgo-loader/csgo-client/Networking/TCPClient.hpp
new file mode 100644
index 0000000..f057cdc
--- /dev/null
+++ b/csgo-loader/csgo-client/Networking/TCPClient.hpp
@@ -0,0 +1,39 @@
+#pragma once
+
+// For encryption wrappers.
+#include <Security/Encryption.hpp>
+
+// WinSocks
+#include <winsock2.h>
+#pragma comment(lib, "ws2_32.lib")
+
+// std::min 
+#include <algorithm>
+
+namespace Networking {
+	// A TCPClient is essentially the same as the TCPConnection counterpart on the server,
+	// however, it independently handles connection.
+	class TCPClient {
+		WSADATA					m_WinSocks;
+		SOCKET					m_Socket;
+		sockaddr_in				m_Context;
+		Wrapper::Encryption		m_Encryption;
+
+	public:
+		TCPClient() = default;
+
+		// Connects to a remote server.
+		// Also handles the initial handshake between server and client.
+		bool Start(uint32_t ServerAddress, uint16_t ServerPort);
+		
+		// Kills the client.
+		void Kill();
+
+		// Wrappers for sending/receiving data.
+		void SendRawBytes(ByteArray &Bytes);
+		ByteArray ReceiveRawBytes();
+
+		void SendBytes(ByteArray &Bytes);
+		ByteArray ReceiveBytes();
+	};
+}
\ No newline at end of file
-- 
cgit v1.2.3