classic.cpp 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273
  1. /*
  2. * classic.cpp
  3. *
  4. * Created on: Sep 15, 2017
  5. * Author: root
  6. */
  7. #include "common.h"
  8. #include "log.h"
  9. #include "git_version.h"
  10. namespace classic
  11. {
  12. using namespace std;
  13. typedef unsigned long long u64_t; //this works on most platform,avoid using the PRId64
  14. typedef long long i64_t;
  15. typedef unsigned int u32_t;
  16. typedef int i32_t;
  17. typedef u64_t anti_replay_seq_t;
  18. int disable_replay_filter=0;
  19. int dup_num=1;
  20. int dup_delay_min=20; //0.1ms
  21. int dup_delay_max=20;
  22. //int dup_first_delay=9000; //0.1ms
  23. int jitter_min=0;
  24. int jitter_max=0;
  25. int iv_min=2;
  26. int iv_max=16;//< 256;
  27. int random_number_fd=-1;
  28. int remote_fd=-1;
  29. int local_fd=-1;
  30. int is_client = 0, is_server = 0;
  31. int local_listen_fd=-1;
  32. int disable_conv_clear=0;
  33. int mtu_warn=1350;
  34. u32_t remote_address_uint32=0;
  35. char local_address[100], remote_address[100];
  36. int local_port = -1, remote_port = -1;
  37. int multi_process_mode=0;
  38. const u32_t anti_replay_buff_size=10000;
  39. char key_string[1000]= "secret key";
  40. int random_drop=0;
  41. u64_t last_report_time=0;
  42. int report_interval=0;
  43. u64_t packet_send_count=0;
  44. u64_t dup_packet_send_count=0;
  45. u64_t packet_recv_count=0;
  46. u64_t dup_packet_recv_count=0;
  47. int max_pending_packet=0;
  48. int random_between(u32_t a,u32_t b)
  49. {
  50. if(a>b)
  51. {
  52. mylog(log_fatal,"min >max?? %d %d\n",a ,b);
  53. myexit(1);
  54. }
  55. if(a==b)return a;
  56. else return a+get_true_random_number()%(b+1-a);
  57. }
  58. int VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV;
  59. struct anti_replay_t
  60. {
  61. u64_t max_packet_received;
  62. u64_t replay_buffer[anti_replay_buff_size];
  63. unordered_set<u64_t> st;
  64. u32_t const_id;
  65. u32_t anti_replay_seq;
  66. int index;
  67. anti_replay_seq_t get_new_seq_for_send()
  68. {
  69. anti_replay_seq_t res=const_id;
  70. res<<=32u;
  71. anti_replay_seq++;
  72. res|=anti_replay_seq;
  73. return res;
  74. }
  75. void prepare()
  76. {
  77. anti_replay_seq=get_true_random_number();//random first seq
  78. const_id=get_true_random_number_nz();
  79. }
  80. anti_replay_t()
  81. {
  82. memset(replay_buffer,0,sizeof(replay_buffer));
  83. st.rehash(anti_replay_buff_size*10);
  84. max_packet_received=0;
  85. index=0;
  86. }
  87. int is_vaild(u64_t seq)
  88. {
  89. //if(disable_replay_filter) return 1;
  90. if(seq==0)
  91. {
  92. mylog(log_debug,"seq=0\n");
  93. return 0;
  94. }
  95. if(st.find(seq)!=st.end() )
  96. {
  97. mylog(log_trace,"seq %llx exist\n",seq);
  98. return 0;
  99. }
  100. if(replay_buffer[index]!=0)
  101. {
  102. assert(st.find(replay_buffer[index])!=st.end());
  103. st.erase(replay_buffer[index]);
  104. }
  105. replay_buffer[index]=seq;
  106. st.insert(seq);
  107. index++;
  108. if(index==int(anti_replay_buff_size)) index=0;
  109. return 1; //for complier check
  110. }
  111. }anti_replay;
  112. struct conn_manager_t //TODO change map to unordered map
  113. {
  114. //typedef hash_map map;
  115. unordered_map<u64_t,u32_t> u64_to_fd; //conv and u64 are both supposed to be uniq
  116. unordered_map<u32_t,u64_t> fd_to_u64;
  117. unordered_map<u32_t,u64_t> fd_last_active_time;
  118. unordered_map<u32_t,u64_t>::iterator clear_it;
  119. unordered_map<u32_t,u64_t>::iterator it;
  120. unordered_map<u32_t,u64_t>::iterator old_it;
  121. //void (*clear_function)(uint64_t u64) ;
  122. long long last_clear_time;
  123. list<int> clear_list;
  124. conn_manager_t()
  125. {
  126. clear_it=fd_last_active_time.begin();
  127. long long last_clear_time=0;
  128. rehash();
  129. //clear_function=0;
  130. }
  131. ~conn_manager_t()
  132. {
  133. clear();
  134. }
  135. int get_size()
  136. {
  137. return fd_to_u64.size();
  138. }
  139. void rehash()
  140. {
  141. u64_to_fd.rehash(10007);
  142. fd_to_u64.rehash(10007);
  143. fd_last_active_time.rehash(10007);
  144. }
  145. void clear()
  146. {
  147. if(disable_conv_clear) return ;
  148. for(it=fd_to_u64.begin();it!=fd_to_u64.end();it++)
  149. {
  150. //int fd=int((it->second<<32u)>>32u);
  151. close( it->first);
  152. }
  153. u64_to_fd.clear();
  154. fd_to_u64.clear();
  155. fd_last_active_time.clear();
  156. clear_it=fd_last_active_time.begin();
  157. }
  158. int exist_fd(u32_t fd)
  159. {
  160. return fd_to_u64.find(fd)!=fd_to_u64.end();
  161. }
  162. int exist_u64(u64_t u64)
  163. {
  164. return u64_to_fd.find(u64)!=u64_to_fd.end();
  165. }
  166. u32_t find_fd_by_u64(u64_t u64)
  167. {
  168. return u64_to_fd[u64];
  169. }
  170. u64_t find_u64_by_fd(u32_t fd)
  171. {
  172. return fd_to_u64[fd];
  173. }
  174. int update_active_time(u32_t fd)
  175. {
  176. return fd_last_active_time[fd]=get_current_time();
  177. }
  178. int insert_fd(u32_t fd,u64_t u64)
  179. {
  180. u64_to_fd[u64]=fd;
  181. fd_to_u64[fd]=u64;
  182. fd_last_active_time[fd]=get_current_time();
  183. return 0;
  184. }
  185. int erase_fd(u32_t fd)
  186. {
  187. if(disable_conv_clear) return 0;
  188. u64_t u64=fd_to_u64[fd];
  189. u32_t ip= (u64 >> 32u);
  190. int port= uint16_t((u64 << 32u) >> 32u);
  191. mylog(log_info,"fd %d cleared,assocated adress %s,%d\n",fd,my_ntoa(ip),port);
  192. close(fd);
  193. fd_to_u64.erase(fd);
  194. u64_to_fd.erase(u64);
  195. fd_last_active_time.erase(fd);
  196. return 0;
  197. }
  198. void check_clear_list()
  199. {
  200. while(!clear_list.empty())
  201. {
  202. int fd=*clear_list.begin();
  203. clear_list.pop_front();
  204. erase_fd(fd);
  205. }
  206. }
  207. int clear_inactive()
  208. {
  209. if(get_current_time()-last_clear_time>conv_clear_interval)
  210. {
  211. last_clear_time=get_current_time();
  212. return clear_inactive0();
  213. }
  214. return 0;
  215. }
  216. int clear_inactive0()
  217. {
  218. if(disable_conv_clear) return 0;
  219. //map<uint32_t,uint64_t>::iterator it;
  220. int cnt=0;
  221. it=clear_it;
  222. int size=fd_last_active_time.size();
  223. int num_to_clean=size/conv_clear_ratio+conv_clear_min; //clear 1/10 each time,to avoid latency glitch
  224. u64_t current_time=get_current_time();
  225. for(;;)
  226. {
  227. if(cnt>=num_to_clean) break;
  228. if(fd_last_active_time.begin()==fd_last_active_time.end()) break;
  229. if(it==fd_last_active_time.end())
  230. {
  231. it=fd_last_active_time.begin();
  232. }
  233. if( current_time -it->second >conv_timeout )
  234. {
  235. //mylog(log_info,"inactive conv %u cleared \n",it->first);
  236. old_it=it;
  237. it++;
  238. u32_t fd= old_it->first;
  239. erase_fd(old_it->first);
  240. }
  241. else
  242. {
  243. it++;
  244. }
  245. cnt++;
  246. }
  247. return 0;
  248. }
  249. }conn_manager;
  250. typedef u64_t my_time_t;
  251. struct delay_data
  252. {
  253. int fd;
  254. int times_left;
  255. char * data;
  256. int len;
  257. u64_t u64;
  258. };
  259. int delay_timer_fd;
  260. int sendto_u64 (int fd,char * buf, int len,int flags, u64_t u64)
  261. {
  262. if(is_server)
  263. {
  264. dup_packet_send_count++;
  265. }
  266. if(is_server&&random_drop!=0)
  267. {
  268. if(get_true_random_number()%10000<(u32_t)random_drop)
  269. {
  270. return 0;
  271. }
  272. }
  273. sockaddr_in tmp_sockaddr;
  274. memset(&tmp_sockaddr,0,sizeof(tmp_sockaddr));
  275. tmp_sockaddr.sin_family = AF_INET;
  276. tmp_sockaddr.sin_addr.s_addr = (u64 >> 32u);
  277. tmp_sockaddr.sin_port = htons(uint16_t((u64 << 32u) >> 32u));
  278. return sendto(fd, buf,
  279. len , 0,
  280. (struct sockaddr *) &tmp_sockaddr,
  281. sizeof(tmp_sockaddr));
  282. }
  283. int send_fd (int fd,char * buf, int len,int flags)
  284. {
  285. if(is_client)
  286. {
  287. dup_packet_send_count++;
  288. }
  289. if(is_client&&random_drop!=0)
  290. {
  291. if(get_true_random_number()%10000<(u32_t)random_drop)
  292. {
  293. return 0;
  294. }
  295. }
  296. return send(fd,buf,len,flags);
  297. }
  298. multimap<my_time_t,delay_data> delay_mp;
  299. int add_to_delay_mp(int fd,int times_left,u32_t delay,char * buf,int len,u64_t u64)
  300. {
  301. if(max_pending_packet!=0&&int(delay_mp.size()) >=max_pending_packet)
  302. {
  303. mylog(log_warn,"max pending packet reached,ignored\n");
  304. return 0;
  305. }
  306. delay_data tmp;
  307. tmp.data = buf;
  308. tmp.fd = fd;
  309. tmp.times_left = times_left;
  310. tmp.len = len;
  311. tmp.u64=u64;
  312. my_time_t tmp_time=get_current_time_us();
  313. tmp_time+=delay*100;
  314. delay_mp.insert(make_pair(tmp_time,tmp));
  315. return 0;
  316. }
  317. int add_and_new(int fd,int times_left,u32_t delay,char * buf,int len,u64_t u64)
  318. {
  319. if(times_left<=0) return -1;
  320. char * str= (char *)malloc(len);
  321. memcpy(str,buf,len);
  322. add_to_delay_mp(fd,times_left,delay,str,len,u64);
  323. return 0;
  324. }
  325. multimap<u64_t,delay_data> new_delay_mp;
  326. void handler(int num) {
  327. int status;
  328. int pid;
  329. while ((pid = waitpid(-1, &status, WNOHANG)) > 0) {
  330. if (WIFEXITED(status)) {
  331. //printf("The child exit with code %d",WEXITSTATUS(status));
  332. }
  333. }
  334. }
  335. void encrypt_0(char * input,int &len,char *key)
  336. {
  337. int i,j;
  338. if(key[0]==0) return;
  339. for(i=0,j=0;i<len;i++,j++)
  340. {
  341. if(key[j]==0)j=0;
  342. input[i]^=key[j];
  343. }
  344. }
  345. void decrypt_0(char * input,int &len,char *key)
  346. {
  347. int i,j;
  348. if(key[0]==0) return;
  349. for(i=0,j=0;i<len;i++,j++)
  350. {
  351. if(key[j]==0)j=0;
  352. input[i]^=key[j];
  353. }
  354. }
  355. int add_seq(char * data,int &data_len )
  356. {
  357. if(data_len<0) return -1;
  358. anti_replay_seq_t seq=anti_replay.get_new_seq_for_send();
  359. seq=hton64(seq);
  360. memcpy(data+data_len,&seq,sizeof(seq));
  361. data_len+=sizeof(seq);
  362. return 0;
  363. }
  364. int remove_seq(char * data,int &data_len)
  365. {
  366. anti_replay_seq_t seq;
  367. if(data_len<int(sizeof(seq))) return -1;
  368. data_len-=sizeof(seq);
  369. memcpy(&seq,data+data_len,sizeof(seq));
  370. seq=ntoh64(seq);
  371. if(anti_replay.is_vaild(seq)==0)
  372. {
  373. if(disable_replay_filter==1)
  374. return 0;
  375. mylog(log_trace,"seq %llx dropped bc of replay-filter\n ",seq);
  376. return -1;
  377. }
  378. packet_recv_count++;
  379. return 0;
  380. }
  381. int do_obscure(const char * input, int in_len,char *output,int &out_len)
  382. {
  383. //memcpy(output,input,in_len);
  384. // out_len=in_len;
  385. //return 0;
  386. int i, j, k;
  387. if (in_len > 65535||in_len<0)
  388. return -1;
  389. int iv_len=iv_min+rand()%(iv_max-iv_min);
  390. get_true_random_chars(output,iv_len);
  391. memcpy(output+iv_len,input,in_len);
  392. output[iv_len+in_len]=(uint8_t)iv_len;
  393. output[iv_len+in_len]^=output[0];
  394. output[iv_len+in_len]^=key_string[0];
  395. for(i=0,j=0,k=1;i<in_len;i++,j++,k++)
  396. {
  397. if(j==iv_len) j=0;
  398. if(key_string[k]==0)k=0;
  399. output[iv_len+i]^=output[j];
  400. output[iv_len+i]^=key_string[k];
  401. }
  402. out_len=iv_len+in_len+1;
  403. return 0;
  404. }
  405. int de_obscure(const char * input, int in_len,char *output,int &out_len)
  406. {
  407. //memcpy(output,input,in_len);
  408. //out_len=in_len;
  409. //return 0;
  410. int i, j, k;
  411. if (in_len > 65535||in_len<0)
  412. {
  413. mylog(log_debug,"in_len > 65535||in_len<0 , %d",in_len);
  414. return -1;
  415. }
  416. int iv_len= int ((uint8_t)(input[in_len-1]^input[0]^key_string[0]) );
  417. out_len=in_len-1-iv_len;
  418. if(out_len<0)
  419. {
  420. mylog(log_debug,"%d %d\n",in_len,out_len);
  421. return -1;
  422. }
  423. for(i=0,j=0,k=1;i<in_len;i++,j++,k++)
  424. {
  425. if(j==iv_len) j=0;
  426. if(key_string[k]==0)k=0;
  427. output[i]=input[iv_len+i]^input[j]^key_string[k];
  428. }
  429. dup_packet_recv_count++;
  430. return 0;
  431. }
  432. void check_delay_map()
  433. {
  434. if(!delay_mp.empty())
  435. {
  436. my_time_t current_time;
  437. multimap<my_time_t,delay_data>::iterator it;
  438. while(1)
  439. {
  440. int ret=0;
  441. it=delay_mp.begin();
  442. if(it==delay_mp.end()) break;
  443. current_time=get_current_time_us();
  444. if(it->first < current_time||it->first ==current_time)
  445. {
  446. if (is_client) {
  447. if (conn_manager.exist_fd(it->second.fd)) {
  448. u64_t u64 = conn_manager.find_u64_by_fd(it->second.fd);
  449. if (u64 != it->second.u64) {
  450. it->second.times_left = 0; //fd has been deleted and recreated
  451. // 偷懒的做法
  452. } else {
  453. char new_data[buf_len];
  454. int new_len = 0;
  455. do_obscure(it->second.data, it->second.len,
  456. new_data, new_len);
  457. ret = send_fd(it->second.fd, new_data, new_len, 0);
  458. }
  459. } else {
  460. it->second.times_left = 0;
  461. }
  462. } else {
  463. if (conn_manager.exist_fd(it->second.fd)) {
  464. u64_t u64 = conn_manager.find_u64_by_fd(it->second.fd);
  465. if (u64 != it->second.u64) {
  466. it->second.times_left = 0;//fd has been deleted and recreated
  467. // 偷懒的做法
  468. } else {
  469. char new_data[buf_len];
  470. int new_len = 0;
  471. do_obscure(it->second.data, it->second.len,
  472. new_data, new_len);
  473. sendto_u64(local_listen_fd, new_data, new_len, 0,
  474. u64);
  475. }
  476. } else {
  477. it->second.times_left = 0;
  478. }
  479. }
  480. if (ret < 0) {
  481. mylog(log_debug, "send return %d at @300", ret);
  482. }
  483. if(it->second.times_left>1)
  484. {
  485. //delay_mp.insert(pair<my_time,delay_data>(current_time));
  486. add_to_delay_mp(it->second.fd,it->second.times_left-1,random_between(dup_delay_min,dup_delay_max),it->second.data,it->second.len,it->second.u64);
  487. }
  488. else
  489. {
  490. free(it->second.data);
  491. }
  492. delay_mp.erase(it);
  493. }
  494. else
  495. {
  496. break;
  497. }
  498. }
  499. if(!delay_mp.empty())
  500. {
  501. itimerspec its;
  502. memset(&its.it_interval,0,sizeof(its.it_interval));
  503. its.it_value.tv_sec=delay_mp.begin()->first/1000000llu;
  504. its.it_value.tv_nsec=(delay_mp.begin()->first%1000000llu)*1000llu;
  505. timerfd_settime(delay_timer_fd,TFD_TIMER_ABSTIME,&its,0);
  506. }
  507. }
  508. }
  509. int create_new_udp(int &new_udp_fd)
  510. {
  511. struct sockaddr_in remote_addr_in;
  512. socklen_t slen = sizeof(sockaddr_in);
  513. memset(&remote_addr_in, 0, sizeof(remote_addr_in));
  514. remote_addr_in.sin_family = AF_INET;
  515. remote_addr_in.sin_port = htons(remote_port);
  516. remote_addr_in.sin_addr.s_addr = remote_address_uint32;
  517. new_udp_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  518. if (new_udp_fd < 0) {
  519. mylog(log_warn, "create udp_fd error\n");
  520. return -1;
  521. }
  522. setnonblocking(new_udp_fd);
  523. set_buf_size(new_udp_fd);
  524. mylog(log_debug, "created new udp_fd %d\n", new_udp_fd);
  525. int ret = connect(new_udp_fd, (struct sockaddr *) &remote_addr_in, slen);
  526. if (ret != 0) {
  527. mylog(log_warn, "udp fd connect fail %d %s\n",ret,strerror(errno));
  528. close(new_udp_fd);
  529. return -1;
  530. }
  531. return 0;
  532. }
  533. int set_timer(int epollfd,int &timer_fd)
  534. {
  535. int ret;
  536. epoll_event ev;
  537. itimerspec its;
  538. memset(&its,0,sizeof(its));
  539. if((timer_fd=timerfd_create(CLOCK_MONOTONIC,TFD_NONBLOCK)) < 0)
  540. {
  541. mylog(log_fatal,"timer_fd create error\n");
  542. myexit(1);
  543. }
  544. its.it_interval.tv_sec=(timer_interval/1000);
  545. its.it_interval.tv_nsec=(timer_interval%1000)*1000ll*1000ll;
  546. its.it_value.tv_nsec=1; //imidiately
  547. timerfd_settime(timer_fd,0,&its,0);
  548. ev.events = EPOLLIN;
  549. ev.data.fd = timer_fd;
  550. ret=epoll_ctl(epollfd, EPOLL_CTL_ADD, timer_fd, &ev);
  551. if (ret < 0) {
  552. mylog(log_fatal,"epoll_ctl return %d\n", ret);
  553. myexit(-1);
  554. }
  555. return 0;
  556. }
  557. int event_loop()
  558. {
  559. struct sockaddr_in local_me, local_other;
  560. local_listen_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  561. int yes = 1;
  562. //setsockopt(local_listen_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
  563. set_buf_size(local_listen_fd,4*1024*1024);
  564. setnonblocking(local_listen_fd);
  565. //char data[buf_len];
  566. //char *data=data0;
  567. socklen_t slen = sizeof(sockaddr_in);
  568. memset(&local_me, 0, sizeof(local_me));
  569. local_me.sin_family = AF_INET;
  570. local_me.sin_port = htons(local_port);
  571. local_me.sin_addr.s_addr = inet_addr(local_address);
  572. if (bind(local_listen_fd, (struct sockaddr*) &local_me, slen) == -1)
  573. {
  574. mylog(log_fatal,"socket bind error");
  575. myexit(1);
  576. }
  577. int epollfd = epoll_create1(0);
  578. const int max_events = 4096;
  579. struct epoll_event ev, events[max_events];
  580. if (epollfd < 0)
  581. {
  582. mylog(log_fatal,"epoll created return %d\n", epollfd);
  583. myexit(-1);
  584. }
  585. ev.events = EPOLLIN;
  586. ev.data.fd = local_listen_fd;
  587. int ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, local_listen_fd, &ev);
  588. if(ret!=0)
  589. {
  590. mylog(log_fatal,"epoll created return %d\n", epollfd);
  591. myexit(-1);
  592. }
  593. int clear_timer_fd=-1;
  594. set_timer(epollfd,clear_timer_fd);
  595. if ((delay_timer_fd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK)) < 0)
  596. {
  597. mylog(log_fatal,"timer_fd create error");
  598. myexit(1);
  599. }
  600. ev.events = EPOLLIN;
  601. ev.data.fd = delay_timer_fd;
  602. itimerspec zero_its;
  603. memset(&zero_its, 0, sizeof(zero_its));
  604. timerfd_settime(delay_timer_fd, TFD_TIMER_ABSTIME, &zero_its, 0);
  605. epoll_ctl(epollfd, EPOLL_CTL_ADD, delay_timer_fd, &ev);
  606. if (ret < 0)
  607. {
  608. mylog(log_fatal,"epoll_ctl return %d\n", ret);
  609. myexit(-1);
  610. }
  611. for (;;)
  612. {
  613. int nfds = epoll_wait(epollfd, events, max_events, 180 * 1000); //3mins
  614. if (nfds < 0)
  615. {
  616. mylog(log_fatal,"epoll_wait return %d\n", nfds);
  617. myexit(-1);
  618. }
  619. int n;
  620. int clear_triggered=0;
  621. for (n = 0; n < nfds; ++n)
  622. {
  623. if (events[n].data.fd == local_listen_fd) //data income from local end
  624. {
  625. char data[buf_len];
  626. int data_len;
  627. slen = sizeof(sockaddr_in);
  628. if ((data_len = recvfrom(local_listen_fd, data, max_data_len, 0,
  629. (struct sockaddr *) &local_other, &slen)) == -1) //<--first packet from a new ip:port turple
  630. {
  631. mylog(log_error,"recv_from error,errno %s,this shouldnt happen,but lets try to pretend it didnt happen",strerror(errno));
  632. //myexit(1);
  633. continue;
  634. }
  635. mylog(log_trace, "received data from listen fd,%s:%d, len=%d\n", my_ntoa(local_other.sin_addr.s_addr),ntohs(local_other.sin_port),data_len);
  636. if(data_len>mtu_warn)
  637. {
  638. 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 ",data_len,mtu_warn);
  639. }
  640. data[data_len] = 0; //for easier debug
  641. u64_t u64=pack_u64(local_other.sin_addr.s_addr,ntohs(local_other.sin_port));
  642. if(!conn_manager.exist_u64(u64))
  643. {
  644. if(int(conn_manager.fd_to_u64.size())>=max_conv_num)
  645. {
  646. mylog(log_info,"new connection from %s:%d ,but ignored,bc of max_conv_num reached\n",my_ntoa(local_other.sin_addr.s_addr),ntohs(local_other.sin_port));
  647. continue;
  648. }
  649. int new_udp_fd;
  650. if(create_new_udp(new_udp_fd)!=0)
  651. {
  652. mylog(log_info,"new connection from %s:%d ,but create udp fd failed\n",my_ntoa(local_other.sin_addr.s_addr),ntohs(local_other.sin_port));
  653. continue;
  654. }
  655. struct epoll_event ev;
  656. mylog(log_trace, "u64: %lld\n", u64);
  657. ev.events = EPOLLIN;
  658. ev.data.fd = new_udp_fd;
  659. ret = epoll_ctl(epollfd, EPOLL_CTL_ADD, new_udp_fd, &ev);
  660. if (ret != 0) {
  661. mylog(log_info,"new connection from %s:%d ,but add to epoll failed\n",my_ntoa(local_other.sin_addr.s_addr),ntohs(local_other.sin_port));
  662. close(new_udp_fd);
  663. continue;
  664. }
  665. mylog(log_info,"new connection from %s:%d ,created new udp fd %d\n",my_ntoa(local_other.sin_addr.s_addr),ntohs(local_other.sin_port),new_udp_fd);
  666. conn_manager.insert_fd(new_udp_fd,u64);
  667. }
  668. int new_udp_fd=conn_manager.find_fd_by_u64(u64);
  669. conn_manager.update_active_time(new_udp_fd);
  670. int ret;
  671. if(is_client)
  672. {
  673. add_seq(data,data_len);
  674. if(jitter_max==0)
  675. {
  676. char new_data[buf_len];
  677. int new_len=0;
  678. do_obscure(data, data_len, new_data, new_len);
  679. ret = send_fd(new_udp_fd, new_data,new_len, 0);
  680. if (ret < 0) {
  681. mylog(log_warn, "send returned %d ,errno:%s\n", ret,strerror(errno));
  682. }
  683. add_and_new(new_udp_fd, dup_num - 1,random_between(dup_delay_min,dup_delay_max), data, data_len,u64);
  684. }
  685. else
  686. {
  687. add_and_new(new_udp_fd, dup_num,random_between(jitter_min,jitter_max), data, data_len,u64);
  688. }
  689. packet_send_count++;
  690. }
  691. else
  692. {
  693. char new_data[buf_len];
  694. int new_len;
  695. if (de_obscure(data, data_len, new_data, new_len) != 0) {
  696. mylog(log_trace,"de_obscure failed \n");
  697. continue;
  698. }
  699. //dup_packet_recv_count++;
  700. if (remove_seq(new_data, new_len) != 0) {
  701. mylog(log_trace,"remove_seq failed \n");
  702. continue;
  703. }
  704. //packet_recv_count++;
  705. ret = send_fd(new_udp_fd, new_data,new_len, 0);
  706. if (ret < 0) {
  707. mylog(log_warn, "send returned %d,%s\n", ret,strerror(errno));
  708. //perror("what happened????");
  709. }
  710. }
  711. }
  712. else if(events[n].data.fd == clear_timer_fd)
  713. {
  714. clear_triggered=1;
  715. if(report_interval!=0 &&get_current_time()-last_report_time>u64_t(report_interval)*1000)
  716. {
  717. last_report_time=get_current_time();
  718. if(is_client)
  719. mylog(log_info,"client-->server: %llu,%llu(include dup); server-->client %llu,%lld(include dup)\n",packet_send_count,
  720. dup_packet_send_count,packet_recv_count,dup_packet_recv_count);
  721. else
  722. mylog(log_info,"client-->server: %llu,%llu(include dup); server-->client %llu,%lld(include dup)\n",packet_recv_count,dup_packet_recv_count,packet_send_count,
  723. dup_packet_send_count);
  724. }
  725. }
  726. else if (events[n].data.fd == delay_timer_fd)
  727. {
  728. uint64_t value;
  729. read(delay_timer_fd, &value, 8);
  730. //printf("<timerfd_triggered, %d>",delay_mp.size());
  731. //fflush(stdout);
  732. }
  733. else
  734. {
  735. int udp_fd=events[n].data.fd;
  736. if(!conn_manager.exist_fd(udp_fd)) continue;
  737. char data[buf_len];
  738. int data_len =recv(udp_fd,data,max_data_len,0);
  739. mylog(log_trace, "received data from udp fd %d, len=%d\n", udp_fd,data_len);
  740. if(data_len<0)
  741. {
  742. if(errno==ECONNREFUSED)
  743. {
  744. //conn_manager.clear_list.push_back(udp_fd);
  745. mylog(log_debug, "recv failed %d ,udp_fd%d,errno:%s\n", data_len,udp_fd,strerror(errno));
  746. }
  747. mylog(log_warn, "recv failed %d ,udp_fd%d,errno:%s\n", data_len,udp_fd,strerror(errno));
  748. continue;
  749. }
  750. if(data_len>mtu_warn)
  751. {
  752. 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 ",data_len,mtu_warn);
  753. }
  754. assert(conn_manager.exist_fd(udp_fd));
  755. conn_manager.update_active_time(udp_fd);
  756. u64_t u64=conn_manager.find_u64_by_fd(udp_fd);
  757. if(is_client)
  758. {
  759. char new_data[buf_len];
  760. int new_len;
  761. if (de_obscure(data, data_len, new_data, new_len) != 0) {
  762. mylog(log_debug,"data_len=%d \n",data_len);
  763. continue;
  764. }
  765. //dup_packet_recv_count++;
  766. if (remove_seq(new_data, new_len) != 0) {
  767. mylog(log_debug,"remove_seq error \n");
  768. continue;
  769. }
  770. //packet_recv_count++;
  771. ret = sendto_u64(local_listen_fd, new_data,
  772. new_len , 0,u64);
  773. if (ret < 0) {
  774. mylog(log_warn, "sento returned %d,%s\n", ret,strerror(errno));
  775. //perror("ret<0");
  776. }
  777. }
  778. else
  779. {
  780. add_seq(data,data_len);
  781. if(jitter_max==0)
  782. {
  783. char new_data[buf_len];
  784. int new_len=0;
  785. do_obscure(data, data_len, new_data, new_len);
  786. ret = sendto_u64(local_listen_fd, new_data,
  787. new_len , 0,u64);
  788. add_and_new(udp_fd, dup_num - 1,random_between(dup_delay_min,dup_delay_max), data, data_len,u64);
  789. if (ret < 0) {
  790. mylog(log_warn, "sento returned %d,%s\n", ret,strerror(errno));
  791. //perror("ret<0");
  792. }
  793. }
  794. else
  795. {
  796. add_and_new(udp_fd, dup_num,random_between(jitter_min,jitter_max), data, data_len,u64);
  797. }
  798. packet_send_count++;
  799. }
  800. //mylog(log_trace, "%s :%d\n", inet_ntoa(tmp_sockaddr.sin_addr),
  801. // ntohs(tmp_sockaddr.sin_port));
  802. //mylog(log_trace, "%d byte sent\n", ret);
  803. }
  804. }
  805. check_delay_map();
  806. conn_manager.check_clear_list();
  807. if(clear_triggered) // 删除操作在epoll event的最后进行,防止event cache中的fd失效。
  808. {
  809. u64_t value;
  810. read(clear_timer_fd, &value, 8);
  811. mylog(log_trace, "timer!\n");
  812. conn_manager.clear_inactive();
  813. }
  814. }
  815. myexit(0);
  816. return 0;
  817. }
  818. void print_help()
  819. {
  820. char git_version_buf[100]={0};
  821. strncpy(git_version_buf,gitversion,10);
  822. printf("UDPspeeder\n");
  823. printf("git version:%s ",git_version_buf);
  824. printf("build date:%s %s\n",__DATE__,__TIME__);
  825. printf("repository: https://github.com/wangyu-/UDPspeeder\n");
  826. printf("\n");
  827. printf("usage:\n");
  828. printf(" run as client : ./this_program -c -l local_listen_ip:local_port -r server_ip:server_port [options]\n");
  829. printf(" run as server : ./this_program -s -l server_listen_ip:server_port -r remote_ip:remote_port [options]\n");
  830. printf("\n");
  831. printf("common option,must be same on both sides:\n");
  832. printf(" -k,--key <string> key for simple xor encryption,default:\"secret key\"\n");
  833. printf("main options:\n");
  834. printf(" -d <number> duplicated packet number, -d 0 means no duplicate. default value:0\n");
  835. printf(" -t <number> duplicated packet delay time, unit: 0.1ms,default value:20(2ms)\n");
  836. printf(" -j <number> simulated jitter.randomly delay first packet for 0~jitter_value*0.1 ms,to\n");
  837. printf(" create simulated jitter.default value:0.do not use if you dont\n");
  838. printf(" know what it means\n");
  839. printf(" --report <number> turn on udp send/recv report,and set a time interval for reporting,unit:s\n");
  840. printf("advanced options:\n");
  841. printf(" -t tmin:tmax simliar to -t above,but delay randomly between tmin and tmax\n");
  842. printf(" -j jmin:jmax simliar to -j above,but create jitter randomly between jmin and jmax\n");
  843. printf(" --random-drop <number> simulate packet loss ,unit:0.01%%\n");
  844. printf(" --disable-filter disable duplicate packet filter.\n");
  845. printf(" -m <number> max pending packets,to prevent the program from eating up all your memory,\n");
  846. printf(" default value:0(disabled).\n");
  847. printf("other options:\n");
  848. printf(" --log-level <number> 0:never 1:fatal 2:error 3:warn \n");
  849. printf(" 4:info (default) 5:debug 6:trace\n");
  850. printf(" --log-position enable file name,function name,line number in log\n");
  851. printf(" --disable-color disable log color\n");
  852. printf(" --sock-buf <number> buf size for socket,>=10 and <=10240,unit:kbyte,default:1024\n");
  853. //printf(" -p use multi-process mode instead of epoll.very costly,only for test,do dont use\n");
  854. printf(" -h,--help print this help message\n");
  855. //printf("common options,these options must be same on both side\n");
  856. }
  857. void process_arg(int argc, char *argv[])
  858. {
  859. int i, j, k;
  860. int opt;
  861. static struct option long_options[] =
  862. {
  863. {"log-level", required_argument, 0, 1},
  864. {"log-position", no_argument, 0, 1},
  865. {"disable-color", no_argument, 0, 1},
  866. {"disable-filter", no_argument, 0, 1},
  867. {"sock-buf", required_argument, 0, 1},
  868. {"random-drop", required_argument, 0, 1},
  869. {"report", required_argument, 0, 1},
  870. {NULL, 0, 0, 0}
  871. };
  872. int option_index = 0;
  873. if (argc == 1)
  874. {
  875. print_help();
  876. myexit( -1);
  877. }
  878. for (i = 0; i < argc; i++)
  879. {
  880. if(strcmp(argv[i],"-h")==0||strcmp(argv[i],"--help")==0)
  881. {
  882. print_help();
  883. myexit(0);
  884. }
  885. }
  886. for (i = 0; i < argc; i++)
  887. {
  888. if(strcmp(argv[i],"--log-level")==0)
  889. {
  890. if(i<argc -1)
  891. {
  892. sscanf(argv[i+1],"%d",&log_level);
  893. if(0<=log_level&&log_level<log_end)
  894. {
  895. }
  896. else
  897. {
  898. log_bare(log_fatal,"invalid log_level\n");
  899. myexit(-1);
  900. }
  901. }
  902. }
  903. if(strcmp(argv[i],"--disable-color")==0)
  904. {
  905. enable_log_color=0;
  906. }
  907. }
  908. mylog(log_info,"argc=%d ", argc);
  909. for (i = 0; i < argc; i++) {
  910. log_bare(log_info, "%s ", argv[i]);
  911. }
  912. log_bare(log_info, "\n");
  913. if (argc == 1)
  914. {
  915. print_help();
  916. myexit(-1);
  917. }
  918. int no_l = 1, no_r = 1;
  919. while ((opt = getopt_long(argc, argv, "l:r:d:t:hcspk:j:m:",long_options,&option_index)) != -1)
  920. {
  921. //string opt_key;
  922. //opt_key+=opt;
  923. switch (opt)
  924. {
  925. case 'p':
  926. multi_process_mode=1;
  927. break;
  928. case 'k':
  929. sscanf(optarg,"%s\n",key_string);
  930. mylog(log_debug,"key=%s\n",key_string);
  931. if(strlen(key_string)==0)
  932. {
  933. mylog(log_fatal,"key len=0??\n");
  934. myexit(-1);
  935. }
  936. break;
  937. case 'm':
  938. sscanf(optarg,"%d\n",&max_pending_packet);
  939. if(max_pending_packet<1000)
  940. {
  941. mylog(log_fatal,"max_pending_packet must be >1000\n");
  942. myexit(-1);
  943. }
  944. break;
  945. case 'j':
  946. if (strchr(optarg, ':') == 0)
  947. {
  948. int jitter;
  949. sscanf(optarg,"%d\n",&jitter);
  950. if(jitter<0 ||jitter>1000*100)
  951. {
  952. mylog(log_fatal,"jitter must be between 0 and 100,000(10 second)\n");
  953. myexit(-1);
  954. }
  955. jitter_min=0;
  956. jitter_max=jitter;
  957. }
  958. else
  959. {
  960. sscanf(optarg,"%d:%d\n",&jitter_min,&jitter_max);
  961. if(jitter_min<0 ||jitter_max<0||jitter_min>jitter_max)
  962. {
  963. mylog(log_fatal," must satisfy 0<=jmin<=jmax\n");
  964. myexit(-1);
  965. }
  966. }
  967. break;
  968. case 't':
  969. if (strchr(optarg, ':') == 0)
  970. {
  971. int dup_delay=-1;
  972. sscanf(optarg,"%d\n",&dup_delay);
  973. if(dup_delay<1||dup_delay>1000*100)
  974. {
  975. mylog(log_fatal,"dup_delay must be between 1 and 100,000(10 second)\n");
  976. myexit(-1);
  977. }
  978. dup_delay_min=dup_delay_max=dup_delay;
  979. }
  980. else
  981. {
  982. sscanf(optarg,"%d:%d\n",&dup_delay_min,&dup_delay_max);
  983. if(dup_delay_min<1 ||dup_delay_max<1||dup_delay_min>dup_delay_max)
  984. {
  985. mylog(log_fatal," must satisfy 1<=dmin<=dmax\n");
  986. myexit(-1);
  987. }
  988. }
  989. break;
  990. case 'd':
  991. dup_num=-1;
  992. sscanf(optarg,"%d\n",&dup_num);
  993. if(dup_num<0 ||dup_num>5)
  994. {
  995. mylog(log_fatal,"dup_num must be between 0 and 5\n");
  996. myexit(-1);
  997. }
  998. dup_num+=1;
  999. break;
  1000. case 'c':
  1001. is_client = 1;
  1002. break;
  1003. case 's':
  1004. is_server = 1;
  1005. break;
  1006. case 'l':
  1007. no_l = 0;
  1008. if (strchr(optarg, ':') != 0)
  1009. {
  1010. sscanf(optarg, "%[^:]:%d", local_address, &local_port);
  1011. }
  1012. else
  1013. {
  1014. mylog(log_fatal," -r ip:port\n");
  1015. myexit(1);
  1016. strcpy(local_address, "127.0.0.1");
  1017. sscanf(optarg, "%d", &local_port);
  1018. }
  1019. break;
  1020. case 'r':
  1021. no_r = 0;
  1022. if (strchr(optarg, ':') != 0)
  1023. {
  1024. //printf("in :\n");
  1025. //printf("%s\n",optarg);
  1026. sscanf(optarg, "%[^:]:%d", remote_address, &remote_port);
  1027. //printf("%d\n",remote_port);
  1028. }
  1029. else
  1030. {
  1031. mylog(log_fatal," -r ip:port\n");
  1032. myexit(1);
  1033. strcpy(remote_address, "127.0.0.1");
  1034. sscanf(optarg, "%d", &remote_port);
  1035. }
  1036. break;
  1037. case 'h':
  1038. break;
  1039. case 1:
  1040. if(strcmp(long_options[option_index].name,"log-level")==0)
  1041. {
  1042. }
  1043. else if(strcmp(long_options[option_index].name,"disable-filter")==0)
  1044. {
  1045. disable_replay_filter=1;
  1046. //enable_log_color=0;
  1047. }
  1048. else if(strcmp(long_options[option_index].name,"disable-color")==0)
  1049. {
  1050. //enable_log_color=0;
  1051. }
  1052. else if(strcmp(long_options[option_index].name,"log-position")==0)
  1053. {
  1054. enable_log_position=1;
  1055. }
  1056. else if(strcmp(long_options[option_index].name,"random-drop")==0)
  1057. {
  1058. sscanf(optarg,"%d",&random_drop);
  1059. if(random_drop<0||random_drop>10000)
  1060. {
  1061. mylog(log_fatal,"random_drop must be between 0 10000 \n");
  1062. myexit(-1);
  1063. }
  1064. }
  1065. else if(strcmp(long_options[option_index].name,"report")==0)
  1066. {
  1067. sscanf(optarg,"%d",&report_interval);
  1068. if(report_interval<=0)
  1069. {
  1070. mylog(log_fatal,"report_interval must be >0 \n");
  1071. myexit(-1);
  1072. }
  1073. }
  1074. else if(strcmp(long_options[option_index].name,"sock-buf")==0)
  1075. {
  1076. int tmp=-1;
  1077. sscanf(optarg,"%d",&tmp);
  1078. if(10<=tmp&&tmp<=10*1024)
  1079. {
  1080. socket_buf_size=tmp*1024;
  1081. }
  1082. else
  1083. {
  1084. mylog(log_fatal,"sock-buf value must be between 1 and 10240 (kbyte) \n");
  1085. myexit(-1);
  1086. }
  1087. }
  1088. else
  1089. {
  1090. mylog(log_fatal,"unknown option\n");
  1091. myexit(-1);
  1092. }
  1093. break;
  1094. default:
  1095. mylog(log_fatal,"unknown option <%x>", opt);
  1096. myexit(-1);
  1097. }
  1098. }
  1099. if (no_l)
  1100. mylog(log_fatal,"error: -i not found\n");
  1101. if (no_r)
  1102. mylog(log_fatal,"error: -o not found\n");
  1103. if (no_l || no_r)
  1104. myexit(-1);
  1105. if (is_client == 0 && is_server == 0)
  1106. {
  1107. mylog(log_fatal,"-s -c hasnt been set\n");
  1108. myexit(-1);
  1109. }
  1110. if (is_client == 1 && is_server == 1)
  1111. {
  1112. mylog(log_fatal,"-s -c cant be both set\n");
  1113. myexit(-1);
  1114. }
  1115. }
  1116. int main(int argc, char *argv[])
  1117. {
  1118. assert(sizeof(u64_t)==8);
  1119. assert(sizeof(i64_t)==8);
  1120. assert(sizeof(u32_t)==4);
  1121. assert(sizeof(i32_t)==4);
  1122. dup2(1, 2); //redirect stderr to stdout
  1123. int i, j, k;
  1124. process_arg(argc,argv);
  1125. init_random_number_fd();
  1126. anti_replay.prepare();
  1127. remote_address_uint32=inet_addr(remote_address);
  1128. if(!multi_process_mode)
  1129. {
  1130. event_loop();
  1131. }
  1132. else
  1133. {
  1134. }
  1135. return 0;
  1136. }
  1137. }