REF: https://bugzilla.tianocore.org/show_bug.cgi?id=2302 The second argument of "UnicodeVSPrintAsciiFormat" is "BufferSize", which takes the size of the buffer in bytes. Replace the currently used MAX_DEBUG_MESSAGE_LENGTH usage, which is the buffer's length, with the actual buffer size. Cc: Michael D Kinney <michael.d.kinney@intel.com> Cc: Liming Gao <liming.gao@intel.com> Signed-off-by: Marvin Haeuser <mhaeuser@outlook.de> Reviewed-by: Liming Gao <liming.gao@intel.com>
		
			
				
	
	
		
			376 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			376 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  UEFI Debug Lib that sends messages to the Standard Error Device in the EFI System Table.
 | 
						|
 | 
						|
  Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>
 | 
						|
  SPDX-License-Identifier: BSD-2-Clause-Patent
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
 | 
						|
#include <Uefi.h>
 | 
						|
 | 
						|
#include <Library/DebugLib.h>
 | 
						|
#include <Library/PrintLib.h>
 | 
						|
#include <Library/PcdLib.h>
 | 
						|
#include <Library/BaseLib.h>
 | 
						|
#include <Library/BaseMemoryLib.h>
 | 
						|
#include <Library/DebugPrintErrorLevelLib.h>
 | 
						|
 | 
						|
//
 | 
						|
// Define the maximum debug and assert message length that this library supports
 | 
						|
//
 | 
						|
#define MAX_DEBUG_MESSAGE_LENGTH  0x100
 | 
						|
 | 
						|
 | 
						|
//
 | 
						|
// VA_LIST can not initialize to NULL for all compiler, so we use this to
 | 
						|
// indicate a null VA_LIST
 | 
						|
//
 | 
						|
VA_LIST     mVaListNull;
 | 
						|
 | 
						|
extern BOOLEAN                mPostEBS;
 | 
						|
extern EFI_SYSTEM_TABLE       *mDebugST;
 | 
						|
 | 
						|
