Added an oemcrypto test for OPK versioning

Merge from Widevine repo of http://go/wvgerrit/125645 bug: 158857733 test: opk_linux_ipc_ta Change-Id: Ibcbb143fa0eb7d2b4e14e3cc9afbe3a219c7c727
2021-10-13 22:00:06 +00:00
parent d7b377f976
commit 882d3ed075
2 changed files with 125 additions and 0 deletions
--- a/libwvdrmengine/oemcrypto/test/oemcrypto_serialization_version_test.cpp
+++ b/libwvdrmengine/oemcrypto/test/oemcrypto_serialization_version_test.cpp
@@ -0,0 +1,107 @@
+// Copyright 2021 Google LLC. All Rights Reserved. This file and proprietary
+// source code may only be used and distributed under the Widevine
+// License Agreement.
+//
+
+#include <iostream>
+
+#include <gtest/gtest.h>
+
+#include "OEMCryptoCENC.h"
+#include "log.h"
+#include "oec_test_data.h"
+
+using namespace std;
+
+namespace wvoec {
+
+class OEMCryptoTest : public ::testing::Test {
+ protected:
+  OEMCryptoTest() {}
+
+  void SetUp() override {
+    ::testing::Test::SetUp();
+    const ::testing::TestInfo* const test_info =
+        ::testing::UnitTest::GetInstance()->current_test_info();
+    LOGD("Running test %s.%s", test_info->test_case_name(), test_info->name());
+    OEMCrypto_SetSandbox(kTestSandbox, sizeof(kTestSandbox));
+    ASSERT_EQ(OEMCrypto_SUCCESS, OEMCrypto_Initialize());
+  }
+
+  void TearDown() override {
+    OEMCrypto_Terminate();
+    ::testing::Test::TearDown();
+  }
+};
+
+TEST_F(OEMCryptoTest, OPK_SerializationVersion) {
+  uint32_t ree_major = 0;
+  uint32_t ree_minor = 0;
+  uint32_t tee_major = 0;
+  uint32_t tee_minor = 0;
+  OEMCryptoResult sts = OEMCrypto_OPK_SerializationVersion(
+      &ree_major, &ree_minor, &tee_major, &tee_minor);
+  if (sts != OEMCrypto_ERROR_NOT_IMPLEMENTED) {
+    EXPECT_EQ(OEMCrypto_SUCCESS, sts);
+    cout << "OPK REE serialization version is " << ree_major << "." << ree_minor
+         << "\n";
+    cout << "OPK TEE serialization version is " << tee_major << "." << tee_minor
+         << "\n";
+    EXPECT_NE(ree_major, 0u);
+    EXPECT_NE(tee_major, 0u);
+
+    const uint32_t orig_ree_major = ree_major;
+    const uint32_t orig_ree_minor = ree_minor;
+    const uint32_t orig_tee_major = tee_major;
+    const uint32_t orig_tee_minor = tee_minor;
+
+    /* test requiring a specific version, should create incompatibility */
+    ree_major = 100;
+    ree_minor = 1;
+    sts = OEMCrypto_OPK_SerializationVersion(&ree_major, &ree_minor, &tee_major,
+                                             &tee_minor);
+    EXPECT_EQ(OPK_ERROR_INCOMPATIBLE_VERSION, sts);
+    EXPECT_EQ(ree_major, 100u);
+    EXPECT_EQ(ree_minor, 1u);
+
+    /* now OEMCrypto_Initialize should fail with incompatible version */
+    OEMCrypto_Terminate();
+    EXPECT_EQ(OPK_ERROR_INCOMPATIBLE_VERSION, OEMCrypto_Initialize());
+
+    /* restore the default versions */
+    ree_major = 0;
+    ree_minor = 0;
+    sts = OEMCrypto_OPK_SerializationVersion(&ree_major, &ree_minor, &tee_major,
+                                             &tee_minor);
+    EXPECT_EQ(OEMCrypto_SUCCESS, sts);
+    EXPECT_EQ(ree_major, orig_ree_major);
+    EXPECT_EQ(ree_minor, orig_ree_minor);
+    EXPECT_EQ(tee_major, orig_tee_major);
+    EXPECT_EQ(tee_minor, orig_tee_minor);
+
+    /* OEMCrypto_Initialize should work now */
+    OEMCrypto_Terminate();
+    EXPECT_EQ(OEMCrypto_SUCCESS, OEMCrypto_Initialize());
+
+    /* changing minor version shouldn't create incompatibility */
+    ree_minor++;
+    sts = OEMCrypto_OPK_SerializationVersion(&ree_major, &ree_minor, &tee_major,
+                                             &tee_minor);
+    EXPECT_EQ(ree_minor, orig_ree_minor + 1);
+    EXPECT_EQ(OEMCrypto_SUCCESS, sts);
+
+    /* OEMCrypto_Initialize should still work */
+    OEMCrypto_Terminate();
+    EXPECT_EQ(OEMCrypto_SUCCESS, OEMCrypto_Initialize());
+
+    /* remove the REE version override so subsequent tests work */
+    ree_major = 0;
+    ree_minor = 0;
+    sts = OEMCrypto_OPK_SerializationVersion(&ree_major, &ree_minor, &tee_major,
+                                             &tee_minor);
+    EXPECT_EQ(OEMCrypto_SUCCESS, sts);
+    EXPECT_EQ(OEMCrypto_SUCCESS, OEMCrypto_Initialize());
+  }
+}
+
+}  // namespace wvoec