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

Re-export the missing types that are used within Args (#87)

Browse source
This commit is contained in:
Andrej Mihajlov 2024-02-10 12:59:48 +01:00 • committed by GitHub
parent ea5ffff82c
commit 9a4bd9f800
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/lib.rs
View file

@ -1,5 +1,4 @@
use crate::{
args::ProxyType,
directions::{IncomingDataEvent, IncomingDirection, OutgoingDirection},
http::HttpManager,
session_info::{IpProtocol, SessionInfo},
@ -21,7 +20,7 @@ use tokio::{
use tproxy_config::is_private_ip;
use udp_stream::UdpStream;
pub use {
args::{ArgVerbosity, Args},
args::{ArgDns, ArgProxy, ArgVerbosity, Args, ProxyType},
error::{Error, Result},
};
@ -117,7 +116,7 @@ where
let key = args.proxy.credentials.clone();
let dns_addr = args.dns_addr;
let ipv6_enabled = args.ipv6_enabled;
let virtual_dns = if args.dns == args::ArgDns::Virtual {
let virtual_dns = if args.dns == ArgDns::Virtual {
Some(Arc::new(Mutex::new(VirtualDns::new())))
} else {
None
@ -183,7 +182,7 @@ where
if is_private_ip(info.dst.ip()) {
info.dst.set_ip(dns_addr);
}
if args.dns == args::ArgDns::OverTcp {
if args.dns == ArgDns::OverTcp {
let proxy_handler = mgr.new_proxy_handler(info, None, false).await?;
tokio::spawn(async move {
if let Err(err) = handle_dns_over_tcp_session(udp, server_addr, proxy_handler, ipv6_enabled).await {
@ -193,7 +192,7 @@ where
});
continue;
}
if args.dns == args::ArgDns::Virtual {
if args.dns == ArgDns::Virtual {
tokio::spawn(async move {
if let Some(virtual_dns) = virtual_dns {
if let Err(err) = handle_virtual_dns_session(udp, virtual_dns).await {
@ -204,7 +203,7 @@ where
});
continue;
}
assert_eq!(args.dns, args::ArgDns::Direct);
assert_eq!(args.dns, ArgDns::Direct);
}
let domain_name = if let Some(virtual_dns) = &virtual_dns {
let mut virtual_dns = virtual_dns.lock().await;