mirror of
https://github.com/cargo-bins/cargo-binstall.git
synced 2025-05-20 02:40:02 +00:00
Minor refactor for binstalk-downloader
(#1365)
- Mv `resolver.rs` => `remote/resolver.rs` - Fix clippy warnings in `resolver.rs` on unix Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
This commit is contained in:
parent
6560225c03
commit
9c4e3a6642
3 changed files with 6 additions and 6 deletions
65
crates/binstalk-downloader/src/remote/resolver.rs
Normal file
65
crates/binstalk-downloader/src/remote/resolver.rs
Normal file
|
@ -0,0 +1,65 @@
|
|||
use std::{net::SocketAddr, sync::Arc};
|
||||
|
||||
use hyper::client::connect::dns::Name;
|
||||
use once_cell::sync::OnceCell;
|
||||
use reqwest::dns::{Addrs, Resolve};
|
||||
#[cfg(windows)]
|
||||
use trust_dns_resolver::config::{NameServerConfig, Protocol, ResolverConfig, ResolverOpts};
|
||||
use trust_dns_resolver::TokioAsyncResolver;
|
||||
|
||||
#[derive(Debug, Default, Clone)]
|
||||
pub struct TrustDnsResolver(Arc<OnceCell<TokioAsyncResolver>>);
|
||||
|
||||
impl Resolve for TrustDnsResolver {
|
||||
fn resolve(&self, name: Name) -> reqwest::dns::Resolving {
|
||||
let resolver = self.clone();
|
||||
Box::pin(async move {
|
||||
let resolver = resolver.0.get_or_try_init(new_resolver)?;
|
||||
|
||||
let lookup = resolver.lookup_ip(name.as_str()).await?;
|
||||
let addrs: Addrs = Box::new(lookup.into_iter().map(|ip| SocketAddr::new(ip, 0)));
|
||||
Ok(addrs)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
fn new_resolver() -> Result<TokioAsyncResolver, Box<dyn std::error::Error + Send + Sync>> {
|
||||
#[cfg(unix)]
|
||||
{
|
||||
let (config, opts) = trust_dns_resolver::system_conf::read_system_conf()?;
|
||||
Ok(TokioAsyncResolver::tokio(config, opts)?)
|
||||
}
|
||||
#[cfg(windows)]
|
||||
{
|
||||
let mut config = ResolverConfig::new();
|
||||
let opts = ResolverOpts::default();
|
||||
|
||||
let current_interface = default_net::get_default_interface()?;
|
||||
ipconfig::get_adapters()?
|
||||
.iter()
|
||||
.filter_map(|adapter| {
|
||||
if adapter.adapter_name() == current_interface.name {
|
||||
Some(adapter.dns_servers())
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.flatten()
|
||||
.for_each(|addr| {
|
||||
let socket_addr = SocketAddr::new(*addr, 53);
|
||||
for protocol in [Protocol::Udp, Protocol::Tcp] {
|
||||
config.add_name_server(NameServerConfig {
|
||||
socket_addr,
|
||||
protocol,
|
||||
tls_dns_name: None,
|
||||
trust_nx_responses: false,
|
||||
#[cfg(feature = "rustls")]
|
||||
tls_config: None,
|
||||
bind_addr: None,
|
||||
})
|
||||
}
|
||||
});
|
||||
|
||||
Ok(TokioAsyncResolver::tokio(config, opts)?)
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue