misc.cpp 20 KB

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