/**
 | 
						|
  Prints a debug message to the debug output device if the specified error level is enabled.
 | 
						|
 | 
						|
  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function
 | 
						|
  GetDebugPrintErrorLevel (), then print the message specified by Format and the
 | 
						|
  associated variable argument list to the debug output device.
 | 
						|
 | 
						|
  If Format is NULL, then ASSERT().
 | 
						|
 | 
						|
  @param  ErrorLevel  The error level of the debug message.
 | 
						|
  @param  Format      The format string for the debug message to print.
 | 
						|
  @param  ...         The variable argument list whose contents are accessed
 | 
						|
                      based on the format string specified by Format.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
DebugPrint (
 | 
						|
  IN  UINTN        ErrorLevel,
 | 
						|
  IN  CONST CHAR8  *Format,
 | 
						|
  ...
 | 
						|
  )
 | 
						|
{
 | 
						|
  VA_LIST         Marker;
 | 
						|
 | 
						|
  VA_START (Marker, Format);
 | 
						|
  DebugVPrint (ErrorLevel, Format, Marker);
 | 
						|
  VA_END (Marker);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Prints a debug message to the debug output device if the specified
 | 
						|
  error level is enabled base on Null-terminated format string and a
 | 
						|
  VA_LIST argument list or a BASE_LIST argument list.
 | 
						|
 | 
						|
  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function
 | 
						|
  GetDebugPrintErrorLevel (), then print the message specified by Format and
 | 
						|
  the associated variable argument list to the debug output device.
 | 
						|
 | 
						|
  If Format is NULL, then ASSERT().
 | 
						|
 | 
						|
  @param  ErrorLevel      The error level of the debug message.
 | 
						|
  @param  Format          Format string for the debug message to print.
 | 
						|
  @param  VaListMarker    VA_LIST marker for the variable argument list.
 | 
						|
  @param  BaseListMarker  BASE_LIST marker for the variable argument list.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
DebugPrintMarker (
 | 
						|
  IN  UINTN         ErrorLevel,
 | 
						|
  IN  CONST CHAR8   *Format,
 | 
						|
  IN  VA_LIST       VaListMarker,
 | 
						|
  IN  BASE_LIST     BaseListMarker
 | 
						|
  )
 | 
						|
{
 | 
						|
  CHAR16   Buffer[MAX_DEBUG_MESSAGE_LENGTH];
 | 
						|
 | 
						|
  if (!mPostEBS) {
 | 
						|
    //
 | 
						|
    // If Format is NULL, then ASSERT().
 | 
						|
    //
 | 
						|
    ASSERT (Format != NULL);
 | 
						|
 | 
						|
    //
 | 
						|
    // Check driver debug mask value and global mask
 | 
						|
    //
 | 
						|
    if ((ErrorLevel & GetDebugPrintErrorLevel ()) == 0) {
 | 
						|
      return;
 | 
						|
    }
 | 
						|
 | 
						|
    //
 | 
						|
    // Convert the DEBUG() message to a Unicode String
 | 
						|
    //
 | 
						|
    if (BaseListMarker == NULL) {
 | 
						|
      UnicodeVSPrintAsciiFormat (Buffer, sizeof (Buffer), Format, VaListMarker);
 | 
						|
    } else {
 | 
						|
      UnicodeBSPrintAsciiFormat (Buffer, sizeof (Buffer), Format, BaseListMarker);
 | 
						|
    }
 | 
						|
 | 
						|
    //
 | 
						|
    // Send the print string to the Standard Error device
 | 
						|
    //
 | 
						|
    if ((mDebugST != NULL) && (mDebugST->StdErr != NULL)) {
 | 
						|
      mDebugST->StdErr->OutputString (mDebugST->StdErr, Buffer);
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Prints a debug message to the debug output device if the specified
 | 
						|
  error level is enabled.
 | 
						|
 | 
						|
  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function
 | 
						|
  GetDebugPrintErrorLevel (), then print the message specified by Format and
 | 
						|
  the associated variable argument list to the debug output device.
 | 
						|
 | 
						|
  If Format is NULL, then ASSERT().
 | 
						|
 | 
						|
  @param  ErrorLevel    The error level of the debug message.
 | 
						|
  @param  Format        Format string for the debug message to print.
 | 
						|
  @param  VaListMarker  VA_LIST marker for the variable argument list.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
DebugVPrint (
 | 
						|
  IN  UINTN         ErrorLevel,
 | 
						|
  IN  CONST CHAR8   *Format,
 | 
						|
  IN  VA_LIST       VaListMarker
 | 
						|
  )
 | 
						|
{
 | 
						|
  DebugPrintMarker (ErrorLevel, Format, VaListMarker, NULL);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Prints a debug message to the debug output device if the specified
 | 
						|
  error level is enabled.
 | 
						|
  This function use BASE_LIST which would provide a more compatible
 | 
						|
  service than VA_LIST.
 | 
						|
 | 
						|
  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function
 | 
						|
  GetDebugPrintErrorLevel (), then print the message specified by Format and
 | 
						|
  the associated variable argument list to the debug output device.
 | 
						|
 | 
						|
  If Format is NULL, then ASSERT().
 | 
						|
 | 
						|
  @param  ErrorLevel      The error level of the debug message.
 | 
						|
  @param  Format          Format string for the debug message to print.
 | 
						|
  @param  BaseListMarker  BASE_LIST marker for the variable argument list.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
DebugBPrint (
 | 
						|
  IN  UINTN         ErrorLevel,
 | 
						|
  IN  CONST CHAR8   *Format,
 | 
						|
  IN  BASE_LIST     BaseListMarker
 | 
						|
  )
 | 
						|
{
 | 
						|
  DebugPrintMarker (ErrorLevel, Format, mVaListNull, BaseListMarker);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Prints an assert message containing a filename, line number, and description.
 | 
						|
  This may be followed by a breakpoint or a dead loop.
 | 
						|
 | 
						|
  Print a message of the form "ASSERT <FileName>(<LineNumber>): <Description>\n"
 | 
						|
  to the debug output device.  If DEBUG_PROPERTY_ASSERT_BREAKPOINT_ENABLED bit of
 | 
						|
  PcdDebugProperyMask is set then CpuBreakpoint() is called. Otherwise, if
 | 
						|
  DEBUG_PROPERTY_ASSERT_DEADLOOP_ENABLED bit of PcdDebugProperyMask is set then
 | 
						|
  CpuDeadLoop() is called.  If neither of these bits are set, then this function
 | 
						|
  returns immediately after the message is printed to the debug output device.
 | 
						|
  DebugAssert() must actively prevent recursion.  If DebugAssert() is called while
 | 
						|
  processing another DebugAssert(), then DebugAssert() must return immediately.
 | 
						|
 | 
						|
  If FileName is NULL, then a <FileName> string of "(NULL) Filename" is printed.
 | 
						|
  If Description is NULL, then a <Description> string of "(NULL) Description" is printed.
 | 
						|
 | 
						|
  @param  FileName     The pointer to the name of the source file that generated
 | 
						|
                       the assert condition.
 | 
						|
  @param  LineNumber   The line number in the source file that generated the
 | 
						|
                       assert condition
 | 
						|
  @param  Description  The pointer to the description of the assert condition.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
DebugAssert (
 | 
						|
  IN CONST CHAR8  *FileName,
 | 
						|
  IN UINTN        LineNumber,
 | 
						|
  IN CONST CHAR8  *Description
 | 
						|
  )
 | 
						|
{
 | 
						|
  CHAR16  Buffer[MAX_DEBUG_MESSAGE_LENGTH];
 | 
						|
 | 
						|
  if (!mPostEBS) {
 | 
						|
    //
 | 
						|
    // Generate the ASSERT() message in Unicode format
 | 
						|
    //
 | 
						|
    UnicodeSPrintAsciiFormat (
 | 
						|
      Buffer,
 | 
						|
      sizeof (Buffer),
 | 
						|
      "ASSERT [%a] %a(%d): %a\n",
 | 
						|
      gEfiCallerBaseName,
 | 
						|
      FileName,
 | 
						|
      LineNumber,
 | 
						|
      Description
 | 
						|
      );
 | 
						|
 | 
						|
    //
 | 
						|
    // Send the print string to the Standard Error device
 | 
						|
    //
 | 
						|
    if ((mDebugST != NULL) && (mDebugST->StdErr != NULL)) {
 | 
						|
      mDebugST->StdErr->OutputString (mDebugST->StdErr, Buffer);
 | 
						|
    }
 | 
						|
 | 
						|
    //
 | 
						|
    // Generate a Breakpoint, DeadLoop, or NOP based on PCD settings
 | 
						|
    //
 | 
						|
    if ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_ASSERT_BREAKPOINT_ENABLED) != 0) {
 | 
						|
      CpuBreakpoint ();
 | 
						|
    } else if ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_ASSERT_DEADLOOP_ENABLED) != 0) {
 | 
						|
      CpuDeadLoop ();
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Fills a target buffer with PcdDebugClearMemoryValue, and returns the target buffer.
 | 
						|
 | 
						|
  This function fills Length bytes of Buffer with the value specified by
 | 
						|
  PcdDebugClearMemoryValue, and returns Buffer.
 | 
						|
 | 
						|
  If Buffer is NULL, then ASSERT().
 | 
						|
  If Length is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT().
 | 
						|
 | 
						|
  @param   Buffer  The pointer to the target buffer to be filled with PcdDebugClearMemoryValue.
 | 
						|
  @param   Length  The number of bytes in Buffer to fill with zeros PcdDebugClearMemoryValue.
 | 
						|
 | 
						|
  @return  Buffer  The pointer to the target buffer filled with PcdDebugClearMemoryValue.
 | 
						|
 | 
						|
**/
 | 
						|
