routecore 0.7.1

A Library with Building Blocks for BGP Routing
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use crate::bgp::message::{Header, MsgType};
use crate::util::parser::ParseError;
use octseq::{FreezeBuilder, Octets, OctetsBuilder, Parser, ShortBuf, Truncate};

/// BGP Keepalive message, variant of the [`Message`] enum.
#[derive(Clone, Debug)]
pub struct KeepaliveMessage<Octets> {
    octets: Octets
}

impl<Octs: Octets> KeepaliveMessage<Octs> {
    pub fn from_octets(octets: Octs) -> Result<Self, ParseError> {
        Self::check(&octets)?;
        Ok(KeepaliveMessage { octets })
    }

    pub fn check(octets: &Octs) -> Result<(), ParseError> {
        let mut parser = Parser::from_ref(octets);
        Header::check(&mut parser)?;
        if parser.remaining() > 0 {
            return Err(ParseError::form_error("KEEPALIVE of >19 bytes"));
        }
        Ok(())
    }
}

impl<Octs: Octets> KeepaliveMessage<Octs> {
    pub fn octets(&self) -> &Octs {
            &self.octets
    }
}

impl<Octs: Octets> AsRef<[u8]> for KeepaliveMessage<Octs> {
    fn as_ref(&self) -> &[u8] {
        self.octets.as_ref()
    }
}


//--- Builder ----------------------------------------------------------------

#[derive(Clone, Debug)]
pub struct KeepaliveBuilder<Target> {
    target: Target,
}

use core::convert::Infallible;
impl<Target: OctetsBuilder + Truncate> KeepaliveBuilder<Target>
where
    Infallible: From<<Target as OctetsBuilder>::AppendError>
{
    pub fn from_target(mut target: Target) -> Result<Self, ShortBuf> {
        target.truncate(0);
        let mut h = Header::<&[u8]>::new();
        h.set_type(MsgType::Keepalive);
        let _ = target.append_slice(h.as_ref());
        Ok(KeepaliveBuilder { target })
    }
}

impl<Target: OctetsBuilder> KeepaliveBuilder<Target> {
    pub fn finish(self) -> Target {
        self.target
    }

    pub fn into_message(
        self
    ) -> KeepaliveMessage<<Target as FreezeBuilder>::Octets>
    where Target: FreezeBuilder {
        KeepaliveMessage{ octets: self.finish().freeze() }
    }
}
impl KeepaliveBuilder<Vec<u8>> {
    pub fn new_vec() -> Self {
        Self::from_target(Vec::new()).unwrap()
    }
}

//--- Tests ------------------------------------------------------------------

#[cfg(test)]
mod tests {

    use super::*;

    mod builder {
        use super::*;

        #[test]
        fn builder() {
            let kap = KeepaliveBuilder::new_vec();
            let res = kap.into_message();
            println!("{:?}", res);
        }
    }

    
}