ironrdp_pdu/rdp/capability_sets/
frame_acknowledge.rs

1use ironrdp_core::{ensure_fixed_part_size, Decode, DecodeResult, Encode, EncodeResult, ReadCursor, WriteCursor};
2
3#[derive(Debug, Clone, PartialEq, Eq)]
4pub struct FrameAcknowledge {
5    pub max_unacknowledged_frame_count: u32,
6}
7
8impl FrameAcknowledge {
9    const NAME: &'static str = "FrameAcknowledge";
10
11    const FIXED_PART_SIZE: usize = 4 /* maxUnackFrameCount */;
12}
13
14impl Encode for FrameAcknowledge {
15    fn encode(&self, dst: &mut WriteCursor<'_>) -> EncodeResult<()> {
16        ensure_fixed_part_size!(in: dst);
17
18        dst.write_u32(self.max_unacknowledged_frame_count);
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 FrameAcknowledge {
33    fn decode(src: &mut ReadCursor<'de>) -> DecodeResult<Self> {
34        ensure_fixed_part_size!(in: src);
35
36        let max_unacknowledged_frame_count = src.read_u32();
37
38        Ok(Self {
39            max_unacknowledged_frame_count,
40        })
41    }
42}
43
44#[cfg(test)]
45mod test {
46    use ironrdp_core::{decode, encode_vec};
47
48    use super::*;
49
50    const FRAME_ACKNOWLEDGE_PDU_BUFFER: [u8; 4] = [0xf4, 0xf3, 0xf2, 0xf1];
51    const FRAME_ACKNOWLEDGE_PDU: FrameAcknowledge = FrameAcknowledge {
52        max_unacknowledged_frame_count: 0xf1f2_f3f4,
53    };
54
55    #[test]
56    fn from_buffer_correctly_parses_frame_acknowledge() {
57        assert_eq!(
58            FRAME_ACKNOWLEDGE_PDU,
59            decode(FRAME_ACKNOWLEDGE_PDU_BUFFER.as_ref()).unwrap()
60        );
61    }
62
63    #[test]
64    fn to_buffer_correctly_serializes_frame_acknowledge() {
65        let expected = FRAME_ACKNOWLEDGE_PDU_BUFFER.as_ref();
66
67        let buffer = encode_vec(&FRAME_ACKNOWLEDGE_PDU).unwrap();
68        assert_eq!(expected, buffer.as_slice());
69    }
70
71    #[test]
72    fn buffer_length_is_correct_for_frame_acknowledge() {
73        assert_eq!(FRAME_ACKNOWLEDGE_PDU_BUFFER.len(), FRAME_ACKNOWLEDGE_PDU.size());
74    }
75}