11255b7426
No-Typo-Check: From a third party header file Bug: 260918793 Test: unit tests Test: atp v2/widevine-eng/drm_compliance Change-Id: I36effd6a10a99bdb2399ab1f4a0fad026d607c70
60 lines
2.3 KiB
C++
60 lines
2.3 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.
|
|
|
|
#include "FuzzedDataProvider.h"
|
|
#include "OEMCryptoCENC.h"
|
|
#include "log.h"
|
|
#include "oec_session_util.h"
|
|
#include "oemcrypto_fuzz_helper.h"
|
|
#include "oemcrypto_fuzz_structs.h"
|
|
#include "oemcrypto_types.h"
|
|
|
|
namespace wvoec {
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
|
|
// Redirect printf and log statements from oemcrypto functions to a file to
|
|
// reduce noise
|
|
RedirectStdoutToFile();
|
|
|
|
// Split data using separator.
|
|
const std::vector<FuzzedData> inputs = SplitFuzzedData(data, size);
|
|
if (inputs.size() < 2) {
|
|
return 0;
|
|
}
|
|
|
|
OEMCrypto_Generic_Api_Fuzz fuzzed_structure;
|
|
if (inputs[0].size < sizeof(fuzzed_structure)) {
|
|
return 0;
|
|
}
|
|
// Copy OEMCrypto_Generic_Api_Fuzz from input data.
|
|
FuzzedDataProvider fuzzed_data(inputs[0].data, inputs[0].size);
|
|
fuzzed_data.ConsumeData(&fuzzed_structure, sizeof(fuzzed_structure));
|
|
ConvertDataToValidEnum(OEMCrypto_CipherMode_MaxValue,
|
|
&fuzzed_structure.cipher_mode);
|
|
ConvertDataToValidEnum(OEMCrypto_Algorithm_MaxValue,
|
|
&fuzzed_structure.algorithm);
|
|
// Copy iv from input data.
|
|
const std::vector<uint8_t> iv = fuzzed_data.ConsumeRemainingBytes<uint8_t>();
|
|
|
|
// Initialize encrypted and clear buffers.
|
|
const std::vector<uint8_t> encrypted_buffer(inputs[1].data,
|
|
inputs[1].data + inputs[1].size);
|
|
std::vector<uint8_t> clear_buffer(encrypted_buffer.size());
|
|
|
|
OEMCryptoLicenseAPIFuzz license_api_fuzz;
|
|
Session* session = license_api_fuzz.session();
|
|
// Load license and call generic_decrypt API.
|
|
license_api_fuzz.LoadLicense();
|
|
vector<uint8_t> key_handle;
|
|
GetKeyHandleIntoVector(session->session_id(),
|
|
session->license().keys[0].key_id,
|
|
session->license().keys[0].key_id_length,
|
|
fuzzed_structure.cipher_mode, key_handle);
|
|
OEMCrypto_Generic_Decrypt(key_handle.data(), key_handle.size(),
|
|
encrypted_buffer.data(), encrypted_buffer.size(),
|
|
iv.data(), fuzzed_structure.algorithm,
|
|
clear_buffer.data());
|
|
return 0;
|
|
}
|
|
} // namespace wvoec
|