e51c9fbbb8
Merge from Widevine repo of http://go/wvgerrit/121950 Remove term "Master" from "Widevine Master License Agreement". Bug: 168562298 Change-Id: I655babf1bc447f4872f6a0f849107262be42df7a
105 lines
3.0 KiB
C++
105 lines
3.0 KiB
C++
// Copyright 2020 Google LLC. All Rights Reserved. This file and proprietary
|
|
// source code may only be used and distributed under the Widevine
|
|
// License Agreement.
|
|
#ifndef OEMCRYPTO_FUZZ_HELPER_H_
|
|
#define OEMCRYPTO_FUZZ_HELPER_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "FuzzedDataProvider.h"
|
|
#include "OEMCryptoCENC.h"
|
|
#include "oec_device_features.h"
|
|
#include "oemcrypto_corpus_generator_helper.h"
|
|
#include "oemcrypto_session_tests_helper.h"
|
|
|
|
namespace wvoec {
|
|
// Initial setup to create a valid OEMCrypto state such as initializing crypto
|
|
// firmware/hardware, installing golden key box etc. in order to fuzz
|
|
// OEMCrypto APIs.
|
|
class InitializeFuzz : public SessionUtil {
|
|
public:
|
|
InitializeFuzz() {
|
|
wvoec::global_features.Initialize();
|
|
OEMCrypto_SetSandbox(kTestSandbox, sizeof(kTestSandbox));
|
|
OEMCrypto_Initialize();
|
|
EnsureTestKeys();
|
|
}
|
|
|
|
~InitializeFuzz() { OEMCrypto_Terminate(); }
|
|
};
|
|
|
|
class OEMCryptoLicenseAPIFuzz : public InitializeFuzz {
|
|
public:
|
|
OEMCryptoLicenseAPIFuzz() : license_messages_(&session_) {
|
|
session_.open();
|
|
InstallTestRSAKey(&session_);
|
|
session_.GenerateNonce();
|
|
}
|
|
|
|
~OEMCryptoLicenseAPIFuzz() { session_.close(); }
|
|
|
|
LicenseRoundTrip& license_messages() { return license_messages_; }
|
|
|
|
Session* session() { return &session_; }
|
|
|
|
void LoadLicense() {
|
|
license_messages_.SignAndVerifyRequest();
|
|
license_messages_.CreateDefaultResponse();
|
|
license_messages_.EncryptAndSignResponse();
|
|
license_messages_.LoadResponse();
|
|
}
|
|
|
|
private:
|
|
Session session_;
|
|
LicenseRoundTrip license_messages_;
|
|
};
|
|
|
|
class OEMCryptoProvisioningAPIFuzz : public InitializeFuzz {
|
|
public:
|
|
OEMCryptoProvisioningAPIFuzz()
|
|
: provisioning_messages_(&session_, encoded_rsa_key_) {
|
|
// Opens a session and Generates Nonce.
|
|
provisioning_messages_.PrepareSession(keybox_);
|
|
}
|
|
|
|
~OEMCryptoProvisioningAPIFuzz() { session_.close(); }
|
|
|
|
ProvisioningRoundTrip& provisioning_messages() {
|
|
return provisioning_messages_;
|
|
}
|
|
|
|
private:
|
|
Session session_;
|
|
ProvisioningRoundTrip provisioning_messages_;
|
|
};
|
|
|
|
// Initial setup to create a valid state such as creating session, installing
|
|
// golden key box etc. in order to fuzz Load Renewal API.
|
|
class OEMCryptoRenewalAPIFuzz : public OEMCryptoLicenseAPIFuzz {
|
|
public:
|
|
OEMCryptoRenewalAPIFuzz() : renewal_messages_(&license_messages()) {}
|
|
|
|
RenewalRoundTrip& renewal_messages() { return renewal_messages_; }
|
|
|
|
private:
|
|
RenewalRoundTrip renewal_messages_;
|
|
};
|
|
|
|
// Convert data to valid enum value.
|
|
template <typename T>
|
|
void ConvertDataToValidEnum(T max_enum_value, T* t) {
|
|
FuzzedDataProvider fuzzed_enum_data(reinterpret_cast<uint8_t*>(t), sizeof(T));
|
|
*t = static_cast<T>(fuzzed_enum_data.ConsumeIntegralInRange<uint32_t>(
|
|
0, static_cast<uint32_t>(max_enum_value)));
|
|
}
|
|
|
|
// Redirect printf and log statements from oemcrypto functions to a file to
|
|
// reduce noise
|
|
void RedirectStdoutToFile();
|
|
|
|
// Function to split fuzzer input using delimiter "-_^_".
|
|
std::vector<std::vector<uint8_t>> SplitInput(const uint8_t* data, size_t size);
|
|
} // namespace wvoec
|
|
|
|
#endif // OEMCRYPTO_FUZZ_HELPER_H_
|