#ifndef CPROXY_H
#define CPROXY_H

#include <arpa/inet.h>
#include <errno.h>
#include <libgen.h>
#include <netdb.h>
#include <resolv.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<fcntl.h>

#define BUF_SIZE 8192
#define BUF_SIZES 1024

#define SERVER_SOCKET_ERROR -1
#define SERVER_SETSOCKOPT_ERROR -2
#define SERVER_BIND_ERROR -3
#define SERVER_LISTEN_ERROR -4
#define CLIENT_SOCKET_ERROR -5
#define CLIENT_RESOLVE_ERROR -6
#define CLIENT_CONNECT_ERROR -7

char remote_host[128];
int remote_port;
int local_port;

int server_sock;
int client_sock;
int remote_sock;

char *header_buffer;
int len_header_buffer;
char *complete_data;
int len_complete_data;

int SIGN;

// 配置文件结构
typedef struct CONF {
    int server_port;            // server module
    char *server_pid_file;

    int http_port;              // http module
    char *http_ip, *http_del, *http_first;

    int https_port;             // https module
    char *https_ip, *https_del, *https_first;

    int len_server_pid_file;    // length
    int len_http_ip, len_http_del, len_http_first;
    int len_https_ip, len_https_del, len_https_first;
} conf;

// 请求类型
#define OTHER 1
#define HTTP 2
#define HTTP_OTHERS 3
#define HTTP_CONNECT 4

void read_conf(char *file, conf *p);
void free_conf(conf * p);

void server_loop(conf * configure);
void handle_client(int client_sock, struct sockaddr_in client_addr, conf *configure);
void forward_data(int source_sock, int destination_sock);
int send_data(int socket, char *buffer, int len);
int receive_data(int socket, char *buffer, int len);

int create_connection(conf * configure, int SIGN);
int _main(int argc, char *argv[]);
int extract_host(char *header);
int replacement_http_head(char *header_buffer, char *remote_host, int *remote_port, int *SIGN, conf *p);
uint8_t request_type(char *req);
void forward_header(int destination_sock);
void rewrite_header();
char help_information(void);

#endif