Nvidia-gpu: Encode/decode APIs for GPU Inventory
Added support for encoding and decoding Get Inventory command.
The command supports fetching inventory properties including Serial
Number, Part Number, Marketing Name, etc for Nvidia GPUs
Tested
- added new UT for encode and decode APIs which pass
Change-Id: I9e5afbe356b64fd7ae4f7a2a65043f3eeffa3807
Signed-off-by: Rohit PAI <ropai@nvidia.com>
diff --git a/src/nvidia-gpu/tests/NvidiaDeviceInventoryMctpVdm.cpp b/src/nvidia-gpu/tests/NvidiaDeviceInventoryMctpVdm.cpp
new file mode 100644
index 0000000..09bea75
--- /dev/null
+++ b/src/nvidia-gpu/tests/NvidiaDeviceInventoryMctpVdm.cpp
@@ -0,0 +1,115 @@
+#include "NvidiaGpuMctpVdm.hpp"
+#include "OcpMctpVdm.hpp"
+
+#include <endian.h>
+
+#include <array>
+#include <cstdint>
+#include <cstring>
+#include <string>
+#include <variant>
+#include <vector>
+
+#include <gtest/gtest.h>
+
+using namespace gpu;
+// NOLINTBEGIN(cppcoreguidelines-pro-type-reinterpret-cast)
+TEST(NvidiaGpuMctpVdmTest, EncodeGetInventoryInformationRequest)
+{
+ std::array<uint8_t, 256> buf{};
+ uint8_t instanceId = 1;
+ uint8_t propertyId =
+ static_cast<uint8_t>(InventoryPropertyId::BOARD_PART_NUMBER);
+
+ auto rc = encodeGetInventoryInformationRequest(instanceId, propertyId, buf);
+ EXPECT_EQ(rc, 0);
+
+ auto* msg = reinterpret_cast<GetInventoryInformationRequest*>(buf.data());
+ EXPECT_EQ(msg->hdr.command,
+ static_cast<uint8_t>(
+ PlatformEnvironmentalCommands::GET_INVENTORY_INFORMATION));
+ EXPECT_EQ(msg->hdr.data_size, sizeof(propertyId));
+ EXPECT_EQ(msg->property_id, propertyId);
+}
+
+TEST(NvidiaGpuMctpVdmTest, DecodeInventoryString)
+{
+ std::array<uint8_t, 256> buf{};
+ auto* response =
+ reinterpret_cast<ocp::accelerator_management::CommonResponse*>(
+ buf.data());
+
+ // Fill header
+ response->msgHdr.hdr.pci_vendor_id = htobe16(0x10DE); // NVIDIA vendor ID
+ response->msgHdr.hdr.instance_id = 0x01; // Instance ID
+ response->msgHdr.hdr.ocp_version = 0x89; // OCP version and type
+ response->msgHdr.hdr.ocp_accelerator_management_msg_type =
+ static_cast<uint8_t>(
+ ocp::accelerator_management::MessageType::RESPONSE);
+
+ response->command = static_cast<uint8_t>(
+ PlatformEnvironmentalCommands::GET_INVENTORY_INFORMATION);
+ response->completion_code = static_cast<uint8_t>(
+ ocp::accelerator_management::CompletionCode::SUCCESS);
+ response->reserved = 0;
+ response->data_size = htole16(5); // 5 bytes for "TEST1"
+
+ const char* testStr = "TEST1";
+ memcpy(buf.data() + sizeof(ocp::accelerator_management::CommonResponse),
+ testStr, 5);
+
+ ocp::accelerator_management::CompletionCode cc =
+ ocp::accelerator_management::CompletionCode::ERROR;
+ uint16_t reasonCode = 0;
+ InventoryValue info;
+
+ auto rc = decodeGetInventoryInformationResponse(
+ buf, cc, reasonCode, InventoryPropertyId::BOARD_PART_NUMBER, info);
+ EXPECT_EQ(rc, 0);
+ EXPECT_EQ(cc, ocp::accelerator_management::CompletionCode::SUCCESS);
+ EXPECT_EQ(reasonCode, 0);
+ EXPECT_TRUE(std::holds_alternative<std::string>(info));
+ EXPECT_EQ(std::get<std::string>(info), "TEST1");
+}
+
+TEST(NvidiaGpuMctpVdmTest, DecodeInventoryDeviceGuid)
+{
+ std::array<uint8_t, 256> buf{};
+ auto* response =
+ reinterpret_cast<ocp::accelerator_management::CommonResponse*>(
+ buf.data());
+
+ // Fill header
+ response->msgHdr.hdr.pci_vendor_id = htobe16(0x10DE); // NVIDIA vendor ID
+ response->msgHdr.hdr.instance_id = 0x01; // Instance ID
+ response->msgHdr.hdr.ocp_version = 0x89; // OCP version and type
+ response->msgHdr.hdr.ocp_accelerator_management_msg_type =
+ static_cast<uint8_t>(
+ ocp::accelerator_management::MessageType::RESPONSE);
+
+ response->command = static_cast<uint8_t>(
+ PlatformEnvironmentalCommands::GET_INVENTORY_INFORMATION);
+ response->completion_code = static_cast<uint8_t>(
+ ocp::accelerator_management::CompletionCode::SUCCESS);
+ response->reserved = 0;
+ response->data_size = htole16(8); // 8 bytes for DEVICE_GUID
+
+ std::vector<uint8_t> dummyGuid = {0xDE, 0xAD, 0xBE, 0xEF,
+ 0x01, 0x23, 0x45, 0x67};
+ memcpy(buf.data() + sizeof(ocp::accelerator_management::CommonResponse),
+ dummyGuid.data(), dummyGuid.size());
+
+ ocp::accelerator_management::CompletionCode cc =
+ ocp::accelerator_management::CompletionCode::ERROR;
+ uint16_t reasonCode = 0;
+ InventoryValue info;
+
+ auto rc = decodeGetInventoryInformationResponse(
+ buf, cc, reasonCode, InventoryPropertyId::DEVICE_GUID, info);
+ EXPECT_EQ(rc, 0);
+ EXPECT_EQ(cc, ocp::accelerator_management::CompletionCode::SUCCESS);
+ EXPECT_EQ(reasonCode, 0);
+ EXPECT_TRUE(std::holds_alternative<std::vector<uint8_t>>(info));
+ EXPECT_EQ(std::get<std::vector<uint8_t>>(info), dummyGuid);
+}
+// NOLINTEND(cppcoreguidelines-pro-type-reinterpret-cast)
diff --git a/src/nvidia-gpu/tests/meson.build b/src/nvidia-gpu/tests/meson.build
index 4923868..99ed721 100644
--- a/src/nvidia-gpu/tests/meson.build
+++ b/src/nvidia-gpu/tests/meson.build
@@ -19,6 +19,7 @@
executable(
'nvidiagpusensor_test',
'NvidiaGpuSensorTest.cpp',
+ 'NvidiaDeviceInventoryMctpVdm.cpp',
'../OcpMctpVdm.cpp',
'../NvidiaGpuMctpVdm.cpp',
implicit_include_directories: false,