1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
/*
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sli.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using testing::ElementsAreArray;
using testing::make_tuple;
using webrtc::rtcp::RawPacket;
using webrtc::rtcp::Sli;
using webrtc::RTCPUtility::RtcpCommonHeader;
using webrtc::RTCPUtility::RtcpParseCommonHeader;
namespace webrtc {
namespace {
const uint32_t kSenderSsrc = 0x12345678;
const uint32_t kRemoteSsrc = 0x23456789;
const uint8_t kPictureId = 0x3f;
const uint16_t kFirstMb = 0x1e61;
const uint16_t kNumberOfMb = 0x1a0a;
const uint32_t kSliItem = (static_cast<uint32_t>(kFirstMb) << 19) |
(static_cast<uint32_t>(kNumberOfMb) << 6) |
static_cast<uint32_t>(kPictureId);
// Manually created Sli packet matching constants above.
const uint8_t kPacket[] = {0x82, 206, 0x00, 0x03,
0x12, 0x34, 0x56, 0x78,
0x23, 0x45, 0x67, 0x89,
(kSliItem >> 24) & 0xff,
(kSliItem >> 16) & 0xff,
(kSliItem >> 8) & 0xff,
kSliItem & 0xff};
const size_t kPacketLength = sizeof(kPacket);
bool ParseSli(const uint8_t* buffer, size_t length, Sli* sli) {
RtcpCommonHeader header;
EXPECT_TRUE(RtcpParseCommonHeader(buffer, length, &header));
EXPECT_EQ(length, header.BlockSize());
return sli->Parse(header, buffer + RtcpCommonHeader::kHeaderSizeBytes);
}
TEST(RtcpPacketSliTest, Create) {
Sli sli;
sli.From(kSenderSsrc);
sli.To(kRemoteSsrc);
sli.WithPictureId(kPictureId, kFirstMb, kNumberOfMb);
rtc::scoped_ptr<RawPacket> packet(sli.Build());
EXPECT_THAT(make_tuple(packet->Buffer(), packet->Length()),
ElementsAreArray(kPacket));
}
TEST(RtcpPacketSliTest, Parse) {
Sli mutable_parsed;
EXPECT_TRUE(ParseSli(kPacket, kPacketLength, &mutable_parsed));
const Sli& parsed = mutable_parsed; // Read values from constant object.
EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc());
EXPECT_EQ(kRemoteSsrc, parsed.media_ssrc());
EXPECT_EQ(1u, parsed.macroblocks().size());
EXPECT_EQ(kFirstMb, parsed.macroblocks()[0].first());
EXPECT_EQ(kNumberOfMb, parsed.macroblocks()[0].number());
EXPECT_EQ(kPictureId, parsed.macroblocks()[0].picture_id());
}
TEST(RtcpPacketSliTest, ParseFailsOnTooSmallPacket) {
Sli sli;
sli.From(kSenderSsrc);
sli.To(kRemoteSsrc);
sli.WithPictureId(kPictureId, kFirstMb, kNumberOfMb);
rtc::scoped_ptr<RawPacket> packet(sli.Build());
packet->MutableBuffer()[3]--; // Decrease size by 1 word (4 bytes).
EXPECT_FALSE(ParseSli(packet->Buffer(), packet->Length() - 4, &sli));
}
} // namespace
} // namespace webrtc
|
