vp8_decrypt_test.cc revision 2ec72e65689c948e92b826ae1e867bf369e72f13
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, 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 36void 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 44namespace libvpx_test { 45 46TEST(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