Index: source/libvpx/test/decode_api_test.cc |
=================================================================== |
--- source/libvpx/test/decode_api_test.cc (revision 0) |
+++ source/libvpx/test/decode_api_test.cc (revision 0) |
@@ -0,0 +1,126 @@ |
+/* |
+ * 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 "test/ivf_video_source.h" |
+#include "./vpx_config.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_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, |
+ }; |
+ 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()); |
+ EXPECT_EQ(VPX_CODEC_MEM_ERROR, |
+ vpx_codec_decode(&dec, video.cxdata(), frame_size, NULL, 0)); |
+ 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)); |
+} |
+#endif // CONFIG_VP9_DECODER |
+ |
+} // namespace |