use super::super::iana::OptionCode;
use super::super::message_builder::OptBuilder;
use super::super::name::{Dname, ToDname};
use super::super::octets::{
Compose, OctetsBuilder, OctetsRef, Parse, ParseError, Parser, ShortBuf
};
use super::CodeOptData;
#[derive(Clone)]
pub struct Chain<Octets> {
start: Dname<Octets>,
}
impl<Octets> Chain<Octets> {
pub fn new(start: Dname<Octets>) -> Self {
Chain { start }
}
pub fn push<Target, N>(
builder: &mut OptBuilder<Target>,
start: &N
) -> Result<(), ShortBuf>
where
Target: OctetsBuilder + AsRef<[u8]> + AsMut<[u8]>,
N: ToDname,
{
builder.push_raw_option(OptionCode::Chain, |target| {
target.append_all(|target| {
for label in start.iter_labels() {
label.compose(target)?
}
Ok(())
})
})
}
pub fn start(&self) -> &Dname<Octets> {
&self.start
}
}
impl<Ref: OctetsRef> Parse<Ref> for Chain<Ref::Range> {
fn parse(parser: &mut Parser<Ref>) -> Result<Self, ParseError> {
Dname::parse(parser).map(Self::new)
}
fn skip(parser: &mut Parser<Ref>) -> Result<(), ParseError> {
Dname::skip(parser)
}
}
impl<Octets: AsRef<[u8]>> Compose for Chain<Octets> {
fn compose<T: OctetsBuilder + AsMut<[u8]>>(
&self,
target: &mut T
) -> Result<(), ShortBuf> {
self.start.compose(target)
}
}
impl<Octets> CodeOptData for Chain<Octets> {
const CODE: OptionCode = OptionCode::Chain;
}