/* * Copyright 2017 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 "video/video_receive_stream2.h" #include #include #include #include #include "api/task_queue/default_task_queue_factory.h" #include "api/test/video/function_video_decoder_factory.h" #include "api/video_codecs/video_decoder.h" #include "call/rtp_stream_receiver_controller.h" #include "common_video/test/utilities.h" #include "media/base/fake_video_renderer.h" #include "modules/pacing/packet_router.h" #include "modules/rtp_rtcp/source/rtp_packet_to_send.h" #include "modules/utility/include/process_thread.h" #include "modules/video_coding/encoded_frame.h" #include "rtc_base/event.h" #include "system_wrappers/include/clock.h" #include "test/fake_decoder.h" #include "test/field_trial.h" #include "test/gmock.h" #include "test/gtest.h" #include "test/run_loop.h" #include "test/time_controller/simulated_time_controller.h" #include "test/video_decoder_proxy_factory.h" #include "video/call_stats2.h" namespace webrtc { namespace { using ::testing::_; using ::testing::ElementsAreArray; using ::testing::Invoke; using ::testing::IsEmpty; using ::testing::SizeIs; constexpr int kDefaultTimeOutMs = 50; class MockTransport : public Transport { public: MOCK_METHOD(bool, SendRtp, (const uint8_t*, size_t length, const PacketOptions& options), (override)); MOCK_METHOD(bool, SendRtcp, (const uint8_t*, size_t length), (override)); }; class MockVideoDecoder : public VideoDecoder { public: MOCK_METHOD(int32_t, InitDecode, (const VideoCodec*, int32_t number_of_cores), (override)); MOCK_METHOD(int32_t, Decode, (const EncodedImage& input, bool missing_frames, int64_t render_time_ms), (override)); MOCK_METHOD(int32_t, RegisterDecodeCompleteCallback, (DecodedImageCallback*), (override)); MOCK_METHOD(int32_t, Release, (), (override)); const char* ImplementationName() const { return "MockVideoDecoder"; } }; class FrameObjectFake : public video_coding::EncodedFrame { public: void SetPayloadType(uint8_t payload_type) { _payloadType = payload_type; } void SetRotation(const VideoRotation& rotation) { rotation_ = rotation; } void SetNtpTime(int64_t ntp_time_ms) { ntp_time_ms_ = ntp_time_ms; } int64_t ReceivedTime() const override { return 0; } int64_t RenderTime() const override { return _renderTimeMs; } }; } // namespace class VideoReceiveStream2Test : public ::testing::Test { public: VideoReceiveStream2Test() : process_thread_(ProcessThread::Create("TestThread")), task_queue_factory_(CreateDefaultTaskQueueFactory()), config_(&mock_transport_), call_stats_(Clock::GetRealTimeClock(), loop_.task_queue()), h264_decoder_factory_(&mock_h264_video_decoder_), null_decoder_factory_(&mock_null_video_decoder_) {} void SetUp() { constexpr int kDefaultNumCpuCores = 2; config_.rtp.remote_ssrc = 1111; config_.rtp.local_ssrc = 2222; config_.renderer = &fake_renderer_; VideoReceiveStream::Decoder h264_decoder; h264_decoder.payload_type = 99; h264_decoder.video_format = SdpVideoFormat("H264"); h264_decoder.video_format.parameters.insert( {"sprop-parameter-sets", "Z0IACpZTBYmI,aMljiA=="}); h264_decoder.decoder_factory = &h264_decoder_factory_; config_.decoders.push_back(h264_decoder); VideoReceiveStream::Decoder null_decoder; null_decoder.payload_type = 98; null_decoder.video_format = SdpVideoFormat("null"); null_decoder.decoder_factory = &null_decoder_factory_; config_.decoders.push_back(null_decoder); clock_ = Clock::GetRealTimeClock(); timing_ = new VCMTiming(clock_); video_receive_stream_ = std::make_unique( task_queue_factory_.get(), loop_.task_queue(), &rtp_stream_receiver_controller_, kDefaultNumCpuCores, &packet_router_, config_.Copy(), process_thread_.get(), &call_stats_, clock_, timing_); } protected: test::RunLoop loop_; std::unique_ptr process_thread_; const std::unique_ptr task_queue_factory_; VideoReceiveStream::Config config_; internal::CallStats call_stats_; MockVideoDecoder mock_h264_video_decoder_; MockVideoDecoder mock_null_video_decoder_; test::VideoDecoderProxyFactory h264_decoder_factory_; test::VideoDecoderProxyFactory null_decoder_factory_; cricket::FakeVideoRenderer fake_renderer_; MockTransport mock_transport_; PacketRouter packet_router_; RtpStreamReceiverController rtp_stream_receiver_controller_; std::unique_ptr video_receive_stream_; Clock* clock_; VCMTiming* timing_; }; TEST_F(VideoReceiveStream2Test, CreateFrameFromH264FmtpSpropAndIdr) { constexpr uint8_t idr_nalu[] = {0x05, 0xFF, 0xFF, 0xFF}; RtpPacketToSend rtppacket(nullptr); uint8_t* payload = rtppacket.AllocatePayload(sizeof(idr_nalu)); memcpy(payload, idr_nalu, sizeof(idr_nalu)); rtppacket.SetMarker(true); rtppacket.SetSsrc(1111); rtppacket.SetPayloadType(99); rtppacket.SetSequenceNumber(1); rtppacket.SetTimestamp(0); rtc::Event init_decode_event_; EXPECT_CALL(mock_h264_video_decoder_, InitDecode(_, _)) .WillOnce(Invoke([&init_decode_event_](const VideoCodec* config, int32_t number_of_cores) { init_decode_event_.Set(); return 0; })); EXPECT_CALL(mock_h264_video_decoder_, RegisterDecodeCompleteCallback(_)); video_receive_stream_->Start(); EXPECT_CALL(mock_h264_video_decoder_, Decode(_, false, _)); RtpPacketReceived parsed_packet; ASSERT_TRUE(parsed_packet.Parse(rtppacket.data(), rtppacket.size())); rtp_stream_receiver_controller_.OnRtpPacket(parsed_packet); EXPECT_CALL(mock_h264_video_decoder_, Release()); // Make sure the decoder thread had a chance to run. init_decode_event_.Wait(kDefaultTimeOutMs); } TEST_F(VideoReceiveStream2Test, PlayoutDelay) { const PlayoutDelay kPlayoutDelayMs = {123, 321}; std::unique_ptr test_frame(new FrameObjectFake()); test_frame->id.picture_id = 0; test_frame->SetPlayoutDelay(kPlayoutDelayMs); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); EXPECT_EQ(kPlayoutDelayMs.min_ms, timing_->min_playout_delay()); EXPECT_EQ(kPlayoutDelayMs.max_ms, timing_->max_playout_delay()); // Check that the biggest minimum delay is chosen. video_receive_stream_->SetMinimumPlayoutDelay(400); EXPECT_EQ(400, timing_->min_playout_delay()); // Check base minimum delay validation. EXPECT_FALSE(video_receive_stream_->SetBaseMinimumPlayoutDelayMs(12345)); EXPECT_FALSE(video_receive_stream_->SetBaseMinimumPlayoutDelayMs(-1)); EXPECT_TRUE(video_receive_stream_->SetBaseMinimumPlayoutDelayMs(500)); EXPECT_EQ(500, timing_->min_playout_delay()); // Check that intermidiate values are remembered and the biggest remembered // is chosen. video_receive_stream_->SetBaseMinimumPlayoutDelayMs(0); EXPECT_EQ(400, timing_->min_playout_delay()); video_receive_stream_->SetMinimumPlayoutDelay(0); EXPECT_EQ(123, timing_->min_playout_delay()); } TEST_F(VideoReceiveStream2Test, PlayoutDelayPreservesDefaultMaxValue) { const int default_max_playout_latency = timing_->max_playout_delay(); const PlayoutDelay kPlayoutDelayMs = {123, -1}; std::unique_ptr test_frame(new FrameObjectFake()); test_frame->id.picture_id = 0; test_frame->SetPlayoutDelay(kPlayoutDelayMs); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); // Ensure that -1 preserves default maximum value from |timing_|. EXPECT_EQ(kPlayoutDelayMs.min_ms, timing_->min_playout_delay()); EXPECT_NE(kPlayoutDelayMs.max_ms, timing_->max_playout_delay()); EXPECT_EQ(default_max_playout_latency, timing_->max_playout_delay()); } TEST_F(VideoReceiveStream2Test, PlayoutDelayPreservesDefaultMinValue) { const int default_min_playout_latency = timing_->min_playout_delay(); const PlayoutDelay kPlayoutDelayMs = {-1, 321}; std::unique_ptr test_frame(new FrameObjectFake()); test_frame->id.picture_id = 0; test_frame->SetPlayoutDelay(kPlayoutDelayMs); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); // Ensure that -1 preserves default minimum value from |timing_|. EXPECT_NE(kPlayoutDelayMs.min_ms, timing_->min_playout_delay()); EXPECT_EQ(kPlayoutDelayMs.max_ms, timing_->max_playout_delay()); EXPECT_EQ(default_min_playout_latency, timing_->min_playout_delay()); } class VideoReceiveStream2TestWithFakeDecoder : public ::testing::Test { public: VideoReceiveStream2TestWithFakeDecoder() : fake_decoder_factory_( []() { return std::make_unique(); }), process_thread_(ProcessThread::Create("TestThread")), task_queue_factory_(CreateDefaultTaskQueueFactory()), config_(&mock_transport_), call_stats_(Clock::GetRealTimeClock(), loop_.task_queue()) {} void SetUp() { config_.rtp.remote_ssrc = 1111; config_.rtp.local_ssrc = 2222; config_.renderer = &fake_renderer_; VideoReceiveStream::Decoder fake_decoder; fake_decoder.payload_type = 99; fake_decoder.video_format = SdpVideoFormat("VP8"); fake_decoder.decoder_factory = &fake_decoder_factory_; config_.decoders.push_back(fake_decoder); clock_ = Clock::GetRealTimeClock(); ReCreateReceiveStream(VideoReceiveStream::RecordingState()); } void ReCreateReceiveStream(VideoReceiveStream::RecordingState state) { constexpr int kDefaultNumCpuCores = 2; video_receive_stream_ = nullptr; timing_ = new VCMTiming(clock_); video_receive_stream_.reset(new webrtc::internal::VideoReceiveStream2( task_queue_factory_.get(), loop_.task_queue(), &rtp_stream_receiver_controller_, kDefaultNumCpuCores, &packet_router_, config_.Copy(), process_thread_.get(), &call_stats_, clock_, timing_)); video_receive_stream_->SetAndGetRecordingState(std::move(state), false); } protected: test::RunLoop loop_; test::FunctionVideoDecoderFactory fake_decoder_factory_; std::unique_ptr process_thread_; const std::unique_ptr task_queue_factory_; VideoReceiveStream::Config config_; internal::CallStats call_stats_; cricket::FakeVideoRenderer fake_renderer_; MockTransport mock_transport_; PacketRouter packet_router_; RtpStreamReceiverController rtp_stream_receiver_controller_; std::unique_ptr video_receive_stream_; Clock* clock_; VCMTiming* timing_; }; TEST_F(VideoReceiveStream2TestWithFakeDecoder, PassesNtpTime) { const int64_t kNtpTimestamp = 12345; auto test_frame = std::make_unique(); test_frame->SetPayloadType(99); test_frame->id.picture_id = 0; test_frame->SetNtpTime(kNtpTimestamp); video_receive_stream_->Start(); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); EXPECT_TRUE(fake_renderer_.WaitForRenderedFrame(kDefaultTimeOutMs)); EXPECT_EQ(kNtpTimestamp, fake_renderer_.ntp_time_ms()); } TEST_F(VideoReceiveStream2TestWithFakeDecoder, PassesRotation) { const webrtc::VideoRotation kRotation = webrtc::kVideoRotation_180; auto test_frame = std::make_unique(); test_frame->SetPayloadType(99); test_frame->id.picture_id = 0; test_frame->SetRotation(kRotation); video_receive_stream_->Start(); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); EXPECT_TRUE(fake_renderer_.WaitForRenderedFrame(kDefaultTimeOutMs)); EXPECT_EQ(kRotation, fake_renderer_.rotation()); } TEST_F(VideoReceiveStream2TestWithFakeDecoder, PassesPacketInfos) { auto test_frame = std::make_unique(); test_frame->SetPayloadType(99); test_frame->id.picture_id = 0; RtpPacketInfos packet_infos = CreatePacketInfos(3); test_frame->SetPacketInfos(packet_infos); video_receive_stream_->Start(); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); EXPECT_TRUE(fake_renderer_.WaitForRenderedFrame(kDefaultTimeOutMs)); EXPECT_THAT(fake_renderer_.packet_infos(), ElementsAreArray(packet_infos)); } TEST_F(VideoReceiveStream2TestWithFakeDecoder, RenderedFrameUpdatesGetSources) { constexpr uint32_t kSsrc = 1111; constexpr uint32_t kCsrc = 9001; constexpr uint32_t kRtpTimestamp = 12345; // Prepare one video frame with per-packet information. auto test_frame = std::make_unique(); test_frame->SetPayloadType(99); test_frame->id.picture_id = 0; RtpPacketInfos packet_infos; { RtpPacketInfos::vector_type infos; RtpPacketInfo info; info.set_ssrc(kSsrc); info.set_csrcs({kCsrc}); info.set_rtp_timestamp(kRtpTimestamp); info.set_receive_time_ms(clock_->TimeInMilliseconds() - 5000); infos.push_back(info); info.set_receive_time_ms(clock_->TimeInMilliseconds() - 3000); infos.push_back(info); info.set_receive_time_ms(clock_->TimeInMilliseconds() - 2000); infos.push_back(info); info.set_receive_time_ms(clock_->TimeInMilliseconds() - 4000); infos.push_back(info); packet_infos = RtpPacketInfos(std::move(infos)); } test_frame->SetPacketInfos(packet_infos); // Start receive stream. video_receive_stream_->Start(); EXPECT_THAT(video_receive_stream_->GetSources(), IsEmpty()); // Render one video frame. int64_t timestamp_ms_min = clock_->TimeInMilliseconds(); video_receive_stream_->OnCompleteFrame(std::move(test_frame)); EXPECT_TRUE(fake_renderer_.WaitForRenderedFrame(kDefaultTimeOutMs)); int64_t timestamp_ms_max = clock_->TimeInMilliseconds(); // Verify that the per-packet information is passed to the renderer. EXPECT_THAT(fake_renderer_.packet_infos(), ElementsAreArray(packet_infos)); // Verify that the per-packet information also updates |GetSources()|. std::vector sources = video_receive_stream_->GetSources(); ASSERT_THAT(sources, SizeIs(2)); { auto it = std::find_if(sources.begin(), sources.end(), [](const RtpSource& source) { return source.source_type() == RtpSourceType::SSRC; }); ASSERT_NE(it, sources.end()); EXPECT_EQ(it->source_id(), kSsrc); EXPECT_EQ(it->source_type(), RtpSourceType::SSRC); EXPECT_EQ(it->rtp_timestamp(), kRtpTimestamp); EXPECT_GE(it->timestamp_ms(), timestamp_ms_min); EXPECT_LE(it->timestamp_ms(), timestamp_ms_max); } { auto it = std::find_if(sources.begin(), sources.end(), [](const RtpSource& source) { return source.source_type() == RtpSourceType::CSRC; }); ASSERT_NE(it, sources.end()); EXPECT_EQ(it->source_id(), kCsrc); EXPECT_EQ(it->source_type(), RtpSourceType::CSRC); EXPECT_EQ(it->rtp_timestamp(), kRtpTimestamp); EXPECT_GE(it->timestamp_ms(), timestamp_ms_min); EXPECT_LE(it->timestamp_ms(), timestamp_ms_max); } } std::unique_ptr MakeFrame(VideoFrameType frame_type, int picture_id) { auto frame = std::make_unique(); frame->SetPayloadType(99); frame->id.picture_id = picture_id; frame->SetFrameType(frame_type); return frame; } TEST_F(VideoReceiveStream2TestWithFakeDecoder, PassesFrameWhenEncodedFramesCallbackSet) { testing::MockFunction callback; video_receive_stream_->Start(); // Expect a keyframe request to be generated EXPECT_CALL(mock_transport_, SendRtcp); EXPECT_CALL(callback, Call); video_receive_stream_->SetAndGetRecordingState( VideoReceiveStream::RecordingState(callback.AsStdFunction()), true); video_receive_stream_->OnCompleteFrame( MakeFrame(VideoFrameType::kVideoFrameKey, 0)); EXPECT_TRUE(fake_renderer_.WaitForRenderedFrame(kDefaultTimeOutMs)); video_receive_stream_->Stop(); } TEST_F(VideoReceiveStream2TestWithFakeDecoder, MovesEncodedFrameDispatchStateWhenReCreating) { testing::MockFunction callback; video_receive_stream_->Start(); // Expect a key frame request over RTCP. EXPECT_CALL(mock_transport_, SendRtcp).Times(1); video_receive_stream_->SetAndGetRecordingState( VideoReceiveStream::RecordingState(callback.AsStdFunction()), true); video_receive_stream_->Stop(); VideoReceiveStream::RecordingState old_state = video_receive_stream_->SetAndGetRecordingState( VideoReceiveStream::RecordingState(), false); ReCreateReceiveStream(std::move(old_state)); video_receive_stream_->Stop(); } class VideoReceiveStream2TestWithSimulatedClock : public ::testing::Test { public: class FakeDecoder2 : public test::FakeDecoder { public: explicit FakeDecoder2(std::function decode_callback) : callback_(decode_callback) {} int32_t Decode(const EncodedImage& input, bool missing_frames, int64_t render_time_ms) override { int32_t result = FakeDecoder::Decode(input, missing_frames, render_time_ms); callback_(); return result; } private: std::function callback_; }; static VideoReceiveStream::Config GetConfig( Transport* transport, VideoDecoderFactory* decoder_factory, rtc::VideoSinkInterface* renderer) { VideoReceiveStream::Config config(transport); config.rtp.remote_ssrc = 1111; config.rtp.local_ssrc = 2222; config.renderer = renderer; VideoReceiveStream::Decoder fake_decoder; fake_decoder.payload_type = 99; fake_decoder.video_format = SdpVideoFormat("VP8"); fake_decoder.decoder_factory = decoder_factory; config.decoders.push_back(fake_decoder); return config; } VideoReceiveStream2TestWithSimulatedClock() : time_controller_(Timestamp::Millis(4711)), fake_decoder_factory_([this] { return std::make_unique([this] { OnFrameDecoded(); }); }), process_thread_(time_controller_.CreateProcessThread("ProcessThread")), config_(GetConfig(&mock_transport_, &fake_decoder_factory_, &fake_renderer_)), call_stats_(time_controller_.GetClock(), loop_.task_queue()), video_receive_stream_(time_controller_.GetTaskQueueFactory(), loop_.task_queue(), &rtp_stream_receiver_controller_, /*num_cores=*/2, &packet_router_, config_.Copy(), process_thread_.get(), &call_stats_, time_controller_.GetClock(), new VCMTiming(time_controller_.GetClock())) { video_receive_stream_.Start(); } void OnFrameDecoded() { event_->Set(); } void PassEncodedFrameAndWait( std::unique_ptr frame) { event_ = std::make_unique(); // This call will eventually end up in the Decoded method where the // event is set. video_receive_stream_.OnCompleteFrame(std::move(frame)); event_->Wait(rtc::Event::kForever); } protected: GlobalSimulatedTimeController time_controller_; test::RunLoop loop_; test::FunctionVideoDecoderFactory fake_decoder_factory_; std::unique_ptr process_thread_; MockTransport mock_transport_; cricket::FakeVideoRenderer fake_renderer_; VideoReceiveStream::Config config_; internal::CallStats call_stats_; PacketRouter packet_router_; RtpStreamReceiverController rtp_stream_receiver_controller_; webrtc::internal::VideoReceiveStream2 video_receive_stream_; std::unique_ptr event_; }; TEST_F(VideoReceiveStream2TestWithSimulatedClock, RequestsKeyFramesUntilKeyFrameReceived) { auto tick = TimeDelta::Millis( internal::VideoReceiveStream2::kMaxWaitForKeyFrameMs / 2); EXPECT_CALL(mock_transport_, SendRtcp).Times(1).WillOnce(Invoke([this]() { loop_.Quit(); return 0; })); video_receive_stream_.GenerateKeyFrame(); PassEncodedFrameAndWait(MakeFrame(VideoFrameType::kVideoFrameDelta, 0)); time_controller_.AdvanceTime(tick); PassEncodedFrameAndWait(MakeFrame(VideoFrameType::kVideoFrameDelta, 1)); loop_.Run(); testing::Mock::VerifyAndClearExpectations(&mock_transport_); // T+200ms: still no key frame received, expect key frame request sent again. EXPECT_CALL(mock_transport_, SendRtcp).Times(1).WillOnce(Invoke([this]() { loop_.Quit(); return 0; })); time_controller_.AdvanceTime(tick); PassEncodedFrameAndWait(MakeFrame(VideoFrameType::kVideoFrameDelta, 2)); loop_.Run(); testing::Mock::VerifyAndClearExpectations(&mock_transport_); // T+200ms: now send a key frame - we should not observe new key frame // requests after this. EXPECT_CALL(mock_transport_, SendRtcp).Times(0); PassEncodedFrameAndWait(MakeFrame(VideoFrameType::kVideoFrameKey, 3)); time_controller_.AdvanceTime(2 * tick); PassEncodedFrameAndWait(MakeFrame(VideoFrameType::kVideoFrameDelta, 4)); loop_.PostTask([this]() { loop_.Quit(); }); loop_.Run(); } } // namespace webrtc