1
use super::{CodecExtra, Depacketizer, PacketError, Packetizer};
2
 
3
/// Packetizes Opus RTP packets.
4
///
5
/// ## Unversioned API surface
6
///
7
/// This struct is not currently versioned according to semver rules.
8
/// Breaking changes may be made in minor or patch releases.
9
#[derive(Default, Debug, Copy, Clone)]
10
pub struct OpusPacketizer;
11
 
12
impl Packetizer for OpusPacketizer {
13
    fn packetize(&mut self, mtu: usize, payload: &[u8]) -> Result<Vec<Vec<u8>>, PacketError> {
14
        if payload.is_empty() || mtu == 0 {
15
            return Ok(vec![]);
16
        }
17
 
18
        let mut out = vec![];
19
 
20
        let mut cur = 0;
21
        while cur < payload.len() {
22
            let min = mtu.min(payload.len() - cur);
23
            out.push(payload[cur..(cur + min)].to_vec());
24
            cur += mtu;
25
        }
26
 
27
        Ok(out)
28
    }
29
 
30
    fn is_marker(&mut self, _data: &[u8], _previous: Option<&[u8]>, _last: bool) -> bool {
31
        // TODO: dtx
32
        false
33
    }
34
}
35
 
36
/// Depacketizes Opus RTP packets.
37
///
38
/// ## Unversioned API surface
39
///
40
/// This struct is not currently versioned according to semver rules.
41
/// Breaking changes may be made in minor or patch releases.
42
#[derive(PartialEq, Eq, Debug, Default, Clone)]
43
pub struct OpusDepacketizer;
44
 
45
impl Depacketizer for OpusDepacketizer {
46
    fn out_size_hint(&self, packets_size: usize) -> Option<usize> {
47
        Some(packets_size)
48
    }
49
 
50
    fn depacketize(
51
        &mut self,
52
        packet: &[u8],
53
        out: &mut Vec<u8>,
54
        _: &mut CodecExtra,
55
    ) -> Result<(), PacketError> {
56
        if !packet.is_empty() {
57
            out.extend_from_slice(packet);
58
        }
59
 
60
        Ok(())
61
    }
62
 
63
    fn is_partition_head(&self, _payload: &[u8]) -> bool {
64
        true
65
    }
66
 
67
    fn is_partition_tail(&self, _marker: bool, _payload: &[u8]) -> bool {
68
        true
69
    }
70
}
71
 
72
#[cfg(test)]
73
mod test {
74
    use super::*;
75
 
76
    #[test]
77
    fn test_opus_unmarshal() -> Result<(), PacketError> {
78
        let mut pck = OpusDepacketizer;
79
        let mut extra = CodecExtra::None;
80
 
81
        // Empty packet
82
        let empty_bytes = &[];
83
        let mut out = Vec::new();
84
        let result = pck.depacketize(empty_bytes, &mut out, &mut extra);
85
        assert!(
86
            result.is_ok(),
87
            "Result should not be err in case of empty packet"
88
        );
89
 
90
        // Normal packet
91
        let raw_bytes: &[u8] = &[0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x90];
92
        let mut out = Vec::new();
93
        pck.depacketize(raw_bytes, &mut out, &mut extra)?;
94
        assert_eq!(raw_bytes, &out, "Payload must be same");
95
 
96
        Ok(())
97
    }
98
 
99
    #[test]
100
    fn test_opus_payload() -> Result<(), PacketError> {
101
        let mut pck = OpusPacketizer;
102
        let empty = &[];
103
        let payload = &[0x90, 0x90, 0x90];
104
 
105
        // Positive MTU, empty payload
106
        let result = pck.packetize(1, empty)?;
107
        assert!(result.is_empty(), "Generated payload should be empty");
108
 
109
        // Positive MTU, small payload
110
        let result = pck.packetize(1, payload)?;
111
        assert_eq!(result.len(), 3, "Generated payload should be 3");
112
 
113
        // Positive MTU, small payload
114
        let result = pck.packetize(2, payload)?;
115
        assert_eq!(result.len(), 2, "Generated payload should be the 2");
116
 
117
        Ok(())
118
    }
119
 
120
    #[test]
121
    fn test_opus_is_partition_head() -> Result<(), PacketError> {
122
        let opus = OpusDepacketizer;
123
        //"NormalPacket"
124
        assert!(
125
            opus.is_partition_head(&[0x00, 0x00]),
126
            "All OPUS RTP packet should be the head of a new partition"
127
        );
128
 
129
        Ok(())
130
    }
131
 
132
    #[test]
133
    fn packetize_respects_mtu() -> Result<(), PacketError> {
134
        let payload = vec![0xABu8; 2000];
135
        for &mtu in &[100usize, 300, 600, 1200] {
136
            let mut pck = OpusPacketizer;
137
            let pkts = pck.packetize(mtu, &payload)?;
138
            assert!(!pkts.is_empty(), "Opus produced no packets at mtu {mtu}");
139
            for (i, pkt) in pkts.iter().enumerate() {
140
                assert!(
141
                    pkt.len() <= mtu,
142
                    "Opus packet {i} size {} > mtu {mtu}",
143
                    pkt.len()
144
                );
145
            }
146
        }
147
        Ok(())
148
    }
149
}