123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950 |
- /*
- * tun.cpp
- *
- * Created on: Oct 26, 2017
- * Author: root
- */
- #include "common.h"
- #include "log.h"
- #include "misc.h"
- #include <netinet/tcp.h> //Provides declarations for tcp header
- #include <netinet/udp.h>
- #include <netinet/ip.h> //Provides declarations for ip header
- #include <netinet/if_ether.h>
- #include <linux/if.h>
- #include <linux/if_tun.h>
- my_time_t last_keep_alive_time=0;
- int keep_alive_interval=3000;//3000ms
- int get_tun_fd(char * dev_name)
- {
- int tun_fd=open("/dev/net/tun",O_RDWR);
- if(tun_fd <0)
- {
- mylog(log_fatal,"open /dev/net/tun failed");
- myexit(-1);
- }
- struct ifreq ifr;
- memset(&ifr, 0, sizeof(ifr));
- ifr.ifr_flags = IFF_TUN|IFF_NO_PI;
- strncpy(ifr.ifr_name, dev_name, IFNAMSIZ);
- if(ioctl(tun_fd, TUNSETIFF, (void *)&ifr) != 0)
- {
- mylog(log_fatal,"open /dev/net/tun failed");
- myexit(-1);
- }
- return tun_fd;
- }
- int set_if(char *if_name,u32_t local_ip,u32_t remote_ip,int mtu)
- {
- //printf("i m here1\n");
- struct ifreq ifr;
- struct sockaddr_in sai;
- memset(&ifr,0,sizeof(ifr));
- memset(&sai, 0, sizeof(struct sockaddr));
- int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
- strncpy(ifr.ifr_name, if_name, IFNAMSIZ);
- sai.sin_family = AF_INET;
- sai.sin_port = 0;
- sai.sin_addr.s_addr = local_ip;
- memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
- assert(ioctl(sockfd, SIOCSIFADDR, &ifr)==0); //set source ip
- sai.sin_addr.s_addr = remote_ip;
- memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
- assert(ioctl(sockfd, SIOCSIFDSTADDR, &ifr)==0);//set dest ip
- ifr.ifr_mtu=mtu;
- assert(ioctl(sockfd, SIOCSIFMTU, &ifr)==0);//set mtu
- assert(ioctl(sockfd, SIOCGIFFLAGS, &ifr)==0);
- // ifr.ifr_flags |= ( IFF_UP|IFF_POINTOPOINT|IFF_RUNNING|IFF_NOARP|IFF_MULTICAST );
- ifr.ifr_flags = ( IFF_UP|IFF_POINTOPOINT|IFF_RUNNING|IFF_NOARP|IFF_MULTICAST );//set interface flags
- assert(ioctl(sockfd, SIOCSIFFLAGS, &ifr)==0);
- //printf("i m here2\n");
- return 0;
- }
- const char header_normal=1;
- const char header_new_connect=2;
- const char header_reject=3;
- const char header_keep_alive=4;
- int put_header(char header,char * data,int &len)
- {
- assert(len>=0);
- data[len]=header;
- len+=1;
- return 0;
- }
- int get_header(char &header,char * data,int &len)
- {
- assert(len>=0);
- if(len<1) return -1;
- len-=1;
- header=data[len];
- return 0;
- }
- int from_normal_to_fec2(conn_info_t & conn_info,dest_t &dest,char * data,int len,char header)
- {
- int out_n;char **out_arr;int *out_len;my_time_t *out_delay;
- from_normal_to_fec(conn_info,data,len,out_n,out_arr,out_len,out_delay);
- for(int i=0;i<out_n;i++)
- {
- char tmp_buf[buf_len];
- int tmp_len=out_len[i];
- memcpy(tmp_buf,out_arr[i],out_len[i]);
- put_header(header,tmp_buf,tmp_len);
- delay_send(out_delay[i],dest,tmp_buf,tmp_len);//this is slow but safer.just use this one
- //put_header(header,out_arr[i],out_len[i]);//modify in place
- //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
- }
- return 0;
- }
- int from_fec_to_normal2(conn_info_t & conn_info,dest_t &dest,char * data,int len)
- {
- int out_n;char **out_arr;int *out_len;my_time_t *out_delay;
- from_fec_to_normal(conn_info,data,len,out_n,out_arr,out_len,out_delay);
- for(int i=0;i<out_n;i++)
- {
- #ifndef NOLIMIT
- if(program_mode==server_mode)
- {
- char * tmp_data=out_arr[i];
- int tmp_len=out_len[i];
- iphdr * iph;
- iph = (struct iphdr *) tmp_data;
- if(tmp_len>=int(sizeof(iphdr))&&iph->version==4)
- {
- u32_t dest_ip=iph->daddr;
- //printf("%s\n",my_ntoa(dest_ip));
- if( ( ntohl(sub_net_uint32)&0xFFFFFF00 ) != ( ntohl (dest_ip) &0xFFFFFF00) )
- {
- string sub=my_ntoa(dest_ip);
- string dst=my_ntoa( htonl( ntohl (sub_net_uint32) &0xFFFFFF00) );
- 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());
- continue;
- }
- }
- }
- #endif
- delay_send(out_delay[i],dest,out_arr[i],out_len[i]);
- }
- return 0;
- }
- int do_mssfix(char * s,int len)
- {
- if(mssfix==0)
- {
- return 0;
- }
- if(len<int(sizeof(iphdr)))
- {
- mylog(log_debug,"packet from tun len=%d <20\n",len);
- return -1;
- }
- iphdr * iph;
- iph = (struct iphdr *) s;
- if(iph->version!=4)
- {
- //mylog(log_trace,"not ipv4");
- return 0;
- }
- if(iph->protocol!=IPPROTO_TCP)
- {
- //mylog(log_trace,"not tcp");
- return 0;
- }
- int ip_len=ntohs(iph->tot_len);
- int ip_hdr_len=iph->ihl*4;
- if(len<ip_hdr_len)
- {
- mylog(log_debug,"len<ip_hdr_len,%d %d\n",len,ip_hdr_len);
- return -1;
- }
- if(len<ip_len)
- {
- mylog(log_debug,"len<ip_len,%d %d\n",len,ip_len);
- return -1;
- }
- if(ip_hdr_len>ip_len)
- {
- mylog(log_debug,"ip_hdr_len<ip_len,%d %d\n",ip_hdr_len,ip_len);
- return -1;
- }
- if( ( ntohs(iph->frag_off) &(short)(0x1FFF) ) !=0 )
- {
- //not first segment
- //printf("line=%d %x %x \n",__LINE__,(u32_t)ntohs(iph->frag_off),u32_t( ntohs(iph->frag_off) &0xFFF8));
- return 0;
- }
- if( ( ntohs(iph->frag_off) &(short)(0x80FF) ) !=0 )
- {
- //not whole segment
- //printf("line=%d \n",__LINE__);
- return 0;
- }
- char * tcp_begin=s+ip_hdr_len;
- int tcp_len=ip_len-ip_hdr_len;
- if(tcp_len<20)
- {
- mylog(log_debug,"tcp_len<20,%d\n",tcp_len);
- return -1;
- }
- tcphdr * tcph=(struct tcphdr*)tcp_begin;
- if(int(tcph->syn)==0) //fast fail
- {
- mylog(log_trace,"tcph->syn==0\n");
- return 0;
- }
- int tcp_hdr_len = tcph->doff*4;
- if(tcp_len<tcp_hdr_len)
- {
- mylog(log_debug,"tcp_len <tcp_hdr_len, %d %d\n",tcp_len,tcp_hdr_len);
- return -1;
- }
- /*
- if(tcp_hdr_len==20)
- {
- //printf("line=%d\n",__LINE__);
- mylog(log_trace,"no tcp option\n");
- return 0;
- }*/
- char *ptr=tcp_begin+20;
- char *option_end=tcp_begin+tcp_hdr_len;
- while(ptr<option_end)
- {
- if(*ptr==0)
- {
- return 0;
- }
- else if(*ptr==1)
- {
- ptr++;
- }
- else if(*ptr==2)
- {
- if(ptr+1>=option_end)
- {
- mylog(log_debug,"invaild option ptr+1==option_end,for mss\n");
- return -1;
- }
- if(*(ptr+1)!=4)
- {
- mylog(log_debug,"invaild mss len\n");
- return -1;
- }
- if(ptr+3>=option_end)
- {
- mylog(log_debug,"ptr+4>option_end for mss\n");
- return -1;
- }
- int mss= read_u16(ptr+2);//uint8_t(ptr[2])*256+uint8_t(ptr[3]);
- int new_mss=mss;
- if(new_mss>::mssfix-40-10) //minus extra 10 for safe
- {
- new_mss=::mssfix-40-10;
- }
- write_u16(ptr+2,(unsigned short)new_mss);
- pseudo_header psh;
- psh.source_address =iph->saddr;
- psh.dest_address = iph->daddr;
- psh.placeholder = 0;
- psh.protocol = iph->protocol;
- psh.tcp_length = htons(tcp_len);
- tcph->check=0;
- tcph->check=tcp_csum(psh,(unsigned short *)tcph,tcp_len);
- mylog(log_trace,"mss=%d syn=%d ack=%d, changed mss to %d \n",mss,(int)tcph->syn,(int)tcph->ack,new_mss);
- //printf("test=%x\n",u32_t(1));
- //printf("frag=%x\n",u32_t( ntohs(iph->frag_off) ));
- return 0;
- }
- else
- {
- if(ptr+1>=option_end)
- {
- mylog(log_debug,"invaild option ptr+1==option_end\n");
- return -1;
- }
- else
- {
- int len=(unsigned char)*(ptr+1);
- if(len<=1)
- {
- mylog(log_debug,"invaild option len %d\n",len);
- return -1;
- }
- ptr+=len;
- }
- }
- }
- return 0;
- }
- int do_keep_alive(dest_t & dest)
- {
- if(get_current_time()-last_keep_alive_time>u64_t(keep_alive_interval))
- {
- last_keep_alive_time=get_current_time();
- char data[buf_len];int len;
- data[0]=header_keep_alive;
- len=1;
- assert(dest.cook==1);
- //do_cook(data,len);
- delay_send(0,dest,data,len);
- }
- return 0;
- }
- int tun_dev_client_event_loop()
- {
- char data[buf_len];
- int len;
- int i,j,k,ret;
- int epoll_fd,tun_fd;
- int remote_fd;
- fd64_t remote_fd64;
- tun_fd=get_tun_fd(tun_dev);
- assert(tun_fd>0);
- assert(new_connected_socket2(remote_fd,remote_addr)==0);
- remote_fd64=fd_manager.create(remote_fd);
- assert(set_if(tun_dev,htonl((ntohl(sub_net_uint32)&0xFFFFFF00)|2),htonl((ntohl(sub_net_uint32)&0xFFFFFF00 )|1),tun_mtu)==0);
- epoll_fd = epoll_create1(0);
- assert(epoll_fd>0);
- const int max_events = 4096;
- struct epoll_event ev, events[max_events];
- if (epoll_fd < 0) {
- mylog(log_fatal,"epoll return %d\n", epoll_fd);
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = remote_fd64;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, remote_fd, &ev);
- if (ret!=0) {
- mylog(log_fatal,"add remote_fd64 error\n");
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = tun_fd;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, tun_fd, &ev);
- if (ret!=0) {
- mylog(log_fatal,"add tun_fd error\n");
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = delay_manager.get_timer_fd();
- mylog(log_debug,"delay_manager.get_timer_fd()=%d\n",delay_manager.get_timer_fd());
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
- myexit(-1);
- }
- conn_info_t *conn_info_p=new conn_info_t;
- conn_info_t &conn_info=*conn_info_p; //huge size of conn_info,do not allocate on stack
- u64_t tmp_timer_fd64=conn_info.fec_encode_manager.get_timer_fd64();
- ev.events = EPOLLIN;
- ev.data.u64 = tmp_timer_fd64;
- mylog(log_debug,"conn_info.fec_encode_manager.get_timer_fd64()=%llu\n",conn_info.fec_encode_manager.get_timer_fd64());
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(tmp_timer_fd64), &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add fec_encode_manager.get_timer_fd64() error\n");
- myexit(-1);
- }
- conn_info.timer.add_fd_to_epoll(epoll_fd);
- conn_info.timer.set_timer_repeat_us(timer_interval*1000);
- int fifo_fd=-1;
- if(fifo_file[0]!=0)
- {
- fifo_fd=create_fifo(fifo_file);
- ev.events = EPOLLIN;
- ev.data.u64 = fifo_fd;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add fifo_fd to epoll error %s\n",strerror(errno));
- myexit(-1);
- }
- mylog(log_info,"fifo_file=%s\n",fifo_file);
- }
- dest_t udp_dest;
- udp_dest.cook=1;
- udp_dest.type=type_fd64;
- udp_dest.inner.fd64=remote_fd64;
- dest_t tun_dest;
- tun_dest.type=type_write_fd;
- tun_dest.inner.fd=tun_fd;
- int got_feed_back=0;
- while(1)////////////////////////
- {
- if(about_to_exit) myexit(0);
- int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
- if (nfds < 0) { //allow zero
- if(errno==EINTR )
- {
- mylog(log_info,"epoll interrupted by signal,continue\n");
- //myexit(0);
- }
- else
- {
- mylog(log_fatal,"epoll_wait return %d,%s\n", nfds,strerror(errno));
- myexit(-1);
- }
- }
- int idx;
- for (idx = 0; idx < nfds; ++idx)
- {
- if(events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd())
- {
- uint64_t value;
- read(conn_info.timer.get_timer_fd(), &value, 8);
- mylog(log_trace,"events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd()\n");
- conn_info.stat.report_as_client();
- if(got_feed_back) do_keep_alive(udp_dest);
- }
- else if(events[idx].data.u64==conn_info.fec_encode_manager.get_timer_fd64())
- {
- fd64_t fd64=events[idx].data.u64;
- mylog(log_trace,"events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64()\n");
- uint64_t value;
- if(!fd_manager.exist(fd64)) //fd64 has been closed
- {
- mylog(log_trace,"!fd_manager.exist(fd64)");
- continue;
- }
- if((ret=read(fd_manager.to_fd(fd64), &value, 8))!=8)
- {
- mylog(log_trace,"(ret=read(fd_manager.to_fd(fd64), &value, 8))!=8,ret=%d\n",ret);
- continue;
- }
- if(value==0)
- {
- mylog(log_debug,"value==0\n");
- continue;
- }
- assert(value==1);
- char header=(got_feed_back==0?header_new_connect:header_normal);
- from_normal_to_fec2(conn_info,udp_dest,0,0,header);
- }
- else if(events[idx].data.u64==(u64_t)tun_fd)
- {
- len=read(tun_fd,data,max_data_len+1);
- if(len==max_data_len+1)
- {
- mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
- continue;
- }
- if(len<0)
- {
- mylog(log_warn,"read from tun_fd return %d,errno=%s\n",len,strerror(errno));
- continue;
- }
- do_mssfix(data,len);
- mylog(log_trace,"Received packet from tun,len: %d\n",len);
- char header=(got_feed_back==0?header_new_connect:header_normal);
- from_normal_to_fec2(conn_info,udp_dest,data,len,header);
- }
- else if(events[idx].data.u64==(u64_t)remote_fd64)
- {
- fd64_t fd64=events[idx].data.u64;
- int fd=fd_manager.to_fd(fd64);
- len=recv(fd,data,max_data_len+1,0);
- if(len==max_data_len+1)
- {
- mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
- continue;
- }
- if(len<0)
- {
- mylog(log_warn,"recv return %d,errno=%s\n",len,strerror(errno));
- continue;
- }
- if(de_cook(data,len)<0)
- {
- mylog(log_warn,"de_cook(data,len)failed \n");
- continue;
- }
- char header=0;
- if(get_header(header,data,len)!=0)
- {
- mylog(log_warn,"get_header failed\n");
- continue;
- }
- if(header==header_keep_alive)
- {
- mylog(log_trace,"got keep_alive packet\n");
- continue;
- }
- if(header==header_reject)
- {
- if(keep_reconnect==0)
- {
- mylog(log_fatal,"server restarted or switched to handle another client,exited\n");
- myexit(-1);
- }
- else
- {
- if(got_feed_back==1)
- mylog(log_warn,"server restarted or switched to handle another client,but keep-reconnect enabled,trying to reconnect\n");
- got_feed_back=0;
- }
- continue;
- }
- else if(header==header_normal)
- {
- if(got_feed_back==0)
- mylog(log_info,"connection accepted by server\n");
- got_feed_back=1;
- }
- else
- {
- mylog(log_warn,"invalid header %d %d\n",int(header),len);
- continue;
- }
- mylog(log_trace,"Received packet from udp,len: %d\n",len);
- from_fec_to_normal2(conn_info,tun_dest,data,len);
- }
- else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd())
- {
- uint64_t value;
- read(delay_manager.get_timer_fd(), &value, 8);
- mylog(log_trace,"events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()\n");
- }
- else if (events[idx].data.u64 == (u64_t)fifo_fd)
- {
- char buf[buf_len];
- int len=read (fifo_fd, buf, sizeof (buf));
- if(len<0)
- {
- mylog(log_warn,"fifo read failed len=%d,errno=%s\n",len,strerror(errno));
- continue;
- }
- buf[len]=0;
- handle_command(buf);
- }
- else
- {
- assert(0==1);
- }
- }
- delay_manager.check();
- }
- return 0;
- }
- int tun_dev_server_event_loop()
- {
- char data[buf_len];
- int len;
- int i,j,k,ret;
- int epoll_fd,tun_fd;
- int local_listen_fd;
- tun_fd=get_tun_fd(tun_dev);
- assert(tun_fd>0);
- assert(new_listen_socket2(local_listen_fd,local_addr)==0);
- assert(set_if(tun_dev,htonl((ntohl(sub_net_uint32)&0xFFFFFF00)|1),htonl((ntohl(sub_net_uint32)&0xFFFFFF00 )|2),tun_mtu)==0);
- epoll_fd = epoll_create1(0);
- assert(epoll_fd>0);
- const int max_events = 4096;
- struct epoll_event ev, events[max_events];
- if (epoll_fd < 0) {
- mylog(log_fatal,"epoll return %d\n", epoll_fd);
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = local_listen_fd;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, local_listen_fd, &ev);
- if (ret!=0) {
- mylog(log_fatal,"add udp_listen_fd error\n");
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = tun_fd;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, tun_fd, &ev);
- if (ret!=0) {
- mylog(log_fatal,"add tun_fd error\n");
- myexit(-1);
- }
- ev.events = EPOLLIN;
- ev.data.u64 = delay_manager.get_timer_fd();
- mylog(log_debug,"delay_manager.get_timer_fd()=%d\n",delay_manager.get_timer_fd());
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, delay_manager.get_timer_fd(), &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add delay_manager.get_timer_fd() error\n");
- myexit(-1);
- }
- conn_info_t *conn_info_p=new conn_info_t;
- conn_info_t &conn_info=*conn_info_p; //huge size of conn_info,do not allocate on stack
- u64_t tmp_timer_fd64=conn_info.fec_encode_manager.get_timer_fd64();
- ev.events = EPOLLIN;
- ev.data.u64 = tmp_timer_fd64;
- mylog(log_debug,"conn_info.fec_encode_manager.get_timer_fd64()=%llu\n",conn_info.fec_encode_manager.get_timer_fd64());
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd_manager.to_fd(tmp_timer_fd64), &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add fec_encode_manager.get_timer_fd64() error\n");
- myexit(-1);
- }
- conn_info.timer.add_fd_to_epoll(epoll_fd);
- conn_info.timer.set_timer_repeat_us(timer_interval*1000);
- int fifo_fd=-1;
- if(fifo_file[0]!=0)
- {
- fifo_fd=create_fifo(fifo_file);
- ev.events = EPOLLIN;
- ev.data.u64 = fifo_fd;
- ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev);
- if (ret!= 0) {
- mylog(log_fatal,"add fifo_fd to epoll error %s\n",strerror(errno));
- myexit(-1);
- }
- mylog(log_info,"fifo_file=%s\n",fifo_file);
- }
- dest_t udp_dest;
- udp_dest.cook=1;
- udp_dest.type=type_fd_addr;
- udp_dest.inner.fd_addr.fd=local_listen_fd;
- udp_dest.inner.fd_addr.addr.clear();
- dest_t tun_dest;
- tun_dest.type=type_write_fd;
- tun_dest.inner.fd=tun_fd;
- while(1)////////////////////////
- {
- if(about_to_exit) myexit(0);
- int nfds = epoll_wait(epoll_fd, events, max_events, 180 * 1000);
- if (nfds < 0) { //allow zero
- if(errno==EINTR )
- {
- mylog(log_info,"epoll interrupted by signal,continue\n");
- //myexit(0);
- }
- else
- {
- mylog(log_fatal,"epoll_wait return %d,%s\n", nfds,strerror(errno));
- myexit(-1);
- }
- }
- int idx;
- for (idx = 0; idx < nfds; ++idx)
- {
- if(events[idx].data.u64==(u64_t)conn_info.timer.get_timer_fd())
- {
- uint64_t value;
- read(conn_info.timer.get_timer_fd(), &value, 8);
- if(udp_dest.inner.fd_addr.addr.is_vaild()==0)
- {
- continue;
- }
- conn_info.stat.report_as_server(udp_dest.inner.fd_addr.addr);
- do_keep_alive(udp_dest);
- }
- else if(events[idx].data.u64==conn_info.fec_encode_manager.get_timer_fd64())
- {
- assert(udp_dest.inner.fd_addr.addr.is_vaild()!=0);
- mylog(log_trace,"events[idx].data.u64 == conn_info.fec_encode_manager.get_timer_fd64()\n");
- uint64_t fd64=events[idx].data.u64;
- //mylog(log_info,"timer!!!\n");
- uint64_t value;
- if(!fd_manager.exist(fd64)) //fd64 has been closed
- {
- mylog(log_trace,"!fd_manager.exist(fd64)");
- continue;
- }
- if((ret=read(fd_manager.to_fd(fd64), &value, 8))!=8)
- {
- mylog(log_trace,"(ret=read(fd_manager.to_fd(fd64), &value, 8))!=8,ret=%d\n",ret);
- continue;
- }
- if(value==0)
- {
- mylog(log_debug,"value==0\n");
- continue;
- }
- assert(value==1);
- from_normal_to_fec2(conn_info,udp_dest,0,0,header_normal);
- }
- else if(events[idx].data.u64==(u64_t)local_listen_fd)
- {
- //struct sockaddr_in udp_new_addr_in={0};
- //socklen_t udp_new_addr_len = sizeof(sockaddr_in);
- address_t::storage_t udp_new_addr_in={0};
- socklen_t udp_new_addr_len = sizeof(address_t::storage_t);
- if ((len = recvfrom(local_listen_fd, data, max_data_len+1, 0,
- (struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) < 0) {
- mylog(log_error,"recv_from error,this shouldnt happen,err=%s,but we can try to continue\n",strerror(errno));
- continue;
- //myexit(1);
- };
- address_t new_addr;
- new_addr.from_sockaddr((struct sockaddr *) &udp_new_addr_in,udp_new_addr_len);
- if(len==max_data_len+1)
- {
- mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
- continue;
- }
- if(de_cook(data,len)<0)
- {
- mylog(log_warn,"de_cook(data,len)failed \n");
- continue;
- }
- char header=0;
- if(get_header(header,data,len)!=0)
- {
- mylog(log_warn,"get_header failed\n");
- continue;
- }
- if(udp_dest.inner.fd_addr.addr==new_addr)
- {
- if(header==header_keep_alive)
- {
- mylog(log_trace,"got keep_alive packet\n");
- continue;
- }
- if(header!=header_new_connect&& header!=header_normal)
- {
- mylog(log_warn,"invalid header\n");
- continue;
- }
- }
- else
- {
- if(header==header_keep_alive)
- {
- mylog(log_debug,"got keep_alive packet from unexpected client\n");
- continue;
- }
- if(header==header_new_connect)
- {
- mylog(log_info,"new connection from %s\n", new_addr.get_str());
- udp_dest.inner.fd_addr.addr=new_addr;
- //udp_dest.inner.fd_ip_port.ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
- //udp_dest.inner.fd_ip_port.ip_port.port=ntohs(udp_new_addr_in.sin_port);
- conn_info.fec_decode_manager.clear();
- conn_info.fec_encode_manager.clear_data();
- memset(&conn_info.stat,0,sizeof(conn_info.stat));
- }
- else if(header==header_normal)
- {
- mylog(log_debug,"rejected connection from %s\n", new_addr.get_str());
- len=1;
- data[0]=header_reject;
- do_cook(data,len);
- dest_t tmp_dest;
- tmp_dest.type=type_fd_addr;
- tmp_dest.inner.fd_addr.fd=local_listen_fd;
- tmp_dest.inner.fd_addr.addr=new_addr;
- delay_manager.add(0,tmp_dest,data,len);;
- continue;
- }
- else
- {
- mylog(log_warn,"invalid header\n");
- }
- }
- mylog(log_trace,"Received packet from %s,len: %d\n", new_addr.get_str(),len);
- from_fec_to_normal2(conn_info,tun_dest,data,len);
- }
- else if(events[idx].data.u64==(u64_t)tun_fd)
- {
- len=read(tun_fd,data,max_data_len+1);
- if(len==max_data_len+1)
- {
- mylog(log_warn,"huge packet, data_len > %d,dropped\n",max_data_len);
- continue;
- }
- if(len<0)
- {
- mylog(log_warn,"read from tun_fd return %d,errno=%s\n",len,strerror(errno));
- continue;
- }
- do_mssfix(data,len);
- mylog(log_trace,"Received packet from tun,len: %d\n",len);
- if(udp_dest.inner.fd_addr.addr.is_vaild()==0)
- {
- mylog(log_debug,"received packet from tun,but there is no client yet,dropped packet\n");
- continue;
- }
- from_normal_to_fec2(conn_info,udp_dest,data,len,header_normal);
- }
- else if (events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd())
- {
- uint64_t value;
- read(delay_manager.get_timer_fd(), &value, 8);
- mylog(log_trace,"events[idx].data.u64 == (u64_t)delay_manager.get_timer_fd()\n");
- }
- else if (events[idx].data.u64 == (u64_t)fifo_fd)
- {
- char buf[buf_len];
- int len=read (fifo_fd, buf, sizeof (buf));
- if(len<0)
- {
- mylog(log_warn,"fifo read failed len=%d,errno=%s\n",len,strerror(errno));
- continue;
- }
- buf[len]=0;
- handle_command(buf);
- }
- else
- {
- assert(0==1);
- }
- }
- delay_manager.check();
- }
- return 0;
- }
|