zettabgp 0.4.1

This is a BGP (parsing and composing) and BMP (only parsing) protocols driver library for Rust
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

// Copyright 2021 Vladimir Melnikov.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

//! BMP termination message

use crate::*;

/// BMP termination message
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct BmpMessageTermination {
    /// reason string
    pub str0: Option<String>,
    /// reason code
    pub reason: Option<u16>,
}

impl BmpMessageTermination {
    pub fn new() -> BmpMessageTermination {
        BmpMessageTermination {
            str0: None,
            reason: None,
        }
    }
    pub fn decode_from(buf: &[u8]) -> Result<(BmpMessageTermination, usize), BgpError> {
        let mut pos: usize = 0;
        let mut ret: BmpMessageTermination = BmpMessageTermination::new();
        while buf.len() - pos >= 4 {
            let infotype = getn_u16(&buf[pos..]);
            let infolen = getn_u16(&buf[pos + 2..]) as usize;
            pos += 4;
            if buf.len() - pos < infolen {
                return Err(BgpError::InsufficientBufferSize(file!(), line!()));
            }
            match infotype {
                0 => ret.str0 = Some(core::str::from_utf8(&buf[pos..pos + infolen])?.to_string()),
                1 => ret.reason = Some(getn_u16(&buf[pos..])),
                _ => {}
            }
            pos += infolen;
        }
        Ok((ret, pos))
    }
    pub fn encode_to(&self, buf: &mut [u8]) -> Result<usize, BgpError> {
        let mut curpos: usize = 0;
        if let Some(str0) = &self.str0 {
            let str0 = str0.as_bytes();
            if buf.len() - curpos < 4 + str0.len() {
                return Err(BgpError::InsufficientBufferSize(file!(), line!()));
            }
            setn_u16(0, &mut buf[curpos..]);
            curpos += 2;
            setn_u16(str0.len() as u16, &mut buf[curpos..]);
            curpos += 2;
            buf[curpos..curpos + str0.len()].copy_from_slice(str0);
            curpos += str0.len();
        }
        if let Some(reason) = self.reason {
            if buf.len() - curpos < 6 {
                return Err(BgpError::InsufficientBufferSize(file!(), line!()));
            }
            setn_u16(1, &mut buf[curpos..]);
            curpos += 2;
            setn_u16(2, &mut buf[curpos..]);
            curpos += 2;
            setn_u16(reason, &mut buf[curpos..]);
            curpos += 2;
        }
        Ok(curpos)
    }
}
impl Default for BmpMessageTermination {
    fn default() -> Self {
        Self::new()
    }
}