[mirror_edk2.git] / StandaloneMmPkg / Library / StandaloneMmCoreEntryPoint / AArch64 / SetPermissions.c

/** @file\r
  Locate, get and update PE/COFF permissions during Standalone MM\r
  Foundation Entry point on ARM platforms.\r
\r
Copyright (c) 2017 - 2018, ARM Ltd. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
\r
#include <PiMm.h>\r
\r
#include <PiPei.h>\r
#include <Guid/MmramMemoryReserve.h>\r
#include <Guid/MpInformation.h>\r
\r
#include <Library/AArch64/StandaloneMmCoreEntryPoint.h>\r
#include <Library/ArmMmuLib.h>\r
#include <Library/ArmSvcLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/HobLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/SerialPortLib.h>\r
\r
#include <IndustryStandard/ArmStdSmc.h>\r
\r
EFI_STATUS\r
EFIAPI\r
UpdateMmFoundationPeCoffPermissions (\r
  IN  CONST PE_COFF_LOADER_IMAGE_CONTEXT      *ImageContext,\r
  IN  EFI_PHYSICAL_ADDRESS                    ImageBase,\r
  IN  UINT32                                  SectionHeaderOffset,\r
  IN  CONST  UINT16                           NumberOfSections,\r
  IN  REGION_PERMISSION_UPDATE_FUNC           TextUpdater,\r
  IN  REGION_PERMISSION_UPDATE_FUNC           ReadOnlyUpdater,\r
  IN  REGION_PERMISSION_UPDATE_FUNC           ReadWriteUpdater\r
  )\r
{\r
  EFI_IMAGE_SECTION_HEADER         SectionHeader;\r
  RETURN_STATUS                    Status;\r
  EFI_PHYSICAL_ADDRESS             Base;\r
  UINTN                            Size;\r
  UINTN                            ReadSize;\r
  UINTN                            Index;\r
\r
  ASSERT (ImageContext != NULL);\r
\r
  //\r
  // Iterate over the sections\r
  //\r
  for (Index = 0; Index < NumberOfSections; Index++) {\r
    //\r
    // Read section header from file\r
    //\r
    Size = sizeof (EFI_IMAGE_SECTION_HEADER);\r
    ReadSize = Size;\r
    Status = ImageContext->ImageRead (\r
                             ImageContext->Handle,\r
                             SectionHeaderOffset,\r
                             &Size,\r
                             &SectionHeader\r
                             );\r
\r
    if (RETURN_ERROR (Status) || (Size != ReadSize)) {\r
      DEBUG ((DEBUG_ERROR,\r
              "%a: ImageContext->ImageRead () failed (Status = %r)\n",\r
              __FUNCTION__, Status));\r
      return Status;\r
    }\r
\r
    DEBUG ((DEBUG_INFO,\r
            "%a: Section %d of image at 0x%lx has 0x%x permissions\n",\r
            __FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Characteristics));\r
    DEBUG ((DEBUG_INFO,\r
            "%a: Section %d of image at 0x%lx has %a name\n",\r
            __FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Name));\r
    DEBUG ((DEBUG_INFO,\r
            "%a: Section %d of image at 0x%lx has 0x%x address\n",\r
            __FUNCTION__, Index, ImageContext->ImageAddress,\r
            ImageContext->ImageAddress + SectionHeader.VirtualAddress));\r
    DEBUG ((DEBUG_INFO,\r
            "%a: Section %d of image at 0x%lx has 0x%x data\n",\r
            __FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.PointerToRawData));\r
\r
    //\r
    // If the section is marked as XN then remove the X attribute. Furthermore,\r
    // if it is a writeable section then mark it appropriately as well.\r
    //\r
    if ((SectionHeader.Characteristics & EFI_IMAGE_SCN_MEM_EXECUTE) == 0) {\r
      Base = ImageBase + SectionHeader.VirtualAddress;\r
\r
      TextUpdater (Base, SectionHeader.Misc.VirtualSize);\r
\r
      if ((SectionHeader.Characteristics & EFI_IMAGE_SCN_MEM_WRITE) != 0) {\r
        ReadWriteUpdater (Base, SectionHeader.Misc.VirtualSize);\r
        DEBUG ((DEBUG_INFO,\r
                "%a: Mapping section %d of image at 0x%lx with RW-XN permissions\n",\r
                __FUNCTION__, Index, ImageContext->ImageAddress));\r
      } else {\r
        DEBUG ((DEBUG_INFO,\r
                "%a: Mapping section %d of image at 0x%lx with RO-XN permissions\n",\r
                __FUNCTION__, Index, ImageContext->ImageAddress));\r
      }\r
    } else {\r
        DEBUG ((DEBUG_INFO,\r
                "%a: Ignoring section %d of image at 0x%lx with 0x%x permissions\n",\r
                __FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Characteristics));\r
    }\r
    SectionHeaderOffset += sizeof (EFI_IMAGE_SECTION_HEADER);\r
  }\r
\r
  return RETURN_SUCCESS;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
LocateStandaloneMmCorePeCoffData (\r
  IN        EFI_FIRMWARE_VOLUME_HEADER      *BfvAddress,\r
  IN  OUT   VOID                            **TeData,\r
  IN  OUT   UINTN                           *TeDataSize\r
  )\r
{\r
  EFI_FFS_FILE_HEADER             *FileHeader = NULL;\r
  EFI_STATUS                      Status;\r
\r
  Status = FfsFindNextFile (\r
             EFI_FV_FILETYPE_SECURITY_CORE,\r
             BfvAddress,\r
             &FileHeader\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM FFS file - 0x%x\n",\r
            Status));\r
    return Status;\r
  }\r
\r
  Status = FfsFindSectionData (EFI_SECTION_PE32, FileHeader, TeData, TeDataSize);\r
  if (EFI_ERROR (Status)) {\r
    Status = FfsFindSectionData (EFI_SECTION_TE, FileHeader, TeData, TeDataSize);\r
    if (EFI_ERROR (Status)) {\r
        DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM Section data - %r\n",\r
                Status));\r
      return Status;\r
    }\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "Found Standalone MM PE data - 0x%x\n", *TeData));\r
  return Status;\r
}\r
\r
STATIC\r
EFI_STATUS\r
GetPeCoffSectionInformation (\r
  IN  OUT   PE_COFF_LOADER_IMAGE_CONTEXT      *ImageContext,\r
      OUT   EFI_PHYSICAL_ADDRESS              *ImageBase,\r
      OUT   UINT32                            *SectionHeaderOffset,\r
      OUT   UINT16                            *NumberOfSections\r
  )\r
{\r
  RETURN_STATUS                         Status;\r
  EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION   Hdr;\r
  EFI_IMAGE_OPTIONAL_HEADER_UNION       HdrData;\r
  UINTN                                 Size;\r
  UINTN                                 ReadSize;\r
\r
  ASSERT (ImageContext != NULL);\r
  ASSERT (SectionHeaderOffset != NULL);\r
  ASSERT (NumberOfSections != NULL);\r
\r
  Status = PeCoffLoaderGetImageInfo (ImageContext);\r
  if (RETURN_ERROR (Status)) {\r
    DEBUG ((DEBUG_ERROR,\r
            "%a: PeCoffLoaderGetImageInfo () failed (Status == %r)\n",\r
            __FUNCTION__, Status));\r
    return Status;\r
  }\r
\r
  if (ImageContext->SectionAlignment < EFI_PAGE_SIZE) {\r
    //\r
    // The sections need to be at least 4 KB aligned, since that is the\r
    // granularity at which we can tighten permissions.\r
    //\r
    if (!ImageContext->IsTeImage) {\r
      DEBUG ((DEBUG_WARN,\r
              "%a: non-TE Image at 0x%lx has SectionAlignment < 4 KB (%lu)\n",\r
              __FUNCTION__, ImageContext->ImageAddress, ImageContext->SectionAlignment));\r
      return RETURN_UNSUPPORTED;\r
    }\r
    ImageContext->SectionAlignment = EFI_PAGE_SIZE;\r
  }\r
\r
  //\r
  // Read the PE/COFF Header. For PE32 (32-bit) this will read in too much\r
  // data, but that should not hurt anything. Hdr.Pe32->OptionalHeader.Magic\r
  // determines if this is a PE32 or PE32+ image. The magic is in the same\r
  // location in both images.\r
  //\r
  Hdr.Union = &HdrData;\r
  Size = sizeof (EFI_IMAGE_OPTIONAL_HEADER_UNION);\r
  ReadSize = Size;\r
  Status = ImageContext->ImageRead (\r
                         ImageContext->Handle,\r
                         ImageContext->PeCoffHeaderOffset,\r
                         &Size,\r
                         Hdr.Pe32\r
                         );\r
\r
  if (RETURN_ERROR (Status) || (Size != ReadSize)) {\r
    DEBUG ((DEBUG_ERROR,\r
            "%a: TmpContext->ImageRead () failed (Status = %r)\n",\r
            __FUNCTION__, Status));\r
    return Status;\r
  }\r
\r
  *ImageBase = ImageContext->ImageAddress;\r
  if (!ImageContext->IsTeImage) {\r
    ASSERT (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE);\r
\r
    *SectionHeaderOffset = ImageContext->PeCoffHeaderOffset + sizeof (UINT32) +\r
                          sizeof (EFI_IMAGE_FILE_HEADER);\r
    *NumberOfSections    = Hdr.Pe32->FileHeader.NumberOfSections;\r
\r
    switch (Hdr.Pe32->OptionalHeader.Magic) {\r
    case EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC:\r
      *SectionHeaderOffset += Hdr.Pe32->FileHeader.SizeOfOptionalHeader;\r
      break;\r
    case EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC:\r
      *SectionHeaderOffset += Hdr.Pe32Plus->FileHeader.SizeOfOptionalHeader;\r
      break;\r
    default:\r
      ASSERT (FALSE);\r
    }\r
  } else {\r
    *SectionHeaderOffset = (UINTN)(sizeof (EFI_TE_IMAGE_HEADER));\r
    *NumberOfSections = Hdr.Te->NumberOfSections;\r
    *ImageBase -= (UINT32)Hdr.Te->StrippedSize - sizeof (EFI_TE_IMAGE_HEADER);\r
  }\r
  return RETURN_SUCCESS;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
GetStandaloneMmCorePeCoffSections (\r
  IN        VOID                            *TeData,\r
  IN  OUT   PE_COFF_LOADER_IMAGE_CONTEXT    *ImageContext,\r
      OUT   EFI_PHYSICAL_ADDRESS            *ImageBase,\r
  IN  OUT   UINT32                          *SectionHeaderOffset,\r
  IN  OUT   UINT16                          *NumberOfSections\r
  )\r
{\r
  EFI_STATUS                   Status;\r
\r
  // Initialize the Image Context\r
  ZeroMem (ImageContext, sizeof (PE_COFF_LOADER_IMAGE_CONTEXT));\r
  ImageContext->Handle    = TeData;\r
  ImageContext->ImageRead = PeCoffLoaderImageReadFromMemory;\r
\r
  DEBUG ((DEBUG_INFO, "Found Standalone MM PE data - 0x%x\n", TeData));\r
\r
  Status = GetPeCoffSectionInformation (ImageContext, ImageBase,\r
             SectionHeaderOffset, NumberOfSections);\r
  if (EFI_ERROR (Status)) {\r
    DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM Core PE-COFF Section information - %r\n", Status));\r
    return Status;\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "Standalone MM Core PE-COFF SectionHeaderOffset - 0x%x, NumberOfSections - %d\n",\r
          *SectionHeaderOffset, *NumberOfSections));\r
\r
  return Status;\r
}\r
Commit	Line	Data
184558d0 SV	1	/** @file\r
	2	Locate, get and update PE/COFF permissions during Standalone MM\r
	3	Foundation Entry point on ARM platforms.\r
	4	\r
	5	Copyright (c) 2017 - 2018, ARM Ltd. All rights reserved.<BR>\r
86094561	6	SPDX-License-Identifier: BSD-2-Clause-Patent\r
184558d0 SV	7	\r
	8	**/\r
	9	\r
	10	\r
	11	#include <PiMm.h>\r
	12	\r
	13	#include <PiPei.h>\r
	14	#include <Guid/MmramMemoryReserve.h>\r
	15	#include <Guid/MpInformation.h>\r
	16	\r
	17	#include <Library/AArch64/StandaloneMmCoreEntryPoint.h>\r
	18	#include <Library/ArmMmuLib.h>\r
	19	#include <Library/ArmSvcLib.h>\r
	20	#include <Library/DebugLib.h>\r
	21	#include <Library/HobLib.h>\r
	22	#include <Library/BaseLib.h>\r
	23	#include <Library/BaseMemoryLib.h>\r
	24	#include <Library/SerialPortLib.h>\r
	25	\r
	26	#include <IndustryStandard/ArmStdSmc.h>\r
	27	\r
	28	EFI_STATUS\r
	29	EFIAPI\r
	30	UpdateMmFoundationPeCoffPermissions (\r
	31	IN CONST PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext,\r
493f2c69	32	IN EFI_PHYSICAL_ADDRESS ImageBase,\r
184558d0 SV	33	IN UINT32 SectionHeaderOffset,\r
	34	IN CONST UINT16 NumberOfSections,\r
	35	IN REGION_PERMISSION_UPDATE_FUNC TextUpdater,\r
	36	IN REGION_PERMISSION_UPDATE_FUNC ReadOnlyUpdater,\r
	37	IN REGION_PERMISSION_UPDATE_FUNC ReadWriteUpdater\r
	38	)\r
	39	{\r
	40	EFI_IMAGE_SECTION_HEADER SectionHeader;\r
	41	RETURN_STATUS Status;\r
	42	EFI_PHYSICAL_ADDRESS Base;\r
	43	UINTN Size;\r
	44	UINTN ReadSize;\r
	45	UINTN Index;\r
	46	\r
	47	ASSERT (ImageContext != NULL);\r
	48	\r
	49	//\r
	50	// Iterate over the sections\r
	51	//\r
	52	for (Index = 0; Index < NumberOfSections; Index++) {\r
	53	//\r
	54	// Read section header from file\r
	55	//\r
	56	Size = sizeof (EFI_IMAGE_SECTION_HEADER);\r
	57	ReadSize = Size;\r
	58	Status = ImageContext->ImageRead (\r
	59	ImageContext->Handle,\r
	60	SectionHeaderOffset,\r
	61	&Size,\r
	62	&SectionHeader\r
	63	);\r
	64	\r
	65	if (RETURN_ERROR (Status) \|\| (Size != ReadSize)) {\r
	66	DEBUG ((DEBUG_ERROR,\r
	67	"%a: ImageContext->ImageRead () failed (Status = %r)\n",\r
	68	__FUNCTION__, Status));\r
	69	return Status;\r
	70	}\r
	71	\r
	72	DEBUG ((DEBUG_INFO,\r
	73	"%a: Section %d of image at 0x%lx has 0x%x permissions\n",\r
	74	__FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Characteristics));\r
	75	DEBUG ((DEBUG_INFO,\r
41915a19	76	"%a: Section %d of image at 0x%lx has %a name\n",\r
184558d0 SV	77	__FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Name));\r
	78	DEBUG ((DEBUG_INFO,\r
	79	"%a: Section %d of image at 0x%lx has 0x%x address\n",\r
	80	__FUNCTION__, Index, ImageContext->ImageAddress,\r
	81	ImageContext->ImageAddress + SectionHeader.VirtualAddress));\r
	82	DEBUG ((DEBUG_INFO,\r
	83	"%a: Section %d of image at 0x%lx has 0x%x data\n",\r
	84	__FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.PointerToRawData));\r
	85	\r
	86	//\r
	87	// If the section is marked as XN then remove the X attribute. Furthermore,\r
	88	// if it is a writeable section then mark it appropriately as well.\r
	89	//\r
	90	if ((SectionHeader.Characteristics & EFI_IMAGE_SCN_MEM_EXECUTE) == 0) {\r
493f2c69	91	Base = ImageBase + SectionHeader.VirtualAddress;\r
184558d0 SV	92	\r
	93	TextUpdater (Base, SectionHeader.Misc.VirtualSize);\r
	94	\r
	95	if ((SectionHeader.Characteristics & EFI_IMAGE_SCN_MEM_WRITE) != 0) {\r
	96	ReadWriteUpdater (Base, SectionHeader.Misc.VirtualSize);\r
	97	DEBUG ((DEBUG_INFO,\r
	98	"%a: Mapping section %d of image at 0x%lx with RW-XN permissions\n",\r
	99	__FUNCTION__, Index, ImageContext->ImageAddress));\r
	100	} else {\r
	101	DEBUG ((DEBUG_INFO,\r
	102	"%a: Mapping section %d of image at 0x%lx with RO-XN permissions\n",\r
	103	__FUNCTION__, Index, ImageContext->ImageAddress));\r
	104	}\r
	105	} else {\r
	106	DEBUG ((DEBUG_INFO,\r
	107	"%a: Ignoring section %d of image at 0x%lx with 0x%x permissions\n",\r
	108	__FUNCTION__, Index, ImageContext->ImageAddress, SectionHeader.Characteristics));\r
	109	}\r
	110	SectionHeaderOffset += sizeof (EFI_IMAGE_SECTION_HEADER);\r
	111	}\r
	112	\r
	113	return RETURN_SUCCESS;\r
	114	}\r
	115	\r
	116	EFI_STATUS\r
	117	EFIAPI\r
	118	LocateStandaloneMmCorePeCoffData (\r
	119	IN EFI_FIRMWARE_VOLUME_HEADER *BfvAddress,\r
	120	IN OUT VOID **TeData,\r
	121	IN OUT UINTN *TeDataSize\r
	122	)\r
	123	{\r
	124	EFI_FFS_FILE_HEADER *FileHeader = NULL;\r
	125	EFI_STATUS Status;\r
	126	\r
	127	Status = FfsFindNextFile (\r
	128	EFI_FV_FILETYPE_SECURITY_CORE,\r
	129	BfvAddress,\r
	130	&FileHeader\r
	131	);\r
	132	\r
	133	if (EFI_ERROR (Status)) {\r
	134	DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM FFS file - 0x%x\n",\r
	135	Status));\r
	136	return Status;\r
	137	}\r
	138	\r
	139	Status = FfsFindSectionData (EFI_SECTION_PE32, FileHeader, TeData, TeDataSize);\r
	140	if (EFI_ERROR (Status)) {\r
4b28452d AB	141	Status = FfsFindSectionData (EFI_SECTION_TE, FileHeader, TeData, TeDataSize);\r
	142	if (EFI_ERROR (Status)) {\r
	143	DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM Section data - %r\n",\r
	144	Status));\r
	145	return Status;\r
	146	}\r
