misc.cpp 24 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105
  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. 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. {"mode", required_argument, 0,1},
  524. {"timeout", required_argument, 0,1},
  525. {"decode-buf", required_argument, 0,1},
  526. {"queue-len", required_argument, 0,'q'},
  527. {"fec", required_argument, 0,'f'},
  528. {"jitter", required_argument, 0,'j'},
  529. {"out-addr", required_argument, 0,1},
  530. {"out-interface", required_argument, 0, 1},
  531. {"key", required_argument, 0,'k'},
  532. {"header-overhead", required_argument, 0, 1},
  533. //{"debug-fec", no_argument, 0, 1},
  534. {"debug-fec-enc", no_argument, 0, 1},
  535. {"debug-fec-dec", no_argument, 0, 1},
  536. {"fifo", required_argument, 0, 1},
  537. {"sub-net", required_argument, 0, 1},
  538. {"tun-dev", required_argument, 0, 1},
  539. {"tun-mtu", required_argument, 0, 1},
  540. {"mssfix", required_argument, 0, 1},
  541. {"keep-reconnect", no_argument, 0, 1},
  542. {"persist-tun", no_argument, 0, 1},
  543. {"manual-set-tun", no_argument, 0, 1},
  544. {"interval", required_argument, 0,'i'},
  545. {NULL, 0, 0, 0}
  546. };
  547. int option_index = 0;
  548. assert(g_fec_par.rs_from_str(rs_par_str)==0);
  549. for (i = 0; i < argc; i++)
  550. {
  551. if(strcmp(argv[i],"--unit-test")==0)
  552. {
  553. unit_test();
  554. myexit(0);
  555. }
  556. }
  557. for (i = 0; i < argc; i++)
  558. {
  559. if(strcmp(argv[i],"--log-level")==0)
  560. {
  561. if(i<argc -1)
  562. {
  563. sscanf(argv[i+1],"%d",&log_level);
  564. if(0<=log_level&&log_level<log_end)
  565. {
  566. }
  567. else
  568. {
  569. log_bare(log_fatal,"invalid log_level\n");
  570. myexit(-1);
  571. }
  572. }
  573. }
  574. if(strcmp(argv[i],"--enable-color")==0)
  575. {
  576. enable_log_color=1;
  577. }
  578. if(strcmp(argv[i],"--disable-color")==0)
  579. {
  580. enable_log_color=0;
  581. }
  582. }
  583. mylog(log_info,"argc=%d ", argc);
  584. for (i = 0; i < argc; i++) {
  585. log_bare(log_info, "%s ", argv[i]);
  586. }
  587. log_bare(log_info, "\n");
  588. int no_l = 1, no_r = 1;
  589. while ((opt = getopt_long(argc, argv, "l:r:hcsk:j:f:p:n:i:q:",long_options,&option_index)) != -1)
  590. {
  591. //string opt_key;
  592. //opt_key+=opt;
  593. switch (opt)
  594. {
  595. case 'k':
  596. sscanf(optarg,"%s\n",key_string);
  597. mylog(log_debug,"key=%s\n",key_string);
  598. if(strlen(key_string)==0)
  599. {
  600. mylog(log_fatal,"key len=0??\n");
  601. myexit(-1);
  602. }
  603. break;
  604. case 'j':
  605. if (strchr(optarg, ':') == 0)
  606. {
  607. int jitter;
  608. sscanf(optarg,"%d\n",&jitter);
  609. if(jitter<0 ||jitter>1000*10)
  610. {
  611. mylog(log_fatal,"jitter must be between 0 and 10,000(10 second)\n");
  612. myexit(-1);
  613. }
  614. jitter_min=0;
  615. jitter_max=jitter;
  616. }
  617. else
  618. {
  619. sscanf(optarg,"%d:%d\n",&jitter_min,&jitter_max);
  620. if(jitter_min<0 ||jitter_max<0||jitter_min>jitter_max)
  621. {
  622. mylog(log_fatal," must satisfy 0<=jmin<=jmax\n");
  623. myexit(-1);
  624. }
  625. }
  626. jitter_min*=1000;
  627. jitter_max*=1000;
  628. break;
  629. case 'i':
  630. if (strchr(optarg, ':') == 0)
  631. {
  632. int output_interval=-1;
  633. sscanf(optarg,"%d\n",&output_interval);
  634. if(output_interval<0||output_interval>1000*10)
  635. {
  636. mylog(log_fatal,"output_interval must be between 0 and 10,000(10 second)\n");
  637. myexit(-1);
  638. }
  639. output_interval_min=output_interval_max=output_interval;
  640. }
  641. else
  642. {
  643. sscanf(optarg,"%d:%d\n",&output_interval_min,&output_interval_max);
  644. if(output_interval_min<0 ||output_interval_max<0||output_interval_min>output_interval_max)
  645. {
  646. mylog(log_fatal," must satisfy 0<=output_interval_min<=output_interval_max\n");
  647. myexit(-1);
  648. }
  649. }
  650. output_interval_min*=1000;
  651. output_interval_max*=1000;
  652. break;
  653. case 'f':
  654. if (strchr(optarg, ':') == 0)
  655. {
  656. mylog(log_fatal,"invalid format for f");
  657. myexit(-1);
  658. }
  659. else
  660. {
  661. strcpy(rs_par_str,optarg);
  662. //sscanf(optarg,"%d:%d\n",&g_fec_data_num,&g_fec_redundant_num);
  663. /*
  664. if(g_fec_data_num<1 ||g_fec_redundant_num<0||g_fec_data_num+g_fec_redundant_num>254)
  665. {
  666. mylog(log_fatal,"fec_data_num<1 ||fec_redundant_num<0||fec_data_num+fec_redundant_num>254\n");
  667. myexit(-1);
  668. }*/
  669. }
  670. break;
  671. case 'q':
  672. sscanf(optarg,"%d",&g_fec_par.queue_len);
  673. if(g_fec_par.queue_len<1||g_fec_par.queue_len>10000)
  674. {
  675. mylog(log_fatal,"fec_pending_num should be between 1 and 10000\n");
  676. myexit(-1);
  677. }
  678. break;
  679. case 'c':
  680. is_client = 1;
  681. break;
  682. case 's':
  683. is_server = 1;
  684. break;
  685. case 'l':
  686. no_l = 0;
  687. local_addr.from_str(optarg);
  688. break;
  689. case 'r':
  690. no_r = 0;
  691. remote_addr.from_str(optarg);
  692. break;
  693. case 'h':
  694. break;
  695. case 1:
  696. if(strcmp(long_options[option_index].name,"log-level")==0)
  697. {
  698. }
  699. else if(strcmp(long_options[option_index].name,"disable-filter")==0)
  700. {
  701. disable_replay_filter=1;
  702. //enable_log_color=0;
  703. }
  704. else if(strcmp(long_options[option_index].name,"disable-color")==0)
  705. {
  706. //enable_log_color=0;
  707. }
  708. else if(strcmp(long_options[option_index].name,"enable-color")==0)
  709. {
  710. //enable_log_color=0;
  711. }
  712. else if(strcmp(long_options[option_index].name,"disable-fec")==0)
  713. {
  714. disable_fec=1;
  715. }
  716. else if(strcmp(long_options[option_index].name,"disable-obscure")==0)
  717. {
  718. mylog(log_info,"obscure disabled\n");
  719. disable_obscure=1;
  720. }
  721. else if(strcmp(long_options[option_index].name,"disable-xor")==0)
  722. {
  723. mylog(log_info,"xor disabled\n");
  724. disable_xor=1;
  725. }
  726. else if(strcmp(long_options[option_index].name,"disable-checksum")==0)
  727. {
  728. disable_checksum=1;
  729. mylog(log_warn,"checksum disabled\n");
  730. }
  731. else if(strcmp(long_options[option_index].name,"fix-latency")==0)
  732. {
  733. mylog(log_info,"fix-latency enabled\n");
  734. fix_latency=1;
  735. }
  736. else if(strcmp(long_options[option_index].name,"log-position")==0)
  737. {
  738. enable_log_position=1;
  739. }
  740. else if(strcmp(long_options[option_index].name,"random-drop")==0)
  741. {
  742. sscanf(optarg,"%d",&random_drop);
  743. if(random_drop<0||random_drop>10000)
  744. {
  745. mylog(log_fatal,"random_drop must be between 0 10000 \n");
  746. myexit(-1);
  747. }
  748. mylog(log_info,"random_drop=%d\n",random_drop);
  749. }
  750. else if(strcmp(long_options[option_index].name,"delay-capacity")==0)
  751. {
  752. sscanf(optarg,"%d",&delay_capacity);
  753. if(delay_capacity<0)
  754. {
  755. mylog(log_fatal,"delay_capacity must be >=0 \n");
  756. myexit(-1);
  757. }
  758. }
  759. else if(strcmp(long_options[option_index].name,"report")==0)
  760. {
  761. sscanf(optarg,"%d",&report_interval);
  762. if(report_interval<=0)
  763. {
  764. mylog(log_fatal,"report_interval must be >0 \n");
  765. myexit(-1);
  766. }
  767. }
  768. else if(strcmp(long_options[option_index].name,"sock-buf")==0)
  769. {
  770. int tmp=-1;
  771. sscanf(optarg,"%d",&tmp);
  772. if(10<=tmp&&tmp<=10*1024)
  773. {
  774. socket_buf_size=tmp*1024;
  775. }
  776. else
  777. {
  778. mylog(log_fatal,"sock-buf value must be between 1 and 10240 (kbyte) \n");
  779. myexit(-1);
  780. }
  781. }
  782. else if(strcmp(long_options[option_index].name,"decode-buf")==0)
  783. {
  784. sscanf(optarg,"%d",&fec_buff_num);
  785. if(fec_buff_num<300 || fec_buff_num>20000)
  786. {
  787. mylog(log_fatal,"decode-buf value must be between 300 and 20000 (kbyte) \n");
  788. myexit(-1);
  789. }
  790. mylog(log_info,"decode-buf=%d\n",fec_buff_num);
  791. }
  792. else if(strcmp(long_options[option_index].name,"mode")==0)
  793. {
  794. sscanf(optarg,"%d",&g_fec_par.mode);
  795. if(g_fec_par.mode!=0&&g_fec_par.mode!=1)
  796. {
  797. mylog(log_fatal,"mode should be 0 or 1\n");
  798. myexit(-1);
  799. }
  800. }
  801. else if(strcmp(long_options[option_index].name,"mtu")==0)
  802. {
  803. sscanf(optarg,"%d",&g_fec_par.mtu);
  804. if(g_fec_par.mtu<100||g_fec_par.mtu>2000)
  805. {
  806. mylog(log_fatal,"fec_mtu should be between 100 and 2000\n");
  807. myexit(-1);
  808. }
  809. }
  810. else if(strcmp(long_options[option_index].name,"out-addr")==0)
  811. {
  812. //has_b = true;
  813. mylog(log_debug,"out-addr=%s\n",optarg);
  814. out_addr=new address_t();
  815. out_addr->from_str(optarg);
  816. }
  817. else if(strcmp(long_options[option_index].name,"out-interface")==0)
  818. {
  819. out_interface=new char[strlen(optarg)+10];
  820. sscanf(optarg,"%s\n",out_interface);
  821. mylog(log_debug,"out-interface=%s\n",out_interface);
  822. if(strlen(out_interface)==0)
  823. {
  824. mylog(log_fatal,"out_interface string len=0??\n");
  825. myexit(-1);
  826. }
  827. }
  828. else if(strcmp(long_options[option_index].name,"timeout")==0)
  829. {
  830. sscanf(optarg,"%d",&g_fec_par.timeout);
  831. if(g_fec_par.timeout<0||g_fec_par.timeout>1000)
  832. {
  833. mylog(log_fatal,"fec_pending_time should be between 0 and 1000(1s)\n");
  834. myexit(-1);
  835. }
  836. g_fec_par.timeout*=1000;
  837. }
  838. else if(strcmp(long_options[option_index].name,"debug-fec-enc")==0)
  839. {
  840. debug_fec_enc=1;
  841. mylog(log_info,"debug_fec_enc enabled\n");
  842. }
  843. else if(strcmp(long_options[option_index].name,"debug-fec-dec")==0)
  844. {
  845. debug_fec_dec=1;
  846. mylog(log_info,"debug_fec_dec enabled\n");
  847. }
  848. else if(strcmp(long_options[option_index].name,"fifo")==0)
  849. {
  850. sscanf(optarg,"%s",fifo_file);
  851. mylog(log_info,"fifo_file =%s \n",fifo_file);
  852. }
  853. else if(strcmp(long_options[option_index].name,"keep-reconnect")==0)
  854. {
  855. keep_reconnect=1;
  856. mylog(log_info,"keep_reconnect enabled\n");
  857. }
  858. else if(strcmp(long_options[option_index].name,"manual-set-tun")==0)
  859. {
  860. manual_set_tun=1;
  861. mylog(log_info,"manual_set_tun enabled\n");
  862. }
  863. else if(strcmp(long_options[option_index].name,"persist-tun")==0)
  864. {
  865. persist_tun=1;
  866. mylog(log_info,"persist_tun enabled\n");
  867. }
  868. else if(strcmp(long_options[option_index].name,"sub-net")==0)
  869. {
  870. sscanf(optarg,"%s",sub_net);
  871. mylog(log_info,"sub_net %s\n",sub_net);
  872. }
  873. else if(strcmp(long_options[option_index].name,"tun-dev")==0)
  874. {
  875. sscanf(optarg,"%s",tun_dev);
  876. mylog(log_info,"tun_dev=%s\n",tun_dev);
  877. }
  878. else if(strcmp(long_options[option_index].name,"tun-mtu")==0)
  879. {
  880. sscanf(optarg,"%d",&tun_mtu);
  881. mylog(log_warn,"changed tun_mtu,tun_mtu=%d\n",tun_mtu);
  882. }
  883. else if(strcmp(long_options[option_index].name,"header-overhead")==0)
  884. {
  885. sscanf(optarg,"%d",&header_overhead);
  886. mylog(log_warn,"changed header_overhead,header_overhead=%d\n",header_overhead);
  887. }
  888. else if(strcmp(long_options[option_index].name,"mssfix")==0)
  889. {
  890. sscanf(optarg,"%d",&mssfix);
  891. mylog(log_warn,"mssfix=%d\n",mssfix);
  892. }
  893. else
  894. {
  895. mylog(log_fatal,"unknown option\n");
  896. myexit(-1);
  897. }
  898. break;
  899. default:
  900. mylog(log_fatal,"unknown option <%x>", opt);
  901. myexit(-1);
  902. }
  903. }
  904. if (is_client == 0 && is_server == 0)
  905. {
  906. mylog(log_fatal,"-s -c hasnt been set\n");
  907. myexit(-1);
  908. }
  909. if (is_client == 1 && is_server == 1)
  910. {
  911. mylog(log_fatal,"-s -c cant be both set\n");
  912. myexit(-1);
  913. }
  914. if(is_client==1)
  915. {
  916. program_mode=client_mode;
  917. }
  918. else
  919. {
  920. program_mode=server_mode;
  921. }
  922. if(working_mode==tunnel_mode)
  923. {
  924. if (no_l)
  925. mylog(log_fatal,"error: -l not found\n");
  926. if (no_r)
  927. mylog(log_fatal,"error: -r not found\n");
  928. if (no_l || no_r)
  929. myexit(-1);
  930. }
  931. else if(working_mode==tun_dev_mode)
  932. {
  933. if(program_mode==client_mode&&no_r)
  934. {
  935. mylog(log_fatal,"error: -r not found\n");
  936. myexit(-1);
  937. }
  938. else if(program_mode==server_mode&&no_l)
  939. {
  940. mylog(log_fatal,"error: -l not found\n");
  941. myexit(-1);
  942. }
  943. }
  944. int ret=g_fec_par.rs_from_str(rs_par_str);
  945. if(ret!=0)
  946. {
  947. mylog(log_fatal,"failed to parse [%s]\n",rs_par_str);
  948. myexit(-1);
  949. }
  950. print_parameter();
  951. }