// Copyright 2020 Google LLC. All Rights Reserved.

#include <stddef.h>
#include <stdint.h>

#include <memory>
#include <vector>

#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/data_source.h"
#include "benchmarking/measurements.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace widevine {
namespace {

constexpr size_t kMessageSize = 4 * 1024;
constexpr size_t kSignatureSize = 256;
constexpr size_t kIterations = 100;

}  // namespace

class LicenseWhiteboxSignBenchmark : public LicenseWhiteboxBenchmark {
 protected:
  void SetUp() override {
    LicenseWhiteboxBenchmark::SetUp();

    message_ = Data().Get(kMessageSize);
    signature_.resize(kSignatureSize);

    const auto init_data = GetLicenseInitData();
    ASSERT_EQ(WB_License_Create(init_data.data(), init_data.size(), &whitebox_),
              WB_RESULT_OK);

    const auto license = CreateLicense();
    ASSERT_EQ(WB_License_ProcessLicenseResponse(
                  whitebox_, 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);
  }

  void TearDown() override { WB_License_Delete(whitebox_); }

  WB_License_Whitebox* whitebox_;

  std::vector<uint8_t> message_;
  std::vector<uint8_t> signature_;
};

TEST_F(LicenseWhiteboxSignBenchmark, SignLicenseRequest) {
  Timer timer;
  Sampler sampler;

  for (size_t i = 0; i < kIterations; i++) {
    size_t signature_size = signature_.size();

    timer.Reset();
    ASSERT_EQ(WB_RESULT_OK, WB_License_SignLicenseRequest(
                                whitebox_, message_.data(), message_.size(),
                                signature_.data(), &signature_size));
    sampler.Push(timer.Get());
  }

  PrettyPrint("License White-box Sign License Request Duration", sampler,
              message_.size());
}

TEST_F(LicenseWhiteboxSignBenchmark, SignRenewalRequest) {
  Timer timer;
  Sampler sampler;

  for (size_t i = 0; i < kIterations; i++) {
    size_t signature_size = signature_.size();

    timer.Reset();
    ASSERT_EQ(WB_RESULT_OK, WB_License_SignRenewalRequest(
                                whitebox_, message_.data(), message_.size(),
                                signature_.data(), &signature_size));
    sampler.Push(timer.Get());
  }

  PrettyPrint("License White-box Sign Renewal Request Duration", sampler,
              message_.size());
}
}  // namespace widevine