proto-packet 0.1.0-rc.13

This library provides the dependencies for generated ProtoPacket code.
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

use crate::io::DecodingError;
use crate::io::DecodingError::InvalidWireType;
use crate::io::WireType::List;
use crate::io::{Decoder, ListHeader, WireType};
use enc::{DecodeFromReadPrefix, read_single_byte};
use std::io::{Read, Take};
use uuid::Uuid;

impl Decoder {
    //! Decode: `Vec<Uuid>`

    /// Decodes a `Vec<Uuid>` from the `Read` prefix with the `first` byte.
    pub fn decode_uuid_slice<R>(
        &self,
        wire: WireType,
        r: &mut R,
        first: u8,
    ) -> Result<Vec<Uuid>, DecodingError>
    where
        R: Read,
    {
        if wire != List {
            return Err(InvalidWireType {
                semantic: "Vec<Uuid>",
                wire,
            });
        }

        let header: ListHeader = ListHeader::decode_from_read_prefix_with_first_byte(r, first)
            .map_err(DecodingError::from_length_prefix_error)?;

        if header.wire() != WireType::Fixed16Byte {
            return Err(InvalidWireType {
                semantic: "Vec<Uuid>",
                wire: header.wire(),
            });
        }

        const _: () = assert!(usize::BITS <= 64);
        let mut r: Take<&mut R> = r.take(header.size() as u64);
        let mut result: Vec<Uuid> = Vec::with_capacity(header.element_capacity_hint());

        while r.limit() > 0 {
            let first: u8 = read_single_byte(&mut r).map_err(DecodingError::from)?;
            let uuid: Uuid = self.decode_uuid(header.wire(), &mut r, first)?;
            result.push(uuid);
        }

        Ok(result)
    }
}

#[cfg(test)]
mod tests {
    use crate::io::WireType::*;
    use crate::io::{Decoder, DecodingError};
    use uuid::Uuid;

    #[test]
    fn decode_uuid_slice() {
        let decoder: Decoder = Decoder::default();
        // ListHeader: wire=Fixed16Byte(4), size=32 -> overflow: 0x9F, VarIntSize(2) = 0x02
        let mut data: Vec<u8> = vec![0x02];
        data.extend_from_slice(&[1; 16]);
        data.extend_from_slice(&[2; 16]);

        let result: Vec<Uuid> = decoder
            .decode_uuid_slice(List, &mut data.as_slice(), 0x9F)
            .unwrap();
        assert_eq!(
            result,
            vec![Uuid::from_bytes([1; 16]), Uuid::from_bytes([2; 16])]
        );
    }

    #[test]
    fn decode_uuid_slice_empty() {
        let decoder: Decoder = Decoder::default();
        // ListHeader: wire=Fixed16Byte(4), size=0 -> 0x80
        let result: Vec<Uuid> = decoder.decode_uuid_slice(List, &mut &[][..], 0x80).unwrap();
        assert!(result.is_empty());
    }

    #[test]
    fn decode_uuid_slice_invalid_wire() {
        let decoder: Decoder = Decoder::default();
        let result: Result<Vec<Uuid>, DecodingError> =
            decoder.decode_uuid_slice(VarInt, &mut &[][..], 0);
        assert!(matches!(
            result,
            Err(DecodingError::InvalidWireType {
                semantic: "Vec<Uuid>",
                wire: VarInt
            })
        ));
    }
}