[mirror_edk2.git] / MdePkg / Library / PciSegmentLibSegmentInfo / DxeRuntimePciSegmentLib.c

/** @file\r
  Instance of Runtime PCI Segment Library that support multi-segment PCI configuration access.\r
\r
  PCI Segment Library that consumes segment information provided by PciSegmentInfoLib to\r
   support multi-segment PCI configuration access through enhanced configuration access mechanism.\r
\r
  Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "PciSegmentLibCommon.h"\r
#include <PiDxe.h>\r
#include <Guid/EventGroup.h>\r
#include <Library/UefiRuntimeLib.h>\r
#include <Library/DxeServicesTableLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/PciSegmentInfoLib.h>\r
\r
///\r
/// Define table for mapping PCI Segment MMIO physical addresses to virtual addresses at OS runtime\r
///\r
typedef struct {\r
  UINTN    PhysicalAddress;\r
  UINTN    VirtualAddress;\r
} PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE;\r
\r
///\r
/// Set Virtual Address Map Event\r
///\r
EFI_EVENT  mDxeRuntimePciSegmentLibVirtualNotifyEvent = NULL;\r
\r
///\r
/// The number of PCI devices that have been registered for runtime access.\r
///\r
UINTN  mDxeRuntimePciSegmentLibNumberOfRuntimeRanges = 0;\r
\r
///\r
/// The table of PCI devices that have been registered for runtime access.\r
///\r
PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE  *mDxeRuntimePciSegmentLibRegistrationTable = NULL;\r
\r
///\r
/// The table index of the most recent virtual address lookup.\r
///\r
UINTN  mDxeRuntimePciSegmentLibLastRuntimeRange = 0;\r
\r
/**\r
  Convert the physical PCI Express MMIO addresses for all registered PCI devices\r
  to virtual addresses.\r
\r
  @param[in]    Event   The event that is being processed.\r
  @param[in]    Context The Event Context.\r
**/\r
VOID\r
EFIAPI\r
DxeRuntimePciSegmentLibVirtualNotify (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  UINTN       Index;\r
  EFI_STATUS  Status;\r
\r
  //\r
  // If there have been no runtime registrations, then just return\r
  //\r
  if (mDxeRuntimePciSegmentLibRegistrationTable == NULL) {\r
    return;\r
  }\r
\r
  //\r
  // Convert physical addresses associated with the set of registered PCI devices to\r
  // virtual addresses.\r
  //\r
  for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
    Status = EfiConvertPointer (0, (VOID **)&(mDxeRuntimePciSegmentLibRegistrationTable[Index].VirtualAddress));\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  //\r
  // Convert table pointer that is allocated from EfiRuntimeServicesData to a virtual address.\r
  //\r
  Status = EfiConvertPointer (0, (VOID **)&mDxeRuntimePciSegmentLibRegistrationTable);\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
/**\r
  The constructor function caches the PCI Express Base Address and creates a\r
  Set Virtual Address Map event to convert physical address to virtual addresses.\r
\r
  @param  ImageHandle   The firmware allocated handle for the EFI image.\r
  @param  SystemTable   A pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS   The constructor completed successfully.\r
  @retval Other value   The constructor did not complete successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DxeRuntimePciSegmentLibConstructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  //\r
  // Register SetVirtualAddressMap () notify function\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,\r
                  TPL_NOTIFY,\r
                  DxeRuntimePciSegmentLibVirtualNotify,\r
                  NULL,\r
                  &mDxeRuntimePciSegmentLibVirtualNotifyEvent\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return Status;\r
}\r
\r
/**\r
  The destructor function frees any allocated buffers and closes the Set Virtual\r
  Address Map event.\r
\r
  @param  ImageHandle   The firmware allocated handle for the EFI image.\r
  @param  SystemTable   A pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS   The destructor completed successfully.\r
  @retval Other value   The destructor did not complete successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DxeRuntimePciSegmentLibDestructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  //\r
  // If one or more PCI devices have been registered for runtime access, then\r
  // free the registration table.\r
  //\r
  if (mDxeRuntimePciSegmentLibRegistrationTable != NULL) {\r
    FreePool (mDxeRuntimePciSegmentLibRegistrationTable);\r
  }\r
\r
  //\r
  // Close the Set Virtual Address Map event\r
  //\r
  Status = gBS->CloseEvent (mDxeRuntimePciSegmentLibVirtualNotifyEvent);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return Status;\r
}\r
\r
/**\r
  Register a PCI device so PCI configuration registers may be accessed after\r
  SetVirtualAddressMap().\r
\r
  If any reserved bits in Address are set, then ASSERT().\r
\r
  @param  Address The address that encodes the PCI Bus, Device, Function and\r
                  Register.\r
\r
  @retval RETURN_SUCCESS           The PCI device was registered for runtime access.\r
  @retval RETURN_UNSUPPORTED       An attempt was made to call this function\r
                                   after ExitBootServices().\r
  @retval RETURN_UNSUPPORTED       The resources required to access the PCI device\r
                                   at runtime could not be mapped.\r
  @retval RETURN_OUT_OF_RESOURCES  There are not enough resources available to\r
                                   complete the registration.\r
\r
**/\r
RETURN_STATUS\r
EFIAPI\r
PciSegmentRegisterForRuntimeAccess (\r
  IN UINTN  Address\r
  )\r
{\r
  RETURN_STATUS                    Status;\r
  EFI_GCD_MEMORY_SPACE_DESCRIPTOR  Descriptor;\r
  UINTN                            Index;\r
  VOID                             *NewTable;\r
  UINTN                            Count;\r
  PCI_SEGMENT_INFO                 *SegmentInfo;\r
  UINT64                           EcamAddress;\r
\r
  //\r
  // Convert Address to a ECAM address at the beginning of the PCI Configuration\r
  // header for the specified PCI Bus/Dev/Func\r
  //\r
  Address    &= ~(UINTN)EFI_PAGE_MASK;\r
  SegmentInfo = GetPciSegmentInfo (&Count);\r
  EcamAddress = PciSegmentLibGetEcamAddress (Address, SegmentInfo, Count);\r
\r
  //\r
  // Return an error if this function is called after ExitBootServices().\r
  //\r
  if (EfiAtRuntime ()) {\r
    return RETURN_UNSUPPORTED;\r
  }\r
\r
  if (sizeof (UINTN) == sizeof (UINT32)) {\r
    ASSERT (EcamAddress < BASE_4GB);\r
  }\r
\r
  Address = (UINTN)EcamAddress;\r
\r
  //\r
  // See if Address has already been registered for runtime access\r
  //\r
  for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
    if (mDxeRuntimePciSegmentLibRegistrationTable[Index].PhysicalAddress == Address) {\r
      return RETURN_SUCCESS;\r
    }\r
  }\r
\r
  //\r
  // Get the GCD Memory Descriptor for the ECAM Address\r
  //\r
  Status = gDS->GetMemorySpaceDescriptor (Address, &Descriptor);\r
  if (EFI_ERROR (Status)) {\r
    return RETURN_UNSUPPORTED;\r
  }\r
\r
  //\r
  // Mark the 4KB region for the PCI Express Bus/Dev/Func as EFI_RUNTIME_MEMORY so the OS\r
  // will allocate a virtual address range for the 4KB PCI Configuration Header.\r
  //\r
  Status = gDS->SetMemorySpaceAttributes (Address, EFI_PAGE_SIZE, Descriptor.Attributes | EFI_MEMORY_RUNTIME);\r
  if (EFI_ERROR (Status)) {\r
    return RETURN_UNSUPPORTED;\r
  }\r
\r
  //\r
  // Grow the size of the registration table\r
  //\r
  NewTable = ReallocateRuntimePool (\r
               (mDxeRuntimePciSegmentLibNumberOfRuntimeRanges + 0) * sizeof (PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE),\r
               (mDxeRuntimePciSegmentLibNumberOfRuntimeRanges + 1) * sizeof (PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE),\r
               mDxeRuntimePciSegmentLibRegistrationTable\r
               );\r
  if (NewTable == NULL) {\r
    return RETURN_OUT_OF_RESOURCES;\r
  }\r
\r
  mDxeRuntimePciSegmentLibRegistrationTable                                                                = NewTable;\r
  mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibNumberOfRuntimeRanges].PhysicalAddress = Address;\r
  mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibNumberOfRuntimeRanges].VirtualAddress  = Address;\r
  mDxeRuntimePciSegmentLibNumberOfRuntimeRanges++;\r
