aldrin_core/message/
subscribe_event.rs

1use super::message_ops::Sealed;
2use super::{Message, MessageKind, MessageOps, OptionKind};
3use crate::ids::ServiceCookie;
4use crate::message_deserializer::{MessageDeserializeError, MessageWithoutValueDeserializer};
5use crate::message_serializer::{MessageSerializeError, MessageSerializer};
6use crate::serialized_value::SerializedValueSlice;
7use bytes::BytesMut;
8
9#[derive(Debug, Copy, Clone, PartialEq, Eq)]
10#[cfg_attr(feature = "fuzzing", derive(arbitrary::Arbitrary))]
11pub struct SubscribeEvent {
12    pub serial: Option<u32>,
13    pub service_cookie: ServiceCookie,
14    pub event: u32,
15}
16
17impl MessageOps for SubscribeEvent {
18    fn kind(&self) -> MessageKind {
19        MessageKind::SubscribeEvent
20    }
21
22    fn serialize_message(self) -> Result<BytesMut, MessageSerializeError> {
23        let mut serializer = MessageSerializer::without_value(MessageKind::SubscribeEvent);
24
25        match self.serial {
26            None => {
27                serializer.put_discriminant_u8(OptionKind::None);
28            }
29
30            Some(serial) => {
31                serializer.put_discriminant_u8(OptionKind::Some);
32                serializer.put_varint_u32_le(serial);
33            }
34        }
35
36        serializer.put_uuid(self.service_cookie.0);
37        serializer.put_varint_u32_le(self.event);
38
39        serializer.finish()
40    }
41
42    fn deserialize_message(buf: BytesMut) -> Result<Self, MessageDeserializeError> {
43        let mut deserializer =
44            MessageWithoutValueDeserializer::new(buf, MessageKind::SubscribeEvent)?;
45
46        let serial = match deserializer.try_get_discriminant_u8()? {
47            OptionKind::None => None,
48            OptionKind::Some => deserializer.try_get_varint_u32_le().map(Some)?,
49        };
50
51        let service_cookie = deserializer.try_get_uuid().map(ServiceCookie)?;
52        let event = deserializer.try_get_varint_u32_le()?;
53
54        deserializer.finish()?;
55        Ok(Self {
56            serial,
57            service_cookie,
58            event,
59        })
60    }
61
62    fn value(&self) -> Option<&SerializedValueSlice> {
63        None
64    }
65}
66
67impl Sealed for SubscribeEvent {}
68
69impl From<SubscribeEvent> for Message {
70    fn from(msg: SubscribeEvent) -> Self {
71        Self::SubscribeEvent(msg)
72    }
73}
74
75#[cfg(test)]
76mod test {
77    use super::super::test::{assert_deserialize_eq, assert_serialize_eq};
78    use super::super::Message;
79    use super::SubscribeEvent;
80    use crate::ids::ServiceCookie;
81    use uuid::uuid;
82
83    #[test]
84    fn no_serial() {
85        let serialized = [
86            23, 0, 0, 0, 13, 0, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc, 0x4e,
87            0x0e, 0x84, 0x80, 0xdf, 0xad, 1,
88        ];
89
90        let msg = SubscribeEvent {
91            serial: None,
92            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
93            event: 1,
94        };
95        assert_serialize_eq(&msg, serialized);
96        assert_deserialize_eq(&msg, serialized);
97
98        let msg = Message::SubscribeEvent(msg);
99        assert_serialize_eq(&msg, serialized);
100        assert_deserialize_eq(&msg, serialized);
101    }
102
103    #[test]
104    fn serial() {
105        let serialized = [
106            24, 0, 0, 0, 13, 1, 2, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc,
107            0x4e, 0x0e, 0x84, 0x80, 0xdf, 0xad, 3,
108        ];
109
110        let msg = SubscribeEvent {
111            serial: Some(2),
112            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
113            event: 3,
114        };
115        assert_serialize_eq(&msg, serialized);
116        assert_deserialize_eq(&msg, serialized);
117
118        let msg = Message::SubscribeEvent(msg);
119        assert_serialize_eq(&msg, serialized);
120        assert_deserialize_eq(&msg, serialized);
121    }
122}
aldrin_core/message/subscribe_event.rs

aldrin_core/message/
subscribe_event.rs
