rust-dnsbox/lib/dnsbox-base/src/common_types/uri.rs

use crate::errors::*;
use crate::ser::packet::{remaining_bytes, DnsPacketData, DnsPacketWriteContext};
use crate::ser::text::*;
use bytes::{BufMut, Bytes};
use std::fmt;
use std::io::Cursor;

/// A single quoted non-empty URL.
///
/// Actually shouldn't allow escapes (i.e. no backslash in the content);
/// but to make sure we can export and import any data we allow standard
/// escape mechanisms and even accept unquoted input.
///
/// No whitespace allowed, last field.
#[derive(Clone, PartialEq, Eq, Debug)]
pub struct UriText(Bytes);

impl DnsPacketData for UriText {
	fn deserialize(data: &mut Cursor<Bytes>) -> Result<Self> {
		let raw = remaining_bytes(data);
		failure::ensure!(!raw.is_empty(), "URI must not be empty");
		Ok(UriText(raw))
	}

	fn serialize(&self, _context: &mut DnsPacketWriteContext, packet: &mut Vec<u8>) -> Result<()> {
		failure::ensure!(!self.0.is_empty(), "URI must not be empty");
		packet.reserve(self.0.len());
		packet.put_slice(&self.0);
		Ok(())
	}
}

impl DnsTextData for UriText {
	fn dns_parse(_context: &DnsTextContext, data: &mut &str) -> crate::errors::Result<Self> {
		let raw = next_quoted_field(data)?;
		failure::ensure!(!raw.is_empty(), "URI must not be empty");
		Ok(UriText(raw.into()))
	}

	fn dns_format(&self, f: &mut DnsTextFormatter) -> fmt::Result {
		write!(f, "{}", quote(&self.0))
	}
}