789377fed2
In this code drop we introduce the ODK dependency. The reference implementation has been updated to make use of the ODK and the related tests have been included. In addition, we have included an example of how a shared libraries can be created. This will allow make it easier to test and verify different implementations of the API. Most other changes introduce by this code drop were made to clean-up the reference implementation and limit dependencies.
140 lines
4.7 KiB
C++
140 lines
4.7 KiB
C++
// Copyright 2020 Google LLC. All Rights Reserved.
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#include "api/license_whitebox.h"
|
|
#include "api/license_whitebox_benchmark.h"
|
|
#include "api/result.h"
|
|
#include "api/test_data.h"
|
|
#include "api/test_license_builder.h"
|
|
#include "base/logging.h"
|
|
#include "benchmarking/measurements.h"
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
|
|
|
namespace widevine {
|
|
namespace {
|
|
// The white-box implementation is slow, so the build bots keep timing out. To
|
|
// work around this, use a fixed number of iterations.
|
|
constexpr size_t kIterations = 100;
|
|
} // namespace
|
|
class LicenseWhiteboxProcessLicenseResponseBenchmark
|
|
: public LicenseWhiteboxBenchmark {
|
|
protected:
|
|
void SetUp() override {
|
|
LicenseWhiteboxBenchmark::SetUp();
|
|
license_ = CreateLicense();
|
|
}
|
|
|
|
void TearDown() override { WB_License_Delete(whitebox_); }
|
|
|
|
WB_License_Whitebox* whitebox_;
|
|
License license_;
|
|
};
|
|
|
|
TEST_F(LicenseWhiteboxProcessLicenseResponseBenchmark,
|
|
CreateAndProcessLicenseResponseAndDelete) {
|
|
Timer timer;
|
|
Sampler sampler;
|
|
|
|
for (size_t i = 0; i < kIterations; i++) {
|
|
const auto init_data = GetLicenseInitData();
|
|
|
|
// We can only call ProcessLicenseResponse() once per whitebox instance. So
|
|
// we need to create a new instance each time. Collect samples for create +
|
|
// process license + delete so that we know the cost of getting a new
|
|
// license.
|
|
timer.Reset();
|
|
|
|
ASSERT_EQ(WB_License_Create(init_data.data(), init_data.size(), &whitebox_),
|
|
WB_RESULT_OK);
|
|
|
|
ASSERT_EQ(WB_License_ProcessLicenseResponse(
|
|
whitebox_, license_.core_message.data(),
|
|
license_.core_message.size(), license_.message.data(),
|
|
license_.message.size(), license_.signature.data(),
|
|
license_.signature.size(), license_.session_key.data(),
|
|
license_.session_key.size(), license_.request.data(),
|
|
license_.request.size()),
|
|
WB_RESULT_OK);
|
|
|
|
WB_License_Delete(whitebox_);
|
|
|
|
sampler.Push(timer.Get());
|
|
|
|
whitebox_ = nullptr;
|
|
}
|
|
|
|
PrettyPrint("License White-box Create + Process License + Delete Duration",
|
|
sampler, license_.message.size());
|
|
}
|
|
|
|
TEST_F(LicenseWhiteboxProcessLicenseResponseBenchmark,
|
|
CreateAndProcessLicenseResponse) {
|
|
Timer timer;
|
|
Sampler sampler;
|
|
|
|
for (size_t i = 0; i < kIterations; i++) {
|
|
const auto init_data = GetLicenseInitData();
|
|
|
|
// We can only call ProcessLicenseResponse() once per whitebox instance. So
|
|
// we need to create a new instance each time. Collect samples for create +
|
|
// process license so that we know the true start-up cost.
|
|
timer.Reset();
|
|
|
|
ASSERT_EQ(WB_License_Create(init_data.data(), init_data.size(), &whitebox_),
|
|
WB_RESULT_OK);
|
|
|
|
ASSERT_EQ(WB_License_ProcessLicenseResponse(
|
|
whitebox_, license_.core_message.data(),
|
|
license_.core_message.size(), license_.message.data(),
|
|
license_.message.size(), license_.signature.data(),
|
|
license_.signature.size(), license_.session_key.data(),
|
|
license_.session_key.size(), license_.request.data(),
|
|
license_.request.size()),
|
|
WB_RESULT_OK);
|
|
|
|
sampler.Push(timer.Get());
|
|
|
|
WB_License_Delete(whitebox_);
|
|
whitebox_ = nullptr;
|
|
}
|
|
|
|
PrettyPrint("License White-box Create + Process License Duration", sampler,
|
|
license_.message.size());
|
|
}
|
|
|
|
TEST_F(LicenseWhiteboxProcessLicenseResponseBenchmark, ProcessLicenseResponse) {
|
|
Timer timer;
|
|
Sampler sampler;
|
|
|
|
for (size_t i = 0; i < kIterations; i++) {
|
|
// We can only call ProcessLicenseResponse() once per whitebox instance. So
|
|
// we need to create a new instance each time. Do this before we reset the
|
|
// timer so that we are not counting it in the execution time.
|
|
const auto init_data = GetLicenseInitData();
|
|
ASSERT_EQ(WB_License_Create(init_data.data(), init_data.size(), &whitebox_),
|
|
WB_RESULT_OK);
|
|
|
|
timer.Reset();
|
|
|
|
ASSERT_EQ(WB_License_ProcessLicenseResponse(
|
|
whitebox_, license_.core_message.data(),
|
|
license_.core_message.size(), license_.message.data(),
|
|
license_.message.size(), license_.signature.data(),
|
|
license_.signature.size(), license_.session_key.data(),
|
|
license_.session_key.size(), license_.request.data(),
|
|
license_.request.size()),
|
|
WB_RESULT_OK);
|
|
|
|
sampler.Push(timer.Get());
|
|
|
|
WB_License_Delete(whitebox_);
|
|
whitebox_ = nullptr;
|
|
}
|
|
|
|
PrettyPrint("License White-box Process License Duration", sampler,
|
|
license_.message.size());
|
|
}
|
|
} // namespace widevine
|