github/wangyu-udp2raw
1
0
Fork 0
mirror of https://github.com/wangyu-/udp2raw.git synced 2025-01-19 14:29:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

everything works now

Browse Source
This commit is contained in:
wangyu 2017-07-14 21:32:37 +08:00 committed by wangyu
parent a25b9db3e8
commit 8d039a839b
2 changed files with 67 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
main.cpp
View File

@ -48,6 +48,13 @@ int epollfd ;
uint32_t session_id=0; uint32_t session_id=0;
uint32_t received_session_id=0; uint32_t received_session_id=0;
const int handshake_timeout=1000;
const int heartbeat_timeout=10000;
const int udp_timeout=2000;
//const uint16_t tcp_window=50000;
const int buf_len = 20480; const int buf_len = 20480;
const int server_mode=2; const int server_mode=2;
@ -58,6 +65,13 @@ const int RETRY_TIME=3;
const int debug_mode=0; const int debug_mode=0;
int bind_fd; int bind_fd;
const int seq_mode=2; //0 increase /1 dont increase //increase randomly,about every 10 packet
const uint64_t epoll_timer_fd_sn=1;
const uint64_t epoll_raw_recv_fd_sn=2;
uint64_t epoll_udp_fd_sn=256;
void handler(int num) { void handler(int num) {
int status; int status;
int pid; int pid;
@ -360,7 +374,9 @@ int send_raw(packet_info_t &info,char * payload,int payloadlen)
tcph->ack=info.ack; tcph->ack=info.ack;
tcph->urg=0; tcph->urg=0;
tcph->window = htons((uint16_t)129600); tcph->window = htons((uint16_t)(40000+random()%20000));
tcph->check = 0; //leave checksum 0 now, filled later by pseudo header tcph->check = 0; //leave checksum 0 now, filled later by pseudo header
tcph->urg_ptr = 0; tcph->urg_ptr = 0;
@ -387,7 +403,20 @@ int send_raw(packet_info_t &info,char * payload,int payloadlen)
int ret = sendto(raw_send_fd, raw_send_buf, iph->tot_len , 0, (struct sockaddr *) &sin, sizeof (sin)); int ret = sendto(raw_send_fd, raw_send_buf, iph->tot_len , 0, (struct sockaddr *) &sin, sizeof (sin));
if(g_packet_info.syn==0&&g_packet_info.ack==1&&payloadlen!=0) if(g_packet_info.syn==0&&g_packet_info.ack==1&&payloadlen!=0)
{ {
g_packet_info.seq+=payloadlen; if(seq_mode==0)
{
g_packet_info.seq+=payloadlen;
}
else if(seq_mode==1)
{
}
else if(seq_mode==3)
{
if(random()% 20==5 )
g_packet_info.seq+=payloadlen;
}
} }
if(debug_mode) printf("<ret:%d>\n",ret); if(debug_mode) printf("<ret:%d>\n",ret);
if(ret<0) if(ret<0)
@ -514,7 +543,7 @@ int fake_tcp_keep_connection_client() //for client
g_packet_info.ack_seq=get_true_random_number(); g_packet_info.ack_seq=get_true_random_number();
send_sync(/*sync*/);//send sync send_sync(/*sync*/);//send sync
} }
if(client_current_state==client_syn_sent &&get_current_time()-last_state_time>1000ll) if(client_current_state==client_syn_sent &&get_current_time()-last_state_time>handshake_timeout)
{ {
if(retry_counter==0) if(retry_counter==0)
{ {
@ -530,7 +559,7 @@ int fake_tcp_keep_connection_client() //for client
last_state_time=get_current_time(); last_state_time=get_current_time();
} }
} }
if(client_current_state==client_ack_sent &&get_current_time()-last_state_time>1000) if(client_current_state==client_ack_sent &&get_current_time()-last_state_time>handshake_timeout)
{ {
printf("!!!!!\n"); printf("!!!!!\n");
fflush(stdout); fflush(stdout);
@ -553,7 +582,7 @@ int fake_tcp_keep_connection_client() //for client
if(client_current_state==client_ready) if(client_current_state==client_ready)
{ {
if(debug_mode)printf("time %lld %lld\n",get_current_time(),last_state_time); if(debug_mode)printf("time %lld %lld\n",get_current_time(),last_state_time);
if(get_current_time()-last_hb_recv_time>5*1000ll) if(get_current_time()-last_hb_recv_time>heartbeat_timeout)
{ {
client_current_state=client_nothing; client_current_state=client_nothing;
printf("state back to nothing\n"); printf("state back to nothing\n");
@ -586,7 +615,7 @@ int fake_tcp_keep_connection_server()
{ {
return 0; return 0;
} }
if(server_current_state==server_syn_ack_sent &&get_current_time()-last_state_time>1000ll ) if(server_current_state==server_syn_ack_sent &&get_current_time()-last_state_time>handshake_timeout )
{ {
if(retry_counter==0) if(retry_counter==0)
{ {
@ -605,7 +634,9 @@ int fake_tcp_keep_connection_server()
} }
if(server_current_state==server_heartbeat_sent||server_current_state==server_ready) if(server_current_state==server_heartbeat_sent||server_current_state==server_ready)
{ {
if(get_current_time()-last_hb_recv_time>5*1000ll) if( (server_current_state==server_heartbeat_sent&&get_current_time()-last_hb_recv_time>handshake_timeout )
|| ( server_current_state==server_ready&&get_current_time()-last_hb_recv_time>heartbeat_timeout )
)
{ {
printf("%lld %lld",get_current_time(),last_state_time); printf("%lld %lld",get_current_time(),last_state_time);
server_current_state=server_nothing; server_current_state=server_nothing;
@ -654,7 +685,7 @@ int set_timer(int epollfd,int &timer_fd)
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = timer_fd; ev.data.u64 = epoll_timer_fd_sn;
epoll_ctl(epollfd, EPOLL_CTL_ADD, timer_fd, &ev); epoll_ctl(epollfd, EPOLL_CTL_ADD, timer_fd, &ev);
if (ret < 0) { if (ret < 0) {
@ -746,7 +777,7 @@ int client_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
printf("client session id mismatch%x %x,ignore\n",tmp_session_id,session_id); printf("client session id mismatch%x %x,ignore\n",tmp_session_id,session_id);
return 0; return 0;
} }
last_hb_recv_time=get_current_time();
int ret=sendto(udp_fd,data+1+sizeof(session_id)*2,data_len -(1+sizeof(session_id)*2),0,(struct sockaddr *)&udp_old_addr_in,sizeof(udp_old_addr_in)); int ret=sendto(udp_fd,data+1+sizeof(session_id)*2,data_len -(1+sizeof(session_id)*2),0,(struct sockaddr *)&udp_old_addr_in,sizeof(udp_old_addr_in));
if(ret<0)perror("ret<0"); if(ret<0)perror("ret<0");
printf("%d byte sent\n",ret); printf("%d byte sent\n",ret);
@ -759,6 +790,7 @@ using namespace std;
set<uint32_t> died_session_id; set<uint32_t> died_session_id;
int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len) int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
{ {
if(server_current_state==server_nothing) if(server_current_state==server_nothing)
@ -819,10 +851,8 @@ int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
if(udp_fd!=-1) if(udp_fd!=-1)
{ {
ev.events = EPOLLIN; int ret = epoll_ctl(epollfd, EPOLL_CTL_DEL, udp_fd, 0);
ev.data.fd = udp_fd; close(udp_fd);
int ret = epoll_ctl(epollfd, EPOLL_CTL_DEL, udp_fd, &ev);
//close(udp_fd);
udp_fd=-1; udp_fd=-1;
} }
received_session_id=session_id; received_session_id=session_id;
@ -902,7 +932,8 @@ int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
struct epoll_event ev; struct epoll_event ev;
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = udp_fd; epoll_udp_fd_sn+=256;
ev.data.u64 = epoll_udp_fd_sn;
ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev); ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev);
@ -913,10 +944,8 @@ int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
if(old_fd!=-1) if(old_fd!=-1)
{ {
ev.events = EPOLLIN; epoll_ctl(epollfd, EPOLL_CTL_DEL, old_fd, 0);
ev.data.fd = old_fd; close(old_fd);
ret = epoll_ctl(epollfd, EPOLL_CTL_DEL, old_fd, &ev);
//close(old_fd);
} }
} }
@ -927,6 +956,7 @@ int server_raw_recv(iphdr * iph,tcphdr *tcph,char * data,int data_len)
printf("created new udp_fd"); printf("created new udp_fd");
} }
printf("received a data from fake tcp,len:%d\n",data_len); printf("received a data from fake tcp,len:%d\n",data_len);
last_hb_recv_time=get_current_time();
int ret=send(udp_fd,data+1+sizeof(session_id)*2,data_len -(1+sizeof(session_id)*2),0); int ret=send(udp_fd,data+1+sizeof(session_id)*2,data_len -(1+sizeof(session_id)*2),0);
printf("%d byte sent\n",ret); printf("%d byte sent\n",ret);
} }
@ -1034,7 +1064,8 @@ int on_raw_recv()
if(data_len>0&&tcph->syn==0&&tcph->ack==1) if(data_len>0&&tcph->syn==0&&tcph->ack==1)
{ {
g_packet_info.ack_seq=ntohl(tcph->seq)+(uint32_t)data_len; //if(seq_increse)
g_packet_info.ack_seq=ntohl(tcph->seq)+(uint32_t)data_len;
} }
@ -1108,14 +1139,14 @@ int client()
} }
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = udp_fd; ev.data.u64 = epoll_udp_fd_sn;
ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev); ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev);
if (ret!=0) { if (ret!=0) {
printf("add udp_listen_fd error\n"); printf("add udp_listen_fd error\n");
exit(-1); exit(-1);
} }
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = raw_recv_fd; ev.data.u64 = epoll_raw_recv_fd_sn;
ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev); ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev);
if (ret!= 0) { if (ret!= 0) {
@ -1141,7 +1172,7 @@ int client()
} }
int n; int n;
for (n = 0; n < nfds; ++n) { for (n = 0; n < nfds; ++n) {
if (events[n].data.fd == raw_recv_fd) if (events[n].data.u64 == epoll_raw_recv_fd_sn)
{ {
on_raw_recv(); on_raw_recv();
/*if(is_sync_ack) /*if(is_sync_ack)
@ -1157,7 +1188,7 @@ int client()
sendto(); sendto();
}*/ }*/
} }
if(events[n].data.fd ==timer_fd) if(events[n].data.u64 ==epoll_timer_fd_sn)
{ {
//printf("timer!\n"); //printf("timer!\n");
//fflush(stdout); //fflush(stdout);
@ -1165,7 +1196,7 @@ int client()
read(timer_fd, &value, 8); read(timer_fd, &value, 8);
fake_tcp_keep_connection_client(); fake_tcp_keep_connection_client();
} }
if (events[n].data.fd == udp_fd) if (events[n].data.u64 == epoll_udp_fd_sn)
{ {
socklen_t recv_len; socklen_t recv_len;
@ -1187,7 +1218,7 @@ int client()
else if(udp_new_addr_in.sin_addr.s_addr!=udp_old_addr_in.sin_addr.s_addr else if(udp_new_addr_in.sin_addr.s_addr!=udp_old_addr_in.sin_addr.s_addr
||udp_new_addr_in.sin_port!=udp_old_addr_in.sin_port) ||udp_new_addr_in.sin_port!=udp_old_addr_in.sin_port)
{ {
if(get_current_time()- last_udp_recv_time <3*1000) if(get_current_time()- last_udp_recv_time <udp_timeout)
{ {
printf("new <ip,port> connected in,ignored,bc last connection is still active\n"); printf("new <ip,port> connected in,ignored,bc last connection is still active\n");
continue; continue;
@ -1256,7 +1287,9 @@ int server()
} }
ev.events = EPOLLIN; ev.events = EPOLLIN;
ev.data.fd = raw_recv_fd; ev.data.u64 = epoll_raw_recv_fd_sn;
ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev); ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev);
if (ret!= 0) { if (ret!= 0) {
@ -1276,7 +1309,7 @@ int server()
const int MTU=1440; const int MTU=1440;
for (n = 0; n < nfds; ++n) for (n = 0; n < nfds; ++n)
{ {
if (events[n].data.fd == udp_fd) if (events[n].data.u64 == epoll_udp_fd_sn)
{ {
int recv_len=recv(udp_fd,buf,buf_len,0); int recv_len=recv(udp_fd,buf,buf_len,0);
printf("received a packet from udp_fd,len:%d\n",recv_len); printf("received a packet from udp_fd,len:%d\n",recv_len);
@ -1290,13 +1323,13 @@ int server()
send_data(g_packet_info,buf,recv_len,received_session_id,session_id); send_data(g_packet_info,buf,recv_len,received_session_id,session_id);
} }
//printf("%d %d %d %d\n",timer_fd,raw_recv_fd,raw_send_fd,n); //printf("%d %d %d %d\n",timer_fd,raw_recv_fd,raw_send_fd,n);
if (events[n].data.fd == timer_fd) if (events[n].data.u64 == epoll_timer_fd_sn)
{ {
uint64_t value; uint64_t value;
read(timer_fd, &value, 8); read(timer_fd, &value, 8);
fake_tcp_keep_connection_server(); fake_tcp_keep_connection_server();
} }
if (events[n].data.fd == raw_recv_fd) if (events[n].data.u64 == epoll_raw_recv_fd_sn)
{ {
on_raw_recv(); on_raw_recv();
} }

5
makefile Normal file
View File

@ -0,0 +1,5 @@
ccmips=mips-openwrt-linux-g++
all:
g++ main.cpp -o raw -static -lrt -ggdb
${ccmips} main.cpp -o rawmips -static -lgcc_eh -lrt