widevine-partner/android
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify OEMCrypto_InstallOemPrivateKey fuzzer

Browse Source 
Merged from https://widevine-internal-review.googlesource.com/174890
Merged from https://widevine-internal-review.googlesource.com/174856

Change-Id: If863e4a8f739ca5645311d7f7b08e6aec4cc7aca
This commit is contained in:
Ian Benz
2023-05-10 14:46:52 +00:00
committed by Robert Shih
parent 75d4778531
commit 708670c98b
1 changed files with 4 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
libwvdrmengine/oemcrypto/test/fuzz_tests/oemcrypto_install_oem_private_key_fuzz.cc
View File

@@ -9,27 +9,17 @@
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
wvoec::RedirectStdoutToFile(); wvoec::RedirectStdoutToFile();
wvoec::LicenseWithUsageEntryFuzz entry; wvoec::SessionFuzz session_fuzz;
entry.Initialize(); session_fuzz.Initialize();
entry.CreateUsageTableHeader();
// Open a session, create a usage entry.
wvoec::Session* session = entry.license_messages().session();
session->open();
entry.InstallTestDrmKey(session);
session->GenerateNonce();
session->CreateNewUsageEntry();
std::vector<uint8_t> encrypted_usage_header;
session->UpdateUsageEntry(&encrypted_usage_header);
FuzzedDataProvider fuzzed_data(data, size); FuzzedDataProvider fuzzed_data(data, size);
const OEMCrypto_PrivateKeyType key_type = wvoec::ConvertDataToValidEnum( const OEMCrypto_PrivateKeyType key_type = wvoec::ConvertDataToValidEnum(
fuzzed_data, OEMCrypto_PrivateKeyType_MaxValue); fuzzed_data, OEMCrypto_PrivateKeyType_MaxValue);
const std::vector<uint8_t> wrapped_private_key = const std::vector<uint8_t> wrapped_private_key =
fuzzed_data.ConsumeRemainingBytes<uint8_t>(); fuzzed_data.ConsumeRemainingBytes<uint8_t>();
OEMCrypto_InstallOemPrivateKey(session->session_id(), key_type, OEMCrypto_InstallOemPrivateKey(session_fuzz.session().session_id(), key_type,
wrapped_private_key.data(), wrapped_private_key.data(),
wrapped_private_key.size()); wrapped_private_key.size());
session->close(); session_fuzz.Terminate();
OEMCrypto_Terminate();
return 0; return 0;
} }