]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdePkg/Library/DxeRuntimePciExpressLib/PciExpressLib.c
UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm
[mirror_edk2.git] / MdePkg / Library / DxeRuntimePciExpressLib / PciExpressLib.c
index 6e784763be73c6cc8d366be21774b32662c12003..a9f380776c001bcf6b6a72364ac1d15741c2eba4 100644 (file)
@@ -10,7 +10,6 @@
 \r
 **/\r
 \r
-\r
 #include <PiDxe.h>\r
 \r
 #include <Guid/EventGroup.h>\r
 #include <Library/DxeServicesTableLib.h>\r
 #include <Library/UefiRuntimeLib.h>\r
 \r
+/**\r
+ Assert the validity of a PCI address. A valid PCI address should contain 1's\r
+ only in the low 28 bits.\r
+\r
+ @param A The address to validate.\r
+\r
+**/\r
+#define ASSERT_INVALID_PCI_ADDRESS(A) \\r
+ ASSERT (((A) & ~0xfffffff) == 0)\r
+\r
 ///\r
 /// Define table for mapping PCI Express MMIO physical addresses to virtual addresses at OS runtime\r
 ///\r
 typedef struct {\r
-  UINTN  PhysicalAddress;\r
-  UINTN  VirtualAddress;\r
+  UINTN    PhysicalAddress;\r
+  UINTN    VirtualAddress;\r
 } PCI_EXPRESS_RUNTIME_REGISTRATION_TABLE;\r
 \r
 ///\r
 /// Set Virtual Address Map Event\r
 ///\r
-EFI_EVENT                               mDxeRuntimePciExpressLibVirtualNotifyEvent = NULL;\r
+EFI_EVENT  mDxeRuntimePciExpressLibVirtualNotifyEvent = NULL;\r
 \r
 ///\r
-/// Module global that contains the base physical address of the PCI Express MMIO range.\r
+/// Module global that contains the base physical address and size of the PCI Express MMIO range.\r
 ///\r
-UINTN                                   mDxeRuntimePciExpressLibPciExpressBaseAddress = 0;\r
+UINTN  mDxeRuntimePciExpressLibPciExpressBaseAddress = 0;\r
+UINTN  mDxeRuntimePciExpressLibPciExpressBaseSize    = 0;\r
 \r
 ///\r
 /// The number of PCI devices that have been registered for runtime access.\r
 ///\r
-UINTN                                   mDxeRuntimePciExpressLibNumberOfRuntimeRanges = 0;\r
+UINTN  mDxeRuntimePciExpressLibNumberOfRuntimeRanges = 0;\r
 \r
 ///\r
 /// The table of PCI devices that have been registered for runtime access.\r
@@ -56,8 +66,7 @@ PCI_EXPRESS_RUNTIME_REGISTRATION_TABLE  *mDxeRuntimePciExpressLibRegistrationTab
 ///\r
 /// The table index of the most recent virtual address lookup.\r
 ///\r
