scion-stack 0.7.0

// Copyright 2026 Anapaya Systems
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//! TXT-based SCION address resolution (TSAR).
//!
//! TSAR encodes SCION addresses in DNS TXT records to support dual-stack
//! resolution. The record format is defined as:
//!
//! ```text
//! scion-txt     = "scion=" version separator address-list
//! version       = "v1"          ; Versioning for future extensibility
//! separator     = ";"
//! address-list  = address *( "," address )
//! address       = "[" isd-as "," host "]"
//! isd-as        = 1*DIGIT "-" 1*HEXDIG ":" 1*HEXDIG ":" 1*HEXDIG
//! host          = ipv4-address / ipv6-address
//! ipv4-address  = 1*3DIGIT "." 1*3DIGIT "." 1*3DIGIT "." 1*3DIGIT
//! ipv6-address  = <RFC5952 compliant string>
//! ```
//!
//! Example records:
//!
//! ```text
//! example.com. IN TXT "scion=v1;[19-ff00:0:110,192.0.2.1]"
//! example.com. IN TXT "scion=v1;[19-ff00:0:110,2001:db8::1]"
//! example.com. IN TXT "scion=v1;[19-ff00:0:110,192.0.2.1],[19-ff00:0:111,203.0.113.5]"
//! ```

use std::{collections::HashMap, net::IpAddr, str::FromStr};

use async_trait::async_trait;
use hickory_resolver::{
    ResolveErrorKind, ResolverBuilder, TokioResolver, name_server::TokioConnectionProvider,
    proto::rr::rdata::TXT,
};
use sciparse::{address::ip_addr::ScionIpAddr, identifier::isd_asn::IsdAsn};
use thiserror::Error;

use super::{InvalidEntry, ResolveError, ScionDnsResolver};

const SCION_TXT_PREFIX: &str = "scion=v1;";

/// Resolver that interprets TXT records using the TSAR format.
///
/// Use this resolver to look up `scion=v1;...` TXT records and translate them into `ScionIpAddr`
/// values. Construction errors are reported via `TxtResolverError`, while lookup failures and
/// parsing outcomes are reported through `ResolveError` from `ScionDnsResolver::resolve`.
///
/// Domain-specific overrides can be added to bypass DNS and always return the configured addresses.
///
/// # Caching
///
/// This type holds no cache of its own. Lookups are cached by the DNS core underneath, at the
/// records' own TTLs, and that cache covers negative answers too: a name without TXT records is
/// remembered for the negative TTL of the zone's SOA. Clones share that cache; dropping the last
/// clone drops the cache with it.
///
/// Callers who need different bounds, or no caching at all, set `ResolverOpts` (`cache_size`,
/// `positive_min_ttl`, `positive_max_ttl`, `negative_min_ttl`, `negative_max_ttl`) on the builder
/// returned by [`ScionTxtDnsResolver::builder`] before passing it to
/// [`ScionTxtDnsResolver::from_builder`]. Bounding staleness means bounding it there; a cache added
/// above this type could not do so, because on expiry it would only re-read the copy the DNS core
/// still holds.
#[derive(Clone, Debug)]
pub struct ScionTxtDnsResolver {
    resolver: TokioResolver,
    overrides: HashMap<String, Vec<ScionIpAddr>>,
}

impl ScionTxtDnsResolver {
    /// Create a resolver using the system DNS configuration.
    ///
    /// This uses the OS resolver configuration (for example `/etc/resolv.conf`)
    /// and then applies the default hickory-dns options for lookups.
    ///
    /// # Errors
    ///
    /// Returns `TxtResolverError` if the system configuration cannot be loaded.
    pub fn new() -> Result<Self, TxtResolverError> {
        let builder = Self::builder()?;
        Self::from_builder(builder)
    }

    /// Override DNS resolution for a specific domain.
    #[must_use]
    pub fn with_override(self, domain: &str, addrs: Vec<ScionIpAddr>) -> Self {
        self.with_overrides(vec![(domain, addrs)])
    }

