tun_dev.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950
  1. /*
  2. * tun.cpp
  3. *
  4. * Created on: Oct 26, 2017
  5. * Author: root
  6. */
  7. #include "common.h"
  8. #include "log.h"
  9. #include "misc.h"
  10. #include <netinet/tcp.h> //Provides declarations for tcp header
  11. #include <netinet/udp.h>
  12. #include <netinet/ip.h> //Provides declarations for ip header
  13. #include <netinet/if_ether.h>
  14. #include <linux/if.h>
  15. #include <linux/if_tun.h>
  16. my_time_t last_keep_alive_time=0;
  17. int keep_alive_interval=3000;//3000ms
  18. int get_tun_fd(char * dev_name)
  19. {
  20. int tun_fd=open("/dev/net/tun",O_RDWR);
  21. if(tun_fd <0)
  22. {
  23. mylog(log_fatal,"open /dev/net/tun failed");
  24. myexit(-1);
  25. }
  26. struct ifreq ifr;
  27. memset(&ifr, 0, sizeof(ifr));
  28. ifr.ifr_flags = IFF_TUN|IFF_NO_PI;
  29. strncpy(ifr.ifr_name, dev_name, IFNAMSIZ);
  30. if(ioctl(tun_fd, TUNSETIFF, (void *)&ifr) != 0)
  31. {
  32. mylog(log_fatal,"open /dev/net/tun failed");
  33. myexit(-1);
  34. }
  35. return tun_fd;
  36. }
  37. int set_if(char *if_name,u32_t local_ip,u32_t remote_ip,int mtu)
  38. {
  39. //printf("i m here1\n");
  40. struct ifreq ifr;
  41. struct sockaddr_in sai;
  42. memset(&ifr,0,sizeof(ifr));
  43. memset(&sai, 0, sizeof(struct sockaddr));
  44. int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
  45. strncpy(ifr.ifr_name, if_name, IFNAMSIZ);
  46. sai.sin_family = AF_INET;
  47. sai.sin_port = 0;
  48. sai.sin_addr.s_addr = local_ip;
  49. memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
  50. assert(ioctl(sockfd, SIOCSIFADDR, &ifr)==0); //set source ip
  51. sai.sin_addr.s_addr = remote_ip;
  52. memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
  53. assert(ioctl(sockfd, SIOCSIFDSTADDR, &ifr)==0);//set dest ip
  54. ifr.ifr_mtu=mtu;
  55. assert(ioctl(sockfd, SIOCSIFMTU, &ifr)==0);//set mtu
  56. assert(ioctl(sockfd, SIOCGIFFLAGS, &ifr)==0);
  57. // ifr.ifr_flags |= ( IFF_UP|IFF_POINTOPOINT|IFF_RUNNING|IFF_NOARP|IFF_MULTICAST );
  58. ifr.ifr_flags = ( IFF_UP|IFF_POINTOPOINT|IFF_RUNNING|IFF_NOARP|IFF_MULTICAST );//set interface flags
  59. assert(ioctl(sockfd, SIOCSIFFLAGS, &ifr)==0);
  60. //printf("i m here2\n");
  61. return 0;
  62. }
  63. const char header_normal=1;
  64. const char header_new_connect=2;
  65. const char header_reject=3;
  66. const char header_keep_alive=4;
  67. int put_header(char header,char * data,int &len)
  68. {
  69. assert(len>=0);
  70. data[len]=header;
  71. len+=1;
  72. return 0;
  73. }
  74. int get_header(char &header,char * data,int &len)
  75. {
  76. assert(len>=0);
  77. if(len<1) return -1;
  78. len-=1;
  79. header=data[len];
  80. return 0;
  81. }
  82. int from_normal_to_fec2(conn_info_t & conn_info,dest_t &dest,char * data,int len,char header)
  83. {
  84. int out_n;char **out_arr;int *out_len;my_time_t *out_delay;
  85. from_normal_to_fec(conn_info,data,len,out_n,out_arr,out_len,out_delay);
  86. for(int i=0;i<out_n;i++)
  87. {
  88. char tmp_buf[buf_len];
  89. int tmp_len=out_len[i];
  90. memcpy(tmp_buf,out_arr[i],out_len[i]);
  91. put_header(header,tmp_buf,tmp_len);
  92. delay_send(out_delay[i],dest,tmp_buf,tmp_len);//this is slow but safer.just use this one
  93. //put_header(header,out_arr[i],out_len[i]);//modify in place
  94. //delay_send(out_delay[i],dest,out_arr[i],out_len[i]);//warning this is currently okay,but if you modified fec encoder,you may have to use the above code
  95. }
  96. return 0;
  97. }
  98. int from_fec_to_normal2(conn_info_t & conn_info,dest_t &dest,char * data,int len)
  99. {
  100. int out_n;char **out_arr;int *out_len;my_time_t *out_delay;
  101. from_fec_to_normal(conn_info,data,len,out_n,out_arr,out_len,out_delay);
  102. for(int i=0;i<out_n;i++)
  103. {
  104. #ifndef NOLIMIT
  105. if(program_mode==server_mode)
  106. {
  107. char * tmp_data=out_arr[i];
  108. int tmp_len=out_len[i];
  109. iphdr * iph;
  110. iph = (struct iphdr *) tmp_data;
  111. if(tmp_len>=int(sizeof(iphdr))&&iph->version==4)
  112. {
  113. u32_t dest_ip=iph->daddr;
  114. //printf("%s\n",my_ntoa(dest_ip));
  115. if( ( ntohl(sub_net_uint32)&0xFFFFFF00 ) != ( ntohl (dest_ip) &0xFFFFFF00) )
  116. {
  117. string sub=my_ntoa(dest_ip);
  118. string dst=my_ntoa( htonl( ntohl (sub_net_uint32) &0xFFFFFF00) );
  119. mylog(log_warn,"[restriction]packet's dest ip [%s] not in subnet [%s],dropped, maybe you need to compile an un-restricted server\n", sub.c_str(), dst.c_str());
  120. continue;
  121. }
  122. }
  123. }
  124. #endif
  125. delay_send(out_delay[i],dest,out_arr[i],out_len[i]);
  126. }
  127. return 0;
  128. }
  129. int do_mssfix(char * s,int len)
  130. {
  131. if(mssfix==0)
  132. {
  133. return 0;
  134. }
  135. if(len<int(sizeof(iphdr)))
  136. {
  137. mylog(log_debug,"packet from tun len=%d <20\n",len);
  138. return -1;
  139. }
  140. iphdr * iph;
  141. iph = (struct iphdr *) s;
  142. if(iph->version!=4)
  143. {
  144. //mylog(log_trace,"not ipv4");
  145. return 0;
  146. }
  147. if(iph->protocol!=IPPROTO_TCP)
  148. {
  149. //mylog(log_trace,"not tcp");
  150. return 0;
  151. }
  152. int ip_len=ntohs(iph->tot_len);
  153. int ip_hdr_len=iph->ihl*4;
  154. if(len<ip_hdr_len)
  155. {
  156. mylog(log_debug,"len<ip_hdr_len,%d %d\n",len,ip_hdr_len);
  157. return -1;
  158. }
  159. if(len<ip_len)
  160. {
  161. mylog(log_debug,"len<ip_len,%d %d\n",len,ip_len);
  162. return -1;
  163. }
  164. if(ip_hdr_len>ip_len)
  165. {
  166. mylog(log_debug,"ip_hdr_len<ip_len,%d %d\n",ip_hdr_len,ip_len);
  167. return -1;
  168. }
  169. if( ( ntohs(iph->frag_off) &(short)(0x1FFF) ) !=0 )
  170. {
  171. //not first segment
  172. //printf("line=%d %x %x \n",__LINE__,(u32_t)ntohs(iph->frag_off),u32_t( ntohs(iph->frag_off) &0xFFF8));
  173. return 0;
  174. }
  175. if( ( ntohs(iph->frag_off) &(short)(0x80FF) ) !=0 )
  176. {
  177. //not whole segment
  178. //printf("line=%d \n",__LINE__);
  179. return 0;
  180. }
  181. char * tcp_begin=s+ip_hdr_len;
  182. int tcp_len=ip_len-ip_hdr_len;
  183. if(tcp_len<20)
  184. {
  185. mylog(log_debug,"tcp_len<20,%d\n",tcp_len);
  186. return -1;
  187. }
  188. tcphdr * tcph=(struct tcphdr*)tcp_begin;
  189. if(int(tcph->syn)==0) //fast fail
  190. {
  191. mylog(log_trace,"tcph->syn==0\n");
  192. return 0;
  193. }
  194. int tcp_hdr_len = tcph->doff*4;
  195. if(tcp_len<tcp_hdr_len)
  196. {
  197. mylog(log_debug,"tcp_len <tcp_hdr_len, %d %d\n",tcp_len,tcp_hdr_len);
  198. return -1;
  199. }
  200. /*
  201. if(tcp_hdr_len==20)
  202. {
  203. //printf("line=%d\n",__LINE__);
  204. mylog(log_trace,"no tcp option\n");
  205. return 0;
  206. }*/
  207. char *ptr=tcp_begin+20;
  208. char *option_end=tcp_begin+tcp_hdr_len;
  209. while(ptr<option_end)
  210. {
  211. if(*ptr==0)
  212. {
  213. return 0;
  214. }
  215. else if(*ptr==1)
  216. {
  217. ptr++;
  218. }
  219. else if(*ptr==2)
  220. {
  221. if(ptr+1>=option_end)
  222. {
  223. mylog(log_debug,"invaild option ptr+1==option_end,for mss\n");
  224. return -1;
  225. }
  226. if(*(ptr+1)!=4)
  227. {
  228. mylog(log_debug,"invaild mss len\n");
  229. return -1;
  230. }
  231. if(ptr+3>=option_end)
  232. {
  233. mylog(log_debug,"ptr+4>option_end for mss\n");
  234. return -1;
  235. }
  236. int mss= read_u16(ptr+2);//uint8_t(ptr[2])*256+uint8_t(ptr[3]);
  237. int new_mss=mss;
  238. if(new_mss>::mssfix-40-10) //minus extra 10 for safe
  239. {
  240. new_mss=::mssfix-40-10;
  241. }
  242. write_u16(ptr+2,(unsigned short)new_mss);
  243. pseudo_header psh;
  244. psh.source_address =iph->saddr;
  245. psh.dest_address = iph->daddr;
  246. psh.placeholder = 0;
  247. psh.protocol = iph->protocol;
  248. psh.tcp_length = htons(tcp_len);
  249. tcph->check=0;
  250. tcph->check=tcp_csum(psh,(unsigned short *)tcph,tcp_len);
  251. mylog(log_trace,"mss=%d syn=%d ack=%d, changed mss to %d \n",mss,(int)tcph->syn,(int)tcph->ack,new_mss);
  252. //printf("test=%x\n",u32_t(1));
  253. //printf("frag=%x\n",u32_t( ntohs(iph->frag_off) ));
  254. return 0;
  255. }
  256. else
  257. {
  258. if(ptr+1>=option_end)
  259. {
  260. mylog(log_debug,"invaild option ptr+1==option_end\n");
  261. return -1;
  262. }
  263. else
  264. {
  265. int len=(unsigned char)*(ptr+1);
  266. if(len<=1)
  267. {
  268. mylog(log_debug,"invaild option len %d\n",len);
  269. return -1;
  270. }
  271. ptr+=len;
  272. }
  273. }
  274. }
  275. return 0;
  276. }
  277. int do_keep_alive(dest_t & dest)
  278. {
  279. if(get_current_time()-last_keep_alive_time>u64_t(keep_alive_interval))
  280. {
  281. last_keep_alive_time=get_current_time();
  282. char data[buf_len];int len;
  283. data[0]=header_keep_alive;
  284. len=1;
  285. assert(dest.cook==1);
  286. //do_cook(data,len);
  287. delay_send(0,dest,data,len);
  288. }
  289. return 0;
  290. }
  291. int tun_dev_client_event_loop()
  292. {
  293. char data[buf_len];
  294. int len;
  295. int i,j,k,ret;
  296. int epoll_fd,tun_fd;
  297. int remote_fd;
  298. fd64_t remote_fd64;
  299. tun_fd=get_tun_fd(tun_dev);
  300. assert(tun_fd>0);
  301. assert(new_connected_socket2(remote_fd,remote_addr)==0);
  302. remote_fd64=fd_manager.create(remote_fd);
  303. assert(set_if(tun_dev,htonl((ntohl(sub_net_uint32)&0xFFFFFF00)|2),htonl((ntohl(sub_net_uint32)&0xFFFFFF00 )|1),tun_mtu)==0);
  304. epoll_fd = epoll_create1(0);
  305. assert(epoll_fd>0);
  306. const int max_events = 4096;
  307. struct epoll_event ev, events[max_events];
  308. if (epoll_fd < 0) {
  309. mylog(log_fatal,"epoll return %d\n", epoll_fd);
  310. myexit(-1);
  311. }
  312. ev.events = EPOLLIN;
  313. ev.data.u64 = remote_fd64;
  314. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, remote_fd, &ev);
  315. if (ret!=0) {
  316. mylog(log_fatal,"add remote_fd64 error\n");
  317. myexit(-1);
  318. }
  319. ev.events = EPOLLIN;
  320. ev.data.u64 = tun_fd;
  321. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, tun_fd, &ev);
  322. if (ret!=0) {
  323. mylog(log_fatal,"add tun_fd error\n");
  324. myexit(-1);
  325. }
  326. ev.events = EPOLLIN;
  327. ev.data.u64 = delay_manager.get_timer_fd();
  328. mylog(log_debug,"delay_manager.get_timer_fd()=%d\n",delay_manager.get_timer_fd());
  329. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
  330. if (ret!= 0) {
  331. mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
  332. myexit(-1);
  333. }
  334. conn_info_t *conn_info_p=new conn_info_t;
  335. conn_info_t &conn_info=*conn_info_p; //huge size of conn_info,do not allocate on stack
  336. u64_t tmp_timer_fd64=conn_info.fec_encode_manager.get_timer_fd64();
  337. ev.events = EPOLLIN;
  338. ev.data.u64 = tmp_timer_fd64;
  339. mylog(log_debug,"conn_info.fec_encode_manager.get_timer_fd64()=%llu\n",conn_info.fec_encode_manager.get_timer_fd64());
  340. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(tmp_timer_fd64), &ev);
  341. if (ret!= 0) {
  342. mylog(log_fatal,"add fec_encode_manager.get_timer_fd64() error\n");
  343. myexit(-1);
  344. }
  345. conn_info.timer.add_fd_to_epoll(epoll_fd);
  346. conn_info.timer.set_timer_repeat_us(timer_interval*1000);
  347. int fifo_fd=-1;
  348. if(fifo_file[0]!=0)
  349. {
  350. fifo_fd=create_fifo(fifo_file);
  351. ev.events = EPOLLIN;
  352. ev.data.u64 = fifo_fd;
  353. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev);
  354. if (ret!= 0) {
  355. mylog(log_fatal,"add fifo_fd to epoll error %s\n",strerror(errno));
  356. myexit(-1);
  357. }
  358. mylog(log_info,"fifo_file=%s\n",fifo_file);
  359. }
  360. dest_t udp_dest;
  361. udp_dest.cook=1;
  362. udp_dest.type=type_fd64;
  363. udp_dest.inner.fd64=remote_fd64;
  364. dest_t tun_dest;
  365. tun_dest.type=type_write_fd;
  366. tun_dest.inner.fd=tun_fd;
  367. int got_feed_back=0;
  368. while(1)////////////////////////
  369. {
  370. if(about_to_exit) myexit(0);
  371. int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
  372. if (nfds < 0) { //allow zero
  373. if(errno==EINTR )
  374. {
  375. mylog(log_info,"epoll interrupted by signal,continue\n");
  376. //myexit(0);
  377. }
  378. else
  379. {
  380. mylog(log_fatal,"epoll_wait return %d,%s\n", nfds,strerror(errno));
  381. myexit(-1);
  382. }
  383. }
  384. int idx;
  385. for (idx = 0; idx < nfds; ++idx)
  386. {
  387. if(events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd())
  388. {
  389. uint64_t value;
  390. read(conn_info.timer.get_timer_fd(), &value, 8);
  391. mylog(log_trace,"events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd()\n");
  392. conn_info.stat.report_as_client();
  393. if(got_feed_back) do_keep_alive(udp_dest);
  394. }
  395. else if(events[idx].data.u64==conn_info.fec_encode_manager.get_timer_fd64())
  396. {
  397. fd64_t fd64=events[idx].data.u64;
  398. mylog(log_trace,"events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64()\n");
  399. uint64_t value;
  400. if(!fd_manager.exist(fd64)) //fd64 has been closed
  401. {
  402. mylog(log_trace,"!fd_manager.exist(fd64)");
  403. continue;
  404. }
  405. if((ret=read(fd_manager.to_fd(fd64), &value, 8))!=8)
  406. {
  407. mylog(log_trace,"(ret=read(fd_manager.to_fd(fd64), &value, 8))!=8,ret=%d\n",ret);
  408. continue;
  409. }
  410. if(value==0)
  411. {
  412. mylog(log_debug,"value==0\n");
  413. continue;
  414. }
  415. assert(value==1);
  416. char header=(got_feed_back==0?header_new_connect:header_normal);
  417. from_normal_to_fec2(conn_info,udp_dest,0,0,header);
  418. }
  419. else if(events[idx].data.u64==(u64_t)tun_fd)
  420. {
  421. len=read(tun_fd,data,max_data_len+1);
  422. if(len==max_data_len+1)
  423. {
  424. mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
  425. continue;
  426. }
  427. if(len<0)
  428. {
  429. mylog(log_warn,"read from tun_fd return %d,errno=%s\n",len,strerror(errno));
  430. continue;
  431. }
  432. do_mssfix(data,len);
  433. mylog(log_trace,"Received packet from tun,len: %d\n",len);
  434. char header=(got_feed_back==0?header_new_connect:header_normal);
  435. from_normal_to_fec2(conn_info,udp_dest,data,len,header);
  436. }
  437. else if(events[idx].data.u64==(u64_t)remote_fd64)
  438. {
  439. fd64_t fd64=events[idx].data.u64;
  440. int fd=fd_manager.to_fd(fd64);
  441. len=recv(fd,data,max_data_len+1,0);
  442. if(len==max_data_len+1)
  443. {
  444. mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
  445. continue;
  446. }
  447. if(len<0)
  448. {
  449. mylog(log_warn,"recv return %d,errno=%s\n",len,strerror(errno));
  450. continue;
  451. }
  452. if(de_cook(data,len)<0)
  453. {
  454. mylog(log_warn,"de_cook(data,len)failed \n");
  455. continue;
  456. }
  457. char header=0;
  458. if(get_header(header,data,len)!=0)
  459. {
  460. mylog(log_warn,"get_header failed\n");
  461. continue;
  462. }
  463. if(header==header_keep_alive)
  464. {
  465. mylog(log_trace,"got keep_alive packet\n");
  466. continue;
  467. }
  468. if(header==header_reject)
  469. {
  470. if(keep_reconnect==0)
  471. {
  472. mylog(log_fatal,"server restarted or switched to handle another client,exited\n");
  473. myexit(-1);
  474. }
  475. else
  476. {
  477. if(got_feed_back==1)
  478. mylog(log_warn,"server restarted or switched to handle another client,but keep-reconnect enabled,trying to reconnect\n");
  479. got_feed_back=0;
  480. }
  481. continue;
  482. }
  483. else if(header==header_normal)
  484. {
  485. if(got_feed_back==0)
  486. mylog(log_info,"connection accepted by server\n");
  487. got_feed_back=1;
  488. }
  489. else
  490. {
  491. mylog(log_warn,"invalid header %d %d\n",int(header),len);
  492. continue;
  493. }
  494. mylog(log_trace,"Received packet from udp,len: %d\n",len);
  495. from_fec_to_normal2(conn_info,tun_dest,data,len);
  496. }
  497. else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd())
  498. {
  499. uint64_t value;
  500. read(delay_manager.get_timer_fd(), &value, 8);
  501. mylog(log_trace,"events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()\n");
  502. }
  503. else if (events[idx].data.u64 == (u64_t)fifo_fd)
  504. {
  505. char buf[buf_len];
  506. int len=read (fifo_fd, buf, sizeof (buf));
  507. if(len<0)
  508. {
  509. mylog(log_warn,"fifo read failed len=%d,errno=%s\n",len,strerror(errno));
  510. continue;
  511. }
  512. buf[len]=0;
  513. handle_command(buf);
  514. }
  515. else
  516. {
  517. assert(0==1);
  518. }
  519. }
  520. delay_manager.check();
  521. }
  522. return 0;
  523. }
  524. int tun_dev_server_event_loop()
  525. {
  526. char data[buf_len];
  527. int len;
  528. int i,j,k,ret;
  529. int epoll_fd,tun_fd;
  530. int local_listen_fd;
  531. tun_fd=get_tun_fd(tun_dev);
  532. assert(tun_fd>0);
  533. assert(new_listen_socket2(local_listen_fd,local_addr)==0);
  534. assert(set_if(tun_dev,htonl((ntohl(sub_net_uint32)&0xFFFFFF00)|1),htonl((ntohl(sub_net_uint32)&0xFFFFFF00 )|2),tun_mtu)==0);
  535. epoll_fd = epoll_create1(0);
  536. assert(epoll_fd>0);
  537. const int max_events = 4096;
  538. struct epoll_event ev, events[max_events];
  539. if (epoll_fd < 0) {
  540. mylog(log_fatal,"epoll return %d\n", epoll_fd);
  541. myexit(-1);
  542. }
  543. ev.events = EPOLLIN;
  544. ev.data.u64 = local_listen_fd;
  545. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, local_listen_fd, &ev);
  546. if (ret!=0) {
  547. mylog(log_fatal,"add udp_listen_fd error\n");
  548. myexit(-1);
  549. }
  550. ev.events = EPOLLIN;
  551. ev.data.u64 = tun_fd;
  552. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, tun_fd, &ev);
  553. if (ret!=0) {
  554. mylog(log_fatal,"add tun_fd error\n");
  555. myexit(-1);
  556. }
  557. ev.events = EPOLLIN;
  558. ev.data.u64 = delay_manager.get_timer_fd();
  559. mylog(log_debug,"delay_manager.get_timer_fd()=%d\n",delay_manager.get_timer_fd());
  560. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
  561. if (ret!= 0) {
  562. mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
  563. myexit(-1);
  564. }
  565. conn_info_t *conn_info_p=new conn_info_t;
  566. conn_info_t &conn_info=*conn_info_p; //huge size of conn_info,do not allocate on stack
  567. u64_t tmp_timer_fd64=conn_info.fec_encode_manager.get_timer_fd64();
  568. ev.events = EPOLLIN;
  569. ev.data.u64 = tmp_timer_fd64;
  570. mylog(log_debug,"conn_info.fec_encode_manager.get_timer_fd64()=%llu\n",conn_info.fec_encode_manager.get_timer_fd64());
  571. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(tmp_timer_fd64), &ev);
  572. if (ret!= 0) {
  573. mylog(log_fatal,"add fec_encode_manager.get_timer_fd64() error\n");
  574. myexit(-1);
  575. }
  576. conn_info.timer.add_fd_to_epoll(epoll_fd);
  577. conn_info.timer.set_timer_repeat_us(timer_interval*1000);
  578. int fifo_fd=-1;
  579. if(fifo_file[0]!=0)
  580. {
  581. fifo_fd=create_fifo(fifo_file);
  582. ev.events = EPOLLIN;
  583. ev.data.u64 = fifo_fd;
  584. ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev);
  585. if (ret!= 0) {
  586. mylog(log_fatal,"add fifo_fd to epoll error %s\n",strerror(errno));
  587. myexit(-1);
  588. }
  589. mylog(log_info,"fifo_file=%s\n",fifo_file);
  590. }
  591. dest_t udp_dest;
  592. udp_dest.cook=1;
  593. udp_dest.type=type_fd_addr;
  594. udp_dest.inner.fd_addr.fd=local_listen_fd;
  595. udp_dest.inner.fd_addr.addr.clear();
  596. dest_t tun_dest;
  597. tun_dest.type=type_write_fd;
  598. tun_dest.inner.fd=tun_fd;
  599. while(1)////////////////////////
  600. {
  601. if(about_to_exit) myexit(0);
  602. int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
  603. if (nfds < 0) { //allow zero
  604. if(errno==EINTR )
  605. {
  606. mylog(log_info,"epoll interrupted by signal,continue\n");
  607. //myexit(0);
  608. }
  609. else
  610. {
  611. mylog(log_fatal,"epoll_wait return %d,%s\n", nfds,strerror(errno));
  612. myexit(-1);
  613. }
  614. }
  615. int idx;
  616. for (idx = 0; idx < nfds; ++idx)
  617. {
  618. if(events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd())
  619. {
  620. uint64_t value;
  621. read(conn_info.timer.get_timer_fd(), &value, 8);
  622. if(udp_dest.inner.fd_addr.addr.is_vaild()==0)
  623. {
  624. continue;
  625. }
  626. conn_info.stat.report_as_server(udp_dest.inner.fd_addr.addr);
  627. do_keep_alive(udp_dest);
  628. }
  629. else if(events[idx].data.u64==conn_info.fec_encode_manager.get_timer_fd64())
  630. {
  631. assert(udp_dest.inner.fd_addr.addr.is_vaild()!=0);
  632. mylog(log_trace,"events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64()\n");
  633. uint64_t fd64=events[idx].data.u64;
  634. //mylog(log_info,"timer!!!\n");
  635. uint64_t value;
  636. if(!fd_manager.exist(fd64)) //fd64 has been closed
  637. {
  638. mylog(log_trace,"!fd_manager.exist(fd64)");
  639. continue;
  640. }
  641. if((ret=read(fd_manager.to_fd(fd64), &value, 8))!=8)
  642. {
  643. mylog(log_trace,"(ret=read(fd_manager.to_fd(fd64), &value, 8))!=8,ret=%d\n",ret);
  644. continue;
  645. }
  646. if(value==0)
  647. {
  648. mylog(log_debug,"value==0\n");
  649. continue;
  650. }
  651. assert(value==1);
  652. from_normal_to_fec2(conn_info,udp_dest,0,0,header_normal);
  653. }
  654. else if(events[idx].data.u64==(u64_t)local_listen_fd)
  655. {
  656. //struct sockaddr_in udp_new_addr_in={0};
  657. //socklen_t udp_new_addr_len = sizeof(sockaddr_in);
  658. address_t::storage_t udp_new_addr_in={0};
  659. socklen_t udp_new_addr_len = sizeof(address_t::storage_t);
  660. if ((len = recvfrom(local_listen_fd, data, max_data_len+1, 0,
  661. (struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) < 0) {
  662. mylog(log_error,"recv_from error,this shouldnt happen,err=%s,but we can try to continue\n",strerror(errno));
  663. continue;
  664. //myexit(1);
  665. };
  666. address_t new_addr;
  667. new_addr.from_sockaddr((struct sockaddr *) &udp_new_addr_in,udp_new_addr_len);
  668. if(len==max_data_len+1)
  669. {
  670. mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
  671. continue;
  672. }
  673. if(de_cook(data,len)<0)
  674. {
  675. mylog(log_warn,"de_cook(data,len)failed \n");
  676. continue;
  677. }
  678. char header=0;
  679. if(get_header(header,data,len)!=0)
  680. {
  681. mylog(log_warn,"get_header failed\n");
  682. continue;
  683. }
  684. if(udp_dest.inner.fd_addr.addr==new_addr)
  685. {
  686. if(header==header_keep_alive)
  687. {
  688. mylog(log_trace,"got keep_alive packet\n");
  689. continue;
  690. }
  691. if(header!=header_new_connect&& header!=header_normal)
  692. {
  693. mylog(log_warn,"invalid header\n");
  694. continue;
  695. }
  696. }
  697. else
  698. {
  699. if(header==header_keep_alive)
  700. {
  701. mylog(log_debug,"got keep_alive packet from unexpected client\n");
  702. continue;
  703. }
  704. if(header==header_new_connect)
  705. {
  706. mylog(log_info,"new connection from %s\n", new_addr.get_str());
  707. udp_dest.inner.fd_addr.addr=new_addr;
  708. //udp_dest.inner.fd_ip_port.ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
  709. //udp_dest.inner.fd_ip_port.ip_port.port=ntohs(udp_new_addr_in.sin_port);
  710. conn_info.fec_decode_manager.clear();
  711. conn_info.fec_encode_manager.clear_data();
  712. memset(&conn_info.stat,0,sizeof(conn_info.stat));
  713. }
  714. else if(header==header_normal)
  715. {
  716. mylog(log_debug,"rejected connection from %s\n", new_addr.get_str());
  717. len=1;
  718. data[0]=header_reject;
  719. do_cook(data,len);
  720. dest_t tmp_dest;
  721. tmp_dest.type=type_fd_addr;
  722. tmp_dest.inner.fd_addr.fd=local_listen_fd;
  723. tmp_dest.inner.fd_addr.addr=new_addr;
  724. delay_manager.add(0,tmp_dest,data,len);;
  725. continue;
  726. }
  727. else
  728. {
  729. mylog(log_warn,"invalid header\n");
  730. }
  731. }
  732. mylog(log_trace,"Received packet from %s,len: %d\n", new_addr.get_str(),len);
  733. from_fec_to_normal2(conn_info,tun_dest,data,len);
  734. }
  735. else if(events[idx].data.u64==(u64_t)tun_fd)
  736. {
  737. len=read(tun_fd,data,max_data_len+1);
  738. if(len==max_data_len+1)
  739. {
  740. mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
  741. continue;
  742. }
  743. if(len<0)
  744. {
  745. mylog(log_warn,"read from tun_fd return %d,errno=%s\n",len,strerror(errno));
  746. continue;
  747. }
  748. do_mssfix(data,len);
  749. mylog(log_trace,"Received packet from tun,len: %d\n",len);
  750. if(udp_dest.inner.fd_addr.addr.is_vaild()==0)
  751. {
  752. mylog(log_debug,"received packet from tun,but there is no client yet,dropped packet\n");
  753. continue;
  754. }
  755. from_normal_to_fec2(conn_info,udp_dest,data,len,header_normal);
  756. }
  757. else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd())
  758. {
  759. uint64_t value;
  760. read(delay_manager.get_timer_fd(), &value, 8);
  761. mylog(log_trace,"events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()\n");
  762. }
  763. else if (events[idx].data.u64 == (u64_t)fifo_fd)
  764. {
  765. char buf[buf_len];
  766. int len=read (fifo_fd, buf, sizeof (buf));
  767. if(len<0)
  768. {
  769. mylog(log_warn,"fifo read failed len=%d,errno=%s\n",len,strerror(errno));
  770. continue;
  771. }
  772. buf[len]=0;
  773. handle_command(buf);
  774. }
  775. else
  776. {
  777. assert(0==1);
  778. }
  779. }
  780. delay_manager.check();
  781. }
  782. return 0;
  783. }