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

pub mod arp;
pub mod ipv4;
pub mod ipv6;

pub use arp::Arp as Arp;
pub use ipv4::IPv4 as IPv4;
pub use ipv6::IPv6 as IPv6;

use log::*;

///
/// Available layer 3 representations
///
#[derive(Clone, Copy, Debug)]
pub enum Layer3<'a> {
    Arp(Arp),
    IPv4(IPv4<'a>),
    IPv6(IPv6<'a>),
    //Lldp(lldp::Lldp)
}

///
/// IP Protocol numbers https://en.wikipedia.org/wiki/List_of_IP_protocol_numbers
///
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum InternetProtocolId {
    AuthenticationHeader,
    HopByHop,
    EncapsulatingSecurityPayload,
    ICMP,
    IPv6Route,
    IPv6Fragment,
    IPv6NoNext,
    IPv6Options,
    Tcp,
    Udp,
}

impl InternetProtocolId {
    pub fn value(&self) -> u8 {
        match self {
            InternetProtocolId::AuthenticationHeader => 50,
            InternetProtocolId::HopByHop => 0,
            InternetProtocolId::EncapsulatingSecurityPayload => 51,
            InternetProtocolId::ICMP => 1,
            InternetProtocolId::IPv6Route => 43,
            InternetProtocolId::IPv6Fragment => 44,
            InternetProtocolId::IPv6NoNext => 59,
            InternetProtocolId::IPv6Options => 60,
            InternetProtocolId::Tcp => 6,
            InternetProtocolId::Udp => 17,
        }
    }
    pub fn new(value: u8) -> Option<InternetProtocolId> {
        match value {
            0 => Some(InternetProtocolId::HopByHop),
            1 => Some(InternetProtocolId::ICMP),
            6 => Some(InternetProtocolId::Tcp),
            17 => Some(InternetProtocolId::Udp),
            43 => Some(InternetProtocolId::IPv6Route),
            44 => Some(InternetProtocolId::IPv6Fragment),
            50 => Some(InternetProtocolId::AuthenticationHeader),
            51 => Some(InternetProtocolId::EncapsulatingSecurityPayload),
            59 => Some(InternetProtocolId::IPv6NoNext),
            60 => Some(InternetProtocolId::IPv6Options),
            _ => {
                //TODO: change to warn once list is more complete
                debug!("Encountered {:02x} when parsing layer 4 id", value);
                None
            }
        }
    }

    pub fn has_next_option(v: InternetProtocolId) -> bool {
        match v {
            InternetProtocolId::AuthenticationHeader => true,
            InternetProtocolId::EncapsulatingSecurityPayload => true,
            InternetProtocolId::HopByHop => true,
            InternetProtocolId::IPv6Route => true,
            InternetProtocolId::IPv6Fragment => true,
            InternetProtocolId::IPv6Options => true,
            _ => false,
        }
    }
}