    /// Override DNS resolution for multiple domains at once.
    #[must_use]
    pub fn with_overrides<D, I>(mut self, overrides: I) -> Self
    where
        D: Into<String>,
        I: IntoIterator<Item = (D, Vec<ScionIpAddr>)>,
    {
        for (domain, addrs) in overrides {
            self.overrides.insert(domain.into(), addrs);
        }
        self
    }

    /// Constructs a resolver from a pre-configured hickory `ResolverBuilder`.
    ///
    /// This allows callers to customize resolver options (timeouts, retries, name servers) via
    /// hickory-dns before constructing the resolver.
    ///
    /// # Errors
    ///
    /// This function is currently infallible, but returns `Result` for future compatibility with
    /// hickory-dns builder changes.
    // Intentionally exposes hickory-dns's builder type as the escape hatch for full control over
    // DNS resolution. See API_CONVENTIONS.md.
    pub fn from_builder(
        builder: ResolverBuilder<TokioConnectionProvider>,
    ) -> Result<Self, TxtResolverError> {
        Ok(Self {
            resolver: builder.build(),
            overrides: HashMap::new(),
        })
    }

    /// Creates a builder for configuring resolver options.
    ///
    /// On Linux/macOS the builder is initialized from the system DNS
    /// configuration (`/etc/resolv.conf`). On Android and iOS, which do not
    /// expose `/etc/resolv.conf`, Google Public DNS is used as a fallback.
    ///
    /// The returned builder can be adjusted before calling
    /// [`ScionTxtDnsResolver::from_builder`].
    ///
    /// # Errors
    ///
    /// Returns [`TxtResolverError`] if system configuration cannot be loaded (non-Android/iOS
    /// platforms only).
    pub fn builder() -> Result<ResolverBuilder<TokioConnectionProvider>, TxtResolverError> {
        #[cfg(any(target_os = "android", target_os = "ios"))]
        {
            use hickory_resolver::config::ResolverConfig;
            // Android and iOS do not have /etc/resolv.conf.
            // Fall back to Google Public DNS for SCION TXT record resolution.
            Ok(TokioResolver::builder_with_config(
                ResolverConfig::google(),
                TokioConnectionProvider::default(),
            ))
        }
        #[cfg(not(any(target_os = "android", target_os = "ios")))]
        {
            Ok(TokioResolver::builder_tokio()?)
        }
    }
}

#[async_trait]
impl ScionDnsResolver for ScionTxtDnsResolver {
    async fn resolve(&self, domain: &str) -> Result<Vec<ScionIpAddr>, ResolveError> {
        if let Some(addrs) = self.overrides.get(domain) {
            return Ok(addrs.clone());
        }

        let lookup = self
            .resolver
            .txt_lookup(domain)
            .await
            .map_err(|err| classify_lookup_error(domain, &err))?;

        let mut txt_records = Vec::new();
        let mut invalid_entries = Vec::new();
        for txt in lookup.iter() {
            match txt_record_to_string(txt) {
                Ok(txt_record) => txt_records.push(txt_record),
                Err(err) => invalid_entries.push(err),
            }
        }

        resolve_txt_records_with_invalid(domain, txt_records, invalid_entries)
    }
}

/// Errors returned while constructing a TXT resolver.
///
/// The underlying cause is available through [`std::error::Error::source`]; the concrete source
/// type is intentionally not exposed, so the DNS backend can change without breaking the public
/// API.
#[derive(Debug, Error)]
#[non_exhaustive]
pub enum TxtResolverError {
    /// DNS resolver configuration failed.
    #[error("dns resolver configuration failed: {message}")]
    DnsConfig {
        /// Human-readable description of the configuration failure.
        message: String,
        /// The underlying cause.
        #[source]
        source: Box<dyn std::error::Error + Send + Sync>,
    },
}

