zqz/platform-external-webrtc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Verify sender ssrc when receiving rtcp target bitrate

Browse Source 
BUG=webrtc:8137

Review-Url: https://codereview.webrtc.org/3000373002
Cr-Commit-Position: refs/heads/master@{#19524}
This commit is contained in:
sprang
2017-08-25 09:06:16 -07:00
committed by Commit Bot
parent cb2bbce4e1
commit a7a4beb419
3 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
webrtc/modules/rtp_rtcp/source/rtcp_receiver.cc
View File

@ -688,8 +688,10 @@ void RTCPReceiver::HandleXr(const CommonHeader& rtcp_block,
for (const rtcp::ReceiveTimeInfo& time_info : xr.dlrr().sub_blocks()) for (const rtcp::ReceiveTimeInfo& time_info : xr.dlrr().sub_blocks())
HandleXrDlrrReportBlock(time_info); HandleXrDlrrReportBlock(time_info);
if (xr.target_bitrate()) if (xr.target_bitrate()) {
HandleXrTargetBitrate(*xr.target_bitrate(), packet_information); HandleXrTargetBitrate(xr.sender_ssrc(), *xr.target_bitrate(),
packet_information);
}
} }
void RTCPReceiver::HandleXrReceiveReferenceTime(uint32_t sender_ssrc, void RTCPReceiver::HandleXrReceiveReferenceTime(uint32_t sender_ssrc,
@ -722,8 +724,13 @@ void RTCPReceiver::HandleXrDlrrReportBlock(const rtcp::ReceiveTimeInfo& rti) {
} }
void RTCPReceiver::HandleXrTargetBitrate( void RTCPReceiver::HandleXrTargetBitrate(
uint32_t ssrc,
const rtcp::TargetBitrate& target_bitrate, const rtcp::TargetBitrate& target_bitrate,
PacketInformation* packet_information) { PacketInformation* packet_information) {
if (ssrc != remote_ssrc_) {
return; // Not for us.
}
BitrateAllocation bitrate_allocation; BitrateAllocation bitrate_allocation;
for (const auto& item : target_bitrate.GetTargetBitrates()) { for (const auto& item : target_bitrate.GetTargetBitrates()) {
if (item.spatial_layer >= kMaxSpatialLayers || if (item.spatial_layer >= kMaxSpatialLayers ||

3
webrtc/modules/rtp_rtcp/source/rtcp_receiver.h
View File

@ -165,7 +165,8 @@ class RTCPReceiver {
void HandleXrDlrrReportBlock(const rtcp::ReceiveTimeInfo& rti) void HandleXrDlrrReportBlock(const rtcp::ReceiveTimeInfo& rti)
EXCLUSIVE_LOCKS_REQUIRED(rtcp_receiver_lock_); EXCLUSIVE_LOCKS_REQUIRED(rtcp_receiver_lock_);
void HandleXrTargetBitrate(const rtcp::TargetBitrate& target_bitrate, void HandleXrTargetBitrate(uint32_t ssrc,
const rtcp::TargetBitrate& target_bitrate,
PacketInformation* packet_information) PacketInformation* packet_information)
EXCLUSIVE_LOCKS_REQUIRED(rtcp_receiver_lock_); EXCLUSIVE_LOCKS_REQUIRED(rtcp_receiver_lock_);

10
webrtc/modules/rtp_rtcp/source/rtcp_receiver_unittest.cc
View File

@ -1225,6 +1225,15 @@ TEST_F(RtcpReceiverTest, ReceivesTargetBitrate) {
rtcp::ExtendedReports xr; rtcp::ExtendedReports xr;
xr.SetTargetBitrate(bitrate); xr.SetTargetBitrate(bitrate);
// Wrong sender ssrc, target bitrate should be discarded.
xr.SetSenderSsrc(kSenderSsrc + 1);
EXPECT_CALL(bitrate_allocation_observer_,
OnBitrateAllocationUpdated(expected_allocation))
.Times(0);
InjectRtcpPacket(xr);
// Set correct ssrc, callback should be called once.
xr.SetSenderSsrc(kSenderSsrc);
EXPECT_CALL(bitrate_allocation_observer_, EXPECT_CALL(bitrate_allocation_observer_,
OnBitrateAllocationUpdated(expected_allocation)); OnBitrateAllocationUpdated(expected_allocation));
InjectRtcpPacket(xr); InjectRtcpPacket(xr);
@ -1241,6 +1250,7 @@ TEST_F(RtcpReceiverTest, HandlesIncorrectTargetBitrate) {
rtcp::ExtendedReports xr; rtcp::ExtendedReports xr;
xr.SetTargetBitrate(bitrate); xr.SetTargetBitrate(bitrate);
xr.SetSenderSsrc(kSenderSsrc);
EXPECT_CALL(bitrate_allocation_observer_, EXPECT_CALL(bitrate_allocation_observer_,
OnBitrateAllocationUpdated(expected_allocation)); OnBitrateAllocationUpdated(expected_allocation));