#pragma once

#include <tfe_stream.h>
#include <event2/event.h>
#include <sender_scm.h>
#include <ssl_stream_core.h>

struct ssl_mgr;
struct key_keeper;
struct acceptor_kni_v2;
struct watchdog_kni;
struct breakpad_instance;

enum TFE_STAT_FIELD
{
	STAT_SIGPIPE,
	/* FDs */
	STAT_FD_OPEN_BY_KNI_ACCEPT,
	STAT_FD_CLOSE_BY_KNI_ACCEPT_FAIL,
	STAT_FD_CLOSE,

	/* Stream */
	STAT_STREAM_OPEN,
	STAT_STREAM_CLS,
	STAT_STREAM_CLS_DOWN_EOF,
	STAT_STREAM_CLS_UP_EOF,
	STAT_STREAM_CLS_DOWN_ERR,
	STAT_STREAM_CLS_UP_ERR,
	STAT_STREAM_CLS_KILL,

	/* Action */
	STAT_STREAM_INTERCEPT,
	STAT_STREAM_BYPASS,
	STAT_STREAM_INCPT_BYTES,
	STAT_STREAM_INCPT_DOWN_BYTES,
	STAT_STREAM_INCPT_UP_BYTES,

	/* Protocol */
	STAT_STREAM_TCP_PLAIN,
	STAT_STREAM_TCP_SSL,
	TFE_STAT_MAX
};

struct tfe_proxy_tcp_options
{
	/* TCP OPTIONS */
	int sz_rcv_buffer;
	int sz_snd_buffer;

    /* TRACE FOR DEBUG */
    int enable_overwrite;
    int tcp_nodelay;
	int so_keepalive;
    int tcp_keepidle;
    int tcp_keepintvl;
    int tcp_keepcnt;
    int tcp_user_timeout;

    int tcp_ttl_upstream;
    int tcp_ttl_downstream;
};

struct tfe_proxy_rate_limit_options
{
	unsigned int read_rate;
	unsigned int read_burst;
	unsigned int write_rate;
	unsigned int write_burst;
};

struct tfe_proxy_accept_para
{
	/* Both upstream and downstream FDs */
	evutil_socket_t upstream_fd;
	evutil_socket_t downstream_fd;

	/* Session Type */
	bool is_set_session_type;
	enum tfe_stream_proto session_type;
	bool passthrough;

	/* addition info */
	unsigned int keyring_id;
};

enum tfe_load_balance_algo
{
	LEAST_CONN = 0,
	ROUND_ROBIN = 1,
};

struct tfe_proxy
{
	char name[TFE_SYMBOL_MAX];
	struct event_base * evbase;
	struct event * sev[8];
	struct event * gcev;

	void * logger;
	void * fs_handle;
	unsigned int nr_work_threads;
	struct tfe_thread_ctx * work_threads[TFE_THREAD_MAX];
	int make_work_thread_sleep;

	unsigned int nr_modules;
	struct tfe_plugin * modules;

	struct ssl_mgr * ssl_mgr_handler;
	struct ssl_policy_enforcer* ssl_ply_enforcer;
	struct key_keeper * key_keeper_handler;

	unsigned int en_kni_v1_acceptor;
    unsigned int en_kni_v2_acceptor;
	unsigned int en_kni_v3_acceptor;

    struct acceptor_kni_v1 * kni_v1_acceptor;
    struct acceptor_kni_v2 * kni_v2_acceptor;
	struct acceptor_kni_v3 * kni_v3_acceptor;
    struct sender_scm * scm_sender;
    struct watchdog_kni * watchdog_kni;
	struct watchdog_tfe * watchdog_tfe;

	/* DEBUG OPTIONS */
	unsigned int tcp_all_passthrough;
	struct tfe_proxy_tcp_options tcp_options;

	/* GLOBAL RATELIMIT */
	unsigned int en_rate_limit;
	struct tfe_proxy_rate_limit_options rate_limit_options;

	/* PERFOMANCE MONIOTR VARIABLES*/
	long long stat_val[TFE_STAT_MAX];
	int fs_id[TFE_STAT_MAX];

	/* Crash Report */
	struct breakpad_instance * breakpad;

	/* cpu affinity */
	unsigned int enable_cpu_affinity;
	unsigned int cpu_affinity_mask[TFE_THREAD_MAX];

	/* load balancing */
	enum tfe_load_balance_algo load_balance;
};

extern struct tfe_proxy * g_default_proxy;

#define TFE_PROXY_STAT_INCREASE(field, val)			\
do { __atomic_fetch_add(&g_default_proxy->stat_val[field], val, __ATOMIC_RELAXED); } while(0)

struct tfe_thread_ctx * tfe_proxy_thread_ctx_acquire(struct tfe_proxy * ctx);
void tfe_proxy_thread_ctx_release(struct tfe_thread_ctx * thread_ctx);

struct tfe_proxy * tfe_proxy_new(const char * profile);
int tfe_proxy_fds_accept(struct tfe_proxy * ctx, int fd_downstream, int fd_upstream, struct tfe_cmsg * cmsg);
void tfe_proxy_run(struct tfe_proxy * proxy);
int tfe_thread_set_affinity(int core_id);
