misc.cpp 24 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100
  1. /*
  2. * misc.cpp
  3. *
  4. * Created on: Oct 26, 2017
  5. * Author: root
  6. */
  7. #include "misc.h"
  8. char fifo_file[1000]="";
  9. int mtu_warn=1350;
  10. int disable_mtu_warn=1;
  11. int disable_fec=0;
  12. int disable_checksum=0;
  13. int debug_force_flush_fec=0;
  14. int jitter_min=0*1000;
  15. int jitter_max=0*1000;
  16. int output_interval_min=0*1000;
  17. int output_interval_max=0*1000;
  18. int fix_latency=0;
  19. char interface_string[16];
  20. bool has_b=false;
  21. address_t local_addr,remote_addr,bind_addr;
  22. //u32_t local_ip_uint32,remote_ip_uint32=0;
  23. //char local_ip[100], remote_ip[100];
  24. //int local_port = -1, remote_port = -1;
  25. conn_manager_t conn_manager;
  26. delay_manager_t delay_manager;
  27. fd_manager_t fd_manager;
  28. int time_mono_test=0;
  29. int delay_capacity=0;
  30. char sub_net[100]="10.22.22.0";
  31. u32_t sub_net_uint32=0;
  32. char tun_dev[100]="";
  33. int keep_reconnect=0;
  34. int tun_mtu=1500;
  35. int mssfix=default_mtu;
  36. int manual_set_tun=0;
  37. int persist_tun=0;
  38. char rs_par_str[rs_str_len]="20:10";
  39. int from_normal_to_fec(conn_info_t & conn_info,char *data,int len,int & out_n,char **&out_arr,int *&out_len,my_time_t *&out_delay)
  40. {
  41. static my_time_t out_delay_buf[max_fec_packet_num+100]={0};
  42. //static int out_len_buf[max_fec_packet_num+100]={0};
  43. //static int counter=0;
  44. out_delay=out_delay_buf;
  45. //out_len=out_len_buf;
  46. inner_stat_t &inner_stat=conn_info.stat.normal_to_fec;
  47. if(disable_fec)
  48. {
  49. if(data==0)
  50. {
  51. out_n=0;
  52. return 0;
  53. }
  54. //assert(data!=0);
  55. inner_stat.input_packet_num++;
  56. inner_stat.input_packet_size+=len;
  57. inner_stat.output_packet_num++;
  58. inner_stat.output_packet_size+=len;
  59. out_n=1;
  60. static char *data_static;
  61. data_static=data;
  62. static int len_static;
  63. len_static=len;
  64. out_arr=&data_static;
  65. out_len=&len_static;
  66. out_delay[0]=0;
  67. }
  68. else
  69. {
  70. if(data!=0)
  71. {
  72. inner_stat.input_packet_num++;
  73. inner_stat.input_packet_size+=len;
  74. }
  75. //counter++;
  76. conn_info.fec_encode_manager.input(data,len);
  77. //if(counter%5==0)
  78. //conn_info.fec_encode_manager.input(0,0);
  79. //int n;
  80. //char **s_arr;
  81. //int s_len;
  82. conn_info.fec_encode_manager.output(out_n,out_arr,out_len);
  83. if(out_n>0)
  84. {
  85. my_time_t common_latency=0;
  86. my_time_t first_packet_time=conn_info.fec_encode_manager.get_first_packet_time();
  87. if(fix_latency==1&&conn_info.fec_encode_manager.get_type()==0)
  88. {
  89. my_time_t current_time=get_current_time_us();
  90. my_time_t tmp;
  91. assert(first_packet_time!=0);
  92. //mylog(log_info,"current_time=%llu first_packlet_time=%llu fec_pending_time=%llu\n",current_time,first_packet_time,(my_time_t)fec_pending_time);
  93. if((my_time_t)conn_info.fec_encode_manager.get_pending_time() >=(current_time - first_packet_time))
  94. {
  95. tmp=(my_time_t)conn_info.fec_encode_manager.get_pending_time()-(current_time - first_packet_time);
  96. //mylog(log_info,"tmp=%llu\n",tmp);
  97. }
  98. else
  99. {
  100. tmp=0;
  101. //mylog(log_info,"0\n");
  102. }
  103. common_latency+=tmp;
  104. }
  105. common_latency+=random_between(jitter_min,jitter_max);
  106. out_delay_buf[0]=common_latency;
  107. for(int i=1;i<out_n;i++)
  108. {
  109. out_delay_buf[i]=out_delay_buf[i-1]+ (my_time_t)( random_between(output_interval_min,output_interval_max)/(out_n-1) );
  110. }
  111. }
  112. if(out_n>0)
  113. {
  114. log_bare(log_trace,"seq= %u ",read_u32(out_arr[0]));
  115. }
  116. for(int i=0;i<out_n;i++)
  117. {
  118. inner_stat.output_packet_num++;
  119. inner_stat.output_packet_size+=out_len[i];
  120. log_bare(log_trace,"%d ",out_len[i]);
  121. }
  122. log_bare(log_trace,"\n");
  123. }
  124. mylog(log_trace,"from_normal_to_fec input_len=%d,output_n=%d\n",len,out_n);
  125. //for(int i=0;i<n;i++)
  126. //{
  127. //delay_send(0,dest,s_arr[i],s_len);
  128. //}
  129. //delay_send(0,dest,data,len);
  130. //delay_send(1000*1000,dest,data,len);
  131. return 0;
  132. }
  133. int from_fec_to_normal(conn_info_t & conn_info,char *data,int len,int & out_n,char **&out_arr,int *&out_len,my_time_t *&out_delay)
  134. {
  135. static my_time_t out_delay_buf[max_blob_packet_num+100]={0};
  136. out_delay=out_delay_buf;
  137. inner_stat_t &inner_stat=conn_info.stat.fec_to_normal;
  138. if(disable_fec)
  139. {
  140. assert(data!=0);
  141. inner_stat.input_packet_num++;
  142. inner_stat.input_packet_size+=len;
  143. inner_stat.output_packet_num++;
  144. inner_stat.output_packet_size+=len;
  145. if(data==0)
  146. {
  147. out_n=0;
  148. return 0;
  149. }
  150. out_n=1;
  151. static char *data_static;
  152. data_static=data;
  153. static int len_static;
  154. len_static=len;
  155. out_arr=&data_static;
  156. out_len=&len_static;
  157. out_delay[0]=0;
  158. }
  159. else
  160. {
  161. if(data!=0)
  162. {
  163. inner_stat.input_packet_num++;
  164. inner_stat.input_packet_size+=len;
  165. }
  166. conn_info.fec_decode_manager.input(data,len);
  167. //int n;char ** s_arr;int* len_arr;
  168. conn_info.fec_decode_manager.output(out_n,out_arr,out_len);
  169. for(int i=0;i<out_n;i++)
  170. {
  171. out_delay_buf[i]=0;
  172. inner_stat.output_packet_num++;
  173. inner_stat.output_packet_size+=out_len[i];
  174. }
  175. }
  176. mylog(log_trace,"from_fec_to_normal input_len=%d,output_n=%d,input_seq=%u\n",len,out_n,read_u32(data));
  177. // printf("<n:%d>",n);
  178. /*
  179. for(int i=0;i<n;i++)
  180. {
  181. delay_send(0,dest,s_arr[i],len_arr[i]);
  182. //s_arr[i][len_arr[i]]=0;
  183. //printf("<%s>\n",s_arr[i]);
  184. }*/
  185. //my_send(dest,data,len);
  186. return 0;
  187. }
  188. int delay_send(my_time_t delay,const dest_t &dest,char *data,int len)
  189. {
  190. //int rand=random()%100;
  191. //mylog(log_info,"rand = %d\n",rand);
  192. if (dest.cook&&random_drop != 0) {
  193. if (get_fake_random_number() % 10000 < (u32_t) random_drop) {
  194. return 0;
  195. }
  196. }
  197. return delay_manager.add(delay,dest,data,len);;
  198. }
  199. int print_parameter()
  200. {
  201. mylog(log_info,"jitter_min=%d jitter_max=%d output_interval_min=%d output_interval_max=%d fec_timeout=%d fec_mtu=%d fec_queue_len=%d fec_mode=%d\n",
  202. jitter_min/1000,jitter_max/1000,output_interval_min/1000,output_interval_max/1000,g_fec_par.timeout/1000,g_fec_par.mtu,g_fec_par.queue_len,g_fec_par.mode);
  203. mylog(log_info,"fec_str=%s\n",rs_par_str);
  204. mylog(log_info,"fec_inner_parameter=%s\n",g_fec_par.rs_to_str());
  205. return 0;
  206. }
  207. int handle_command(char *s)
  208. {
  209. int len=strlen(s);
  210. while(len>=1&&s[len-1]=='\n')
  211. s[len-1]=0;
  212. mylog(log_info,"got data from fifo,len=%d,s=[%s]\n",len,s);
  213. int a=-1,b=-1;
  214. if(strncmp(s,"fec",strlen("fec"))==0)
  215. {
  216. mylog(log_info,"got command [fec]\n");
  217. char tmp_str[max_fec_packet_num*10+100];
  218. fec_parameter_t tmp_par;
  219. sscanf(s,"fec %s",tmp_str);
  220. /*
  221. if(a<1||b<0||a+b>254)
  222. {
  223. mylog(log_warn,"invaild value\n");
  224. return -1;
  225. }*/
  226. int ret=tmp_par.rs_from_str(tmp_str);
  227. if(ret!=0)
  228. {
  229. mylog(log_warn,"failed to parse [%s]\n",tmp_str);
  230. return -1;
  231. }
  232. int version=g_fec_par.version;
  233. g_fec_par.clone(tmp_par);
  234. g_fec_par.version=version;
  235. g_fec_par.version++;
  236. strcpy(rs_par_str,tmp_str);
  237. //g_fec_data_num=a;
  238. //g_fec_redundant_num=b;
  239. }
  240. else if(strncmp(s,"mtu",strlen("mtu"))==0)
  241. {
  242. mylog(log_info,"got command [mtu]\n");
  243. sscanf(s,"mtu %d",&a);
  244. if(a<100||a>2000)
  245. {
  246. mylog(log_warn,"invaild value\n");
  247. return -1;
  248. }
  249. g_fec_par.mtu=a;
  250. }
  251. else if(strncmp(s,"queue-len",strlen("queue-len"))==0)
  252. {
  253. mylog(log_info,"got command [queue-len]\n");
  254. sscanf(s,"queue-len %d",&a);
  255. if(a<1||a>10000)
  256. {
  257. mylog(log_warn,"invaild value\n");
  258. return -1;
  259. }
  260. g_fec_par.queue_len=a;
  261. }
  262. else if(strncmp(s,"mode",strlen("mode"))==0)
  263. {
  264. mylog(log_info,"got command [mode]\n");
  265. sscanf(s,"mode %d",&a);
  266. if(a!=0&&a!=1)
  267. {
  268. mylog(log_warn,"invaild value\n");
  269. return -1;
  270. }
  271. if(g_fec_par.mode!=a)
  272. {
  273. g_fec_par.mode=a;
  274. assert(g_fec_par.rs_from_str(rs_par_str)==0); //re parse rs_par_str,not necessary at the moment, for futher use
  275. g_fec_par.version++;
  276. }
  277. }
  278. else if(strncmp(s,"timeout",strlen("timeout"))==0)
  279. {
  280. mylog(log_info,"got command [timeout]\n");
  281. sscanf(s,"timeout %d",&a);
  282. if(a<0||a>1000)
  283. {
  284. mylog(log_warn,"invaild value\n");
  285. return -1;
  286. }
  287. g_fec_par.timeout=a*1000;
  288. }
  289. else
  290. {
  291. mylog(log_info,"unknown command\n");
  292. }
  293. print_parameter();
  294. return 0;
  295. }
  296. static void empty_cb(struct ev_loop *loop, struct ev_timer *watcher, int revents)
  297. {
  298. }
  299. int unit_test()
  300. {
  301. {
  302. union test_t
  303. {
  304. u64_t u64;
  305. char arry[8];
  306. }test111;
  307. assert((void*)&test111.u64==(void*)&test111.arry[0]);
  308. //printf("%llx,%llx\n",&ttt.u64,&ttt.arry[0]);
  309. // printf("%lld\n",get_fake_random_number_64());
  310. // printf("%lld\n",get_fake_random_number_64());
  311. // printf("%lld\n",get_fake_random_number_64());
  312. // printf("%x\n",get_fake_random_number());
  313. // printf("%x\n",get_fake_random_number());
  314. // printf("%x\n",get_fake_random_number());
  315. char buf[10];
  316. get_fake_random_chars(buf,10);
  317. for(int i=0;i<10;i++)
  318. printf("<%d>",(int)buf[i]);
  319. printf("\n");
  320. get_fake_random_chars(buf,10);
  321. for(int i=0;i<10;i++)
  322. printf("<%d>",(int)buf[i]);
  323. printf("\n");
  324. }
  325. int i,j,k;
  326. void *code=fec_new(3,6);
  327. char arr[6][100]=
  328. {
  329. "aaa","bbb","ccc"
  330. ,"ddd","eee","fff"
  331. };
  332. char *data[6];
  333. for(i=0;i<6;i++)
  334. {
  335. data[i]=arr[i];
  336. }
  337. rs_encode2(3,6,data,3);
  338. //printf("%d %d",(int)(unsigned char)arr[5][0],(int)('a'^'b'^'c'^'d'^'e'));
  339. for(i=0;i<6;i++)
  340. {
  341. printf("<%s>",data[i]);
  342. }
  343. data[0]=0;
  344. //data[1]=0;
  345. //data[5]=0;
  346. int ret=rs_decode2(3,6,data,3);
  347. printf("ret:%d\n",ret);
  348. for(i=0;i<6;i++)
  349. {
  350. printf("<%s>",data[i]);
  351. }
  352. fec_free(code);
  353. char arr2[6][100]=
  354. {
  355. "aaa11111","","ccc333333333"
  356. ,"ddd444","eee5555","ff6666"
  357. };
  358. blob_encode_t blob_encode;
  359. for(int i=0;i<6;i++)
  360. blob_encode.input(arr2[i],strlen(arr2[i]));
  361. char **output;
  362. int shard_len;
  363. blob_encode.output(7,output,shard_len);
  364. printf("<shard_len:%d>",shard_len);
  365. blob_decode_t blob_decode;
  366. for(int i=0;i<7;i++)
  367. {
  368. blob_decode.input(output[i],shard_len);
  369. }
  370. char **decode_output;
  371. int * len_arr;
  372. int num;
  373. ret=blob_decode.output(num,decode_output,len_arr);
  374. printf("<num:%d,ret:%d>\n",num,ret);
  375. for(int i=0;i<num;i++)
  376. {
  377. char buf[1000]={0};
  378. memcpy(buf,decode_output[i],len_arr[i]);
  379. printf("<%d:%s>",len_arr[i],buf);
  380. }
  381. printf("\n");
  382. static fec_encode_manager_t fec_encode_manager;
  383. static fec_decode_manager_t fec_decode_manager;
  384. //dynamic_update_fec=0;
  385. fec_encode_manager.set_loop_and_cb(ev_default_loop(0),empty_cb);
  386. {
  387. string a = "11111";
  388. string b = "22";
  389. string c = "33333333";
  390. fec_encode_manager.input((char *) a.c_str(), a.length());
  391. fec_encode_manager.input((char *) b.c_str(), b.length());
  392. fec_encode_manager.input((char *) c.c_str(), c.length());
  393. fec_encode_manager.input(0, 0);
  394. int n;
  395. char **s_arr;
  396. int *len;
  397. fec_encode_manager.output(n,s_arr,len);
  398. printf("<n:%d,len:%d>",n,len[0]);
  399. for(int i=0;i<n;i++)
  400. {
  401. fec_decode_manager.input(s_arr[i],len[i]);
  402. }
  403. {
  404. int n;char ** s_arr;int* len_arr;
  405. fec_decode_manager.output(n,s_arr,len_arr);
  406. printf("<n:%d>",n);
  407. for(int i=0;i<n;i++)
  408. {
  409. s_arr[i][len_arr[i]]=0;
  410. printf("<%s>\n",s_arr[i]);
  411. }
  412. }
  413. }
  414. {
  415. string a = "aaaaaaa";
  416. string b = "bbbbbbbbbbbbb";
  417. string c = "ccc";
  418. fec_encode_manager.input((char *) a.c_str(), a.length());
  419. fec_encode_manager.input((char *) b.c_str(), b.length());
  420. fec_encode_manager.input((char *) c.c_str(), c.length());
  421. fec_encode_manager.input(0, 0);
  422. int n;
  423. char **s_arr;
  424. int * len;
  425. fec_encode_manager.output(n,s_arr,len);
  426. printf("<n:%d,len:%d>",n,len[0]);
  427. for(int i=0;i<n;i++)
  428. {
  429. if(i==1||i==3||i==5||i==0)
  430. fec_decode_manager.input(s_arr[i],len[i]);
  431. }
  432. {
  433. int n;char ** s_arr;int* len_arr;
  434. fec_decode_manager.output(n,s_arr,len_arr);
  435. printf("<n:%d>",n);
  436. for(int i=0;i<n;i++)
  437. {
  438. s_arr[i][len_arr[i]]=0;
  439. printf("<%s>\n",s_arr[i]);
  440. }
  441. }
  442. }
  443. printf("ok here.\n");
  444. for(int i=0;i<10;i++)
  445. {
  446. string a = "aaaaaaaaaaaaaaaaaaaaaaa";
  447. string b = "bbbbbbbbbbbbb";
  448. string c = "cccccccccccccccccc";
  449. printf("======\n");
  450. int n;
  451. char **s_arr;
  452. int * len;
  453. fec_decode_manager.output(n,s_arr,len);
  454. //fec_encode_manager.reset_fec_parameter(3,2,g_fec_mtu,g_fec_queue_len,g_fec_timeout,1);
  455. fec_parameter_t &fec_par=fec_encode_manager.get_fec_par();
  456. fec_par.mtu=g_fec_par.mtu;
  457. fec_par.queue_len=g_fec_par.queue_len;
  458. fec_par.timeout=g_fec_par.timeout;
  459. fec_par.mode=1;
  460. fec_par.rs_from_str((char *)"3:2");
  461. fec_encode_manager.input((char *) a.c_str(), a.length());
  462. fec_encode_manager.output(n,s_arr,len);
  463. printf("n=<%d>\n",n);
  464. assert(n==1);
  465. fec_decode_manager.input(s_arr[0],len[0]);
  466. fec_decode_manager.output(n,s_arr,len);
  467. assert(n==1);
  468. printf("%s\n",s_arr[0]);
  469. fec_encode_manager.input((char *) b.c_str(), b.length());
  470. fec_encode_manager.output(n,s_arr,len);
  471. assert(n==1);
  472. //fec_decode_manager.input(s_arr[0],len[0]);
  473. fec_encode_manager.input((char *) c.c_str(), c.length());
  474. fec_encode_manager.output(n,s_arr,len);
  475. assert(n==3);
  476. fec_decode_manager.input(s_arr[0],len[0]);
  477. //printf("n=%d\n",n);
  478. {
  479. int n;
  480. char **s_arr;
  481. int * len;
  482. fec_decode_manager.output(n,s_arr,len);
  483. assert(n==1);
  484. printf("%s\n",s_arr[0]);
  485. }
  486. fec_decode_manager.input(s_arr[1],len[1]);
  487. {
  488. int n;
  489. char **s_arr;
  490. int * len;
  491. fec_decode_manager.output(n,s_arr,len);
  492. assert(n==1);
  493. printf("n=%d\n",n);
  494. s_arr[0][len[0]]=0;
  495. printf("%s\n",s_arr[0]);
  496. }
  497. }
  498. myexit(0);
  499. return 0;
  500. }
  501. void process_arg(int argc, char *argv[])
  502. {
  503. int is_client=0,is_server=0;
  504. int i, j, k;
  505. int opt;
  506. static struct option long_options[] =
  507. {
  508. {"log-level", required_argument, 0, 1},
  509. {"log-position", no_argument, 0, 1},
  510. {"disable-color", no_argument, 0, 1},
  511. {"enable-color", no_argument, 0, 1},
  512. {"disable-filter", no_argument, 0, 1},
  513. {"disable-fec", no_argument, 0, 1},
  514. {"disable-obscure", no_argument, 0, 1},
  515. {"disable-xor", no_argument, 0, 1},
  516. {"disable-checksum", no_argument, 0, 1},
  517. {"fix-latency", no_argument, 0, 1},
  518. {"sock-buf", required_argument, 0, 1},
  519. {"random-drop", required_argument, 0, 1},
  520. {"report", required_argument, 0, 1},
  521. {"delay-capacity", required_argument, 0, 1},
  522. {"mtu", required_argument, 0, 1},
  523. {"interface", required_argument, 0, 1},
  524. {"mode", required_argument, 0,1},
  525. {"timeout", required_argument, 0,1},
  526. {"decode-buf", required_argument, 0,1},
  527. {"queue-len", required_argument, 0,'q'},
  528. {"fec", required_argument, 0,'f'},
  529. {"jitter", required_argument, 0,'j'},
  530. {"bind", required_argument, 0,'b'},
  531. {"header-overhead", required_argument, 0, 1},
  532. //{"debug-fec", no_argument, 0, 1},
  533. {"debug-fec-enc", no_argument, 0, 1},
  534. {"debug-fec-dec", no_argument, 0, 1},
  535. {"fifo", required_argument, 0, 1},
  536. {"sub-net", required_argument, 0, 1},
  537. {"tun-dev", required_argument, 0, 1},
  538. {"tun-mtu", required_argument, 0, 1},
  539. {"mssfix", required_argument, 0, 1},
  540. {"keep-reconnect", no_argument, 0, 1},
  541. {"persist-tun", no_argument, 0, 1},
  542. {"manual-set-tun", no_argument, 0, 1},
  543. {"interval", required_argument, 0,'i'},
  544. {NULL, 0, 0, 0}
  545. };
  546. int option_index = 0;
  547. assert(g_fec_par.rs_from_str(rs_par_str)==0);
  548. for (i = 0; i < argc; i++)
  549. {
  550. if(strcmp(argv[i],"--unit-test")==0)
  551. {
  552. unit_test();
  553. myexit(0);
  554. }
  555. }
  556. for (i = 0; i < argc; i++)
  557. {
  558. if(strcmp(argv[i],"--log-level")==0)
  559. {
  560. if(i<argc -1)
  561. {
  562. sscanf(argv[i+1],"%d",&log_level);
  563. if(0<=log_level&&log_level<log_end)
  564. {
  565. }
  566. else
  567. {
  568. log_bare(log_fatal,"invalid log_level\n");
  569. myexit(-1);
  570. }
  571. }
  572. }
  573. if(strcmp(argv[i],"--enable-color")==0)
  574. {
  575. enable_log_color=1;
  576. }
  577. if(strcmp(argv[i],"--disable-color")==0)
  578. {
  579. enable_log_color=0;
  580. }
  581. }
  582. mylog(log_info,"argc=%d ", argc);
  583. for (i = 0; i < argc; i++) {
  584. log_bare(log_info, "%s ", argv[i]);
  585. }
  586. log_bare(log_info, "\n");
  587. int no_l = 1, no_r = 1;
  588. while ((opt = getopt_long(argc, argv, "l:r:hcsk:j:f:p:n:i:q:b",long_options,&option_index)) != -1)
  589. {
  590. //string opt_key;
  591. //opt_key+=opt;
  592. switch (opt)
  593. {
  594. case 'k':
  595. sscanf(optarg,"%s\n",key_string);
  596. mylog(log_debug,"key=%s\n",key_string);
  597. if(strlen(key_string)==0)
  598. {
  599. mylog(log_fatal,"key len=0??\n");
  600. myexit(-1);
  601. }
  602. break;
  603. case 'j':
  604. if (strchr(optarg, ':') == 0)
  605. {
  606. int jitter;
  607. sscanf(optarg,"%d\n",&jitter);
  608. if(jitter<0 ||jitter>1000*10)
  609. {
  610. mylog(log_fatal,"jitter must be between 0 and 10,000(10 second)\n");
  611. myexit(-1);
  612. }
  613. jitter_min=0;
  614. jitter_max=jitter;
  615. }
  616. else
  617. {
  618. sscanf(optarg,"%d:%d\n",&jitter_min,&jitter_max);
  619. if(jitter_min<0 ||jitter_max<0||jitter_min>jitter_max)
  620. {
  621. mylog(log_fatal," must satisfy 0<=jmin<=jmax\n");
  622. myexit(-1);
  623. }
  624. }
  625. jitter_min*=1000;
  626. jitter_max*=1000;
  627. break;
  628. case 'i':
  629. if (strchr(optarg, ':') == 0)
  630. {
  631. int output_interval=-1;
  632. sscanf(optarg,"%d\n",&output_interval);
  633. if(output_interval<0||output_interval>1000*10)
  634. {
  635. mylog(log_fatal,"output_interval must be between 0 and 10,000(10 second)\n");
  636. myexit(-1);
  637. }
  638. output_interval_min=output_interval_max=output_interval;
  639. }
  640. else
  641. {
  642. sscanf(optarg,"%d:%d\n",&output_interval_min,&output_interval_max);
  643. if(output_interval_min<0 ||output_interval_max<0||output_interval_min>output_interval_max)
  644. {
  645. mylog(log_fatal," must satisfy 0<=output_interval_min<=output_interval_max\n");
  646. myexit(-1);
  647. }
  648. }
  649. output_interval_min*=1000;
  650. output_interval_max*=1000;
  651. break;
  652. case 'f':
  653. if (strchr(optarg, ':') == 0)
  654. {
  655. mylog(log_fatal,"invalid format for f");
  656. myexit(-1);
  657. }
  658. else
  659. {
  660. strcpy(rs_par_str,optarg);
  661. //sscanf(optarg,"%d:%d\n",&g_fec_data_num,&g_fec_redundant_num);
  662. /*
  663. if(g_fec_data_num<1 ||g_fec_redundant_num<0||g_fec_data_num+g_fec_redundant_num>254)
  664. {
  665. mylog(log_fatal,"fec_data_num<1 ||fec_redundant_num<0||fec_data_num+fec_redundant_num>254\n");
  666. myexit(-1);
  667. }*/
  668. }
  669. break;
  670. case 'q':
  671. sscanf(optarg,"%d",&g_fec_par.queue_len);
  672. if(g_fec_par.queue_len<1||g_fec_par.queue_len>10000)
  673. {
  674. mylog(log_fatal,"fec_pending_num should be between 1 and 10000\n");
  675. myexit(-1);
  676. }
  677. break;
  678. case 'c':
  679. is_client = 1;
  680. break;
  681. case 's':
  682. is_server = 1;
  683. break;
  684. case 'l':
  685. no_l = 0;
  686. local_addr.from_str(optarg);
  687. break;
  688. case 'r':
  689. no_r = 0;
  690. remote_addr.from_str(optarg);
  691. break;
  692. case 'b':
  693. has_b = true;
  694. bind_addr.from_str(optarg);
  695. break;
  696. case 'h':
  697. break;
  698. case 1:
  699. if(strcmp(long_options[option_index].name,"log-level")==0)
  700. {
  701. }
  702. else if(strcmp(long_options[option_index].name,"disable-filter")==0)
  703. {
  704. disable_replay_filter=1;
  705. //enable_log_color=0;
  706. }
  707. else if(strcmp(long_options[option_index].name,"disable-color")==0)
  708. {
  709. //enable_log_color=0;
  710. }
  711. else if(strcmp(long_options[option_index].name,"enable-color")==0)
  712. {
  713. //enable_log_color=0;
  714. }
  715. else if(strcmp(long_options[option_index].name,"disable-fec")==0)
  716. {
  717. disable_fec=1;
  718. }
  719. else if(strcmp(long_options[option_index].name,"disable-obscure")==0)
  720. {
  721. mylog(log_info,"obscure disabled\n");
  722. disable_obscure=1;
  723. }
  724. else if(strcmp(long_options[option_index].name,"disable-xor")==0)
  725. {
  726. mylog(log_info,"xor disabled\n");
  727. disable_xor=1;
  728. }
  729. else if(strcmp(long_options[option_index].name,"disable-checksum")==0)
  730. {
  731. disable_checksum=1;
  732. mylog(log_warn,"checksum disabled\n");
  733. }
  734. else if(strcmp(long_options[option_index].name,"fix-latency")==0)
  735. {
  736. mylog(log_info,"fix-latency enabled\n");
  737. fix_latency=1;
  738. }
  739. else if(strcmp(long_options[option_index].name,"log-position")==0)
  740. {
  741. enable_log_position=1;
  742. }
  743. else if(strcmp(long_options[option_index].name,"random-drop")==0)
  744. {
  745. sscanf(optarg,"%d",&random_drop);
  746. if(random_drop<0||random_drop>10000)
  747. {
  748. mylog(log_fatal,"random_drop must be between 0 10000 \n");
  749. myexit(-1);
  750. }
  751. mylog(log_info,"random_drop=%d\n",random_drop);
  752. }
  753. else if(strcmp(long_options[option_index].name,"delay-capacity")==0)
  754. {
  755. sscanf(optarg,"%d",&delay_capacity);
  756. if(delay_capacity<0)
  757. {
  758. mylog(log_fatal,"delay_capacity must be >=0 \n");
  759. myexit(-1);
  760. }
  761. }
  762. else if(strcmp(long_options[option_index].name,"report")==0)
  763. {
  764. sscanf(optarg,"%d",&report_interval);
  765. if(report_interval<=0)
  766. {
  767. mylog(log_fatal,"report_interval must be >0 \n");
  768. myexit(-1);
  769. }
  770. }
  771. else if(strcmp(long_options[option_index].name,"sock-buf")==0)
  772. {
  773. int tmp=-1;
  774. sscanf(optarg,"%d",&tmp);
  775. if(10<=tmp&&tmp<=10*1024)
  776. {
  777. socket_buf_size=tmp*1024;
  778. }
  779. else
  780. {
  781. mylog(log_fatal,"sock-buf value must be between 1 and 10240 (kbyte) \n");
  782. myexit(-1);
  783. }
  784. }
  785. else if(strcmp(long_options[option_index].name,"decode-buf")==0)
  786. {
  787. sscanf(optarg,"%d",&fec_buff_num);
  788. if(fec_buff_num<300 || fec_buff_num>20000)
  789. {
  790. mylog(log_fatal,"decode-buf value must be between 300 and 20000 (kbyte) \n");
  791. myexit(-1);
  792. }
  793. mylog(log_info,"decode-buf=%d\n",fec_buff_num);
  794. }
  795. else if(strcmp(long_options[option_index].name,"mode")==0)
  796. {
  797. sscanf(optarg,"%d",&g_fec_par.mode);
  798. if(g_fec_par.mode!=0&&g_fec_par.mode!=1)
  799. {
  800. mylog(log_fatal,"mode should be 0 or 1\n");
  801. myexit(-1);
  802. }
  803. }
  804. else if(strcmp(long_options[option_index].name,"mtu")==0)
  805. {
  806. sscanf(optarg,"%d",&g_fec_par.mtu);
  807. if(g_fec_par.mtu<100||g_fec_par.mtu>2000)
  808. {
  809. mylog(log_fatal,"fec_mtu should be between 100 and 2000\n");
  810. myexit(-1);
  811. }
  812. }
  813. else if(strcmp(long_options[option_index].name,"interface")==0)
  814. {
  815. sscanf(optarg,"%s\n",interface_string);
  816. mylog(log_debug,"interface=%s\n",interface_string);
  817. if(strlen(interface_string)==0)
  818. {
  819. mylog(log_fatal,"interface_string len=0??\n");
  820. myexit(-1);
  821. }
  822. }
  823. else if(strcmp(long_options[option_index].name,"timeout")==0)
  824. {
  825. sscanf(optarg,"%d",&g_fec_par.timeout);
  826. if(g_fec_par.timeout<0||g_fec_par.timeout>1000)
  827. {
  828. mylog(log_fatal,"fec_pending_time should be between 0 and 1000(1s)\n");
  829. myexit(-1);
  830. }
  831. g_fec_par.timeout*=1000;
  832. }
  833. else if(strcmp(long_options[option_index].name,"debug-fec-enc")==0)
  834. {
  835. debug_fec_enc=1;
  836. mylog(log_info,"debug_fec_enc enabled\n");
  837. }
  838. else if(strcmp(long_options[option_index].name,"debug-fec-dec")==0)
  839. {
  840. debug_fec_dec=1;
  841. mylog(log_info,"debug_fec_dec enabled\n");
  842. }
  843. else if(strcmp(long_options[option_index].name,"fifo")==0)
  844. {
  845. sscanf(optarg,"%s",fifo_file);
  846. mylog(log_info,"fifo_file =%s \n",fifo_file);
  847. }
  848. else if(strcmp(long_options[option_index].name,"keep-reconnect")==0)
  849. {
  850. keep_reconnect=1;
  851. mylog(log_info,"keep_reconnect enabled\n");
  852. }
  853. else if(strcmp(long_options[option_index].name,"manual-set-tun")==0)
  854. {
  855. manual_set_tun=1;
  856. mylog(log_info,"manual_set_tun enabled\n");
  857. }
  858. else if(strcmp(long_options[option_index].name,"persist-tun")==0)
  859. {
  860. persist_tun=1;
  861. mylog(log_info,"persist_tun enabled\n");
  862. }
  863. else if(strcmp(long_options[option_index].name,"sub-net")==0)
  864. {
  865. sscanf(optarg,"%s",sub_net);
  866. mylog(log_info,"sub_net %s\n",sub_net);
  867. }
  868. else if(strcmp(long_options[option_index].name,"tun-dev")==0)
  869. {
  870. sscanf(optarg,"%s",tun_dev);
  871. mylog(log_info,"tun_dev=%s\n",tun_dev);
  872. }
  873. else if(strcmp(long_options[option_index].name,"tun-mtu")==0)
  874. {
  875. sscanf(optarg,"%d",&tun_mtu);
  876. mylog(log_warn,"changed tun_mtu,tun_mtu=%d\n",tun_mtu);
  877. }
  878. else if(strcmp(long_options[option_index].name,"header-overhead")==0)
  879. {
  880. sscanf(optarg,"%d",&header_overhead);
  881. mylog(log_warn,"changed header_overhead,header_overhead=%d\n",header_overhead);
  882. }
  883. else if(strcmp(long_options[option_index].name,"mssfix")==0)
  884. {
  885. sscanf(optarg,"%d",&mssfix);
  886. mylog(log_warn,"mssfix=%d\n",mssfix);
  887. }
  888. else
  889. {
  890. mylog(log_fatal,"unknown option\n");
  891. myexit(-1);
  892. }
  893. break;
  894. default:
  895. mylog(log_fatal,"unknown option <%x>", opt);
  896. myexit(-1);
  897. }
  898. }
  899. if (is_client == 0 && is_server == 0)
  900. {
  901. mylog(log_fatal,"-s -c hasnt been set\n");
  902. myexit(-1);
  903. }
  904. if (is_client == 1 && is_server == 1)
  905. {
  906. mylog(log_fatal,"-s -c cant be both set\n");
  907. myexit(-1);
  908. }
  909. if(is_client==1)
  910. {
  911. program_mode=client_mode;
  912. }
  913. else
  914. {
  915. program_mode=server_mode;
  916. }
  917. if(working_mode==tunnel_mode)
  918. {
  919. if (no_l)
  920. mylog(log_fatal,"error: -l not found\n");
  921. if (no_r)
  922. mylog(log_fatal,"error: -r not found\n");
  923. if (no_l || no_r)
  924. myexit(-1);
  925. }
  926. else if(working_mode==tun_dev_mode)
  927. {
  928. if(program_mode==client_mode&&no_r)
  929. {
  930. mylog(log_fatal,"error: -r not found\n");
  931. myexit(-1);
  932. }
  933. else if(program_mode==server_mode&&no_l)
  934. {
  935. mylog(log_fatal,"error: -l not found\n");
  936. myexit(-1);
  937. }
  938. }
  939. int ret=g_fec_par.rs_from_str(rs_par_str);
  940. if(ret!=0)
  941. {
  942. mylog(log_fatal,"failed to parse [%s]\n",rs_par_str);
  943. myexit(-1);
  944. }
  945. print_parameter();
  946. }