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
19 namespace {
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.
24 const 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
encrypt_buffer(const uint8_t * src,uint8_t * dst,size_t size,ptrdiff_t offset)29 void encrypt_buffer(const uint8_t *src, uint8_t *dst, size_t size,
30 ptrdiff_t offset) {
31 for (size_t i = 0; i < size; ++i) {
32 dst[i] = src[i] ^ test_key[(offset + i) & 15];
33 }
34 }
35
test_decrypt_cb(void * decrypt_state,const uint8_t * input,uint8_t * output,int count)36 void test_decrypt_cb(void *decrypt_state, const uint8_t *input,
37 uint8_t *output, int count) {
38 encrypt_buffer(input, output, count,
39 input - reinterpret_cast<uint8_t *>(decrypt_state));
40 }
41
42 } // namespace
43
44 namespace libvpx_test {
45
TEST(TestDecrypt,DecryptWorks)46 TEST(TestDecrypt, DecryptWorks) {
47 libvpx_test::IVFVideoSource video("vp80-00-comprehensive-001.ivf");
48 video.Init();
49
50 vpx_codec_dec_cfg_t dec_cfg = {0};
51 VP8Decoder decoder(dec_cfg, 0);
52
53 video.Begin();
54
55 // no decryption
56 vpx_codec_err_t res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
57 ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
58
59 // decrypt frame
60 video.Next();
61
62 #if CONFIG_DECRYPT
63 std::vector<uint8_t> encrypted(video.frame_size());
64 encrypt_buffer(video.cxdata(), &encrypted[0], video.frame_size(), 0);
65 vp8_decrypt_init di = { test_decrypt_cb, &encrypted[0] };
66 decoder.Control(VP8D_SET_DECRYPTOR, &di);
67 #endif // CONFIG_DECRYPT
68
69 res = decoder.DecodeFrame(video.cxdata(), video.frame_size());
70 ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
71 }
72
73 } // namespace libvpx_test
74