mirror of https://github.com/wangyu-/udp2raw.git
156 lines
5.4 KiB
C
156 lines
5.4 KiB
C
/*
|
|
* misc.h
|
|
*
|
|
* Created on: Sep 23, 2017
|
|
* Author: root
|
|
*/
|
|
|
|
#ifndef MISC_H_
|
|
#define MISC_H_
|
|
|
|
#include "common.h"
|
|
#include "log.h"
|
|
#include "network.h"
|
|
|
|
extern int hb_mode;
|
|
extern int hb_len;
|
|
extern char hb_buf[buf_len];
|
|
|
|
extern int mtu_warn;
|
|
|
|
extern int max_rst_allowed;
|
|
extern int max_rst_to_show;
|
|
|
|
extern int enable_dns_resolve;
|
|
|
|
extern int ttl_value;
|
|
|
|
const u32_t max_handshake_conn_num = 10000;
|
|
const u32_t max_ready_conn_num = 1000;
|
|
const u32_t anti_replay_window_size = 4000;
|
|
const int max_conv_num = 10000;
|
|
|
|
const u32_t client_handshake_timeout = 5000; // unit ms
|
|
const u32_t client_retry_interval = 1000; // ms
|
|
|
|
const u32_t server_handshake_timeout = client_handshake_timeout + 5000; // this should be longer than clients. client retry initially ,server retry passtively
|
|
|
|
const int conv_clear_ratio = 30; // conv grabage collecter check 1/30 of all conv one time
|
|
const int conn_clear_ratio = 50;
|
|
const int conv_clear_min = 1;
|
|
const int conn_clear_min = 1;
|
|
|
|
const u32_t conv_clear_interval = 1000; // ms
|
|
const u32_t conn_clear_interval = 1000; // ms
|
|
|
|
const i32_t max_fail_time = 0; // disable
|
|
|
|
const u32_t heartbeat_interval = 600; // ms
|
|
|
|
const u32_t timer_interval = 400; // ms. this should be smaller than heartbeat_interval and retry interval;
|
|
|
|
const uint32_t conv_timeout = 180000; // ms. 120 second
|
|
// const u32_t conv_timeout=30000; //for test
|
|
|
|
const u32_t client_conn_timeout = 10000; // ms.
|
|
const u32_t client_conn_uplink_timeout = client_conn_timeout + 2000; // ms
|
|
|
|
const uint32_t server_conn_timeout = conv_timeout + 60000; // ms. this should be 60s+ longer than conv_timeout,so that conv_manager can destruct convs gradually,to avoid latency glicth
|
|
// const u32_t server_conn_timeout=conv_timeout+10000;//for test
|
|
|
|
const u32_t iptables_rule_keep_interval = 20; // unit: second;
|
|
|
|
enum server_current_state_t { server_idle = 0,
|
|
server_handshake1,
|
|
server_ready }; // server state machine
|
|
enum client_current_state_t { client_idle = 0,
|
|
client_tcp_handshake,
|
|
client_handshake1,
|
|
client_handshake2,
|
|
client_ready,
|
|
client_tcp_handshake_dummy }; // client state machine
|
|
|
|
enum raw_mode_t { mode_faketcp = 0,
|
|
mode_udp,
|
|
mode_icmp,
|
|
mode_end };
|
|
enum program_mode_t { unset_mode = 0,
|
|
client_mode,
|
|
server_mode };
|
|
|
|
union current_state_t {
|
|
server_current_state_t server_current_state;
|
|
client_current_state_t client_current_state;
|
|
};
|
|
|
|
// extern char remote_address[max_address_len];
|
|
// extern char local_ip[100], remote_ip[100],source_ip[100];//local_ip is for -l option,remote_ip for -r option,source for --source-ip
|
|
// extern u32_t local_ip_uint32,remote_ip_uint32,source_ip_uint32;//convert from last line.
|
|
// extern int local_port , remote_port,source_port;//similiar to local_ip remote_ip,buf for port.source_port=0 indicates --source-port is not enabled
|
|
|
|
extern address_t local_addr, remote_addr, source_addr;
|
|
|
|
extern my_ip_t bind_addr;
|
|
|
|
extern int bind_addr_used;
|
|
extern int force_source_ip; // if --source-ip is enabled
|
|
extern int force_source_port;
|
|
extern int source_port;
|
|
|
|
extern my_id_t const_id; // an id used for connection recovery,its generated randomly,it never change since its generated
|
|
|
|
extern int udp_fd; // for client only. client use this fd to listen and handle udp connection
|
|
extern int bind_fd; // bind only,never send or recv. its just a dummy fd for bind,so that other program wont occupy the same port
|
|
extern int epollfd; // fd for epoll
|
|
extern int timer_fd; // the general timer fd for client and server.for server this is not the only timer find,every connection has a timer fd.
|
|
extern int fail_time_counter; // determine if the max_fail_time is reached
|
|
extern int epoll_trigger_counter; // for debug only
|
|
extern int debug_flag; // for debug only
|
|
|
|
extern int simple_rule; // deprecated.
|
|
extern int keep_rule; // whether to monitor the iptables rule periodly,re-add if losted
|
|
extern int auto_add_iptables_rule; // if -a is set
|
|
extern int generate_iptables_rule; // if -g is set
|
|
extern int generate_iptables_rule_add; // if --gen-add is set
|
|
extern int retry_on_error;
|
|
const int retry_on_error_interval = 10;
|
|
|
|
extern int debug_resend; // debug only
|
|
|
|
extern char key_string[1000]; // -k option
|
|
extern char fifo_file[1000];
|
|
|
|
extern raw_mode_t raw_mode;
|
|
extern u32_t raw_ip_version;
|
|
|
|
extern program_mode_t program_mode;
|
|
extern unordered_map<int, const char *> raw_mode_tostring;
|
|
|
|
extern int about_to_exit;
|
|
|
|
extern int socket_buf_size;
|
|
|
|
extern pthread_t keep_thread;
|
|
extern int keep_thread_running;
|
|
|
|
int process_lower_level_arg();
|
|
void print_help();
|
|
void iptables_rule();
|
|
void pre_process_arg(int argc, char *argv[]); // mainly for load conf file;
|
|
int unit_test();
|
|
int set_timer(int epollfd, int &timer_fd);
|
|
int set_timer_server(int epollfd, int &timer_fd, fd64_t &fd64);
|
|
int handle_lower_level(raw_info_t &raw_info);
|
|
|
|
int add_iptables_rule(const char *);
|
|
|
|
int clear_iptables_rule();
|
|
|
|
int iptables_gen_add(const char *s, u32_t const_id);
|
|
int iptables_rule_init(const char *s, u32_t const_id, int keep);
|
|
int keep_iptables_rule();
|
|
|
|
void signal_handler(int sig);
|
|
|
|
#endif /* MISC_H_ */
|