#include <Client.hpp>

/*
	TODO:
	- Finish off shellcode execution wrapper:
		- The shellcode can be executed via two ways
		- Either the code is mapped and called via CreateRemoteThread (allows custom param)
		- or the code is mapped and called via DX9 (does not allow custom param)
		- This will probably be the easiest thing to do.

	- Finish off injection wrapper:
		- Allocate every memory page as a bunch of consecutive 4kb sections (in order to fuck with people trying to dump)

	- Have the loader inject a .DLL :^)

	TODO (Nave):
		- Make the UI look nice.
		- Adapt the server to work with your backend.
		- Add dump protection (closes csgo.exe if a handle is detected, probably explorer shellcode)
*/


#pragma optimize("", off)

int __stdcall WinMain(HINSTANCE inst, HINSTANCE prev, char* str, int cmdshow)
{
	WRAP_IF_DEBUG(
		Utils::OpenConsole();
	);

	///////////////////////////////////////////////////////////////

	//VMProtectBeginMutation("EntryPoint");

	///////////////////////////////////////////////////////////////

	// Create a thread to handle UI.
	std::thread WindowThread([]
	{	
		// Create a window, initialise DirectX context.
		if(!UserInterface->Start())
			ERROR_ASSERT(STR("[000F:00001C00] Failed to initialize. Please contact an administrator."));
		
		UserInterface->RunUiFrame();
	}); WindowThread.detach();

	while(!UserInterface->m_Data.m_Ready) { Sleep(1); }

	// Initialize the syscall manager.
	if(!Syscalls->Start())
		ERROR_ASSERT(STR("[000F:00001B00] Failed to initialize. Please contact an administrator."));

	// Initialize the runtime protection system.
	if(!Protection->Start())
		ERROR_ASSERT(STR("[000F:00001A00] Failed to initialize. Please contact an administrator."));

	// Wait for connection.
	UserInterface->m_Data.m_ExecutionState = UserExperience::EXECUTION_WAITING;

	// Attempt to connect to the remote server.
	WRAP_IF_DEBUG(
		printf("[DEBUG] Server IP: %08x\n", inet_addr("35.165.60.229"));
	);
	
	Networking::TCPClientPtr Client = std::make_unique<Networking::TCPClient>();

	if(!Client->Start(LOCAL_IP, SERVER_PORT))
		ERROR_ASSERT(STR("[000F:0002A000] Server closed the connection unexpectedly."));
	
	// Allow the user to input their log-in data.
	UserInterface->m_Data.m_ExecutionState = UserExperience::EXECUTION_LOG_IN;

	while(UserInterface->m_Data.m_ExecutionState != UserExperience::EXECUTION_WAITING) { Sleep(1); }

	Login::RemoteLoginTransaction LoginTransaction;
	LoginTransaction.Start(UserInterface->m_Data.m_Username, UserInterface->m_Data.m_Password);

	ByteArray Transaction = LoginTransaction.GetHeader();
	Client->SendBytes(Transaction);

	ByteArray LoginResponse = Client->ReceiveBytes();
	if(!LoginTransaction.TranslateResponse(LoginResponse))
		ExitProcess(0);

	// Allow the user to choose a cheat to inject.
	UserInterface->m_Data.m_ExecutionState = UserExperience::EXECUTION_CHOOSE;

	// TODO: Add game selection.
	while(1) { if(GetAsyncKeyState(VK_END) & 0x8000) break; Sleep(1); }

	///////////////////////////////////////////////////////////////

	//VMProtectEnd();

	///////////////////////////////////////////////////////////////
}

#pragma optimize("", on)