MdeModulePkg/NonDiscoverablePciDeviceDxe: expose unique B/D/F identifiers

[mirror_edk2.git] / MdeModulePkg / Bus / Pci / NonDiscoverablePciDeviceDxe / NonDiscoverablePciDeviceDxe.c

diff --git a/MdeModulePkg/Bus/Pci/NonDiscoverablePciDeviceDxe/NonDiscoverablePciDeviceDxe.c b/MdeModulePkg/Bus/Pci/NonDiscoverablePciDeviceDxe/NonDiscoverablePciDeviceDxe.c
index 3e9ff6620d8da4c74ed839c25e045640508bc853..2483c7296912ea5d87044e2b52a9f4c345b9d0c2 100644 (file)
--- a/MdeModulePkg/Bus/Pci/NonDiscoverablePciDeviceDxe/NonDiscoverablePciDeviceDxe.c
+++ b/MdeModulePkg/Bus/Pci/NonDiscoverablePciDeviceDxe/NonDiscoverablePciDeviceDxe.c
@@ -16,6 +16,9 @@
 \r
 #include <Protocol/DriverBinding.h>\r
 \r
+#define MAX_NON_DISCOVERABLE_PCI_DEVICE_ID   (32 * 256)\r
+\r
+STATIC UINTN               mUniqueIdCounter = 0;\r
 EFI_CPU_ARCH_PROTOCOL      *mCpu;\r
 \r
 //\r
@@ -141,6 +144,11 @@ NonDiscoverablePciDeviceStart (
   NON_DISCOVERABLE_PCI_DEVICE   *Dev;\r
   EFI_STATUS                    Status;\r
 \r
+  ASSERT (mUniqueIdCounter < MAX_NON_DISCOVERABLE_PCI_DEVICE_ID);\r
+  if (mUniqueIdCounter >= MAX_NON_DISCOVERABLE_PCI_DEVICE_ID) {\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+\r
   Dev = AllocateZeroPool (sizeof *Dev);\r
   if (Dev == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
@@ -167,6 +175,8 @@ NonDiscoverablePciDeviceStart (
     goto CloseProtocol;\r
   }\r
 \r
+  Dev->UniqueId = mUniqueIdCounter++;\r
+\r
   return EFI_SUCCESS;\r
 \r
 CloseProtocol:\r