OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h" |
| 11 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_to_send.h" |
| 12 |
| 13 #include "testing/gmock/include/gmock/gmock.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 #include "webrtc/base/random.h" |
| 16 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h" |
| 17 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extension.h" |
| 18 |
| 19 using testing::ElementsAreArray; |
| 20 using testing::make_tuple; |
| 21 |
| 22 namespace webrtc { |
| 23 namespace { |
| 24 constexpr int8_t kPayloadType = 100; |
| 25 constexpr uint32_t kSsrc = 0x12345678; |
| 26 constexpr uint16_t kSeqNum = 88; |
| 27 constexpr uint32_t kTimestamp = 0x65431278; |
| 28 constexpr uint8_t kTransmissionOffsetExtensionId = 1; |
| 29 constexpr uint8_t kAudioLevelExtensionId = 9; |
| 30 constexpr int32_t kTimeOffset = 0x56ce; |
| 31 constexpr bool kVoiceActive = true; |
| 32 constexpr uint8_t kAudioLevel = 0x5a; |
| 33 constexpr size_t kMaxPaddingSize = 224u; |
| 34 constexpr uint8_t kMinimumPacket[] = { |
| 35 0x80, kPayloadType, 0x00, kSeqNum, |
| 36 0x65, 0x43, 0x12, 0x78, |
| 37 0x12, 0x34, 0x56, 0x78}; |
| 38 constexpr uint8_t kPacketWithTO[] = { |
| 39 0x90, kPayloadType, 0x00, kSeqNum, |
| 40 0x65, 0x43, 0x12, 0x78, |
| 41 0x12, 0x34, 0x56, 0x78, |
| 42 0xbe, 0xde, 0x00, 0x01, |
| 43 0x12, 0x00, 0x56, 0xce}; |
| 44 |
| 45 constexpr uint8_t kPacketWithTOAndAL[] = { |
| 46 0x90, kPayloadType, 0x00, kSeqNum, |
| 47 0x65, 0x43, 0x12, 0x78, |
| 48 0x12, 0x34, 0x56, 0x78, |
| 49 0xbe, 0xde, 0x00, 0x02, |
| 50 0x12, 0x00, 0x56, 0xce, |
| 51 0x90, 0x80|kAudioLevel, 0x00, 0x00}; |
| 52 |
| 53 constexpr uint32_t kCsrcs[] = {0x34567890, 0x32435465}; |
| 54 constexpr uint8_t kPayload[] = {'p', 'a', 'y', 'l', 'o', 'a', 'd'}; |
| 55 constexpr uint8_t kPacketPaddingSize = 8; |
| 56 constexpr uint8_t kPacket[] = { |
| 57 0xb2, kPayloadType, 0x00, kSeqNum, |
| 58 0x65, 0x43, 0x12, 0x78, |
| 59 0x12, 0x34, 0x56, 0x78, |
| 60 0x34, 0x56, 0x78, 0x90, |
| 61 0x32, 0x43, 0x54, 0x65, |
| 62 0xbe, 0xde, 0x00, 0x01, |
| 63 0x12, 0x00, 0x56, 0xce, |
| 64 'p', 'a', 'y', 'l', 'o', 'a', 'd', |
| 65 'p', 'a', 'd', 'd', 'i', 'n', 'g', kPacketPaddingSize}; |
| 66 |
| 67 } // namespace |
| 68 |
| 69 TEST(RtpPacketTest, CreateMinimum) { |
| 70 RtpPacketToSend packet(nullptr); |
| 71 packet.SetPayloadType(kPayloadType); |
| 72 packet.SetSequenceNumber(kSeqNum); |
| 73 packet.SetTimestamp(kTimestamp); |
| 74 packet.SetSsrc(kSsrc); |
| 75 EXPECT_THAT(kMinimumPacket, ElementsAreArray(packet.data(), packet.size())); |
| 76 } |
| 77 |
| 78 TEST(RtpPacketTest, CreateWithExtension) { |
| 79 RtpPacketToSend::ExtensionManager extensions; |
| 80 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 81 kTransmissionOffsetExtensionId); |
| 82 RtpPacketToSend packet(&extensions); |
| 83 packet.SetPayloadType(kPayloadType); |
| 84 packet.SetSequenceNumber(kSeqNum); |
| 85 packet.SetTimestamp(kTimestamp); |
| 86 packet.SetSsrc(kSsrc); |
| 87 packet.SetExtension<TransmissionOffset>(kTimeOffset); |
| 88 EXPECT_THAT(kPacketWithTO, ElementsAreArray(packet.data(), packet.size())); |
| 89 } |
| 90 |
| 91 TEST(RtpPacketTest, CreateWith2Extensions) { |
| 92 RtpPacketToSend::ExtensionManager extensions; |
| 93 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 94 kTransmissionOffsetExtensionId); |
| 95 extensions.Register(kRtpExtensionAudioLevel, kAudioLevelExtensionId); |
| 96 RtpPacketToSend packet(&extensions); |
| 97 packet.SetPayloadType(kPayloadType); |
| 98 packet.SetSequenceNumber(kSeqNum); |
| 99 packet.SetTimestamp(kTimestamp); |
| 100 packet.SetSsrc(kSsrc); |
| 101 packet.SetExtension<TransmissionOffset>(kTimeOffset); |
| 102 packet.SetExtension<AudioLevel>(kVoiceActive, kAudioLevel); |
| 103 EXPECT_THAT(kPacketWithTOAndAL, |
| 104 ElementsAreArray(packet.data(), packet.size())); |
| 105 } |
| 106 |
| 107 TEST(RtpPacketTest, SetReservedExtensionsAfterPayload) { |
| 108 const size_t kPayloadSize = 4; |
| 109 RtpPacketToSend::ExtensionManager extensions; |
| 110 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 111 kTransmissionOffsetExtensionId); |
| 112 extensions.Register(kRtpExtensionAudioLevel, kAudioLevelExtensionId); |
| 113 RtpPacketToSend packet(&extensions); |
| 114 |
| 115 EXPECT_TRUE(packet.ReserveExtension<TransmissionOffset>()); |
| 116 packet.AllocatePayload(kPayloadSize); |
| 117 // Can't set extension after payload. |
| 118 EXPECT_FALSE(packet.SetExtension<AudioLevel>(kVoiceActive, kAudioLevel)); |
| 119 // Unless reserved. |
| 120 EXPECT_TRUE(packet.SetExtension<TransmissionOffset>(kTimeOffset)); |
| 121 } |
| 122 |
| 123 TEST(RtpPacketTest, CreatePurePadding) { |
| 124 const size_t kPaddingSize = kMaxPaddingSize - 1; |
| 125 RtpPacketToSend packet(nullptr, 12 + kPaddingSize); |
| 126 packet.SetPayloadType(kPayloadType); |
| 127 packet.SetSequenceNumber(kSeqNum); |
| 128 packet.SetTimestamp(kTimestamp); |
| 129 packet.SetSsrc(kSsrc); |
| 130 Random random(0x123456789); |
| 131 |
| 132 EXPECT_LT(packet.size(), packet.capacity()); |
| 133 EXPECT_FALSE(packet.SetPadding(kPaddingSize + 1, &random)); |
| 134 EXPECT_TRUE(packet.SetPadding(kPaddingSize, &random)); |
| 135 EXPECT_EQ(packet.size(), packet.capacity()); |
| 136 } |
| 137 |
| 138 TEST(RtpPacketTest, CreateUnalignedPadding) { |
| 139 const size_t kPayloadSize = 3; // Make padding start at unaligned address. |
| 140 RtpPacketToSend packet(nullptr, 12 + kPayloadSize + kMaxPaddingSize); |
| 141 packet.SetPayloadType(kPayloadType); |
| 142 packet.SetSequenceNumber(kSeqNum); |
| 143 packet.SetTimestamp(kTimestamp); |
| 144 packet.SetSsrc(kSsrc); |
| 145 packet.AllocatePayload(kPayloadSize); |
| 146 Random r(0x123456789); |
| 147 |
| 148 EXPECT_LT(packet.size(), packet.capacity()); |
| 149 EXPECT_TRUE(packet.SetPadding(kMaxPaddingSize, &r)); |
| 150 EXPECT_EQ(packet.size(), packet.capacity()); |
| 151 } |
| 152 |
| 153 TEST(RtpPacketTest, ParseMinimum) { |
| 154 RtpPacketReceived packet; |
| 155 EXPECT_TRUE(packet.Parse(kMinimumPacket, sizeof(kMinimumPacket))); |
| 156 EXPECT_EQ(kPayloadType, packet.PayloadType()); |
| 157 EXPECT_EQ(kSeqNum, packet.SequenceNumber()); |
| 158 EXPECT_EQ(kTimestamp, packet.Timestamp()); |
| 159 EXPECT_EQ(kSsrc, packet.Ssrc()); |
| 160 EXPECT_EQ(0u, packet.padding_size()); |
| 161 EXPECT_EQ(0u, packet.payload_size()); |
| 162 } |
| 163 |
| 164 TEST(RtpPacketTest, ParseBuffer) { |
| 165 rtc::Buffer unparsed(kMinimumPacket); |
| 166 const uint8_t* raw = unparsed.data(); |
| 167 |
| 168 RtpPacketReceived packet; |
| 169 EXPECT_TRUE(packet.Parse(std::move(unparsed))); |
| 170 EXPECT_EQ(raw, packet.data()); // Expect packet took over the buffer. |
| 171 EXPECT_EQ(kSeqNum, packet.SequenceNumber()); |
| 172 EXPECT_EQ(kTimestamp, packet.Timestamp()); |
| 173 EXPECT_EQ(kSsrc, packet.Ssrc()); |
| 174 EXPECT_EQ(0u, packet.padding_size()); |
| 175 EXPECT_EQ(0u, packet.payload_size()); |
| 176 } |
| 177 |
| 178 TEST(RtpPacketTest, ParseWithExtension) { |
| 179 RtpPacketToSend::ExtensionManager extensions; |
| 180 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 181 kTransmissionOffsetExtensionId); |
| 182 |
| 183 RtpPacketReceived packet(&extensions); |
| 184 EXPECT_TRUE(packet.Parse(kPacketWithTO, sizeof(kPacketWithTO))); |
| 185 EXPECT_EQ(kPayloadType, packet.PayloadType()); |
| 186 EXPECT_EQ(kSeqNum, packet.SequenceNumber()); |
| 187 EXPECT_EQ(kTimestamp, packet.Timestamp()); |
| 188 EXPECT_EQ(kSsrc, packet.Ssrc()); |
| 189 int32_t time_offset; |
| 190 EXPECT_TRUE(packet.GetExtension<TransmissionOffset>(&time_offset)); |
| 191 EXPECT_EQ(kTimeOffset, time_offset); |
| 192 EXPECT_EQ(0u, packet.payload_size()); |
| 193 EXPECT_EQ(0u, packet.padding_size()); |
| 194 } |
| 195 |
| 196 TEST(RtpPacketTest, ParseWith2Extensions) { |
| 197 RtpPacketToSend::ExtensionManager extensions; |
| 198 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 199 kTransmissionOffsetExtensionId); |
| 200 extensions.Register(kRtpExtensionAudioLevel, kAudioLevelExtensionId); |
| 201 RtpPacketReceived packet(&extensions); |
| 202 EXPECT_TRUE(packet.Parse(kPacketWithTOAndAL, sizeof(kPacketWithTOAndAL))); |
| 203 int32_t time_offset; |
| 204 EXPECT_TRUE(packet.GetExtension<TransmissionOffset>(&time_offset)); |
| 205 EXPECT_EQ(kTimeOffset, time_offset); |
| 206 bool voice_active; |
| 207 uint8_t audio_level; |
| 208 EXPECT_TRUE(packet.GetExtension<AudioLevel>(&voice_active, &audio_level)); |
| 209 EXPECT_EQ(kVoiceActive, voice_active); |
| 210 EXPECT_EQ(kAudioLevel, audio_level); |
| 211 } |
| 212 |
| 213 TEST(RtpPacketTest, ParseWithAllFeatures) { |
| 214 RtpPacketToSend::ExtensionManager extensions; |
| 215 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 216 kTransmissionOffsetExtensionId); |
| 217 RtpPacketReceived packet(&extensions); |
| 218 EXPECT_TRUE(packet.Parse(kPacket, sizeof(kPacket))); |
| 219 EXPECT_EQ(kPayloadType, packet.PayloadType()); |
| 220 EXPECT_EQ(kSeqNum, packet.SequenceNumber()); |
| 221 EXPECT_EQ(kTimestamp, packet.Timestamp()); |
| 222 EXPECT_EQ(kSsrc, packet.Ssrc()); |
| 223 EXPECT_THAT(packet.Csrcs(), ElementsAreArray(kCsrcs)); |
| 224 EXPECT_THAT(make_tuple(packet.payload(), packet.payload_size()), |
| 225 ElementsAreArray(kPayload)); |
| 226 EXPECT_EQ(kPacketPaddingSize, packet.padding_size()); |
| 227 int32_t time_offset; |
| 228 EXPECT_TRUE(packet.GetExtension<TransmissionOffset>(&time_offset)); |
| 229 } |
| 230 |
| 231 TEST(RtpPacketTest, ParseWithExtensionDelayed) { |
| 232 RtpPacketReceived packet; |
| 233 EXPECT_TRUE(packet.Parse(kPacketWithTO, sizeof(kPacketWithTO))); |
| 234 EXPECT_EQ(kPayloadType, packet.PayloadType()); |
| 235 EXPECT_EQ(kSeqNum, packet.SequenceNumber()); |
| 236 EXPECT_EQ(kTimestamp, packet.Timestamp()); |
| 237 EXPECT_EQ(kSsrc, packet.Ssrc()); |
| 238 |
| 239 RtpPacketToSend::ExtensionManager extensions; |
| 240 extensions.Register(kRtpExtensionTransmissionTimeOffset, |
| 241 kTransmissionOffsetExtensionId); |
| 242 |
| 243 int32_t time_offset; |
| 244 EXPECT_FALSE(packet.GetExtension<TransmissionOffset>(&time_offset)); |
| 245 packet.IdentifyExtensions(&extensions); |
| 246 EXPECT_TRUE(packet.GetExtension<TransmissionOffset>(&time_offset)); |
| 247 EXPECT_EQ(kTimeOffset, time_offset); |
| 248 EXPECT_EQ(0u, packet.payload_size()); |
| 249 EXPECT_EQ(0u, packet.padding_size()); |
| 250 } |
| 251 |
| 252 } // namespace webrtc |
OLD | NEW |