ic-query 0.59.0

#[must_use]
pub fn hex_bytes(bytes: &[u8]) -> String {
    const HEX: &[u8; 16] = b"0123456789abcdef";
    let mut out = String::with_capacity(bytes.len() * 2);
    for byte in bytes {
        out.push(HEX[usize::from(byte >> 4)] as char);
        out.push(HEX[usize::from(byte & 0x0f)] as char);
    }
    out
}

#[must_use]
pub fn is_lowercase_hex(value: &str) -> bool {
    value
        .bytes()
        .all(|byte| byte.is_ascii_digit() || matches!(byte, b'a'..=b'f'))
}

#[must_use]
pub fn is_canonical_lowercase_hex(value: &str) -> bool {
    !value.is_empty() && value.len().is_multiple_of(2) && is_lowercase_hex(value)
}

///
/// decode_lowercase_hex
///
/// Decode even-length lowercase hexadecimal, including an empty byte string.
///

#[cfg(any(test, feature = "sns-host", feature = "subnet-catalog-host"))]
pub fn decode_lowercase_hex(value: &str) -> Option<Vec<u8>> {
    is_lowercase_hex(value).then(|| decode_hex(value)).flatten()
}

/// Decode even-length ASCII hexadecimal with either letter case.
#[cfg(any(
    test,
    feature = "icrc-host",
    feature = "sns-host",
    feature = "subnet-catalog-host"
))]
pub fn decode_hex(value: &str) -> Option<Vec<u8>> {
    if !value.len().is_multiple_of(2) || !value.bytes().all(|byte| byte.is_ascii_hexdigit()) {
        return None;
    }
    (0..value.len())
        .step_by(2)
        .map(|index| u8::from_str_radix(&value[index..index + 2], 16).ok())
        .collect()
}

///
/// HexEvidenceError
///
/// Shape failure for nonempty, byte-counted lowercase hexadecimal evidence.
///

#[cfg(any(feature = "cmc-host", feature = "certified-subnet-catalog-host"))]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum HexEvidenceError {
    Empty,
    ByteCountOverflow,
    LengthMismatch,
    NonLowercaseHex,
}

/// Admit encoded evidence without decoding or allocating its bytes.
#[cfg(any(feature = "cmc-host", feature = "certified-subnet-catalog-host"))]
pub fn validate_nonempty_hex_evidence(
    value: &str,
    byte_count: usize,
) -> Result<(), HexEvidenceError> {
    if byte_count == 0 {
        return Err(HexEvidenceError::Empty);
    }
    let expected_length = byte_count
        .checked_mul(2)
        .ok_or(HexEvidenceError::ByteCountOverflow)?;
    if value.len() != expected_length {
        return Err(HexEvidenceError::LengthMismatch);
    }
    if !is_lowercase_hex(value) {
        return Err(HexEvidenceError::NonLowercaseHex);
    }
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    #[cfg(any(feature = "cmc-host", feature = "certified-subnet-catalog-host"))]
    use HexEvidenceError::{ByteCountOverflow, Empty, LengthMismatch, NonLowercaseHex};

    #[test]
    #[cfg(any(feature = "cmc-host", feature = "certified-subnet-catalog-host"))]
    fn encoded_evidence_requires_consistent_nonempty_bytes() {
        for (value, bytes, expected) in [
            ("", 0, Err(Empty)),
            ("zz", 0, Err(Empty)),
            ("", usize::MAX, Err(ByteCountOverflow)),
            ("", 1, Err(LengthMismatch)),
            ("0", 1, Err(LengthMismatch)),
            ("0A", 1, Err(NonLowercaseHex)),
            ("🦀", 2, Err(NonLowercaseHex)),
            ("ab", 1, Ok(())),
            ("00ff", 2, Ok(())),
        ] {
            assert_eq!(validate_nonempty_hex_evidence(value, bytes), expected);
        }
    }

    #[test]
    fn lowercase_hex_predicates_distinguish_alphabet_and_canonical_length() {
        for (value, lowercase, canonical) in [
            ("", true, false),
            ("0", true, false),
            ("00", true, true),
            ("0a", true, true),
            ("0A", false, false),
            ("0g", false, false),
            ("é", false, false),
        ] {
            assert_eq!(is_lowercase_hex(value), lowercase, "{value:?}");
            assert_eq!(is_canonical_lowercase_hex(value), canonical, "{value:?}");
        }
    }

    #[test]
    fn lowercase_hex_decoder_accepts_complete_bytes_only() {
        assert_eq!(decode_lowercase_hex(""), Some(Vec::new()));
        assert_eq!(decode_lowercase_hex("000aff"), Some(vec![0, 10, 255]));
        assert_eq!(decode_lowercase_hex("0"), None);
        assert_eq!(decode_lowercase_hex("0A"), None);
        assert_eq!(decode_lowercase_hex("0g"), None);
        assert_eq!(decode_lowercase_hex("aé0"), None);
        assert_eq!(decode_lowercase_hex("🦀"), None);
        assert_eq!(decode_lowercase_hex("+a"), None);
        assert_eq!(decode_lowercase_hex(" 0a"), None);
    }
}