VOID *
 | 
						|
EFIAPI
 | 
						|
DebugClearMemory (
 | 
						|
  OUT VOID  *Buffer,
 | 
						|
  IN UINTN  Length
 | 
						|
  )
 | 
						|
{
 | 
						|
  //
 | 
						|
  // If Buffer is NULL, then ASSERT().
 | 
						|
  //
 | 
						|
  ASSERT (Buffer != NULL);
 | 
						|
 | 
						|
  //
 | 
						|
  // SetMem() checks for the the ASSERT() condition on Length and returns Buffer
 | 
						|
  //
 | 
						|
  return SetMem (Buffer, Length, PcdGet8(PcdDebugClearMemoryValue));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Returns TRUE if ASSERT() macros are enabled.
 | 
						|
 | 
						|
  This function returns TRUE if the DEBUG_PROPERTY_DEBUG_ASSERT_ENABLED bit of
 | 
						|
  PcdDebugProperyMask is set.  Otherwise FALSE is returned.
 | 
						|
 | 
						|
  @retval  TRUE    The DEBUG_PROPERTY_DEBUG_ASSERT_ENABLED bit of PcdDebugProperyMask is set.
 | 
						|
  @retval  FALSE   The DEBUG_PROPERTY_DEBUG_ASSERT_ENABLED bit of PcdDebugProperyMask is clear.
 | 
						|
 | 
						|
**/
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DebugAssertEnabled (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  return (BOOLEAN) ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_DEBUG_ASSERT_ENABLED) != 0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Returns TRUE if DEBUG() macros are enabled.
 | 
						|
 | 
						|
  This function returns TRUE if the DEBUG_PROPERTY_DEBUG_PRINT_ENABLED bit of
 | 
						|
  PcdDebugProperyMask is set.  Otherwise FALSE is returned.
 | 
						|
 | 
						|
  @retval  TRUE    The DEBUG_PROPERTY_DEBUG_PRINT_ENABLED bit of PcdDebugProperyMask is set.
 | 
						|
  @retval  FALSE   The DEBUG_PROPERTY_DEBUG_PRINT_ENABLED bit of PcdDebugProperyMask is clear.
 | 
						|
 | 
						|
**/
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DebugPrintEnabled (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  return (BOOLEAN) ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_DEBUG_PRINT_ENABLED) != 0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Returns TRUE if DEBUG_CODE() macros are enabled.
 | 
						|
 | 
						|
  This function returns TRUE if the DEBUG_PROPERTY_DEBUG_CODE_ENABLED bit of
 | 
						|
  PcdDebugProperyMask is set.  Otherwise FALSE is returned.
 | 
						|
 | 
						|
  @retval  TRUE    The DEBUG_PROPERTY_DEBUG_CODE_ENABLED bit of PcdDebugProperyMask is set.
 | 
						|
  @retval  FALSE   The DEBUG_PROPERTY_DEBUG_CODE_ENABLED bit of PcdDebugProperyMask is clear.
 | 
						|
 | 
						|
**/
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DebugCodeEnabled (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  return (BOOLEAN) ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_DEBUG_CODE_ENABLED) != 0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Returns TRUE if DEBUG_CLEAR_MEMORY() macro is enabled.
 | 
						|
 | 
						|
  This function returns TRUE if the DEBUG_PROPERTY_CLEAR_MEMORY_ENABLED bit of
 | 
						|
  PcdDebugProperyMask is set.  Otherwise FALSE is returned.
 | 
						|
 | 
						|
  @retval  TRUE    The DEBUG_PROPERTY_CLEAR_MEMORY_ENABLED bit of PcdDebugProperyMask is set.
 | 
						|
  @retval  FALSE   The DEBUG_PROPERTY_CLEAR_MEMORY_ENABLED bit of PcdDebugProperyMask is clear.
 | 
						|
 | 
						|
**/
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DebugClearMemoryEnabled (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  return (BOOLEAN) ((PcdGet8(PcdDebugPropertyMask) & DEBUG_PROPERTY_CLEAR_MEMORY_ENABLED) != 0);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Returns TRUE if any one of the bit is set both in ErrorLevel and PcdFixedDebugPrintErrorLevel.
 | 
						|
 | 
						|
  This function compares the bit mask of ErrorLevel and PcdFixedDebugPrintErrorLevel.
 | 
						|
 | 
						|
  @retval  TRUE    Current ErrorLevel is supported.
 | 
						|
  @retval  FALSE   Current ErrorLevel is not supported.
 | 
						|
 | 
						|
**/
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DebugPrintLevelEnabled (
 | 
						|
  IN  CONST UINTN        ErrorLevel
 | 
						|
  )
 | 
						|
{
 | 
						|
  return (BOOLEAN) ((ErrorLevel & PcdGet32(PcdFixedDebugPrintErrorLevel)) != 0);
 | 
						|
}
 |