Compare commits
No commits in common. "master" and "9a28c71d6fab4fbbdf42434433ecfc0e96fbca01" have entirely different histories.
master
...
9a28c71d6f
5
README.md
Executable file → Normal file
5
README.md
Executable file → Normal file
@ -2,7 +2,8 @@
|
|||||||
|
|
||||||
修改自mproxy(https://github.com/examplecode/mproxy), 作为CProxy服务端. 仅代理TCP.
|
修改自mproxy(https://github.com/examplecode/mproxy), 作为CProxy服务端. 仅代理TCP.
|
||||||
支持客户端IP白名单.
|
支持客户端IP白名单.
|
||||||
支持IPV4/IPV6.
|
支持IPV4.
|
||||||
|
支持IPV6.
|
||||||
|
|
||||||
## 参数
|
## 参数
|
||||||
|
|
||||||
@ -20,7 +21,7 @@
|
|||||||
// 本地端口
|
// 本地端口
|
||||||
local_port="127";
|
local_port="127";
|
||||||
|
|
||||||
// IO标志 R_C_DEC、W_S_ENC、FLG_NONE. W_S_ENC 在转发数据时对数据进行编码, R_C_DEC 接收数据时解码数据, FLG_NONE 正常数据流不进行编解码
|
// IO标志 R_C_DEC、W_S_ENC. W_S_ENC在转发数据时对数据进行编码, R_C_DEC 接收数据时解码数据
|
||||||
io_flag="R_C_DEC";
|
io_flag="R_C_DEC";
|
||||||
|
|
||||||
// 编码
|
// 编码
|
||||||
|
144
ais.c
Executable file → Normal file
144
ais.c
Executable file → Normal file
@ -48,7 +48,7 @@
|
|||||||
#define LOG(fmt...) do { fprintf(stderr,"%s %s ",__DATE__,__TIME__); fprintf(stderr, ##fmt); } while(0)
|
#define LOG(fmt...) do { fprintf(stderr,"%s %s ",__DATE__,__TIME__); fprintf(stderr, ##fmt); } while(0)
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
char remote_host[270];
|
char remote_host[128];
|
||||||
int remote_port;
|
int remote_port;
|
||||||
int local_port;
|
int local_port;
|
||||||
|
|
||||||
@ -70,9 +70,7 @@ enum {
|
|||||||
static int io_flag; /* 网络io的一些标志位 */
|
static int io_flag; /* 网络io的一些标志位 */
|
||||||
static int m_pid; /* 保存主进程id */
|
static int m_pid; /* 保存主进程id */
|
||||||
|
|
||||||
conf *configure = NULL;
|
void server_loop(int signal, char *conffile);
|
||||||
|
|
||||||
void server_loop(int signal, conf *p);
|
|
||||||
void stop_server();
|
void stop_server();
|
||||||
void handle_client(int client_sock, struct sockaddr_in client_addr);
|
void handle_client(int client_sock, struct sockaddr_in client_addr);
|
||||||
void forward_header(int destination_sock);
|
void forward_header(int destination_sock);
|
||||||
@ -86,7 +84,6 @@ const char *get_work_mode();
|
|||||||
int create_connection6(char *remote_host, int remote_port);
|
int create_connection6(char *remote_host, int remote_port);
|
||||||
int _main(int argc, char *argv[]);
|
int _main(int argc, char *argv[]);
|
||||||
|
|
||||||
|
|
||||||
ssize_t readLine(int fd, void *buffer, size_t n)
|
ssize_t readLine(int fd, void *buffer, size_t n)
|
||||||
{
|
{
|
||||||
ssize_t numRead;
|
ssize_t numRead;
|
||||||
@ -158,7 +155,6 @@ int read_header(int fd, void *buffer)
|
|||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
}
|
}
|
||||||
@ -192,16 +188,15 @@ int isChar(char *string) // string字符串里有字符时返回1
|
|||||||
|
|
||||||
int extract_host(const char *header)
|
int extract_host(const char *header)
|
||||||
{
|
{
|
||||||
|
|
||||||
char *_p = strstr(header, "CONNECT"); // 在 CONNECT 方法中解析 隧道主机名称及端口号
|
char *_p = strstr(header, "CONNECT"); // 在 CONNECT 方法中解析 隧道主机名称及端口号
|
||||||
if (_p) {
|
if (_p) {
|
||||||
if (strchr(header, '[') || strchr(header, ']')) {
|
if (strchr(header, '[') || strchr(header, ']')) {
|
||||||
char *_p1 = strchr(header, '[');
|
char *_p1 = strchr(header, '[');
|
||||||
char *_p2 = strchr(_p1 + 1, ']');
|
char *_p2 = strchr(_p1 + 1, ']');
|
||||||
strncpy(remote_host, _p1 + 1, _p2 - (_p1 + 1));
|
strncpy(remote_host, _p1 + 1, (int)(_p2 - _p1) - 1);
|
||||||
|
|
||||||
char *_p3 = strchr(_p2 + 1, ' ');
|
char *_p3 = strchr(_p2 + 1, ' ');
|
||||||
char s_port[(_p3 - _p2)];
|
char s_port[270];
|
||||||
strncpy(s_port, _p2 + 2, (int)(_p3 - _p2) - 1);
|
strncpy(s_port, _p2 + 2, (int)(_p3 - _p2) - 1);
|
||||||
remote_port = atoi(s_port);
|
remote_port = atoi(s_port);
|
||||||
|
|
||||||
@ -212,31 +207,24 @@ int extract_host(const char *header)
|
|||||||
char *_p2 = strchr(_p1 + 1, ':');
|
char *_p2 = strchr(_p1 + 1, ':');
|
||||||
char *_p3 = strchr(_p1 + 1, ' ');
|
char *_p3 = strchr(_p1 + 1, ' ');
|
||||||
if (_p2) {
|
if (_p2) {
|
||||||
char s_port[270];
|
char s_port[10];
|
||||||
bzero(s_port, 270);
|
bzero(s_port, 10);
|
||||||
|
|
||||||
if ((_p3-_p2) < 0) { // 不带端口, 指向Host的':'
|
strncpy(remote_host, _p1 + 1, (int)(_p2 - _p1) - 1);
|
||||||
strncpy(remote_host, _p1 + 1, _p3 - (_p1+1));
|
|
||||||
remote_port = 80;
|
|
||||||
} else { // 带端口正常情况
|
|
||||||
strncpy(remote_host, _p1 + 1, _p2 - (_p1+1));
|
|
||||||
strncpy(s_port, _p2 + 1, (int)(_p3 - _p2) - 1);
|
strncpy(s_port, _p2 + 1, (int)(_p3 - _p2) - 1);
|
||||||
remote_port = atoi(s_port);
|
remote_port = atoi(s_port);
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
return -1;
|
strncpy(remote_host, _p1 + 1, (int)(_p3 - _p1) - 1);
|
||||||
|
remote_port = 80;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
} else { // 在非 CONNECT 方法中解析主机名称及端口号
|
} else { // 在非 CONNECT 方法中解析主机名称及端口号
|
||||||
char *p = strstr(header, "Host:");
|
char *p = strstr(header, "Host:");
|
||||||
if (p == NULL) {
|
if (!p) {
|
||||||
p = strstr(header, "host:");
|
|
||||||
}
|
|
||||||
if (p == NULL) {
|
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
char *p1 = strchr(p, '\n');
|
char *p1 = strchr(p, '\n');
|
||||||
if (!p1) {
|
if (!p1) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -252,8 +240,8 @@ int extract_host(const char *header)
|
|||||||
char *p4 = NULL;
|
char *p4 = NULL;
|
||||||
if (p3)
|
if (p3)
|
||||||
p4 = strchr(p3 + 1, ':');
|
p4 = strchr(p3 + 1, ':');
|
||||||
{ // IPV6
|
|
||||||
if (p4 != NULL) {
|
if (p4 != NULL) { // IPV6
|
||||||
char *p5 = NULL;
|
char *p5 = NULL;
|
||||||
char *p6 = NULL;
|
char *p6 = NULL;
|
||||||
p5 = strchr(header, ' ');
|
p5 = strchr(header, ' ');
|
||||||
@ -265,22 +253,21 @@ int extract_host(const char *header)
|
|||||||
strncpy(url, p5 + 1, p6 - p5 - 1);
|
strncpy(url, p5 + 1, p6 - p5 - 1);
|
||||||
url[p6 - p5 - 1] = '\0';
|
url[p6 - p5 - 1] = '\0';
|
||||||
if (strstr(url, "http") != NULL) {
|
if (strstr(url, "http") != NULL) {
|
||||||
memcpy(url, url + 7, p6 - p5 - 1 - 7); // 去除 'http://'
|
memcpy(url, url + 7, strlen(url) - 7); // 去除 'http://'
|
||||||
url[(p6 - p5 - 1) - 7] = '\0';
|
url[strlen(url) - 7] = '\0';
|
||||||
char *p7 = strchr(url, '/');
|
char *p7 = strchr(url, '/');
|
||||||
if (p7) { // 去除 uri
|
if (p7) { // 去除 uri
|
||||||
url[p7 - url] = '\0';
|
url[p7 - url] = '\0';
|
||||||
}
|
}
|
||||||
printf("url: %s\n", url);
|
printf("url: %s\n", url);
|
||||||
char *p8 = strchr(url, ']');
|
char *p8 = strchr(url, ']');
|
||||||
char *p9 = strchr(url, '\0');
|
|
||||||
if (p8) {
|
if (p8) {
|
||||||
remote_port = atoi(p8 + 2);
|
remote_port = atoi(p8 + 2);
|
||||||
strncpy(remote_host, url + 1, p8 - (url+1));
|
strncpy(remote_host, url + 1, strlen(url) - strlen(p8) - 1);
|
||||||
|
|
||||||
if ((p9-p8) == 1) { // 如果p8为 ']' 时, 长度为1
|
if (strlen(p8) < 3) { // 如果p8为 ']' 时, 长度为1
|
||||||
remote_port = 80;
|
remote_port = 80;
|
||||||
strncpy(remote_host, url + 1, p8 - (url+1));
|
strncpy(remote_host, url + 1, strlen(url) - strlen(p8) - 1);
|
||||||
}
|
}
|
||||||
} else { // 不包含'['、']'时
|
} else { // 不包含'['、']'时
|
||||||
remote_port = 80;
|
remote_port = 80;
|
||||||
@ -296,7 +283,7 @@ int extract_host(const char *header)
|
|||||||
remote_port = atoi(_p2+2);
|
remote_port = atoi(_p2+2);
|
||||||
if (strlen(_p2) < 3) {
|
if (strlen(_p2) < 3) {
|
||||||
remote_port = 80;
|
remote_port = 80;
|
||||||
strncpy(remote_host, s_host+1, _p2 - (s_host+1));
|
strncpy(remote_host, s_host+1, strlen(s_host)-strlen(_p2)-1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -305,20 +292,19 @@ int extract_host(const char *header)
|
|||||||
|
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if (p2 && p2 < p1) { // http请求头带端口
|
if (p2 && p2 < p1) {
|
||||||
int p_len = (int)(p1 - p2 - 1);
|
int p_len = (int)(p1 - p2 - 1);
|
||||||
char s_port[p_len];
|
char s_port[p_len];
|
||||||
strncpy(s_port, p2 + 1, p_len);
|
strncpy(s_port, p2 + 1, p_len);
|
||||||
s_port[p_len] = '\0';
|
s_port[p_len] = '\0';
|
||||||
remote_port = atoi(s_port);
|
remote_port = atoi(s_port);
|
||||||
|
|
||||||
int h_len = (int)(p2 - (p + 6));
|
int h_len = (int)(p2 - p - 5 - 1);
|
||||||
strncpy(remote_host, p + 5 + 1, h_len);
|
strncpy(remote_host, p + 5 + 1, h_len);
|
||||||
remote_host[h_len] = '\0';
|
remote_host[h_len] = '\0';
|
||||||
} else { // http请求头不带端口
|
} else {
|
||||||
int h_len = (int)((p1-1) - (p + 6));
|
int h_len = (int)(p1 - p - 5 - 1 - 1);
|
||||||
strncpy(remote_host, p + 5 + 1, h_len);
|
strncpy(remote_host, p + 5 + 1, h_len);
|
||||||
remote_host[h_len] = '\0';
|
remote_host[h_len] = '\0';
|
||||||
remote_port = 80;
|
remote_port = 80;
|
||||||
@ -419,32 +405,32 @@ void handle_client(int client_sock, struct sockaddr_in client_addr)
|
|||||||
int is_http_tunnel = 0;
|
int is_http_tunnel = 0;
|
||||||
if (strlen(remote_host) == 0) { /* 未指定远端主机名称从http 请求 HOST 字段中获取 */
|
if (strlen(remote_host) == 0) { /* 未指定远端主机名称从http 请求 HOST 字段中获取 */
|
||||||
#ifdef DEBUG
|
#ifdef DEBUG
|
||||||
LOG(RED " ============ handle new client ============\n" NONE);
|
LOG(" ============ handle new client ============\n");
|
||||||
LOG(RED ">>>Header:%s\n" NONE, header_buffer);
|
LOG(">>>Header:%s\n", header_buffer);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (read_header(client_sock, header_buffer) < 0) {
|
if (read_header(client_sock, header_buffer) < 0) {
|
||||||
LOG(RED "Read Http header failed\n" NONE);
|
LOG("Read Http header failed\n");
|
||||||
return;
|
return;
|
||||||
} else {
|
} else {
|
||||||
char *p = strstr(header_buffer, "CONNECT"); /* 判断是否是http 隧道请求 */
|
char *p = strstr(header_buffer, "CONNECT"); /* 判断是否是http 隧道请求 */
|
||||||
if (p) {
|
if (p) {
|
||||||
LOG(RED "receive CONNECT request\n" NONE);
|
LOG("receive CONNECT request\n");
|
||||||
is_http_tunnel = 1;
|
is_http_tunnel = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (strstr(header_buffer, "GET /AIS") > 0) {
|
if (strstr(header_buffer, "GET /AIS") > 0) {
|
||||||
LOG(RED "====== hand AIS info request ====" NONE);
|
LOG("====== hand AIS info request ====");
|
||||||
hand_mproxy_info_req(client_sock, header_buffer);
|
hand_mproxy_info_req(client_sock, header_buffer);
|
||||||
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (extract_host(header_buffer) < 0) {
|
if (extract_host(header_buffer) < 0) {
|
||||||
LOG(RED "Cannot extract host field, bad http protrotol" NONE);
|
LOG("Cannot extract host field,bad http protrotol");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
LOG(RED "Host:%s port: %d io_flag:%d\n" NONE, remote_host, remote_port, io_flag);
|
LOG("Host:%s port: %d io_flag:%d\n", remote_host, remote_port, io_flag);
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -454,7 +440,7 @@ void handle_client(int client_sock, struct sockaddr_in client_addr)
|
|||||||
//printf("%d\n", remote_port);
|
//printf("%d\n", remote_port);
|
||||||
|
|
||||||
if ((remote_sock = create_connection6(remote_host, remote_port)) < 0) {
|
if ((remote_sock = create_connection6(remote_host, remote_port)) < 0) {
|
||||||
LOG(RED "Cannot connect to host [%s:%d]\n" NONE, remote_host, remote_port);
|
LOG("Cannot connect to host [%s:%d]\n", remote_host, remote_port);
|
||||||
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -491,8 +477,8 @@ void forward_header(int destination_sock)
|
|||||||
{
|
{
|
||||||
rewrite_header();
|
rewrite_header();
|
||||||
#ifdef DEBUG
|
#ifdef DEBUG
|
||||||
LOG(RED "================ The Forward HEAD =================" NONE);
|
LOG("================ The Forward HEAD =================");
|
||||||
LOG(RED "%s\n" NONE, header_buffer);
|
LOG("%s\n", header_buffer);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
int len = strlen(header_buffer);
|
int len = strlen(header_buffer);
|
||||||
@ -643,7 +629,7 @@ int whitelist(char *client_ip, char (*whitelist_ip)[WHITELIST_IP_NUM])
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void server_loop(int signals, conf *configure)
|
void server_loop(int signal, char *conffile)
|
||||||
{
|
{
|
||||||
int i;
|
int i;
|
||||||
char ipstr[WHITELIST_IP_NUM];
|
char ipstr[WHITELIST_IP_NUM];
|
||||||
@ -654,25 +640,27 @@ void server_loop(int signals, conf *configure)
|
|||||||
socklen_t addrlen6 = sizeof(client_addr6);
|
socklen_t addrlen6 = sizeof(client_addr6);
|
||||||
char whitelist_ip[WHITELIST_IP_NUM][WHITELIST_IP_NUM] = { { 0 }, { 0 } };
|
char whitelist_ip[WHITELIST_IP_NUM][WHITELIST_IP_NUM] = { { 0 }, { 0 } };
|
||||||
|
|
||||||
//printf("%s\n", configure->IP_SEGMENT);
|
conf *configure = (struct CONF *)malloc(sizeof(struct CONF));
|
||||||
|
read_conf(conffile, configure);
|
||||||
|
printf("%s\n", configure->IP_SEGMENT);
|
||||||
|
|
||||||
split_string(configure->IP_SEGMENT, " ", whitelist_ip);
|
split_string(configure->IP_SEGMENT, " ", whitelist_ip);
|
||||||
|
|
||||||
for (i = 1; i <= WHITELIST_IP_NUM - 1; i++) {
|
for (i = 1; i <= WHITELIST_IP_NUM - 1; i++) {
|
||||||
if (*whitelist_ip[i] != '\0') ;
|
if (*whitelist_ip[i] != '\0')
|
||||||
//printf("%s\n", whitelist_ip[i]);
|
printf("%s\n", whitelist_ip[i]);
|
||||||
}
|
}
|
||||||
|
|
||||||
while (1) {
|
while (1) {
|
||||||
if (signals == 4) {
|
if (signal == 4) {
|
||||||
client_sock = accept(server_sock, (struct sockaddr *)&client_addr, &addrlen);
|
client_sock = accept(server_sock, (struct sockaddr *)&client_addr, &addrlen);
|
||||||
if (client_sock > 0) {
|
if (client_sock > 0) {
|
||||||
LOG(RED "Client Ip %s Client Port %d\n" NONE, inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, ipstr, sizeof(ipstr)), ntohs(client_addr.sin_port));
|
LOG("Client Ip %s Client Port %d\n", inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, ipstr, sizeof(ipstr)), ntohs(client_addr.sin_port));
|
||||||
strcpy(client_ip, inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, ipstr, sizeof(ipstr))); // 复制客户端IP到client_ip
|
strcpy(client_ip, inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, ipstr, sizeof(ipstr))); // 复制客户端IP到client_ip
|
||||||
|
|
||||||
if (configure->IP_RESTRICTION == 1) {
|
if (configure->IP_RESTRICTION == 1) {
|
||||||
if (whitelist(client_ip, whitelist_ip) == 0) {
|
if (whitelist(client_ip, whitelist_ip) == 0) {
|
||||||
LOG(RED "非法IPV4客户端, 拒绝连接\n" NONE);
|
LOG("非法IPV4客户端, 拒绝连接\n");
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -686,15 +674,15 @@ void server_loop(int signals, conf *configure)
|
|||||||
close(client_sock); // 关闭父进程 client_sock
|
close(client_sock); // 关闭父进程 client_sock
|
||||||
}
|
}
|
||||||
|
|
||||||
if (signals == 6) {
|
if (signal == 6) {
|
||||||
client_sock6 = accept(server_sock6, (struct sockaddr *)&client_addr6, &addrlen6);
|
client_sock6 = accept(server_sock6, (struct sockaddr *)&client_addr6, &addrlen6);
|
||||||
if (client_sock6 > 0) {
|
if (client_sock6 > 0) {
|
||||||
LOG(RED "Client Ip %s Client Port %d\n" NONE, inet_ntop(AF_INET6, &client_addr6.sin6_addr, ipstr, sizeof(ipstr)), ntohs(client_addr6.sin6_port));
|
LOG("Client Ip %s Client Port %d\n", inet_ntop(AF_INET6, &client_addr6.sin6_addr, ipstr, sizeof(ipstr)), ntohs(client_addr6.sin6_port));
|
||||||
strcpy(client_ip, inet_ntop(AF_INET6, &client_addr6.sin6_addr, ipstr, sizeof(ipstr))); // 复制客户端IP到client_ip
|
strcpy(client_ip, inet_ntop(AF_INET6, &client_addr6.sin6_addr, ipstr, sizeof(ipstr))); // 复制客户端IP到client_ip
|
||||||
|
|
||||||
if (configure->IP_RESTRICTION == 1) {
|
if (configure->IP_RESTRICTION == 1) {
|
||||||
if (whitelist(client_ip, whitelist_ip) == 0) {
|
if (whitelist(client_ip, whitelist_ip) == 0) {
|
||||||
LOG(RED "非法IPV6客户端, 拒绝连接\n" NONE);
|
LOG("非法IPV6客户端, 拒绝连接\n");
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -747,7 +735,8 @@ int create_server_socket(int port)
|
|||||||
server_addr.sin_family = AF_INET;
|
server_addr.sin_family = AF_INET;
|
||||||
server_addr.sin_port = htons(port);
|
server_addr.sin_port = htons(port);
|
||||||
server_addr.sin_addr.s_addr = INADDR_ANY;
|
server_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
if (bind(server_sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) != 0) {
|
if (bind(server_sock, (struct sockaddr *)&server_addr, sizeof(server_addr))
|
||||||
|
!= 0) {
|
||||||
return SERVER_BIND_ERROR;
|
return SERVER_BIND_ERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -795,12 +784,11 @@ int create_server_socket6(int port)
|
|||||||
return server_sock;
|
return server_sock;
|
||||||
}
|
}
|
||||||
|
|
||||||
void start_server(int SIGNAL, conf *p)
|
void start_server(int SIGNAL, char *conffile)
|
||||||
{
|
{
|
||||||
//初始化全局变量
|
//初始化全局变量
|
||||||
header_buffer = (char *)malloc(MAX_HEADER_SIZE);
|
header_buffer = (char *)malloc(MAX_HEADER_SIZE);
|
||||||
|
|
||||||
// ipv4
|
|
||||||
if (SIGNAL == 4) {
|
if (SIGNAL == 4) {
|
||||||
if ((server_sock = create_server_socket(local_port)) < 0) { // start server
|
if ((server_sock = create_server_socket(local_port)) < 0) { // start server
|
||||||
LOG("Cannot run server on %d\n", local_port);
|
LOG("Cannot run server on %d\n", local_port);
|
||||||
@ -808,7 +796,6 @@ void start_server(int SIGNAL, conf *p)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// ipv6
|
|
||||||
if (SIGNAL == 6) {
|
if (SIGNAL == 6) {
|
||||||
if ((server_sock6 = create_server_socket6(local_port)) < 0) { // start server
|
if ((server_sock6 = create_server_socket6(local_port)) < 0) { // start server
|
||||||
LOG("Cannot run server on %d\n", local_port);
|
LOG("Cannot run server on %d\n", local_port);
|
||||||
@ -816,16 +803,7 @@ void start_server(int SIGNAL, conf *p)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
server_loop(SIGNAL, p);
|
server_loop(SIGNAL, conffile);
|
||||||
}
|
|
||||||
|
|
||||||
void signal_free(int signal)
|
|
||||||
{
|
|
||||||
printf("PID:%d, signal = %d\n", getpid(), signal);
|
|
||||||
free(header_buffer);
|
|
||||||
|
|
||||||
kill(getpid(), SIGKILL);
|
|
||||||
exit(0);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int _main(int argc, char *argv[])
|
int _main(int argc, char *argv[])
|
||||||
@ -842,9 +820,8 @@ int _main(int argc, char *argv[])
|
|||||||
char *p = NULL;
|
char *p = NULL;
|
||||||
|
|
||||||
char *conffile = "./ais.conf";
|
char *conffile = "./ais.conf";
|
||||||
configure = (struct CONF *)malloc(sizeof(struct CONF));
|
conf *configure = (struct CONF *)malloc(sizeof(struct CONF));
|
||||||
read_conf(conffile, configure);
|
read_conf(conffile, configure);
|
||||||
|
|
||||||
printf("%d\n", configure->local_port);
|
printf("%d\n", configure->local_port);
|
||||||
printf("%s\n", configure->io_flag);
|
printf("%s\n", configure->io_flag);
|
||||||
printf("%d\n", configure->encode);
|
printf("%d\n", configure->encode);
|
||||||
@ -878,8 +855,6 @@ int _main(int argc, char *argv[])
|
|||||||
break;
|
break;
|
||||||
case 'c':
|
case 'c':
|
||||||
conffile = optarg;
|
conffile = optarg;
|
||||||
free_conf(configure);
|
|
||||||
read_conf(conffile, configure);
|
|
||||||
break;
|
break;
|
||||||
case 'E':
|
case 'E':
|
||||||
io_flag = W_S_ENC;
|
io_flag = W_S_ENC;
|
||||||
@ -917,32 +892,23 @@ int _main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//printf("sslEncodeCode: %d\n", sslEncodeCode);
|
printf("sslEncodeCode: %d\n", sslEncodeCode);
|
||||||
get_info(info_buf);
|
get_info(info_buf);
|
||||||
LOG(RED "%s\n" NONE, info_buf);
|
LOG("%s\n", info_buf);
|
||||||
|
|
||||||
{
|
|
||||||
signal(SIGTERM, signal_free);
|
|
||||||
signal(SIGHUP, signal_free);
|
|
||||||
signal(SIGINT, signal_free);
|
|
||||||
signal(SIGABRT, signal_free);
|
|
||||||
signal(SIGILL, signal_free);
|
|
||||||
signal(SIGSEGV, signal_free);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fork() == 0) { // IPV4 进程
|
if (fork() == 0) { // IPV4 进程
|
||||||
start_server(4, configure);
|
start_server(4, conffile);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (fork() == 0) { // IPV6 进程
|
if (fork() == 0) { // IPV6 进程
|
||||||
start_server(6, configure);
|
start_server(6, conffile);
|
||||||
}
|
}
|
||||||
|
|
||||||
free_conf(configure);
|
free_conf(configure);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(int argc, char *argv[], char **envp)
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
return _main(argc, argv);
|
return _main(argc, argv);
|
||||||
}
|
}
|
||||||
|
4
ais.conf
Executable file → Normal file
4
ais.conf
Executable file → Normal file
@ -1,7 +1,7 @@
|
|||||||
global {
|
global {
|
||||||
local_port="129";
|
local_port="127";
|
||||||
io_flag="R_C_DEC";
|
io_flag="R_C_DEC";
|
||||||
encode=128;
|
encode=128;
|
||||||
IP_RESTRICTION = 0;
|
IP_RESTRICTION = 0;
|
||||||
IP_SEGMENT= 127.0.0.1 2408:8221:9913:d040:4804:5d12:4c4a:c3a;
|
IP_SEGMENT= 127.0.0.1;
|
||||||
}
|
}
|
||||||
|
18
ais.h
Executable file → Normal file
18
ais.h
Executable file → Normal file
@ -3,22 +3,4 @@
|
|||||||
|
|
||||||
#define WHITELIST_IP_NUM 2700
|
#define WHITELIST_IP_NUM 2700
|
||||||
|
|
||||||
// 字体颜色
|
|
||||||
#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"
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
17
conf.c
Executable file → Normal file
17
conf.c
Executable file → Normal file
@ -1,16 +1,5 @@
|
|||||||
#include "conf.h"
|
#include "conf.h"
|
||||||
|
|
||||||
int8_t copy_new_mem(char *src, int src_len, char **dest)
|
|
||||||
{
|
|
||||||
*dest = (char *)malloc(src_len + 1);
|
|
||||||
if (*dest == NULL)
|
|
||||||
return 1;
|
|
||||||
memcpy(*dest, src, src_len);
|
|
||||||
*((*dest) + src_len) = '\0';
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* 在content中,设置变量(var)的首地址,值(val)的位置首地址和末地址,返回下一行指针 */
|
/* 在content中,设置变量(var)的首地址,值(val)的位置首地址和末地址,返回下一行指针 */
|
||||||
static char *set_var_val_lineEnd(char *content, char **var, char **val_begin, char **val_end)
|
static char *set_var_val_lineEnd(char *content, char **var, char **val_begin, char **val_end)
|
||||||
{
|
{
|
||||||
@ -85,8 +74,7 @@ static void parse_global_module(char *content, conf * p)
|
|||||||
|
|
||||||
if (strcasecmp(var, "io_flag") == 0) {
|
if (strcasecmp(var, "io_flag") == 0) {
|
||||||
val_begin_len = strlen(val_begin) + 1;
|
val_begin_len = strlen(val_begin) + 1;
|
||||||
//val_begin_len = val_end - val_begin;
|
p->io_flag = (char *)malloc(val_begin_len);
|
||||||
p->io_flag = (char *)malloc(val_begin_len + 1);
|
|
||||||
memset(p->io_flag, 0, val_begin_len);
|
memset(p->io_flag, 0, val_begin_len);
|
||||||
memcpy(p->io_flag, val_begin, val_begin_len);
|
memcpy(p->io_flag, val_begin, val_begin_len);
|
||||||
}
|
}
|
||||||
@ -97,8 +85,7 @@ static void parse_global_module(char *content, conf * p)
|
|||||||
|
|
||||||
if (strcasecmp(var, "IP_SEGMENT") == 0) {
|
if (strcasecmp(var, "IP_SEGMENT") == 0) {
|
||||||
val_begin_len = strlen(val_begin) + 1;
|
val_begin_len = strlen(val_begin) + 1;
|
||||||
//val_begin_len = val_end - val_begin;
|
p->IP_SEGMENT = (char *)malloc(val_begin_len);
|
||||||
p->IP_SEGMENT = (char *)malloc(val_begin_len + 1);
|
|
||||||
memset(p->IP_SEGMENT, 0, val_begin_len);
|
memset(p->IP_SEGMENT, 0, val_begin_len);
|
||||||
memcpy(p->IP_SEGMENT, val_begin, val_begin_len);
|
memcpy(p->IP_SEGMENT, val_begin, val_begin_len);
|
||||||
}
|
}
|
||||||
|
0
stript/start.sh
Executable file → Normal file
0
stript/start.sh
Executable file → Normal file
0
stript/stop.sh
Executable file → Normal file
0
stript/stop.sh
Executable file → Normal file
Loading…
Reference in New Issue
Block a user