Handle new relocation type EFI_IMAGE_REL_BASED_DIR64 for X64.

[mirror_edk2.git] / MdePkg / Library / BasePeCoffLib / x64 / PeCoffLoaderEx.c

diff --git a/MdePkg/Library/BasePeCoffLib/x64/PeCoffLoaderEx.c b/MdePkg/Library/BasePeCoffLib/x64/PeCoffLoaderEx.c
index 7141adc2ad5ddc5d06ad50f1ca38864022c1f47f..8628dc7df12a1ce9ac858fb25d99e54ac819357f 100644 (file)
--- a/MdePkg/Library/BasePeCoffLib/x64/PeCoffLoaderEx.c
+++ b/MdePkg/Library/BasePeCoffLib/x64/PeCoffLoaderEx.c
@@ -1,16 +1,14 @@
 /** @file\r
   x64 Specific relocation fixups.\r
 \r
-  Copyright (c) 2006, Intel Corporation\r
-  All rights reserved. This program and the accompanying materials\r
-  are licensed and made available under the terms and conditions of the BSD License\r
-  which accompanies this distribution.  The full text of the license may be found at\r
-  http://opensource.org/licenses/bsd-license.php\r
-\r
-  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
-\r
-  Module Name:  PeCoffLoaderEx.c\r
+Copyright (c) 2005 - 2006 Intel Corporation. <BR>\r
+All rights reserved. This program and the accompanying materials                          \r
+are licensed and made available under the terms and conditions of the BSD License         \r
+which accompanies this distribution.  The full text of the license may be found at        \r
+http://opensource.org/licenses/bsd-license.php                                            \r
+                                                                                          \r
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
 \r
 **/\r
 \r
@@ -22,13 +20,15 @@
   @param  FixupData   Pointer to a buffer to log the fixups\r
   @param  Adjust      The offset to adjust the fixup\r
 \r
+  @retval RETURN_SUCCESS      Success to perform relocation\r
+  @retval RETURN_UNSUPPORTED  Unsupported.\r
 **/\r
 RETURN_STATUS\r
 PeCoffLoaderRelocateImageEx (\r
-  IN UINT16      *Reloc,\r
-  IN OUT CHAR8   *Fixup, \r
-  IN OUT CHAR8   **FixupData,\r
-  IN UINT64      Adjust\r
+  IN     UINT16       *Reloc,\r
+  IN OUT CHAR8        *Fixup, \r
+  IN OUT CHAR8        **FixupData,\r
+  IN     UINT64       Adjust\r
   )\r
 {\r
   return RETURN_UNSUPPORTED;\r
@@ -64,7 +64,7 @@ PeCoffLoaderImageFormatSupported (
 \r
 \r
 /**\r
-  Performs an Itanium-based specific re-relocation fixup and is a no-op on other\r
+  Performs an X64 specific re-relocation fixup and is a no-op on other\r
   instruction sets. This is used to re-relocated the image into the EFI virtual\r
   space for runtime calls.\r
 \r
@@ -84,5 +84,22 @@ PeHotRelocateImageEx (
   IN UINT64      Adjust\r
   )\r
 {\r
-  return RETURN_UNSUPPORTED;\r
+  UINT64      *F64;\r
+\r
+  switch ((*Reloc) >> 12) {\r
+    case EFI_IMAGE_REL_BASED_DIR64:\r
+      F64 = (UINT64 *) Fixup;\r
+      *FixupData = ALIGN_POINTER(*FixupData, sizeof(UINT64));\r
+      if (*(UINT64 *)(*FixupData) == *F64) {\r
+        *F64  = *F64 + (UINT64) Adjust;\r
+      }\r
+\r
+      *FixupData = *FixupData + sizeof(UINT64);\r
+      break;\r
+\r
+    default:\r
+      return RETURN_UNSUPPORTED;\r
+  }\r
+ \r
+  return RETURN_SUCCESS;\r
 }\r