From 001ede83fd215d2a24b8c1e3e06b2f033a75498d Mon Sep 17 00:00:00 2001
From: Vicky Min <vickymin@google.com>
Date: Tue, 13 Jun 2023 17:41:22 +0000
Subject: [PATCH] Skip OEMCryptoMemoryLoadUsageTableHeaderForHugeHeader when
 usage tables are not supported

Bug: 284922725
Merged from https://widevine-internal-review.googlesource.com/176890

Change-Id: I7d15a85b8a9a25e1261f341b0a583369f84c1fb1
---
 .../test/oemcrypto_security_test.cpp          | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/libwvdrmengine/oemcrypto/test/oemcrypto_security_test.cpp b/libwvdrmengine/oemcrypto/test/oemcrypto_security_test.cpp
index dc39198a..af9b0462 100644
--- a/libwvdrmengine/oemcrypto/test/oemcrypto_security_test.cpp
+++ b/libwvdrmengine/oemcrypto/test/oemcrypto_security_test.cpp
@@ -984,6 +984,9 @@ TEST_P(OEMCryptoGenericCryptoTest,
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryUpdateUsageEntryForHugeHeaderBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     if (buffer_length < encrypted_usage_header_.size()) {
       return OEMCrypto_ERROR_SHORT_BUFFER;
@@ -1011,6 +1014,9 @@ TEST_P(OEMCryptoUsageTableTest,
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryUpdateUsageEntryForHugeUsageEntryBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     entry.license_messages().set_api_version(license_api_version_);
@@ -1035,6 +1041,9 @@ TEST_P(OEMCryptoUsageTableTest,
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryDeactivateUsageEntryForHugePstBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     std::string pst("pst");
@@ -1054,6 +1063,9 @@ TEST_P(OEMCryptoUsageTableTest,
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryLoadUsageTableHeaderForHugeHeader) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     if (buffer_length < encrypted_usage_header_.size()) {
       return OEMCrypto_ERROR_SHORT_BUFFER;
@@ -1078,6 +1090,9 @@ TEST_P(OEMCryptoUsageTableTest,
 TEST_P(
     OEMCryptoUsageTableTest,
     OEMCryptoMemoryLoadUsageTableHeaderForHugeHeaderStartingHeaderLengthFrom1) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     entry.license_messages().set_api_version(license_api_version_);
@@ -1094,6 +1109,9 @@ TEST_P(
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryLoadUsageEntryForHugeUsageEntryBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     entry.license_messages().set_api_version(license_api_version_);
@@ -1121,6 +1139,9 @@ TEST_P(OEMCryptoUsageTableTest,
 }
 
 TEST_P(OEMCryptoUsageTableTest, OEMCryptoMemoryReportUsageForHugeReportBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     entry.license_messages().set_api_version(license_api_version_);
@@ -1147,6 +1168,9 @@ TEST_P(OEMCryptoUsageTableTest, OEMCryptoMemoryReportUsageForHugeReportBuffer) {
 }
 
 TEST_P(OEMCryptoUsageTableTest, OEMCryptoMemoryReportUsageForHugePstBuffer) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   auto oemcrypto_function = [&](size_t buffer_length) {
     LicenseWithUsageEntry entry;
     entry.license_messages().set_api_version(license_api_version_);
@@ -1167,6 +1191,9 @@ TEST_P(OEMCryptoUsageTableTest, OEMCryptoMemoryReportUsageForHugePstBuffer) {
 
 TEST_P(OEMCryptoUsageTableTest,
        OEMCryptoMemoryShrinkUsageTableHeaderForHugeHeaderBufferLength) {
+  if (!wvoec::global_features.usage_table) {
+    GTEST_SKIP() << "Usage tables are not supported.";
+  }
   LicenseWithUsageEntry entry0;
   entry0.set_pst("pst 0");
   entry0.MakeOfflineAndClose(this);