aixiao/CProxy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add to get the HTTPS IPv6 host.

Browse Source
This commit is contained in:
aixiao 2020-12-15 10:43:06 +08:00
parent 3c0d42da76
commit e082b111f7
9 changed files with 79 additions and 119 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
conf.h
View File

@ -45,7 +45,6 @@ typedef struct CONF {
char *https_regrep_aim, *https_regrep_obj;
int https_regrep_aim_len, https_regrep_obj_len;
// http dns_listen
char *addr;
char *http_req;
@ -57,5 +56,4 @@ char *strncpy_(char *dest, const char *src, size_t n);
void read_conf(char *file, conf * p);
void free_conf(conf * p);
#endif

76
http_proxy.c
View File

@ -41,22 +41,18 @@ static char *read_data(conn_t * in, char *data, int *data_len)
return data;
}
void close_connection(conn_t * conn)
{
epoll_ctl(epollfd, EPOLL_CTL_DEL, conn->fd, NULL);
close(conn->fd);
if ((conn - cts) & 1)
{
if ((conn - cts) & 1) {
char *server_data;
server_data = conn->ready_data;
memset(conn, 0, sizeof(conn_t));
conn->ready_data = server_data;
conn--->fd = -1;
}
else
{
} else {
free(conn->ready_data);
free(conn->incomplete_data);
memset(conn, 0, sizeof(conn_t));
@ -66,27 +62,22 @@ void close_connection(conn_t *conn)
close_connection(conn);
}
static void serverToClient(conn_t * server)
{
conn_t *client;
int write_len;
client = server - 1;
while ((server->ready_data_len = read(server->fd, server->ready_data, BUFFER_SIZE)) > 0)
{
while ((server->ready_data_len = read(server->fd, server->ready_data, BUFFER_SIZE)) > 0) {
dataEncode(server->ready_data, server->ready_data_len);
write_len = write(client->fd, server->ready_data, server->ready_data_len);
if (write_len <= 0)
{
if (write_len <= 0) {
if (write_len == 0 || errno != EAGAIN)
close_connection(server);
else
server->sent_len = 0;
return;
}
else if (write_len < server->ready_data_len)
{
} else if (write_len < server->ready_data_len) {
server->sent_len = write_len;
ev.events = EPOLLIN | EPOLLOUT | EPOLLET;
ev.data.ptr = client;
@ -124,20 +115,7 @@ void clientToserver(conn_t * in)
// 判断请求类型
static int8_t request_type(char *data)
{
if (strncmp(data, "GET", 3) == 0 ||
strncmp(data, "POST", 4) == 0 ||
strncmp(data, "CONNECT", 7) == 0 ||
strncmp(data, "HEAD", 4) == 0 ||
strncmp(data, "PUT", 3) == 0 ||
strncmp(data, "OPTIONS", 7) == 0 ||
strncmp(data, "MOVE", 4) == 0 ||
strncmp(data, "COPY", 4) == 0 ||
strncmp(data, "TRACE", 5) == 0 ||
strncmp(data, "DELETE", 6) == 0 ||
strncmp(data, "LINK", 4) == 0 ||
strncmp(data, "UNLINK", 6) == 0 ||
strncmp(data, "PATCH", 5) == 0 ||
strncmp(data, "WRAPPED", 7) == 0)
if (strncmp(data, "GET", 3) == 0 || strncmp(data, "POST", 4) == 0 || strncmp(data, "CONNECT", 7) == 0 || strncmp(data, "HEAD", 4) == 0 || strncmp(data, "PUT", 3) == 0 || strncmp(data, "OPTIONS", 7) == 0 || strncmp(data, "MOVE", 4) == 0 || strncmp(data, "COPY", 4) == 0 || strncmp(data, "TRACE", 5) == 0 || strncmp(data, "DELETE", 6) == 0 || strncmp(data, "LINK", 4) == 0 || strncmp(data, "UNLINK", 6) == 0 || strncmp(data, "PATCH", 5) == 0 || strncmp(data, "WRAPPED", 7) == 0)
return HTTP_TYPE;
return OTHER_TYPE;
}
@ -157,7 +135,8 @@ int check_ipversion(char * address)
return 0;
}
int create_connection6(char *remote_host, int remote_port) {
int create_connection6(char *remote_host, int remote_port)
{
struct addrinfo hints, *res = NULL;
int sock;
int validfamily = 0;
@ -176,7 +155,6 @@ int create_connection6(char *remote_host, int remote_port) {
hints.ai_family = validfamily;
hints.ai_flags |= AI_NUMERICHOST; // remote_host是有效的数字ip，跳过解析
}
// 检查指定的主机是否有效。 如果remote_host是主机名，尝试解析地址
if (getaddrinfo(remote_host, portstr, &hints, &res) != 0) {
errno = EFAULT;
@ -204,8 +182,7 @@ int create_connection6(char *remote_host, int remote_port) {
static int8_t copy_data(conn_t * ct)
{
dataEncode(ct->incomplete_data, ct->incomplete_data_len);
if (ct->ready_data)
{
if (ct->ready_data) {
char *new_data;
new_data = (char *)realloc(ct->ready_data, ct->ready_data_len + ct->incomplete_data_len);
@ -215,9 +192,7 @@ static int8_t copy_data(conn_t *ct)
memcpy(new_data + ct->ready_data_len, ct->incomplete_data, ct->incomplete_data_len);
ct->ready_data_len += ct->incomplete_data_len;
free(ct->incomplete_data);
}
else
{
} else {
ct->ready_data = ct->incomplete_data;
ct->ready_data_len = ct->incomplete_data_len;
}
@ -235,8 +210,7 @@ void tcp_in(conn_t * in, conf * configure)
if (in->fd < 0)
return;
//如果in - cts是奇数，那么是服务端触发事件
if ((in - cts) & 1)
{
if ((in - cts) & 1) {
in->timer = (in - 1)->timer = 0;
if (in->ready_data_len <= 0)
serverToClient(in);
@ -247,8 +221,7 @@ void tcp_in(conn_t * in, conf * configure)
in->incomplete_data = read_data(in, in->incomplete_data, &in->incomplete_data_len);
//printf("%s", in->incomplete_data);
if (in->incomplete_data == NULL)
{
if (in->incomplete_data == NULL) {
close_connection(in);
return;
}
@ -267,12 +240,10 @@ void tcp_in(conn_t * in, conf * configure)
}
if (in->incomplete_data == NULL || copy_data(in) != 0)
{
if (in->incomplete_data == NULL || copy_data(in) != 0) {
close_connection(in);
return;
}
// 这个判断是防止 多次读取客户端数据，但是没有和服务端建立连接，导致报错
if (server->fd >= 0)
tcp_out(server);
@ -291,21 +262,16 @@ void tcp_out(conn_t *to)
from = to + 1;
from->timer = to->timer = 0;
write_len = write(to->fd, from->ready_data + from->sent_len, from->ready_data_len - from->sent_len);
if (write_len == from->ready_data_len - from->sent_len)
{
if (write_len == from->ready_data_len - from->sent_len) {
//服务端的数据可能没全部写入到客户端
if ((from - cts) & 1)
{
if ((from - cts) & 1) {
serverToClient(from);
if (from->fd >= 0 && from->ready_data_len == 0)
{
if (from->fd >= 0 && from->ready_data_len == 0) {
ev.events = EPOLLIN | EPOLLET;
ev.data.ptr = to;
epoll_ctl(epollfd, EPOLL_CTL_MOD, to->fd, &ev);
}
}
else
{
} else {
ev.events = EPOLLIN | EPOLLET;
ev.data.ptr = to;
epoll_ctl(epollfd, EPOLL_CTL_MOD, to->fd, &ev);
@ -313,16 +279,12 @@ void tcp_out(conn_t *to)
from->ready_data = NULL;
from->ready_data_len = 0;
}
}
else if (write_len > 0)
{
} else if (write_len > 0) {
from->sent_len += write_len;
ev.events = EPOLLIN | EPOLLOUT | EPOLLET;
ev.data.ptr = to;
epoll_ctl(epollfd, EPOLL_CTL_MOD, to->fd, &ev);
}
else if (errno != EAGAIN)
{
} else if (errno != EAGAIN) {
close_connection(to);
}
}

5
http_proxy.h
View File

@ -16,12 +16,9 @@ typedef struct tcp_connection {
char *ready_data, *incomplete_data;
int fd, ready_data_len, incomplete_data_len, sent_len, timer;
uint16_t destPort;
unsigned reread_data :1,
request_type :1,
keep_alive :1;
unsigned reread_data:1, request_type:1, keep_alive:1;
} conn_t;
extern conn_t cts[MAX_CONNECTION];
extern void tcp_in(conn_t * in, conf * configure);
extern void tcp_out(conn_t * out);

11
http_request.c
View File

@ -160,6 +160,17 @@ int extract_host(char *header, char *host, char *port)
{
char *_p = strstr(header, "CONNECT"); // 在 CONNECT 方法中解析 隧道主机名称及端口号
if (_p) {
if (strchr(header, '[') || strchr(header, ']')) { // IPv6
char *_p1 = strchr(header, '[');
char *_p2 = strchr(_p1 + 1, ']');
strncpy(host, _p1 + 1, (int)(_p2 - _p1) - 1);
remote_port = 443;
return 0;
}
char *_p1 = strchr(_p, ' ');
char *_p2 = strchr(_p1 + 1, ':');
char *_p3 = strchr(_p1 + 1, ' ');

4
httpdns.c
View File

@ -484,7 +484,8 @@ int udp_listen(char *ip, int port)
return fd;
}
int httpdns_initialize(conf * configure) {
int httpdns_initialize(conf * configure)
{
char *p;
p = strchr(configure->addr, ':');
host_value = configure->addr;
@ -501,4 +502,3 @@ int httpdns_initialize(conf * configure) {
host_value_len = strlen(host_value);
return 0;
}

3
httpdns.h
View File

@ -23,7 +23,6 @@
#define DATA_SIZE 512
#define HTTP_RSP_SIZE 1024
typedef struct dns_connection {
char dns_req[DATA_SIZE];
struct sockaddr_in src_addr;
@ -46,9 +45,7 @@ struct dns_cache {
extern dns_t dns_list[DNS_MAX_CONNECTION];
extern struct epoll_event evs[DNS_MAX_CONNECTION + 1], event;
void *httpdns_loop(void *p);
int httpdns_initialize(conf * configure);
#endif

6
main.c
View File

@ -166,7 +166,6 @@ void *http_proxy_loop(void *p)
conf *configure = (conf *) p;
int n;
ev.events = EPOLLIN;
ev.data.fd = server_sock;
if (-1 == epoll_ctl(epollfd, EPOLL_CTL_ADD, server_sock, &ev)) {
@ -186,8 +185,7 @@ void *http_proxy_loop(void *p)
//printf("accept_client6()!!!\n");
accept_client6();
;
}
else if (events[n].data.fd == server_sock) {
} else if (events[n].data.fd == server_sock) {
//printf("accept_client()!!!\n");
accept_client();
} else {
@ -305,7 +303,6 @@ void server_ini()
perror("daemon");
return;
}
//while (process-- > 1 && fork() == 0);
}
@ -431,7 +428,6 @@ void _main(int argc, char *argv[])
exit(1);
}
if (setegid(configure->uid) == -1 || seteuid(configure->uid) == -1) // 设置uid
exit(1);

3
timeout.c
View File

@ -14,8 +14,7 @@ void *tcp_timeout_check(void *nullPtr)
if (cts[i].fd > -1) {
if (cts[i].timer >= timeout_minute) {
close_connection(cts + i);
}
else
} else
cts[i].timer++;
}
}