16d9abca4c
This change updates the copyright notice to make it more clear that the code is distribued under the Widevine Master License Agreement. It also updates the unit tests and sample code to correct the useage of AES 256. AES 256 is used to decrypt entitled content keys, but it is not used to decrypt key control blocks.
122 lines
4.4 KiB
C++
122 lines
4.4 KiB
C++
// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
|
|
// source code may only be used and distributed under the Widevine Master
|
|
// License Agreement.
|
|
|
|
#include "OEMCryptoCENC.h"
|
|
|
|
#include <string>
|
|
#include <gtest/gtest.h>
|
|
#include "log.h"
|
|
#include "oemcrypto_logging.h"
|
|
#include "oemcrypto_mock.cpp"
|
|
|
|
class OEMCryptoLoggingTest : public ::testing::Test {
|
|
protected:
|
|
OEMCryptoLoggingTest() {}
|
|
|
|
void SetUp() {
|
|
::testing::Test::SetUp();
|
|
ASSERT_EQ(OEMCrypto_SUCCESS, OEMCrypto_Initialize());
|
|
}
|
|
|
|
void TearDown() {
|
|
OEMCrypto_Terminate();
|
|
::testing::Test::TearDown();
|
|
}
|
|
};
|
|
|
|
TEST_F(OEMCryptoLoggingTest, TestDumpHexFunctions) {
|
|
uint8_t vector[] = {0xFA, 0x11, 0x28, 0x33};
|
|
std::string buffer;
|
|
wvoec_mock::dump_hex_helper(buffer, "name", vector, 4u);
|
|
ASSERT_EQ("name = \n wvcdm::a2b_hex(\"FA112833\");\n", buffer);
|
|
|
|
uint8_t vector2[] = {
|
|
0xFA, 0x11, 0x28, 0x33, 0xFA, 0x11, 0x28, 0x33, 0xFA, 0x11,
|
|
0x28, 0x33, 0xFA, 0x11, 0x28, 0x33, 0xFA, 0x11, 0x28, 0x33,
|
|
0xFA, 0x11, 0x28, 0x33, 0x01, 0x14, 0x28, 0xAB, 0xFA, 0xCD,
|
|
0xEF, 0x67, 0x01, 0x14, 0x28, 0xAB, 0xFA, 0xCD, 0xEF, 0x67,
|
|
};
|
|
|
|
buffer.clear(); // dump_hex_helper appends to buffer
|
|
wvoec_mock::dump_hex_helper(buffer, "name", vector2, 40u);
|
|
ASSERT_EQ(
|
|
"name = \n "
|
|
"wvcdm::a2b_hex("
|
|
"\"FA112833FA112833FA112833FA112833FA112833FA112833011428ABFACDEF67\"\n "
|
|
" \"011428ABFACDEF67\");\n",
|
|
buffer);
|
|
|
|
buffer.clear(); // dump_hex_helper appends to buffer
|
|
wvoec_mock::dump_array_part_helper(buffer, "array", 5u, "name", vector2, 40u);
|
|
ASSERT_EQ(
|
|
"std::string s5_name = \n "
|
|
"wvcdm::a2b_hex("
|
|
"\"FA112833FA112833FA112833FA112833FA112833FA112833011428ABFACDEF67\"\n "
|
|
" \"011428ABFACDEF67\");\narray[5].name = message_ptr + "
|
|
"message.find(s5_name.data());\n",
|
|
buffer);
|
|
|
|
buffer.clear(); // dump_hex_helper appends to buffer
|
|
wvoec_mock::dump_array_part_helper(buffer, "array", 5u, "name", NULL, 40u);
|
|
ASSERT_EQ("array[5].name = NULL;\n", buffer);
|
|
}
|
|
|
|
TEST_F(OEMCryptoLoggingTest, TestChangeLoggingLevel) {
|
|
wvoec_mock::SetLoggingLevel(wvcdm::LOG_WARN);
|
|
ASSERT_EQ(wvcdm::LOG_WARN, wvcdm::g_cutoff);
|
|
|
|
wvoec_mock::SetLoggingLevel(wvcdm::LOG_INFO);
|
|
ASSERT_EQ(wvcdm::LOG_INFO, wvcdm::g_cutoff);
|
|
|
|
wvoec_mock::SetLoggingSettings(wvcdm::LOG_WARN,
|
|
wvoec_mock::kLoggingDumpTraceAll);
|
|
ASSERT_EQ(wvcdm::LOG_WARN, wvcdm::g_cutoff);
|
|
ASSERT_TRUE(wvoec_mock::LogCategoryEnabled(wvoec_mock::kLoggingDumpTraceAll));
|
|
wvoec_mock::TurnOffLoggingForAllCategories();
|
|
|
|
wvoec_mock::SetLoggingLevel(wvcdm::LOG_VERBOSE);
|
|
ASSERT_EQ(wvcdm::LOG_VERBOSE, wvcdm::g_cutoff);
|
|
|
|
wvoec_mock::SetLoggingLevel(wvcdm::LOG_WARN);
|
|
}
|
|
|
|
TEST_F(OEMCryptoLoggingTest, TestChangeLoggingCategories) {
|
|
using namespace wvoec_mock;
|
|
TurnOffLoggingForAllCategories();
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingTraceDecryption |
|
|
kLoggingTraceOEMCryptoCalls));
|
|
|
|
AddLoggingForCategories(kLoggingDumpKeyControlBlocks |
|
|
kLoggingDumpDerivedKeys);
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpKeyControlBlocks));
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingTraceUsageTable));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpTraceAll));
|
|
|
|
RemoveLoggingForCategories(kLoggingDumpKeyControlBlocks |
|
|
kLoggingTraceUsageTable);
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingDumpKeyControlBlocks));
|
|
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpDerivedKeys));
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingTraceUsageTable));
|
|
|
|
TurnOffLoggingForAllCategories();
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingTraceUsageTable));
|
|
|
|
AddLoggingForCategories(kLoggingDumpTraceAll);
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpKeyControlBlocks));
|
|
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingTraceOEMCryptoCalls));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpContentKeys));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpKeyControlBlocks));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpDerivedKeys));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingTraceNonce));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingTraceDecryption));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingTraceUsageTable));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingTraceDecryptCalls));
|
|
ASSERT_TRUE(LogCategoryEnabled(kLoggingDumpTraceAll));
|
|
|
|
RemoveLoggingForCategories(kLoggingDumpKeyControlBlocks);
|
|
ASSERT_FALSE(LogCategoryEnabled(kLoggingDumpKeyControlBlocks));
|
|
}
|