impl From<hickory_resolver::ResolveError> for TxtResolverError {
    fn from(error: hickory_resolver::ResolveError) -> Self {
        Self::DnsConfig {
            message: error.to_string(),
            source: Box::new(error),
        }
    }
}

impl PartialEq for TxtResolverError {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (Self::DnsConfig { message: a, .. }, Self::DnsConfig { message: b, .. }) => a == b,
        }
    }
}

/// Splits a hickory lookup failure into the two outcomes of the [`ScionDnsResolver`] contract: a
/// lookup that completed but found no TXT records is a permanent [`ResolveError::NoValidEntries`]
/// verdict, every other failure (timeout, network error, SERVFAIL) is a transient
/// [`ResolveError::DnsLookup`] that a retry may resolve.
///
/// Hickory reports both an empty answer and a non-existent name as `NoRecordsFound`. Both are
/// permanent for as long as the negative answer is valid, so the distinction between "no such name"
/// and "the name exists without TXT records" is not carried over. Failure kinds we do not recognize
/// stay transient, which is the safe default for a retry decision.
fn classify_lookup_error(domain: &str, err: &hickory_resolver::ResolveError) -> ResolveError {
    match err.kind() {
        ResolveErrorKind::Proto(proto) if proto.is_no_records_found() => {
            ResolveError::NoValidEntries {
                domain: domain.to_string(),
                invalid_entries: Vec::new(),
            }
        }
        _ => ResolveError::DnsLookup(err.to_string()),
    }
}

