Update: Adding tests for flashing GFX, OptionROM firmwares

Related-To: VLCLJ-2532

Signed-off-by: Alle, Kalyan <[email protected]>

Author: kalyanalle

conformance_tests/sysman/test_sysman_firmware/src/test_sysman_firmware.cpp

@@ -303,4 +303,137 @@ LZT_TEST_F(
   }
 }
 
+LZT_TEST_F(
+    FIRMWARE_TEST,
+    GivenValidFirmwareHandleWhenFlashingGfxFirmwareThenExpectFirmwareFlashingSuccess) {
+  auto fwDirEnv = getenv("ZE_LZT_FIRMWARE_DIRECTORY");
+  if (nullptr == fwDirEnv) {
+    LOG_INFO << "Skipping test as ZE_LZT_FIRMWARE_DIRECTORY not set";
+    GTEST_SKIP();
+  }
+  std::vector<char> testFwImage;
+  std::string fwDir(fwDirEnv);
+
+  for (auto device : devices) {
+    uint32_t count = 0;
+    auto firmware_handles = lzt::get_firmware_handles(device, count);
+    if (count == 0) {
+      FAIL() << "No handles found: "
+             << _ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+    }
+
+    for (auto firmware_handle : firmware_handles) {
+      ASSERT_NE(nullptr, firmware_handle);
+      auto propFw = lzt::get_firmware_properties(firmware_handle);
+
+      if (propFw.canControl == true &&
+          std::string(reinterpret_cast<char *>(propFw.name)) == "GFX") {
+        std::string fwName(reinterpret_cast<char *>(propFw.name));
+        std::string fwToLoad = fwDir + "/" + fwName + ".bin";
+        LOG_INFO << "Firmware Name: " << fwName;
+        LOG_INFO << "Firmware Path: " << fwToLoad;
+        std::ifstream inFileStream(fwToLoad, std::ios::binary | std::ios::ate);
+        if (!inFileStream.is_open()) {
+          LOG_INFO << "Skipping test as firmware image not found";
+          GTEST_SKIP();
+        }
+        testFwImage.resize(inFileStream.tellg());
+        inFileStream.seekg(0, inFileStream.beg);
+        inFileStream.read(testFwImage.data(), testFwImage.size());
+        lzt::flash_firmware(firmware_handle,
+                            static_cast<void *>(testFwImage.data()),
+                            testFwImage.size());
+      }
+    }
+  }
+}
+
+LZT_TEST_F(
+    FIRMWARE_TEST,
+    GivenValidFirmwareHandleWhenFlashingOptionRomDataThenExpectFirmwareFlashingSuccess) {
+  auto fwDirEnv = getenv("ZE_LZT_FIRMWARE_DIRECTORY");
+  if (nullptr == fwDirEnv) {
+    LOG_INFO << "Skipping test as ZE_LZT_FIRMWARE_DIRECTORY not set";
+    GTEST_SKIP();
+  }
+  std::vector<char> testFwImage;
+  std::string fwDir(fwDirEnv);
+
+  for (auto device : devices) {
+    uint32_t count = 0;
+    auto firmware_handles = lzt::get_firmware_handles(device, count);
+    if (count == 0) {
+      FAIL() << "No handles found: "
+             << _ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+    }
+
+    for (auto firmware_handle : firmware_handles) {
+      ASSERT_NE(nullptr, firmware_handle);
+      auto propFw = lzt::get_firmware_properties(firmware_handle);
+      if (propFw.canControl == true &&
+          std::string(reinterpret_cast<char *>(propFw.name)) == "OptionROM") {
+        std::string fwName(reinterpret_cast<char *>(propFw.name));
+        std::string fwToLoad = fwDir + "/" + fwName + ".bin";
+        LOG_INFO << "Firmware Name: " << fwName;
+        LOG_INFO << "Firmware Path: " << fwToLoad;
+        std::ifstream inFileStream(fwToLoad, std::ios::binary | std::ios::ate);
+        if (!inFileStream.is_open()) {
+          LOG_INFO << "Skipping test as firmware image not found";
+          GTEST_SKIP();
+        }
+        testFwImage.resize(inFileStream.tellg());
+        inFileStream.seekg(0, inFileStream.beg);
+        inFileStream.read(testFwImage.data(), testFwImage.size());
+        lzt::flash_firmware(firmware_handle,
+                            static_cast<void *>(testFwImage.data()),
+                            testFwImage.size());
+      }
+    }
+  }
+}
+
+LZT_TEST_F(
+    FIRMWARE_TEST,
+    GivenValidFirmwareHandleWhenFlashingOptionRomCodeThenExpectFirmwareFlashingSuccess) {
+  auto fwDirEnv = getenv("ZE_LZT_FIRMWARE_DIRECTORY");
+  if (nullptr == fwDirEnv) {
+    LOG_INFO << "Skipping test as ZE_LZT_FIRMWARE_DIRECTORY not set";
+    GTEST_SKIP();
+  }
+  std::vector<char> testFwImage;
+  std::string fwDir(fwDirEnv);
+
+  for (auto device : devices) {
+    uint32_t count = 0;
+    auto firmware_handles = lzt::get_firmware_handles(device, count);
+    if (count == 0) {
+      FAIL() << "No handles found: "
+             << _ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+    }
+    for (auto firmware_handle : firmware_handles) {
+      ASSERT_NE(nullptr, firmware_handle);
+      auto propFw = lzt::get_firmware_properties(firmware_handle);
+      if (propFw.canControl == true &&
+          std::string(reinterpret_cast<char *>(propFw.name)) == "OptionROM") {
+        std::string fwName(reinterpret_cast<char *>(propFw.name));
+        std::string fwToLoad = fwDir + "/" + fwName + ".bin";
+        LOG_INFO << "Firmware Name: " << fwName;
+        LOG_INFO << "Firmware Path: " << fwToLoad;
+        std::ifstream inFileStream(fwToLoad, std::ios::binary | std::ios::ate);
+        if (!inFileStream.is_open()) {
+          LOG_INFO << "Skipping test as firmware image not found";
+          GTEST_SKIP();
+        }
+        testFwImage.resize(inFileStream.tellg());
+        inFileStream.seekg(0, inFileStream.beg);
+        inFileStream.read(testFwImage.data(), testFwImage.size());
+        lzt::flash_firmware(firmware_handle,
+                            static_cast<void *>(testFwImage.data()),
+                            testFwImage.size());
+        track_firmware_flash(firmware_handle);
+      }
+    }
+  }
+}
+
 } // namespace