Source release 17.1.0

oemcrypto/test/fuzz_tests/oemcrypto_generic_encrypt_fuzz.cc

@@ -0,0 +1,61 @@
+// 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 "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.
+  auto inputs = SplitInput(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.
+  memcpy(&fuzzed_structure, data, sizeof(fuzzed_structure));
+  ConvertDataToValidEnum(OEMCrypto_CipherMode_MaxValue,
+                         &fuzzed_structure.cipher_mode);
+  ConvertDataToValidEnum(OEMCrypto_Algorithm_MaxValue,
+                         &fuzzed_structure.algorithm);
+
+  // Copy iv from input data.
+  size_t iv_size = inputs[0].size() - sizeof(fuzzed_structure);
+  if (iv_size == 0) {
+    return 0;
+  }
+  vector<uint8_t> iv(iv_size);
+  memcpy(iv.data(), data + sizeof(fuzzed_structure), iv_size);
+
+  // Copy clear buffer from input data.
+  vector<uint8_t> clear_buffer(inputs[1].size());
+  vector<uint8_t> encrypted_buffer(inputs[1].size());
+  memcpy(clear_buffer.data(), inputs[1].data(), inputs[1].size());
+
+  OEMCryptoLicenseAPIFuzz license_api_fuzz;
+  Session* session = license_api_fuzz.session();
+  // Load license and call generic_encrypt API.
+  license_api_fuzz.LoadLicense();
+  OEMCryptoResult sts = OEMCrypto_SelectKey(
+      session->session_id(), session->license().keys[0].key_id,
+      session->license().keys[0].key_id_length, fuzzed_structure.cipher_mode);
+  CheckStatusAndExitFuzzerOnFailure(sts, OEMCrypto_SUCCESS);
+  OEMCrypto_Generic_Encrypt(
+      session->session_id(), clear_buffer.data(), clear_buffer.size(),
+      iv.data(), fuzzed_structure.algorithm, encrypted_buffer.data());
+  return 0;
+}
+}  // namespace wvoec