| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944 | 
							- /*
 
-  * 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>
 
- my_time_t last_keep_alive_time=0;
 
- int keep_alive_interval=1000;//1000ms
 
- 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(client_or_server==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\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>g_fec_mtu-40-10) //minus extra 10 for safe
 
-     		{
 
-     			new_mss=g_fec_mtu-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_socket(remote_fd,remote_ip_uint32,remote_port)==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_socket(local_listen_fd,local_ip_uint32,local_port)==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_ip_port;
 
- 	udp_dest.inner.fd_ip_port.fd=local_listen_fd;
 
- 	udp_dest.inner.fd_ip_port.ip_port.ip=0;
 
- 	udp_dest.inner.fd_ip_port.ip_port.port=0;
 
- 	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.fd64_ip_port.ip_port.to_u64()==0)
 
- 				{
 
- 					continue;
 
- 				}
 
- 				conn_info.stat.report_as_server(udp_dest.inner.fd_ip_port.ip_port);
 
- 				do_keep_alive(udp_dest);
 
- 			}
 
- 			else if(events[idx].data.u64==conn_info.fec_encode_manager.get_timer_fd64())
 
- 			{
 
- 				assert(udp_dest.inner.fd64_ip_port.ip_port.to_u64()!=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);
 
- 				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);
 
- 				};
 
- 				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_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)))
 
- 				{
 
- 					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:%d \n", inet_ntoa(udp_new_addr_in.sin_addr),
 
- 												ntohs(udp_new_addr_in.sin_port));
 
- 						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();
 
- 						memset(&conn_info.stat,0,sizeof(conn_info.stat));
 
- 					}
 
- 					else if(header==header_normal)
 
- 					{
 
- 						mylog(log_debug,"rejected connection from %s:%d\n", inet_ntoa(udp_new_addr_in.sin_addr),ntohs(udp_new_addr_in.sin_port));
 
- 						len=1;
 
- 						data[0]=header_reject;
 
- 						do_cook(data,len);
 
- 						dest_t tmp_dest;
 
- 						tmp_dest.type=type_fd_ip_port;
 
- 						tmp_dest.inner.fd_ip_port.fd=local_listen_fd;
 
- 						tmp_dest.inner.fd_ip_port.ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
 
- 						tmp_dest.inner.fd_ip_port.ip_port.port=ntohs(udp_new_addr_in.sin_port);
 
- 						delay_manager.add(0,tmp_dest,data,len);;
 
- 						continue;
 
- 					}
 
- 					else
 
- 					{
 
- 						mylog(log_warn,"invalid header\n");
 
- 					}
 
- 				}
 
- 				mylog(log_trace,"Received packet from %s:%d,len: %d\n", inet_ntoa(udp_new_addr_in.sin_addr),
 
- 						ntohs(udp_new_addr_in.sin_port),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.fd64_ip_port.ip_port.to_u64()==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;
 
- }
 
 
  |