// Copyright 2023 Google LLC. All Rights Reserved. This file and proprietary
// source code may only be used and distributed under the Widevine License
// Agreement.
//
//  Reference implementation utilities of OEMCrypto APIs
//
#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include "bcc_validator.h"

using ::testing::AllOf;
using ::testing::Ge;
using ::testing::HasSubstr;
using ::testing::Le;

namespace wvoec {
namespace util {
namespace {
// Self-signed phase 1 BCC generated by OPK reference implementation.
const std::vector<uint8_t> kBcc = {
    0x82, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04, 0x81, 0x02, 0x20, 0x06, 0x21,
    0x58, 0x20, 0x02, 0xab, 0xbe, 0x80, 0x02, 0x41, 0xf1, 0x0b, 0x40, 0xff,
    0xd5, 0xf0, 0xd0, 0x26, 0x65, 0x70, 0xdc, 0x5a, 0x97, 0xa6, 0x80, 0xee,
    0x15, 0x95, 0xec, 0x26, 0x36, 0x70, 0x77, 0xe5, 0xfb, 0x10, 0x84, 0x43,
    0xa1, 0x01, 0x27, 0xa0, 0x58, 0x40, 0xa4, 0x01, 0x60, 0x02, 0x60, 0x3a,
    0x00, 0x47, 0x44, 0x57, 0x58, 0x2d, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04,
    0x81, 0x02, 0x20, 0x06, 0x21, 0x58, 0x20, 0x02, 0xab, 0xbe, 0x80, 0x02,
    0x41, 0xf1, 0x0b, 0x40, 0xff, 0xd5, 0xf0, 0xd0, 0x26, 0x65, 0x70, 0xdc,
    0x5a, 0x97, 0xa6, 0x80, 0xee, 0x15, 0x95, 0xec, 0x26, 0x36, 0x70, 0x77,
    0xe5, 0xfb, 0x10, 0x3a, 0x00, 0x47, 0x44, 0x58, 0x41, 0x20, 0x58, 0x40,
    0x73, 0x02, 0x36, 0xaa, 0x6d, 0x52, 0x50, 0x67, 0x43, 0xc4, 0x0b, 0xf8,
    0x3f, 0x35, 0x2a, 0xd8, 0x44, 0x09, 0xf4, 0x1d, 0xca, 0x91, 0x12, 0x27,
    0x01, 0xdf, 0x73, 0xb7, 0x9b, 0x31, 0x28, 0x8e, 0xae, 0x9b, 0xc6, 0x7a,
    0xdc, 0x07, 0xab, 0x69, 0xd2, 0x85, 0x9a, 0x15, 0x8b, 0xe3, 0x5b, 0xf2,
    0x94, 0x95, 0xee, 0x49, 0x74, 0xc5, 0x85, 0x62, 0x3d, 0x46, 0x4c, 0xeb,
    0x11, 0x89, 0x68, 0x02};
const std::vector<uint8_t> kBccWrongEntryKey = {
    0x82, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04, 0x81, 0x02, 0x20, 0x06, 0x21,
    0x58, 0x20, 0x02, 0xab, 0xbe, 0x80, 0x02, 0x41, 0xf1, 0x0b, 0x40, 0xff,
    0xd5, 0xf0, 0xd0, 0x26, 0x65, 0x70, 0xdc, 0x5a, 0x97, 0xa6, 0x80, 0xee,
    0x15, 0x95, 0xec, 0x26, 0x36, 0x70, 0x77, 0xe5, 0xfb, 0x10, 0x84, 0x43,
    0xa1, 0x01, 0x27, 0xa0, 0x58, 0x40, 0xa4, 0x01, 0x60, 0x02, 0x60, 0x3a,
    0x00, 0x47, 0x44, 0x57, 0x58, 0x2d, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04,
    0x81, 0x02, 0x20, 0x06, 0x21, 0x58, 0x20, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
    0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
    0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
    0xaa, 0xaa, 0xaa, 0x3a, 0x00, 0x47, 0x44, 0x58, 0x41, 0x20, 0x58, 0x40,
    0x89, 0x1d, 0xff, 0xb3, 0x3b, 0xe2, 0xdc, 0xc6, 0xbc, 0xbd, 0xc7, 0xcd,
    0x3f, 0x9c, 0x43, 0xf6, 0xdd, 0xea, 0x58, 0x53, 0x45, 0x8f, 0x87, 0x17,
    0x0a, 0xe4, 0x06, 0xf2, 0xbe, 0x14, 0x69, 0x13, 0x3d, 0x1d, 0xd0, 0x52,
    0x8f, 0x56, 0x4b, 0x0f, 0xad, 0x2e, 0xf0, 0xbf, 0xbb, 0xd1, 0x35, 0x9c,
    0x5a, 0xe8, 0x67, 0xbe, 0xec, 0xff, 0x9d, 0xfe, 0xac, 0x8d, 0x47, 0x4e,
    0x6d, 0xd1, 0xd3, 0x02};
const std::vector<uint8_t> kBccMissingIssuer = {
    0x82, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04, 0x81, 0x02, 0x20, 0x06, 0x21,
    0x58, 0x20, 0x02, 0xab, 0xbe, 0x80, 0x02, 0x41, 0xf1, 0x0b, 0x40, 0xff,
    0xd5, 0xf0, 0xd0, 0x26, 0x65, 0x70, 0xdc, 0x5a, 0x97, 0xa6, 0x80, 0xee,
    0x15, 0x95, 0xec, 0x26, 0x36, 0x70, 0x77, 0xe5, 0xfb, 0x10, 0x84, 0x43,
    0xa1, 0x01, 0x27, 0xa0, 0x58, 0x3e, 0xa3, 0x02, 0x60, 0x3a, 0x00, 0x47,
    0x44, 0x57, 0x58, 0x2d, 0xa5, 0x01, 0x01, 0x03, 0x27, 0x04, 0x81, 0x02,
    0x20, 0x06, 0x21, 0x58, 0x20, 0x02, 0xab, 0xbe, 0x80, 0x02, 0x41, 0xf1,
    0x0b, 0x40, 0xff, 0xd5, 0xf0, 0xd0, 0x26, 0x65, 0x70, 0xdc, 0x5a, 0x97,
    0xa6, 0x80, 0xee, 0x15, 0x95, 0xec, 0x26, 0x36, 0x70, 0x77, 0xe5, 0xfb,
    0x10, 0x3a, 0x00, 0x47, 0x44, 0x58, 0x41, 0x20, 0x58, 0x40, 0xf9, 0x46,
    0x36, 0xbd, 0x95, 0x75, 0xc2, 0x3d, 0xf9, 0xa2, 0xbe, 0x60, 0x8e, 0xbf,
    0x64, 0x89, 0xdf, 0xb9, 0x9c, 0x3c, 0x17, 0x36, 0x23, 0x9a, 0x68, 0x1a,
    0x34, 0x36, 0x51, 0x89, 0x59, 0xf2, 0x54, 0x62, 0xd3, 0x8f, 0xeb, 0x9b,
    0x75, 0x3e, 0xe9, 0xfc, 0xe3, 0xc2, 0x8f, 0x84, 0xb1, 0x71, 0xcd, 0x29,
    0x12, 0x65, 0xeb, 0xab, 0x28, 0x4b, 0xe2, 0x3e, 0x1b, 0xd8, 0x17, 0xdb,
    0x97, 0x0f};
}  // namespace

TEST(OEMCryptoBccValidatorTest, BccParseError) {
  const std::vector<uint8_t> bcc_bad(kBcc.begin(), kBcc.end() - 1);
  BccValidator validator;
  CborMessageStatus result = validator.Parse(bcc_bad);
  EXPECT_EQ(kCborParseError, result);
  result = validator.Validate();
  EXPECT_EQ(kCborParseError, result);
  EXPECT_EQ("", validator.GetRawMessage());
  EXPECT_EQ("", validator.GetFormattedMessage());
}

TEST(OEMCryptoBccValidatorTest, Bcc) {
  BccValidator validator;
  CborMessageStatus result = validator.Parse(kBcc);
  EXPECT_EQ(kCborParseOk, result);
  result = validator.Validate();
  EXPECT_THAT(result, AllOf(Ge(kCborValidateOk), Le(kCborValidateWarning)));
  const std::string out = validator.GetFormattedMessage();
  EXPECT_THAT(out, HasSubstr("key encoding format: DEVICE_KEY_BYTE_STRING"));
  EXPECT_THAT(out, HasSubstr("key algorithm type: EDDSA"));
}

TEST(OEMCryptoBccValidatorTest, BccWrongEntryKey) {
  BccValidator validator;
  CborMessageStatus result = validator.Parse(kBccWrongEntryKey);
  EXPECT_EQ(kCborParseOk, result);
  result = validator.Validate();
  EXPECT_EQ(result, kCborValidateError);
  // Non-fatal validation error should be able to return formatted output.
  const std::string out = validator.GetFormattedMessage();
  EXPECT_THAT(out, HasSubstr("key encoding format: DEVICE_KEY_BYTE_STRING"));
  EXPECT_THAT(out, HasSubstr("key algorithm type: EDDSA"));
  const std::vector<std::pair<CborMessageStatus, std::string>> msgs =
      validator.GetValidateMessages();
  EXPECT_EQ(1u, msgs.size());
  EXPECT_EQ(kCborValidateError, msgs[0].first);
}

TEST(OEMCryptoBccValidatorTest, BccParseThreeTimes) {
  BccValidator validator;
  const std::vector<uint8_t> bcc_bad(kBcc.begin(), kBcc.end() - 2);
  CborMessageStatus result = validator.Parse(bcc_bad);
  EXPECT_EQ(kCborParseError, result);
  result = validator.Validate();
  EXPECT_EQ(kCborParseError, result);

  result = validator.Parse(kBccWrongEntryKey);
  EXPECT_EQ(kCborParseOk, result);
  result = validator.Validate();
  EXPECT_EQ(result, kCborValidateError);

  result = validator.Parse(kBcc);
  EXPECT_EQ(kCborParseOk, result);
  result = validator.Validate();
  EXPECT_EQ(result, kCborValidateOk);
}

TEST(OEMCryptoBccValidatorTest, BccMissingIssuer) {
  BccValidator validator;
  CborMessageStatus result = validator.Parse(kBccMissingIssuer);
  EXPECT_EQ(kCborParseOk, result);
  result = validator.Validate();
  EXPECT_EQ(result, kCborValidateError);
  const std::string out = validator.GetFormattedMessage();
  EXPECT_THAT(out, HasSubstr("key encoding format: DEVICE_KEY_BYTE_STRING"));
  EXPECT_THAT(out, HasSubstr("key algorithm type: EDDSA"));
  const std::vector<std::pair<CborMessageStatus, std::string>> msgs =
      validator.GetValidateMessages();
  EXPECT_EQ(1u, msgs.size());
  EXPECT_EQ(kCborValidateError, msgs[0].first);
  EXPECT_THAT(msgs[0].second, HasSubstr("Missing Issuer"));
}
}  // namespace util
}  // namespace wvoec