47 lines
1.1 KiB
C
47 lines
1.1 KiB
C
#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 pipe_handle;
|
|
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,
|
|
USER, /* client is running as some user */
|
|
MACHINE /* clinet is running as machine - System, NS or LS */
|
|
} client_type;
|
|
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);
|
|
void agent_process_connection(HANDLE);
|
|
void agent_shutdown();
|
|
void agent_cleanup_connection(struct agent_connection*);
|
|
|
|
int load_config();
|
|
int config_log_level(); |