Sync BaseTool trunk (version r2599) into EDKII BaseTools.

[mirror_edk2.git] / BaseTools / Source / C / Common / PeCoffLoaderEx.c

diff --git a/BaseTools/Source/C/Common/PeCoffLoaderEx.c b/BaseTools/Source/C/Common/PeCoffLoaderEx.c
index 2afd441845c4f45d1af191db7f3b3d9096f83c56..8d6b3961f5272243de53cfdc053ef0b566ce2102 100644 (file)
--- a/BaseTools/Source/C/Common/PeCoffLoaderEx.c
+++ b/BaseTools/Source/C/Common/PeCoffLoaderEx.c
@@ -1,6 +1,7 @@
 /** @file\r
 \r
 Copyright (c) 2004 - 2008, Intel Corporation. All rights reserved.<BR>\r
+Portions Copyright (c) 2011 - 2013, ARM Ltd. All rights reserved.<BR>\r
 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
@@ -15,7 +16,7 @@ Module Name:
 \r
 Abstract:\r
 \r
-    IA32, X64 and IPF Specific relocation fixups\r
+    IA32, X64, IPF, ARM and AArch64 Specific relocation fixups\r
 \r
 Revision History\r
 \r
@@ -25,6 +26,7 @@ Revision History
 #include <IndustryStandard/PeImage.h>\r
 #include "PeCoffLib.h"\r
 #include "CommonLib.h"\r
+#include "EfiUtilityMsgs.h"\r
 \r
 \r
 #define EXT_IMM64(Value, Address, Size, InstPos, ValPos)  \\r
@@ -472,3 +474,43 @@ PeCoffLoaderRelocateArmImage (
 \r
   return RETURN_SUCCESS;\r
 }\r
+\r
+RETURN_STATUS\r
+PeCoffLoaderRelocateAArch64Image (\r
+  IN     UINT16       *Reloc,\r
+  IN OUT CHAR8        *Fixup,\r
+  IN OUT CHAR8        **FixupData,\r
+  IN     UINT64       Adjust\r
+  )\r
+/**\r
+  Performs an AArch64 specific relocation fixup\r
+\r
+  @param Reloc        Pointer to the relocation record\r
+  @param Fixup        Pointer to the address to fix up\r
+  @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
+{\r
+  UINT64      *F64;\r
+\r
+  switch ((*Reloc) >> 12) {\r
+\r
+    case EFI_IMAGE_REL_BASED_DIR64:\r
+      F64 = (UINT64 *) Fixup;\r
+      *F64 = *F64 + (UINT64) Adjust;\r
+      if (*FixupData != NULL) {\r
+        *FixupData = ALIGN_POINTER(*FixupData, sizeof(UINT64));\r
+        *(UINT64 *)(*FixupData) = *F64;\r
+        *FixupData = *FixupData + sizeof(UINT64);\r
+      }\r
+      break;\r
+\r
+    default:\r
+      return RETURN_UNSUPPORTED;\r
+  }\r
+\r
+  return RETURN_SUCCESS;\r
+}\r