-UINTN                                   mDxeRuntimePciExpressLibLastRuntimeRange = 0;\r
-\r
+UINTN  mDxeRuntimePciExpressLibLastRuntimeRange = 0;\r
 \r
 /**\r
   Convert the physical PCI Express MMIO addresses for all registered PCI devices\r
@@ -87,13 +96,13 @@ DxeRuntimePciExpressLibVirtualNotify (
   // virtual addresses.\r
   //\r
   for (Index = 0; Index < mDxeRuntimePciExpressLibNumberOfRuntimeRanges; Index++) {\r
-    EfiConvertPointer (0, (VOID **) &(mDxeRuntimePciExpressLibRegistrationTable[Index].VirtualAddress));\r
+    EfiConvertPointer (0, (VOID **)&(mDxeRuntimePciExpressLibRegistrationTable[Index].VirtualAddress));\r
   }\r
 \r
   //\r
   // Convert table pointer that is allocated from EfiRuntimeServicesData to a virtual address.\r
   //\r
-  EfiConvertPointer (0, (VOID **) &mDxeRuntimePciExpressLibRegistrationTable);\r
+  EfiConvertPointer (0, (VOID **)&mDxeRuntimePciExpressLibRegistrationTable);\r
 }\r
 \r
 /**\r
@@ -119,17 +128,17 @@ DxeRuntimePciExpressLibConstructor (
   //\r
   // Cache the physical address of the PCI Express MMIO range into a module global variable\r
   //\r
-  mDxeRuntimePciExpressLibPciExpressBaseAddress = (UINTN) PcdGet64 (PcdPciExpressBaseAddress);\r
+  mDxeRuntimePciExpressLibPciExpressBaseAddress = (UINTN)PcdGet64 (PcdPciExpressBaseAddress);\r
+  mDxeRuntimePciExpressLibPciExpressBaseSize    = (UINTN)PcdGet64 (PcdPciExpressBaseSize);\r
 \r
   //\r
   // Register SetVirtualAddressMap () notify function\r
   //\r
-  Status = gBS->CreateEventEx (\r
-                  EVT_NOTIFY_SIGNAL,\r
+  Status = gBS->CreateEvent (\r
+                  EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,\r
                   TPL_NOTIFY,\r
                   DxeRuntimePciExpressLibVirtualNotify,\r
                   NULL,\r
-                  &gEfiEventVirtualAddressChangeGuid,\r
                   &mDxeRuntimePciExpressLibVirtualNotifyEvent\r
                   );\r
   ASSERT_EFI_ERROR (Status);\r
@@ -180,8 +189,12 @@ DxeRuntimePciExpressLibDestructor (
   This internal functions retrieves PCI Express Base Address via a PCD entry\r
   PcdPciExpressBaseAddress.\r
 \r
-  @param  Address  The address that encodes the PCI Bus, Device, Function and Register.\r
-  @return          The base address of PCI Express.\r
+  If Address > 0x0FFFFFFF, then ASSERT().\r
+\r
+  @param  Address   The address that encodes the PCI Bus, Device, Function and Register.\r
+\r
+  @retval (UINTN)-1 Invalid PCI address.\r
+  @retval other     The base address of PCI Express.\r
 \r
 **/\r
 UINTN\r
@@ -194,7 +207,14 @@ GetPciExpressAddress (
   //\r
   // Make sure Address is valid\r
   //\r
-  ASSERT (((Address) & ~0xfffffff) == 0);\r
+  ASSERT_INVALID_PCI_ADDRESS (Address);\r
+\r
+  //\r
+  // Make sure the Address is in MMCONF address space\r
+  //\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINTN)-1;\r
+  }\r
 \r
   //\r
   // Convert Address to a physical address in the MMIO PCI Express range\r
