Home | History | Annotate | Download | only in test 
      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 
     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 
     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 
     46 TEST(TestDecrypt, DecryptWorksVp8) {
     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   std::vector<uint8_t> encrypted(video.frame_size());
     63   encrypt_buffer(video.cxdata(), &encrypted[0], video.frame_size(), 0);
     64   vpx_decrypt_init di = { test_decrypt_cb, &encrypted[0] };
     65   decoder.Control(VPXD_SET_DECRYPTOR, &di);
     66 
     67   res = decoder.DecodeFrame(&encrypted[0], encrypted.size());
     68   ASSERT_EQ(VPX_CODEC_OK, res) << decoder.DecodeError();
     69 }
     70 
     71 }  // namespace libvpx_test
     72