classic.cpp 30 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274
  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. }//namespace end