Import two CPU Exception Handler Library instances: SecPeiCpuExceptionHandler.inf...

[mirror_edk2.git] / UefiCpuPkg / Library / CpuExceptionHandlerLib / CpuExceptionCommon.c

/** @file\r
  CPU Exception Hanlder Library common functions.\r
\r
  Copyright (c) 2012, Intel Corporation. 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
  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
#include "CpuExceptionCommon.h"\r
\r
//\r
// Error code flag indicating whether or not an error code will be \r
// pushed on the stack if an exception occurs.\r
//\r
// 1 means an error code will be pushed, otherwise 0\r
//\r
UINT32 mErrorCodeFlag = 0x00027d00;\r
\r
//\r
// Define the maximum message length \r
//\r
#define MAX_DEBUG_MESSAGE_LENGTH  0x100\r
\r
/**\r
  Prints a message to the serial port.\r
\r
  @param  Format      Format string for the message to print.\r
  @param  ...         Variable argument list whose contents are accessed \r
                      based on the format string specified by Format.\r
\r
**/\r
VOID\r
EFIAPI\r
InternalPrintMessage (\r
  IN  CONST CHAR8  *Format,\r
  ...\r
  )\r
{\r
  CHAR8    Buffer[MAX_DEBUG_MESSAGE_LENGTH];\r
  VA_LIST  Marker;\r
\r
  //\r
  // Convert the message to an ASCII String\r
  //\r
  VA_START (Marker, Format);\r
  AsciiVSPrint (Buffer, sizeof (Buffer), Format, Marker);\r
  VA_END (Marker);\r
\r
  //\r
  // Send the print string to a Serial Port \r
  //\r
  SerialPortWrite ((UINT8 *)Buffer, AsciiStrLen (Buffer));\r
}\r
\r
/**\r
  Find and display image base address and return image base and its entry point.\r
  \r
  @return EFI_SUCCESS     Image base address.\r
  @return 0               Image header cannot be found.\r
**/\r
UINTN \r
FindModuleImageBase (\r
  IN  UINTN              CurrentEip,\r
  OUT UINTN              *EntryPoint\r
  )\r
{\r
  UINTN                                Pe32Data;\r
  EFI_IMAGE_DOS_HEADER                 *DosHdr;\r
  EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION  Hdr;\r
  VOID                                 *PdbPointer;\r
\r
  //\r
  // Find Image Base\r
  //\r
  Pe32Data = CurrentEip & ~(mImageAlignSize - 1);\r
  while (Pe32Data != 0) {\r
    DosHdr = (EFI_IMAGE_DOS_HEADER *) Pe32Data;\r
    if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
      //\r
      // DOS image header is present, so read the PE header after the DOS image header.\r
      //\r
      Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)(Pe32Data + (UINTN) ((DosHdr->e_lfanew) & 0x0ffff));\r
      if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE) {\r
        //\r
        // It's PE image.\r
        //\r
        InternalPrintMessage ("!!!! Find PE image ");\r
        *EntryPoint = (UINTN)Pe32Data + (UINTN)(Hdr.Pe32->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff);\r
        break;\r
      }\r
    } else {\r
      //\r
      // DOS image header is not present, TE header is at the image base.\r
      //\r
      Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)Pe32Data;\r
      if ((Hdr.Te->Signature == EFI_TE_IMAGE_HEADER_SIGNATURE) &&\r
          ((Hdr.Te->Machine == IMAGE_FILE_MACHINE_I386) || Hdr.Te->Machine == IMAGE_FILE_MACHINE_X64)) {\r
        //\r
        // It's TE image, it TE header and Machine type match\r
        //\r
        InternalPrintMessage ("!!!! Find TE image ");\r
        *EntryPoint = (UINTN)Pe32Data + (UINTN)(Hdr.Te->AddressOfEntryPoint & 0x0ffffffff) + sizeof(EFI_TE_IMAGE_HEADER) - Hdr.Te->StrippedSize;\r
        break;\r
      }\r
    }\r
\r
    //\r
    // Not found the image base, check the previous aligned address\r
    // \r
    Pe32Data -= mImageAlignSize;\r
  }\r
\r
  if (Pe32Data != 0) {\r
    PdbPointer = PeCoffLoaderGetPdbPointer ((VOID *) Pe32Data);\r
    if (PdbPointer != NULL) {\r
      InternalPrintMessage ("%a", PdbPointer);\r
    } else {\r
      InternalPrintMessage ("(No PDB) " );\r
    }\r
  } else {\r
    InternalPrintMessage ("!!!! Can't find image information. !!!!\n");\r
  }\r
\r
  return Pe32Data;\r
}\r
\r