main.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843
  1. #include "common.h"
  2. #include "network.h"
  3. #include "connection.h"
  4. #include "misc.h"
  5. #include "log.h"
  6. #include "lib/md5.h"
  7. #include "encrypt.h"
  8. #include "fd_manager.h"
  9. char hb_buf[buf_len];
  10. int on_epoll_recv_event=0; //TODO, just a flag to help detect epoll infinite shoot
  11. int client_on_timer(conn_info_t &conn_info) //for client. called when a timer is ready in epoll
  12. {
  13. //keep_iptables_rule();
  14. packet_info_t &send_info=conn_info.raw_info.send_info;
  15. packet_info_t &recv_info=conn_info.raw_info.recv_info;
  16. raw_info_t &raw_info=conn_info.raw_info;
  17. conn_info.blob->conv_manager.clear_inactive();
  18. mylog(log_trace,"timer!\n");
  19. mylog(log_trace,"roller my %d,oppsite %d,%lld\n",int(conn_info.my_roller),int(conn_info.oppsite_roller),conn_info.last_oppsite_roller_time);
  20. mylog(log_trace,"<client_on_timer,send_info.ts_ack= %u>\n",send_info.ts_ack);
  21. if(raw_info.disabled)
  22. {
  23. conn_info.state.client_current_state=client_idle;
  24. conn_info.my_id=get_true_random_number_nz();
  25. mylog(log_info,"state back to client_idle\n");
  26. }
  27. if(conn_info.state.client_current_state==client_idle)
  28. {
  29. raw_info.rst_received=0;
  30. raw_info.disabled=0;
  31. fail_time_counter++;
  32. if(max_fail_time>0&&fail_time_counter>max_fail_time)
  33. {
  34. mylog(log_fatal,"max_fail_time exceed\n");
  35. myexit(-1);
  36. }
  37. conn_info.blob->anti_replay.re_init();
  38. conn_info.my_id = get_true_random_number_nz(); ///todo no need to do this everytime
  39. u32_t new_ip=0;
  40. if(!force_source_ip&&get_src_adress(new_ip,remote_ip_uint32,remote_port)==0)
  41. {
  42. if(new_ip!=source_ip_uint32)
  43. {
  44. mylog(log_info,"source ip changed from %s to ",my_ntoa(source_ip_uint32));
  45. log_bare(log_info,"%s\n",my_ntoa(new_ip));
  46. source_ip_uint32=new_ip;
  47. send_info.src_ip=new_ip;
  48. }
  49. }
  50. if (source_port == 0)
  51. {
  52. send_info.src_port = client_bind_to_a_new_port(bind_fd,local_ip_uint32);
  53. }
  54. else
  55. {
  56. send_info.src_port = source_port;
  57. }
  58. if (raw_mode == mode_icmp)
  59. {
  60. send_info.dst_port = send_info.src_port;
  61. }
  62. mylog(log_info, "using port %d\n", send_info.src_port);
  63. init_filter(send_info.src_port);
  64. if(raw_mode==mode_icmp||raw_mode==mode_udp)
  65. {
  66. conn_info.state.client_current_state=client_handshake1;
  67. mylog(log_info,"state changed from client_idle to client_pre_handshake\n");
  68. }
  69. if(raw_mode==mode_faketcp)
  70. {
  71. conn_info.state.client_current_state=client_tcp_handshake;
  72. mylog(log_info,"state changed from client_idle to client_tcp_handshake\n");
  73. }
  74. conn_info.last_state_time=get_current_time();
  75. conn_info.last_hb_sent_time=0;
  76. //dont return;
  77. }
  78. if(conn_info.state.client_current_state==client_tcp_handshake) //send and resend syn
  79. {
  80. assert(raw_mode==mode_faketcp);
  81. if (get_current_time() - conn_info.last_state_time > client_handshake_timeout)
  82. {
  83. conn_info.state.client_current_state = client_idle;
  84. mylog(log_info, "state back to client_idle from client_tcp_handshake\n");
  85. return 0;
  86. }
  87. else if (get_current_time() - conn_info.last_hb_sent_time > client_retry_interval)
  88. {
  89. if (raw_mode == mode_faketcp)
  90. {
  91. if (conn_info.last_hb_sent_time == 0)
  92. {
  93. send_info.psh = 0;
  94. send_info.syn = 1;
  95. send_info.ack = 0;
  96. send_info.ts_ack =0;
  97. send_info.seq=get_true_random_number();
  98. send_info.ack_seq=get_true_random_number();
  99. }
  100. }
  101. send_raw0(raw_info, 0, 0);
  102. conn_info.last_hb_sent_time = get_current_time();
  103. mylog(log_info, "(re)sent tcp syn\n");
  104. return 0;
  105. }
  106. else
  107. {
  108. return 0;
  109. }
  110. return 0;
  111. }
  112. else if(conn_info.state.client_current_state==client_handshake1)//send and resend handshake1
  113. {
  114. if(get_current_time()-conn_info.last_state_time>client_handshake_timeout)
  115. {
  116. conn_info.state.client_current_state=client_idle;
  117. mylog(log_info,"state back to client_idle from client_handshake1\n");
  118. return 0;
  119. }
  120. else if(get_current_time()-conn_info.last_hb_sent_time>client_retry_interval)
  121. {
  122. if(raw_mode==mode_faketcp)
  123. {
  124. if(conn_info.last_hb_sent_time==0)
  125. {
  126. send_info.seq++;
  127. send_info.ack_seq=recv_info.seq+1;
  128. send_info.ts_ack=recv_info.ts;
  129. raw_info.reserved_send_seq=send_info.seq;
  130. }
  131. send_info.seq=raw_info.reserved_send_seq;
  132. send_info.psh = 0;
  133. send_info.syn = 0;
  134. send_info.ack = 1;
  135. send_raw0(raw_info, 0, 0);
  136. send_handshake(raw_info,conn_info.my_id,0,const_id);
  137. send_info.seq+=raw_info.send_info.data_len;
  138. }
  139. else
  140. {
  141. send_handshake(raw_info,conn_info.my_id,0,const_id);
  142. if(raw_mode==mode_icmp)
  143. send_info.icmp_seq++;
  144. }
  145. conn_info.last_hb_sent_time=get_current_time();
  146. mylog(log_info,"(re)sent handshake1\n");
  147. return 0;
  148. }
  149. else
  150. {
  151. return 0;
  152. }
  153. return 0;
  154. }
  155. else if(conn_info.state.client_current_state==client_handshake2)
  156. {
  157. if(get_current_time()-conn_info.last_state_time>client_handshake_timeout)
  158. {
  159. conn_info.state.client_current_state=client_idle;
  160. mylog(log_info,"state back to client_idle from client_handshake2\n");
  161. return 0;
  162. }
  163. else if(get_current_time()-conn_info.last_hb_sent_time>client_retry_interval)
  164. {
  165. if(raw_mode==mode_faketcp)
  166. {
  167. if(conn_info.last_hb_sent_time==0)
  168. {
  169. send_info.ack_seq=recv_info.seq+raw_info.recv_info.data_len;
  170. send_info.ts_ack=recv_info.ts;
  171. raw_info.reserved_send_seq=send_info.seq;
  172. }
  173. send_info.seq=raw_info.reserved_send_seq;
  174. send_handshake(raw_info,conn_info.my_id,conn_info.oppsite_id,const_id);
  175. send_info.seq+=raw_info.send_info.data_len;
  176. }
  177. else
  178. {
  179. send_handshake(raw_info,conn_info.my_id,conn_info.oppsite_id,const_id);
  180. if(raw_mode==mode_icmp)
  181. send_info.icmp_seq++;
  182. }
  183. conn_info.last_hb_sent_time=get_current_time();
  184. mylog(log_info,"(re)sent handshake2\n");
  185. return 0;
  186. }
  187. else
  188. {
  189. return 0;
  190. }
  191. return 0;
  192. }
  193. else if(conn_info.state.client_current_state==client_ready)
  194. {
  195. fail_time_counter=0;
  196. mylog(log_trace,"time %llu,%llu\n",get_current_time(),conn_info.last_state_time);
  197. if(get_current_time()-conn_info.last_hb_recv_time>client_conn_timeout)
  198. {
  199. conn_info.state.client_current_state=client_idle;
  200. conn_info.my_id=get_true_random_number_nz();
  201. mylog(log_info,"state back to client_idle from client_ready bc of server-->client direction timeout\n");
  202. return 0;
  203. }
  204. if(get_current_time()- conn_info.last_oppsite_roller_time>client_conn_uplink_timeout)
  205. {
  206. conn_info.state.client_current_state=client_idle;
  207. conn_info.my_id=get_true_random_number_nz();
  208. mylog(log_info,"state back to client_idle from client_ready bc of client-->server direction timeout\n");
  209. }
  210. if(get_current_time()-conn_info.last_hb_sent_time<heartbeat_interval)
  211. {
  212. return 0;
  213. }
  214. mylog(log_debug,"heartbeat sent <%x,%x>\n",conn_info.oppsite_id,conn_info.my_id);
  215. if(hb_mode==0)
  216. send_safer(conn_info,'h',hb_buf,0);/////////////send
  217. else
  218. send_safer(conn_info,'h',hb_buf,hb_len);
  219. conn_info.last_hb_sent_time=get_current_time();
  220. return 0;
  221. }
  222. else
  223. {
  224. mylog(log_fatal,"unknown state,this shouldnt happen.\n");
  225. myexit(-1);
  226. }
  227. return 0;
  228. }
  229. int client_on_raw_recv(conn_info_t &conn_info) //called when raw fd received a packet.
  230. {
  231. char* data;int data_len;
  232. packet_info_t &send_info=conn_info.raw_info.send_info;
  233. packet_info_t &recv_info=conn_info.raw_info.recv_info;
  234. raw_info_t &raw_info=conn_info.raw_info;
  235. mylog(log_trace,"<client_on_raw_recv,send_info.ts_ack= %u>\n",send_info.ts_ack);
  236. if(conn_info.state.client_current_state==client_idle )
  237. {
  238. recv(raw_recv_fd, 0,0, 0 );
  239. }
  240. else if(conn_info.state.client_current_state==client_tcp_handshake)//received syn ack
  241. {
  242. assert(raw_mode==mode_faketcp);
  243. if(recv_raw0(raw_info,data,data_len)<0)
  244. {
  245. return -1;
  246. }
  247. if(recv_info.src_ip!=send_info.dst_ip||recv_info.src_port!=send_info.dst_port)
  248. {
  249. mylog(log_debug,"unexpected adress %x %x %d %d\n",recv_info.src_ip,send_info.dst_ip,recv_info.src_port,send_info.dst_port);
  250. return -1;
  251. }
  252. if(data_len==0&&raw_info.recv_info.syn==1&&raw_info.recv_info.ack==1)
  253. {
  254. if(recv_info.ack_seq!=send_info.seq+1)
  255. {
  256. mylog(log_debug,"seq ack_seq mis match\n");
  257. return -1;
  258. }
  259. conn_info.state.client_current_state = client_handshake1;
  260. mylog(log_info,"state changed from client_tcp_handshake to client_handshake1\n");
  261. conn_info.last_state_time = get_current_time();
  262. conn_info.last_hb_sent_time=0;
  263. client_on_timer(conn_info);
  264. return 0;
  265. }
  266. else
  267. {
  268. mylog(log_debug,"unexpected packet type,expected:syn ack\n");
  269. return -1;
  270. }
  271. }
  272. else if(conn_info.state.client_current_state==client_handshake1)//recevied respond of handshake1
  273. {
  274. if(recv_bare(raw_info,data,data_len)!=0)
  275. {
  276. mylog(log_debug,"recv_bare failed!\n");
  277. return -1;
  278. }
  279. if(recv_info.src_ip!=send_info.dst_ip||recv_info.src_port!=send_info.dst_port)
  280. {
  281. mylog(log_debug,"unexpected adress %x %x %d %d\n",recv_info.src_ip,send_info.dst_ip,recv_info.src_port,send_info.dst_port);
  282. return -1;
  283. }
  284. if(data_len<int( 3*sizeof(id_t)))
  285. {
  286. mylog(log_debug,"too short to be a handshake\n");
  287. return -1;
  288. }
  289. //id_t tmp_oppsite_id= ntohl(* ((u32_t *)&data[0]));
  290. id_t tmp_oppsite_id;
  291. memcpy(&tmp_oppsite_id,&data[0],sizeof(tmp_oppsite_id));
  292. tmp_oppsite_id=ntohl(tmp_oppsite_id);
  293. //id_t tmp_my_id=ntohl(* ((u32_t *)&data[sizeof(id_t)]));
  294. id_t tmp_my_id;
  295. memcpy(&tmp_my_id,&data[sizeof(id_t)],sizeof(tmp_my_id));
  296. tmp_my_id=ntohl(tmp_my_id);
  297. //id_t tmp_oppsite_const_id=ntohl(* ((u32_t *)&data[sizeof(id_t)*2]));
  298. id_t tmp_oppsite_const_id;
  299. memcpy(&tmp_oppsite_const_id,&data[sizeof(id_t)*2],sizeof(tmp_oppsite_const_id));
  300. tmp_oppsite_const_id=ntohl(tmp_oppsite_const_id);
  301. if(tmp_my_id!=conn_info.my_id)
  302. {
  303. mylog(log_debug,"tmp_my_id doesnt match\n");
  304. return -1;
  305. }
  306. if(raw_mode==mode_faketcp)
  307. {
  308. if(recv_info.ack_seq!=send_info.seq)
  309. {
  310. mylog(log_debug,"seq ack_seq mis match\n");
  311. return -1;
  312. }
  313. if(recv_info.seq!=send_info.ack_seq)
  314. {
  315. mylog(log_debug,"seq ack_seq mis match\n");
  316. return -1;
  317. }
  318. }
  319. conn_info.oppsite_id=tmp_oppsite_id;
  320. mylog(log_info,"changed state from to client_handshake1 to client_handshake2,my_id is %x,oppsite id is %x\n",conn_info.my_id,conn_info.oppsite_id);
  321. //send_handshake(raw_info,conn_info.my_id,conn_info.oppsite_id,const_id); //////////////send
  322. conn_info.state.client_current_state = client_handshake2;
  323. conn_info.last_state_time = get_current_time();
  324. conn_info.last_hb_sent_time=0;
  325. client_on_timer(conn_info);
  326. return 0;
  327. }
  328. else if(conn_info.state.client_current_state==client_handshake2||conn_info.state.client_current_state==client_ready)//received heartbeat or data
  329. {
  330. char type;
  331. if(recv_safer(conn_info,type,data,data_len)!=0)
  332. {
  333. mylog(log_debug,"recv_safer failed!\n");
  334. return -1;
  335. }
  336. if(recv_info.src_ip!=send_info.dst_ip||recv_info.src_port!=send_info.dst_port)
  337. {
  338. mylog(log_warn,"unexpected adress %x %x %d %d,this shouldnt happen.\n",recv_info.src_ip,send_info.dst_ip,recv_info.src_port,send_info.dst_port);
  339. return -1;
  340. }
  341. if(conn_info.state.client_current_state==client_handshake2)
  342. {
  343. mylog(log_info,"changed state from to client_handshake2 to client_ready\n");
  344. conn_info.state.client_current_state=client_ready;
  345. conn_info.last_hb_sent_time=0;
  346. conn_info.last_hb_recv_time=get_current_time();
  347. conn_info.last_oppsite_roller_time=conn_info.last_hb_recv_time;
  348. client_on_timer(conn_info);
  349. }
  350. if(data_len>=0&&type=='h')
  351. {
  352. mylog(log_debug,"[hb]heart beat received,oppsite_roller=%d\n",int(conn_info.oppsite_roller));
  353. conn_info.last_hb_recv_time=get_current_time();
  354. return 0;
  355. }
  356. else if(data_len>= int( sizeof(u32_t))&&type=='d')
  357. {
  358. mylog(log_trace,"received a data from fake tcp,len:%d\n",data_len);
  359. if(hb_mode==0)
  360. conn_info.last_hb_recv_time=get_current_time();
  361. //u32_t tmp_conv_id= ntohl(* ((u32_t *)&data[0]));
  362. u32_t tmp_conv_id;
  363. memcpy(&tmp_conv_id,&data[0],sizeof(tmp_conv_id));
  364. tmp_conv_id=ntohl(tmp_conv_id);
  365. if(!conn_info.blob->conv_manager.is_conv_used(tmp_conv_id))
  366. {
  367. mylog(log_info,"unknow conv %d,ignore\n",tmp_conv_id);
  368. return 0;
  369. }
  370. conn_info.blob->conv_manager.update_active_time(tmp_conv_id);
  371. u64_t u64=conn_info.blob->conv_manager.find_u64_by_conv(tmp_conv_id);
  372. sockaddr_in tmp_sockaddr={0};
  373. tmp_sockaddr.sin_family = AF_INET;
  374. tmp_sockaddr.sin_addr.s_addr=(u64>>32u);
  375. tmp_sockaddr.sin_port= htons(uint16_t((u64<<32u)>>32u));
  376. int ret=sendto(udp_fd,data+sizeof(u32_t),data_len -(sizeof(u32_t)),0,(struct sockaddr *)&tmp_sockaddr,sizeof(tmp_sockaddr));
  377. if(ret<0)
  378. {
  379. mylog(log_warn,"sento returned %d\n",ret);
  380. //perror("ret<0");
  381. }
  382. mylog(log_trace,"%s :%d\n",inet_ntoa(tmp_sockaddr.sin_addr),ntohs(tmp_sockaddr.sin_port));
  383. mylog(log_trace,"%d byte sent\n",ret);
  384. }
  385. else
  386. {
  387. mylog(log_warn,"unknown packet,this shouldnt happen.\n");
  388. return -1;
  389. }
  390. return 0;
  391. }
  392. else
  393. {
  394. mylog(log_fatal,"unknown state,this shouldnt happen.\n");
  395. myexit(-1);
  396. }
  397. return 0;
  398. }
  399. int client_event_loop()
  400. {
  401. char buf[buf_len];
  402. conn_info_t conn_info;
  403. conn_info.my_id=get_true_random_number_nz();
  404. conn_info.prepare();
  405. packet_info_t &send_info=conn_info.raw_info.send_info;
  406. packet_info_t &recv_info=conn_info.raw_info.recv_info;
  407. if(source_ip_uint32==0)
  408. {
  409. mylog(log_info,"get_src_adress called\n");
  410. if(retry_on_error==0)
  411. {
  412. if(get_src_adress(source_ip_uint32,remote_ip_uint32,remote_port)!=0)
  413. {
  414. mylog(log_fatal,"the trick to auto get source ip failed, maybe you dont have internet access\n");
  415. myexit(-1);
  416. }
  417. }
  418. else
  419. {
  420. int ok=0;
  421. while(!ok)
  422. {
  423. if(get_src_adress(source_ip_uint32,remote_ip_uint32,remote_port)!=0)
  424. {
  425. mylog(log_warn,"the trick to auto get source ip failed, maybe you dont have internet access, retry in %d seconds\n",retry_on_error_interval);
  426. sleep(retry_on_error_interval);
  427. }
  428. else
  429. {
  430. ok=1;
  431. }
  432. }
  433. }
  434. }
  435. in_addr tmp;
  436. tmp.s_addr=source_ip_uint32;
  437. mylog(log_info,"source ip = %s\n",inet_ntoa(tmp));
  438. //printf("done\n");
  439. if(try_to_list_and_bind(bind_fd,local_ip_uint32,source_port)!=0)
  440. {
  441. mylog(log_fatal,"bind to source_port:%d fail\n ",source_port);
  442. myexit(-1);
  443. }
  444. send_info.src_port=source_port;
  445. send_info.src_ip = source_ip_uint32;
  446. int i, j, k;int ret;
  447. //init_filter(source_port);
  448. send_info.dst_ip=remote_ip_uint32;
  449. send_info.dst_port=remote_port;
  450. //g_packet_info.src_ip=source_address_uint32;
  451. //g_packet_info.src_port=source_port;
  452. udp_fd=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  453. set_buf_size(udp_fd,socket_buf_size,force_socket_buf);
  454. int yes = 1;
  455. //setsockopt(udp_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
  456. struct sockaddr_in local_me={0};
  457. socklen_t slen = sizeof(sockaddr_in);
  458. //memset(&local_me, 0, sizeof(local_me));
  459. local_me.sin_family = AF_INET;
  460. local_me.sin_port = htons(local_port);
  461. local_me.sin_addr.s_addr = local_ip_uint32;
  462. if (bind(udp_fd, (struct sockaddr*) &local_me, slen) == -1) {
  463. mylog(log_fatal,"socket bind error\n");
  464. //perror("socket bind error");
  465. myexit(1);
  466. }
  467. setnonblocking(udp_fd);
  468. epollfd = epoll_create1(0);
  469. const int max_events = 4096;
  470. struct epoll_event ev, events[max_events];
  471. if (epollfd < 0) {
  472. mylog(log_fatal,"epoll return %d\n", epollfd);
  473. myexit(-1);
  474. }
  475. ev.events = EPOLLIN;
  476. ev.data.u64 = udp_fd;
  477. ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, udp_fd, &ev);
  478. if (ret!=0) {
  479. mylog(log_fatal,"add udp_listen_fd error\n");
  480. myexit(-1);
  481. }
  482. ev.events = EPOLLIN;
  483. ev.data.u64 = raw_recv_fd;
  484. ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, raw_recv_fd, &ev);
  485. if (ret!= 0) {
  486. mylog(log_fatal,"add raw_fd error\n");
  487. myexit(-1);
  488. }
  489. ////add_timer for fake_tcp_keep_connection_client
  490. //sleep(10);
  491. //memset(&udp_old_addr_in,0,sizeof(sockaddr_in));
  492. int unbind=1;
  493. set_timer(epollfd,timer_fd);
  494. mylog(log_debug,"send_raw : from %x %d to %x %d\n",send_info.src_ip,send_info.src_port,send_info.dst_ip,send_info.dst_port);
  495. int fifo_fd=-1;
  496. if(fifo_file[0]!=0)
  497. {
  498. fifo_fd=create_fifo(fifo_file);
  499. ev.events = EPOLLIN;
  500. ev.data.u64 = fifo_fd;
  501. ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, fifo_fd, &ev);
  502. if (ret!= 0) {
  503. mylog(log_fatal,"add fifo_fd to epoll error %s\n",strerror(errno));
  504. myexit(-1);
  505. }
  506. mylog(log_info,"fifo_file=%s\n",fifo_file);
  507. }
  508. while(1)////////////////////////
  509. {
  510. if(about_to_exit) myexit(0);
  511. epoll_trigger_counter++;
  512. int nfds = epoll_wait(epollfd, events, max_events, 180 * 1000);
  513. if (nfds < 0) { //allow zero
  514. if(errno==EINTR )
  515. {
  516. mylog(log_info,"epoll interrupted by signal,continue\n");
  517. //close(fifo_fd);
  518. //myexit(0);
  519. }
  520. else
  521. {
  522. mylog(log_fatal,"epoll_wait return %d,%s\n", nfds,strerror(errno));
  523. myexit(-1);
  524. }
  525. }
  526. int idx;
  527. for (idx = 0; idx < nfds; ++idx) {
  528. if (events[idx].data.u64 == (u64_t)raw_recv_fd)
  529. {
  530. iphdr *iph;tcphdr *tcph;
  531. client_on_raw_recv(conn_info);
  532. }
  533. else if(events[idx].data.u64 ==(u64_t)timer_fd)
  534. {
  535. u64_t value;
  536. read(timer_fd, &value, 8);
  537. client_on_timer(conn_info);
  538. mylog(log_trace,"epoll_trigger_counter: %d \n",epoll_trigger_counter);
  539. epoll_trigger_counter=0;
  540. }
  541. else if (events[idx].data.u64 == (u64_t)fifo_fd)
  542. {
  543. int len=read (fifo_fd, buf, sizeof (buf));
  544. //assert(len>=0);
  545. if(len<0)
  546. {
  547. mylog(log_warn,"fifo read failed len=%d,errno=%s\n",len,strerror(errno));
  548. continue;
  549. }
  550. buf[len]=0;
  551. while(len>=1&&buf[len-1]=='\n')
  552. buf[len-1]=0;
  553. mylog(log_info,"got data from fifo,len=%d,s=[%s]\n",len,buf);
  554. if(strcmp(buf,"reconnect")==0)
  555. {
  556. mylog(log_info,"received command: reconnect\n");
  557. conn_info.state.client_current_state=client_idle;
  558. conn_info.my_id=get_true_random_number_nz();
  559. }
  560. else
  561. {
  562. mylog(log_info,"unknown command\n");
  563. }
  564. }
  565. else if (events[idx].data.u64 == (u64_t)udp_fd)
  566. {
  567. int recv_len;
  568. struct sockaddr_in udp_new_addr_in={0};
  569. socklen_t udp_new_addr_len = sizeof(sockaddr_in);
  570. if ((recv_len = recvfrom(udp_fd, buf, max_data_len+1, 0,
  571. (struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) == -1) {
  572. mylog(log_error,"recv_from error,this shouldnt happen at client\n");
  573. myexit(1);
  574. };
  575. if(recv_len==max_data_len+1)
  576. {
  577. mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
  578. continue;
  579. }
  580. if(recv_len>=mtu_warn)
  581. {
  582. mylog(log_warn,"huge packet,data len=%d (>=%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ",recv_len,mtu_warn);
  583. }
  584. mylog(log_trace,"Received packet from %s:%d,len: %d\n", inet_ntoa(udp_new_addr_in.sin_addr),
  585. ntohs(udp_new_addr_in.sin_port),recv_len);
  586. /*
  587. if(udp_old_addr_in.sin_addr.s_addr==0&&udp_old_addr_in.sin_port==0)
  588. {
  589. memcpy(&udp_old_addr_in,&udp_new_addr_in,sizeof(udp_new_addr_in));
  590. }
  591. else if(udp_new_addr_in.sin_addr.s_addr!=udp_old_addr_in.sin_addr.s_addr
  592. ||udp_new_addr_in.sin_port!=udp_old_addr_in.sin_port)
  593. {
  594. if(get_current_time()- last_udp_recv_time <udp_timeout)
  595. {
  596. printf("new <ip,port> connected in,ignored,bc last connection is still active\n");
  597. continue;
  598. }
  599. else
  600. {
  601. printf("new <ip,port> connected in,accpeted\n");
  602. memcpy(&udp_old_addr_in,&udp_new_addr_in,sizeof(udp_new_addr_in));
  603. conv_id++;
  604. }
  605. }*/
  606. //last_udp_recv_time=get_current_time();
  607. u64_t u64=((u64_t(udp_new_addr_in.sin_addr.s_addr))<<32u)+ntohs(udp_new_addr_in.sin_port);
  608. u32_t conv;
  609. if(!conn_info.blob->conv_manager.is_u64_used(u64))
  610. {
  611. if(conn_info.blob->conv_manager.get_size() >=max_conv_num)
  612. {
  613. mylog(log_warn,"ignored new udp connect bc max_conv_num exceed\n");
  614. continue;
  615. }
  616. conv=conn_info.blob->conv_manager.get_new_conv();
  617. conn_info.blob->conv_manager.insert_conv(conv,u64);
  618. mylog(log_info,"new packet from %s:%d,conv_id=%x\n",inet_ntoa(udp_new_addr_in.sin_addr),ntohs(udp_new_addr_in.sin_port),conv);
  619. }
  620. else
  621. {
  622. conv=conn_info.blob->conv_manager.find_conv_by_u64(u64);
  623. }
  624. conn_info.blob->conv_manager.update_active_time(conv);
  625. if(conn_info.state.client_current_state==client_ready)
  626. {
  627. /*
  628. char buf2[6000];
  629. int ret1=send_raw(conn_info.raw_info,buf2,40);
  630. int ret2=send_raw(conn_info.raw_info,buf2,500);
  631. int ret3=send_raw(conn_info.raw_info,buf2,1000);
  632. int ret4=send_raw(conn_info.raw_info,buf2,2000);
  633. mylog(log_warn,"ret= %d %d %d %d\n",ret1,ret2,ret3,ret4);*/
  634. send_data_safer(conn_info,buf,recv_len,conv);
  635. }
  636. }
  637. else
  638. {
  639. mylog(log_fatal,"unknown fd,this should never happen\n");
  640. myexit(-1);
  641. }
  642. }
  643. }
  644. return 0;
  645. }
  646. /*
  647. int test()
  648. {
  649. char ip_str[100]="8.8.8.8";
  650. u32_t ip=inet_addr(ip_str);
  651. u32_t dest_ip;
  652. string if_name;
  653. string hw;
  654. find_lower_level_info(ip,dest_ip,if_name,hw);
  655. printf("%s %s %s\n",my_ntoa(dest_ip),if_name.c_str(),hw.c_str());
  656. exit(0);
  657. return 0;
  658. }*/
  659. int main(int argc, char *argv[])
  660. {
  661. //printf("%llu\n",u64_t(-1));
  662. //test();
  663. //printf("%s\n",my_ntoa(0x00ffffff));
  664. //auto a=string_to_vec("a b c d ");
  665. //printf("%d\n",(int)a.size());
  666. //printf("%d %d %d %d",larger_than_u32(1,2),larger_than_u32(2,1),larger_than_u32(0xeeaaeebb,2),larger_than_u32(2,0xeeaaeebb));
  667. //assert(0==1);
  668. dup2(1, 2);//redirect stderr to stdout
  669. signal(SIGINT, signal_handler);
  670. signal(SIGHUP, signal_handler);
  671. signal(SIGKILL, signal_handler);
  672. signal(SIGTERM, signal_handler);
  673. signal(SIGQUIT, signal_handler);
  674. pre_process_arg(argc,argv);
  675. if(geteuid() != 0)
  676. {
  677. mylog(log_warn,"root check failed, it seems like you are using a non-root account. we can try to continue, but it may fail. If you want to run udp2raw as non-root, you have to add iptables rule manually, and grant udp2raw CAP_NET_RAW capability, check README.md in repo for more info.\n");
  678. }
  679. else
  680. {
  681. mylog(log_warn,"you can run udp2raw with non-root account for better security. check README.md in repo for more info.\n");
  682. }
  683. local_ip_uint32=inet_addr(local_ip);
  684. source_ip_uint32=inet_addr(source_ip);
  685. //strncpy(remote_ip,remote_address,sizeof(remote_ip)-1);
  686. mylog(log_info,"remote_ip=[%s], make sure this is a vaild IP address\n",remote_ip);
  687. strcpy(remote_ip,remote_address);
  688. remote_ip_uint32=inet_addr(remote_ip);
  689. //current_time_rough=get_current_time();
  690. init_random_number_fd();
  691. srand(get_true_random_number_nz());
  692. const_id=get_true_random_number_nz();
  693. mylog(log_info,"const_id:%x\n",const_id);
  694. char tmp[1000]="";
  695. strcat(tmp,key_string);
  696. strcat(tmp,"key1");
  697. md5((uint8_t*)tmp,strlen(tmp),(uint8_t*)key);
  698. /*
  699. tmp[0]=0;
  700. strcat(tmp,key_string);
  701. strcat(tmp,"key2");
  702. md5((uint8_t*)tmp,strlen(tmp),(uint8_t*)key2);*/
  703. iptables_rule();
  704. init_raw_socket();
  705. if(program_mode==client_mode)
  706. {
  707. client_event_loop();
  708. }
  709. else
  710. {
  711. mylog(log_fatal,"server mode not supported in portable version\n");
  712. myexit(-1);
  713. //server_event_loop();
  714. }
  715. return 0;
  716. }