test/vp9_decrypt_test.cc

   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] = { 0x01, 0x12, 0x23, 0x34, 0x45, 0x56, 0x67, 0x78,
  25                                0x89, 0x9a, 0xab, 0xbc, 0xcd, 0xde, 0xef, 0xf0 };
  26
  27 void encrypt_buffer(const uint8_t *src, uint8_t *dst, size_t size,
  28                     ptrdiff_t offset) {
  29   for (size_t i = 0; i < size; ++i) {
  30     dst[i] = src[i] ^ test_key[(offset + i) & 15];
  31   }
  32 }
  33
  34 void test_decrypt_cb(void *decrypt_state, const uint8_t *input, uint8_t *output,
  35                      int count) {
  36   encrypt_buffer(input, output, count,
  37                  input - reinterpret_cast<uint8_t *>(decrypt_state));
  38 }
  39
  40 }  // namespace
  41
  42 namespace libvpx_test {
  43
  44 TEST(TestDecrypt, DecryptWorksVp9) {
  45   libvpx_test::IVFVideoSource video("vp90-2-05-resize.ivf");
  46   video.Init();
  47
  48   vpx_codec_dec_cfg_t dec_cfg = vpx_codec_dec_cfg_t();
  49   VP9Decoder 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   std::vector<uint8_t> encrypted(video.frame_size());
  61   encrypt_buffer(video.cxdata(), &encrypted[0], video.frame_size(), 0);
  62   vpx_decrypt_init di = { test_decrypt_cb, &encrypted[0] };
  63   decoder.Control(VPXD_SET_DECRYPTOR, &di);
  64
  65   res = decoder.DecodeFrame(&encrypted[0], encrypted.size());
  66   ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
  67 }
  68
  69 }  // namespace libvpx_test