vp8_decrypt_test.cc revision ba164dffc5a6795bce97fae02b51ccf3330e15e4 
1/*
2 *  Copyright (c) 2013 The WebM project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include <cstdio>
12#include <cstdlib>
13#include <string>
14#include <vector>
15#include "third_party/googletest/src/include/gtest/gtest.h"
16#include "test/codec_factory.h"
17#include "test/ivf_video_source.h"
18
19namespace {
20// In a real use the 'decrypt_state' parameter will be a pointer to a struct
21// with whatever internal state the decryptor uses. For testing we'll just
22// xor with a constant key, and decrypt_state will point to the start of
23// the original buffer.
24const uint8_t test_key[16] = {
25  0x01, 0x12, 0x23, 0x34, 0x45, 0x56, 0x67, 0x78,
26  0x89, 0x9a, 0xab, 0xbc, 0xcd, 0xde, 0xef, 0xf0
27};
28
29void encrypt_buffer(const uint8_t *src, uint8_t *dst, int size, int offset = 0) {
30  for (int i = 0; i < size; ++i) {
31    dst[i] = src[i] ^ test_key[(offset + i) & 15];
32  }
33}
34
35void test_decrypt_cb(void *decrypt_state, const uint8_t *input,
36                     uint8_t *output, int count) {
37  encrypt_buffer(input, output, count, input - (uint8_t *)decrypt_state);
38}
39
40} // namespace
41
42namespace libvpx_test {
43
44TEST(TestDecrypt, DecryptWorks) {
45  libvpx_test::IVFVideoSource video("vp80-00-comprehensive-001.ivf");
46  video.Init();
47
48  vpx_codec_dec_cfg_t dec_cfg = {0};
49  VP8Decoder decoder(dec_cfg, 0);
50
51  video.Begin();
52
53  // no decryption
54  vpx_codec_err_t res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
55  ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
56
57  // decrypt frame
58  video.Next();
59
60#if CONFIG_DECRYPT
61  std::vector<uint8_t> encrypted(video.frame_size());
62  encrypt_buffer(video.cxdata(), &encrypted[0], video.frame_size());
63  vp8_decrypt_init di = { test_decrypt_cb, &encrypted[0] };
64  decoder.Control(VP8D_SET_DECRYPTOR, &di);
65#endif  // CONFIG_DECRYPT
66
67  res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
68  ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
69}
70
71}  // namespace libvpx_test
72