rtc_datachannel/message/
mod.rs

1#[cfg(test)]
2mod message_test;
3
4pub mod message_channel_ack;
5pub mod message_channel_close;
6pub mod message_channel_open;
7pub mod message_channel_threshold;
8pub mod message_type;
9
10use bytes::{Buf, BufMut};
11use message_channel_ack::*;
12use message_channel_close::*;
13use message_channel_open::*;
14use message_channel_threshold::*;
15use message_type::*;
16use shared::error::{Error, Result};
17use shared::marshal::*;
18
19/// A parsed DataChannel message
20#[derive(Eq, PartialEq, Clone, Debug)]
21pub enum Message {
22    DataChannelThreshold(DataChannelThreshold), // internal usage only
23    DataChannelClose(DataChannelClose),         // internal usage only
24    DataChannelAck(DataChannelAck),
25    DataChannelOpen(DataChannelOpen),
26}
27
28impl MarshalSize for Message {
29    fn marshal_size(&self) -> usize {
30        match self {
31            Message::DataChannelThreshold(m) => m.marshal_size() + MESSAGE_TYPE_LEN, // internal usage only
32            Message::DataChannelClose(m) => m.marshal_size() + MESSAGE_TYPE_LEN, // internal usage only
33            Message::DataChannelAck(m) => m.marshal_size() + MESSAGE_TYPE_LEN,
34            Message::DataChannelOpen(m) => m.marshal_size() + MESSAGE_TYPE_LEN,
35        }
36    }
37}
38
39impl Marshal for Message {
40    fn marshal_to(&self, mut buf: &mut [u8]) -> Result<usize> {
41        let mut bytes_written = 0;
42        let n = self.message_type().marshal_to(buf)?;
43        buf = &mut buf[n..];
44        bytes_written += n;
45        bytes_written += match self {
46            Message::DataChannelThreshold(threshold) => threshold.marshal_to(buf)?, // internal usage only
47            Message::DataChannelClose(_) => 0, // internal usage only
48            Message::DataChannelAck(_) => 0,
49            Message::DataChannelOpen(open) => open.marshal_to(buf)?,
50        };
51        Ok(bytes_written)
52    }
53}
54
55impl Unmarshal for Message {
56    fn unmarshal<B>(buf: &mut B) -> Result<Self>
57    where
58        Self: Sized,
59        B: Buf,
60    {
61        if buf.remaining() < MESSAGE_TYPE_LEN {
62            return Err(Error::UnexpectedEndOfBuffer {
63                expected: MESSAGE_TYPE_LEN,
64                actual: buf.remaining(),
65            });
66        }
67
68        match MessageType::unmarshal(buf)? {
69            MessageType::DataChannelThreshold => Ok(Self::DataChannelThreshold(
70                DataChannelThreshold::unmarshal(buf)?,
71            )), // internal usage only
72            MessageType::DataChannelClose => Ok(Self::DataChannelClose(DataChannelClose {})), // internal usage only
73            MessageType::DataChannelAck => Ok(Self::DataChannelAck(DataChannelAck {})),
74            MessageType::DataChannelOpen => {
75                Ok(Self::DataChannelOpen(DataChannelOpen::unmarshal(buf)?))
76            }
77        }
78    }
79}
80
81impl Message {
82    pub fn message_type(&self) -> MessageType {
83        match self {
84            Self::DataChannelThreshold(_) => MessageType::DataChannelThreshold, // internal usage only
85            Self::DataChannelClose(_) => MessageType::DataChannelClose, // internal usage only
86            Self::DataChannelAck(_) => MessageType::DataChannelAck,
87            Self::DataChannelOpen(_) => MessageType::DataChannelOpen,
88        }
89    }
90}
rtc_datachannel/message/mod.rs

rtc_datachannel/message/
mod.rs
