hex/wstunnel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix dns-resolver cmd line parsing

Browse source
This commit is contained in:
Σrebe - Romain GERARD 2024-01-02 13:15:04 +01:00
parent 817571c907
commit bffd2470e7
No known key found for this signature in database
GPG key ID: 7A42B4B97E0332F4
3 changed files with 16 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/main.rs
View file

@ -852,7 +852,13 @@ async fn main() {
_ => panic!("invalid protocol for dns resolver"),
};
let sock = match resolver.host().unwrap() {
Host::Domain(_) => panic!("Dns resolver must be an ip address"),
Host::Domain(host) => match Host::parse(host) {
Ok(Host::Ipv4(ip)) => SocketAddr::V4(SocketAddrV4::new(ip, port)),
Ok(Host::Ipv6(ip)) => SocketAddr::V6(SocketAddrV6::new(ip, port, 0, 0)),
Ok(Host::Domain(_)) | Err(_) => {
panic!("Dns resolver must be an ip address, got {}", host)
}
},
Host::Ipv4(ip) => SocketAddr::V4(SocketAddrV4::new(ip, port)),
Host::Ipv6(ip) => SocketAddr::V6(SocketAddrV6::new(ip, port, 0, 0)),
};