github/wangyu-udp2raw
1
0
Fork 0
mirror of https://github.com/wangyu-/udp2raw.git synced 2025-10-30 01:35:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

use tcp for detection

Browse Source
This commit is contained in:
wangyu-
2018-06-15 10:06:34 -05:00
parent 63950e2b22
commit ba6d5e8895
1 changed files with 40 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
main.cpp
View File

@@ -16,6 +16,10 @@ int on_epoll_recv_event=0; //TODO, just a flag to help detect epoll infinite sh
u32_t detect_interval=1500; u32_t detect_interval=1500;
u64_t laste_detect_time=0; u64_t laste_detect_time=0;
int use_udp_for_detection=0;
int use_tcp_for_detection=1;
int client_on_timer(conn_info_t &conn_info) //for client. called when a timer is ready in epoll int client_on_timer(conn_info_t &conn_info) //for client. called when a timer is ready in epoll
{ {
packet_info_t &send_info=conn_info.raw_info.send_info; packet_info_t &send_info=conn_info.raw_info.send_info;
@@ -46,14 +50,17 @@ int client_on_timer(conn_info_t &conn_info) //for client. called when a timer is
socklen_t slen = sizeof(sockaddr_in); socklen_t slen = sizeof(sockaddr_in);
//memset(&remote_addr_in, 0, sizeof(remote_addr_in)); //memset(&remote_addr_in, 0, sizeof(remote_addr_in));
int port=get_true_random_number()%65534+1;
remote_addr_in.sin_family = AF_INET; remote_addr_in.sin_family = AF_INET;
remote_addr_in.sin_port = htons(remote_port); remote_addr_in.sin_port = htons(port);
remote_addr_in.sin_addr.s_addr = remote_ip_uint32; remote_addr_in.sin_addr.s_addr = remote_ip_uint32;
if(use_udp_for_detection)
{
int new_udp_fd=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); int new_udp_fd=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if(new_udp_fd<0) if(new_udp_fd<0)
{ {
mylog(log_warn,"create udp_fd error\n"); mylog(log_warn,"create new_udp_fd error\n");
return -1; return -1;
} }
setnonblocking(new_udp_fd); setnonblocking(new_udp_fd);
@@ -64,8 +71,28 @@ int client_on_timer(conn_info_t &conn_info) //for client. called when a timer is
{ {
mylog(log_warn,"sendto() failed\n"); mylog(log_warn,"sendto() failed\n");
} }
close(new_udp_fd); close(new_udp_fd);
}
if(use_tcp_for_detection)
{
static int last_tcp_fd=-1;
int new_tcp_fd=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if(new_tcp_fd<0)
{
mylog(log_warn,"create new_tcp_fd error\n");
return -1;
}
setnonblocking(new_tcp_fd);
connect(new_tcp_fd,(struct sockaddr *)&remote_addr_in,sizeof(remote_addr_in));
if(last_tcp_fd!=-1)
close(last_tcp_fd);
last_tcp_fd=new_tcp_fd;
//close(new_tcp_fd);
}
mylog(log_info,"waiting for a use-able packet to be captured\n"); mylog(log_info,"waiting for a use-able packet to be captured\n");