MdeModulePkg/PiSmmCore: SmmEntryPoint underflow (CVE-2021-38578)

[mirror_edk2.git] / MdeModulePkg / Library / SmmReportStatusCodeLib / ReportStatusCodeLib.c

/** @file\r
  Report Status Code Library for MM Phase.\r
\r
  Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <Library/ReportStatusCodeLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/MmServicesTableLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/PcdLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
\r
#include <Guid/StatusCodeDataTypeId.h>\r
#include <Guid/StatusCodeDataTypeDebug.h>\r
#include <Protocol/MmStatusCode.h>\r
\r
#include "ReportStatusCodeLib.h"\r
\r
EFI_MM_REPORT_STATUS_CODE    mReportStatusCode    = NULL;\r
EFI_MM_STATUS_CODE_PROTOCOL  *mStatusCodeProtocol = NULL;\r
\r
/**\r
  Locate the report status code service.\r
\r
  @return   Function pointer to the report status code service.\r
            NULL is returned if no status code service is available.\r
\r
**/\r
EFI_MM_REPORT_STATUS_CODE\r
InternalGetReportStatusCode (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  Status = InternalLocateProtocol (&gEfiMmStatusCodeProtocolGuid, NULL, (VOID **)&mStatusCodeProtocol);\r
  if (!EFI_ERROR (Status) && (mStatusCodeProtocol != NULL)) {\r
    return mStatusCodeProtocol->ReportStatusCode;\r
  }\r
\r
  return NULL;\r
}\r
\r
/**\r
  Internal worker function that reports a status code through the status code service.\r
\r
  If status code service is not cached, then this function checks if status code service is\r
  available in system.  If status code service is not available, then EFI_UNSUPPORTED is\r
  returned.  If status code service is present, then it is cached in mReportStatusCode.\r
  Finally this function reports status code through the status code service.\r
\r
  @param  Type              Status code type.\r
  @param  Value             Status code value.\r
  @param  Instance          Status code instance number.\r
  @param  CallerId          Pointer to a GUID that identifies the caller of this\r
                            function.  This is an optional parameter that may be\r
                            NULL.\r
  @param  Data              Pointer to the extended data buffer.  This is an\r
                            optional parameter that may be NULL.\r
\r
  @retval EFI_SUCCESS       The status code was reported.\r
  @retval EFI_UNSUPPORTED   Status code service is not available.\r
  @retval EFI_UNSUPPORTED   Status code type is not supported.\r
\r
**/\r
EFI_STATUS\r
InternalReportStatusCode (\r
  IN EFI_STATUS_CODE_TYPE   Type,\r
  IN EFI_STATUS_CODE_VALUE  Value,\r
  IN UINT32                 Instance,\r
  IN CONST EFI_GUID         *CallerId OPTIONAL,\r
  IN EFI_STATUS_CODE_DATA   *Data     OPTIONAL\r
  )\r
{\r
  if ((ReportProgressCodeEnabled () && (((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_PROGRESS_CODE)) ||\r
      (ReportErrorCodeEnabled () && (((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_ERROR_CODE)) ||\r
      (ReportDebugCodeEnabled () && (((Type) & EFI_STATUS_CODE_TYPE_MASK) == EFI_DEBUG_CODE)))\r
  {\r
    //\r
    // If mReportStatusCode is NULL, then check if status code service is available in system.\r
    //\r
    if (mReportStatusCode == NULL) {\r
      mReportStatusCode = InternalGetReportStatusCode ();\r
      if (mReportStatusCode == NULL) {\r
        return EFI_UNSUPPORTED;\r
      }\r
    }\r
\r
    //\r
    // A status code service is present in system, so pass in all the parameters to the service.\r
    //\r
    return (*mReportStatusCode)(mStatusCodeProtocol, Type, Value, Instance, (EFI_GUID *)CallerId, Data);\r
  }\r
\r
  return EFI_UNSUPPORTED;\r
}\r
\r
/**\r
  Converts a status code to an 8-bit POST code value.\r
\r
  Converts the status code specified by CodeType and Value to an 8-bit POST code\r
  and returns the 8-bit POST code in PostCode.  If CodeType is an\r
  EFI_PROGRESS_CODE or CodeType is an EFI_ERROR_CODE, then bits 0..4 of PostCode\r
  are set to bits 16..20 of Value, and bits 5..7 of PostCode are set to bits\r
  24..26 of Value., and TRUE is returned.  Otherwise, FALSE is returned.\r
\r
  If PostCode is NULL, then ASSERT().\r
\r
  @param  CodeType  The type of status code being converted.\r
  @param  Value     The status code value being converted.\r
  @param  PostCode  A pointer to the 8-bit POST code value to return.\r
\r
  @retval  TRUE   The status code specified by CodeType and Value was converted\r
                  to an 8-bit POST code and returned in  PostCode.\r
  @retval  FALSE  The status code specified by CodeType and Value could not be\r
                  converted to an 8-bit POST code value.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
CodeTypeToPostCode (\r
  IN  EFI_STATUS_CODE_TYPE   CodeType,\r
  IN  EFI_STATUS_CODE_VALUE  Value,\r
  OUT UINT8                  *PostCode\r
  )\r
{\r
  //\r
  // If PostCode is NULL, then ASSERT()\r
  //\r
  ASSERT (PostCode != NULL);\r
\r
  //\r
  // Convert Value to an 8 bit post code\r
  //\r
  if (((CodeType & EFI_STATUS_CODE_TYPE_MASK) == EFI_PROGRESS_CODE) ||\r
      ((CodeType & EFI_STATUS_CODE_TYPE_MASK) == EFI_ERROR_CODE))\r
  {\r
    *PostCode = (UINT8)((((Value & EFI_STATUS_CODE_CLASS_MASK) >> 24) << 5) |\r
                        (((Value & EFI_STATUS_CODE_SUBCLASS_MASK) >> 16) & 0x1f));\r
    return TRUE;\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Extracts ASSERT() information from a status code structure.\r
\r
  Converts the status code specified by CodeType, Value, and Data to the ASSERT()\r
  arguments specified by Filename, Description, and LineNumber.  If CodeType is\r
  an EFI_ERROR_CODE, and CodeType has a severity of EFI_ERROR_UNRECOVERED, and\r
  Value has an operation mask of EFI_SW_EC_ILLEGAL_SOFTWARE_STATE, extract\r
  Filename, Description, and LineNumber from the optional data area of the\r
  status code buffer specified by Data.  The optional data area of Data contains\r
  a Null-terminated ASCII string for the FileName, followed by a Null-terminated\r
  ASCII string for the Description, followed by a 32-bit LineNumber.  If the\r
  ASSERT() information could be extracted from Data, then return TRUE.\r
  Otherwise, FALSE is returned.\r
\r
  If Data is NULL, then ASSERT().\r
  If Filename is NULL, then ASSERT().\r
  If Description is NULL, then ASSERT().\r
  If LineNumber is NULL, then ASSERT().\r
\r
  @param  CodeType     The type of status code being converted.\r
  @param  Value        The status code value being converted.\r
  @param  Data         Pointer to status code data buffer.\r
  @param  Filename     Pointer to the source file name that generated the ASSERT().\r
  @param  Description  Pointer to the description of the ASSERT().\r
  @param  LineNumber   Pointer to source line number that generated the ASSERT().\r
\r
  @retval  TRUE   The status code specified by CodeType, Value, and Data was\r
                  converted ASSERT() arguments specified by Filename, Description,\r
                  and LineNumber.\r
  @retval  FALSE  The status code specified by CodeType, Value, and Data could\r
                  not be converted to ASSERT() arguments.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
ReportStatusCodeExtractAssertInfo (\r
  IN EFI_STATUS_CODE_TYPE        CodeType,\r
  IN EFI_STATUS_CODE_VALUE       Value,\r
  IN CONST EFI_STATUS_CODE_DATA  *Data,\r
  OUT CHAR8                      **Filename,\r
  OUT CHAR8                      **Description,\r
  OUT UINT32                     *LineNumber\r
  )\r
{\r
  EFI_DEBUG_ASSERT_DATA  *AssertData;\r
\r
  ASSERT (Data        != NULL);\r
  ASSERT (Filename    != NULL);\r
  ASSERT (Description != NULL);\r
  ASSERT (LineNumber  != NULL);\r
\r
  if (((CodeType & EFI_STATUS_CODE_TYPE_MASK)      == EFI_ERROR_CODE) &&\r
      ((CodeType & EFI_STATUS_CODE_SEVERITY_MASK)  == EFI_ERROR_UNRECOVERED) &&\r
      ((Value    & EFI_STATUS_CODE_OPERATION_MASK) == EFI_SW_EC_ILLEGAL_SOFTWARE_STATE))\r
  {\r
    AssertData   = (EFI_DEBUG_ASSERT_DATA *)(Data + 1);\r
    *Filename    = (CHAR8 *)(AssertData + 1);\r
    *Description = *Filename + AsciiStrLen (*Filename) + 1;\r
    *LineNumber  = AssertData->LineNumber;\r
    return TRUE;\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Extracts DEBUG() information from a status code structure.\r
\r
  Converts the status code specified by Data to the DEBUG() arguments specified\r
  by ErrorLevel, Marker, and Format.  If type GUID in Data is\r
  EFI_STATUS_CODE_DATA_TYPE_DEBUG_GUID, then extract ErrorLevel, Marker, and\r
  Format from the optional data area of the status code buffer specified by Data.\r
  The optional data area of Data contains a 32-bit ErrorLevel followed by Marker\r
  which is 12 UINTN parameters, followed by a Null-terminated ASCII string for\r
  the Format.  If the DEBUG() information could be extracted from Data, then\r
  return TRUE.  Otherwise, FALSE is returned.\r
\r
  If Data is NULL, then ASSERT().\r
  If ErrorLevel is NULL, then ASSERT().\r
  If Marker is NULL, then ASSERT().\r
  If Format is NULL, then ASSERT().\r
\r
  @param  Data        Pointer to status code data buffer.\r
  @param  ErrorLevel  Pointer to error level mask for a debug message.\r
  @param  Marker      Pointer to the variable argument list associated with Format.\r
  @param  Format      Pointer to a Null-terminated ASCII format string of a\r
                      debug message.\r
\r
  @retval  TRUE   The status code specified by Data was converted DEBUG() arguments\r
                  specified by ErrorLevel, Marker, and Format.\r
  @retval  FALSE  The status code specified by Data could not be converted to\r
                  DEBUG() arguments.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
ReportStatusCodeExtractDebugInfo (\r
  IN CONST EFI_STATUS_CODE_DATA  *Data,\r
  OUT UINT32                     *ErrorLevel,\r
  OUT BASE_LIST                  *Marker,\r
  OUT CHAR8                      **Format\r
  )\r
{\r
  EFI_DEBUG_INFO  *DebugInfo;\r
\r
  ASSERT (Data       != NULL);\r
  ASSERT (ErrorLevel != NULL);\r
  ASSERT (Marker     != NULL);\r
  ASSERT (Format     != NULL);\r
\r
  //\r
  // If the GUID type is not EFI_STATUS_CODE_DATA_TYPE_DEBUG_GUID then return FALSE\r
  //\r
  if (!CompareGuid (&Data->Type, &gEfiStatusCodeDataTypeDebugGuid)) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Retrieve the debug information from the status code record\r
  //\r
  DebugInfo = (EFI_DEBUG_INFO *)(Data + 1);\r
\r
  *ErrorLevel = DebugInfo->ErrorLevel;\r
\r
  //\r
  // The first 12 * sizeof (UINT64) bytes following EFI_DEBUG_INFO are for variable arguments\r
  // of format in DEBUG string. Its address is returned in Marker and has to be 64-bit aligned.\r
  // It must be noticed that EFI_DEBUG_INFO follows EFI_STATUS_CODE_DATA, whose size is\r
  // 20 bytes. The size of EFI_DEBUG_INFO is 4 bytes, so we can ensure that Marker\r
  // returned is 64-bit aligned.\r
  // 64-bit aligned is a must, otherwise retrieving 64-bit parameter from BASE_LIST will\r
  // cause unalignment exception.\r
  //\r
  *Marker = (BASE_LIST)(DebugInfo + 1);\r
  *Format = (CHAR8 *)(((UINT64 *)*Marker) + 12);\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Reports a status code.\r
\r
  Reports the status code specified by the parameters Type and Value.  Status\r
  code also require an instance, caller ID, and extended data.  This function\r
  passed in a zero instance, NULL extended data, and a caller ID of\r
  gEfiCallerIdGuid, which is the GUID for the module.\r
\r
  ReportStatusCode()must actively prevent recusrsion.  If ReportStatusCode()\r
  is called while processing another any other Report Status Code Library function,\r
  then ReportStatusCode() must return immediately.\r
\r
  @param  Type   Status code type.\r
  @param  Value  Status code value.\r
\r
  @retval  EFI_SUCCESS       The status code was reported.\r
  @retval  EFI_DEVICE_ERROR  There status code could not be reported due to a\r
                             device error.\r
  @retval  EFI_UNSUPPORTED   Report status code is not supported\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ReportStatusCode (\r
  IN EFI_STATUS_CODE_TYPE   Type,\r
  IN EFI_STATUS_CODE_VALUE  Value\r
  )\r
{\r
  return InternalReportStatusCode (Type, Value, 0, &gEfiCallerIdGuid, NULL);\r
}\r
\r
/**\r
  Reports a status code with an extended data buffer.\r
\r
  Allocates and fills in the extended data section of a status code with the\r
  extended data specified by ExtendedData and ExtendedDataSize.  ExtendedData\r
  is assumed to be one of the data structures specified in Related Definitions.\r
  These data structure do not have the standard header, so this function is\r
  responsible for allocating a buffer large enough for the standard header and\r
  the extended data passed into this function.  The standard header is filled\r
  in with a GUID of  gEfiStatusCodeSpecificDataGuid.  The status code is reported\r
  with a zero instance and a caller ID of gEfiCallerIdGuid.\r
\r
  ReportStatusCodeWithExtendedData()must actively prevent recursion.  If\r
  ReportStatusCodeWithExtendedData() is called while processing another any other\r
  Report Status Code Library function, then ReportStatusCodeWithExtendedData()\r
  must return EFI_DEVICE_ERROR immediately.\r
\r
  If ExtendedData is NULL, then ASSERT().\r
  If ExtendedDataSize is 0, then ASSERT().\r
\r
  @param  Type              Status code type.\r
  @param  Value             Status code value.\r
  @param  ExtendedData      Pointer to the extended data buffer to be reported.\r
  @param  ExtendedDataSize  The size, in bytes, of the extended data buffer to\r
                            be reported.\r
\r
  @retval  EFI_SUCCESS           The status code was reported with the extended\r
                                 data specified by ExtendedData and ExtendedDataSize.\r
  @retval  EFI_OUT_OF_RESOURCES  There were not enough resources to allocate the\r
                                 extended data section.\r
  @retval  EFI_UNSUPPORTED       Report status code is not supported\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ReportStatusCodeWithExtendedData (\r
  IN EFI_STATUS_CODE_TYPE   Type,\r
  IN EFI_STATUS_CODE_VALUE  Value,\r
  IN CONST VOID             *ExtendedData,\r
  IN UINTN                  ExtendedDataSize\r
  )\r
{\r
  ASSERT (ExtendedData     != NULL);\r
  ASSERT (ExtendedDataSize != 0);\r
  return ReportStatusCodeEx (\r
           Type,\r
           Value,\r
           0,\r
           NULL,\r
           NULL,\r
           ExtendedData,\r
           ExtendedDataSize\r
           );\r
}\r
\r
/**\r
  Reports a status code with full parameters.\r
\r
  The function reports a status code.  If ExtendedData is NULL and ExtendedDataSize\r
  is 0, then an extended data buffer is not reported.  If ExtendedData is not\r
  NULL and ExtendedDataSize is not 0, then an extended data buffer is allocated.\r
  ExtendedData is assumed not have the standard status code header, so this function\r
  is responsible for allocating a buffer large enough for the standard header and\r
  the extended data passed into this function.  The standard header is filled in\r
  with a GUID specified by ExtendedDataGuid.  If ExtendedDataGuid is NULL, then a\r
  GUID of gEfiStatusCodeSpecificDataGuid is used.  The status code is reported with\r
  an instance specified by Instance and a caller ID specified by CallerId.  If\r
  CallerId is NULL, then a caller ID of gEfiCallerIdGuid is used.\r
\r
  ReportStatusCodeEx()must actively prevent recursion. If\r
  ReportStatusCodeEx() is called while processing another any\r
  other Report Status Code Library function, then\r
  ReportStatusCodeEx() must return EFI_DEVICE_ERROR immediately.\r
\r
  If ExtendedData is NULL and ExtendedDataSize is not zero, then ASSERT().\r
  If ExtendedData is not NULL and ExtendedDataSize is zero, then ASSERT().\r
\r
  @param  Type              Status code type.\r
  @param  Value             Status code value.\r
  @param  Instance          Status code instance number.\r
  @param  CallerId          Pointer to a GUID that identifies the caller of this\r
                            function.  If this parameter is NULL, then a caller\r
                            ID of gEfiCallerIdGuid is used.\r
  @param  ExtendedDataGuid  Pointer to the GUID for the extended data buffer.\r
                            If this parameter is NULL, then a the status code\r
                            standard header is filled in with\r
                            gEfiStatusCodeSpecificDataGuid.\r
  @param  ExtendedData      Pointer to the extended data buffer.  This is an\r
                            optional parameter that may be NULL.\r
  @param  ExtendedDataSize  The size, in bytes, of the extended data buffer.\r
\r
  @retval  EFI_SUCCESS           The status code was reported.\r
  @retval  EFI_OUT_OF_RESOURCES  There were not enough resources to allocate\r
                                 the extended data section if it was specified.\r
  @retval  EFI_UNSUPPORTED       Report status code is not supported\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ReportStatusCodeEx (\r
  IN EFI_STATUS_CODE_TYPE   Type,\r
  IN EFI_STATUS_CODE_VALUE  Value,\r
  IN UINT32                 Instance,\r
  IN CONST EFI_GUID         *CallerId          OPTIONAL,\r
  IN CONST EFI_GUID         *ExtendedDataGuid  OPTIONAL,\r
  IN CONST VOID             *ExtendedData      OPTIONAL,\r
  IN UINTN                  ExtendedDataSize\r
  )\r
{\r
  EFI_STATUS            Status;\r
  EFI_STATUS_CODE_DATA  *StatusCodeData;\r
\r
  ASSERT (!((ExtendedData == NULL) && (ExtendedDataSize != 0)));\r
  ASSERT (!((ExtendedData != NULL) && (ExtendedDataSize == 0)));\r
\r
  //\r
  // Allocate space for the Status Code Header and its buffer\r
  //\r
  StatusCodeData = AllocatePool (sizeof (EFI_STATUS_CODE_DATA) + ExtendedDataSize);\r
  if (StatusCodeData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  //\r
  // Fill in the extended data header\r
  //\r
  StatusCodeData->HeaderSize = (UINT16)sizeof (EFI_STATUS_CODE_DATA);\r
  StatusCodeData->Size       = (UINT16)ExtendedDataSize;\r
  if (ExtendedDataGuid == NULL) {\r
    ExtendedDataGuid = &gEfiStatusCodeSpecificDataGuid;\r
  }\r
\r
  CopyGuid (&StatusCodeData->Type, ExtendedDataGuid);\r
\r
  //\r
  // Fill in the extended data buffer\r
  //\r
  if (ExtendedData != NULL) {\r
    CopyMem (StatusCodeData + 1, ExtendedData, ExtendedDataSize);\r
  }\r
\r
  //\r
  // Report the status code\r
  //\r
  if (CallerId == NULL) {\r
    CallerId = &gEfiCallerIdGuid;\r
  }\r
\r
  Status = InternalReportStatusCode (Type, Value, Instance, CallerId, StatusCodeData);\r
\r
  //\r
  // Free the allocated buffer\r
  //\r
  FreePool (StatusCodeData);\r
\r
  return Status;\r
}\r
\r
/**\r
  Returns TRUE if status codes of type EFI_PROGRESS_CODE are enabled\r
\r
  This function returns TRUE if the REPORT_STATUS_CODE_PROPERTY_PROGRESS_CODE_ENABLED\r
  bit of PcdReportStatusCodeProperyMask is set.  Otherwise FALSE is returned.\r
\r
  @retval  TRUE   The REPORT_STATUS_CODE_PROPERTY_PROGRESS_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is set.\r
  @retval  FALSE  The REPORT_STATUS_CODE_PROPERTY_PROGRESS_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is clear.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
ReportProgressCodeEnabled (\r
  VOID\r
  )\r
{\r
  return (BOOLEAN)((PcdGet8 (PcdReportStatusCodePropertyMask) & REPORT_STATUS_CODE_PROPERTY_PROGRESS_CODE_ENABLED) != 0);\r
}\r
\r
/**\r
  Returns TRUE if status codes of type EFI_ERROR_CODE are enabled\r
\r
  This function returns TRUE if the REPORT_STATUS_CODE_PROPERTY_ERROR_CODE_ENABLED\r
  bit of PcdReportStatusCodeProperyMask is set.  Otherwise FALSE is returned.\r
\r
  @retval  TRUE   The REPORT_STATUS_CODE_PROPERTY_ERROR_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is set.\r
  @retval  FALSE  The REPORT_STATUS_CODE_PROPERTY_ERROR_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is clear.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
ReportErrorCodeEnabled (\r
  VOID\r
  )\r
{\r
  return (BOOLEAN)((PcdGet8 (PcdReportStatusCodePropertyMask) & REPORT_STATUS_CODE_PROPERTY_ERROR_CODE_ENABLED) != 0);\r
}\r
\r
/**\r
  Returns TRUE if status codes of type EFI_DEBUG_CODE are enabled\r
\r
  This function returns TRUE if the REPORT_STATUS_CODE_PROPERTY_DEBUG_CODE_ENABLED\r
  bit of PcdReportStatusCodeProperyMask is set.  Otherwise FALSE is returned.\r
\r
  @retval  TRUE   The REPORT_STATUS_CODE_PROPERTY_DEBUG_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is set.\r
  @retval  FALSE  The REPORT_STATUS_CODE_PROPERTY_DEBUG_CODE_ENABLED bit of\r
                  PcdReportStatusCodeProperyMask is clear.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
ReportDebugCodeEnabled (\r
  VOID\r
  )\r
{\r
  return (BOOLEAN)((PcdGet8 (PcdReportStatusCodePropertyMask) & REPORT_STATUS_CODE_PROPERTY_DEBUG_CODE_ENABLED) != 0);\r
}\r