Skip to content

Commit

Permalink
added missing files
Browse files Browse the repository at this point in the history
  • Loading branch information
@wangyu-
wangyu- committed Aug 20, 2018
1 parent 07608b9 commit 35ce5af
Show file tree
Hide file tree
Showing 2 changed files with 607 additions and 0 deletions.
290 changes: 290 additions & 0 deletions tun_dev_client.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,290 @@
#include "tun_dev.h"


int tun_dev_client_event_loop()
{
char data[buf_len];
int len;
int i,j,k,ret;
int 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_tun(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);
///}

struct ev_loop * loop= ev_default_loop(0);
assert(loop != NULL);


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;
}
Loading

0 comments on commit 35ce5af

Please sign in to comment.