misc.cpp 24 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102
  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. address_t local_addr,remote_addr;
  20. address_t *out_addr=0;
  21. char *out_interface=0;
  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. g_fec_par.clone_fec(tmp_par);
  233. strcpy(rs_par_str,tmp_str);
  234. //g_fec_data_num=a;
  235. //g_fec_redundant_num=b;
  236. }
  237. else if(strncmp(s,"mtu",strlen("mtu"))==0)
  238. {
  239. mylog(log_info,"got command [mtu]\n");
  240. sscanf(s,"mtu %d",&a);
  241. if(a<100||a>2000)
  242. {
  243. mylog(log_warn,"invaild value\n");
  244. return -1;
  245. }
  246. g_fec_par.mtu=a;
  247. }
  248. else if(strncmp(s,"queue-len",strlen("queue-len"))==0)
  249. {
  250. mylog(log_info,"got command [queue-len]\n");
  251. sscanf(s,"queue-len %d",&a);
  252. if(a<1||a>10000)
  253. {
  254. mylog(log_warn,"invaild value\n");
  255. return -1;
  256. }
  257. g_fec_par.queue_len=a;
  258. }
  259. else if(strncmp(s,"mode",strlen("mode"))==0)
  260. {
  261. mylog(log_info,"got command [mode]\n");
  262. sscanf(s,"mode %d",&a);
  263. if(a!=0&&a!=1)
  264. {
  265. mylog(log_warn,"invaild value\n");
  266. return -1;
  267. }
  268. if(g_fec_par.mode!=a)
  269. {
  270. g_fec_par.mode=a;
  271. assert(g_fec_par.rs_from_str(rs_par_str)==0); //re parse rs_par_str,not necessary at the moment, for futher use
  272. g_fec_par.version++;
  273. }
  274. }
  275. else if(strncmp(s,"timeout",strlen("timeout"))==0)
  276. {
  277. mylog(log_info,"got command [timeout]\n");
  278. sscanf(s,"timeout %d",&a);
  279. if(a<0||a>1000)
  280. {
  281. mylog(log_warn,"invaild value\n");
  282. return -1;
  283. }
  284. g_fec_par.timeout=a*1000;
  285. }
  286. else
  287. {
  288. mylog(log_info,"unknown command\n");
  289. }
  290. print_parameter();
  291. return 0;
  292. }
  293. static void empty_cb(struct ev_loop *loop, struct ev_timer *watcher, int revents)
  294. {
  295. }
  296. int unit_test()
  297. {
  298. {
  299. union test_t
  300. {
  301. u64_t u64;
  302. char arry[8];
  303. }test111;
  304. assert((void*)&test111.u64==(void*)&test111.arry[0]);
  305. //printf("%llx,%llx\n",&ttt.u64,&ttt.arry[0]);
  306. // printf("%lld\n",get_fake_random_number_64());
  307. // printf("%lld\n",get_fake_random_number_64());
  308. // printf("%lld\n",get_fake_random_number_64());
  309. // printf("%x\n",get_fake_random_number());
  310. // printf("%x\n",get_fake_random_number());
  311. // printf("%x\n",get_fake_random_number());
  312. char buf[10];
  313. get_fake_random_chars(buf,10);
  314. for(int i=0;i<10;i++)
  315. printf("<%d>",(int)buf[i]);
  316. printf("\n");
  317. get_fake_random_chars(buf,10);
  318. for(int i=0;i<10;i++)
  319. printf("<%d>",(int)buf[i]);
  320. printf("\n");
  321. }
  322. int i,j,k;
  323. void *code=fec_new(3,6);
  324. char arr[6][100]=
  325. {
  326. "aaa","bbb","ccc"
  327. ,"ddd","eee","fff"
  328. };
  329. char *data[6];
  330. for(i=0;i<6;i++)
  331. {
  332. data[i]=arr[i];
  333. }
  334. rs_encode2(3,6,data,3);
  335. //printf("%d %d",(int)(unsigned char)arr[5][0],(int)('a'^'b'^'c'^'d'^'e'));
  336. for(i=0;i<6;i++)
  337. {
  338. printf("<%s>",data[i]);
  339. }
  340. data[0]=0;
  341. //data[1]=0;
  342. //data[5]=0;
  343. int ret=rs_decode2(3,6,data,3);
  344. printf("ret:%d\n",ret);
  345. for(i=0;i<6;i++)
  346. {
  347. printf("<%s>",data[i]);
  348. }
  349. fec_free(code);
  350. char arr2[6][100]=
  351. {
  352. "aaa11111","","ccc333333333"
  353. ,"ddd444","eee5555","ff6666"
  354. };
  355. blob_encode_t blob_encode;
  356. for(int i=0;i<6;i++)
  357. blob_encode.input(arr2[i],strlen(arr2[i]));
  358. char **output;
  359. int shard_len;
  360. blob_encode.output(7,output,shard_len);
  361. printf("<shard_len:%d>",shard_len);
  362. blob_decode_t blob_decode;
  363. for(int i=0;i<7;i++)
  364. {
  365. blob_decode.input(output[i],shard_len);
  366. }
  367. char **decode_output;
  368. int * len_arr;
  369. int num;
  370. ret=blob_decode.output(num,decode_output,len_arr);
  371. printf("<num:%d,ret:%d>\n",num,ret);
  372. for(int i=0;i<num;i++)
  373. {
  374. char buf[1000]={0};
  375. memcpy(buf,decode_output[i],len_arr[i]);
  376. printf("<%d:%s>",len_arr[i],buf);
  377. }
  378. printf("\n");
  379. static fec_encode_manager_t fec_encode_manager;
  380. static fec_decode_manager_t fec_decode_manager;
  381. //dynamic_update_fec=0;
  382. fec_encode_manager.set_loop_and_cb(ev_default_loop(0),empty_cb);
  383. {
  384. string a = "11111";
  385. string b = "22";
  386. string c = "33333333";
  387. fec_encode_manager.input((char *) a.c_str(), a.length());
  388. fec_encode_manager.input((char *) b.c_str(), b.length());
  389. fec_encode_manager.input((char *) c.c_str(), c.length());
  390. fec_encode_manager.input(0, 0);
  391. int n;
  392. char **s_arr;
  393. int *len;
  394. fec_encode_manager.output(n,s_arr,len);
  395. printf("<n:%d,len:%d>",n,len[0]);
  396. for(int i=0;i<n;i++)
  397. {
  398. fec_decode_manager.input(s_arr[i],len[i]);
  399. }
  400. {
  401. int n;char ** s_arr;int* len_arr;
  402. fec_decode_manager.output(n,s_arr,len_arr);
  403. printf("<n:%d>",n);
  404. for(int i=0;i<n;i++)
  405. {
  406. s_arr[i][len_arr[i]]=0;
  407. printf("<%s>\n",s_arr[i]);
  408. }
  409. }
  410. }
  411. {
  412. string a = "aaaaaaa";
  413. string b = "bbbbbbbbbbbbb";
  414. string c = "ccc";
  415. fec_encode_manager.input((char *) a.c_str(), a.length());
  416. fec_encode_manager.input((char *) b.c_str(), b.length());
  417. fec_encode_manager.input((char *) c.c_str(), c.length());
  418. fec_encode_manager.input(0, 0);
  419. int n;
  420. char **s_arr;
  421. int * len;
  422. fec_encode_manager.output(n,s_arr,len);
  423. printf("<n:%d,len:%d>",n,len[0]);
  424. for(int i=0;i<n;i++)
  425. {
  426. if(i==1||i==3||i==5||i==0)
  427. fec_decode_manager.input(s_arr[i],len[i]);
  428. }
  429. {
  430. int n;char ** s_arr;int* len_arr;
  431. fec_decode_manager.output(n,s_arr,len_arr);
  432. printf("<n:%d>",n);
  433. for(int i=0;i<n;i++)
  434. {
  435. s_arr[i][len_arr[i]]=0;
  436. printf("<%s>\n",s_arr[i]);
  437. }
  438. }
  439. }
  440. printf("ok here.\n");
  441. for(int i=0;i<10;i++)
  442. {
  443. string a = "aaaaaaaaaaaaaaaaaaaaaaa";
  444. string b = "bbbbbbbbbbbbb";
  445. string c = "cccccccccccccccccc";
  446. printf("======\n");
  447. int n;
  448. char **s_arr;
  449. int * len;
  450. fec_decode_manager.output(n,s_arr,len);
  451. //fec_encode_manager.reset_fec_parameter(3,2,g_fec_mtu,g_fec_queue_len,g_fec_timeout,1);
  452. fec_parameter_t &fec_par=fec_encode_manager.get_fec_par();
  453. fec_par.mtu=g_fec_par.mtu;
  454. fec_par.queue_len=g_fec_par.queue_len;
  455. fec_par.timeout=g_fec_par.timeout;
  456. fec_par.mode=1;
  457. fec_par.rs_from_str((char *)"3:2");
  458. fec_encode_manager.input((char *) a.c_str(), a.length());
  459. fec_encode_manager.output(n,s_arr,len);
  460. printf("n=<%d>\n",n);
  461. assert(n==1);
  462. fec_decode_manager.input(s_arr[0],len[0]);
  463. fec_decode_manager.output(n,s_arr,len);
  464. assert(n==1);
  465. printf("%s\n",s_arr[0]);
  466. fec_encode_manager.input((char *) b.c_str(), b.length());
  467. fec_encode_manager.output(n,s_arr,len);
  468. assert(n==1);
  469. //fec_decode_manager.input(s_arr[0],len[0]);
  470. fec_encode_manager.input((char *) c.c_str(), c.length());
  471. fec_encode_manager.output(n,s_arr,len);
  472. assert(n==3);
  473. fec_decode_manager.input(s_arr[0],len[0]);
  474. //printf("n=%d\n",n);
  475. {
  476. int n;
  477. char **s_arr;
  478. int * len;
  479. fec_decode_manager.output(n,s_arr,len);
  480. assert(n==1);
  481. printf("%s\n",s_arr[0]);
  482. }
  483. fec_decode_manager.input(s_arr[1],len[1]);
  484. {
  485. int n;
  486. char **s_arr;
  487. int * len;
  488. fec_decode_manager.output(n,s_arr,len);
  489. assert(n==1);
  490. printf("n=%d\n",n);
  491. s_arr[0][len[0]]=0;
  492. printf("%s\n",s_arr[0]);
  493. }
  494. }
  495. myexit(0);
  496. return 0;
  497. }
  498. void process_arg(int argc, char *argv[])
  499. {
  500. int is_client=0,is_server=0;
  501. int i, j, k;
  502. int opt;
  503. static struct option long_options[] =
  504. {
  505. {"log-level", required_argument, 0, 1},
  506. {"log-position", no_argument, 0, 1},
  507. {"disable-color", no_argument, 0, 1},
  508. {"enable-color", no_argument, 0, 1},
  509. {"disable-filter", no_argument, 0, 1},
  510. {"disable-fec", no_argument, 0, 1},
  511. {"disable-obscure", no_argument, 0, 1},
  512. {"disable-xor", no_argument, 0, 1},
  513. {"disable-checksum", no_argument, 0, 1},
  514. {"fix-latency", no_argument, 0, 1},
  515. {"sock-buf", required_argument, 0, 1},
  516. {"random-drop", required_argument, 0, 1},
  517. {"report", required_argument, 0, 1},
  518. {"delay-capacity", required_argument, 0, 1},
  519. {"mtu", required_argument, 0, 1},
  520. {"mode", required_argument, 0,1},
  521. {"timeout", required_argument, 0,1},
  522. {"decode-buf", required_argument, 0,1},
  523. {"queue-len", required_argument, 0,'q'},
  524. {"fec", required_argument, 0,'f'},
  525. {"jitter", required_argument, 0,'j'},
  526. {"out-addr", required_argument, 0,1},
  527. {"out-interface", required_argument, 0, 1},
  528. {"key", required_argument, 0,'k'},
  529. {"header-overhead", required_argument, 0, 1},
  530. //{"debug-fec", no_argument, 0, 1},
  531. {"debug-fec-enc", no_argument, 0, 1},
  532. {"debug-fec-dec", no_argument, 0, 1},
  533. {"fifo", required_argument, 0, 1},
  534. {"sub-net", required_argument, 0, 1},
  535. {"tun-dev", required_argument, 0, 1},
  536. {"tun-mtu", required_argument, 0, 1},
  537. {"mssfix", required_argument, 0, 1},
  538. {"keep-reconnect", no_argument, 0, 1},
  539. {"persist-tun", no_argument, 0, 1},
  540. {"manual-set-tun", no_argument, 0, 1},
  541. {"interval", required_argument, 0,'i'},
  542. {NULL, 0, 0, 0}
  543. };
  544. int option_index = 0;
  545. assert(g_fec_par.rs_from_str(rs_par_str)==0);
  546. for (i = 0; i < argc; i++)
  547. {
  548. if(strcmp(argv[i],"--unit-test")==0)
  549. {
  550. unit_test();
  551. myexit(0);
  552. }
  553. }
  554. for (i = 0; i < argc; i++)
  555. {
  556. if(strcmp(argv[i],"--log-level")==0)
  557. {
  558. if(i<argc -1)
  559. {
  560. sscanf(argv[i+1],"%d",&log_level);
  561. if(0<=log_level&&log_level<log_end)
  562. {
  563. }
  564. else
  565. {
  566. log_bare(log_fatal,"invalid log_level\n");
  567. myexit(-1);
  568. }
  569. }
  570. }
  571. if(strcmp(argv[i],"--enable-color")==0)
  572. {
  573. enable_log_color=1;
  574. }
  575. if(strcmp(argv[i],"--disable-color")==0)
  576. {
  577. enable_log_color=0;
  578. }
  579. }
  580. mylog(log_info,"argc=%d ", argc);
  581. for (i = 0; i < argc; i++) {
  582. log_bare(log_info, "%s ", argv[i]);
  583. }
  584. log_bare(log_info, "\n");
  585. int no_l = 1, no_r = 1;
  586. while ((opt = getopt_long(argc, argv, "l:r:hcsk:j:f:p:n:i:q:",long_options,&option_index)) != -1)
  587. {
  588. //string opt_key;
  589. //opt_key+=opt;
  590. switch (opt)
  591. {
  592. case 'k':
  593. sscanf(optarg,"%s\n",key_string);
  594. mylog(log_debug,"key=%s\n",key_string);
  595. if(strlen(key_string)==0)
  596. {
  597. mylog(log_fatal,"key len=0??\n");
  598. myexit(-1);
  599. }
  600. break;
  601. case 'j':
  602. if (strchr(optarg, ':') == 0)
  603. {
  604. int jitter;
  605. sscanf(optarg,"%d\n",&jitter);
  606. if(jitter<0 ||jitter>1000*10)
  607. {
  608. mylog(log_fatal,"jitter must be between 0 and 10,000(10 second)\n");
  609. myexit(-1);
  610. }
  611. jitter_min=0;
  612. jitter_max=jitter;
  613. }
  614. else
  615. {
  616. sscanf(optarg,"%d:%d\n",&jitter_min,&jitter_max);
  617. if(jitter_min<0 ||jitter_max<0||jitter_min>jitter_max)
  618. {
  619. mylog(log_fatal," must satisfy 0<=jmin<=jmax\n");
  620. myexit(-1);
  621. }
  622. }
  623. jitter_min*=1000;
  624. jitter_max*=1000;
  625. break;
  626. case 'i':
  627. if (strchr(optarg, ':') == 0)
  628. {
  629. int output_interval=-1;
  630. sscanf(optarg,"%d\n",&output_interval);
  631. if(output_interval<0||output_interval>1000*10)
  632. {
  633. mylog(log_fatal,"output_interval must be between 0 and 10,000(10 second)\n");
  634. myexit(-1);
  635. }
  636. output_interval_min=output_interval_max=output_interval;
  637. }
  638. else
  639. {
  640. sscanf(optarg,"%d:%d\n",&output_interval_min,&output_interval_max);
  641. if(output_interval_min<0 ||output_interval_max<0||output_interval_min>output_interval_max)
  642. {
  643. mylog(log_fatal," must satisfy 0<=output_interval_min<=output_interval_max\n");
  644. myexit(-1);
  645. }
  646. }
  647. output_interval_min*=1000;
  648. output_interval_max*=1000;
  649. break;
  650. case 'f':
  651. if (strchr(optarg, ':') == 0)
  652. {
  653. mylog(log_fatal,"invalid format for f");
  654. myexit(-1);
  655. }
  656. else
  657. {
  658. strcpy(rs_par_str,optarg);
  659. //sscanf(optarg,"%d:%d\n",&g_fec_data_num,&g_fec_redundant_num);
  660. /*
  661. if(g_fec_data_num<1 ||g_fec_redundant_num<0||g_fec_data_num+g_fec_redundant_num>254)
  662. {
  663. mylog(log_fatal,"fec_data_num<1 ||fec_redundant_num<0||fec_data_num+fec_redundant_num>254\n");
  664. myexit(-1);
  665. }*/
  666. }
  667. break;
  668. case 'q':
  669. sscanf(optarg,"%d",&g_fec_par.queue_len);
  670. if(g_fec_par.queue_len<1||g_fec_par.queue_len>10000)
  671. {
  672. mylog(log_fatal,"fec_pending_num should be between 1 and 10000\n");
  673. myexit(-1);
  674. }
  675. break;
  676. case 'c':
  677. is_client = 1;
  678. break;
  679. case 's':
  680. is_server = 1;
  681. break;
  682. case 'l':
  683. no_l = 0;
  684. local_addr.from_str(optarg);
  685. break;
  686. case 'r':
  687. no_r = 0;
  688. remote_addr.from_str(optarg);
  689. break;
  690. case 'h':
  691. break;
  692. case 1:
  693. if(strcmp(long_options[option_index].name,"log-level")==0)
  694. {
  695. }
  696. else if(strcmp(long_options[option_index].name,"disable-filter")==0)
  697. {
  698. disable_replay_filter=1;
  699. //enable_log_color=0;
  700. }
  701. else if(strcmp(long_options[option_index].name,"disable-color")==0)
  702. {
  703. //enable_log_color=0;
  704. }
  705. else if(strcmp(long_options[option_index].name,"enable-color")==0)
  706. {
  707. //enable_log_color=0;
  708. }
  709. else if(strcmp(long_options[option_index].name,"disable-fec")==0)
  710. {
  711. disable_fec=1;
  712. }
  713. else if(strcmp(long_options[option_index].name,"disable-obscure")==0)
  714. {
  715. mylog(log_info,"obscure disabled\n");
  716. disable_obscure=1;
  717. }
  718. else if(strcmp(long_options[option_index].name,"disable-xor")==0)
  719. {
  720. mylog(log_info,"xor disabled\n");
  721. disable_xor=1;
  722. }
  723. else if(strcmp(long_options[option_index].name,"disable-checksum")==0)
  724. {
  725. disable_checksum=1;
  726. mylog(log_warn,"checksum disabled\n");
  727. }
  728. else if(strcmp(long_options[option_index].name,"fix-latency")==0)
  729. {
  730. mylog(log_info,"fix-latency enabled\n");
  731. fix_latency=1;
  732. }
  733. else if(strcmp(long_options[option_index].name,"log-position")==0)
  734. {
  735. enable_log_position=1;
  736. }
  737. else if(strcmp(long_options[option_index].name,"random-drop")==0)
  738. {
  739. sscanf(optarg,"%d",&random_drop);
  740. if(random_drop<0||random_drop>10000)
  741. {
  742. mylog(log_fatal,"random_drop must be between 0 10000 \n");
  743. myexit(-1);
  744. }
  745. mylog(log_info,"random_drop=%d\n",random_drop);
  746. }
  747. else if(strcmp(long_options[option_index].name,"delay-capacity")==0)
  748. {
  749. sscanf(optarg,"%d",&delay_capacity);
  750. if(delay_capacity<0)
  751. {
  752. mylog(log_fatal,"delay_capacity must be >=0 \n");
  753. myexit(-1);
  754. }
  755. }
  756. else if(strcmp(long_options[option_index].name,"report")==0)
  757. {
  758. sscanf(optarg,"%d",&report_interval);
  759. if(report_interval<=0)
  760. {
  761. mylog(log_fatal,"report_interval must be >0 \n");
  762. myexit(-1);
  763. }
  764. }
  765. else if(strcmp(long_options[option_index].name,"sock-buf")==0)
  766. {
  767. int tmp=-1;
  768. sscanf(optarg,"%d",&tmp);
  769. if(10<=tmp&&tmp<=10*1024)
  770. {
  771. socket_buf_size=tmp*1024;
  772. }
  773. else
  774. {
  775. mylog(log_fatal,"sock-buf value must be between 1 and 10240 (kbyte) \n");
  776. myexit(-1);
  777. }
  778. }
  779. else if(strcmp(long_options[option_index].name,"decode-buf")==0)
  780. {
  781. sscanf(optarg,"%d",&fec_buff_num);
  782. if(fec_buff_num<300 || fec_buff_num>20000)
  783. {
  784. mylog(log_fatal,"decode-buf value must be between 300 and 20000 (kbyte) \n");
  785. myexit(-1);
  786. }
  787. mylog(log_info,"decode-buf=%d\n",fec_buff_num);
  788. }
  789. else if(strcmp(long_options[option_index].name,"mode")==0)
  790. {
  791. sscanf(optarg,"%d",&g_fec_par.mode);
  792. if(g_fec_par.mode!=0&&g_fec_par.mode!=1)
  793. {
  794. mylog(log_fatal,"mode should be 0 or 1\n");
  795. myexit(-1);
  796. }
  797. }
  798. else if(strcmp(long_options[option_index].name,"mtu")==0)
  799. {
  800. sscanf(optarg,"%d",&g_fec_par.mtu);
  801. if(g_fec_par.mtu<100||g_fec_par.mtu>2000)
  802. {
  803. mylog(log_fatal,"fec_mtu should be between 100 and 2000\n");
  804. myexit(-1);
  805. }
  806. }
  807. else if(strcmp(long_options[option_index].name,"out-addr")==0)
  808. {
  809. //has_b = true;
  810. mylog(log_debug,"out-addr=%s\n",optarg);
  811. out_addr=new address_t();
  812. out_addr->from_str(optarg);
  813. }
  814. else if(strcmp(long_options[option_index].name,"out-interface")==0)
  815. {
  816. out_interface=new char[strlen(optarg)+10];
  817. sscanf(optarg,"%s\n",out_interface);
  818. mylog(log_debug,"out-interface=%s\n",out_interface);
  819. if(strlen(out_interface)==0)
  820. {
  821. mylog(log_fatal,"out_interface string len=0??\n");
  822. myexit(-1);
  823. }
  824. }
  825. else if(strcmp(long_options[option_index].name,"timeout")==0)
  826. {
  827. sscanf(optarg,"%d",&g_fec_par.timeout);
  828. if(g_fec_par.timeout<0||g_fec_par.timeout>1000)
  829. {
  830. mylog(log_fatal,"fec_pending_time should be between 0 and 1000(1s)\n");
  831. myexit(-1);
  832. }
  833. g_fec_par.timeout*=1000;
  834. }
  835. else if(strcmp(long_options[option_index].name,"debug-fec-enc")==0)
  836. {
  837. debug_fec_enc=1;
  838. mylog(log_info,"debug_fec_enc enabled\n");
  839. }
  840. else if(strcmp(long_options[option_index].name,"debug-fec-dec")==0)
  841. {
  842. debug_fec_dec=1;
  843. mylog(log_info,"debug_fec_dec enabled\n");
  844. }
  845. else if(strcmp(long_options[option_index].name,"fifo")==0)
  846. {
  847. sscanf(optarg,"%s",fifo_file);
  848. mylog(log_info,"fifo_file =%s \n",fifo_file);
  849. }
  850. else if(strcmp(long_options[option_index].name,"keep-reconnect")==0)
  851. {
  852. keep_reconnect=1;
  853. mylog(log_info,"keep_reconnect enabled\n");
  854. }
  855. else if(strcmp(long_options[option_index].name,"manual-set-tun")==0)
  856. {
  857. manual_set_tun=1;
  858. mylog(log_info,"manual_set_tun enabled\n");
  859. }
  860. else if(strcmp(long_options[option_index].name,"persist-tun")==0)
  861. {
  862. persist_tun=1;
  863. mylog(log_info,"persist_tun enabled\n");
  864. }
  865. else if(strcmp(long_options[option_index].name,"sub-net")==0)
  866. {
  867. sscanf(optarg,"%s",sub_net);
  868. mylog(log_info,"sub_net %s\n",sub_net);
  869. }
  870. else if(strcmp(long_options[option_index].name,"tun-dev")==0)
  871. {
  872. sscanf(optarg,"%s",tun_dev);
  873. mylog(log_info,"tun_dev=%s\n",tun_dev);
  874. }
  875. else if(strcmp(long_options[option_index].name,"tun-mtu")==0)
  876. {
  877. sscanf(optarg,"%d",&tun_mtu);
  878. mylog(log_warn,"changed tun_mtu,tun_mtu=%d\n",tun_mtu);
  879. }
  880. else if(strcmp(long_options[option_index].name,"header-overhead")==0)
  881. {
  882. sscanf(optarg,"%d",&header_overhead);
  883. mylog(log_warn,"changed header_overhead,header_overhead=%d\n",header_overhead);
  884. }
  885. else if(strcmp(long_options[option_index].name,"mssfix")==0)
  886. {
  887. sscanf(optarg,"%d",&mssfix);
  888. mylog(log_warn,"mssfix=%d\n",mssfix);
  889. }
  890. else
  891. {
  892. mylog(log_fatal,"unknown option\n");
  893. myexit(-1);
  894. }
  895. break;
  896. default:
  897. mylog(log_fatal,"unknown option <%x>", opt);
  898. myexit(-1);
  899. }
  900. }
  901. if (is_client == 0 && is_server == 0)
  902. {
  903. mylog(log_fatal,"-s -c hasnt been set\n");
  904. myexit(-1);
  905. }
  906. if (is_client == 1 && is_server == 1)
  907. {
  908. mylog(log_fatal,"-s -c cant be both set\n");
  909. myexit(-1);
  910. }
  911. if(is_client==1)
  912. {
  913. program_mode=client_mode;
  914. }
  915. else
  916. {
  917. program_mode=server_mode;
  918. }
  919. if(working_mode==tunnel_mode)
  920. {
  921. if (no_l)
  922. mylog(log_fatal,"error: -l not found\n");
  923. if (no_r)
  924. mylog(log_fatal,"error: -r not found\n");
  925. if (no_l || no_r)
  926. myexit(-1);
  927. }
  928. else if(working_mode==tun_dev_mode)
  929. {
  930. if(program_mode==client_mode&&no_r)
  931. {
  932. mylog(log_fatal,"error: -r not found\n");
  933. myexit(-1);
  934. }
  935. else if(program_mode==server_mode&&no_l)
  936. {
  937. mylog(log_fatal,"error: -l not found\n");
  938. myexit(-1);
  939. }
  940. }
  941. int ret=g_fec_par.rs_from_str(rs_par_str);
  942. if(ret!=0)
  943. {
  944. mylog(log_fatal,"failed to parse [%s]\n",rs_par_str);
  945. myexit(-1);
  946. }
  947. print_parameter();
  948. }