1
0

misc.cpp 31 KB

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