#include <Windows.h>
#include <stdio.h>
#define MAX_MESSAGE_SIZE 256 * 1024

#define SSH_ROOT L"SOFTWARE\\SSH"
#define SSH_AGENT_ROOT SSH_ROOT L"\\Agent"
#define SSH_KEYS_KEY L"Keys"
#define SSH_KEYS_ROOT SSH_ROOT L"\\" SSH_KEYS_KEY

#define HEADER_SIZE 4

struct agent_connection {
	OVERLAPPED ol;
	HANDLE connection;
	struct {
		DWORD num_bytes;
		DWORD transferred;
		char buf[MAX_MESSAGE_SIZE];
		DWORD buf_size;
	} io_buf;
	enum {
		LISTENING = 0,
		READING_HEADER,
		READING,
		WRITING,
		DONE
	} state;
	enum {
		UNKNOWN = 0,
		OTHER,
		LOCAL_SYSTEM,
		SSHD,
		NETWORK_SERVICE
	} client_process;
        HANDLE auth_token;
        HANDLE hProfile;
};

void agent_connection_on_io(struct agent_connection*, DWORD, OVERLAPPED*);
void agent_connection_on_error(struct agent_connection* , DWORD );
void agent_connection_disconnect(struct agent_connection*);

void agent_start(BOOL, BOOL, HANDLE);
void agent_shutdown();
void agent_cleanup_connection(struct agent_connection*);

int load_config();
int config_log_level();