#ifndef _SERVER_H
#define _SERVER_H

#include <ev.h>
#include <time.h>

#include "encrypt.h"
#include "jconf.h"

struct listen_ctx {
	ev_io io;
	int fd;
    struct timeval timeout;
	struct sockaddr sock;
};

struct server_ctx {
	ev_io io;
	int connected;
	struct server *server;
};

struct server {
	int fd;
	char buf[BUF_SIZE]; // server send from, remote recv into
    char stage;
	int buf_len;
    struct timeval timeout;
    struct rc4_state *e_ctx;
    struct rc4_state *d_ctx;
	struct server_ctx *recv_ctx;
	struct server_ctx *send_ctx;
	struct remote *remote;
};

struct remote_ctx {
	ev_io io;
    ev_timer watcher;
	int connected;
	struct remote *remote;
};

struct remote {
	int fd;
	char buf[BUF_SIZE]; // remote send from, server recv into
	int buf_len;
	struct remote_ctx *recv_ctx;
	struct remote_ctx *send_ctx;
	struct server *server;
};


static void accept_cb (EV_P_ ev_io *w, int revents);
static void server_recv_cb (EV_P_ ev_io *w, int revents);
static void server_send_cb (EV_P_ ev_io *w, int revents);
static void remote_recv_cb (EV_P_ ev_io *w, int revents);
static void remote_send_cb (EV_P_ ev_io *w, int revents);

struct remote* new_remote(int fd, struct timeval timeout);
struct remote *connect_to_remote(char *remote_host, char *remote_port, struct timeval timeout);
void free_remote(struct remote *remote);
void close_and_free_remote(EV_P_ struct remote *remote);
struct server* new_server(int fd);
void free_server(struct server *server);
void close_and_free_server(EV_P_ struct server *server);

#endif // _SERVER_H