text: /*
   *  Copyright (c) 2014 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 "third_party/googletest/src/include/gtest/gtest.h"
  
  #include "./vpx_config.h"
  #include "test/ivf_video_source.h"
  #include "vpx/vp8dx.h"
  #include "vpx/vpx_decoder.h"
  
  namespace {
  
  #define NELEMENTS(x) static_cast<int>(sizeof(x) / sizeof(x[0]))
  
  TEST(DecodeAPI, InvalidParams) {
    static const vpx_codec_iface_t *kCodecs[] = {
  #if CONFIG_VP8_DECODER
      &vpx_codec_vp8_dx_algo,
  #endif
  #if CONFIG_VP9_DECODER
      &vpx_codec_vp9_dx_algo,
  #endif
    };
    uint8_t buf[1] = { 0 };
    vpx_codec_ctx_t dec;
  
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_dec_init(NULL, NULL, NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_dec_init(&dec, NULL, NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_decode(NULL, NULL, 0, NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_decode(NULL, buf, 0, NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
              vpx_codec_decode(NULL, buf, NELEMENTS(buf), NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
              vpx_codec_decode(NULL, NULL, NELEMENTS(buf), NULL, 0));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_destroy(NULL));
    EXPECT_TRUE(vpx_codec_error(NULL) != NULL);
  
    for (int i = 0; i < NELEMENTS(kCodecs); ++i) {
      EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
                vpx_codec_dec_init(NULL, kCodecs[i], NULL, 0));
  
      EXPECT_EQ(VPX_CODEC_OK, vpx_codec_dec_init(&dec, kCodecs[i], NULL, 0));
      EXPECT_EQ(VPX_CODEC_UNSUP_BITSTREAM,
                vpx_codec_decode(&dec, buf, NELEMENTS(buf), NULL, 0));
      EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
                vpx_codec_decode(&dec, NULL, NELEMENTS(buf), NULL, 0));
      EXPECT_EQ(VPX_CODEC_INVALID_PARAM, vpx_codec_decode(&dec, buf, 0, NULL, 0));
  
      EXPECT_EQ(VPX_CODEC_OK, vpx_codec_destroy(&dec));
    }
  }
  
  #if CONFIG_VP8_DECODER
  TEST(DecodeAPI, OptionalParams) {
    vpx_codec_ctx_t dec;
  
  #if CONFIG_ERROR_CONCEALMENT
    EXPECT_EQ(VPX_CODEC_OK, vpx_codec_dec_init(&dec, &vpx_codec_vp8_dx_algo, NULL,
                                               VPX_CODEC_USE_ERROR_CONCEALMENT));
  #else
    EXPECT_EQ(VPX_CODEC_INCAPABLE,
              vpx_codec_dec_init(&dec, &vpx_codec_vp8_dx_algo, NULL,
                                 VPX_CODEC_USE_ERROR_CONCEALMENT));
  #endif  // CONFIG_ERROR_CONCEALMENT
  }
  #endif  // CONFIG_VP8_DECODER
  
  #if CONFIG_VP9_DECODER
  // Test VP9 codec controls after a decode error to ensure the code doesn't
  // misbehave.
  void TestVp9Controls(vpx_codec_ctx_t *dec) {
    static const int kControls[] = { VP8D_GET_LAST_REF_UPDATES,
                                     VP8D_GET_FRAME_CORRUPTED,
                                     VP9D_GET_DISPLAY_SIZE, VP9D_GET_FRAME_SIZE };
    int val[2];
  
    for (int i = 0; i < NELEMENTS(kControls); ++i) {
      const vpx_codec_err_t res = vpx_codec_control_(dec, kControls[i], val);
      switch (kControls[i]) {
        case VP8D_GET_FRAME_CORRUPTED:
          EXPECT_EQ(VPX_CODEC_ERROR, res) << kControls[i];
          break;
        default: EXPECT_EQ(VPX_CODEC_OK, res) << kControls[i]; break;
      }
      EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
                vpx_codec_control_(dec, kControls[i], NULL));
    }
  
    vp9_ref_frame_t ref;
    ref.idx = 0;
    EXPECT_EQ(VPX_CODEC_ERROR, vpx_codec_control(dec, VP9_GET_REFERENCE, &ref));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
              vpx_codec_control(dec, VP9_GET_REFERENCE, NULL));
  
    vpx_ref_frame_t ref_copy;
    const int width = 352;
    const int height = 288;
    ASSERT_TRUE(
        vpx_img_alloc(&ref_copy.img, VPX_IMG_FMT_I420, width, height, 1) != NULL);
    ref_copy.frame_type = VP8_LAST_FRAME;
    EXPECT_EQ(VPX_CODEC_ERROR,
              vpx_codec_control(dec, VP8_COPY_REFERENCE, &ref_copy));
    EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
              vpx_codec_control(dec, VP8_COPY_REFERENCE, NULL));
    vpx_img_free(&ref_copy.img);
  }
  
  TEST(DecodeAPI, Vp9InvalidDecode) {
    const vpx_codec_iface_t *const codec = &vpx_codec_vp9_dx_algo;
    const char filename[] =
        "invalid-vp90-2-00-quantizer-00.webm.ivf.s5861_r01-05_b6-.v2.ivf";
    libvpx_test::IVFVideoSource video(filename);
    video.Init();
    video.Begin();
    ASSERT_TRUE(!HasFailure());
  
    vpx_codec_ctx_t dec;
    EXPECT_EQ(VPX_CODEC_OK, vpx_codec_dec_init(&dec, codec, NULL, 0));
    const uint32_t frame_size = static_cast<uint32_t>(video.frame_size());
  #if CONFIG_VP9_HIGHBITDEPTH
    EXPECT_EQ(VPX_CODEC_MEM_ERROR,
              vpx_codec_decode(&dec, video.cxdata(), frame_size, NULL, 0));
  #else
    EXPECT_EQ(VPX_CODEC_UNSUP_BITSTREAM,
              vpx_codec_decode(&dec, video.cxdata(), frame_size, NULL, 0));
  #endif
    vpx_codec_iter_t iter = NULL;
    EXPECT_EQ(NULL, vpx_codec_get_frame(&dec, &iter));
  
    TestVp9Controls(&dec);
    EXPECT_EQ(VPX_CODEC_OK, vpx_codec_destroy(&dec));
  }
  
  void TestPeekInfo(const uint8_t *const data, uint32_t data_sz,
                    uint32_t peek_size) {
    const vpx_codec_iface_t *const codec = &vpx_codec_vp9_dx_algo;
    // Verify behavior of vpx_codec_decode. vpx_codec_decode doesn't even get
    // to decoder_peek_si_internal on frames of size < 8.
    if (data_sz >= 8) {
      vpx_codec_ctx_t dec;
      EXPECT_EQ(VPX_CODEC_OK, vpx_codec_dec_init(&dec, codec, NULL, 0));
      EXPECT_EQ((data_sz < peek_size) ? VPX_CODEC_UNSUP_BITSTREAM
                                      : VPX_CODEC_CORRUPT_FRAME,
                vpx_codec_decode(&dec, data, data_sz, NULL, 0));
      vpx_codec_iter_t iter = NULL;
      EXPECT_EQ(NULL, vpx_codec_get_frame(&dec, &iter));
      EXPECT_EQ(VPX_CODEC_OK, vpx_codec_destroy(&dec));
    }
  
    // Verify behavior of vpx_codec_peek_stream_info.
    vpx_codec_stream_info_t si;
    si.sz = sizeof(si);
    EXPECT_EQ((data_sz < peek_size) ? VPX_CODEC_UNSUP_BITSTREAM : VPX_CODEC_OK,
              vpx_codec_peek_stream_info(codec, data, data_sz, &si));
  }
  
  TEST(DecodeAPI, Vp9PeekStreamInfo) {
    // The first 9 bytes are valid and the rest of the bytes are made up. Until
    // size 10, this should return VPX_CODEC_UNSUP_BITSTREAM and after that it
    // should return VPX_CODEC_CORRUPT_FRAME.
    const uint8_t data[32] = {
      0x85, 0xa4, 0xc1, 0xa1, 0x38, 0x81, 0xa3, 0x49, 0x83, 0xff, 0xff,
      0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
      0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    };
  
    for (uint32_t data_sz = 1; data_sz <= 32; ++data_sz) {
      TestPeekInfo(data, data_sz, 10);
    }
  }
  
  TEST(DecodeAPI, Vp9PeekStreamInfoTruncated) {
    // This profile 1 header requires 10.25 bytes, ensure
    // vpx_codec_peek_stream_info doesn't over read.
    const uint8_t profile1_data[10] = { 0xa4, 0xe9, 0x30, 0x68, 0x53,
                                        0xe9, 0x30, 0x68, 0x53, 0x04 };
  
    for (uint32_t data_sz = 1; data_sz <= 10; ++data_sz) {
      TestPeekInfo(profile1_data, data_sz, 11);
    }
  }
  #endif  // CONFIG_VP9_DECODER
  
  TEST(DecodeAPI, HighBitDepthCapability) {
  // VP8 should not claim VP9 HBD as a capability.
  #if CONFIG_VP8_DECODER
    const vpx_codec_caps_t vp8_caps = vpx_codec_get_caps(&vpx_codec_vp8_dx_algo);
    EXPECT_EQ(vp8_caps & VPX_CODEC_CAP_HIGHBITDEPTH, 0);
  #endif
  
  #if CONFIG_VP9_DECODER
    const vpx_codec_caps_t vp9_caps = vpx_codec_get_caps(&vpx_codec_vp9_dx_algo);
  #if CONFIG_VP9_HIGHBITDEPTH
    EXPECT_EQ(vp9_caps & VPX_CODEC_CAP_HIGHBITDEPTH, VPX_CODEC_CAP_HIGHBITDEPTH);
  #else
    EXPECT_EQ(vp9_caps & VPX_CODEC_CAP_HIGHBITDEPTH, 0);
  #endif
  #endif
  }
  
  }  // namespace
  