\r
  return RETURN_SUCCESS;\r
}\r
\r
/**\r
  Return the linear address for the physical address.\r
\r
  @param  Address  The physical address.\r
\r
  @retval The linear address.\r
**/\r
UINTN\r
PciSegmentLibVirtualAddress (\r
  IN UINTN  Address\r
  )\r
{\r
  UINTN  Index;\r
\r
  //\r
  // If SetVirtualAddressMap() has not been called, then just return the physical address\r
  //\r
  if (!EfiGoneVirtual ()) {\r
    return Address;\r
  }\r
\r
  //\r
  // See if there is a physical address match at the exact same index as the last address match\r
  //\r
  if (mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibLastRuntimeRange].PhysicalAddress == (Address & (~(UINTN)EFI_PAGE_MASK))) {\r
    //\r
    // Convert the physical address to a virtual address and return the virtual address\r
    //\r
    return (Address & EFI_PAGE_MASK) + mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibLastRuntimeRange].VirtualAddress;\r
  }\r
\r
  //\r
  // Search the entire table for a physical address match\r
  //\r
  for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
    if (mDxeRuntimePciSegmentLibRegistrationTable[Index].PhysicalAddress == (Address & (~(UINTN)EFI_PAGE_MASK))) {\r
      //\r
      // Cache the matching index value\r
      //\r
      mDxeRuntimePciSegmentLibLastRuntimeRange = Index;\r
      //\r
      // Convert the physical address to a virtual address and return the virtual address\r
      //\r
      return (Address & EFI_PAGE_MASK) + mDxeRuntimePciSegmentLibRegistrationTable[Index].VirtualAddress;\r
    }\r
  }\r