@@ -237,8 +257,7 @@ GetPciExpressAddress (
   //\r
   // No match was found.  This is a critical error at OS runtime, so ASSERT() and force a breakpoint.\r
   //\r
-  ASSERT (FALSE);\r
-  CpuBreakpoint();\r
+  CpuBreakpoint ();\r
 \r
   //\r
   // Return the physical address\r
@@ -289,7 +308,14 @@ PciExpressRegisterForRuntimeAccess (
   //\r
   // Make sure Address is valid\r
   //\r
-  ASSERT (((Address) & ~0xfffffff) == 0);\r
+  ASSERT_INVALID_PCI_ADDRESS (Address);\r
+\r
+  //\r
+  // Make sure the Address is in MMCONF address space\r
+  //\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return RETURN_UNSUPPORTED;\r
+  }\r
 \r
   //\r
   // Convert Address to a physical address in the MMIO PCI Express range\r
@@ -299,7 +325,7 @@ PciExpressRegisterForRuntimeAccess (
   Address = GetPciExpressAddress (Address & 0x0ffff000);\r
 \r
   //\r
-  // See if Address has already been registerd for runtime access\r
+  // See if Address has already been registered for runtime access\r
   //\r
   for (Index = 0; Index < mDxeRuntimePciExpressLibNumberOfRuntimeRanges; Index++) {\r
     if (mDxeRuntimePciExpressLibRegistrationTable[Index].PhysicalAddress == Address) {\r
@@ -335,7 +361,8 @@ PciExpressRegisterForRuntimeAccess (
   if (NewTable == NULL) {\r
     return RETURN_OUT_OF_RESOURCES;\r
   }\r
-  mDxeRuntimePciExpressLibRegistrationTable = NewTable;\r
+\r
+  mDxeRuntimePciExpressLibRegistrationTable                                                                = NewTable;\r
   mDxeRuntimePciExpressLibRegistrationTable[mDxeRuntimePciExpressLibNumberOfRuntimeRanges].PhysicalAddress = Address;\r
   mDxeRuntimePciExpressLibRegistrationTable[mDxeRuntimePciExpressLibNumberOfRuntimeRanges].VirtualAddress  = Address;\r
   mDxeRuntimePciExpressLibNumberOfRuntimeRanges++;\r
@@ -343,7 +370,6 @@ PciExpressRegisterForRuntimeAccess (
   return RETURN_SUCCESS;\r
 }\r
 \r
-\r
 /**\r
   Reads an 8-bit PCI configuration register.\r
 \r
@@ -355,16 +381,21 @@ PciExpressRegisterForRuntimeAccess (
 \r
   @param  Address The address that encodes the PCI Bus, Device, Function and\r
                   Register.\r
-\r
-  @return The read value from the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The read value from the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressRead8 (\r
-  IN      UINTN                     Address\r
+  IN      UINTN  Address\r
   )\r
 {\r
+  ASSERT_INVALID_PCI_ADDRESS (Address);\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioRead8 (GetPciExpressAddress (Address));\r
 }\r
 \r
@@ -381,16 +412,21 @@ PciExpressRead8 (
                   Register.\r
   @param  Value   The value to write.\r
 \r
-  @return The value written to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressWrite8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT8                     Value\r
+  IN      UINTN  Address,\r
+  IN      UINT8  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioWrite8 (GetPciExpressAddress (Address), Value);\r
 }\r
 \r
@@ -411,16 +447,21 @@ PciExpressWrite8 (
                   Register.\r
   @param  OrData  The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressOr8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT8                     OrData\r
+  IN      UINTN  Address,\r
+  IN      UINT8  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioOr8 (GetPciExpressAddress (Address), OrData);\r
 }\r
 \r
@@ -441,16 +482,21 @@ PciExpressOr8 (
                   Register.\r
   @param  AndData The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other  The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressAnd8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT8                     AndData\r
+  IN      UINTN  Address,\r
+  IN      UINT8  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioAnd8 (GetPciExpressAddress (Address), AndData);\r
 }\r
 \r
@@ -473,17 +519,22 @@ PciExpressAnd8 (
   @param  AndData The value to AND with the PCI configuration register.\r
   @param  OrData  The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressAndThenOr8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT8                     AndData,\r
-  IN      UINT8                     OrData\r
+  IN      UINTN  Address,\r
+  IN      UINT8  AndData,\r
+  IN      UINT8  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioAndThenOr8 (\r
            GetPciExpressAddress (Address),\r
            AndData,\r
@@ -509,17 +560,22 @@ PciExpressAndThenOr8 (
   @param  EndBit    The ordinal of the most significant bit in the bit field.\r
                     Range 0..7.\r
 \r
-  @return The value of the bit field read from the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value of the bit field read from the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressBitFieldRead8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioBitFieldRead8 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -548,18 +604,23 @@ PciExpressBitFieldRead8 (
                     Range 0..7.\r
   @param  Value     The new value of the bit field.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressBitFieldWrite8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT8                     Value\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit,\r
+  IN      UINT8  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioBitFieldWrite8 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -592,18 +653,23 @@ PciExpressBitFieldWrite8 (
                     Range 0..7.\r
   @param  OrData    The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressBitFieldOr8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT8                     OrData\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit,\r
+  IN      UINT8  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioBitFieldOr8 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -636,18 +702,23 @@ PciExpressBitFieldOr8 (
                     Range 0..7.\r
   @param  AndData   The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressBitFieldAnd8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT8                     AndData\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit,\r
+  IN      UINT8  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioBitFieldAnd8 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -684,19 +755,24 @@ PciExpressBitFieldAnd8 (
   @param  AndData   The value to AND with the PCI configuration register.\r
   @param  OrData    The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFF  Invalid PCI address.\r
+  @retval other The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT8\r
 EFIAPI\r
 PciExpressBitFieldAndThenOr8 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT8                     AndData,\r
-  IN      UINT8                     OrData\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit,\r
+  IN      UINT8  AndData,\r
+  IN      UINT8  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT8)-1;\r
+  }\r
+\r
   return MmioBitFieldAndThenOr8 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -719,15 +795,20 @@ PciExpressBitFieldAndThenOr8 (
   @param  Address The address that encodes the PCI Bus, Device, Function and\r
                   Register.\r
 \r
-  @return The read value from the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The read value from the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressRead16 (\r
-  IN      UINTN                     Address\r
+  IN      UINTN  Address\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioRead16 (GetPciExpressAddress (Address));\r
 }\r
 \r
@@ -745,16 +826,21 @@ PciExpressRead16 (
                   Register.\r
   @param  Value   The value to write.\r
 \r
-  @return The value written to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressWrite16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT16                    Value\r
+  IN      UINTN   Address,\r
+  IN      UINT16  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioWrite16 (GetPciExpressAddress (Address), Value);\r
 }\r
 \r
@@ -776,16 +862,21 @@ PciExpressWrite16 (
                   Register.\r
   @param  OrData  The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressOr16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT16                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINT16  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioOr16 (GetPciExpressAddress (Address), OrData);\r
 }\r
 \r
@@ -807,16 +898,21 @@ PciExpressOr16 (
                   Register.\r
   @param  AndData The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressAnd16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT16                    AndData\r
+  IN      UINTN   Address,\r
+  IN      UINT16  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioAnd16 (GetPciExpressAddress (Address), AndData);\r
 }\r
 \r
@@ -840,17 +936,22 @@ PciExpressAnd16 (
   @param  AndData The value to AND with the PCI configuration register.\r
   @param  OrData  The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressAndThenOr16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT16                    AndData,\r
-  IN      UINT16                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINT16  AndData,\r
+  IN      UINT16  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioAndThenOr16 (\r
            GetPciExpressAddress (Address),\r
            AndData,\r
@@ -877,17 +978,22 @@ PciExpressAndThenOr16 (
   @param  EndBit    The ordinal of the most significant bit in the bit field.\r
                     Range 0..15.\r
 \r
-  @return The value of the bit field read from the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value of the bit field read from the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressBitFieldRead16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioBitFieldRead16 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -917,18 +1023,23 @@ PciExpressBitFieldRead16 (
                     Range 0..15.\r
   @param  Value     The new value of the bit field.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressBitFieldWrite16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT16                    Value\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT16  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioBitFieldWrite16 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -962,18 +1073,23 @@ PciExpressBitFieldWrite16 (
                     Range 0..15.\r
   @param  OrData    The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressBitFieldOr16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT16                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT16  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioBitFieldOr16 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1007,18 +1123,23 @@ PciExpressBitFieldOr16 (
                     Range 0..15.\r
   @param  AndData   The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressBitFieldAnd16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT16                    AndData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT16  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioBitFieldAnd16 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1056,19 +1177,24 @@ PciExpressBitFieldAnd16 (
   @param  AndData   The value to AND with the PCI configuration register.\r
   @param  OrData    The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT16\r
 EFIAPI\r
 PciExpressBitFieldAndThenOr16 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT16                    AndData,\r
-  IN      UINT16                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT16  AndData,\r
+  IN      UINT16  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT16)-1;\r
+  }\r
+\r
   return MmioBitFieldAndThenOr16 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1091,15 +1217,20 @@ PciExpressBitFieldAndThenOr16 (
   @param  Address The address that encodes the PCI Bus, Device, Function and\r
                   Register.\r
 \r
-  @return The read value from the PCI configuration register.\r
+  @retval 0xFFFF  Invalid PCI address.\r
+  @retval other   The read value from the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressRead32 (\r
-  IN      UINTN                     Address\r
+  IN      UINTN  Address\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioRead32 (GetPciExpressAddress (Address));\r
 }\r
 \r
@@ -1117,16 +1248,21 @@ PciExpressRead32 (
                   Register.\r
   @param  Value   The value to write.\r
 \r
-  @return The value written to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressWrite32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT32                    Value\r
+  IN      UINTN   Address,\r
+  IN      UINT32  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioWrite32 (GetPciExpressAddress (Address), Value);\r
 }\r
 \r
@@ -1148,16 +1284,21 @@ PciExpressWrite32 (
                   Register.\r
   @param  OrData  The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressOr32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT32                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINT32  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioOr32 (GetPciExpressAddress (Address), OrData);\r
 }\r
 \r
@@ -1179,16 +1320,21 @@ PciExpressOr32 (
                   Register.\r
   @param  AndData The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressAnd32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT32                    AndData\r
+  IN      UINTN   Address,\r
+  IN      UINT32  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioAnd32 (GetPciExpressAddress (Address), AndData);\r
 }\r
 \r
@@ -1212,17 +1358,22 @@ PciExpressAnd32 (
   @param  AndData The value to AND with the PCI configuration register.\r
   @param  OrData  The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressAndThenOr32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINT32                    AndData,\r
-  IN      UINT32                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINT32  AndData,\r
+  IN      UINT32  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioAndThenOr32 (\r
            GetPciExpressAddress (Address),\r
            AndData,\r
@@ -1249,17 +1400,22 @@ PciExpressAndThenOr32 (
   @param  EndBit    The ordinal of the most significant bit in the bit field.\r
                     Range 0..31.\r
 \r
-  @return The value of the bit field read from the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value of the bit field read from the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressBitFieldRead32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit\r
+  IN      UINTN  Address,\r
+  IN      UINTN  StartBit,\r
+  IN      UINTN  EndBit\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioBitFieldRead32 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1289,18 +1445,23 @@ PciExpressBitFieldRead32 (
                     Range 0..31.\r
   @param  Value     The new value of the bit field.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressBitFieldWrite32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT32                    Value\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT32  Value\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioBitFieldWrite32 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1334,18 +1495,23 @@ PciExpressBitFieldWrite32 (
                     Range 0..31.\r
   @param  OrData    The value to OR with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressBitFieldOr32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT32                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT32  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioBitFieldOr32 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1379,18 +1545,23 @@ PciExpressBitFieldOr32 (
                     Range 0..31.\r
   @param  AndData   The value to AND with the PCI configuration register.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressBitFieldAnd32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT32                    AndData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT32  AndData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioBitFieldAnd32 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1428,19 +1599,24 @@ PciExpressBitFieldAnd32 (
   @param  AndData   The value to AND with the PCI configuration register.\r
   @param  OrData    The value to OR with the result of the AND operation.\r
 \r
-  @return The value written back to the PCI configuration register.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      The value written back to the PCI configuration register.\r
 \r
 **/\r
 UINT32\r
 EFIAPI\r
 PciExpressBitFieldAndThenOr32 (\r
-  IN      UINTN                     Address,\r
-  IN      UINTN                     StartBit,\r
-  IN      UINTN                     EndBit,\r
-  IN      UINT32                    AndData,\r
-  IN      UINT32                    OrData\r
+  IN      UINTN   Address,\r
+  IN      UINTN   StartBit,\r
+  IN      UINTN   EndBit,\r
+  IN      UINT32  AndData,\r
+  IN      UINT32  OrData\r
   )\r
 {\r
+  if (Address >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINT32)-1;\r
+  }\r
+\r
   return MmioBitFieldAndThenOr32 (\r
            GetPciExpressAddress (Address),\r
            StartBit,\r
@@ -1457,7 +1633,7 @@ PciExpressBitFieldAndThenOr32 (
   Size into the buffer specified by Buffer. This function only allows the PCI\r
   configuration registers from a single PCI function to be read. Size is\r
   returned. When possible 32-bit PCI configuration read cycles are used to read\r
-  from StartAdress to StartAddress + Size. Due to alignment restrictions, 8-bit\r
+  from StartAddress to StartAddress + Size. Due to alignment restrictions, 8-bit\r
   and 16-bit PCI configuration read cycles may be used at the beginning and the\r
   end of the range.\r
 \r
@@ -1470,25 +1646,33 @@ PciExpressBitFieldAndThenOr32 (
   @param  Size          The size in bytes of the transfer.\r
   @param  Buffer        The pointer to a buffer receiving the data read.\r
 \r
-  @return Size read data from StartAddress.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      Size read data from StartAddress.\r
 \r
 **/\r
 UINTN\r
 EFIAPI\r
 PciExpressReadBuffer (\r
-  IN      UINTN                     StartAddress,\r
-  IN      UINTN                     Size,\r
-  OUT     VOID                      *Buffer\r
+  IN      UINTN  StartAddress,\r
+  IN      UINTN  Size,\r
+  OUT     VOID   *Buffer\r
   )\r
 {\r
-  UINTN   ReturnValue;\r
+  UINTN  ReturnValue;\r
 \r
   //\r
   // Make sure Address is valid\r
   //\r
-  ASSERT (((StartAddress) & ~0xfffffff) == 0);\r
+  ASSERT_INVALID_PCI_ADDRESS (StartAddress);\r
   ASSERT (((StartAddress & 0xFFF) + Size) <= 0x1000);\r
 \r
+  //\r
+  // Make sure the Address is in MMCONF address space\r
+  //\r
+  if (StartAddress >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINTN)-1;\r
+  }\r
+\r
   if (Size == 0) {\r
     return Size;\r
   }\r
@@ -1505,41 +1689,41 @@ PciExpressReadBuffer (
     // Read a byte if StartAddress is byte aligned\r
     //\r
     *(volatile UINT8 *)Buffer = PciExpressRead8 (StartAddress);\r
-    StartAddress += sizeof (UINT8);\r
-    Size -= sizeof (UINT8);\r
-    Buffer = (UINT8*)Buffer + 1;\r
+    StartAddress             += sizeof (UINT8);\r
+    Size                     -= sizeof (UINT8);\r
+    Buffer                    = (UINT8 *)Buffer + 1;\r
   }\r
 \r
-  if (Size >= sizeof (UINT16) && (StartAddress & 2) != 0) {\r
+  if ((Size >= sizeof (UINT16)) && ((StartAddress & 2) != 0)) {\r
     //\r
     // Read a word if StartAddress is word aligned\r
     //\r
-    WriteUnaligned16 ((UINT16 *) Buffer, (UINT16) PciExpressRead16 (StartAddress));\r
+    WriteUnaligned16 ((UINT16 *)Buffer, (UINT16)PciExpressRead16 (StartAddress));\r
 \r
     StartAddress += sizeof (UINT16);\r
-    Size -= sizeof (UINT16);\r
-    Buffer = (UINT16*)Buffer + 1;\r
+    Size         -= sizeof (UINT16);\r
+    Buffer        = (UINT16 *)Buffer + 1;\r
   }\r
 \r
   while (Size >= sizeof (UINT32)) {\r
     //\r
     // Read as many double words as possible\r
     //\r
-    WriteUnaligned32 ((UINT32 *) Buffer, (UINT32) PciExpressRead32 (StartAddress));\r
+    WriteUnaligned32 ((UINT32 *)Buffer, (UINT32)PciExpressRead32 (StartAddress));\r
 \r
     StartAddress += sizeof (UINT32);\r
-    Size -= sizeof (UINT32);\r
-    Buffer = (UINT32*)Buffer + 1;\r
+    Size         -= sizeof (UINT32);\r
+    Buffer        = (UINT32 *)Buffer + 1;\r
   }\r
 \r
   if (Size >= sizeof (UINT16)) {\r
     //\r
     // Read the last remaining word if exist\r
     //\r
-    WriteUnaligned16 ((UINT16 *) Buffer, (UINT16) PciExpressRead16 (StartAddress));\r
+    WriteUnaligned16 ((UINT16 *)Buffer, (UINT16)PciExpressRead16 (StartAddress));\r
     StartAddress += sizeof (UINT16);\r
-    Size -= sizeof (UINT16);\r
-    Buffer = (UINT16*)Buffer + 1;\r
+    Size         -= sizeof (UINT16);\r
+    Buffer        = (UINT16 *)Buffer + 1;\r
   }\r
 \r
   if (Size >= sizeof (UINT8)) {\r
@@ -1560,7 +1744,7 @@ PciExpressReadBuffer (
   Size from the buffer specified by Buffer. This function only allows the PCI\r
   configuration registers from a single PCI function to be written. Size is\r
   returned. When possible 32-bit PCI configuration write cycles are used to\r
-  write from StartAdress to StartAddress + Size. Due to alignment restrictions,\r
+  write from StartAddress to StartAddress + Size. Due to alignment restrictions,\r
   8-bit and 16-bit PCI configuration write cycles may be used at the beginning\r
   and the end of the range.\r
 \r
@@ -1573,25 +1757,33 @@ PciExpressReadBuffer (
   @param  Size          The size in bytes of the transfer.\r
   @param  Buffer        The pointer to a buffer containing the data to write.\r
 \r
-  @return Size written to StartAddress.\r
+  @retval 0xFFFFFFFF Invalid PCI address.\r
+  @retval other      Size written to StartAddress.\r
 \r
 **/\r
 UINTN\r
 EFIAPI\r
 PciExpressWriteBuffer (\r
-  IN      UINTN                     StartAddress,\r
-  IN      UINTN                     Size,\r
-  IN      VOID                      *Buffer\r
+  IN      UINTN  StartAddress,\r
+  IN      UINTN  Size,\r
+  IN      VOID   *Buffer\r
   )\r
 {\r
-  UINTN                             ReturnValue;\r
+  UINTN  ReturnValue;\r
 \r
   //\r
   // Make sure Address is valid\r
   //\r
-  ASSERT (((StartAddress) & ~0xfffffff) == 0);\r
+  ASSERT_INVALID_PCI_ADDRESS (StartAddress);\r
   ASSERT (((StartAddress & 0xFFF) + Size) <= 0x1000);\r
 \r
+  //\r
+  // Make sure the Address is in MMCONF address space\r
+  //\r
+  if (StartAddress >= mDxeRuntimePciExpressLibPciExpressBaseSize) {\r
+    return (UINTN)-1;\r
+  }\r
+\r
   if (Size == 0) {\r
     return 0;\r
   }\r
@@ -1607,47 +1799,47 @@ PciExpressWriteBuffer (
     //\r
     // Write a byte if StartAddress is byte aligned\r
     //\r
-    PciExpressWrite8 (StartAddress, *(UINT8*)Buffer);\r
+    PciExpressWrite8 (StartAddress, *(UINT8 *)Buffer);\r
     StartAddress += sizeof (UINT8);\r
-    Size -= sizeof (UINT8);\r
-    Buffer = (UINT8*)Buffer + 1;\r
+    Size         -= sizeof (UINT8);\r
+    Buffer        = (UINT8 *)Buffer + 1;\r
   }\r
 \r
-  if (Size >= sizeof (UINT16) && (StartAddress & 2) != 0) {\r
+  if ((Size >= sizeof (UINT16)) && ((StartAddress & 2) != 0)) {\r
     //\r
     // Write a word if StartAddress is word aligned\r
     //\r
-    PciExpressWrite16 (StartAddress, ReadUnaligned16 ((UINT16*)Buffer));\r
+    PciExpressWrite16 (StartAddress, ReadUnaligned16 ((UINT16 *)Buffer));\r
     StartAddress += sizeof (UINT16);\r
-    Size -= sizeof (UINT16);\r
-    Buffer = (UINT16*)Buffer + 1;\r
+    Size         -= sizeof (UINT16);\r
+    Buffer        = (UINT16 *)Buffer + 1;\r
   }\r
 \r
   while (Size >= sizeof (UINT32)) {\r
     //\r
     // Write as many double words as possible\r
     //\r
-    PciExpressWrite32 (StartAddress, ReadUnaligned32 ((UINT32*)Buffer));\r
+    PciExpressWrite32 (StartAddress, ReadUnaligned32 ((UINT32 *)Buffer));\r
     StartAddress += sizeof (UINT32);\r
-    Size -= sizeof (UINT32);\r
-    Buffer = (UINT32*)Buffer + 1;\r
+    Size         -= sizeof (UINT32);\r
+    Buffer        = (UINT32 *)Buffer + 1;\r
   }\r
 \r
   if (Size >= sizeof (UINT16)) {\r
     //\r
     // Write the last remaining word if exist\r
     //\r
-    PciExpressWrite16 (StartAddress, ReadUnaligned16 ((UINT16*)Buffer));\r
+    PciExpressWrite16 (StartAddress, ReadUnaligned16 ((UINT16 *)Buffer));\r
     StartAddress += sizeof (UINT16);\r
-    Size -= sizeof (UINT16);\r
-    Buffer = (UINT16*)Buffer + 1;\r
+    Size         -= sizeof (UINT16);\r
+    Buffer        = (UINT16 *)Buffer + 1;\r
   }\r
 \r
   if (Size >= sizeof (UINT8)) {\r
     //\r
     // Write the last remaining byte if exist\r
     //\r
-    PciExpressWrite8 (StartAddress, *(UINT8*)Buffer);\r
+    PciExpressWrite8 (StartAddress, *(UINT8 *)Buffer);\r
   }\r
 \r
   return ReturnValue;\r