luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-04-21 22:39:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Run cargo fmt

Browse source
This commit is contained in:
B. Blechschmidt 2023-03-20 17:56:54 +01:00
parent 4ce2bdf0ba
commit 2f4be7a6cd
3 changed files with 25 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/main.rs
View file

@ -1,7 +1,7 @@
use tun2proxy::{ProxyType, main_entry};
use clap::{Parser, ValueEnum}; use clap::{Parser, ValueEnum};
use env_logger::Env; use env_logger::Env;
use std::net::SocketAddr; use std::net::SocketAddr;
use tun2proxy::{main_entry, ProxyType};
/// Tunnel interface to proxy /// Tunnel interface to proxy
#[derive(Parser)] #[derive(Parser)]

29
src/tun2proxy.rs
View file

@ -6,15 +6,17 @@ use mio::unix::SourceFd;
use mio::{Events, Interest, Poll, Token}; use mio::{Events, Interest, Poll, Token};
use smoltcp::iface::{Config, Interface, SocketHandle, SocketSet}; use smoltcp::iface::{Config, Interface, SocketHandle, SocketSet};
use smoltcp::phy::{Device, Medium, RxToken, TunTapInterface, TxToken}; use smoltcp::phy::{Device, Medium, RxToken, TunTapInterface, TxToken};
use smoltcp::socket::tcp;
use smoltcp::time::Instant; use smoltcp::time::Instant;
use smoltcp::wire::{IpAddress, IpCidr, Ipv4Address, Ipv4Packet, Ipv6Address, Ipv6Packet, TcpPacket, UdpPacket}; use smoltcp::wire::{
use std::collections::{HashMap}; IpAddress, IpCidr, Ipv4Address, Ipv4Packet, Ipv6Address, Ipv6Packet, TcpPacket, UdpPacket,
};
use std::collections::HashMap;
use std::convert::From; use std::convert::From;
use std::io::{Read, Write}; use std::io::{Read, Write};
use std::net::Shutdown::Both; use std::net::Shutdown::Both;
use std::net::{IpAddr, Shutdown, SocketAddr}; use std::net::{IpAddr, Shutdown, SocketAddr};
use std::os::unix::io::AsRawFd; use std::os::unix::io::AsRawFd;
use smoltcp::socket::tcp;
pub struct ProxyError { pub struct ProxyError {
message: String, message: String,
@ -205,11 +207,21 @@ impl<'a> TunToProxy<'a> {
let mut virt = VirtualTunDevice::new(tun.capabilities()); let mut virt = VirtualTunDevice::new(tun.capabilities());
let mut iface = Interface::new(config, &mut virt); let mut iface = Interface::new(config, &mut virt);
iface.update_ip_addrs(|ip_addrs| { iface.update_ip_addrs(|ip_addrs| {
ip_addrs.push(IpCidr::new(IpAddress::v4(0, 0, 0, 1), 0)).unwrap(); ip_addrs
ip_addrs.push(IpCidr::new(IpAddress::v6(0, 0, 0, 0, 0, 0, 0, 1), 0)).unwrap() .push(IpCidr::new(IpAddress::v4(0, 0, 0, 1), 0))
.unwrap();
ip_addrs
.push(IpCidr::new(IpAddress::v6(0, 0, 0, 0, 0, 0, 0, 1), 0))
.unwrap()
}); });
iface.routes_mut().add_default_ipv4_route(Ipv4Address::new(0, 0, 0, 1)).unwrap(); iface
iface.routes_mut().add_default_ipv6_route(Ipv6Address::new(0, 0, 0, 0, 0, 0, 0, 1)).unwrap(); .routes_mut()
.add_default_ipv4_route(Ipv4Address::new(0, 0, 0, 1))
.unwrap();
iface
.routes_mut()
.add_default_ipv6_route(Ipv6Address::new(0, 0, 0, 0, 0, 0, 0, 1))
.unwrap();
iface.set_any_ip(true); iface.set_any_ip(true);
Self { Self {
@ -232,7 +244,8 @@ impl<'a> TunToProxy<'a> {
} }
fn expect_smoltcp_send(&mut self) { fn expect_smoltcp_send(&mut self) {
self.iface.poll(Instant::now(), &mut self.device, &mut self.sockets); self.iface
.poll(Instant::now(), &mut self.device, &mut self.sockets);
while let Some(vec) = self.device.exfiltrate_packet() { while let Some(vec) = self.device.exfiltrate_packet() {
let slice = vec.as_slice(); let slice = vec.as_slice();

2
src/virtdevice.rs
View file

@ -6,7 +6,7 @@ use smoltcp::time::Instant;
pub struct VirtualTunDevice { pub struct VirtualTunDevice {
capabilities: DeviceCapabilities, capabilities: DeviceCapabilities,
inbuf: Vec<Vec<u8>>, inbuf: Vec<Vec<u8>>,
outbuf: Vec<Vec<u8>> outbuf: Vec<Vec<u8>>,
} }
impl VirtualTunDevice { impl VirtualTunDevice {