\r
  //\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
\r
  //\r
  // Return the physical address\r
  //\r
  return Address;\r
}\r
Commit	Line	Data
5c9bb86f RN	1	/** @file\r
	2	Instance of Runtime PCI Segment Library that support multi-segment PCI configuration access.\r
	3	\r
	4	PCI Segment Library that consumes segment information provided by PciSegmentInfoLib to\r
	5	support multi-segment PCI configuration access through enhanced configuration access mechanism.\r
	6	\r
	7	Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>\r
9344f092	8	SPDX-License-Identifier: BSD-2-Clause-Patent\r
5c9bb86f RN	9	\r
	10	**/\r
	11	\r
	12	#include "PciSegmentLibCommon.h"\r
	13	#include <PiDxe.h>\r
	14	#include <Guid/EventGroup.h>\r
	15	#include <Library/UefiRuntimeLib.h>\r
	16	#include <Library/DxeServicesTableLib.h>\r
	17	#include <Library/UefiBootServicesTableLib.h>\r
	18	#include <Library/MemoryAllocationLib.h>\r
	19	#include <Library/PciSegmentInfoLib.h>\r
	20	\r
	21	///\r
	22	/// Define table for mapping PCI Segment MMIO physical addresses to virtual addresses at OS runtime\r
	23	///\r
	24	typedef struct {\r
2f88bd3a MK	25	UINTN PhysicalAddress;\r
2f88bd3a MK	26	UINTN VirtualAddress;\r
5c9bb86f RN	27	} PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE;\r
	28	\r
	29	///\r
	30	/// Set Virtual Address Map Event\r
	31	///\r
2f88bd3a	32	EFI_EVENT mDxeRuntimePciSegmentLibVirtualNotifyEvent = NULL;\r
5c9bb86f RN	33	\r
	34	///\r
	35	/// The number of PCI devices that have been registered for runtime access.\r
	36	///\r
2f88bd3a	37	UINTN mDxeRuntimePciSegmentLibNumberOfRuntimeRanges = 0;\r
5c9bb86f RN	38	\r
	39	///\r
	40	/// The table of PCI devices that have been registered for runtime access.\r
	41	///\r
	42	PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE *mDxeRuntimePciSegmentLibRegistrationTable = NULL;\r
	43	\r
	44	///\r
	45	/// The table index of the most recent virtual address lookup.\r
	46	///\r
