100 lines
3.3 KiB
C++
100 lines
3.3 KiB
C++
// Copyright 2016 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "packager/base/strings/string_number_conversions.h"
|
|
#include "packager/media/base/fixed_key_source.h"
|
|
#include "packager/media/base/test/status_test_util.h"
|
|
|
|
#define EXPECT_HEX_EQ(expected, actual) \
|
|
do { \
|
|
std::vector<uint8_t> decoded_; \
|
|
ASSERT_TRUE(base::HexStringToBytes((expected), &decoded_)); \
|
|
EXPECT_EQ(decoded_, (actual)); \
|
|
} while (false)
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
namespace {
|
|
const char kKeyIdHex[] = "0101020305080d1522375990e9000000";
|
|
const char kKeyHex[] = "00100100200300500801302103405500";
|
|
const char kIvHex[] = "000102030405060708090a0b0c0d0e0f";
|
|
const char kPsshBox1Hex[] =
|
|
"000000427073736801000000"
|
|
"020305070b0d1113171d1f25292b2f35"
|
|
"00000001"
|
|
"0101020305080d1522375990e9000000"
|
|
"0000000e"
|
|
"6544617368207061636b61676572";
|
|
const char kPsshBox2Hex[] =
|
|
"0000002e7073736800000000"
|
|
"bbbbbbbbbaaaaaaaaaaaaddddddddddd"
|
|
"0000000e"
|
|
"fffffffff000000000000ddddddd";
|
|
const char kDefaultPsshBoxHex[] =
|
|
"000000347073736801000000"
|
|
"1077efecc0b24d02ace33c1e52e2fb4b"
|
|
"00000001"
|
|
"0101020305080d1522375990e9000000"
|
|
"00000000";
|
|
}
|
|
|
|
TEST(FixedKeySourceTest, CreateFromHexStrings_Succes) {
|
|
std::string pssh_boxes = std::string(kPsshBox1Hex) + kPsshBox2Hex;
|
|
scoped_ptr<FixedKeySource> key_source = FixedKeySource::CreateFromHexStrings(
|
|
kKeyIdHex, kKeyHex, pssh_boxes, kIvHex);
|
|
ASSERT_NE(nullptr, key_source);
|
|
|
|
EncryptionKey key;
|
|
ASSERT_OK(key_source->GetKey(KeySource::TRACK_TYPE_SD, &key));
|
|
|
|
EXPECT_HEX_EQ(kKeyIdHex, key.key_id);
|
|
EXPECT_HEX_EQ(kKeyHex, key.key);
|
|
EXPECT_HEX_EQ(kIvHex, key.iv);
|
|
|
|
ASSERT_EQ(2u, key.key_system_info.size());
|
|
EXPECT_HEX_EQ(kPsshBox1Hex, key.key_system_info[0].CreateBox());
|
|
EXPECT_HEX_EQ(kPsshBox2Hex, key.key_system_info[1].CreateBox());
|
|
}
|
|
|
|
TEST(FixedKeySourceTest, CreateFromHexStrings_EmptyPssh) {
|
|
scoped_ptr<FixedKeySource> key_source = FixedKeySource::CreateFromHexStrings(
|
|
kKeyIdHex, kKeyHex, "", kIvHex);
|
|
ASSERT_NE(nullptr, key_source);
|
|
|
|
EncryptionKey key;
|
|
ASSERT_OK(key_source->GetKey(KeySource::TRACK_TYPE_SD, &key));
|
|
|
|
EXPECT_HEX_EQ(kKeyIdHex, key.key_id);
|
|
EXPECT_HEX_EQ(kKeyHex, key.key);
|
|
EXPECT_HEX_EQ(kIvHex, key.iv);
|
|
|
|
ASSERT_EQ(1u, key.key_system_info.size());
|
|
EXPECT_HEX_EQ(kDefaultPsshBoxHex, key.key_system_info[0].CreateBox());
|
|
}
|
|
|
|
TEST(FixedKeySourceTest, CreateFromHexStrings_Failure) {
|
|
scoped_ptr<FixedKeySource> key_source = FixedKeySource::CreateFromHexStrings(
|
|
kKeyIdHex, "invalid_hex_value", kPsshBox1Hex, kIvHex);
|
|
EXPECT_EQ(nullptr, key_source);
|
|
|
|
// Invalid key id size.
|
|
key_source = FixedKeySource::CreateFromHexStrings("000102030405", kKeyHex,
|
|
kPsshBox1Hex, kIvHex);
|
|
EXPECT_EQ(nullptr, key_source);
|
|
|
|
// Invalid pssh box.
|
|
key_source = FixedKeySource::CreateFromHexStrings(kKeyIdHex, kKeyHex,
|
|
"000102030405", kIvHex);
|
|
EXPECT_EQ(nullptr, key_source);
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|