use crate::error::*;
use crate::buffer::{self, Buffer};
use crate::builder::{Builder as Build, Finalization};
use crate::ip::{v4, v6};
#[derive(Debug)]
pub struct Builder<B: Buffer = buffer::Dynamic> {
buffer: B,
finalizer: Finalization,
}
impl<B: Buffer> Build<B> for Builder<B> {
fn with(buffer: B) -> Result<Self> {
Ok(Builder {
buffer: buffer,
finalizer: Default::default(),
})
}
fn finalizer(&mut self) -> &mut Finalization {
&mut self.finalizer
}
fn build(self) -> Result<B::Inner> {
Err(Error::InvalidPacket)
}
}
impl Default for Builder<buffer::Dynamic> {
fn default() -> Self {
Builder::with(buffer::Dynamic::default()).unwrap()
}
}
impl<B: Buffer> Builder<B> {
pub fn v4(self) -> Result<v4::Builder<B>> {
let mut v4 = v4::Builder::with(self.buffer)?;
v4.finalizer().extend(self.finalizer);
Ok(v4)
}
pub fn v6(self) -> Result<v6::Builder<B>> {
let mut v6 = v6::Builder::with(self.buffer)?;
v6.finalizer().extend(self.finalizer);
Ok(v6)
}
}