2f88bd3a	47	UINTN mDxeRuntimePciSegmentLibLastRuntimeRange = 0;\r
5c9bb86f RN	48	\r
	49	/**\r
	50	Convert the physical PCI Express MMIO addresses for all registered PCI devices\r
	51	to virtual addresses.\r
	52	\r
	53	@param[in] Event The event that is being processed.\r
	54	@param[in] Context The Event Context.\r
	55	**/\r
	56	VOID\r
	57	EFIAPI\r
	58	DxeRuntimePciSegmentLibVirtualNotify (\r
	59	IN EFI_EVENT Event,\r
	60	IN VOID *Context\r
	61	)\r
	62	{\r
2f88bd3a MK	63	UINTN Index;\r
2f88bd3a MK	64	EFI_STATUS Status;\r
5c9bb86f RN	65	\r
	66	//\r
	67	// If there have been no runtime registrations, then just return\r
	68	//\r
	69	if (mDxeRuntimePciSegmentLibRegistrationTable == NULL) {\r
	70	return;\r
	71	}\r
	72	\r
	73	//\r
	74	// Convert physical addresses associated with the set of registered PCI devices to\r
	75	// virtual addresses.\r
	76	//\r
	77	for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
2f88bd3a	78	Status = EfiConvertPointer (0, (VOID **)&(mDxeRuntimePciSegmentLibRegistrationTable[Index].VirtualAddress));\r
5c9bb86f RN	79	ASSERT_EFI_ERROR (Status);\r
	80	}\r
	81	\r
	82	//\r
	83	// Convert table pointer that is allocated from EfiRuntimeServicesData to a virtual address.\r
	84	//\r
2f88bd3a	85	Status = EfiConvertPointer (0, (VOID **)&mDxeRuntimePciSegmentLibRegistrationTable);\r
5c9bb86f RN	86	ASSERT_EFI_ERROR (Status);\r
	87	}\r
	88	\r
	89	/**\r
	90	The constructor function caches the PCI Express Base Address and creates a\r
	91	Set Virtual Address Map event to convert physical address to virtual addresses.\r
	92	\r
	93	@param ImageHandle The firmware allocated handle for the EFI image.\r
	94	@param SystemTable A pointer to the EFI System Table.\r
	95	\r
	96	@retval EFI_SUCCESS The constructor completed successfully.\r
	97	@retval Other value The constructor did not complete successfully.\r
	98	\r
	99	**/\r
	100	EFI_STATUS\r
	101	EFIAPI\r
	102	DxeRuntimePciSegmentLibConstructor (\r
	103	IN EFI_HANDLE ImageHandle,\r
	104	IN EFI_SYSTEM_TABLE *SystemTable\r
	105	)\r
	106	{\r
	107	EFI_STATUS Status;\r
	108	\r
	109	//\r
	110	// Register SetVirtualAddressMap () notify function\r
	111	//\r
df851da3 VC	112	Status = gBS->CreateEvent (\r
df851da3 VC	113	EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,\r
5c9bb86f RN	114	TPL_NOTIFY,\r
	115	DxeRuntimePciSegmentLibVirtualNotify,\r
	116	NULL,\r
5c9bb86f RN	117	&mDxeRuntimePciSegmentLibVirtualNotifyEvent\r
	118	);\r
	119	ASSERT_EFI_ERROR (Status);\r
	120	\r
	121	return Status;\r
	122	}\r
	123	\r
	124	/**\r
	125	The destructor function frees any allocated buffers and closes the Set Virtual\r
	126	Address Map event.\r
	127	\r
	128	@param ImageHandle The firmware allocated handle for the EFI image.\r
	129	@param SystemTable A pointer to the EFI System Table.\r
	130	\r
	131	@retval EFI_SUCCESS The destructor completed successfully.\r
	132	@retval Other value The destructor did not complete successfully.\r
	133	\r
	134	**/\r
	135	EFI_STATUS\r
	136	EFIAPI\r
	137	DxeRuntimePciSegmentLibDestructor (\r
	138	IN EFI_HANDLE ImageHandle,\r
	139	IN EFI_SYSTEM_TABLE *SystemTable\r
	140	)\r
	141	{\r
	142	EFI_STATUS Status;\r
	143	\r
	144	//\r
	145	// If one or more PCI devices have been registered for runtime access, then\r
	146	// free the registration table.\r
	147	//\r
	148	if (mDxeRuntimePciSegmentLibRegistrationTable != NULL) {\r
	149	FreePool (mDxeRuntimePciSegmentLibRegistrationTable);\r
	150	}\r
	151	\r
	152	//\r
	153	// Close the Set Virtual Address Map event\r
	154	//\r
	155	Status = gBS->CloseEvent (mDxeRuntimePciSegmentLibVirtualNotifyEvent);\r
	156	ASSERT_EFI_ERROR (Status);\r
	157	\r
	158	return Status;\r
	159	}\r
	160	\r
	161	/**\r
	162	Register a PCI device so PCI configuration registers may be accessed after\r
	163	SetVirtualAddressMap().\r
	164	\r
	165	If any reserved bits in Address are set, then ASSERT().\r
	166	\r
	167	@param Address The address that encodes the PCI Bus, Device, Function and\r
	168	Register.\r
	169	\r
	170	@retval RETURN_SUCCESS The PCI device was registered for runtime access.\r
	171	@retval RETURN_UNSUPPORTED An attempt was made to call this function\r
	172	after ExitBootServices().\r
	173	@retval RETURN_UNSUPPORTED The resources required to access the PCI device\r
	174	at runtime could not be mapped.\r
	175	@retval RETURN_OUT_OF_RESOURCES There are not enough resources available to\r
	176	complete the registration.\r
	177	\r
	178	**/\r
	179	RETURN_STATUS\r
	180	EFIAPI\r
