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

cleanup

Browse source
This commit is contained in:
Σrebe - Romain GERARD 2023-12-19 23:16:06 +01:00
parent c766b89553
commit 001759afac
No known key found for this signature in database
GPG key ID: 7A42B4B97E0332F4
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/dns.rs
View file

@ -1,4 +1,3 @@
use anyhow::Context;
use hickory_resolver::TokioAsyncResolver; use hickory_resolver::TokioAsyncResolver;
use std::net::{IpAddr, SocketAddr, SocketAddrV4, SocketAddrV6}; use std::net::{IpAddr, SocketAddr, SocketAddrV4, SocketAddrV6};
@ -12,13 +11,11 @@ impl DnsResolver {
pub async fn lookup_host(&self, domain: &str, port: u16) -> anyhow::Result<Vec<SocketAddr>> { pub async fn lookup_host(&self, domain: &str, port: u16) -> anyhow::Result<Vec<SocketAddr>> {
let addrs: Vec<SocketAddr> = match self { let addrs: Vec<SocketAddr> = match self {
DnsResolver::System => tokio::net::lookup_host(format!("{}:{}", domain, port)) DnsResolver::System => tokio::net::lookup_host(format!("{}:{}", domain, port))
.await .await?
.with_context(|| format!("cannot resolve domain: {}", domain))?
.collect(), .collect(),
DnsResolver::TrustDns(dns_resolver) => dns_resolver DnsResolver::TrustDns(dns_resolver) => dns_resolver
.lookup_ip(domain) .lookup_ip(domain)
.await .await?
.with_context(|| format!("cannot resolve domain: {}", domain))?
.into_iter() .into_iter()
.map(|ip| match ip { .map(|ip| match ip {
IpAddr::V4(ip) => SocketAddr::V4(SocketAddrV4::new(ip, port)), IpAddr::V4(ip) => SocketAddr::V4(SocketAddrV4::new(ip, port)),