/* *  Copyright (c) 2018 The WebM 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 <tuple> #include "third_party/googletest/src/include/gtest/gtest.h" #include "test/codec_factory.h"#include "test/encode_test_driver.h"#include "test/util.h"#include "test/i420_video_source.h"#include "vpx_mem/vpx_mem.h" namespace { class DecodeCorruptedFrameTest    : public ::libvpx_test::EncoderTest,      public ::testing::TestWithParam<          std::tuple<const libvpx_test::CodecFactory *> > { public:  DecodeCorruptedFrameTest() : EncoderTest(GET_PARAM(0)) {}  protected:  virtual ~DecodeCorruptedFrameTest() {}   virtual void SetUp() {    InitializeConfig();    SetMode(::libvpx_test::kRealTime);    cfg_.g_lag_in_frames = 0;    cfg_.rc_end_usage = VPX_CBR;    cfg_.rc_buf_sz = 1000;    cfg_.rc_buf_initial_sz = 500;    cfg_.rc_buf_optimal_sz = 600;     // Set small key frame distance such that we insert more key frames.    cfg_.kf_max_dist = 3;    dec_cfg_.threads = 1;  }   virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video,                                  ::libvpx_test::Encoder *encoder) {    if (video->frame() == 0) encoder->Control(VP8E_SET_CPUUSED, 7);  }   virtual void MismatchHook(const vpx_image_t * /*img1*/,                            const vpx_image_t * /*img2*/) {}   virtual const vpx_codec_cx_pkt_t *MutateEncoderOutputHook(      const vpx_codec_cx_pkt_t *pkt) {    // Don't edit frame packet on key frame.    if (pkt->data.frame.flags & VPX_FRAME_IS_KEY) return pkt;    if (pkt->kind != VPX_CODEC_CX_FRAME_PKT) return pkt;     memcpy(&modified_pkt_, pkt, sizeof(*pkt));     // Halve the size so it's corrupted to decoder.    modified_pkt_.data.frame.sz = modified_pkt_.data.frame.sz / 2;     return &modified_pkt_;  }   virtual bool HandleDecodeResult(const vpx_codec_err_t res_dec,                                  const libvpx_test::VideoSource & /*video*/,                                  libvpx_test::Decoder *decoder) {    EXPECT_NE(res_dec, VPX_CODEC_MEM_ERROR) << decoder->DecodeError();    return VPX_CODEC_MEM_ERROR != res_dec;  }   vpx_codec_cx_pkt_t modified_pkt_;}; TEST_P(DecodeCorruptedFrameTest, DecodeCorruptedFrame) {  cfg_.rc_target_bitrate = 200;  cfg_.g_error_resilient = 0;   ::libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288,                                       30, 1, 0, 300);   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));} #if CONFIG_VP9INSTANTIATE_TEST_SUITE_P(    VP9, DecodeCorruptedFrameTest,    ::testing::Values(        static_cast<const libvpx_test::CodecFactory *>(&libvpx_test::kVP9)));#endif  // CONFIG_VP9 #if CONFIG_VP8INSTANTIATE_TEST_SUITE_P(    VP8, DecodeCorruptedFrameTest,    ::testing::Values(        static_cast<const libvpx_test::CodecFactory *>(&libvpx_test::kVP8)));#endif  // CONFIG_VP8 }  // namespace