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 | |
11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sli.h" | |
12 | |
13 #include "testing/gmock/include/gmock/gmock.h" | |
14 #include "testing/gtest/include/gtest/gtest.h" | |
15 | |
16 using testing::ElementsAreArray; | |
17 using testing::Eq; | |
18 using testing::make_tuple; | |
19 using webrtc::rtcp::RawPacket; | |
20 using webrtc::rtcp::Sli; | |
21 | |
22 namespace webrtc { | |
23 namespace { | |
24 | |
25 const uint32_t kSenderSsrc = 0x12345678; | |
26 const uint32_t kRemoteSsrc = 0x23456789; | |
27 | |
28 const uint8_t kPictureId = 0x3c; | |
29 const uint16_t kFirstMb = 0x1e61; | |
30 const uint16_t kNumberOfMb = 0x1a0a; | |
31 const uint32_t kSliItem = (kFirstMb << 19) | (kNumberOfMb << 6) | kPictureId; | |
32 | |
33 // Manually created Sli packet matching constants above. | |
34 const uint8_t kPacket[] = {0x82, 206, 0x00, 0x03, | |
35 0x12, 0x34, 0x56, 0x78, | |
36 0x23, 0x45, 0x67, 0x89, | |
37 (kSliItem >> 24) & 0xff, | |
38 (kSliItem >> 16) & 0xff, | |
39 (kSliItem >> 8) & 0xff, | |
40 kSliItem & 0xff}; | |
41 const size_t kPacketLength = sizeof(kPacket); | |
42 | |
43 bool ParseSli(const uint8_t* buffer, size_t length, Sli* sli) { | |
44 using RTCPUtility::RtcpCommonHeader; | |
45 using RTCPUtility::RtcpParseCommonHeader; | |
åsapersson
2016/01/11 12:13:54
move line 44, 45 to other using above?
danilchap
2016/01/11 15:36:32
Because of plans to drop dependency on RTCPUtility
åsapersson
2016/01/12 12:35:26
Acknowledged.
danilchap
2016/01/12 12:56:35
Reconsidered: dropping will be equally easy from b
| |
46 RtcpCommonHeader header; | |
47 EXPECT_TRUE(RtcpParseCommonHeader(buffer, length, &header)); | |
48 EXPECT_THAT(header.BlockSize(), Eq(length)); | |
åsapersson
2016/01/11 12:13:54
EXPECT_EQ(length, header.BlockSize());
danilchap
2016/01/11 15:36:32
EXPECT_THAT syntax makes it more clear which value
| |
49 return sli->Parse(header, buffer + RtcpCommonHeader::kHeaderSizeBytes); | |
50 } | |
51 | |
52 TEST(RtcpPacketSliTest, Create) { | |
53 Sli sli; | |
54 sli.From(kSenderSsrc); | |
55 sli.To(kRemoteSsrc); | |
56 sli.WithMacroblocks(kPictureId, kFirstMb, kNumberOfMb); | |
57 | |
58 rtc::scoped_ptr<RawPacket> packet(sli.Build()); | |
59 | |
60 EXPECT_THAT(make_tuple(packet->Buffer(), packet->Length()), | |
61 ElementsAreArray(kPacket)); | |
62 } | |
63 | |
64 TEST(RtcpPacketSliTest, Parse) { | |
65 Sli mutable_parsed; | |
66 EXPECT_TRUE(ParseSli(kPacket, kPacketLength, &mutable_parsed)); | |
67 const Sli& parsed = mutable_parsed; // Read values from constant object. | |
68 | |
69 EXPECT_THAT(parsed.sender_ssrc(), Eq(kSenderSsrc)); | |
åsapersson
2016/01/11 12:13:54
use EXPECT_EQ?
EXPECT_EQ(kSenderSsrc, parsed.sende
| |
70 EXPECT_THAT(parsed.media_ssrc(), Eq(kRemoteSsrc)); | |
71 EXPECT_THAT(parsed.macroblocks()[0].first(), Eq(kFirstMb)); | |
72 EXPECT_THAT(parsed.macroblocks()[0].number(), Eq(kNumberOfMb)); | |
73 EXPECT_THAT(parsed.macroblocks()[0].picture_id(), Eq(kPictureId)); | |
74 } | |
75 | |
76 TEST(RtcpPacketSliTest, ParseFailesOnSmallPacket) { | |
åsapersson
2016/01/11 12:13:54
s/Failes/Fails
åsapersson
2016/01/11 12:13:54
maybe ParseFailsOnTooSmallPacket
danilchap
2016/01/11 15:36:32
Done.
| |
77 Sli sli; | |
78 sli.From(kSenderSsrc); | |
79 sli.To(kRemoteSsrc); | |
80 sli.WithMacroblocks(kPictureId, kFirstMb, kNumberOfMb); | |
81 | |
82 rtc::scoped_ptr<RawPacket> packet(sli.Build()); | |
83 packet->MutableBuffer()[3]--; // Decrease size by 1 word (4 bytes). | |
84 | |
85 EXPECT_FALSE(ParseSli(packet->Buffer(), packet->Length() - 4, &sli)); | |
86 } | |
87 | |
88 } // namespace | |
89 } // namespace webrtc | |
OLD | NEW |