mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 14:50:39 +01:00
8037fc6ffa
Bug: webrtc:342905193 No-Try: True Change-Id: Icc968be43b8830038ea9a1f5f604307220457807 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/361021 Auto-Submit: Florent Castelli <orphis@webrtc.org> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Florent Castelli <orphis@webrtc.org> Cr-Commit-Position: refs/heads/main@{#42911}
96 lines
3.7 KiB
C++
96 lines
3.7 KiB
C++
/*
|
|
* 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 "modules/rtp_rtcp/source/rtcp_packet/target_bitrate.h"
|
|
|
|
#include "modules/rtp_rtcp/source/byte_io.h"
|
|
#include "modules/rtp_rtcp/source/rtcp_packet/extended_reports.h"
|
|
#include "rtc_base/buffer.h"
|
|
#include "test/gtest.h"
|
|
#include "test/rtcp_packet_parser.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
using BitrateItem = rtcp::TargetBitrate::BitrateItem;
|
|
using rtcp::TargetBitrate;
|
|
using test::ParseSinglePacket;
|
|
|
|
constexpr uint32_t kSsrc = 0x12345678;
|
|
|
|
// clang-format off
|
|
const uint8_t kPacket[] = { TargetBitrate::kBlockType, // Block ID.
|
|
0x00, // Reserved.
|
|
0x00, 0x04, // Length = 4 words.
|
|
0x00, 0x01, 0x02, 0x03, // S0T0 0x010203 kbps.
|
|
0x01, 0x02, 0x03, 0x04, // S0T1 0x020304 kbps.
|
|
0x10, 0x03, 0x04, 0x05, // S1T0 0x030405 kbps.
|
|
0x11, 0x04, 0x05, 0x06 }; // S1T1 0x040506 kbps.
|
|
constexpr size_t kPacketLengthBlocks = ((sizeof(kPacket) + 3) / 4) - 1;
|
|
// clang-format on
|
|
|
|
void ExpectBirateItemEquals(const BitrateItem& expected,
|
|
const BitrateItem& actual) {
|
|
EXPECT_EQ(expected.spatial_layer, actual.spatial_layer);
|
|
EXPECT_EQ(expected.temporal_layer, actual.temporal_layer);
|
|
EXPECT_EQ(expected.target_bitrate_kbps, actual.target_bitrate_kbps);
|
|
}
|
|
|
|
void CheckBitrateItems(const std::vector<BitrateItem>& bitrates) {
|
|
EXPECT_EQ(4U, bitrates.size());
|
|
ExpectBirateItemEquals(BitrateItem(0, 0, 0x010203), bitrates[0]);
|
|
ExpectBirateItemEquals(BitrateItem(0, 1, 0x020304), bitrates[1]);
|
|
ExpectBirateItemEquals(BitrateItem(1, 0, 0x030405), bitrates[2]);
|
|
ExpectBirateItemEquals(BitrateItem(1, 1, 0x040506), bitrates[3]);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(TargetBitrateTest, Parse) {
|
|
TargetBitrate target_bitrate;
|
|
target_bitrate.Parse(kPacket, kPacketLengthBlocks);
|
|
CheckBitrateItems(target_bitrate.GetTargetBitrates());
|
|
}
|
|
|
|
TEST(TargetBitrateTest, FullPacket) {
|
|
const size_t kXRHeaderSize = 8; // RTCP header (4) + SSRC (4).
|
|
const size_t kTotalSize = kXRHeaderSize + sizeof(kPacket);
|
|
uint8_t kRtcpPacket[kTotalSize] = {2 << 6, 207, 0x00, (kTotalSize / 4) - 1,
|
|
0x12, 0x34, 0x56, 0x78}; // SSRC.
|
|
memcpy(&kRtcpPacket[kXRHeaderSize], kPacket, sizeof(kPacket));
|
|
rtcp::ExtendedReports xr;
|
|
EXPECT_TRUE(ParseSinglePacket(kRtcpPacket, &xr));
|
|
EXPECT_EQ(kSsrc, xr.sender_ssrc());
|
|
const std::optional<TargetBitrate>& target_bitrate = xr.target_bitrate();
|
|
ASSERT_TRUE(static_cast<bool>(target_bitrate));
|
|
CheckBitrateItems(target_bitrate->GetTargetBitrates());
|
|
}
|
|
|
|
TEST(TargetBitrateTest, Create) {
|
|
TargetBitrate target_bitrate;
|
|
target_bitrate.AddTargetBitrate(0, 0, 0x010203);
|
|
target_bitrate.AddTargetBitrate(0, 1, 0x020304);
|
|
target_bitrate.AddTargetBitrate(1, 0, 0x030405);
|
|
target_bitrate.AddTargetBitrate(1, 1, 0x040506);
|
|
|
|
uint8_t buffer[sizeof(kPacket)] = {};
|
|
ASSERT_EQ(sizeof(kPacket), target_bitrate.BlockLength());
|
|
target_bitrate.Create(buffer);
|
|
|
|
EXPECT_EQ(0, memcmp(kPacket, buffer, sizeof(kPacket)));
|
|
}
|
|
|
|
TEST(TargetBitrateTest, ParseNullBitratePacket) {
|
|
const uint8_t kNullPacket[] = {TargetBitrate::kBlockType, 0x00, 0x00, 0x00};
|
|
TargetBitrate target_bitrate;
|
|
target_bitrate.Parse(kNullPacket, 0);
|
|
EXPECT_TRUE(target_bitrate.GetTargetBitrates().empty());
|
|
}
|
|
|
|
} // namespace webrtc
|