181	PciSegmentRegisterForRuntimeAccess (\r
182	IN UINTN Address\r
183	)\r
184	{\r
185	RETURN_STATUS Status;\r
186	EFI_GCD_MEMORY_SPACE_DESCRIPTOR Descriptor;\r
187	UINTN Index;\r
188	VOID *NewTable;\r
189	UINTN Count;\r
190	PCI_SEGMENT_INFO *SegmentInfo;\r
191	UINT64 EcamAddress;\r
192	\r
193	//\r
194	// Convert Address to a ECAM address at the beginning of the PCI Configuration\r
195	// header for the specified PCI Bus/Dev/Func\r
196	//\r
2f88bd3a	197	Address &= ~(UINTN)EFI_PAGE_MASK;\r
5c9bb86f RN	198	SegmentInfo = GetPciSegmentInfo (&Count);\r
	199	EcamAddress = PciSegmentLibGetEcamAddress (Address, SegmentInfo, Count);\r
	200	\r
	201	//\r
	202	// Return an error if this function is called after ExitBootServices().\r
	203	//\r
	204	if (EfiAtRuntime ()) {\r
	205	return RETURN_UNSUPPORTED;\r
	206	}\r
2f88bd3a	207	\r
5c9bb86f RN	208	if (sizeof (UINTN) == sizeof (UINT32)) {\r
	209	ASSERT (EcamAddress < BASE_4GB);\r
	210	}\r
2f88bd3a	211	\r
5c9bb86f RN	212	Address = (UINTN)EcamAddress;\r
	213	\r
	214	//\r
a8ecf980	215	// See if Address has already been registered for runtime access\r
5c9bb86f RN	216	//\r
	217	for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
	218	if (mDxeRuntimePciSegmentLibRegistrationTable[Index].PhysicalAddress == Address) {\r
	219	return RETURN_SUCCESS;\r
	220	}\r
	221	}\r
	222	\r
	223	//\r
	224	// Get the GCD Memory Descriptor for the ECAM Address\r
	225	//\r
	226	Status = gDS->GetMemorySpaceDescriptor (Address, &Descriptor);\r
	227	if (EFI_ERROR (Status)) {\r
	228	return RETURN_UNSUPPORTED;\r
	229	}\r
	230	\r
	231	//\r
	232	// Mark the 4KB region for the PCI Express Bus/Dev/Func as EFI_RUNTIME_MEMORY so the OS\r
	233	// will allocate a virtual address range for the 4KB PCI Configuration Header.\r
	234	//\r
	235	Status = gDS->SetMemorySpaceAttributes (Address, EFI_PAGE_SIZE, Descriptor.Attributes \| EFI_MEMORY_RUNTIME);\r
	236	if (EFI_ERROR (Status)) {\r
	237	return RETURN_UNSUPPORTED;\r
	238	}\r
	239	\r
	240	//\r
	241	// Grow the size of the registration table\r
	242	//\r
	243	NewTable = ReallocateRuntimePool (\r
	244	(mDxeRuntimePciSegmentLibNumberOfRuntimeRanges + 0) * sizeof (PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE),\r
	245	(mDxeRuntimePciSegmentLibNumberOfRuntimeRanges + 1) * sizeof (PCI_SEGMENT_RUNTIME_REGISTRATION_TABLE),\r
	246	mDxeRuntimePciSegmentLibRegistrationTable\r
	247	);\r
	248	if (NewTable == NULL) {\r
	249	return RETURN_OUT_OF_RESOURCES;\r
	250	}\r
