#ifndef MAIN_H
#define MAIN_H

#include <netinet/in.h>
#include <arpa/inet.h>
#include "common.h"
#include "httpdns.h"
#include "conf.h"

// 字体颜色
#define NONE "\033[m"
#define RED "\033[0;32;31m"
#define LIGHT_RED "\033[1;31m"
#define GREEN "\033[0;32;32m"
#define LIGHT_GREEN "\033[1;32m"
#define BLUE "\033[0;32;34m"
#define LIGHT_BLUE "\033[1;34m"
#define DARY_GRAY "\033[1;30m"
#define CYAN "\033[0;36m"
#define LIGHT_CYAN "\033[1;36m"
#define PURPLE "\033[0;35m"
#define LIGHT_PURPLE "\033[1;35m"
#define BROWN "\033[0;33m"
#define YELLOW "\033[1;33m"
#define LIGHT_GRAY "\033[0;37m"
#define WHITE "\033[1;37m"

#define LOG(fmt...)  do { fprintf(stderr, ##fmt); } while(0)

/* 数据类型 */
#define OTHER 1
#define HTTP 2
#define HTTP_OTHERS 3
#define HTTP_CONNECT 4
/* 处理TCP请求模式 */
#define WAP 1
#define WAP_CONNECT 2
#define NET_CONNECT 3
#define NET_PROXY 4

struct httpudp {
    struct sockaddr_in dst;
    char *http_request, *original_http_request; //original_http_request为初始化生成的请求头，用来配合use_hdr语法
    int http_request_len, original_http_request_len;
    unsigned encodeCode,        //数据编码传输
     httpsProxy_encodeCode;     //CONNECT代理编码
};

typedef struct connection {
    struct sockaddr_in original_dst;
    char *ready_data;           //已就绪的数据，可以发送
    char *incomplete_data;      //存放不是完整的请求头
    char *host;
    char *connect;              //存放CONNECT请求
    int connect_len;            //CONNECT请求的长度
    int incomplete_data_len, ready_data_len, sent_len, fd, timer;
    uint16_t original_port;
    unsigned reqType:3,         //请求类型
     first_connection:1,        //发送客户端数据前是否首先进行CONNECT连接
     is_httpsProxy:1;           //如果真 则表示连接是走HTTPS模块
} tcp_t;

struct global {
    int tcp_listen_fd, dns_listen_fd, udp_listen_fd, uid, procs, timeout_m;
    unsigned mode:3, strict_modify:1;
};

extern struct global global;

extern uint16_t tcp_listen_port;
extern struct httpudp udp;

#endif