44 lines
1.3 KiB
Rust
44 lines
1.3 KiB
Rust
use bytes::{Bytes, BufMut};
|
|
use errors::*;
|
|
use ser::packet::{DnsPacketData, DnsPacketWriteContext, remaining_bytes};
|
|
use ser::text::*;
|
|
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);
|
|
ensure!(!raw.is_empty(), "URI must not be empty");
|
|
Ok(UriText(raw))
|
|
}
|
|
|
|
fn serialize(&self, _context: &mut DnsPacketWriteContext, packet: &mut Vec<u8>) -> Result<()> {
|
|
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) -> ::errors::Result<Self> {
|
|
let raw = next_quoted_field(data)?;
|
|
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))
|
|
}
|
|
}
|