BaseTools/GenFw: Enhance GenFw to support PRM GCC build

[mirror_edk2.git] / BaseTools / Source / C / GenFw / ElfConvert.c

/** @file\r
Elf convert solution\r
\r
Copyright (c) 2010 - 2018, Intel Corporation. All rights reserved.<BR>\r
\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "WinNtInclude.h"\r
\r
#ifndef __GNUC__\r
#include <windows.h>\r
#include <io.h>\r
#endif\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
#include <time.h>\r
#include <ctype.h>\r
#include <assert.h>\r
\r
#include <Common/UefiBaseTypes.h>\r
#include <IndustryStandard/PeImage.h>\r
\r
#include "EfiUtilityMsgs.h"\r
\r
#include "GenFw.h"\r
#include "ElfConvert.h"\r
#include "Elf32Convert.h"\r
#include "Elf64Convert.h"\r
\r
//\r
// Result Coff file in memory.\r
//\r
UINT8 *mCoffFile = NULL;\r
\r
//\r
// COFF relocation data\r
//\r
EFI_IMAGE_BASE_RELOCATION *mCoffBaseRel;\r
UINT16                    *mCoffEntryRel;\r
\r
//\r
// Current offset in coff file.\r
//\r
UINT32 mCoffOffset;\r
\r
//\r
// Offset in Coff file of headers and sections.\r
//\r
UINT32 mTableOffset;\r
\r
//\r
//mFileBufferSize\r
//\r
UINT32 mFileBufferSize;\r
\r
//\r
//*****************************************************************************\r
// Common ELF Functions\r
//*****************************************************************************\r
//\r
\r
VOID\r
CoffAddFixupEntry(\r
  UINT16 Val\r
  )\r
{\r
  *mCoffEntryRel = Val;\r
  mCoffEntryRel++;\r
  mCoffBaseRel->SizeOfBlock += 2;\r
  mCoffOffset += 2;\r
}\r
\r
VOID\r
CoffAddFixup(\r
  UINT32 Offset,\r
  UINT8  Type\r
  )\r
{\r
  if (mCoffBaseRel == NULL\r
      || mCoffBaseRel->VirtualAddress != (Offset & ~0xfff)) {\r
    if (mCoffBaseRel != NULL) {\r
      //\r
      // Add a null entry (is it required ?)\r
      //\r
      CoffAddFixupEntry (0);\r
\r
      //\r
      // Pad for alignment.\r
      //\r
      if (mCoffOffset % 4 != 0)\r
        CoffAddFixupEntry (0);\r
    }\r
\r
    mCoffFile = realloc (\r
      mCoffFile,\r
      mCoffOffset + sizeof(EFI_IMAGE_BASE_RELOCATION) + 2 * MAX_COFF_ALIGNMENT\r
      );\r
    if (mCoffFile == NULL) {\r
      Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
    }\r
    assert (mCoffFile != NULL);\r
    memset (\r
      mCoffFile + mCoffOffset, 0,\r
      sizeof(EFI_IMAGE_BASE_RELOCATION) + 2 * MAX_COFF_ALIGNMENT\r
      );\r
\r
    mCoffBaseRel = (EFI_IMAGE_BASE_RELOCATION*)(mCoffFile + mCoffOffset);\r
    mCoffBaseRel->VirtualAddress = Offset & ~0xfff;\r
    mCoffBaseRel->SizeOfBlock = sizeof(EFI_IMAGE_BASE_RELOCATION);\r
\r
    mCoffEntryRel = (UINT16 *)(mCoffBaseRel + 1);\r
    mCoffOffset += sizeof(EFI_IMAGE_BASE_RELOCATION);\r
  }\r
\r
  //\r
  // Fill the entry.\r
  //\r
  CoffAddFixupEntry((UINT16) ((Type << 12) | (Offset & 0xfff)));\r
}\r
\r
VOID\r
CreateSectionHeader (\r
  const CHAR8 *Name,\r
  UINT32      Offset,\r
  UINT32      Size,\r
  UINT32      Flags\r
  )\r
{\r
  EFI_IMAGE_SECTION_HEADER *Hdr;\r
  Hdr = (EFI_IMAGE_SECTION_HEADER*)(mCoffFile + mTableOffset);\r
\r
  strcpy((char *)Hdr->Name, Name);\r
  Hdr->Misc.VirtualSize = Size;\r
  Hdr->VirtualAddress = Offset;\r
  Hdr->SizeOfRawData = Size;\r
  Hdr->PointerToRawData = Offset;\r
  Hdr->PointerToRelocations = 0;\r
  Hdr->PointerToLinenumbers = 0;\r
  Hdr->NumberOfRelocations = 0;\r
  Hdr->NumberOfLinenumbers = 0;\r
  Hdr->Characteristics = Flags;\r
\r
  mTableOffset += sizeof (EFI_IMAGE_SECTION_HEADER);\r
}\r
\r
//\r
//*****************************************************************************\r
// Functions called from GenFw main code.\r
//*****************************************************************************\r
//\r
\r
INTN\r
IsElfHeader (\r
  UINT8  *FileBuffer\r
)\r
{\r
  return (FileBuffer[EI_MAG0] == ELFMAG0 &&\r
          FileBuffer[EI_MAG1] == ELFMAG1 &&\r
          FileBuffer[EI_MAG2] == ELFMAG2 &&\r
          FileBuffer[EI_MAG3] == ELFMAG3);\r
}\r
\r
BOOLEAN\r
ConvertElf (\r
  UINT8  **FileBuffer,\r
  UINT32 *FileLength\r
  )\r
{\r
  ELF_FUNCTION_TABLE              ElfFunctions;\r
  UINT8                           EiClass;\r
\r
  mFileBufferSize = *FileLength;\r
  //\r
  // Determine ELF type and set function table pointer correctly.\r
  //\r
  VerboseMsg ("Check Elf Image Header");\r
  EiClass = (*FileBuffer)[EI_CLASS];\r
  if (EiClass == ELFCLASS32) {\r
    if (!InitializeElf32 (*FileBuffer, &ElfFunctions)) {\r
      return FALSE;\r
    }\r
  } else if (EiClass == ELFCLASS64) {\r
    if (!InitializeElf64 (*FileBuffer, &ElfFunctions)) {\r
      return FALSE;\r
    }\r
  } else {\r
    Error (NULL, 0, 3000, "Unsupported", "ELF EI_CLASS not supported.");\r
    return FALSE;\r
  }\r
\r
  //\r
  // Compute sections new address.\r
  //\r
  VerboseMsg ("Compute sections new address.");\r
  ElfFunctions.ScanSections ();\r
\r
  //\r
  // Write and relocate sections.\r
  //\r
  VerboseMsg ("Write and relocate sections.");\r
  if (!ElfFunctions.WriteSections (SECTION_TEXT)) {\r
    return FALSE;\r
  }\r
  if (!ElfFunctions.WriteSections (SECTION_DATA)) {\r
    return FALSE;\r
  }\r
  if (!ElfFunctions.WriteSections (SECTION_HII)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Translate and write relocations.\r
  //\r
  VerboseMsg ("Translate and write relocations.");\r
  ElfFunctions.WriteRelocations ();\r
\r
  //\r
  // Write debug info.\r
  //\r
  VerboseMsg ("Write debug info.");\r
  ElfFunctions.WriteDebug ();\r
\r
  //\r
  // For PRM Driver to Write export info.\r
  //\r
  if (mExportFlag) {\r
    VerboseMsg ("Write export info.");\r
    ElfFunctions.WriteExport ();\r
  }\r
\r
  //\r
  // Make sure image size is correct before returning the new image.\r
  //\r
  VerboseMsg ("Set image size.");\r
  ElfFunctions.SetImageSize ();\r
\r
  //\r
  // Replace.\r
  //\r
  free (*FileBuffer);\r
  *FileBuffer = mCoffFile;\r
  *FileLength = mCoffOffset;\r
\r
  //\r
  // Free resources used by ELF functions.\r
  //\r
  ElfFunctions.CleanUp ();\r
\r
  return TRUE;\r
}\r