]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Library/DxePrintLibPrint2Protocol/PrintLib.c
MdePkg/BasePrintLib: Refine the SPrint functions
[mirror_edk2.git] / MdeModulePkg / Library / DxePrintLibPrint2Protocol / PrintLib.c
index 3b4d4e03adb26c5c105d5158229dd8f74daab088..01378687fd58e7303c0c42bfc946eace70ed6cd9 100644 (file)
@@ -6,7 +6,7 @@
   protocol related to this implementation, not in the public spec. So, this \r
   library instance is only for this code base.\r
 \r
-Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2009 - 2017, 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
@@ -71,7 +71,8 @@ PrintLibConstructor (
   @param  BaseListMarker  BASE_LIST style variable argument list consumed by processing Format.\r
   @param  Size            The size, in bytes, of the BaseListMarker buffer.\r
 \r
-  @return The number of bytes in BaseListMarker.  0 if BaseListMarker is too small.\r
+  @return TRUE   The VA_LIST has been converted to BASE_LIST.\r
+  @return FALSE  The VA_LIST has not been converted to BASE_LIST.\r
 \r
 **/\r
 BOOLEAN\r
@@ -177,6 +178,7 @@ DxePrintLibPrint2ProtocolVaListToBaseList (
         }\r
       case 'X':\r
       case 'x':\r
+      case 'u':\r
       case 'd':\r
         if (Long) {\r
           BASE_ARG (BaseListMarker, INT64) = VA_ARG (VaListMarker, INT64);\r
@@ -204,6 +206,7 @@ DxePrintLibPrint2ProtocolVaListToBaseList (
     // If BASE_LIST is larger than Size, then return FALSE\r
     //\r
     if ((UINTN)((UINT8 *)BaseListMarker - (UINT8 *)BaseListStart) > Size) {\r
+      DEBUG ((DEBUG_ERROR, "The input variable argument list is too long. Please consider breaking into multiple print calls.\n"));\r
       return FALSE;\r
     }\r
 \r
@@ -263,15 +266,19 @@ UnicodeVSPrint (
   IN  VA_LIST       Marker\r
   )\r
 {\r
-  UINT64  BaseListMarker[256 / sizeof (UINT64)];\r
-\r
-  DxePrintLibPrint2ProtocolVaListToBaseList (\r
-    FALSE, \r
-    (CHAR8 *)FormatString, \r
-    Marker, \r
-    (BASE_LIST)BaseListMarker, \r
-    sizeof (BaseListMarker) - 8\r
-    );\r
+  UINT64   BaseListMarker[256 / sizeof (UINT64)];\r
+  BOOLEAN  Converted;\r
+\r
+  Converted = DxePrintLibPrint2ProtocolVaListToBaseList (\r
+                FALSE,\r
+                (CHAR8 *)FormatString,\r
+                Marker,\r
+                (BASE_LIST)BaseListMarker,\r
+                sizeof (BaseListMarker) - 8\r
+                );\r
+  if (!Converted) {\r
+    return 0;\r
+  }\r
 \r
   return UnicodeBSPrint (StartOfBuffer, BufferSize, FormatString, (BASE_LIST)BaseListMarker);\r
 }\r
@@ -366,9 +373,12 @@ UnicodeSPrint (
   )\r
 {\r
   VA_LIST Marker;\r
+  UINTN   NumberOfPrinted;\r
 \r
   VA_START (Marker, FormatString);\r
-  return UnicodeVSPrint (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  NumberOfPrinted = UnicodeVSPrint (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  VA_END (Marker);\r
+  return NumberOfPrinted;\r
 }\r
 \r
 /**\r
@@ -413,15 +423,19 @@ UnicodeVSPrintAsciiFormat (
   IN  VA_LIST      Marker\r
   )\r
 {\r
-  UINT64  BaseListMarker[256 / sizeof (UINT64)];\r
-\r
-  DxePrintLibPrint2ProtocolVaListToBaseList (\r
-    TRUE, \r
-    FormatString, \r
-    Marker, \r
-    (BASE_LIST)BaseListMarker, \r
-    sizeof (BaseListMarker) - 8\r
-    );\r
+  UINT64   BaseListMarker[256 / sizeof (UINT64)];\r
+  BOOLEAN  Converted;\r
+\r
+  Converted = DxePrintLibPrint2ProtocolVaListToBaseList (\r
+                TRUE,\r
+                FormatString,\r
+                Marker,\r
+                (BASE_LIST)BaseListMarker,\r
+                sizeof (BaseListMarker) - 8\r
+                );\r
+  if (!Converted) {\r
+    return 0;\r
+  }\r
 \r
   return UnicodeBSPrintAsciiFormat (StartOfBuffer, BufferSize, FormatString, (BASE_LIST)BaseListMarker);\r
 }\r
@@ -515,9 +529,12 @@ UnicodeSPrintAsciiFormat (
   )\r
 {\r
   VA_LIST Marker;\r
+  UINTN   NumberOfPrinted;\r
 \r
   VA_START (Marker, FormatString);\r
-  return UnicodeVSPrintAsciiFormat (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  NumberOfPrinted = UnicodeVSPrintAsciiFormat (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  VA_END (Marker);\r
+  return NumberOfPrinted;\r
 }\r
 \r
 /**\r
@@ -614,15 +631,19 @@ AsciiVSPrint (
   IN  VA_LIST       Marker\r
   )\r
 {\r
-  UINT64  BaseListMarker[256 / sizeof (UINT64)];\r
-\r
-  DxePrintLibPrint2ProtocolVaListToBaseList (\r
-    TRUE, \r
-    FormatString, \r
-    Marker, \r
-    (BASE_LIST)BaseListMarker, \r
-    sizeof (BaseListMarker) - 8\r
-    );\r
+  UINT64   BaseListMarker[256 / sizeof (UINT64)];\r
+  BOOLEAN  Converted;\r
+\r
+  Converted = DxePrintLibPrint2ProtocolVaListToBaseList (\r
+                TRUE,\r
+                FormatString,\r
+                Marker,\r
+                (BASE_LIST)BaseListMarker,\r
+                sizeof (BaseListMarker) - 8\r
+                );\r
+  if (!Converted) {\r
+    return 0;\r
+  }\r
 \r
   return AsciiBSPrint (StartOfBuffer, BufferSize, FormatString, (BASE_LIST)BaseListMarker);\r
 }\r
@@ -714,9 +735,12 @@ AsciiSPrint (
   )\r
 {\r
   VA_LIST Marker;\r
+  UINTN   NumberOfPrinted;\r
 \r
   VA_START (Marker, FormatString);\r
-  return AsciiVSPrint (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  NumberOfPrinted = AsciiVSPrint (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  VA_END (Marker);\r
+  return NumberOfPrinted;\r
 }\r
 \r
 /**\r
@@ -761,15 +785,19 @@ AsciiVSPrintUnicodeFormat (
   IN  VA_LIST       Marker\r
   )\r
 {\r
-  UINT64  BaseListMarker[256 / sizeof (UINT64)];\r
-\r
-  DxePrintLibPrint2ProtocolVaListToBaseList (\r
-    FALSE, \r
-    (CHAR8 *)FormatString, \r
-    Marker, \r
-    (BASE_LIST)BaseListMarker, \r
-    sizeof (BaseListMarker) - 8\r
-    );\r
+  UINT64   BaseListMarker[256 / sizeof (UINT64)];\r
+  BOOLEAN  Converted;\r
+\r
+  Converted = DxePrintLibPrint2ProtocolVaListToBaseList (\r
+                FALSE,\r
+                (CHAR8 *)FormatString,\r
+                Marker,\r
+                (BASE_LIST)BaseListMarker,\r
+                sizeof (BaseListMarker) - 8\r
+                );\r
+  if (!Converted) {\r
+    return 0;\r
+  }\r
 \r
   return AsciiBSPrintUnicodeFormat (StartOfBuffer, BufferSize, FormatString, (BASE_LIST)BaseListMarker);\r
 }\r
@@ -863,9 +891,12 @@ AsciiSPrintUnicodeFormat (
   )\r
 {\r
   VA_LIST Marker;\r
+  UINTN   NumberOfPrinted;\r
 \r
   VA_START (Marker, FormatString);\r
-  return AsciiVSPrintUnicodeFormat (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  NumberOfPrinted = AsciiVSPrintUnicodeFormat (StartOfBuffer, BufferSize, FormatString, Marker);\r
+  VA_END (Marker);\r
+  return NumberOfPrinted;\r
 }\r
 \r
 \r
@@ -931,6 +962,7 @@ AsciiValueToString (
 #define PRECISION             BIT11\r
 #define ARGUMENT_REVERSED     BIT12\r
 #define COUNT_ONLY_NO_PRINT   BIT13\r
+#define UNSIGNED_TYPE         BIT14\r
 \r
 //\r
 // Record date and time information\r
@@ -1056,20 +1088,24 @@ InternalPrintLibSPrint (
   )\r
 {\r
   VA_LIST  Marker;\r
+  UINTN    NumberOfPrinted;\r
 \r
   VA_START (Marker, FormatString);\r
-  return InternalPrintLibSPrintMarker (StartOfBuffer, BufferSize, Flags, FormatString, Marker, NULL);\r
+  NumberOfPrinted = InternalPrintLibSPrintMarker (StartOfBuffer, BufferSize, Flags, FormatString, Marker, NULL);\r
+  VA_END (Marker);\r
+  return NumberOfPrinted;\r
 }\r
 \r
-#define WARNING_STATUS_NUMBER         4\r
-#define ERROR_STATUS_NUMBER           24\r
+#define WARNING_STATUS_NUMBER         5\r
+#define ERROR_STATUS_NUMBER           33\r
 \r
-GLOBAL_REMOVE_IF_UNREFERENCED CONST CHAR8 *mStatusString[] = {\r
+GLOBAL_REMOVE_IF_UNREFERENCED CONST CHAR8 * CONST mStatusString[] = {\r
   "Success",                      //  RETURN_SUCCESS                = 0\r
   "Warning Unknown Glyph",        //  RETURN_WARN_UNKNOWN_GLYPH     = 1\r
   "Warning Delete Failure",       //  RETURN_WARN_DELETE_FAILURE    = 2\r
   "Warning Write Failure",        //  RETURN_WARN_WRITE_FAILURE     = 3\r
   "Warning Buffer Too Small",     //  RETURN_WARN_BUFFER_TOO_SMALL  = 4\r
+  "Warning Stale Data",           //  RETURN_WARN_STALE_DATA        = 5\r
   "Load Error",                   //  RETURN_LOAD_ERROR             = 1  | MAX_BIT\r
   "Invalid Parameter",            //  RETURN_INVALID_PARAMETER      = 2  | MAX_BIT\r
   "Unsupported",                  //  RETURN_UNSUPPORTED            = 3  | MAX_BIT\r
@@ -1093,9 +1129,56 @@ GLOBAL_REMOVE_IF_UNREFERENCED CONST CHAR8 *mStatusString[] = {
   "Aborted",                      //  RETURN_ABORTED                = 21 | MAX_BIT\r
   "ICMP Error",                   //  RETURN_ICMP_ERROR             = 22 | MAX_BIT\r
   "TFTP Error",                   //  RETURN_TFTP_ERROR             = 23 | MAX_BIT\r
-  "Protocol Error"                //  RETURN_PROTOCOL_ERROR         = 24 | MAX_BIT\r
+  "Protocol Error",               //  RETURN_PROTOCOL_ERROR         = 24 | MAX_BIT\r
+  "Incompatible Version",         //  RETURN_INCOMPATIBLE_VERSION   = 25 | MAX_BIT\r
+  "Security Violation",           //  RETURN_SECURITY_VIOLATION     = 26 | MAX_BIT\r
+  "CRC Error",                    //  RETURN_CRC_ERROR              = 27 | MAX_BIT\r
+  "End of Media",                 //  RETURN_END_OF_MEDIA           = 28 | MAX_BIT\r
+  "Reserved (29)",                //  RESERVED                      = 29 | MAX_BIT\r
+  "Reserved (30)",                //  RESERVED                      = 30 | MAX_BIT\r
+  "End of File",                  //  RETURN_END_OF_FILE            = 31 | MAX_BIT\r
+  "Invalid Language",             //  RETURN_INVALID_LANGUAGE       = 32 | MAX_BIT\r
+  "Compromised Data"              //  RETURN_COMPROMISED_DATA       = 33 | MAX_BIT\r
 };\r
 \r
+/**\r
+  Internal function that places the character into the Buffer.\r
+\r
+  Internal function that places ASCII or Unicode character into the Buffer.\r
+\r
+  @param  Buffer      The buffer to place the Unicode or ASCII string.\r
+  @param  EndBuffer   The end of the input Buffer. No characters will be\r
+                      placed after that. \r
+  @param  Length      The count of character to be placed into Buffer.\r
+                      (Negative value indicates no buffer fill.)\r
+  @param  Character   The character to be placed into Buffer.\r
+  @param  Increment   The character increment in Buffer.\r
+\r
+  @return Buffer.\r
+\r
+**/\r
+CHAR8 *\r
+InternalPrintLibFillBuffer (\r
+  OUT CHAR8   *Buffer,\r
+  IN  CHAR8   *EndBuffer,\r
+  IN  INTN    Length,\r
+  IN  UINTN   Character,\r
+  IN  INTN    Increment\r
+  )\r
+{\r
+  INTN  Index;\r
+  \r
+  for (Index = 0; Index < Length && Buffer < EndBuffer; Index++) {\r
+    *Buffer = (CHAR8) Character;\r
+    if (Increment != 1) {\r
+      *(Buffer + 1) = (CHAR8)(Character >> 8);\r
+    }\r
+    Buffer += Increment;\r
+  }\r
+\r
+  return Buffer;\r
+}\r
+\r
 /**\r
   Worker function that produces a Null-terminated string in an output buffer \r
   based on a Null-terminated format string and a VA_LIST argument list.\r
@@ -1134,6 +1217,7 @@ InternalPrintLibSPrintMarker (
   IN  BASE_LIST    BaseListMarker  OPTIONAL\r
   )\r
 {\r
+  CHAR8             *OriginalBuffer;\r
   CHAR8             *EndBuffer;\r
   CHAR8             ValueBuffer[MAXIMUM_VALUE_CHARACTERS];\r
   UINT32            BytesPerOutputCharacter;\r
@@ -1170,7 +1254,19 @@ InternalPrintLibSPrintMarker (
   // DxePrintLibPrint2Protocol (both PrintLib instances).\r
   //\r
 \r
-  ASSERT(Flags & COUNT_ONLY_NO_PRINT);\r
+  if ((Flags & COUNT_ONLY_NO_PRINT) != 0) {\r
+    if (BufferSize == 0) {\r
+      Buffer = NULL;\r
+    }\r
+  } else {\r
+    //\r
+    // We can run without a Buffer for counting only.\r
+    //\r
+    if (BufferSize == 0) {\r
+      return 0;\r
+    }\r
+    ASSERT (Buffer != NULL);\r
+  }\r
 \r
   if ((Flags & OUTPUT_UNICODE) != 0) {\r
     BytesPerOutputCharacter = 2;\r
@@ -1179,16 +1275,21 @@ InternalPrintLibSPrintMarker (
   }\r
 \r
   LengthToReturn = 0;\r
+  EndBuffer = NULL;\r
+  OriginalBuffer = NULL;\r
 \r
   //\r
   // Reserve space for the Null terminator.\r
   //\r
-  BufferSize--;\r
+  if (Buffer != NULL) {\r
+    BufferSize--;\r
+    OriginalBuffer = Buffer;\r
 \r
-  //\r
-  // Set the tag for the end of the input Buffer.\r
-  //\r
-  EndBuffer      = Buffer + BufferSize * BytesPerOutputCharacter;\r
+    //\r
+    // Set the tag for the end of the input Buffer.\r
+    //\r
+    EndBuffer = Buffer + BufferSize * BytesPerOutputCharacter;\r
+  }\r
 \r
   if ((Flags & FORMAT_UNICODE) != 0) {\r
     //\r
@@ -1216,11 +1317,14 @@ InternalPrintLibSPrintMarker (
   //\r
   // Loop until the end of the format string is reached or the output buffer is full\r
   //\r
-  while (FormatCharacter != 0 && Buffer < EndBuffer) {\r
+  while (FormatCharacter != 0) {\r
+    if ((Buffer != NULL) && (Buffer >= EndBuffer)) {\r
+      break;\r
+    }\r
     //\r
     // Clear all the flag bits except those that may have been passed in\r
     //\r
-    Flags &= (OUTPUT_UNICODE | FORMAT_UNICODE | COUNT_ONLY_NO_PRINT);\r
+    Flags &= (UINTN) (OUTPUT_UNICODE | FORMAT_UNICODE | COUNT_ONLY_NO_PRINT);\r
 \r
     //\r
     // Set the default width to zero, and the default precision to 1\r
@@ -1328,10 +1432,13 @@ InternalPrintLibSPrintMarker (
         //\r
         // Flag space, +, 0, L & l are invalid for type p.\r
         //\r
-        Flags &= ~(PREFIX_BLANK | PREFIX_SIGN | PREFIX_ZERO | LONG_TYPE);\r
+        Flags &= ~((UINTN) (PREFIX_BLANK | PREFIX_SIGN | PREFIX_ZERO | LONG_TYPE));\r
         if (sizeof (VOID *) > 4) {\r
           Flags |= LONG_TYPE;\r
         }\r
+        //\r
+        // break skipped on purpose\r
+        //\r
       case 'X':\r
         Flags |= PREFIX_ZERO;\r
         //\r
@@ -1342,10 +1449,18 @@ InternalPrintLibSPrintMarker (
         //\r
         // break skipped on purpose\r
         //\r
+      case 'u':\r
+        if ((Flags & RADIX_HEX) == 0) {\r
+          Flags &= ~((UINTN) (PREFIX_SIGN));\r
+          Flags |= UNSIGNED_TYPE;\r
+        }\r
+        //\r
+        // break skipped on purpose\r
+        //\r
       case 'd':\r
         if ((Flags & LONG_TYPE) == 0) {\r
           //\r
-          // 'd','x', and 'X' that are not preceded by 'l' or 'L' are assumed to be type "int".\r
+          // 'd', 'u', 'x', and 'X' that are not preceded by 'l' or 'L' are assumed to be type "int".\r
           // This assumption is made so the format string definition is compatible with the ANSI C\r
           // Specification for formatted strings.  It is recommended that the Base Types be used \r
           // everywhere, but in this one case, compliance with ANSI C is more important, and \r
@@ -1376,20 +1491,30 @@ InternalPrintLibSPrintMarker (
         if ((Flags & RADIX_HEX) == 0) {\r
           Radix = 10;\r
           if (Comma) {\r
-            Flags &= (~PREFIX_ZERO);\r
+            Flags &= ~((UINTN) PREFIX_ZERO);\r
             Precision = 1;\r
           }\r
-          if (Value < 0) {\r
+          if (Value < 0 && (Flags & UNSIGNED_TYPE) == 0) {\r
             Flags |= PREFIX_SIGN;\r
             Prefix = '-';\r
             Value = -Value;\r
+          } else if ((Flags & UNSIGNED_TYPE) != 0 && (Flags & LONG_TYPE) == 0) {\r
+            //\r
+            // 'd', 'u', 'x', and 'X' that are not preceded by 'l' or 'L' are assumed to be type "int".\r
+            // This assumption is made so the format string definition is compatible with the ANSI C\r
+            // Specification for formatted strings.  It is recommended that the Base Types be used \r
+            // everywhere, but in this one case, compliance with ANSI C is more important, and \r
+            // provides an implementation that is compatible with that largest possible set of CPU \r
+            // architectures.  This is why the type "unsigned int" is used in this one case.\r
+            //\r
+            Value = (unsigned int)Value;\r
           }\r
         } else {\r
           Radix = 16;\r
           Comma = FALSE;\r
           if ((Flags & LONG_TYPE) == 0 && Value < 0) {\r
             //\r
-            // 'd','x', and 'X' that are not preceded by 'l' or 'L' are assumed to be type "int".\r
+            // 'd', 'u', 'x', and 'X' that are not preceded by 'l' or 'L' are assumed to be type "int".\r
             // This assumption is made so the format string definition is compatible with the ANSI C\r
             // Specification for formatted strings.  It is recommended that the Base Types be used \r
             // everywhere, but in this one case, compliance with ANSI C is more important, and \r
@@ -1445,7 +1570,7 @@ InternalPrintLibSPrintMarker (
           ArgumentString = BASE_ARG (BaseListMarker, CHAR8 *);\r
         }\r
         if (ArgumentString == NULL) {\r
-          Flags &= (~ARGUMENT_UNICODE);\r
+          Flags &= (~(UINTN)ARGUMENT_UNICODE);\r
           ArgumentString = "<null string>";\r
         }\r
         //\r
@@ -1658,17 +1783,32 @@ InternalPrintLibSPrintMarker (
     //\r
     if ((Flags & (PAD_TO_WIDTH | LEFT_JUSTIFY)) == (PAD_TO_WIDTH)) {\r
       LengthToReturn += ((Width - Precision) * BytesPerOutputCharacter);\r
+      if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+        Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, Width - Precision, ' ', BytesPerOutputCharacter);\r
+      }\r
     }\r
 \r
     if (ZeroPad) {\r
       if (Prefix != 0) {\r
         LengthToReturn += (1 * BytesPerOutputCharacter);\r
+        if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+          Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, 1, Prefix, BytesPerOutputCharacter);\r
+        }\r
       }\r
       LengthToReturn += ((Precision - Count) * BytesPerOutputCharacter);\r
+      if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+        Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, Precision - Count, '0', BytesPerOutputCharacter);\r
+      }\r
     } else {\r
       LengthToReturn += ((Precision - Count) * BytesPerOutputCharacter);\r
+      if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+        Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, Precision - Count, ' ', BytesPerOutputCharacter);\r
+      }\r
       if (Prefix != 0) {\r
         LengthToReturn += (1 * BytesPerOutputCharacter);\r
+        if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+          Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, 1, Prefix, BytesPerOutputCharacter);\r
+        }\r
       }\r
     }\r
 \r
@@ -1687,6 +1827,9 @@ InternalPrintLibSPrintMarker (
       ArgumentCharacter = ((*ArgumentString & 0xff) | (*(ArgumentString + 1) << 8)) & ArgumentMask;\r
 \r
       LengthToReturn += (1 * BytesPerOutputCharacter);\r
+      if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+        Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, 1, ArgumentCharacter, BytesPerOutputCharacter);\r
+      }\r
       ArgumentString    += BytesPerArgumentCharacter;\r
       Index++;\r
       if (Comma) {\r
@@ -1696,6 +1839,9 @@ InternalPrintLibSPrintMarker (
           Index++;\r
           if (Index < Count) {\r
             LengthToReturn += (1 * BytesPerOutputCharacter);\r
+            if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+              Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, 1, ',', BytesPerOutputCharacter);\r
+            }\r
           }\r
         }\r
       }\r
@@ -1706,6 +1852,9 @@ InternalPrintLibSPrintMarker (
     //\r
     if ((Flags & (PAD_TO_WIDTH | LEFT_JUSTIFY)) == (PAD_TO_WIDTH | LEFT_JUSTIFY)) {\r
       LengthToReturn += ((Width - Precision) * BytesPerOutputCharacter);\r
+      if ((Flags & COUNT_ONLY_NO_PRINT) == 0 && Buffer != NULL) {\r
+        Buffer = InternalPrintLibFillBuffer (Buffer, EndBuffer, Width - Precision, ' ', BytesPerOutputCharacter);\r
+      }\r
     }\r
 \r
     //\r
@@ -1719,7 +1868,27 @@ InternalPrintLibSPrintMarker (
     FormatCharacter = ((*Format & 0xff) | (*(Format + 1) << 8)) & FormatMask;\r
   }\r
 \r
-  return (LengthToReturn / BytesPerOutputCharacter);\r
+  if ((Flags & COUNT_ONLY_NO_PRINT) != 0) {\r
+    return (LengthToReturn / BytesPerOutputCharacter);\r
+  }\r
+\r
+  ASSERT (Buffer != NULL);\r
+  //\r
+  // Null terminate the Unicode or ASCII string\r
+  //\r
+  InternalPrintLibFillBuffer (Buffer, EndBuffer + BytesPerOutputCharacter, 1, 0, BytesPerOutputCharacter);\r
+  //\r
+  // Make sure output buffer cannot contain more than PcdMaximumUnicodeStringLength\r
+  // Unicode characters if PcdMaximumUnicodeStringLength is not zero. \r
+  //\r
+  ASSERT ((((Flags & OUTPUT_UNICODE) == 0)) || (StrSize ((CHAR16 *) OriginalBuffer) != 0));\r
+  //\r
+  // Make sure output buffer cannot contain more than PcdMaximumAsciiStringLength\r
+  // ASCII characters if PcdMaximumAsciiStringLength is not zero. \r
+  //\r
+  ASSERT ((((Flags & OUTPUT_UNICODE) != 0)) || (AsciiStrSize (OriginalBuffer) != 0));\r
+\r
+  return ((Buffer - OriginalBuffer) / BytesPerOutputCharacter);\r
 }\r
 \r
 /**\r