tnnl 0.1.17

tnnl gives you full control over whether and when your IoT devices should be reachable from the internet
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

use std::collections::HashMap;

use tokio::sync::mpsc::{self, Sender};

use crate::{
    channel::Channel,
    command_channel::{COMMAND_CHANNEL_ID, CommandChannel},
    model::packet::Packet,
    rules::Rule,
    tcp_channel::TcpChannel,
};

pub struct ChannelHandler {
    channel_senders: HashMap<i32, Sender<Vec<u8>>>,
    to_server_tx: Sender<Vec<u8>>,

    last_channel: i32,
    rule: Rule,
}

impl ChannelHandler {
    pub fn new(to_server_tx: Sender<Vec<u8>>, rule: Rule) -> (ChannelHandler, CommandChannel) {
        let mut channel_senders = HashMap::new();

        let (from_server_tx, from_server_rx) = mpsc::channel::<Vec<u8>>(8);

        let command_channel =
            Channel::new(to_server_tx.clone(), from_server_rx, COMMAND_CHANNEL_ID);
        let command_channel = CommandChannel::new(command_channel);

        channel_senders.insert(COMMAND_CHANNEL_ID, from_server_tx);

        (
            ChannelHandler {
                to_server_tx,
                last_channel: 0,

                channel_senders,
                rule,
            },
            command_channel,
        )
    }

    pub async fn route(&mut self, packet: Packet) {
        // cleanup old closed channels
        self.channel_senders
            .retain(|_channel_id, channel| !channel.is_closed());

        let channel = self.channel_senders.get(&packet.channel_id);

        match channel {
            Some(channel) => {
                if packet.data.is_empty() {
                    self.channel_senders.remove(&packet.channel_id);
                    return;
                }

                match channel.send(packet.data).await {
                    Ok(_) => {}
                    Err(_) => {
                        log::warn!("failed to post data to receive channel")
                    }
                }
            }

            None => {
                if packet.channel_id <= self.last_channel {
                    log::warn!(
                        "dropping packet for already closed channel {:}",
                        packet.channel_id
                    );
                    return;
                }
                self.last_channel = packet.channel_id;

                log::trace!("new connection for {:}", packet.channel_id);

                let (from_server_tx, from_server_rx) = mpsc::channel::<Vec<u8>>(8);

                let channel =
                    Channel::new(self.to_server_tx.clone(), from_server_rx, packet.channel_id);
                self.channel_senders
                    .insert(packet.channel_id, from_server_tx);

                let tcp_channel = TcpChannel::new(channel);
                tcp_channel.spawn(packet, &self.rule);
            }
        }
    }
}