Change ReceiveStatistics to implement RtpPacketSinkInterface, part 1

Add new method OnRtpPacket, but leave
ReceiveStatisticsImpl::IncomingPacket and most of the implementation
unchanged. Deleting the old method and converting implementation from
RTPHeader to RtpPacketreceived is planned for a followup, after
downstream code is updated.

Bug: webrtc:7135, webrtc:8016
Change-Id: I697ec12804618859f8d69415622d1b957e1d0847
Reviewed-on: https://webrtc-review.googlesource.com/100104
Reviewed-by: Fredrik Solenberg <solenberg@webrtc.org>
Reviewed-by: Stefan Holmer <stefan@webrtc.org>
Reviewed-by: Danil Chapovalov <danilchap@webrtc.org>
Commit-Queue: Niels Moller <nisse@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#24889}
This commit is contained in:
Niels Möller
2018-09-14 08:26:32 +02:00
committed by Commit Bot
parent 2fbb83b16b
commit 1f3206cca4
17 changed files with 214 additions and 176 deletions

View File

@ -866,20 +866,22 @@ void Channel::OnRtpPacket(const RtpPacketReceived& packet) {
contributing_sources_.Update(now_ms, csrcs); contributing_sources_.Update(now_ms, csrcs);
} }
RTPHeader header;
packet.GetHeader(&header);
// Store playout timestamp for the received RTP packet // Store playout timestamp for the received RTP packet
UpdatePlayoutTimestamp(false); UpdatePlayoutTimestamp(false);
const auto& it = payload_type_frequencies_.find(header.payloadType); const auto& it = payload_type_frequencies_.find(packet.PayloadType());
if (it == payload_type_frequencies_.end()) if (it == payload_type_frequencies_.end())
return; return;
header.payload_type_frequency = it->second; // TODO(nisse): Set payload_type_frequency earlier, when packet is parsed.
RtpPacketReceived packet_copy(packet);
packet_copy.set_payload_type_frequency(it->second);
rtp_receive_statistics_->IncomingPacket(header, packet.size()); rtp_receive_statistics_->OnRtpPacket(packet_copy);
ReceivePacket(packet.data(), packet.size(), header); RTPHeader header;
packet_copy.GetHeader(&header);
ReceivePacket(packet_copy.data(), packet_copy.size(), header);
} }
bool Channel::ReceivePacket(const uint8_t* packet, bool Channel::ReceivePacket(const uint8_t* packet,

View File

@ -177,9 +177,7 @@ void FlexfecReceiveStreamImpl::OnRtpPacket(const RtpPacketReceived& packet) {
// Do not report media packets in the RTCP RRs generated by |rtp_rtcp_|. // Do not report media packets in the RTCP RRs generated by |rtp_rtcp_|.
if (packet.Ssrc() == config_.remote_ssrc) { if (packet.Ssrc() == config_.remote_ssrc) {
RTPHeader header; rtp_receive_statistics_->OnRtpPacket(packet);
packet.GetHeader(&header);
rtp_receive_statistics_->IncomingPacket(header, packet.size());
} }
} }

View File

@ -36,9 +36,7 @@ RtxReceiveStream::~RtxReceiveStream() = default;
void RtxReceiveStream::OnRtpPacket(const RtpPacketReceived& rtx_packet) { void RtxReceiveStream::OnRtpPacket(const RtpPacketReceived& rtx_packet) {
if (rtp_receive_statistics_) { if (rtp_receive_statistics_) {
RTPHeader header; rtp_receive_statistics_->OnRtpPacket(rtx_packet);
rtx_packet.GetHeader(&header);
rtp_receive_statistics_->IncomingPacket(header, rtx_packet.size());
} }
rtc::ArrayView<const uint8_t> payload = rtx_packet.payload(); rtc::ArrayView<const uint8_t> payload = rtx_packet.payload();

View File

@ -14,6 +14,7 @@
#include <sstream> #include <sstream>
#include "modules/rtp_rtcp/source/rtp_header_extensions.h"
#include "rtc_base/constructormagic.h" #include "rtc_base/constructormagic.h"
#include "rtc_base/numerics/safe_minmax.h" #include "rtc_base/numerics/safe_minmax.h"
#include "rtc_base/system/unused.h" #include "rtc_base/system/unused.h"
@ -137,6 +138,15 @@ void MediaPacket::SetAbsSendTimeMs(int64_t abs_send_time_ms) {
0x00fffffful; 0x00fffffful;
} }
// Copies payload size and sequence number.
RtpPacketReceived MediaPacket::GetRtpPacket() const {
RtpPacketReceived packet;
packet.SetPayloadSize(payload_size());
packet.SetSequenceNumber(header_.sequenceNumber);
return packet;
}
BbrBweFeedback::BbrBweFeedback( BbrBweFeedback::BbrBweFeedback(
int flow_id, int flow_id,
int64_t send_time_us, int64_t send_time_us,

View File

@ -57,8 +57,7 @@ void NadaBweReceiver::ReceivePacket(int64_t arrival_time_ms,
const int64_t kDelayMaxThresholdMs = 400; // Referred as d_max. const int64_t kDelayMaxThresholdMs = 400; // Referred as d_max.
clock_.AdvanceTimeMilliseconds(arrival_time_ms - clock_.TimeInMilliseconds()); clock_.AdvanceTimeMilliseconds(arrival_time_ms - clock_.TimeInMilliseconds());
recv_stats_->IncomingPacket(media_packet.header(), recv_stats_->OnRtpPacket(media_packet.GetRtpPacket());
media_packet.payload_size());
// Refered as x_n. // Refered as x_n.
int64_t delay_ms = arrival_time_ms - media_packet.sender_timestamp_ms(); int64_t delay_ms = arrival_time_ms - media_packet.sender_timestamp_ms();

View File

@ -84,8 +84,7 @@ RembReceiver::~RembReceiver() {}
void RembReceiver::ReceivePacket(int64_t arrival_time_ms, void RembReceiver::ReceivePacket(int64_t arrival_time_ms,
const MediaPacket& media_packet) { const MediaPacket& media_packet) {
recv_stats_->IncomingPacket(media_packet.header(), recv_stats_->OnRtpPacket(media_packet.GetRtpPacket());
media_packet.payload_size());
latest_estimate_bps_ = -1; latest_estimate_bps_ = -1;

View File

@ -19,6 +19,7 @@
#include "common_types.h" // NOLINT(build/include) #include "common_types.h" // NOLINT(build/include)
#include "modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h" #include "modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h"
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h" #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
namespace webrtc { namespace webrtc {
namespace testing { namespace testing {
@ -77,6 +78,7 @@ class MediaPacket : public Packet {
const RTPHeader& header() const { return header_; } const RTPHeader& header() const { return header_; }
Packet::Type GetPacketType() const override; Packet::Type GetPacketType() const override;
uint16_t sequence_number() const { return header_.sequenceNumber; } uint16_t sequence_number() const { return header_.sequenceNumber; }
RtpPacketReceived GetRtpPacket() const;
private: private:
static const int kAbsSendTimeFraction = 18; static const int kAbsSendTimeFraction = 18;

View File

@ -206,6 +206,7 @@ rtc_static_library("rtp_rtcp") {
"../../api/video:video_bitrate_allocator", "../../api/video:video_bitrate_allocator",
"../../api/video:video_frame", "../../api/video:video_frame",
"../../api/video_codecs:video_codecs_api", "../../api/video_codecs:video_codecs_api",
"../../call:rtp_interfaces",
"../../common_video", "../../common_video",
"../../logging:rtc_event_audio", "../../logging:rtc_event_audio",
"../../logging:rtc_event_log_api", "../../logging:rtc_event_log_api",

View File

@ -14,6 +14,7 @@
#include <map> #include <map>
#include <vector> #include <vector>
#include "call/rtp_packet_sink_interface.h"
#include "modules/include/module.h" #include "modules/include/module.h"
#include "modules/include/module_common_types.h" #include "modules/include/module_common_types.h"
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h" #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
@ -48,28 +49,22 @@ class StreamStatistician {
virtual uint32_t BitrateReceived() const = 0; virtual uint32_t BitrateReceived() const = 0;
}; };
class ReceiveStatistics : public ReceiveStatisticsProvider { class ReceiveStatistics : public ReceiveStatisticsProvider,
public RtpPacketSinkInterface {
public: public:
~ReceiveStatistics() override = default; ~ReceiveStatistics() override = default;
static ReceiveStatistics* Create(Clock* clock); static ReceiveStatistics* Create(Clock* clock);
// Updates the receive statistics with this packet. // Updates the receive statistics with this packet.
// TODO(bugs.webrtc.org/8016): Deprecated. Delete as soon as
// downstream code is updated to use OnRtpPacket.
RTC_DEPRECATED
virtual void IncomingPacket(const RTPHeader& rtp_header, virtual void IncomingPacket(const RTPHeader& rtp_header,
size_t packet_length) = 0; size_t packet_length) = 0;
// TODO(nisse): Wrapper for backwards compatibility. Delete as soon as
// downstream callers are updated.
RTC_DEPRECATED
void IncomingPacket(const RTPHeader& rtp_header,
size_t packet_length,
bool retransmitted) {
IncomingPacket(rtp_header, packet_length);
}
// Increment counter for number of FEC packets received. // Increment counter for number of FEC packets received.
virtual void FecPacketReceived(const RTPHeader& header, virtual void FecPacketReceived(const RtpPacketReceived& packet) = 0;
size_t packet_length) = 0;
// Returns a pointer to the statistician of an ssrc. // Returns a pointer to the statistician of an ssrc.
virtual StreamStatistician* GetStatistician(uint32_t ssrc) const = 0; virtual StreamStatistician* GetStatistician(uint32_t ssrc) const = 0;

View File

@ -16,6 +16,7 @@
#include <vector> #include <vector>
#include "modules/remote_bitrate_estimator/test/bwe_test_logging.h" #include "modules/remote_bitrate_estimator/test/bwe_test_logging.h"
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
#include "modules/rtp_rtcp/source/rtp_rtcp_config.h" #include "modules/rtp_rtcp/source/rtp_rtcp_config.h"
#include "modules/rtp_rtcp/source/time_util.h" #include "modules/rtp_rtcp/source/time_util.h"
#include "rtc_base/logging.h" #include "rtc_base/logging.h"
@ -372,6 +373,12 @@ ReceiveStatisticsImpl::~ReceiveStatisticsImpl() {
} }
} }
void ReceiveStatisticsImpl::OnRtpPacket(const RtpPacketReceived& packet) {
RTPHeader header;
packet.GetHeader(&header);
IncomingPacket(header, packet.size());
}
void ReceiveStatisticsImpl::IncomingPacket(const RTPHeader& header, void ReceiveStatisticsImpl::IncomingPacket(const RTPHeader& header,
size_t packet_length) { size_t packet_length) {
StreamStatisticianImpl* impl; StreamStatisticianImpl* impl;
@ -394,18 +401,19 @@ void ReceiveStatisticsImpl::IncomingPacket(const RTPHeader& header,
impl->IncomingPacket(header, packet_length); impl->IncomingPacket(header, packet_length);
} }
void ReceiveStatisticsImpl::FecPacketReceived(const RTPHeader& header, void ReceiveStatisticsImpl::FecPacketReceived(const RtpPacketReceived& packet) {
size_t packet_length) {
StreamStatisticianImpl* impl; StreamStatisticianImpl* impl;
{ {
rtc::CritScope cs(&receive_statistics_lock_); rtc::CritScope cs(&receive_statistics_lock_);
auto it = statisticians_.find(header.ssrc); auto it = statisticians_.find(packet.Ssrc());
// Ignore FEC if it is the first packet. // Ignore FEC if it is the first packet.
if (it == statisticians_.end()) if (it == statisticians_.end())
return; return;
impl = it->second; impl = it->second;
} }
impl->FecPacketReceived(header, packet_length); RTPHeader header;
packet.GetHeader(&header);
impl->FecPacketReceived(header, packet.size());
} }
StreamStatistician* ReceiveStatisticsImpl::GetStatistician( StreamStatistician* ReceiveStatisticsImpl::GetStatistician(

View File

@ -105,10 +105,12 @@ class ReceiveStatisticsImpl : public ReceiveStatistics,
// Implement ReceiveStatisticsProvider. // Implement ReceiveStatisticsProvider.
std::vector<rtcp::ReportBlock> RtcpReportBlocks(size_t max_blocks) override; std::vector<rtcp::ReportBlock> RtcpReportBlocks(size_t max_blocks) override;
// Implement RtpPacketSinkInterface
void OnRtpPacket(const RtpPacketReceived& packet) override;
// Implement ReceiveStatistics. // Implement ReceiveStatistics.
void IncomingPacket(const RTPHeader& header, size_t packet_length) override; void IncomingPacket(const RTPHeader& header, size_t packet_length) override;
void FecPacketReceived(const RTPHeader& header, void FecPacketReceived(const RtpPacketReceived& packet) override;
size_t packet_length) override;
StreamStatistician* GetStatistician(uint32_t ssrc) const override; StreamStatistician* GetStatistician(uint32_t ssrc) const override;
void SetMaxReorderingThreshold(int max_reordering_threshold) override; void SetMaxReorderingThreshold(int max_reordering_threshold) override;
void EnableRetransmitDetection(uint32_t ssrc, bool enable) override; void EnableRetransmitDetection(uint32_t ssrc, bool enable) override;

View File

@ -12,6 +12,8 @@
#include <vector> #include <vector>
#include "modules/rtp_rtcp/include/receive_statistics.h" #include "modules/rtp_rtcp/include/receive_statistics.h"
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
#include "rtc_base/random.h"
#include "system_wrappers/include/clock.h" #include "system_wrappers/include/clock.h"
#include "test/gmock.h" #include "test/gmock.h"
#include "test/gtest.h" #include "test/gtest.h"
@ -29,40 +31,71 @@ const uint32_t kSsrc2 = 202;
const uint32_t kSsrc3 = 203; const uint32_t kSsrc3 = 203;
const uint32_t kSsrc4 = 304; const uint32_t kSsrc4 = 304;
RTPHeader CreateRtpHeader(uint32_t ssrc) { RtpPacketReceived CreateRtpPacket(uint32_t ssrc,
RTPHeader header; size_t header_size,
memset(&header, 0, sizeof(header)); size_t payload_size,
header.ssrc = ssrc; size_t padding_size) {
header.sequenceNumber = 100; RtpPacketReceived packet;
header.payload_type_frequency = 90000; packet.SetSsrc(ssrc);
return header; packet.SetSequenceNumber(100);
packet.set_payload_type_frequency(90000);
RTC_CHECK_GE(header_size, 12);
RTC_CHECK_EQ(header_size % 4, 0);
if (header_size > 12) {
// Insert csrcs to increase header size.
const int num_csrcs = (header_size - 12) / 4;
std::vector<uint32_t> csrcs(num_csrcs);
packet.SetCsrcs(csrcs);
}
packet.SetPayloadSize(payload_size);
if (padding_size > 0) {
Random random(17);
packet.SetPadding(padding_size, &random);
}
return packet;
}
RtpPacketReceived CreateRtpPacket(uint32_t ssrc, size_t packet_size) {
return CreateRtpPacket(ssrc, 12, packet_size - 12, 0);
}
void IncrementSequenceNumber(RtpPacketReceived* packet, uint16_t incr) {
packet->SetSequenceNumber(packet->SequenceNumber() + incr);
}
void IncrementSequenceNumber(RtpPacketReceived* packet) {
IncrementSequenceNumber(packet, 1);
}
void IncrementTimestamp(RtpPacketReceived* packet, uint32_t incr) {
packet->SetTimestamp(packet->Timestamp() + incr);
} }
class ReceiveStatisticsTest : public ::testing::Test { class ReceiveStatisticsTest : public ::testing::Test {
public: public:
ReceiveStatisticsTest() ReceiveStatisticsTest()
: clock_(0), receive_statistics_(ReceiveStatistics::Create(&clock_)) { : clock_(0), receive_statistics_(ReceiveStatistics::Create(&clock_)) {
header1_ = CreateRtpHeader(kSsrc1); packet1_ = CreateRtpPacket(kSsrc1, kPacketSize1);
header2_ = CreateRtpHeader(kSsrc2); packet2_ = CreateRtpPacket(kSsrc2, kPacketSize2);
} }
protected: protected:
SimulatedClock clock_; SimulatedClock clock_;
std::unique_ptr<ReceiveStatistics> receive_statistics_; std::unique_ptr<ReceiveStatistics> receive_statistics_;
RTPHeader header1_; RtpPacketReceived packet1_;
RTPHeader header2_; RtpPacketReceived packet2_;
}; };
TEST_F(ReceiveStatisticsTest, TwoIncomingSsrcs) { TEST_F(ReceiveStatisticsTest, TwoIncomingSsrcs) {
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
receive_statistics_->IncomingPacket(header2_, kPacketSize2); receive_statistics_->OnRtpPacket(packet2_);
++header2_.sequenceNumber; IncrementSequenceNumber(&packet2_);
clock_.AdvanceTimeMilliseconds(100); clock_.AdvanceTimeMilliseconds(100);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
receive_statistics_->IncomingPacket(header2_, kPacketSize2); receive_statistics_->OnRtpPacket(packet2_);
++header2_.sequenceNumber; IncrementSequenceNumber(&packet2_);
StreamStatistician* statistician = StreamStatistician* statistician =
receive_statistics_->GetStatistician(kSsrc1); receive_statistics_->GetStatistician(kSsrc1);
@ -84,10 +117,10 @@ TEST_F(ReceiveStatisticsTest, TwoIncomingSsrcs) {
EXPECT_EQ(2u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(2u, receive_statistics_->RtcpReportBlocks(3).size());
// Add more incoming packets and verify that they are registered in both // Add more incoming packets and verify that they are registered in both
// access methods. // access methods.
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
receive_statistics_->IncomingPacket(header2_, kPacketSize2); receive_statistics_->OnRtpPacket(packet2_);
++header2_.sequenceNumber; IncrementSequenceNumber(&packet2_);
receive_statistics_->GetStatistician(kSsrc1)->GetDataCounters( receive_statistics_->GetStatistician(kSsrc1)->GetDataCounters(
&bytes_received, &packets_received); &bytes_received, &packets_received);
@ -101,12 +134,12 @@ TEST_F(ReceiveStatisticsTest, TwoIncomingSsrcs) {
TEST_F(ReceiveStatisticsTest, TEST_F(ReceiveStatisticsTest,
RtcpReportBlocksReturnsMaxBlocksWhenThereAreMoreStatisticians) { RtcpReportBlocksReturnsMaxBlocksWhenThereAreMoreStatisticians) {
RTPHeader header1 = CreateRtpHeader(kSsrc1); RtpPacketReceived packet1 = CreateRtpPacket(kSsrc1, kPacketSize1);
RTPHeader header2 = CreateRtpHeader(kSsrc2); RtpPacketReceived packet2 = CreateRtpPacket(kSsrc2, kPacketSize1);
RTPHeader header3 = CreateRtpHeader(kSsrc3); RtpPacketReceived packet3 = CreateRtpPacket(kSsrc3, kPacketSize1);
receive_statistics_->IncomingPacket(header1, kPacketSize1); receive_statistics_->OnRtpPacket(packet1);
receive_statistics_->IncomingPacket(header2, kPacketSize1); receive_statistics_->OnRtpPacket(packet2);
receive_statistics_->IncomingPacket(header3, kPacketSize1); receive_statistics_->OnRtpPacket(packet3);
EXPECT_THAT(receive_statistics_->RtcpReportBlocks(2), SizeIs(2)); EXPECT_THAT(receive_statistics_->RtcpReportBlocks(2), SizeIs(2));
EXPECT_THAT(receive_statistics_->RtcpReportBlocks(2), SizeIs(2)); EXPECT_THAT(receive_statistics_->RtcpReportBlocks(2), SizeIs(2));
@ -115,14 +148,14 @@ TEST_F(ReceiveStatisticsTest,
TEST_F(ReceiveStatisticsTest, TEST_F(ReceiveStatisticsTest,
RtcpReportBlocksReturnsAllObservedSsrcsWithMultipleCalls) { RtcpReportBlocksReturnsAllObservedSsrcsWithMultipleCalls) {
RTPHeader header1 = CreateRtpHeader(kSsrc1); RtpPacketReceived packet1 = CreateRtpPacket(kSsrc1, kPacketSize1);
RTPHeader header2 = CreateRtpHeader(kSsrc2); RtpPacketReceived packet2 = CreateRtpPacket(kSsrc2, kPacketSize1);
RTPHeader header3 = CreateRtpHeader(kSsrc3); RtpPacketReceived packet3 = CreateRtpPacket(kSsrc3, kPacketSize1);
RTPHeader header4 = CreateRtpHeader(kSsrc4); RtpPacketReceived packet4 = CreateRtpPacket(kSsrc4, kPacketSize1);
receive_statistics_->IncomingPacket(header1, kPacketSize1); receive_statistics_->OnRtpPacket(packet1);
receive_statistics_->IncomingPacket(header2, kPacketSize1); receive_statistics_->OnRtpPacket(packet2);
receive_statistics_->IncomingPacket(header3, kPacketSize1); receive_statistics_->OnRtpPacket(packet3);
receive_statistics_->IncomingPacket(header4, kPacketSize1); receive_statistics_->OnRtpPacket(packet4);
std::vector<uint32_t> observed_ssrcs; std::vector<uint32_t> observed_ssrcs;
std::vector<rtcp::ReportBlock> report_blocks = std::vector<rtcp::ReportBlock> report_blocks =
@ -141,11 +174,11 @@ TEST_F(ReceiveStatisticsTest,
} }
TEST_F(ReceiveStatisticsTest, ActiveStatisticians) { TEST_F(ReceiveStatisticsTest, ActiveStatisticians) {
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
clock_.AdvanceTimeMilliseconds(1000); clock_.AdvanceTimeMilliseconds(1000);
receive_statistics_->IncomingPacket(header2_, kPacketSize2); receive_statistics_->OnRtpPacket(packet2_);
++header2_.sequenceNumber; IncrementSequenceNumber(&packet2_);
// Nothing should time out since only 1000 ms has passed since the first // Nothing should time out since only 1000 ms has passed since the first
// packet came in. // packet came in.
EXPECT_EQ(2u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(2u, receive_statistics_->RtcpReportBlocks(3).size());
@ -158,8 +191,8 @@ TEST_F(ReceiveStatisticsTest, ActiveStatisticians) {
// kSsrc2 should have timed out. // kSsrc2 should have timed out.
EXPECT_EQ(0u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(0u, receive_statistics_->RtcpReportBlocks(3).size());
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
// kSsrc1 should be active again and the data counters should have survived. // kSsrc1 should be active again and the data counters should have survived.
EXPECT_EQ(1u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(1u, receive_statistics_->RtcpReportBlocks(3).size());
StreamStatistician* statistician = StreamStatistician* statistician =
@ -179,12 +212,12 @@ TEST_F(ReceiveStatisticsTest,
EXPECT_TRUE(receive_statistics_->GetStatistician(kSsrc1) != nullptr); EXPECT_TRUE(receive_statistics_->GetStatistician(kSsrc1) != nullptr);
EXPECT_EQ(0u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(0u, receive_statistics_->RtcpReportBlocks(3).size());
// Receive first packet // Receive first packet
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
EXPECT_EQ(1u, receive_statistics_->RtcpReportBlocks(3).size()); EXPECT_EQ(1u, receive_statistics_->RtcpReportBlocks(3).size());
} }
TEST_F(ReceiveStatisticsTest, GetReceiveStreamDataCounters) { TEST_F(ReceiveStatisticsTest, GetReceiveStreamDataCounters) {
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
StreamStatistician* statistician = StreamStatistician* statistician =
receive_statistics_->GetStatistician(kSsrc1); receive_statistics_->GetStatistician(kSsrc1);
ASSERT_TRUE(statistician != NULL); ASSERT_TRUE(statistician != NULL);
@ -194,7 +227,7 @@ TEST_F(ReceiveStatisticsTest, GetReceiveStreamDataCounters) {
EXPECT_GT(counters.first_packet_time_ms, -1); EXPECT_GT(counters.first_packet_time_ms, -1);
EXPECT_EQ(1u, counters.transmitted.packets); EXPECT_EQ(1u, counters.transmitted.packets);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
statistician->GetReceiveStreamDataCounters(&counters); statistician->GetReceiveStreamDataCounters(&counters);
EXPECT_GT(counters.first_packet_time_ms, -1); EXPECT_GT(counters.first_packet_time_ms, -1);
EXPECT_EQ(2u, counters.transmitted.packets); EXPECT_EQ(2u, counters.transmitted.packets);
@ -225,23 +258,23 @@ TEST_F(ReceiveStatisticsTest, RtcpCallbacks) {
receive_statistics_->EnableRetransmitDetection(kSsrc1, true); receive_statistics_->EnableRetransmitDetection(kSsrc1, true);
// Add some arbitrary data, with loss and jitter. // Add some arbitrary data, with loss and jitter.
header1_.sequenceNumber = 1; packet1_.SetSequenceNumber(1);
clock_.AdvanceTimeMilliseconds(7); clock_.AdvanceTimeMilliseconds(7);
header1_.timestamp += 3; IncrementTimestamp(&packet1_, 3);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
header1_.sequenceNumber += 2; IncrementSequenceNumber(&packet1_, 2);
clock_.AdvanceTimeMilliseconds(9); clock_.AdvanceTimeMilliseconds(9);
header1_.timestamp += 9; IncrementTimestamp(&packet1_, 9);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
--header1_.sequenceNumber; IncrementSequenceNumber(&packet1_, -1);
clock_.AdvanceTimeMilliseconds(13); clock_.AdvanceTimeMilliseconds(13);
header1_.timestamp += 47; IncrementTimestamp(&packet1_, 47);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
header1_.sequenceNumber += 3; IncrementSequenceNumber(&packet1_, 3);
clock_.AdvanceTimeMilliseconds(11); clock_.AdvanceTimeMilliseconds(11);
header1_.timestamp += 17; IncrementTimestamp(&packet1_, 17);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
EXPECT_EQ(0u, callback.num_calls_); EXPECT_EQ(0u, callback.num_calls_);
@ -265,23 +298,23 @@ TEST_F(ReceiveStatisticsTest, RtcpCallbacks) {
receive_statistics_->RegisterRtcpStatisticsCallback(NULL); receive_statistics_->RegisterRtcpStatisticsCallback(NULL);
// Add some more data. // Add some more data.
header1_.sequenceNumber = 1; packet1_.SetSequenceNumber(1);
clock_.AdvanceTimeMilliseconds(7); clock_.AdvanceTimeMilliseconds(7);
header1_.timestamp += 3; IncrementTimestamp(&packet1_, 3);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
header1_.sequenceNumber += 2; IncrementSequenceNumber(&packet1_, 2);
clock_.AdvanceTimeMilliseconds(9); clock_.AdvanceTimeMilliseconds(9);
header1_.timestamp += 9; IncrementTimestamp(&packet1_, 9);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
--header1_.sequenceNumber; IncrementSequenceNumber(&packet1_, -1);
clock_.AdvanceTimeMilliseconds(13); clock_.AdvanceTimeMilliseconds(13);
header1_.timestamp += 47; IncrementTimestamp(&packet1_, 47);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
header1_.sequenceNumber += 3; IncrementSequenceNumber(&packet1_, 3);
clock_.AdvanceTimeMilliseconds(11); clock_.AdvanceTimeMilliseconds(11);
header1_.timestamp += 17; IncrementTimestamp(&packet1_, 17);
receive_statistics_->IncomingPacket(header1_, kPacketSize1); receive_statistics_->OnRtpPacket(packet1_);
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1_);
receive_statistics_->GetStatistician(kSsrc1)->GetStatistics(&statistics, receive_statistics_->GetStatistician(kSsrc1)->GetStatistics(&statistics,
true); true);
@ -334,9 +367,10 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacks) {
const size_t kHeaderLength = 20; const size_t kHeaderLength = 20;
const size_t kPaddingLength = 9; const size_t kPaddingLength = 9;
// One packet of size kPacketSize1. // One packet with payload size kPacketSize1.
header1_.headerLength = kHeaderLength; RtpPacketReceived packet1 =
receive_statistics_->IncomingPacket(header1_, kPacketSize1 + kHeaderLength); CreateRtpPacket(kSsrc1, kHeaderLength, kPacketSize1, 0);
receive_statistics_->OnRtpPacket(packet1);
StreamDataCounters expected; StreamDataCounters expected;
expected.transmitted.payload_bytes = kPacketSize1; expected.transmitted.payload_bytes = kPacketSize1;
expected.transmitted.header_bytes = kHeaderLength; expected.transmitted.header_bytes = kHeaderLength;
@ -349,12 +383,12 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacks) {
expected.fec.packets = 0; expected.fec.packets = 0;
callback.Matches(1, kSsrc1, expected); callback.Matches(1, kSsrc1, expected);
++header1_.sequenceNumber;
clock_.AdvanceTimeMilliseconds(5);
header1_.paddingLength = 9;
// Another packet of size kPacketSize1 with 9 bytes padding. // Another packet of size kPacketSize1 with 9 bytes padding.
receive_statistics_->IncomingPacket( RtpPacketReceived packet2 =
header1_, kPacketSize1 + kHeaderLength + kPaddingLength); CreateRtpPacket(kSsrc1, kHeaderLength, kPacketSize1, 9);
packet2.SetSequenceNumber(packet1.SequenceNumber() + 1);
clock_.AdvanceTimeMilliseconds(5);
receive_statistics_->OnRtpPacket(packet2);
expected.transmitted.payload_bytes = kPacketSize1 * 2; expected.transmitted.payload_bytes = kPacketSize1 * 2;
expected.transmitted.header_bytes = kHeaderLength * 2; expected.transmitted.header_bytes = kHeaderLength * 2;
expected.transmitted.padding_bytes = kPaddingLength; expected.transmitted.padding_bytes = kPaddingLength;
@ -363,8 +397,7 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacks) {
clock_.AdvanceTimeMilliseconds(5); clock_.AdvanceTimeMilliseconds(5);
// Retransmit last packet. // Retransmit last packet.
receive_statistics_->IncomingPacket( receive_statistics_->OnRtpPacket(packet2);
header1_, kPacketSize1 + kHeaderLength + kPaddingLength);
expected.transmitted.payload_bytes = kPacketSize1 * 3; expected.transmitted.payload_bytes = kPacketSize1 * 3;
expected.transmitted.header_bytes = kHeaderLength * 3; expected.transmitted.header_bytes = kHeaderLength * 3;
expected.transmitted.padding_bytes = kPaddingLength * 2; expected.transmitted.padding_bytes = kPaddingLength * 2;
@ -375,13 +408,11 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacks) {
expected.retransmitted.packets = 1; expected.retransmitted.packets = 1;
callback.Matches(3, kSsrc1, expected); callback.Matches(3, kSsrc1, expected);
header1_.paddingLength = 0;
++header1_.sequenceNumber;
clock_.AdvanceTimeMilliseconds(5);
// One FEC packet. // One FEC packet.
receive_statistics_->IncomingPacket(header1_, kPacketSize1 + kHeaderLength); packet1.SetSequenceNumber(packet2.SequenceNumber() + 1);
receive_statistics_->FecPacketReceived(header1_, clock_.AdvanceTimeMilliseconds(5);
kPacketSize1 + kHeaderLength); receive_statistics_->OnRtpPacket(packet1);
receive_statistics_->FecPacketReceived(packet1);
expected.transmitted.payload_bytes = kPacketSize1 * 4; expected.transmitted.payload_bytes = kPacketSize1 * 4;
expected.transmitted.header_bytes = kHeaderLength * 4; expected.transmitted.header_bytes = kHeaderLength * 4;
expected.transmitted.packets = 4; expected.transmitted.packets = 4;
@ -393,9 +424,9 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacks) {
receive_statistics_->RegisterRtpStatisticsCallback(NULL); receive_statistics_->RegisterRtpStatisticsCallback(NULL);
// New stats, but callback should not be called. // New stats, but callback should not be called.
++header1_.sequenceNumber; IncrementSequenceNumber(&packet1);
clock_.AdvanceTimeMilliseconds(5); clock_.AdvanceTimeMilliseconds(5);
receive_statistics_->IncomingPacket(header1_, kPacketSize1 + kHeaderLength); receive_statistics_->OnRtpPacket(packet1);
callback.Matches(5, kSsrc1, expected); callback.Matches(5, kSsrc1, expected);
} }
@ -404,14 +435,13 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacksFecFirst) {
receive_statistics_->RegisterRtpStatisticsCallback(&callback); receive_statistics_->RegisterRtpStatisticsCallback(&callback);
const uint32_t kHeaderLength = 20; const uint32_t kHeaderLength = 20;
header1_.headerLength = kHeaderLength; RtpPacketReceived packet =
CreateRtpPacket(kSsrc1, kHeaderLength, kPacketSize1, 0);
// If first packet is FEC, ignore it. // If first packet is FEC, ignore it.
receive_statistics_->FecPacketReceived(header1_, receive_statistics_->FecPacketReceived(packet);
kPacketSize1 + kHeaderLength);
EXPECT_EQ(0u, callback.num_calls_); EXPECT_EQ(0u, callback.num_calls_);
receive_statistics_->IncomingPacket(header1_, kPacketSize1 + kHeaderLength); receive_statistics_->OnRtpPacket(packet);
StreamDataCounters expected; StreamDataCounters expected;
expected.transmitted.payload_bytes = kPacketSize1; expected.transmitted.payload_bytes = kPacketSize1;
expected.transmitted.header_bytes = kHeaderLength; expected.transmitted.header_bytes = kHeaderLength;
@ -420,8 +450,7 @@ TEST_F(ReceiveStatisticsTest, RtpCallbacksFecFirst) {
expected.fec.packets = 0; expected.fec.packets = 0;
callback.Matches(1, kSsrc1, expected); callback.Matches(1, kSsrc1, expected);
receive_statistics_->FecPacketReceived(header1_, receive_statistics_->FecPacketReceived(packet);
kPacketSize1 + kHeaderLength);
expected.fec.payload_bytes = kPacketSize1; expected.fec.payload_bytes = kPacketSize1;
expected.fec.header_bytes = kHeaderLength; expected.fec.header_bytes = kHeaderLength;
expected.fec.packets = 1; expected.fec.packets = 1;

View File

@ -14,6 +14,7 @@
#include "modules/rtp_rtcp/source/rtcp_packet/bye.h" #include "modules/rtp_rtcp/source/rtcp_packet/bye.h"
#include "modules/rtp_rtcp/source/rtcp_packet/common_header.h" #include "modules/rtp_rtcp/source/rtcp_packet/common_header.h"
#include "modules/rtp_rtcp/source/rtcp_sender.h" #include "modules/rtp_rtcp/source/rtcp_sender.h"
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
#include "modules/rtp_rtcp/source/rtp_rtcp_impl.h" #include "modules/rtp_rtcp/source/rtp_rtcp_impl.h"
#include "rtc_base/rate_limiter.h" #include "rtc_base/rate_limiter.h"
#include "test/gmock.h" #include "test/gmock.h"
@ -93,13 +94,12 @@ class RtcpSenderTest : public ::testing::Test {
} }
void InsertIncomingPacket(uint32_t remote_ssrc, uint16_t seq_num) { void InsertIncomingPacket(uint32_t remote_ssrc, uint16_t seq_num) {
RTPHeader header; RtpPacketReceived packet;
header.ssrc = remote_ssrc; packet.SetSsrc(remote_ssrc);
header.sequenceNumber = seq_num; packet.SetSequenceNumber(seq_num);
header.timestamp = 12345; packet.SetTimestamp(12345);
header.headerLength = 12; packet.SetPayloadSize(100 - 12);
size_t kPacketLength = 100; receive_statistics_->OnRtpPacket(packet);
receive_statistics_->IncomingPacket(header, kPacketLength);
} }
test::RtcpPacketParser* parser() { return &test_transport_.parser_; } test::RtcpPacketParser* parser() { return &test_transport_.parser_; }

View File

@ -17,6 +17,7 @@
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h" #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
#include "modules/rtp_rtcp/source/rtcp_packet.h" #include "modules/rtp_rtcp/source/rtcp_packet.h"
#include "modules/rtp_rtcp/source/rtcp_packet/nack.h" #include "modules/rtp_rtcp/source/rtcp_packet/nack.h"
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
#include "modules/rtp_rtcp/source/rtp_rtcp_impl.h" #include "modules/rtp_rtcp/source/rtp_rtcp_impl.h"
#include "rtc_base/rate_limiter.h" #include "rtc_base/rate_limiter.h"
#include "test/gmock.h" #include "test/gmock.h"
@ -318,12 +319,12 @@ TEST_F(RtpRtcpImplTest, SetSelectiveRetransmissions_HigherLayers) {
} }
TEST_F(RtpRtcpImplTest, Rtt) { TEST_F(RtpRtcpImplTest, Rtt) {
RTPHeader header; RtpPacketReceived packet;
header.timestamp = 1; packet.SetTimestamp(1);
header.sequenceNumber = 123; packet.SetSequenceNumber(123);
header.ssrc = kSenderSsrc; packet.SetSsrc(kSenderSsrc);
header.headerLength = 12; packet.AllocatePayload(100 - 12);
receiver_.receive_statistics_->IncomingPacket(header, 100); receiver_.receive_statistics_->OnRtpPacket(packet);
// Send Frame before sending an SR. // Send Frame before sending an SR.
SendFrame(&sender_, kBaseLayerTid); SendFrame(&sender_, kBaseLayerTid);

View File

@ -14,6 +14,7 @@
#include <memory> #include <memory>
#include <string> #include <string>
#include "modules/rtp_rtcp/source/rtp_packet_received.h"
#include "rtc_base/checks.h" #include "rtc_base/checks.h"
#include "rtc_base/rate_limiter.h" #include "rtc_base/rate_limiter.h"
#include "test/null_transport.h" #include "test/null_transport.h"
@ -43,22 +44,23 @@ bool LoopBackTransport::SendRtp(const uint8_t* data,
return true; return true;
} }
} }
RTPHeader header; RtpPacketReceived parsed_packet;
std::unique_ptr<RtpHeaderParser> parser(RtpHeaderParser::Create()); if (!parsed_packet.Parse(data, len)) {
if (!parser->Parse(data, len, &header)) {
return false; return false;
} }
const auto pl = const auto pl =
rtp_payload_registry_->PayloadTypeToPayload(header.payloadType); rtp_payload_registry_->PayloadTypeToPayload(parsed_packet.PayloadType());
if (!pl) { if (!pl) {
return false; return false;
} }
const uint8_t* payload = data + header.headerLength; receive_statistics_->OnRtpPacket(parsed_packet);
RTC_CHECK_GE(len, header.headerLength); RTPHeader header;
const size_t payload_length = len - header.headerLength; parsed_packet.GetHeader(&header);
receive_statistics_->IncomingPacket(header, len); return rtp_receiver_->IncomingRtpPacket(
return rtp_receiver_->IncomingRtpPacket(header, payload, payload_length, header, parsed_packet.payload().data(),
pl->typeSpecific); // RtpReceiver expects length including padding.
parsed_packet.payload_size() + parsed_packet.padding_size(),
pl->typeSpecific);
} }
bool LoopBackTransport::SendRtcp(const uint8_t* data, size_t len) { bool LoopBackTransport::SendRtcp(const uint8_t* data, size_t len) {

View File

@ -320,11 +320,7 @@ void RtpVideoStreamReceiver::OnRtpPacket(const RtpPacketReceived& packet) {
// Receive statistics will be reset if the payload type changes (make sure // Receive statistics will be reset if the payload type changes (make sure
// that the first packet is included in the stats). // that the first packet is included in the stats).
if (!packet.recovered()) { if (!packet.recovered()) {
RTPHeader header; rtp_receive_statistics_->OnRtpPacket(packet);
packet.GetHeader(&header);
// TODO(nisse): We should pass a recovered flag to stats, to aid
// fixing bug bugs.webrtc.org/6339.
rtp_receive_statistics_->IncomingPacket(header, packet.size());
} }
for (RtpPacketSinkInterface* secondary_sink : secondary_sinks_) { for (RtpPacketSinkInterface* secondary_sink : secondary_sinks_) {
@ -421,9 +417,7 @@ void RtpVideoStreamReceiver::ReceivePacket(const RtpPacketReceived& packet) {
return; return;
} }
if (packet.PayloadType() == config_.rtp.red_payload_type) { if (packet.PayloadType() == config_.rtp.red_payload_type) {
RTPHeader header; ParseAndHandleEncapsulatingHeader(packet);
packet.GetHeader(&header);
ParseAndHandleEncapsulatingHeader(packet.data(), packet.size(), header);
return; return;
} }
@ -509,21 +503,21 @@ void RtpVideoStreamReceiver::ReceivePacket(const RtpPacketReceived& packet) {
} }
void RtpVideoStreamReceiver::ParseAndHandleEncapsulatingHeader( void RtpVideoStreamReceiver::ParseAndHandleEncapsulatingHeader(
const uint8_t* packet, const RtpPacketReceived& packet) {
size_t packet_length,
const RTPHeader& header) {
RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_); RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_task_checker_);
if (header.payloadType == config_.rtp.red_payload_type && if (packet.PayloadType() == config_.rtp.red_payload_type &&
packet_length > header.headerLength + header.paddingLength) { packet.payload_size() > 0) {
if (packet[header.headerLength] == config_.rtp.ulpfec_payload_type) { if (packet.payload()[0] == config_.rtp.ulpfec_payload_type) {
rtp_receive_statistics_->FecPacketReceived(header, packet_length); rtp_receive_statistics_->FecPacketReceived(packet);
// Notify video_receiver about received FEC packets to avoid NACKing these // Notify video_receiver about received FEC packets to avoid NACKing these
// packets. // packets.
NotifyReceiverOfEmptyPacket(header.sequenceNumber); NotifyReceiverOfEmptyPacket(packet.SequenceNumber());
} }
RTPHeader header;
packet.GetHeader(&header);
if (ulpfec_receiver_->AddReceivedRedPacket( if (ulpfec_receiver_->AddReceivedRedPacket(
header, packet, packet_length, config_.rtp.ulpfec_payload_type) != header, packet.data(), packet.size(),
0) { config_.rtp.ulpfec_payload_type) != 0) {
return; return;
} }
ulpfec_receiver_->ProcessReceivedFec(); ulpfec_receiver_->ProcessReceivedFec();

View File

@ -143,9 +143,7 @@ class RtpVideoStreamReceiver : public RtpData,
void ReceivePacket(const RtpPacketReceived& packet); void ReceivePacket(const RtpPacketReceived& packet);
// Parses and handles RED headers. // Parses and handles RED headers.
// This function assumes that it's being called from only one thread. // This function assumes that it's being called from only one thread.
void ParseAndHandleEncapsulatingHeader(const uint8_t* packet, void ParseAndHandleEncapsulatingHeader(const RtpPacketReceived& packet);
size_t packet_length,
const RTPHeader& header);
void NotifyReceiverOfEmptyPacket(uint16_t seq_num); void NotifyReceiverOfEmptyPacket(uint16_t seq_num);
void UpdateHistograms(); void UpdateHistograms();
bool IsRedEnabled() const; bool IsRedEnabled() const;