ironrdp_pdu/rdp/capability_sets/
multifragment_update.rs

1use ironrdp_core::{ensure_fixed_part_size, Decode, DecodeResult, Encode, EncodeResult, ReadCursor, WriteCursor};
2
3#[derive(Debug, PartialEq, Eq, Clone)]
4pub struct MultifragmentUpdate {
5    pub max_request_size: u32,
6}
7
8impl MultifragmentUpdate {
9    const NAME: &'static str = "MultifragmentUpdate";
10
11    const FIXED_PART_SIZE: usize = 4;
12}
13
14impl Encode for MultifragmentUpdate {
15    fn encode(&self, dst: &mut WriteCursor<'_>) -> EncodeResult<()> {
16        ensure_fixed_part_size!(in: dst);
17
18        dst.write_u32(self.max_request_size);
19
20        Ok(())
21    }
22
23    fn name(&self) -> &'static str {
24        Self::NAME
25    }
26
27    fn size(&self) -> usize {
28        Self::FIXED_PART_SIZE
29    }
30}
31
32impl<'de> Decode<'de> for MultifragmentUpdate {
33    fn decode(src: &mut ReadCursor<'de>) -> DecodeResult<Self> {
34        ensure_fixed_part_size!(in: src);
35
36        let max_request_size = src.read_u32();
37
38        Ok(Self { max_request_size })
39    }
40}
41
42#[cfg(test)]
43mod test {
44    use ironrdp_core::{decode, encode_vec};
45
46    use super::*;
47
48    const MULTIFRAGMENT_UPDATE_PDU_BUFFER: [u8; 4] = [0xf4, 0xf3, 0xf2, 0xf1];
49    const MULTIFRAGMENT_UPDATE_PDU: MultifragmentUpdate = MultifragmentUpdate {
50        max_request_size: 0xf1f2_f3f4,
51    };
52
53    #[test]
54    fn from_buffer_correctly_parses_multifragment_update() {
55        assert_eq!(
56            MULTIFRAGMENT_UPDATE_PDU,
57            decode(MULTIFRAGMENT_UPDATE_PDU_BUFFER.as_ref()).unwrap()
58        );
59    }
60
61    #[test]
62    fn to_buffer_correctly_serializes_multifragment_update() {
63        let expected = MULTIFRAGMENT_UPDATE_PDU_BUFFER.as_ref();
64
65        let buffer = encode_vec(&MULTIFRAGMENT_UPDATE_PDU).unwrap();
66        assert_eq!(expected, buffer.as_slice());
67    }
68
69    #[test]
70    fn buffer_length_is_correct_for_multifragment_update() {
71        assert_eq!(MULTIFRAGMENT_UPDATE_PDU_BUFFER.len(), MULTIFRAGMENT_UPDATE_PDU.size());
72    }
73}
ironrdp_pdu/rdp/capability_sets/multifragment_update.rs

ironrdp_pdu/rdp/capability_sets/
multifragment_update.rs