#[derive(Debug, Error)]
enum TxtParseError {
    #[error("missing TXT address list")]
    MissingAddressList,
    #[error("expected '[' at: {0}")]
    ExpectedOpenBracket(String),
    #[error("missing closing ']' in: {0}")]
    MissingCloseBracket(String),
    #[error("expected comma separator in: {0}")]
    MissingSeparator(String),
    #[error("invalid ISD-AS: {0}")]
    InvalidIsdAsn(#[from] sciparse::address::AddressParseError),
    #[error("invalid host address: {0}")]
    InvalidHost(#[from] std::net::AddrParseError),
    #[error("expected ',' after entry in: {0}")]
    ExpectedComma(String),
}

#[cfg(test)]
fn resolve_txt_records(
    domain: &str,
    records: impl IntoIterator<Item = String>,
) -> Result<Vec<ScionIpAddr>, ResolveError> {
    resolve_txt_records_with_invalid(domain, records, Vec::new())
}

fn resolve_txt_records_with_invalid(
    domain: &str,
    records: impl IntoIterator<Item = String>,
    mut invalid: Vec<InvalidEntry>,
) -> Result<Vec<ScionIpAddr>, ResolveError> {
    let mut valid = Vec::new();

    for record in records {
        let Some(payload) = record.strip_prefix(SCION_TXT_PREFIX) else {
            continue;
        };

        match parse_txt_payload(payload) {
            Ok(mut addresses) => valid.append(&mut addresses),
            Err(err) => invalid.push(InvalidEntry::new(record, err.to_string())),
        }
    }

    if valid.is_empty() {
        return Err(ResolveError::NoValidEntries {
            domain: domain.to_string(),
            invalid_entries: invalid,
        });
    }

    if !invalid.is_empty() {
        let details = format_invalid_entries(&invalid);
        tracing::info!(
            domain,
            invalid_entries = invalid.len(),
            details = ?details,
            "Ignoring invalid SCION TXT entries"
        );
    }

    Ok(valid)
}

fn parse_txt_payload(payload: &str) -> Result<Vec<ScionIpAddr>, TxtParseError> {
    let mut remaining = payload.trim();
    if remaining.is_empty() {
        return Err(TxtParseError::MissingAddressList);
    }

    let mut addresses = Vec::new();
    while !remaining.is_empty() {
        if !remaining.starts_with('[') {
            return Err(TxtParseError::ExpectedOpenBracket(remaining.to_string()));
        }

        let close_idx = remaining
            .find(']')
            .ok_or_else(|| TxtParseError::MissingCloseBracket(remaining.to_string()))?;
        let entry = remaining[1..close_idx].trim();
        let rest = remaining[close_idx + 1..].trim();

        let (isd_asn_str, host_str) = entry
            .split_once(',')
            .ok_or_else(|| TxtParseError::MissingSeparator(entry.to_string()))?;

        let isd_asn = IsdAsn::from_str(isd_asn_str.trim())?;
        let host = IpAddr::from_str(host_str.trim())?;

        addresses.push(ScionIpAddr::new(isd_asn, host));

        if rest.is_empty() {
            break;
        }

        if !rest.starts_with(',') {
            return Err(TxtParseError::ExpectedComma(rest.to_string()));
        }

        remaining = rest[1..].trim();
    }

    Ok(addresses)
}

fn txt_record_to_string(txt: &TXT) -> Result<String, InvalidEntry> {
    let bytes: Vec<u8> = txt
        .txt_data()
        .iter()
        .flat_map(|chunk| chunk.iter())
        .copied()
        .collect();

    String::from_utf8(bytes)
        .map_err(|_| InvalidEntry::new("<invalid-utf8>", "TXT entry is not valid UTF-8"))
}

fn format_invalid_entries(entries: &[InvalidEntry]) -> Vec<String> {
    entries
        .iter()
        .map(|entry| format!("{} ({})", entry.raw(), entry.reason()))
        .collect()
}

#[cfg(test)]
mod tests {
    use hickory_resolver::proto::{
        ProtoError, ProtoErrorKind,
        op::{Query, ResponseCode},
        rr::{Name, RecordType},
    };

    use super::*;

    /// Builds the failure hickory reports for a TXT query that found no records, which is also how
    /// it reports a name that does not exist (`response_code` tells the two apart).
    fn no_records_error(
        domain: &str,
        response_code: ResponseCode,
    ) -> hickory_resolver::ResolveError {
        let query = Query::query(Name::from_str(domain).expect("valid name"), RecordType::TXT);
        ResolveErrorKind::Proto(ProtoError::nx_error(
            Box::new(query),
            None,
            None,
            None,
            response_code,
            false,
            None,
        ))
        .into()
    }

    #[test]
    fn classify_empty_answer_as_no_valid_entries() {
        let raw = no_records_error("example.com.", ResponseCode::NoError);

        let err = classify_lookup_error("example.com", &raw);

        assert_eq!(
            err,
            ResolveError::NoValidEntries {
                domain: "example.com".to_string(),
                invalid_entries: Vec::new(),
            }
        );
        assert!(!err.is_transient());
    }

    #[test]
    fn classify_nonexistent_name_as_no_valid_entries() {
        let raw = no_records_error("example.com.", ResponseCode::NXDomain);

        let err = classify_lookup_error("example.com", &raw);

        assert_eq!(
            err,
            ResolveError::NoValidEntries {
                domain: "example.com".to_string(),
                invalid_entries: Vec::new(),
            }
        );
        assert!(!err.is_transient());
    }

    #[test]
    fn classify_timeout_as_transient_lookup_failure() {
        let raw: hickory_resolver::ResolveError =
            ResolveErrorKind::Proto(ProtoError::from(ProtoErrorKind::Timeout)).into();

        let err = classify_lookup_error("example.com", &raw);

        assert!(matches!(err, ResolveError::DnsLookup(_)), "{err:?}");
        assert!(err.is_transient());
    }

    #[test]
    fn parse_txt_payload_single() {
        let addrs = parse_txt_payload("[19-ff00:0:110,192.0.2.1]").expect("valid payload");
        assert_eq!(addrs.len(), 1);
        assert_eq!(
            addrs[0],
            ScionIpAddr::from_str("19-ff00:0:110,192.0.2.1").unwrap()
        );
    }

    #[test]
    fn parse_txt_payload_multiple() {
        let addrs = parse_txt_payload("[19-ff00:0:110,192.0.2.1],[19-ff00:0:111,2001:db8::1]")
            .expect("valid payload");
        assert_eq!(addrs.len(), 2);
    }

    #[test]
    fn resolve_txt_records_mixed_validity() {
        let records = vec![
            "scion=v1;[19-ff00:0:110,192.0.2.1]".to_string(),
            "scion=v1;[bad,192.0.2.2]".to_string(),
        ];

        let resolved = resolve_txt_records("example.com", records).expect("valid addresses");
        assert_eq!(resolved.len(), 1);
    }

    #[test]
    fn resolve_txt_records_no_valid_entries() {
        let records = vec!["scion=v1;[bad,192.0.2.2]".to_string()];

        let err = resolve_txt_records("example.com", records).expect_err("no valid entries");
        match err {
            ResolveError::NoValidEntries { domain, .. } => {
                assert_eq!(domain, "example.com");
            }
            other => panic!("unexpected error: {other:?}"),
        }
    }

    #[test]
    fn parse_txt_payload_allows_whitespace_between_entries() {
        let addrs = parse_txt_payload("[19-ff00:0:110,192.0.2.1] , [19-ff00:0:111,2001:db8::1]")
            .expect("valid payload");
        assert_eq!(addrs.len(), 2);
    }

    #[tokio::test]
    async fn with_override_returns_single_address() {
        let addr = ScionIpAddr::from_str("19-ff00:0:110,192.0.2.1").unwrap();
        let resolver = ScionTxtDnsResolver::new()
            .unwrap()
            .with_override("example.com", vec![addr]);

        let result = ScionDnsResolver::resolve(&resolver, "example.com")
            .await
            .unwrap();
        assert_eq!(result, vec![addr]);
    }

    #[tokio::test]
    async fn with_overrides_returns_all_addresses() {
        let addr1 = ScionIpAddr::from_str("19-ff00:0:110,192.0.2.1").unwrap();
        let addr2 = ScionIpAddr::from_str("19-ff00:0:111,2001:db8::1").unwrap();
        let resolver = ScionTxtDnsResolver::new()
            .unwrap()
            .with_override("example.com", vec![addr1, addr2]);

        let result = ScionDnsResolver::resolve(&resolver, "example.com")
            .await
            .unwrap();
        assert_eq!(result, vec![addr1, addr2]);
    }

    #[tokio::test]
    async fn with_multi_overrides_handles_multiple_domains() {
        let addr1 = ScionIpAddr::from_str("19-ff00:0:110,192.0.2.1").unwrap();
        let addr2 = ScionIpAddr::from_str("19-ff00:0:111,192.0.2.2").unwrap();
        let resolver = ScionTxtDnsResolver::new().unwrap().with_overrides([
            ("first.example.com", vec![addr1]),
            ("second.example.com", vec![addr2]),
        ]);

        let result1 = ScionDnsResolver::resolve(&resolver, "first.example.com")
            .await
            .unwrap();
        assert_eq!(result1, vec![addr1]);

        let result2 = ScionDnsResolver::resolve(&resolver, "second.example.com")
            .await
            .unwrap();
        assert_eq!(result2, vec![addr2]);
    }

    #[tokio::test]
    async fn with_override_later_call_replaces_previous() {
        let addr1 = ScionIpAddr::from_str("19-ff00:0:110,192.0.2.1").unwrap();
        let addr2 = ScionIpAddr::from_str("19-ff00:0:111,192.0.2.2").unwrap();
        let resolver = ScionTxtDnsResolver::new()
            .unwrap()
            .with_override("example.com", vec![addr1])
            .with_override("example.com", vec![addr2]);

        let result = ScionDnsResolver::resolve(&resolver, "example.com")
            .await
            .unwrap();
        assert_eq!(result, vec![addr2]);
    }
}