Index: webrtc/video/end_to_end_tests.cc |
diff --git a/webrtc/video/end_to_end_tests.cc b/webrtc/video/end_to_end_tests.cc |
index f59044cf01ad61b561d12cf46227a7c223afc4de..525a0299e4d216bfe0208b2e8745a65a2e6badb1 100644 |
--- a/webrtc/video/end_to_end_tests.cc |
+++ b/webrtc/video/end_to_end_tests.cc |
@@ -26,6 +26,7 @@ |
#include "webrtc/modules/include/module_common_types.h" |
#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" |
#include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
+#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/rapid_resync_request.h" |
#include "webrtc/modules/rtp_rtcp/source/rtcp_utility.h" |
#include "webrtc/modules/video_coding/codecs/h264/include/h264.h" |
#include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" |
@@ -111,7 +112,7 @@ class EndToEndTest : public test::CallTest { |
void RespectsRtcpMode(RtcpMode rtcp_mode); |
void TestXrReceiverReferenceTimeReport(bool enable_rrtr); |
void TestSendsSetSsrcs(size_t num_ssrcs, bool send_single_ssrc_first); |
- void TestRtpStatePreservation(bool use_rtx); |
+ void TestRtpStatePreservation(bool use_rtx, bool wait_rtcp); |
void VerifyHistogramStats(bool use_rtx, bool use_red, bool screenshare); |
void VerifyNewVideoSendStreamsRespectNetworkState( |
MediaType network_to_bring_down, |
@@ -2871,7 +2872,7 @@ TEST_F(EndToEndTest, DISABLED_RedundantPayloadsTransmittedOnAllSsrcs) { |
RunBaseTest(&test); |
} |
-void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
+void EndToEndTest::TestRtpStatePreservation(bool use_rtx, bool wait_rtcp) { |
class RtpSequenceObserver : public test::RtpRtcpObserver { |
public: |
explicit RtpSequenceObserver(bool use_rtx) |
@@ -2891,6 +2892,23 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
} |
private: |
+ void ValidateTimestampGap(uint32_t ssrc, uint32_t timestamp) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_) { |
+ static const int32_t kMaxTimestampGap = kDefaultTimeoutMs * 90; |
+ auto timestamp_it = last_observed_timestamp_.find(ssrc); |
+ if (timestamp_it == last_observed_timestamp_.end()) { |
+ last_observed_timestamp_[ssrc] = timestamp; |
+ } else { |
+ // Verify timestamps are reasonably close. |
+ uint32_t latest_observed = timestamp_it->second; |
+ int32_t timestamp_gap = rtc::TimeDiff(timestamp, latest_observed); |
+ EXPECT_LE(std::abs(timestamp_gap), kMaxTimestampGap) |
+ << "Gap in timestamps (" << latest_observed << " -> " << timestamp |
+ << ") too large for SSRC: " << ssrc << "."; |
+ timestamp_it->second = timestamp; |
+ } |
+ } |
+ |
Action OnSendRtp(const uint8_t* packet, size_t length) override { |
RTPHeader header; |
EXPECT_TRUE(parser_->Parse(packet, length, &header)); |
@@ -2927,21 +2945,9 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
} |
} |
- static const int32_t kMaxTimestampGap = kDefaultTimeoutMs * 90; |
- auto timestamp_it = last_observed_timestamp_.find(ssrc); |
- if (timestamp_it == last_observed_timestamp_.end()) { |
- last_observed_timestamp_[ssrc] = timestamp; |
- } else { |
- // Verify timestamps are reasonably close. |
- uint32_t latest_observed = timestamp_it->second; |
- int32_t timestamp_gap = rtc::TimeDiff(timestamp, latest_observed); |
- EXPECT_LE(std::abs(timestamp_gap), kMaxTimestampGap) |
- << "Gap in timestamps (" << latest_observed << " -> " |
- << timestamp << ") too large for SSRC: " << ssrc << "."; |
- timestamp_it->second = timestamp; |
- } |
- |
rtc::CritScope lock(&crit_); |
+ ValidateTimestampGap(ssrc, timestamp); |
+ |
// Wait for media packets on all ssrcs. |
if (!ssrc_observed_[ssrc] && !only_padding) { |
ssrc_observed_[ssrc] = true; |
@@ -2952,6 +2958,19 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
return SEND_PACKET; |
} |
+ Action OnSendRtcp(const uint8_t* packet, size_t length) override { |
+ test::RtcpPacketParser rtcp_parser; |
+ rtcp_parser.Parse(packet, length); |
+ if (rtcp_parser.sender_report()->num_packets() > 0) { |
+ uint32_t ssrc = rtcp_parser.sender_report()->Ssrc(); |
+ uint32_t rtcp_timestamp = rtcp_parser.sender_report()->RtpTimestamp(); |
+ |
+ rtc::CritScope lock(&crit_); |
+ ValidateTimestampGap(ssrc, rtcp_timestamp); |
+ } |
+ return SEND_PACKET; |
+ } |
+ |
SequenceNumberUnwrapper seq_numbers_unwrapper_; |
std::map<uint32_t, std::list<int64_t>> last_observed_seq_numbers_; |
std::map<uint32_t, uint32_t> last_observed_timestamp_; |
@@ -3021,6 +3040,17 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
video_send_stream_ = |
sender_call_->CreateVideoSendStream(video_send_config_, one_stream); |
video_send_stream_->Start(); |
+ if (wait_rtcp) { |
+ // Wait for SR rtcp packet before generating rtp packets. |
+ // There should be no rtcp packet. |
+ |
+ // Rapid Resync Request forces sending RTCP Sender Report back. |
+ // Alternative approach is to wait several seconds for SR to be generated. |
+ rtcp::RapidResyncRequest rrr; |
pbos-webrtc
2016/06/16 11:39:43
call rrr something understandable
danilchap
2016/06/16 12:05:02
Done.
|
+ rtc::Buffer packet = rrr.Build(); |
+ static_cast<webrtc::test::DirectTransport&>(receive_transport) |
+ .SendRtcp(packet.data(), packet.size()); |
+ } |
CreateFrameGeneratorCapturer(); |
frame_generator_capturer_->Start(); |
@@ -3053,11 +3083,15 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
} |
TEST_F(EndToEndTest, RestartingSendStreamPreservesRtpState) { |
- TestRtpStatePreservation(false); |
+ TestRtpStatePreservation(false, false); |
} |
TEST_F(EndToEndTest, RestartingSendStreamPreservesRtpStatesWithRtx) { |
- TestRtpStatePreservation(true); |
+ TestRtpStatePreservation(true, false); |
+} |
+ |
+TEST_F(EndToEndTest, RestartingSendStreamKeepsRtpAndRtcpTimestampsSynced) { |
+ TestRtpStatePreservation(true, true); |
} |
TEST_F(EndToEndTest, RespectsNetworkState) { |