tunneler-core 0.9.2

Provides the Core functionality for the Tunneler software
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

use std::convert::TryInto;

use crate::message::MessageType;

/// The Header of a single Message
#[derive(Debug, PartialEq, Clone)]
pub struct MessageHeader {
    /// The ID of the Connection the assosicated Message belongs to
    pub id: u32, // 4 bytes
    /// The Type of Message
    pub kind: MessageType, // 1 byte
    /// The Length of the Data assosicated with this Message
    pub length: u64, // 8 bytes
}

impl MessageHeader {
    /// Creates a new Header with the given Metadata
    pub fn new(id: u32, kind: MessageType, length: u64) -> MessageHeader {
        MessageHeader { id, kind, length }
    }

    /// Deserializes a 13-Byte array into the fitting Message-Header
    pub fn deserialize(raw_data: &[u8; 13]) -> Option<MessageHeader> {
        let id_part = &raw_data[0..4];
        let kind_part = raw_data[4];
        let length_part = &raw_data[5..13];

        let id = u32::from_le_bytes(id_part.try_into().unwrap());
        let kind = MessageType::deserialize(kind_part);
        kind.as_ref()?;
        let kind = kind.unwrap();
        let length = u64::from_le_bytes(length_part.try_into().unwrap());

        Some(MessageHeader { id, kind, length })
    }

    /// Serializes the Header itself into a 13-Byte array
    pub fn serialize(&self, target: &mut [u8; 13]) {
        let id = self.id.to_le_bytes();
        let length = self.length.to_le_bytes();

        target[0] = id[0];
        target[1] = id[1];
        target[2] = id[2];
        target[3] = id[3];

        target[4] = self.kind.serialize();

        target[5] = length[0];
        target[6] = length[1];
        target[7] = length[2];
        target[8] = length[3];
        target[9] = length[4];
        target[10] = length[5];
        target[11] = length[6];
        target[12] = length[7];
    }

    /// Returns the ID of the Connection this message is meant for
    pub fn get_id(&self) -> u32 {
        self.id
    }
    /// Returns the Type of message
    pub fn get_kind(&self) -> &MessageType {
        &self.kind
    }
    /// Returns the Length of the data that belongs to this message
    pub fn get_length(&self) -> u64 {
        self.length
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn message_header_serialize_connect() {
        let input = vec![13, 0, 0, 0, 1, 20, 0, 0, 0, 0, 0, 0, 0];
        let mut output = [0; 13];
        MessageHeader {
            id: 13,
            kind: MessageType::Connect,
            length: 20,
        }
        .serialize(&mut output);

        assert_eq!(&input[0..13], &output[0..13],);
    }

    #[test]
    fn message_header_deserialize_connect() {
        let mut input = [0; 13];
        input[0] = 13;
        input[4] = 1;
        input[5] = 20;
        assert_eq!(
            Some(MessageHeader {
                id: 13,
                kind: MessageType::Connect,
                length: 20,
            }),
            MessageHeader::deserialize(&input)
        );
    }

    #[test]
    fn serialize_deserialize() {
        let first = MessageHeader::new(123, MessageType::Data, 123);
        let mut serialized = [0; 13];
        first.serialize(&mut serialized);
        let deserialized = MessageHeader::deserialize(&serialized);

        assert_eq!(true, deserialized.is_some());
        assert_eq!(first, deserialized.unwrap());
    }
}