use std::net::{IpAddr, SocketAddr};
use std::time::Duration;
use hickory_resolver::config::{ResolverConfig, GOOGLE};
use hickory_resolver::net::runtime::TokioRuntimeProvider;
use hickory_resolver::TokioResolver;
use reqwest::Url;
use std::sync::LazyLock;
use tokio::net::lookup_host;
use tracing::debug;
use crate::error::{Result, SeerError};
static FALLBACK_RESOLVER: LazyLock<Option<TokioResolver>> = LazyLock::new(|| {
let mut builder = TokioResolver::builder_with_config(
ResolverConfig::udp_and_tcp(&GOOGLE),
TokioRuntimeProvider::default(),
);
{
crate::dns::apply_standard_opts(builder.options_mut(), Duration::from_secs(5));
}
builder.build().ok()
});
const PRIMARY_RESOLVE_TIMEOUT: Duration = Duration::from_secs(5);
pub fn is_reserved_ip(ip: IpAddr) -> bool {
match ip {
IpAddr::V4(v4) => {
let o = v4.octets();
v4.is_loopback()
|| v4.is_private()
|| v4.is_link_local()
|| v4.is_multicast()
|| v4.is_broadcast()
|| v4.is_unspecified()
|| v4.is_documentation()
|| o[0] == 0
|| o[0] >= 240
|| (o[0] == 100 && (o[1] & 0xC0) == 64)
|| (o[0] == 192 && o[1] == 0 && o[2] == 0)
|| (o[0] == 198 && (o[1] == 18 || o[1] == 19))
}
IpAddr::V6(v6) => {
if v6.is_loopback() || v6.is_multicast() || v6.is_unspecified() {
return true;
}
let seg = v6.segments();
if (seg[0] & 0xfe00) == 0xfc00 {
return true;
}
if (seg[0] & 0xffc0) == 0xfe80 {
return true;
}
if (seg[0] & 0xffc0) == 0xfec0 {
return true;
}
if seg[0] == 0x2001 && seg[1] == 0x0db8 {
return true;
}
if seg[0] == 0x2002 {
return true;
}
if seg[0] == 0x0064
&& seg[1] == 0xff9b
&& seg[2] == 0
&& seg[3] == 0
&& seg[4] == 0
&& seg[5] == 0
{
return true;
}
if seg[0] == 0x0064 && seg[1] == 0xff9b && seg[2] == 0x0001 {
return true;
}
if seg[0] == 0x2001 && seg[1] == 0 {
return true;
}
if seg[0] == 0
&& seg[1] == 0
&& seg[2] == 0
&& seg[3] == 0
&& seg[4] == 0xffff
&& seg[5] == 0
{
let embedded = std::net::Ipv4Addr::from(((seg[6] as u32) << 16) | seg[7] as u32);
if is_reserved_ip(IpAddr::V4(embedded)) {
return true;
}
}
if v6
.to_ipv4_mapped()
.is_some_and(|v4| is_reserved_ip(IpAddr::V4(v4)))
{
return true;
}
if seg[0] == 0
&& seg[1] == 0
&& seg[2] == 0
&& seg[3] == 0
&& seg[4] == 0
&& seg[5] == 0
{
let embedded = std::net::Ipv4Addr::from(((seg[6] as u32) << 16) | seg[7] as u32);
if is_reserved_ip(IpAddr::V4(embedded)) {
return true;
}
}
false
}
}
}
pub async fn validate_public_host(host: &str, port: u16) -> Result<()> {
resolve_public_host(host, port).await.map(|_| ())
}
pub async fn resolve_public_host(host: &str, port: u16) -> Result<Vec<SocketAddr>> {
if let Ok(ip) = host.parse::<IpAddr>() {
if is_reserved_ip(ip) {
return Err(SeerError::InvalidInput(format!(
"refusing to connect to reserved address: {}",
ip
)));
}
return Ok(vec![SocketAddr::new(ip, port)]);
}
let addrs: Vec<SocketAddr> =
match tokio::time::timeout(PRIMARY_RESOLVE_TIMEOUT, lookup_host((host, port))).await {
Ok(Ok(iter)) => iter.collect(),
os_failure => {
let os_err = match os_failure {
Ok(Err(e)) => e.to_string(),
_ => format!(
"OS resolver timed out after {}s",
PRIMARY_RESOLVE_TIMEOUT.as_secs()
),
};
debug!(
host = %host,
error = %os_err,
"OS resolver could not resolve host; trying hickory fallback"
);
let Some(resolver) = FALLBACK_RESOLVER.as_ref() else {
return Err(SeerError::DnsError(format!(
"DNS resolution failed for {host}: {os_err} (no fallback resolver available)"
)));
};
match resolver.lookup_ip(host).await {
Ok(resp) => resp.iter().map(|ip| SocketAddr::new(ip, port)).collect(),
Err(fallback_err) => {
return Err(SeerError::DnsError(format!(
"DNS resolution failed for {host}: {os_err} (fallback: {fallback_err})"
)));
}
}
}
};
if addrs.is_empty() {
return Err(SeerError::DnsError(format!(
"no addresses resolved for {host}"
)));
}
for sa in &addrs {
if is_reserved_ip(sa.ip()) {
debug!(
host = %host,
resolved_ip = %sa.ip(),
"refusing host: resolves to a reserved (non-public) address"
);
return Err(SeerError::InvalidInput(format!(
"{host} is not permitted (resolves to a non-public address)"
)));
}
}
Ok(addrs)
}
pub(crate) const USER_AGENT: &str = concat!("Seer/", env!("CARGO_PKG_VERSION"));
pub(crate) fn client_builder(timeout: Duration) -> reqwest::ClientBuilder {
reqwest::Client::builder()
.timeout(timeout)
.redirect(reqwest::redirect::Policy::none())
}
pub(crate) fn url_host(url: &Url) -> Option<String> {
Some(match url.host()? {
url::Host::Domain(d) => d.to_string(),
url::Host::Ipv4(ip) => ip.to_string(),
url::Host::Ipv6(ip) => ip.to_string(),
})
}
pub(crate) async fn validate_http_url(url: &Url) -> Result<Vec<SocketAddr>> {
let scheme = url.scheme();
if scheme != "https" && scheme != "http" {
return Err(SeerError::HttpError(format!(
"unsupported URL scheme: {}",
scheme
)));
}
if !url.username().is_empty() || url.password().is_some() {
return Err(SeerError::HttpError(
"URL credentials are not allowed".to_string(),
));
}
let host = url_host(url).ok_or_else(|| SeerError::HttpError("missing URL host".to_string()))?;
let port = url.port_or_known_default().unwrap_or(443);
if port != 80 && port != 443 {
return Err(SeerError::HttpError(format!(
"non-standard port {} is not allowed in redirects",
port
)));
}
resolve_public_host(&host, port)
.await
.map_err(|e| SeerError::HttpError(e.to_string()))
}
#[cfg(test)]
#[allow(clippy::unwrap_used)]
mod tests {
use super::*;
#[test]
fn fallback_resolver_opts_match_the_shared_standard() {
use hickory_resolver::config::{ResolverOpts, ServerOrderingStrategy};
let mut opts = ResolverOpts::default();
crate::dns::apply_standard_opts(&mut opts, Duration::from_secs(5));
assert_eq!(
opts.server_ordering_strategy,
ServerOrderingStrategy::UserProvidedOrder
);
assert_eq!(opts.timeout, Duration::from_secs(5));
assert_eq!(opts.attempts, 2);
}
#[test]
fn rejects_loopback_v4() {
assert!(is_reserved_ip("127.0.0.1".parse().unwrap()));
}
#[test]
fn rejects_metadata_v4() {
assert!(is_reserved_ip("169.254.169.254".parse().unwrap()));
}
#[test]
fn rejects_rfc1918() {
assert!(is_reserved_ip("10.0.0.1".parse().unwrap()));
assert!(is_reserved_ip("172.16.0.1".parse().unwrap()));
assert!(is_reserved_ip("192.168.1.1".parse().unwrap()));
}
#[test]
fn rejects_cgnat() {
assert!(is_reserved_ip("100.64.0.1".parse().unwrap()));
}
#[test]
fn rejects_benchmarking() {
assert!(is_reserved_ip("198.18.0.1".parse().unwrap()));
}
#[test]
fn rejects_ipv6_loopback() {
assert!(is_reserved_ip("::1".parse().unwrap()));
}
#[test]
fn rejects_ipv6_ula() {
assert!(is_reserved_ip("fd00::1".parse().unwrap()));
}
#[test]
fn rejects_ipv6_site_local() {
assert!(is_reserved_ip("fec0::1".parse().unwrap()));
assert!(is_reserved_ip("feff::1".parse().unwrap()));
}
#[test]
fn rejects_ipv4_mapped_loopback() {
assert!(is_reserved_ip("::ffff:127.0.0.1".parse().unwrap()));
}
#[test]
fn rejects_ipv4_mapped_cgnat() {
assert!(is_reserved_ip("::ffff:100.64.0.1".parse().unwrap()));
}
#[test]
fn rejects_class_e_reserved() {
assert!(is_reserved_ip("240.0.0.1".parse().unwrap()));
assert!(is_reserved_ip("250.1.2.3".parse().unwrap()));
}
#[test]
fn rejects_this_network_0_8() {
assert!(is_reserved_ip("0.1.2.3".parse().unwrap()));
}
#[test]
fn rejects_ipv6_nat64_of_metadata() {
assert!(is_reserved_ip("64:ff9b::a9fe:a9fe".parse().unwrap()));
}
#[test]
fn rejects_ipv6_6to4() {
assert!(is_reserved_ip("2002:a9fe:a9fe::".parse().unwrap()));
}
#[test]
fn rejects_teredo_and_siit_translated_forms() {
assert!(is_reserved_ip(
"2001:0:4136:e378:8000:63bf:3fff:fdd2".parse().unwrap()
));
assert!(is_reserved_ip("::ffff:0:a9fe:a9fe".parse().unwrap()));
assert!(is_reserved_ip("::ffff:0:a00:1".parse().unwrap()));
assert!(!is_reserved_ip("::ffff:0:808:808".parse().unwrap()));
assert!(!is_reserved_ip("2001:4860:4860::8888".parse().unwrap()));
}
#[test]
fn rejects_ipv6_nat64_rfc8215_local_use_prefix() {
assert!(is_reserved_ip("64:ff9b:1::a9fe:a9fe".parse().unwrap()));
assert!(is_reserved_ip("64:ff9b:1:ffff::1".parse().unwrap()));
assert!(!is_reserved_ip("64:ff9b:2::1".parse().unwrap()));
}
#[test]
fn rejects_ipv4_compatible_metadata() {
assert!(is_reserved_ip("::a9fe:a9fe".parse().unwrap()));
}
#[test]
fn rejects_ipv6_documentation() {
assert!(is_reserved_ip("2001:db8::1".parse().unwrap()));
}
#[test]
fn allows_ipv4_compatible_public() {
assert!(!is_reserved_ip("::808:808".parse().unwrap()));
}
#[test]
fn allows_public_v4() {
assert!(!is_reserved_ip("8.8.8.8".parse().unwrap()));
assert!(!is_reserved_ip("1.1.1.1".parse().unwrap()));
}
#[test]
fn allows_public_v6() {
assert!(!is_reserved_ip("2606:4700:4700::1111".parse().unwrap()));
}
#[tokio::test]
async fn validate_rejects_ip_literal_loopback() {
let err = validate_public_host("127.0.0.1", 80).await.unwrap_err();
assert!(matches!(err, SeerError::InvalidInput(_)));
}
#[tokio::test]
async fn validate_rejects_ip_literal_metadata() {
let err = validate_public_host("169.254.169.254", 80)
.await
.unwrap_err();
assert!(matches!(err, SeerError::InvalidInput(_)));
}
#[tokio::test]
async fn validate_allows_public_ip_literal() {
validate_public_host("8.8.8.8", 53).await.unwrap();
}
#[tokio::test]
async fn validate_http_url_unbrackets_ipv6_literals() {
let url = Url::parse("https://[::1]/").unwrap();
let err = validate_http_url(&url).await.unwrap_err();
assert!(
matches!(err, SeerError::HttpError(ref s) if s.contains("reserved")),
"got: {err:?}"
);
}
#[tokio::test]
async fn validate_http_url_returns_pinnable_addrs_for_public_literal() {
let url = Url::parse("https://8.8.8.8/").unwrap();
let addrs = validate_http_url(&url).await.unwrap();
assert_eq!(addrs, vec!["8.8.8.8:443".parse().unwrap()]);
}
#[tokio::test]
async fn reserved_resolution_error_does_not_leak_resolved_ip() {
let err = validate_public_host("localhost", 80).await.unwrap_err();
assert!(matches!(err, SeerError::InvalidInput(_)));
let msg = err.to_string();
assert!(
!msg.contains("127.0.0.1"),
"must not leak resolved IPv4: {msg}"
);
assert!(!msg.contains("::1"), "must not leak resolved IPv6: {msg}");
assert!(
!msg.contains("resolves to reserved address"),
"must not echo the resolved-address detail: {msg}"
);
}
#[tokio::test]
#[ignore = "requires network — hits Google DNS via hickory fallback"]
async fn validate_rejects_unresolvable_via_fallback() {
let err = validate_public_host("nonexistent.host.invalid.", 443)
.await
.unwrap_err();
assert!(matches!(err, SeerError::DnsError(_)), "got: {err:?}");
let msg = format!("{err}");
assert!(msg.contains("DNS resolution failed"), "got: {msg}");
assert!(msg.contains("fallback"), "got: {msg}");
assert_eq!(err.sanitized_message(), "DNS resolution failed");
}
}