2f88bd3a MK	251	\r
2f88bd3a MK	252	mDxeRuntimePciSegmentLibRegistrationTable = NewTable;\r
5c9bb86f RN	253	mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibNumberOfRuntimeRanges].PhysicalAddress = Address;\r
	254	mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibNumberOfRuntimeRanges].VirtualAddress = Address;\r
	255	mDxeRuntimePciSegmentLibNumberOfRuntimeRanges++;\r
	256	\r
	257	return RETURN_SUCCESS;\r
	258	}\r
	259	\r
	260	/**\r
	261	Return the linear address for the physical address.\r
	262	\r
	263	@param Address The physical address.\r
	264	\r
	265	@retval The linear address.\r
	266	**/\r
	267	UINTN\r
	268	PciSegmentLibVirtualAddress (\r
2f88bd3a	269	IN UINTN Address\r
5c9bb86f RN	270	)\r
5c9bb86f RN	271	{\r
2f88bd3a MK	272	UINTN Index;\r
2f88bd3a MK	273	\r
5c9bb86f RN	274	//\r
	275	// If SetVirtualAddressMap() has not been called, then just return the physical address\r
	276	//\r
	277	if (!EfiGoneVirtual ()) {\r
	278	return Address;\r
	279	}\r
	280	\r
	281	//\r
	282	// See if there is a physical address match at the exact same index as the last address match\r
	283	//\r
	284	if (mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibLastRuntimeRange].PhysicalAddress == (Address & (~(UINTN)EFI_PAGE_MASK))) {\r
	285	//\r
	286	// Convert the physical address to a virtual address and return the virtual address\r
	287	//\r
	288	return (Address & EFI_PAGE_MASK) + mDxeRuntimePciSegmentLibRegistrationTable[mDxeRuntimePciSegmentLibLastRuntimeRange].VirtualAddress;\r
	289	}\r
	290	\r
	291	//\r
	292	// Search the entire table for a physical address match\r
	293	//\r
	294	for (Index = 0; Index < mDxeRuntimePciSegmentLibNumberOfRuntimeRanges; Index++) {\r
	295	if (mDxeRuntimePciSegmentLibRegistrationTable[Index].PhysicalAddress == (Address & (~(UINTN)EFI_PAGE_MASK))) {\r
	296	//\r
	297	// Cache the matching index value\r
	298	//\r
	299	mDxeRuntimePciSegmentLibLastRuntimeRange = Index;\r
	300	//\r
	301	// Convert the physical address to a virtual address and return the virtual address\r
	302	//\r
	303	return (Address & EFI_PAGE_MASK) + mDxeRuntimePciSegmentLibRegistrationTable[Index].VirtualAddress;\r
	304	}\r
	305	}\r
	306	\r
	307	//\r
	308	// No match was found. This is a critical error at OS runtime, so ASSERT() and force a breakpoint.\r
	309	//\r
	310	ASSERT (FALSE);\r
	311	CpuBreakpoint ();\r
	312	\r
	313	//\r
	314	// Return the physical address\r
	315	//\r
	316	return Address;\r
	317	}\r