first

1 files changed, 130 insertions, 0 deletions

diff --git a/client/connect.hpp b/client/connect.hpp
new file mode 100644
index 0000000..8b36687
--- /dev/null
+++ b/client/connect.hpp
@@ -0,0 +1,130 @@
+#pragma once
+
+#include <Windows.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#pragma comment( lib, "ws2_32.lib" )
+
+#include <vector>
+
+#include "err.hpp"
+#include "util.hpp"
+
+/*
+protocol rules:
+every msg must be xored
+first byte is the xorkey
+
+*/
+
+namespace client
+{
+	constexpr auto PORT_NUM = 6969;
+	constexpr auto BUFFER_SIZE = 255;
+
+	class c_connect {
+	public:
+		c_connect( const char* ip ) : 
+			m_ip( inet_addr( ip ) ) { }
+
+		~c_connect( ) {
+			if( m_socket )
+				closesocket( m_socket );
+
+			WSACleanup( );
+		}
+
+		bool setup( ) {
+			int code{ };
+
+			if( WSAStartup( MAKEWORD( 2, 2 ), &m_wsdata ) )
+				code = err::ERR_WSA;
+			else {
+				m_socket = socket( AF_INET, SOCK_STREAM, 0 );
+				if( m_socket == INVALID_SOCKET )
+					code = err::ERR_WSA;
+			}
+
+			if( code != err::ERR_NONE ) {
+				MessageBoxA( nullptr, err::translate_err( code ), "", MB_OK );
+				return false;
+			}
+
+			return true;
+		}
+
+		bool connect( ) {
+			sockaddr_in server_address{ };
+			int code{ };
+
+			server_address.sin_addr.s_addr = m_ip;
+			server_address.sin_port = htons( PORT_NUM );
+			server_address.sin_family = AF_INET;
+			
+			code = ::connect( m_socket, ( sockaddr* )( &server_address ), 
+				sizeof( server_address ) );
+
+			if( code == -1 ) {
+				MessageBoxA( nullptr, err::translate_err( err::ERR_CONNECT ), "", MB_OK );
+				return false;
+			}
+
+			return true;
+		}
+
+		void decode_buffer( uint8_t* buf, size_t length ) {
+			auto key = buf[ 0 ];
+			for( size_t i{ 1 }; i < length; ++i )
+				buf[ i ] ^= key;
+		}
+
+		std::vector< uint8_t > get_msg( ) {
+			std::vector< uint8_t > ret;
+			char buffer[ BUFFER_SIZE ];
+			int received = 0;
+			
+			while( true ) {
+				received = recv( m_socket, buffer, BUFFER_SIZE, 0 );
+				if( received < 0 )
+					break;
+
+				for( int i{ }; i < received; ++i )
+					ret.push_back( buffer[ i ] );
+
+				if( received < BUFFER_SIZE )
+					break;
+			}
+
+			if( ret.size( ) ) {
+				decode_buffer( ret.data( ), ret.size( ) );
+				ret.erase( ret.begin( ) );
+			}
+			return ret;
+		}
+
+		void send_msg( const uint8_t* msg, size_t length ) {
+			auto new_buffer = ( uint8_t* )( malloc( length + 1 ) );
+			auto key = util::random_number( 0, 255 ) & 0xff;
+
+			new_buffer[ 0 ] = key;
+			memcpy( new_buffer + 1,
+				msg,
+				length );
+
+			for( size_t i = 1; i < length + 1; ++i ) {
+				new_buffer[ i ] ^= key;
+			}
+
+			send( m_socket, ( char* )new_buffer, length + 1, 0 );
+
+			free( new_buffer );
+		}
+
+	private:
+		SOCKET  m_socket;
+		WSADATA m_wsdata;
+		int     m_ip;
+	};
+}
\ No newline at end of file