yeyinlaoying/vnt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加参数关闭ip代理

Browse Source
This commit is contained in:
lubeilin
2023-10-09 20:56:04 +08:00
parent 9ad0525216
commit fc104d5dee
10 changed files with 74 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vnt/Cargo.toml
+1 -1
View File
@@ -44,7 +44,7 @@ protobuf-codegen = "3.2.0"
protoc-bin-vendored = "3.0.0"
[features]
default = ["server_encrypt","aes_gcm","aes_cbc","aes_ecb","sm4_cbc"]
default = ["server_encrypt","aes_gcm","aes_cbc","aes_ecb","sm4_cbc","ip_proxy"]
openssl = ["openssl-sys"]
# 从源码编译
openssl-vendored = ["openssl-sys/vendored"]
vnt/src/core/mod.rs
+6 -1
View File
@@ -302,7 +302,7 @@ impl VntUtil {
Some(ExternalRoute::new(config.in_ips))
};
#[cfg(feature = "ip_proxy")]
let (tcp_proxy, udp_proxy, ip_proxy_map) = if config.out_ips.is_empty() {
let (tcp_proxy, udp_proxy, ip_proxy_map) = if config.out_ips.is_empty() || config.no_proxy {
(None, None, None)
} else {
let (tcp_proxy, udp_proxy, ip_proxy_map) = crate::ip_proxy::init_proxy(
@@ -582,6 +582,8 @@ pub struct Config {
pub tcp: bool,
pub ip: Option<Ipv4Addr>,
pub relay: bool,
#[cfg(feature = "ip_proxy")]
pub no_proxy: bool,
pub server_encrypt: bool,
pub parallel: usize,
pub cipher_model: CipherModel,
@@ -607,6 +609,7 @@ impl Config {
tcp: bool,
ip: Option<Ipv4Addr>,
relay: bool,
#[cfg(feature = "ip_proxy")] no_proxy: bool,
server_encrypt: bool,
parallel: usize,
cipher_model: CipherModel,
@@ -635,6 +638,8 @@ impl Config {
tcp,
ip,
relay,
#[cfg(feature = "ip_proxy")]
no_proxy,
server_encrypt,
parallel,
cipher_model,
vnt/src/handle/recv_handler.rs
+2 -50
View File
@@ -233,57 +233,9 @@ impl ChannelDataHandler {
if self.out_external_route.allow(&ipv4.destination_ip()) {
#[cfg(feature = "ip_proxy")]
if let Some(ip_proxy_map) = &self.ip_proxy_map {
match ipv4.protocol() {
ipv4::protocol::Protocol::Tcp => {
if ip_proxy_map.tcp_handler.recv_handle(
&mut ipv4,
source,
destination,
)? {
return Ok(());
}
}
ipv4::protocol::Protocol::Udp => {
if ip_proxy_map.udp_handler.recv_handle(
&mut ipv4,
source,
destination,
)? {
return Ok(());
}
}
#[cfg(not(target_os = "android"))]
ipv4::protocol::Protocol::Icmp => {
if ip_proxy_map.icmp_handler.recv_handle(
&mut ipv4,
source,
destination,
)? {
return Ok(());
}
}
_ => {
log::warn!(
"不支持的ip代理ipv4协议{:?}:{}->{}->{}",
ipv4.protocol(),
source,
destination,
ipv4.destination_ip()
);
return Err(Error::Warn(
"不支持的ip代理ipv4协议".to_string(),
));
}
if ip_proxy_map.recv_handle(&mut ipv4, source, destination)? {
return Ok(());
}
} else {
log::warn!(
"不支持ip代理{:?}:{}->{}->{}",
ipv4.protocol(),
source,
destination,
ipv4.destination_ip()
);
return Err(Error::Warn("不支持ip代理".to_string()));
}
} else {
log::warn!(
vnt/src/handle/tun_tap/mod.rs
+2 -11
View File
@@ -188,17 +188,8 @@ pub fn base_handle(
}
#[cfg(feature = "ip_proxy")]
if let Some(proxy_map) = proxy_map {
match protocol {
Protocol::Tcp => {
let mut ipv4_packet = IpV4Packet::new(net_packet.payload_mut())?;
proxy_map.tcp_handler.send_handle(&mut ipv4_packet)?;
}
Protocol::Udp => {
let mut ipv4_packet = IpV4Packet::new(net_packet.payload_mut())?;
proxy_map.udp_handler.send_handle(&mut ipv4_packet)?;
}
_ => {}
}
let mut ipv4_packet = IpV4Packet::new(net_packet.payload_mut())?;
proxy_map.send_handle(&mut ipv4_packet)?;
}
client_cipher.encrypt_ipv4(&mut net_packet)?;
//优先发到直连到地址
vnt/src/ip_proxy/mod.rs
+41
View File
@@ -6,6 +6,7 @@ use std::{io, thread};
use crossbeam_utils::atomic::AtomicCell;
use dashmap::DashMap;
use packet::ip::ipv4;
use tokio::net::UdpSocket;
use packet::ip::ipv4::packet::IpV4Packet;
@@ -173,3 +174,43 @@ pub fn send(
}
}
}
impl ProxyHandler for IpProxyMap {
fn recv_handle(
&self,
ipv4: &mut IpV4Packet<&mut [u8]>,
source: Ipv4Addr,
destination: Ipv4Addr,
) -> io::Result<bool> {
match ipv4.protocol() {
ipv4::protocol::Protocol::Tcp => {
self.tcp_handler.recv_handle(ipv4, source, destination)
}
ipv4::protocol::Protocol::Udp => {
self.udp_handler.recv_handle(ipv4, source, destination)
}
#[cfg(not(target_os = "android"))]
ipv4::protocol::Protocol::Icmp => {
self.icmp_handler.recv_handle(ipv4, source, destination)
}
_ => {
log::warn!(
"不支持的ip代理ipv4协议{:?}:{}->{}->{}",
ipv4.protocol(),
source,
destination,
ipv4.destination_ip()
);
Ok(false)
}
}
}
fn send_handle(&self, ipv4: &mut IpV4Packet<&mut [u8]>) -> io::Result<()> {
match ipv4.protocol() {
ipv4::protocol::Protocol::Tcp => self.tcp_handler.send_handle(ipv4),
ipv4::protocol::Protocol::Udp => self.udp_handler.send_handle(ipv4),
_ => Ok(()),
}
}
}