mirror of
https://github.com/wangyu-/UDPspeeder.git
synced 2025-01-19 06:19:34 +08:00
298 lines
6.3 KiB
C++
298 lines
6.3 KiB
C++
/*
|
|
* tun.cpp
|
|
*
|
|
* Created on: Oct 26, 2017
|
|
* Author: root
|
|
*/
|
|
|
|
|
|
#include "common.h"
|
|
#include "log.h"
|
|
#include "misc.h"
|
|
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,char * local_ip,char * 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 = inet_addr(local_ip);
|
|
memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
|
|
assert(ioctl(sockfd, SIOCSIFADDR, &ifr)==0);
|
|
|
|
|
|
|
|
sai.sin_addr.s_addr = inet_addr(local_ip);
|
|
memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
|
|
assert(ioctl(sockfd, SIOCSIFADDR, &ifr)==0);
|
|
|
|
|
|
|
|
sai.sin_addr.s_addr = inet_addr(remote_ip);
|
|
memcpy(&ifr.ifr_addr,&sai, sizeof(struct sockaddr));
|
|
assert(ioctl(sockfd, SIOCSIFDSTADDR, &ifr)==0);
|
|
|
|
ifr.ifr_mtu=mtu;
|
|
assert(ioctl(sockfd, SIOCSIFMTU, &ifr)==0);
|
|
|
|
|
|
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 );
|
|
assert(ioctl(sockfd, SIOCSIFFLAGS, &ifr)==0);
|
|
|
|
//printf("i m here2\n");
|
|
return 0;
|
|
}
|
|
|
|
|
|
int tun_dev_client_event_loop()
|
|
{
|
|
char buf[buf_len+1];
|
|
//char *data=buf+1;
|
|
int len;
|
|
int i,j,k,ret;
|
|
int epoll_fd,tun_fd;
|
|
|
|
int remote_fd;
|
|
fd64_t remote_fd64;
|
|
|
|
|
|
tun_fd=get_tun_fd("tun11");
|
|
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("tun11","10.0.0.2","10.0.0.1",1000)==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);
|
|
}
|
|
|
|
dest_t dest;
|
|
dest.type=type_fd64;
|
|
dest.inner.fd64=remote_fd64;
|
|
//dest.conv=conv;
|
|
//dest.inner.ip_port=dest_ip_port;
|
|
//dest.cook=1;
|
|
|
|
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\n");
|
|
myexit(0);
|
|
}
|
|
else
|
|
{
|
|
mylog(log_fatal,"epoll_wait return %d\n", nfds);
|
|
myexit(-1);
|
|
}
|
|
}
|
|
int idx;
|
|
for (idx = 0; idx < nfds; ++idx)
|
|
{
|
|
if(events[idx].data.u64==(u64_t)tun_fd)
|
|
{
|
|
len=read(tun_fd,buf,max_data_len);
|
|
assert(len>=0);
|
|
|
|
mylog(log_trace,"Received packet from tun,len: %d\n",len);
|
|
|
|
delay_manager.add(0,dest,buf,len);;
|
|
}
|
|
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,buf,max_data_len,0);
|
|
|
|
mylog(log_trace,"Received packet from udp,len: %d\n",len);
|
|
assert(len>=0);
|
|
|
|
assert(write(tun_fd,buf,len)>0);
|
|
}
|
|
}
|
|
delay_manager.check();
|
|
}
|
|
|
|
|
|
return 0;
|
|
}
|
|
|
|
int tun_dev_server_event_loop()
|
|
{
|
|
char buf[buf_len+1];
|
|
char *data=buf+1;
|
|
int len;
|
|
int i,j,k,ret;
|
|
int epoll_fd,tun_fd;
|
|
|
|
tun_fd=get_tun_fd("tun11");
|
|
assert(tun_fd>0);
|
|
|
|
assert(new_listen_socket(local_listen_fd,local_ip_uint32,local_port)==0);
|
|
|
|
assert(set_if("tun11","10.0.0.1","10.0.0.2",1000)==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);
|
|
}
|
|
|
|
//ip_port_t dest_ip_port;
|
|
|
|
dest_t dest;
|
|
dest.type=type_ip_port;
|
|
dest.inner.ip_port.ip=0;
|
|
dest.inner.ip_port.port=0;
|
|
//dest.conv=conv;
|
|
//dest.inner.ip_port=dest_ip_port;
|
|
//dest.cook=1;
|
|
|
|
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\n");
|
|
myexit(0);
|
|
}
|
|
else
|
|
{
|
|
mylog(log_fatal,"epoll_wait return %d\n", nfds);
|
|
myexit(-1);
|
|
}
|
|
}
|
|
int idx;
|
|
for (idx = 0; idx < nfds; ++idx)
|
|
{
|
|
|
|
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, buf, max_data_len, 0,
|
|
(struct sockaddr *) &udp_new_addr_in, &udp_new_addr_len)) == -1) {
|
|
mylog(log_error,"recv_from error,this shouldnt happen,err=%s,but we can try to continue\n",strerror(errno));
|
|
continue;
|
|
//myexit(1);
|
|
};
|
|
|
|
dest.inner.ip_port.ip=udp_new_addr_in.sin_addr.s_addr;
|
|
dest.inner.ip_port.port=ntohs(udp_new_addr_in.sin_port);
|
|
|
|
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);
|
|
|
|
assert(write(tun_fd,buf,len)>0);
|
|
|
|
}
|
|
else if(events[idx].data.u64==(u64_t)tun_fd)
|
|
{
|
|
len=read(tun_fd,buf,max_data_len);
|
|
assert(len>=0);
|
|
|
|
mylog(log_trace,"Received packet from tun,len: %d\n",len);
|
|
|
|
if(dest.inner.ip_port.to_u64()==0)
|
|
{
|
|
mylog(log_warn,"there is no client yet\n");
|
|
continue;
|
|
}
|
|
|
|
delay_manager.add(0,dest,buf,len);;
|
|
|
|
|
|
}
|
|
}
|
|
delay_manager.check();
|
|
}
|
|
|
|
|
|
return 0;
|
|
}
|