184558d0 SV	147	}\r
	148	\r
	149	DEBUG ((DEBUG_INFO, "Found Standalone MM PE data - 0x%x\n", *TeData));\r
	150	return Status;\r
	151	}\r
	152	\r
	153	STATIC\r
	154	EFI_STATUS\r
	155	GetPeCoffSectionInformation (\r
4b28452d	156	IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext,\r
493f2c69	157	OUT EFI_PHYSICAL_ADDRESS *ImageBase,\r
4b28452d AB	158	OUT UINT32 *SectionHeaderOffset,\r
4b28452d AB	159	OUT UINT16 *NumberOfSections\r
184558d0 SV	160	)\r
	161	{\r
	162	RETURN_STATUS Status;\r
	163	EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr;\r
	164	EFI_IMAGE_OPTIONAL_HEADER_UNION HdrData;\r
	165	UINTN Size;\r
	166	UINTN ReadSize;\r
	167	\r
	168	ASSERT (ImageContext != NULL);\r
184558d0 SV	169	ASSERT (SectionHeaderOffset != NULL);\r
	170	ASSERT (NumberOfSections != NULL);\r
	171	\r
4b28452d AB	172	Status = PeCoffLoaderGetImageInfo (ImageContext);\r
	173	if (RETURN_ERROR (Status)) {\r
	174	DEBUG ((DEBUG_ERROR,\r
	175	"%a: PeCoffLoaderGetImageInfo () failed (Status == %r)\n",\r
	176	__FUNCTION__, Status));\r
	177	return Status;\r
184558d0 SV	178	}\r
184558d0 SV	179	\r
4b28452d	180	if (ImageContext->SectionAlignment < EFI_PAGE_SIZE) {\r
184558d0 SV	181	//\r
	182	// The sections need to be at least 4 KB aligned, since that is the\r
	183	// granularity at which we can tighten permissions.\r
	184	//\r
4b28452d	185	if (!ImageContext->IsTeImage) {\r
184558d0 SV	186	DEBUG ((DEBUG_WARN,\r
184558d0 SV	187	"%a: non-TE Image at 0x%lx has SectionAlignment < 4 KB (%lu)\n",\r
4b28452d AB	188	__FUNCTION__, ImageContext->ImageAddress, ImageContext->SectionAlignment));\r
4b28452d AB	189	return RETURN_UNSUPPORTED;\r
184558d0	190	}\r
4b28452d	191	ImageContext->SectionAlignment = EFI_PAGE_SIZE;\r
184558d0 SV	192	}\r
	193	\r
	194	//\r
	195	// Read the PE/COFF Header. For PE32 (32-bit) this will read in too much\r
	196	// data, but that should not hurt anything. Hdr.Pe32->OptionalHeader.Magic\r
	197	// determines if this is a PE32 or PE32+ image. The magic is in the same\r
	198	// location in both images.\r
	199	//\r
	200	Hdr.Union = &HdrData;\r
	201	Size = sizeof (EFI_IMAGE_OPTIONAL_HEADER_UNION);\r
	202	ReadSize = Size;\r
4b28452d AB	203	Status = ImageContext->ImageRead (\r
	204	ImageContext->Handle,\r
	205	ImageContext->PeCoffHeaderOffset,\r
184558d0 SV	206	&Size,\r
	207	Hdr.Pe32\r
	208	);\r
	209	\r
	210	if (RETURN_ERROR (Status) \|\| (Size != ReadSize)) {\r
	211	DEBUG ((DEBUG_ERROR,\r
	212	"%a: TmpContext->ImageRead () failed (Status = %r)\n",\r
	213	__FUNCTION__, Status));\r
	214	return Status;\r
	215	}\r
	216	\r
493f2c69	217	*ImageBase = ImageContext->ImageAddress;\r
4b28452d AB	218	if (!ImageContext->IsTeImage) {\r
	219	ASSERT (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE);\r
	220	\r
	221	*SectionHeaderOffset = ImageContext->PeCoffHeaderOffset + sizeof (UINT32) +\r
	222	sizeof (EFI_IMAGE_FILE_HEADER);\r
	223	*NumberOfSections = Hdr.Pe32->FileHeader.NumberOfSections;\r
	224	\r
	225	switch (Hdr.Pe32->OptionalHeader.Magic) {\r
	226	case EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC:\r
	227	*SectionHeaderOffset += Hdr.Pe32->FileHeader.SizeOfOptionalHeader;\r
	228	break;\r
	229	case EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC:\r
	230	*SectionHeaderOffset += Hdr.Pe32Plus->FileHeader.SizeOfOptionalHeader;\r
	231	break;\r
	232	default:\r
	233	ASSERT (FALSE);\r
	234	}\r
	235	} else {\r
	236	*SectionHeaderOffset = (UINTN)(sizeof (EFI_TE_IMAGE_HEADER));\r
	237	*NumberOfSections = Hdr.Te->NumberOfSections;\r
493f2c69	238	*ImageBase -= (UINT32)Hdr.Te->StrippedSize - sizeof (EFI_TE_IMAGE_HEADER);\r
184558d0	239	}\r
184558d0 SV	240	return RETURN_SUCCESS;\r
	241	}\r
	242	\r
	243	EFI_STATUS\r
	244	EFIAPI\r
	245	GetStandaloneMmCorePeCoffSections (\r
	246	IN VOID *TeData,\r
	247	IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext,\r
493f2c69	248	OUT EFI_PHYSICAL_ADDRESS *ImageBase,\r
184558d0 SV	249	IN OUT UINT32 *SectionHeaderOffset,\r
	250	IN OUT UINT16 *NumberOfSections\r
	251	)\r
	252	{\r
	253	EFI_STATUS Status;\r
184558d0 SV	254	\r
	255	// Initialize the Image Context\r
	256	ZeroMem (ImageContext, sizeof (PE_COFF_LOADER_IMAGE_CONTEXT));\r
	257	ImageContext->Handle = TeData;\r
	258	ImageContext->ImageRead = PeCoffLoaderImageReadFromMemory;\r
	259	\r
	260	DEBUG ((DEBUG_INFO, "Found Standalone MM PE data - 0x%x\n", TeData));\r
	261	\r
493f2c69 AB	262	Status = GetPeCoffSectionInformation (ImageContext, ImageBase,\r
493f2c69 AB	263	SectionHeaderOffset, NumberOfSections);\r
184558d0	264	if (EFI_ERROR (Status)) {\r
4b28452d	265	DEBUG ((DEBUG_ERROR, "Unable to locate Standalone MM Core PE-COFF Section information - %r\n", Status));\r
184558d0 SV	266	return Status;\r
	267	}\r
	268	\r
	269	DEBUG ((DEBUG_INFO, "Standalone MM Core PE-COFF SectionHeaderOffset - 0x%x, NumberOfSections - %d\n",\r
	270	SectionHeaderOffset, NumberOfSections));\r
	271	\r
	272	return Status;\r
	273	}\r