misc.cpp 23 KB

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