| Index: source/libvpx/test/vp8_decrypt_test.cc
|
| ===================================================================
|
| --- source/libvpx/test/vp8_decrypt_test.cc (revision 0)
|
| +++ source/libvpx/test/vp8_decrypt_test.cc (revision 0)
|
| @@ -0,0 +1,65 @@
|
| +/*
|
| + * Copyright (c) 2013 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 <cstdio>
|
| +#include <cstdlib>
|
| +#include <string>
|
| +#include "third_party/googletest/src/include/gtest/gtest.h"
|
| +#include "test/decode_test_driver.h"
|
| +#include "test/ivf_video_source.h"
|
| +
|
| +#if CONFIG_DECRYPT
|
| +
|
| +namespace {
|
| +
|
| +const uint8_t decrypt_key[32] = {
|
| + 255, 0, 0, 0, 0, 0, 0, 0,
|
| + 0, 0, 0, 0, 0, 0, 0, 0,
|
| + 0, 0, 0, 0, 0, 0, 0, 0,
|
| + 0, 0, 0, 0, 0, 0, 0, 0,
|
| +};
|
| +
|
| +} // namespace
|
| +
|
| +namespace libvpx_test {
|
| +
|
| +TEST(TestDecrypt, NullKey) {
|
| + vpx_codec_dec_cfg_t cfg = {0};
|
| + vpx_codec_ctx_t decoder = {0};
|
| + vpx_codec_err_t res = vpx_codec_dec_init(&decoder, &vpx_codec_vp8_dx_algo,
|
| + &cfg, 0);
|
| + ASSERT_EQ(VPX_CODEC_OK, res);
|
| +
|
| + res = vpx_codec_control(&decoder, VP8_SET_DECRYPT_KEY, NULL);
|
| + ASSERT_EQ(VPX_CODEC_INVALID_PARAM, res);
|
| +}
|
| +
|
| +TEST(TestDecrypt, DecryptWorks) {
|
| + libvpx_test::IVFVideoSource video("vp80-00-comprehensive-001.ivf");
|
| + video.Init();
|
| +
|
| + vpx_codec_dec_cfg_t dec_cfg = {0};
|
| + Decoder decoder(dec_cfg, 0);
|
| +
|
| + // Zero decrypt key (by default)
|
| + video.Begin();
|
| + vpx_codec_err_t res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
|
| + ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
|
| +
|
| + // Non-zero decrypt key
|
| + video.Next();
|
| + decoder.Control(VP8_SET_DECRYPT_KEY, decrypt_key);
|
| + res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
|
| + ASSERT_NE(VPX_CODEC_OK, res) << decoder.DecodeError();
|
| +}
|
| +
|
| +} // namespace libvpx_test
|
| +
|
| +#endif // CONFIG_DECRYPT
|
|
|