geiserx_ts_bart 0.21.0

fast, memory-efficient routing table
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137

use core::net::IpAddr;

use crate::{RouteModification, RoutingTable, iptrie, node, table};

/// Routing table that segregates routes into independent IPv4 and IPv6 tables.
///
/// This is likely the kind of table you want unless you know you'll be
/// operating in an exclusively-single-stack environment and want to save the
/// slight runtime overhead of discriminating on address type.
#[derive(Debug, Copy, Clone, Default, PartialEq, Eq, Hash)]
pub struct SplitStackTable<Node> {
    table4: table::SimpleTable<Node>,
    table6: table::SimpleTable<Node>,
}

impl<T, C> SplitStackTable<crate::Node<T, C>>
where
    C: ?Sized + crate::Storage,
{
    /// The empty table.
    pub const EMPTY: Self = Self {
        table4: table::SimpleTable::EMPTY,
        table6: table::SimpleTable::EMPTY,
    };
}

impl<Node> SplitStackTable<Node>
where
    Node: node::StrideOps,
{
    /// Report the total number of IPv4 routes stored in the table.
    #[inline]
    pub fn size4(&self) -> usize {
        self.table4.size()
    }

    /// Report the total number of IPv6 routes stored in the table.
    #[inline]
    pub fn size6(&self) -> usize {
        self.table6.size()
    }

    /// Get a reference to the root node for the given ip stack.
    #[inline]
    pub const fn root(&self, ipv4: bool) -> &Node {
        if ipv4 {
            self.table4.root()
        } else {
            self.table6.root()
        }
    }

    #[inline]
    const fn stack_table(&self, ipv4: bool) -> &table::SimpleTable<Node> {
        if ipv4 { &self.table4 } else { &self.table6 }
    }

    #[inline]
    const fn stack_table_mut(&mut self, ipv4: bool) -> &mut table::SimpleTable<Node> {
        if ipv4 {
            &mut self.table4
        } else {
            &mut self.table6
        }
    }
}

impl<Node> RoutingTable for SplitStackTable<Node>
where
    Node: node::StrideOps,
{
    type Value = Node::T;

    #[inline]
    fn contains(&self, ip: IpAddr) -> bool {
        self.stack_table(ip.is_ipv4()).contains(ip)
    }

    #[inline]
    fn insert(&mut self, prefix: ipnet::IpNet, val: Node::T) -> Option<Node::T> {
        self.stack_table_mut(prefix.addr().is_ipv4())
            .insert(prefix, val)
    }

    #[inline]
    fn remove(&mut self, prefix: ipnet::IpNet) -> Option<Node::T> {
        self.stack_table_mut(prefix.addr().is_ipv4()).remove(prefix)
    }

    #[inline]
    fn modify_impl(
        &mut self,
        prefix: ipnet::IpNet,
        modify: &mut dyn FnMut(Option<&mut Node::T>) -> RouteModification<Node::T>,
    ) -> Option<Node::T> {
        self.stack_table_mut(prefix.addr().is_ipv4())
            .modify_impl(prefix, modify)
    }

    #[inline]
    fn clear(&mut self) {
        self.table4.clear();
        self.table6.clear();
    }

    #[inline]
    fn lookup(&self, ip: IpAddr) -> Option<&Node::T> {
        self.stack_table(ip.is_ipv4()).lookup(ip)
    }

    fn lookup_all(&self, ip: IpAddr) -> iptrie::LookupIter<'_, Self::Value> {
        self.stack_table(ip.is_ipv4()).lookup_all(ip)
    }

    #[inline]
    fn lookup_prefix_exact(&self, prefix: ipnet::IpNet) -> Option<&Node::T> {
        self.stack_table(prefix.addr().is_ipv4())
            .lookup_prefix_exact(prefix)
    }

    #[inline]
    fn lookup_prefix(&self, prefix: ipnet::IpNet) -> Option<&Node::T> {
        self.stack_table(prefix.addr().is_ipv4())
            .lookup_prefix(prefix)
    }

    #[inline]
    fn lookup_prefix_lpm(&self, prefix: ipnet::IpNet) -> Option<(ipnet::IpNet, &Node::T)> {
        self.stack_table(prefix.addr().is_ipv4())
            .lookup_prefix_lpm(prefix)
    }

    #[inline]
    fn size(&self) -> usize {
        self.table4.size() + self.table6.size()
    }
}