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

use priv_prelude::*;

pub struct ImplNode<N> {
    nat_builder: NatV4Builder,
    node: N,
}

/// Create a node for an Ipv4 NAT.
pub fn nat_v4<N>(nat_builder: NatV4Builder, node: N) -> ImplNode<N>
where
    N: Ipv4Node,
{
    ImplNode {
        nat_builder,
        node,
    }
}

impl<N> Ipv4Node for ImplNode<N>
where
    N: Ipv4Node,
{
    type Output = N::Output;

    fn build(
        self,
        handle: &Handle,
        subnet: SubnetV4,
    ) -> (JoinHandle<N::Output>, Ipv4Plug) {
        let ip = subnet.random_client_addr();

        let private_subnet = {
            self.nat_builder
            .get_subnet()
            .unwrap_or_else(SubnetV4::random_local)
        };
        let nat_builder = self.nat_builder.subnet(private_subnet);
        let (join_handle, client_plug) = self.node.build(handle, private_subnet);
        let (public_plug_0, public_plug_1) = Ipv4Plug::new_wire();
        nat_builder.spawn(handle, public_plug_1, client_plug, ip);
        (join_handle, public_plug_0)
    }
}