#ifndef YTB_CONFIG_H
#define YTB_CONFIG_H

#ifndef KERNEL_SPACE
#define USER_SPACE
#endif

#include "raw_replacements.h"
#include "types.h"

typedef int (*raw_send_t)(const unsigned char *data, unsigned int data_len);
/**
 * Sends the packet after delay_ms. The function should schedule send and return immediately
 * (for example, open daemon thread)
 */
typedef int (*delayed_send_t)(const unsigned char *data, unsigned int data_len, unsigned int delay_ms);

struct instance_config_t {
	raw_send_t send_raw_packet;
	delayed_send_t send_delayed_packet;
};
extern struct instance_config_t instance_config;

struct udp_dport_range {
	uint16_t start;
	uint16_t end;
};

struct section_config_t {
	const char *domains_str;
	unsigned int domains_strlen;

	int fragmentation_strategy;
	int frag_sni_reverse;
	int frag_sni_faked;
	int faking_strategy;
	int frag_middle_sni;
	int frag_sni_pos;
	unsigned char faking_ttl;
	int fake_sni;
	unsigned int fake_sni_seq_len;

#define FAKE_PAYLOAD_RANDOM	0
#define FAKE_PAYLOAD_CUSTOM	1
// In default mode all other options will be skipped.
#define FAKE_PAYLOAD_DEFAULT	2
	int fake_sni_type;

	/* In milliseconds */
	unsigned int seg2_delay;
	int synfake;
	unsigned int synfake_len;

	const char *exclude_domains_str;
	unsigned int exclude_domains_strlen;
	unsigned int all_domains;

	const char *fake_sni_pkt;
	unsigned int fake_sni_pkt_sz;

	const char *fake_custom_pkt;
	unsigned int fake_custom_pkt_sz;

	unsigned int fk_winsize;
	int fakeseq_offset;

#define SNI_DETECTION_PARSE 0
#define SNI_DETECTION_BRUTE 1
	int sni_detection;

	int udp_mode;
	unsigned int udp_fake_seq_len;
	unsigned int udp_fake_len;
	int udp_faking_strategy;

	struct udp_dport_range *udp_dport_range;
	int udp_dport_range_len;
	int udp_filter_quic;
};

#define MAX_CONFIGLIST_LEN 64

struct config_t {
	unsigned int queue_start_num;
	int threads;
	int use_gso;
	int use_ipv6;
	unsigned int mark;
	int daemonize;
	// Same as daemon() noclose
	int noclose;
	int syslog;

#define VERBOSE_INFO	0
#define VERBOSE_DEBUG	1
#define VERBOSE_TRACE	2
	int verbose;

	struct section_config_t default_config;
	struct section_config_t custom_configs[MAX_CONFIGLIST_LEN];
	int custom_configs_len;
};

extern struct config_t config;

#define ITER_CONFIG_SECTIONS(section) \
for (struct section_config_t *section = &config.default_config + config.custom_configs_len; section >= &config.default_config; section--)

#define CONFIG_SECTION_NUMBER(section) (int)((section) - &config.default_config)

#define MAX_THREADS 16

#ifndef THREADS_NUM
#define THREADS_NUM 1
#endif

#if THREADS_NUM > MAX_THREADS
#error "Too much threads"
#endif

#ifndef NOUSE_GSO
#define USE_GSO
#endif

#define FRAG_STRAT_TCP	0
#define FRAG_STRAT_IP	1
#define FRAG_STRAT_NONE	2

#ifndef FRAGMENTATION_STRATEGY
#define FRAGMENTATION_STRATEGY FRAG_STRAT_TCP
#endif

#define DEFAULT_RAWSOCKET_MARK (1 << 15)

#ifdef USE_SEG2_DELAY
#define SEG2_DELAY 100
#endif

#define FAKE_TTL 8

// Will invalidate fake packets by out-of-ack_seq out-of-seq request
#define FAKE_STRAT_RAND_SEQ	(1 << 0)
// Will assume that GGC server is located further than FAKE_TTL
// Thus, Fake packet will be eliminated automatically.
#define FAKE_STRAT_TTL		(1 << 1)
#define FAKE_STRAT_PAST_SEQ	(1 << 2)
#define FAKE_STRAT_TCP_CHECK	(1 << 3)
#define FAKE_STRAT_TCP_MD5SUM	(1 << 4)
#define FAKE_STRAT_UDP_CHECK	(1 << 5)

#define FAKE_STRAT_COUNT	6

/**
 * This macros iterates through all faking strategies and executes code under it.
 * destination strategy will be available under name of `strategy` variable.
 */
#define ITER_FAKE_STRAT(fake_bitmask, strategy) \
for (int strategy = 1; strategy <= (1 << FAKE_STRAT_COUNT); strategy <<= 1) \
if ((fake_bitmask) & strategy) 

#ifndef FAKING_STRATEGY
#define FAKING_STRATEGY FAKE_STRAT_PAST_SEQ
#endif

#define MAX_FAKE_SIZE 1300

#if !defined(SILENT) && !defined(KERNEL_SPACE)
#define DEBUG
#endif

// The Maximum Transmission Unit size for rawsocket
// Larger packets will be fragmented. Applicable for Chrome's kyber.
#define AVAILABLE_MTU 1400

#define DEFAULT_QUEUE_NUM 537

#define MAX_PACKET_SIZE 8192

#define DEFAULT_SNISTR "googlevideo.com,ggpht.com,ytimg.com,youtube.com,play.google.com,youtu.be,googleapis.com,googleusercontent.com,gstatic.com,l.google.com"

static const char defaul_snistr[] = DEFAULT_SNISTR;

enum {
	UDP_MODE_DROP,
	UDP_MODE_FAKE,
};

enum {
	UDP_FILTER_QUIC_DISABLED,
	UDP_FILTER_QUIC_ALL,
};

#define default_section_config {				\
	.frag_sni_reverse = 1,                                  \
	.frag_sni_faked = 0,                                    \
	.fragmentation_strategy = FRAGMENTATION_STRATEGY,       \
	.faking_strategy = FAKING_STRATEGY,                     \
	.faking_ttl = FAKE_TTL,                                 \
	.fake_sni = 1,                                          \
	.fake_sni_seq_len = 1,                                  \
	.fake_sni_type = FAKE_PAYLOAD_DEFAULT,                  \
	.frag_middle_sni = 1,                                   \
	.frag_sni_pos = 1,                                      \
	.fakeseq_offset = 10000,                                \
	.synfake = 0,                                           \
	.synfake_len = 0,                                       \
                                                                \
	.seg2_delay = 0,                                        \
                                                                \
	.domains_str = defaul_snistr,                           \
	.domains_strlen = sizeof(defaul_snistr),                \
                                                                \
	.exclude_domains_str = "",                              \
	.exclude_domains_strlen = 0,                            \
                                                                \
	.fake_sni_pkt = fake_sni_old,                           \
	.fake_sni_pkt_sz = sizeof(fake_sni_old) - 1,		\
	.fake_custom_pkt = custom_fake_buf,                     \
	.fake_custom_pkt_sz = 0,                                \
	.sni_detection = SNI_DETECTION_PARSE,                   \
								\
	.udp_mode = UDP_MODE_FAKE,				\
	.udp_fake_seq_len = 6,					\
	.udp_fake_len = 64,					\
	.udp_faking_strategy = FAKE_STRAT_UDP_CHECK,		\
	.udp_dport_range = NULL,				\
	.udp_dport_range_len = 0,				\
	.udp_filter_quic = UDP_FILTER_QUIC_DISABLED,		\
}

#endif /* YTB_CONFIG_H */