Merge "Unit test for OEMCrypto_LoadUsageEntry in multiple sessions" into rvc-dev

libwvdrmengine/oemcrypto/test/oemcrypto_test.cpp

@@ -5190,6 +5190,32 @@ TEST_P(OEMCryptoUsageTableTest, CreateAndLoadMultipleEntriesAPI16) {
       OEMCrypto_CreateNewUsageEntry(s2.session_id(), &usage_entry_number));
 }
 
+// An entry can be loaded in only one session at a time.
+TEST_P(OEMCryptoUsageTableTest, LoadEntryInMultipleSessions) {
+  // Entry Count: we start each test with an empty header.
+  LicenseWithUsageEntry entry;
+  entry.license_messages().set_api_version(license_api_version_);
+  Session& s = entry.session();
+  // Make first entry 0.
+  ASSERT_NO_FATAL_FAILURE(entry.MakeOfflineAndClose(this));
+  const uint32_t usage_entry_number = s.usage_entry_number();
+  EXPECT_EQ(usage_entry_number, 0u);  // Should be only entry in this test.
+
+  // Load an entry, then try to create a second.
+  ASSERT_NO_FATAL_FAILURE(s.open());
+  // Reload entry 0.
+  ASSERT_NO_FATAL_FAILURE(s.ReloadUsageEntry());
+
+  // Create an entry, then try to load a second.
+  Session s2;
+  ASSERT_NO_FATAL_FAILURE(s2.open());
+  // Try to load entry 0 into session 2.
+  ASSERT_EQ(OEMCrypto_ERROR_INVALID_SESSION,
+            OEMCrypto_LoadUsageEntry(s2.session_id(), usage_entry_number,
+                                     s.encrypted_usage_entry().data(),
+                                     s.encrypted_usage_entry().size()));
+}
+
 // Test generic encrypt when the license uses a PST.
 TEST_P(OEMCryptoUsageTableTest, GenericCryptoEncrypt) {
   LicenseWithUsageEntry entry;