mirror of https://github.com/acidanthera/audk.git
138 lines
4.4 KiB
C
138 lines
4.4 KiB
C
/** @file
|
|
PE/Coff loader for LoongArch PE image
|
|
|
|
Copyright (c) 2022, Loongson Technology Corporation Limited. All rights reserved.<BR>
|
|
|
|
SPDX-License-Identifier: BSD-2-Clause-Patent
|
|
**/
|
|
|
|
#include "BasePeCoffLibInternals.h"
|
|
#include <Library/BaseLib.h>
|
|
|
|
/**
|
|
Performs an LoongArch specific relocation fixup and is a no-op on other
|
|
instruction sets.
|
|
|
|
@param[in] Reloc Pointer to the relocation record.
|
|
@param[in, out] Fixup Pointer to the address to fix up.
|
|
@param[in, out] FixupData Pointer to a buffer to log the fixups.
|
|
@param[in] Adjust The offset to adjust the fixup.
|
|
|
|
@return Status code.
|
|
|
|
**/
|
|
RETURN_STATUS
|
|
PeCoffLoaderRelocateImageEx (
|
|
IN UINT16 *Reloc,
|
|
IN OUT CHAR8 *Fixup,
|
|
IN OUT CHAR8 **FixupData,
|
|
IN UINT64 Adjust
|
|
)
|
|
{
|
|
UINT8 RelocType;
|
|
UINT64 Value;
|
|
UINT64 Tmp1;
|
|
UINT64 Tmp2;
|
|
|
|
RelocType = (*Reloc) >> 12;
|
|
Value = 0;
|
|
Tmp1 = 0;
|
|
Tmp2 = 0;
|
|
|
|
switch (RelocType) {
|
|
case EFI_IMAGE_REL_BASED_LOONGARCH64_MARK_LA:
|
|
// The next four instructions are used to load a 64 bit address, relocate all of them
|
|
Value = (*(UINT32 *)Fixup & 0x1ffffe0) << 7 | // lu12i.w 20bits from bit5
|
|
(*((UINT32 *)Fixup + 1) & 0x3ffc00) >> 10; // ori 12bits from bit10
|
|
Tmp1 = *((UINT32 *)Fixup + 2) & 0x1ffffe0; // lu32i.d 20bits from bit5
|
|
Tmp2 = *((UINT32 *)Fixup + 3) & 0x3ffc00; // lu52i.d 12bits from bit10
|
|
Value = Value | (Tmp1 << 27) | (Tmp2 << 42);
|
|
Value += Adjust;
|
|
|
|
*(UINT32 *)Fixup = (*(UINT32 *)Fixup & ~0x1ffffe0) | (((Value >> 12) & 0xfffff) << 5);
|
|
if (*FixupData != NULL) {
|
|
*FixupData = ALIGN_POINTER (*FixupData, sizeof (UINT32));
|
|
*(UINT32 *)(*FixupData) = *(UINT32 *)Fixup;
|
|
*FixupData = *FixupData + sizeof (UINT32);
|
|
}
|
|
|
|
Fixup += sizeof (UINT32);
|
|
*(UINT32 *)Fixup = (*(UINT32 *)Fixup & ~0x3ffc00) | ((Value & 0xfff) << 10);
|
|
if (*FixupData != NULL) {
|
|
*FixupData = ALIGN_POINTER (*FixupData, sizeof (UINT32));
|
|
*(UINT32 *)(*FixupData) = *(UINT32 *)Fixup;
|
|
*FixupData = *FixupData + sizeof (UINT32);
|
|
}
|
|
|
|
Fixup += sizeof (UINT32);
|
|
*(UINT32 *)Fixup = (*(UINT32 *)Fixup & ~0x1ffffe0) | (((Value >> 32) & 0xfffff) << 5);
|
|
if (*FixupData != NULL) {
|
|
*FixupData = ALIGN_POINTER (*FixupData, sizeof (UINT32));
|
|
*(UINT32 *)(*FixupData) = *(UINT32 *)Fixup;
|
|
*FixupData = *FixupData + sizeof (UINT32);
|
|
}
|
|
|
|
Fixup += sizeof (UINT32);
|
|
*(UINT32 *)Fixup = (*(UINT32 *)Fixup & ~0x3ffc00) | (((Value >> 52) & 0xfff) << 10);
|
|
if (*FixupData != NULL) {
|
|
*FixupData = ALIGN_POINTER (*FixupData, sizeof (UINT32));
|
|
*(UINT32 *)(*FixupData) = *(UINT32 *)Fixup;
|
|
*FixupData = *FixupData + sizeof (UINT32);
|
|
}
|
|
|
|
break;
|
|
default:
|
|
return RETURN_UNSUPPORTED;
|
|
}
|
|
|
|
return RETURN_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
Returns TRUE if the machine type of PE/COFF image is supported. Supported
|
|
does not mean the image can be executed it means the PE/COFF loader supports
|
|
loading and relocating of the image type. It's up to the caller to support
|
|
the entry point.
|
|
|
|
@param[in] Machine Machine type from the PE Header.
|
|
|
|
@return TRUE if this PE/COFF loader can load the image
|
|
|
|
**/
|
|
BOOLEAN
|
|
PeCoffLoaderImageFormatSupported (
|
|
IN UINT16 Machine
|
|
)
|
|
{
|
|
if (Machine == IMAGE_FILE_MACHINE_LOONGARCH64) {
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/**
|
|
Performs an LOONGARCH-based specific re-relocation fixup and is a no-op on other
|
|
instruction sets. This is used to re-relocated the image into the EFI virtual
|
|
space for runtime calls.
|
|
|
|
@param[in] Reloc The pointer to the relocation record.
|
|
@param[in, out] Fixup The pointer to the address to fix up.
|
|
@param[in, out] FixupData The pointer to a buffer to log the fixups.
|
|
@param[in] Adjust The offset to adjust the fixup.
|
|
|
|
@return Status code.
|
|
|
|
**/
|
|
RETURN_STATUS
|
|
PeHotRelocateImageEx (
|
|
IN UINT16 *Reloc,
|
|
IN OUT CHAR8 *Fixup,
|
|
IN OUT CHAR8 **FixupData,
|
|
IN UINT64 Adjust
|
|
)
|
|
{
|
|
// To check
|
|
return PeCoffLoaderRelocateImageEx (Reloc, Fixup, FixupData, Adjust);
|
|
}
|