main.cpp 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335
  1. #include "common.h"
  2. #include "log.h"
  3. #include "git_version.h"
  4. #include "lib/rs.h"
  5. #include "packet.h"
  6. #include "connection.h"
  7. #include "fd_manager.h"
  8. #include "delay_manager.h"
  9. #include "fec_manager.h"
  10. using namespace std;
  11. typedef unsigned long long u64_t; //this works on most platform,avoid using the PRId64
  12. typedef long long i64_t;
  13. typedef unsigned int u32_t;
  14. typedef int i32_t;
  15. int dup_num=1;
  16. int dup_delay_min=20; //0.1ms
  17. int dup_delay_max=20;
  18. int jitter_min=0;
  19. int jitter_max=0;
  20. //int random_number_fd=-1;
  21. int mtu_warn=1350;
  22. int fec_data_num=20;
  23. int fec_redundant_num=8;
  24. int fec_mtu=1200;
  25. int fec_pending_num=200;
  26. int fec_pending_time=50000;
  27. u32_t local_ip_uint32,remote_ip_uint32=0;
  28. char local_ip[100], remote_ip[100];
  29. int local_port = -1, remote_port = -1;
  30. u64_t last_report_time=0;
  31. int report_interval=0;
  32. //conn_manager_t conn_manager;
  33. delay_manager_t delay_manager;
  34. fd_manager_t fd_manager;
  35. const int disable_conv_clear=0;
  36. int socket_buf_size=1024*1024;
  37. int VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV;
  38. int init_listen_socket()
  39. {
  40. local_listen_fd =socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  41. int yes = 1;
  42. //setsockopt(udp_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
  43. struct sockaddr_in local_me={0};
  44. socklen_t slen = sizeof(sockaddr_in);
  45. //memset(&local_me, 0, sizeof(local_me));
  46. local_me.sin_family = AF_INET;
  47. local_me.sin_port = htons(local_port);
  48. local_me.sin_addr.s_addr = local_ip_uint32;
  49. if (bind(local_listen_fd, (struct sockaddr*) &local_me, slen) == -1) {
  50. mylog(log_fatal,"socket bind error\n");
  51. //perror("socket bind error");
  52. myexit(1);
  53. }
  54. setnonblocking(local_listen_fd);
  55. set_buf_size(local_listen_fd,socket_buf_size);
  56. return 0;
  57. }
  58. int new_connected_socket(int &fd,u32_t ip,int port)
  59. {
  60. char ip_port[40];
  61. sprintf(ip_port,"%s:%d",my_ntoa(ip),port);
  62. struct sockaddr_in remote_addr_in = { 0 };
  63. socklen_t slen = sizeof(sockaddr_in);
  64. //memset(&remote_addr_in, 0, sizeof(remote_addr_in));
  65. remote_addr_in.sin_family = AF_INET;
  66. remote_addr_in.sin_port = htons(port);
  67. remote_addr_in.sin_addr.s_addr = ip;
  68. fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
  69. if (fd < 0) {
  70. mylog(log_warn, "[%s]create udp_fd error\n", ip_port);
  71. return -1;
  72. }
  73. setnonblocking(fd);
  74. set_buf_size(fd, socket_buf_size);
  75. mylog(log_debug, "[%s]created new udp_fd %d\n", ip_port, fd);
  76. int ret = connect(fd, (struct sockaddr *) &remote_addr_in, slen);
  77. if (ret != 0) {
  78. mylog(log_warn, "[%s]fd connect fail\n",ip_port);
  79. close(fd);
  80. return -1;
  81. }
  82. return 0;
  83. }
  84. int delay_send(my_time_t delay,const dest_t &dest,char *data,int len)
  85. {
  86. int rand=random()%100;
  87. //mylog(log_info,"rand = %d\n",rand);
  88. if(rand>=80)
  89. {
  90. //return 0;
  91. //mylog(log_info,"dropped!\n");
  92. }
  93. return delay_manager.add(delay,dest,data,len);;
  94. }
  95. int from_normal_to_fec(conn_info_t & conn_info,char *data,int len,int & out_n,char **&out_arr,int *&out_len,int *&out_delay)
  96. {
  97. static int out_delay_buf[max_fec_packet_num+100]={0};
  98. static int out_len_buf[max_fec_packet_num+100]={0};
  99. static int counter=0;
  100. out_delay=out_delay_buf;
  101. out_len=out_len_buf;
  102. if(0)
  103. {
  104. if(data==0) return 0;
  105. out_n=1;
  106. static char *data_static;
  107. data_static=data;
  108. static int len_static;
  109. len_static=len;
  110. out_arr=&data_static;
  111. out_len=&len_static;
  112. }
  113. else
  114. {
  115. counter++;
  116. conn_info.fec_encode_manager.input(data,len);
  117. //if(counter%5==0)
  118. //conn_info.fec_encode_manager.input(0,0);
  119. //int n;
  120. //char **s_arr;
  121. //int s_len;
  122. int tmp_out_len;
  123. conn_info.fec_encode_manager.output(out_n,out_arr,tmp_out_len);
  124. for(int i=0;i<out_n;i++)
  125. {
  126. out_len_buf[i]=tmp_out_len;
  127. // out_delay_buf[i]=100*i;
  128. }
  129. }
  130. //for(int i=0;i<n;i++)
  131. //{
  132. //delay_send(0,dest,s_arr[i],s_len);
  133. //}
  134. //delay_send(0,dest,data,len);
  135. //delay_send(1000*1000,dest,data,len);
  136. return 0;
  137. }
  138. int from_fec_to_normal(conn_info_t & conn_info,char *data,int len,int & out_n,char **&out_arr,int *&out_len,int *&out_delay)
  139. {
  140. static int out_delay_buf[max_normal_packet_num+100]={0};
  141. out_delay=out_delay_buf;
  142. if(0)
  143. {
  144. if(data==0) return 0;
  145. out_n=1;
  146. static char *data_static;
  147. data_static=data;
  148. static int len_static;
  149. len_static=len;
  150. out_arr=&data_static;
  151. out_len=&len_static;
  152. //out_len_buf[0]=len;
  153. }
  154. else
  155. {
  156. conn_info.fec_decode_manager.input(data,len);
  157. //int n;char ** s_arr;int* len_arr;
  158. conn_info.fec_decode_manager.output(out_n,out_arr,out_len);
  159. for(int i=0;i<out_n;i++)
  160. {
  161. //out_delay_buf[i]=100*i;
  162. }
  163. }
  164. // printf("<n:%d>",n);
  165. /*
  166. for(int i=0;i<n;i++)
  167. {
  168. delay_send(0,dest,s_arr[i],len_arr[i]);
  169. //s_arr[i][len_arr[i]]=0;
  170. //printf("<%s>\n",s_arr[i]);
  171. }*/
  172. //my_send(dest,data,len);
  173. return 0;
  174. }
  175. int client_event_loop()
  176. {
  177. //char buf[buf_len];
  178. int i, j, k;int ret;
  179. int yes = 1;
  180. int epoll_fd;
  181. int remote_fd;
  182. fd64_t remote_fd64;
  183. conn_info_t *conn_info_p=new conn_info_t;
  184. conn_info_t &conn_info=*conn_info_p; //huge size of conn_info,do not allocate on stack
  185. conn_info.conv_manager.reserve();
  186. conn_info.fec_encode_manager.re_init(fec_data_num,fec_redundant_num,fec_mtu,fec_pending_num,fec_pending_time);
  187. init_listen_socket();
  188. epoll_fd = epoll_create1(0);
  189. const int max_events = 4096;
  190. struct epoll_event ev, events[max_events];
  191. if (epoll_fd < 0) {
  192. mylog(log_fatal,"epoll return %d\n", epoll_fd);
  193. myexit(-1);
  194. }
  195. ev.events = EPOLLIN;
  196. ev.data.u64 = local_listen_fd;
  197. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, local_listen_fd, &ev);
  198. if (ret!=0) {
  199. mylog(log_fatal,"add udp_listen_fd error\n");
  200. myexit(-1);
  201. }
  202. assert(new_connected_socket(remote_fd,remote_ip_uint32,remote_port)==0);
  203. remote_fd64=fd_manager.create(remote_fd);
  204. ev.events = EPOLLIN;
  205. ev.data.u64 = remote_fd64;
  206. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, remote_fd, &ev);
  207. if (ret!= 0) {
  208. mylog(log_fatal,"add raw_fd error\n");
  209. myexit(-1);
  210. }
  211. ev.events = EPOLLIN;
  212. ev.data.u64 = delay_manager.get_timer_fd();
  213. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
  214. if (ret!= 0) {
  215. mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
  216. myexit(-1);
  217. }
  218. u64_t fd64=conn_info.fec_encode_manager.get_timer_fd64();
  219. ev.events = EPOLLIN;
  220. ev.data.u64 = fd64;
  221. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(fd64), &ev);
  222. if (ret!= 0) {
  223. mylog(log_fatal,"add fec_encode_manager.get_timer_fd64() error\n");
  224. myexit(-1);
  225. }
  226. //my_timer_t timer;
  227. conn_info.timer.add_fd_to_epoll(epoll_fd);
  228. conn_info.timer.set_timer_repeat_us(timer_interval*1000);
  229. while(1)////////////////////////
  230. {
  231. if(about_to_exit) myexit(0);
  232. int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
  233. if (nfds < 0) { //allow zero
  234. if(errno==EINTR )
  235. {
  236. mylog(log_info,"epoll interrupted by signal\n");
  237. myexit(0);
  238. }
  239. else
  240. {
  241. mylog(log_fatal,"epoll_wait return %d\n", nfds);
  242. myexit(-1);
  243. }
  244. }
  245. int idx;
  246. for (idx = 0; idx < nfds; ++idx) {
  247. if(events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd())
  248. {
  249. uint64_t value;
  250. read(conn_info.timer.get_timer_fd(), &value, 8);
  251. conn_info.conv_manager.clear_inactive();
  252. }
  253. else if (events[idx].data.u64 == (u64_t)local_listen_fd||events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64())
  254. {
  255. char data[buf_len];
  256. int data_len;
  257. ip_port_t ip_port;
  258. u32_t conv;
  259. int out_n;char **out_arr;int *out_len;int *out_delay;
  260. dest_t dest;
  261. dest.type=type_fd64;
  262. dest.inner.fd64=remote_fd64;
  263. dest.cook=1;
  264. if(events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64())
  265. {
  266. mylog(log_info,"timer!!!\n");
  267. uint64_t value;
  268. if(!fd_manager.exist(fd64)) //fd64 has been closed
  269. {
  270. continue;
  271. }
  272. if(read(fd_manager.to_fd(fd64), &value, 8)!=8)
  273. {
  274. mylog(log_info,"unknow!!!\n");
  275. continue;
  276. }
  277. if(value==0)
  278. {
  279. mylog(log_info,"cancel!!!\n");
  280. continue;
  281. }
  282. assert(value==1);
  283. from_normal_to_fec(conn_info,0,0,out_n,out_arr,out_len,out_delay);
  284. //from_normal_to_fec(conn_info,0,0,out_n,out_arr,out_len,out_delay);
  285. }
  286. else
  287. {
  288. struct sockaddr_in udp_new_addr_in={0};
  289. socklen_t udp_new_addr_len = sizeof(sockaddr_in);
  290. if ((data_len = recvfrom(local_listen_fd, data, max_data_len, 0,
  291. (struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) == -1) {
  292. mylog(log_error,"recv_from error,this shouldnt happen at client\n");
  293. myexit(1);
  294. };
  295. if(data_len>=mtu_warn)
  296. {
  297. mylog(log_warn,"huge packet,data len=%d (>=%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ",data_len,mtu_warn);
  298. }
  299. mylog(log_trace,"Received packet from %s:%d,len: %d\n", inet_ntoa(udp_new_addr_in.sin_addr),
  300. ntohs(udp_new_addr_in.sin_port),data_len);
  301. ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
  302. ip_port.port=ntohs(udp_new_addr_in.sin_port);
  303. u64_t u64=ip_port.to_u64();
  304. if(!conn_info.conv_manager.is_u64_used(u64))
  305. {
  306. if(conn_info.conv_manager.get_size() >=max_conv_num)
  307. {
  308. mylog(log_warn,"ignored new udp connect bc max_conv_num exceed\n");
  309. continue;
  310. }
  311. conv=conn_info.conv_manager.get_new_conv();
  312. conn_info.conv_manager.insert_conv(conv,u64);
  313. mylog(log_info,"new packet from %s:%d,conv_id=%x\n",inet_ntoa(udp_new_addr_in.sin_addr),ntohs(udp_new_addr_in.sin_port),conv);
  314. }
  315. else
  316. {
  317. conv=conn_info.conv_manager.find_conv_by_u64(u64);
  318. }
  319. conn_info.conv_manager.update_active_time(conv);
  320. char * new_data;
  321. int new_len;
  322. put_conv(conv,data,data_len,new_data,new_len);
  323. //dest.conv=conv;
  324. from_normal_to_fec(conn_info,new_data,new_len,out_n,out_arr,out_len,out_delay);
  325. }
  326. if(out_n!=-1)
  327. {
  328. mylog(log_debug,"n=%d\n",out_n);
  329. }
  330. for(int i=0;i<out_n;i++)
  331. {
  332. delay_send(out_delay[i],dest,out_arr[i],out_len[i]);
  333. }
  334. //my_send(dest,data,data_len);
  335. }
  336. else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()) {
  337. uint64_t value;
  338. read(delay_manager.get_timer_fd(), &value, 8);
  339. //printf("<timerfd_triggered, %d>",delay_mp.size());
  340. //fflush(stdout);
  341. }
  342. else if(events[idx].data.u64>u32_t(-1) )
  343. {
  344. char data[buf_len];
  345. if(!fd_manager.exist(events[idx].data.u64)) //fd64 has been closed
  346. {
  347. continue;
  348. }
  349. assert(events[idx].data.u64==remote_fd64);
  350. int fd=fd_manager.to_fd(remote_fd64);
  351. int data_len =recv(fd,data,max_data_len,0);
  352. mylog(log_trace, "received data from udp fd %d, len=%d\n", remote_fd,data_len);
  353. if(data_len<0)
  354. {
  355. if(errno==ECONNREFUSED)
  356. {
  357. //conn_manager.clear_list.push_back(udp_fd);
  358. mylog(log_debug, "recv failed %d ,udp_fd%d,errno:%s\n", data_len,remote_fd,strerror(errno));
  359. }
  360. mylog(log_warn, "recv failed %d ,udp_fd%d,errno:%s\n", data_len,remote_fd,strerror(errno));
  361. continue;
  362. }
  363. if(data_len>mtu_warn)
  364. {
  365. mylog(log_warn,"huge packet,data len=%d (>%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ",data_len,mtu_warn);
  366. }
  367. if(rm_crc32(data,data_len)!=0)
  368. {
  369. mylog(log_debug,"crc32 check error");
  370. }
  371. int out_n;char **out_arr;int *out_len;int *out_delay;
  372. from_fec_to_normal(conn_info,data,data_len,out_n,out_arr,out_len,out_delay);
  373. for(int i=0;i<out_n;i++)
  374. {
  375. u32_t conv;
  376. char *new_data;
  377. int new_len;
  378. if(get_conv(conv,out_arr[i],out_len[i],new_data,new_len)!=0)
  379. continue;
  380. if(!conn_info.conv_manager.is_conv_used(conv))continue;
  381. conn_info.conv_manager.update_active_time(conv);
  382. u64_t u64=conn_info.conv_manager.find_u64_by_conv(conv);
  383. dest_t dest;
  384. dest.inner.ip_port.from_u64(u64);
  385. dest.type=type_ip_port;
  386. //dest.conv=conv;
  387. delay_send(out_delay[i],dest,new_data,new_len);
  388. }
  389. //mylog(log_trace,"[%s] send packet\n",dest.inner.ip_port.to_s());
  390. }
  391. /*
  392. else if(events[idx].data.u64 ==(u64_t)timer_fd)
  393. {
  394. u64_t value;
  395. read(timer_fd, &value, 8);
  396. client_on_timer(conn_info);
  397. mylog(log_trace,"epoll_trigger_counter: %d \n",epoll_trigger_counter);
  398. epoll_trigger_counter=0;
  399. }*/
  400. else
  401. {
  402. mylog(log_fatal,"unknown fd,this should never happen\n");
  403. myexit(-1);
  404. }
  405. }
  406. delay_manager.check();
  407. }
  408. return 0;
  409. }
  410. int server_event_loop()
  411. {
  412. //char buf[buf_len];
  413. int i, j, k;int ret;
  414. int yes = 1;
  415. int epoll_fd;
  416. int remote_fd;
  417. // conn_info_t conn_info;
  418. init_listen_socket();
  419. epoll_fd = epoll_create1(0);
  420. const int max_events = 4096;
  421. struct epoll_event ev, events[max_events];
  422. if (epoll_fd < 0) {
  423. mylog(log_fatal,"epoll return %d\n", epoll_fd);
  424. myexit(-1);
  425. }
  426. ev.events = EPOLLIN;
  427. ev.data.u64 = local_listen_fd;
  428. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, local_listen_fd, &ev);
  429. if (ret!=0) {
  430. mylog(log_fatal,"add udp_listen_fd error\n");
  431. myexit(-1);
  432. }
  433. ev.events = EPOLLIN;
  434. ev.data.u64 = delay_manager.get_timer_fd();
  435. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
  436. if (ret!= 0) {
  437. mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
  438. myexit(-1);
  439. }
  440. mylog(log_info,"now listening at %s:%d\n",my_ntoa(local_ip_uint32),local_port);
  441. my_timer_t timer;
  442. timer.add_fd_to_epoll(epoll_fd);
  443. timer.set_timer_repeat_us(timer_interval*1000);
  444. while(1)////////////////////////
  445. {
  446. if(about_to_exit) myexit(0);
  447. int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
  448. if (nfds < 0) { //allow zero
  449. if(errno==EINTR )
  450. {
  451. mylog(log_info,"epoll interrupted by signal\n");
  452. myexit(0);
  453. }
  454. else
  455. {
  456. mylog(log_fatal,"epoll_wait return %d\n", nfds);
  457. myexit(-1);
  458. }
  459. }
  460. int idx;
  461. for (idx = 0; idx < nfds; ++idx)
  462. {
  463. /*
  464. if ((events[idx].data.u64 ) == (u64_t)timer_fd)
  465. {
  466. conn_manager.clear_inactive();
  467. u64_t dummy;
  468. read(timer_fd, &dummy, 8);
  469. //current_time_rough=get_current_time();
  470. }
  471. else */
  472. if(events[idx].data.u64==(u64_t)timer.get_timer_fd())
  473. {
  474. uint64_t value;
  475. read(timer.get_timer_fd(), &value, 8);
  476. conn_manager.clear_inactive();
  477. //conn_info.conv_manager.clear_inactive();
  478. }
  479. else if (events[idx].data.u64 == (u64_t)local_listen_fd)
  480. {
  481. //int recv_len;
  482. char data[buf_len];
  483. int data_len;
  484. struct sockaddr_in udp_new_addr_in={0};
  485. socklen_t udp_new_addr_len = sizeof(sockaddr_in);
  486. if ((data_len = recvfrom(local_listen_fd, data, max_data_len, 0,
  487. (struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) == -1) {
  488. mylog(log_error,"recv_from error,this shouldnt happen at client\n");
  489. myexit(1);
  490. };
  491. if(data_len>=mtu_warn)
  492. {
  493. mylog(log_warn,"huge packet,data len=%d (>=%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ",data_len,mtu_warn);
  494. }
  495. if(rm_crc32(data,data_len)!=0)
  496. {
  497. mylog(log_debug,"crc32 check error");
  498. }
  499. mylog(log_trace,"Received packet from %s:%d,len: %d\n", inet_ntoa(udp_new_addr_in.sin_addr),
  500. ntohs(udp_new_addr_in.sin_port),data_len);
  501. ip_port_t ip_port;
  502. ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
  503. ip_port.port=ntohs(udp_new_addr_in.sin_port);
  504. if(!conn_manager.exist(ip_port))
  505. {
  506. conn_manager.insert(ip_port);
  507. conn_info_t &conn_info=conn_manager.find(ip_port);
  508. conn_info.fec_encode_manager.re_init(fec_data_num,fec_redundant_num,fec_mtu,fec_pending_num,fec_pending_time);
  509. conn_info.conv_manager.reserve();
  510. u64_t fec_fd64=conn_info.fec_encode_manager.get_timer_fd64();
  511. ev.events = EPOLLIN;
  512. ev.data.u64 = fec_fd64;
  513. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(fec_fd64), &ev);
  514. fd_manager.get_info(fec_fd64).ip_port=ip_port;
  515. conn_info.timer.add_fd64_to_epoll(epoll_fd);
  516. conn_info.timer.set_timer_repeat_us(timer_interval*1000);
  517. u64_t timer_fd64=conn_info.timer.get_timer_fd64();
  518. fd_manager.get_info(timer_fd64).ip_port=ip_port;
  519. mylog(log_info,"new connection from %s\n",ip_port.to_s());
  520. }
  521. conn_info_t &conn_info=conn_manager.find(ip_port);
  522. int out_n;char **out_arr;int *out_len;int *out_delay;
  523. from_fec_to_normal(conn_info,data,data_len,out_n,out_arr,out_len,out_delay);
  524. for(int i=0;i<out_n;i++)
  525. {
  526. u32_t conv;
  527. char *new_data;
  528. int new_len;
  529. if(get_conv(conv,out_arr[i],out_len[i],new_data,new_len)!=0)
  530. continue;
  531. /*
  532. id_t tmp_conv_id;
  533. memcpy(&tmp_conv_id,&data_[0],sizeof(tmp_conv_id));
  534. tmp_conv_id=ntohl(tmp_conv_id);*/
  535. if (!conn_info.conv_manager.is_conv_used(conv))
  536. {
  537. int new_udp_fd;
  538. ret=new_connected_socket(new_udp_fd,remote_ip_uint32,remote_port);
  539. if (ret != 0) {
  540. mylog(log_warn, "[%s:%d]new_connected_socket failed\n",my_ntoa(ip_port.ip),ip_port.port);
  541. continue;
  542. }
  543. fd64_t fd64 = fd_manager.create(new_udp_fd);
  544. ev.events = EPOLLIN;
  545. ev.data.u64 = fd64;
  546. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, new_udp_fd, &ev);
  547. conn_info.conv_manager.insert_conv(conv, fd64);
  548. fd_manager.get_info(fd64).ip_port=ip_port;
  549. mylog(log_info,"[%s]new conv %x,fd %d created\n",ip_port.to_s(),conv,new_udp_fd);
  550. //assert(!conn_manager.exist_fd64(fd64));
  551. //conn_manager.insert_fd64(fd64,ip_port);
  552. }
  553. conn_info.conv_manager.update_active_time(conv);
  554. fd64_t fd64= conn_info.conv_manager.find_u64_by_conv(conv);
  555. //int fd=fd_manager.fd64_to_fd(fd64);
  556. dest_t dest;
  557. dest.type=type_fd64;
  558. dest.inner.fd64=fd64;
  559. //dest.conv=conv;
  560. delay_send(out_delay[i],dest,new_data,new_len);
  561. }
  562. //int fd = int((u64 << 32u) >> 32u);
  563. //////////////////////////////todo
  564. //u64_t u64=((u64_t(udp_new_addr_in.sin_addr.s_addr))<<32u)+ntohs(udp_new_addr_in.sin_port);
  565. }
  566. /*
  567. else if ((events[idx].data.u64 >>32u) == 2u)
  568. {
  569. if(debug_flag)begin_time=get_current_time();
  570. int fd=get_u64_l(events[idx].data.u64);
  571. u64_t dummy;
  572. read(fd, &dummy, 8);
  573. if(conn_manager.timer_fd_mp.find(fd)==conn_manager.timer_fd_mp.end()) //this can happen,when fd is a just closed fd
  574. {
  575. mylog(log_info,"timer_fd no longer exits\n");
  576. continue;
  577. }
  578. conn_info_t* p_conn_info=conn_manager.timer_fd_mp[fd];
  579. u32_t ip=p_conn_info->raw_info.recv_info.src_ip;
  580. u32_t port=p_conn_info->raw_info.recv_info.src_port;
  581. assert(conn_manager.exist(ip,port));//TODO remove this for peformance
  582. assert(p_conn_info->state.server_current_state == server_ready); //TODO remove this for peformance
  583. //conn_info_t &conn_info=conn_manager.find(ip,port);
  584. char ip_port[40];
  585. sprintf(ip_port,"%s:%d",my_ntoa(ip),port);
  586. server_on_timer_multi(*p_conn_info,ip_port);
  587. if(debug_flag)
  588. {
  589. end_time=get_current_time();
  590. mylog(log_debug,"(events[idx].data.u64 >>32u) == 2u ,%llu,%llu,%llu \n",begin_time,end_time,end_time-begin_time);
  591. }
  592. }*/
  593. else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()) {
  594. uint64_t value;
  595. read(delay_manager.get_timer_fd(), &value, 8);
  596. //printf("<timerfd_triggered, %d>",delay_mp.size());
  597. //fflush(stdout);
  598. }
  599. else if (events[idx].data.u64 >u32_t(-1))
  600. {
  601. char data[buf_len];
  602. int data_len;
  603. u32_t conv;
  604. fd64_t fd64=events[idx].data.u64;
  605. if(!fd_manager.exist(fd64)) //fd64 has been closed
  606. {
  607. continue;
  608. }
  609. //assert(conn_manager.exist_fd64(fd64));
  610. assert(fd_manager.exist_info(fd64));
  611. ip_port_t ip_port=fd_manager.get_info(fd64).ip_port;
  612. assert(conn_manager.exist(ip_port));
  613. //conn_info_t* p_conn_info=conn_manager.find_insert_p(ip_port);
  614. conn_info_t &conn_info=conn_manager.find(ip_port);
  615. int out_n;char **out_arr;int *out_len;int *out_delay;
  616. dest_t dest;
  617. dest.type=type_ip_port;
  618. //dest.conv=conv;
  619. dest.inner.ip_port=ip_port;
  620. dest.cook=1;
  621. if(fd64==conn_info.fec_encode_manager.get_timer_fd64())
  622. {
  623. mylog(log_info,"timer!!!\n");
  624. uint64_t value;
  625. if(read(fd_manager.to_fd(fd64), &value, 8)!=8)
  626. {
  627. continue;
  628. }
  629. if(value==0)
  630. {
  631. continue;
  632. }
  633. assert(value==1);
  634. from_normal_to_fec(conn_info,0,0,out_n,out_arr,out_len,out_delay);
  635. }
  636. else if(fd64==conn_info.timer.get_timer_fd64())
  637. {
  638. uint64_t value;
  639. read(conn_info.timer.get_timer_fd(), &value, 8);
  640. conn_info.conv_manager.clear_inactive();
  641. }
  642. else
  643. {
  644. assert(conn_info.conv_manager.is_u64_used(fd64));
  645. conv=conn_info.conv_manager.find_conv_by_u64(fd64);
  646. conn_info.conv_manager.update_active_time(conv);
  647. int fd=fd_manager.to_fd(fd64);
  648. data_len=recv(fd,data,max_data_len,0);
  649. mylog(log_trace,"received a packet from udp_fd,len:%d\n",data_len);
  650. if(data_len<0)
  651. {
  652. mylog(log_debug,"udp fd,recv_len<0 continue,%s\n",strerror(errno));
  653. continue;
  654. }
  655. if(data_len>=mtu_warn)
  656. {
  657. mylog(log_warn,"huge packet,data len=%d (>=%d).strongly suggested to set a smaller mtu at upper level,to get rid of this warn\n ",data_len,mtu_warn);
  658. }
  659. char * new_data;
  660. int new_len;
  661. put_conv(conv,data,data_len,new_data,new_len);
  662. from_normal_to_fec(conn_info,new_data,new_len,out_n,out_arr,out_len,out_delay);
  663. }
  664. for(int i=0;i<out_n;i++)
  665. {
  666. delay_send(out_delay[i],dest,out_arr[i],out_len[i]);
  667. }
  668. //mylog(log_trace,"[%s] send packet\n",ip_port.to_s());
  669. }
  670. else
  671. {
  672. mylog(log_fatal,"unknown fd,this should never happen\n");
  673. myexit(-1);
  674. }
  675. }
  676. delay_manager.check();
  677. }
  678. return 0;
  679. }
  680. int unit_test()
  681. {
  682. int i,j,k;
  683. void *code=fec_new(3,6);
  684. char arr[6][100]=
  685. {
  686. "aaa","bbb","ccc"
  687. ,"ddd","eee","fff"
  688. };
  689. char *data[6];
  690. for(i=0;i<6;i++)
  691. {
  692. data[i]=arr[i];
  693. }
  694. rs_encode2(3,6,data,3);
  695. //printf("%d %d",(int)(unsigned char)arr[5][0],(int)('a'^'b'^'c'^'d'^'e'));
  696. for(i=0;i<6;i++)
  697. {
  698. printf("<%s>",data[i]);
  699. }
  700. data[0]=0;
  701. //data[1]=0;
  702. //data[5]=0;
  703. int ret=rs_decode2(3,6,data,3);
  704. printf("ret:%d\n",ret);
  705. for(i=0;i<6;i++)
  706. {
  707. printf("<%s>",data[i]);
  708. }
  709. fec_free(code);
  710. char arr2[6][100]=
  711. {
  712. "aaa11111","","ccc333333333"
  713. ,"ddd444","eee5555","ff6666"
  714. };
  715. blob_encode_t blob_encode;
  716. for(int i=0;i<6;i++)
  717. blob_encode.input(arr2[i],strlen(arr2[i]));
  718. char **output;
  719. int shard_len;
  720. blob_encode.output(7,output,shard_len);
  721. printf("<shard_len:%d>",shard_len);
  722. blob_decode_t blob_decode;
  723. for(int i=0;i<7;i++)
  724. {
  725. blob_decode.input(output[i],shard_len);
  726. }
  727. char **decode_output;
  728. int * len_arr;
  729. int num;
  730. ret=blob_decode.output(num,decode_output,len_arr);
  731. printf("<num:%d,ret:%d>\n",num,ret);
  732. for(int i=0;i<num;i++)
  733. {
  734. char buf[1000]={0};
  735. memcpy(buf,decode_output[i],len_arr[i]);
  736. printf("<%d:%s>",len_arr[i],buf);
  737. }
  738. printf("\n");
  739. static fec_encode_manager_t fec_encode_manager;
  740. static fec_decode_manager_t fec_decode_manager;
  741. {
  742. string a = "11111";
  743. string b = "22";
  744. string c = "33333333";
  745. fec_encode_manager.input((char *) a.c_str(), a.length());
  746. fec_encode_manager.input((char *) b.c_str(), b.length());
  747. fec_encode_manager.input((char *) c.c_str(), c.length());
  748. fec_encode_manager.input(0, 0);
  749. int n;
  750. char **s_arr;
  751. int len;
  752. fec_encode_manager.output(n,s_arr,len);
  753. printf("<n:%d,len:%d>",n,len);
  754. for(int i=0;i<n;i++)
  755. {
  756. fec_decode_manager.input(s_arr[i],len);
  757. }
  758. {
  759. int n;char ** s_arr;int* len_arr;
  760. fec_decode_manager.output(n,s_arr,len_arr);
  761. printf("<n:%d>",n);
  762. for(int i=0;i<n;i++)
  763. {
  764. s_arr[i][len_arr[i]]=0;
  765. printf("<%s>\n",s_arr[i]);
  766. }
  767. }
  768. }
  769. {
  770. string a = "aaaaaaa";
  771. string b = "bbbbbbbbbbbbb";
  772. string c = "ccc";
  773. fec_encode_manager.input((char *) a.c_str(), a.length());
  774. fec_encode_manager.input((char *) b.c_str(), b.length());
  775. fec_encode_manager.input((char *) c.c_str(), c.length());
  776. fec_encode_manager.input(0, 0);
  777. int n;
  778. char **s_arr;
  779. int len;
  780. fec_encode_manager.output(n,s_arr,len);
  781. printf("<n:%d,len:%d>",n,len);
  782. for(int i=0;i<n;i++)
  783. {
  784. if(i==1||i==3||i==5||i==0)
  785. fec_decode_manager.input(s_arr[i],len);
  786. }
  787. {
  788. int n;char ** s_arr;int* len_arr;
  789. fec_decode_manager.output(n,s_arr,len_arr);
  790. printf("<n:%d>",n);
  791. for(int i=0;i<n;i++)
  792. {
  793. s_arr[i][len_arr[i]]=0;
  794. printf("<%s>\n",s_arr[i]);
  795. }
  796. }
  797. }
  798. return 0;
  799. }
  800. void print_help()
  801. {
  802. char git_version_buf[100]={0};
  803. strncpy(git_version_buf,gitversion,10);
  804. printf("UDPspeeder\n");
  805. printf("git version:%s ",git_version_buf);
  806. printf("build date:%s %s\n",__DATE__,__TIME__);
  807. printf("repository: https://github.com/wangyu-/UDPspeeder\n");
  808. printf("\n");
  809. printf("usage:\n");
  810. printf(" run as client : ./this_program -c -l local_listen_ip:local_port -r server_ip:server_port [options]\n");
  811. printf(" run as server : ./this_program -s -l server_listen_ip:server_port -r remote_ip:remote_port [options]\n");
  812. printf("\n");
  813. printf("common option,must be same on both sides:\n");
  814. printf(" -k,--key <string> key for simple xor encryption,default:\"secret key\"\n");
  815. printf("main options:\n");
  816. printf(" -d <number> duplicated packet number, -d 0 means no duplicate. default value:0\n");
  817. printf(" -t <number> duplicated packet delay time, unit: 0.1ms,default value:20(2ms)\n");
  818. printf(" -j <number> simulated jitter.randomly delay first packet for 0~jitter_value*0.1 ms,to\n");
  819. printf(" create simulated jitter.default value:0.do not use if you dont\n");
  820. printf(" know what it means\n");
  821. printf(" --report <number> turn on udp send/recv report,and set a time interval for reporting,unit:s\n");
  822. printf("advanced options:\n");
  823. printf(" -t tmin:tmax simliar to -t above,but delay randomly between tmin and tmax\n");
  824. printf(" -j jmin:jmax simliar to -j above,but create jitter randomly between jmin and jmax\n");
  825. printf(" --random-drop <number> simulate packet loss ,unit:0.01%%\n");
  826. printf(" --disable-filter disable duplicate packet filter.\n");
  827. printf(" -m <number> max pending packets,to prevent the program from eating up all your memory,\n");
  828. printf(" default value:0(disabled).\n");
  829. printf("other options:\n");
  830. printf(" --log-level <number> 0:never 1:fatal 2:error 3:warn \n");
  831. printf(" 4:info (default) 5:debug 6:trace\n");
  832. printf(" --log-position enable file name,function name,line number in log\n");
  833. printf(" --disable-color disable log color\n");
  834. printf(" --sock-buf <number> buf size for socket,>=10 and <=10240,unit:kbyte,default:1024\n");
  835. //printf(" -p use multi-process mode instead of epoll.very costly,only for test,do dont use\n");
  836. printf(" -h,--help print this help message\n");
  837. //printf("common options,these options must be same on both side\n");
  838. }
  839. void process_arg(int argc, char *argv[])
  840. {
  841. int is_client=0,is_server=0;
  842. int i, j, k;
  843. int opt;
  844. static struct option long_options[] =
  845. {
  846. {"log-level", required_argument, 0, 1},
  847. {"log-position", no_argument, 0, 1},
  848. {"disable-color", no_argument, 0, 1},
  849. {"disable-filter", no_argument, 0, 1},
  850. {"sock-buf", required_argument, 0, 1},
  851. {"random-drop", required_argument, 0, 1},
  852. {"report", required_argument, 0, 1},
  853. {NULL, 0, 0, 0}
  854. };
  855. int option_index = 0;
  856. if (argc == 1)
  857. {
  858. print_help();
  859. myexit( -1);
  860. }
  861. for (i = 0; i < argc; i++)
  862. {
  863. if(strcmp(argv[i],"--unit-test")==0)
  864. {
  865. unit_test();
  866. myexit(0);
  867. }
  868. }
  869. for (i = 0; i < argc; i++)
  870. {
  871. if(strcmp(argv[i],"-h")==0||strcmp(argv[i],"--help")==0)
  872. {
  873. print_help();
  874. myexit(0);
  875. }
  876. }
  877. for (i = 0; i < argc; i++)
  878. {
  879. if(strcmp(argv[i],"--log-level")==0)
  880. {
  881. if(i<argc -1)
  882. {
  883. sscanf(argv[i+1],"%d",&log_level);
  884. if(0<=log_level&&log_level<log_end)
  885. {
  886. }
  887. else
  888. {
  889. log_bare(log_fatal,"invalid log_level\n");
  890. myexit(-1);
  891. }
  892. }
  893. }
  894. if(strcmp(argv[i],"--disable-color")==0)
  895. {
  896. enable_log_color=0;
  897. }
  898. }
  899. mylog(log_info,"argc=%d ", argc);
  900. for (i = 0; i < argc; i++) {
  901. log_bare(log_info, "%s ", argv[i]);
  902. }
  903. log_bare(log_info, "\n");
  904. if (argc == 1)
  905. {
  906. print_help();
  907. myexit(-1);
  908. }
  909. int no_l = 1, no_r = 1;
  910. while ((opt = getopt_long(argc, argv, "l:r:d:t:hcspk:j:m:",long_options,&option_index)) != -1)
  911. {
  912. //string opt_key;
  913. //opt_key+=opt;
  914. switch (opt)
  915. {
  916. case 'p':
  917. //multi_process_mode=1;
  918. break;
  919. case 'k':
  920. sscanf(optarg,"%s\n",key_string);
  921. mylog(log_debug,"key=%s\n",key_string);
  922. if(strlen(key_string)==0)
  923. {
  924. mylog(log_fatal,"key len=0??\n");
  925. myexit(-1);
  926. }
  927. break;
  928. case 'm':
  929. sscanf(optarg,"%d\n",&max_pending_packet);
  930. if(max_pending_packet<1000)
  931. {
  932. mylog(log_fatal,"max_pending_packet must be >1000\n");
  933. myexit(-1);
  934. }
  935. break;
  936. case 'j':
  937. if (strchr(optarg, ':') == 0)
  938. {
  939. int jitter;
  940. sscanf(optarg,"%d\n",&jitter);
  941. if(jitter<0 ||jitter>1000*100)
  942. {
  943. mylog(log_fatal,"jitter must be between 0 and 100,000(10 second)\n");
  944. myexit(-1);
  945. }
  946. jitter_min=0;
  947. jitter_max=jitter;
  948. }
  949. else
  950. {
  951. sscanf(optarg,"%d:%d\n",&jitter_min,&jitter_max);
  952. if(jitter_min<0 ||jitter_max<0||jitter_min>jitter_max)
  953. {
  954. mylog(log_fatal," must satisfy 0<=jmin<=jmax\n");
  955. myexit(-1);
  956. }
  957. }
  958. break;
  959. case 't':
  960. if (strchr(optarg, ':') == 0)
  961. {
  962. int dup_delay=-1;
  963. sscanf(optarg,"%d\n",&dup_delay);
  964. if(dup_delay<1||dup_delay>1000*100)
  965. {
  966. mylog(log_fatal,"dup_delay must be between 1 and 100,000(10 second)\n");
  967. myexit(-1);
  968. }
  969. dup_delay_min=dup_delay_max=dup_delay;
  970. }
  971. else
  972. {
  973. sscanf(optarg,"%d:%d\n",&dup_delay_min,&dup_delay_max);
  974. if(dup_delay_min<1 ||dup_delay_max<1||dup_delay_min>dup_delay_max)
  975. {
  976. mylog(log_fatal," must satisfy 1<=dmin<=dmax\n");
  977. myexit(-1);
  978. }
  979. }
  980. break;
  981. case 'd':
  982. dup_num=-1;
  983. sscanf(optarg,"%d\n",&dup_num);
  984. if(dup_num<0 ||dup_num>5)
  985. {
  986. mylog(log_fatal,"dup_num must be between 0 and 5\n");
  987. myexit(-1);
  988. }
  989. dup_num+=1;
  990. break;
  991. case 'c':
  992. is_client = 1;
  993. break;
  994. case 's':
  995. is_server = 1;
  996. break;
  997. case 'l':
  998. no_l = 0;
  999. if (strchr(optarg, ':') != 0)
  1000. {
  1001. sscanf(optarg, "%[^:]:%d", local_ip, &local_port);
  1002. }
  1003. else
  1004. {
  1005. mylog(log_fatal," -r ip:port\n");
  1006. myexit(1);
  1007. strcpy(local_ip, "127.0.0.1");
  1008. sscanf(optarg, "%d", &local_port);
  1009. }
  1010. break;
  1011. case 'r':
  1012. no_r = 0;
  1013. if (strchr(optarg, ':') != 0)
  1014. {
  1015. //printf("in :\n");
  1016. //printf("%s\n",optarg);
  1017. sscanf(optarg, "%[^:]:%d", remote_ip, &remote_port);
  1018. //printf("%d\n",remote_port);
  1019. }
  1020. else
  1021. {
  1022. mylog(log_fatal," -r ip:port\n");
  1023. myexit(1);
  1024. strcpy(remote_ip, "127.0.0.1");
  1025. sscanf(optarg, "%d", &remote_port);
  1026. }
  1027. break;
  1028. case 'h':
  1029. break;
  1030. case 1:
  1031. if(strcmp(long_options[option_index].name,"log-level")==0)
  1032. {
  1033. }
  1034. else if(strcmp(long_options[option_index].name,"disable-filter")==0)
  1035. {
  1036. disable_replay_filter=1;
  1037. //enable_log_color=0;
  1038. }
  1039. else if(strcmp(long_options[option_index].name,"disable-color")==0)
  1040. {
  1041. //enable_log_color=0;
  1042. }
  1043. else if(strcmp(long_options[option_index].name,"log-position")==0)
  1044. {
  1045. enable_log_position=1;
  1046. }
  1047. else if(strcmp(long_options[option_index].name,"random-drop")==0)
  1048. {
  1049. sscanf(optarg,"%d",&random_drop);
  1050. if(random_drop<0||random_drop>10000)
  1051. {
  1052. mylog(log_fatal,"random_drop must be between 0 10000 \n");
  1053. myexit(-1);
  1054. }
  1055. }
  1056. else if(strcmp(long_options[option_index].name,"report")==0)
  1057. {
  1058. sscanf(optarg,"%d",&report_interval);
  1059. if(report_interval<=0)
  1060. {
  1061. mylog(log_fatal,"report_interval must be >0 \n");
  1062. myexit(-1);
  1063. }
  1064. }
  1065. else if(strcmp(long_options[option_index].name,"sock-buf")==0)
  1066. {
  1067. int tmp=-1;
  1068. sscanf(optarg,"%d",&tmp);
  1069. if(10<=tmp&&tmp<=10*1024)
  1070. {
  1071. socket_buf_size=tmp*1024;
  1072. }
  1073. else
  1074. {
  1075. mylog(log_fatal,"sock-buf value must be between 1 and 10240 (kbyte) \n");
  1076. myexit(-1);
  1077. }
  1078. }
  1079. else
  1080. {
  1081. mylog(log_fatal,"unknown option\n");
  1082. myexit(-1);
  1083. }
  1084. break;
  1085. default:
  1086. mylog(log_fatal,"unknown option <%x>", opt);
  1087. myexit(-1);
  1088. }
  1089. }
  1090. if (no_l)
  1091. mylog(log_fatal,"error: -i not found\n");
  1092. if (no_r)
  1093. mylog(log_fatal,"error: -o not found\n");
  1094. if (no_l || no_r)
  1095. myexit(-1);
  1096. if (is_client == 0 && is_server == 0)
  1097. {
  1098. mylog(log_fatal,"-s -c hasnt been set\n");
  1099. myexit(-1);
  1100. }
  1101. if (is_client == 1 && is_server == 1)
  1102. {
  1103. mylog(log_fatal,"-s -c cant be both set\n");
  1104. myexit(-1);
  1105. }
  1106. if(is_client==1)
  1107. {
  1108. program_mode=client_mode;
  1109. }
  1110. else
  1111. {
  1112. program_mode=server_mode;
  1113. }
  1114. }
  1115. int main(int argc, char *argv[])
  1116. {
  1117. /*
  1118. if(argc==1||argc==0)
  1119. {
  1120. printf("this_program classic\n");
  1121. printf("this_program fec\n");
  1122. return 0;
  1123. }*/
  1124. /*
  1125. if(argc>=2&&strcmp(argv[1],"fec")!=0)
  1126. {
  1127. printf("running into classic mode!\n");
  1128. return classic::main(argc,argv);
  1129. }*/
  1130. assert(sizeof(u64_t)==8);
  1131. assert(sizeof(i64_t)==8);
  1132. assert(sizeof(u32_t)==4);
  1133. assert(sizeof(i32_t)==4);
  1134. dup2(1, 2); //redirect stderr to stdout
  1135. int i, j, k;
  1136. process_arg(argc,argv);
  1137. delay_manager.capacity=max_pending_packet;
  1138. init_random_number_fd();
  1139. local_ip_uint32=inet_addr(local_ip);
  1140. remote_ip_uint32=inet_addr(remote_ip);
  1141. fd_manager.reserve(10007);
  1142. if(program_mode==client_mode)
  1143. {
  1144. client_event_loop();
  1145. }
  1146. else
  1147. {
  1148. server_event_loop();
  1149. }
  1150. return 0;
  1151. }