X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdePkg%2FLibrary%2FBasePrintLib%2FPrintLibInternal.c;fp=MdePkg%2FLibrary%2FBasePrintLib%2FPrintLibInternal.c;h=e61aed9fda649c9ed279445b22d15c0961b4eb6e;hp=c4fd755eec029a4bd02f717c0bddc44a1992041e;hb=28087ef09f3ade5072ce1505a8acc8aff85787cd;hpb=85e2e0854bd158a561ee829d84635a1ff8128581 diff --git a/MdePkg/Library/BasePrintLib/PrintLibInternal.c b/MdePkg/Library/BasePrintLib/PrintLibInternal.c index c4fd755eec..e61aed9fda 100644 --- a/MdePkg/Library/BasePrintLib/PrintLibInternal.c +++ b/MdePkg/Library/BasePrintLib/PrintLibInternal.c @@ -373,17 +373,21 @@ BasePrintLibSPrintMarker ( } LengthToReturn = 0; + EndBuffer = NULL; + OriginalBuffer = NULL; // // Reserve space for the Null terminator. // - BufferSize--; - OriginalBuffer = Buffer; + if (Buffer != NULL) { + BufferSize--; + OriginalBuffer = Buffer; - // - // Set the tag for the end of the input Buffer. - // - EndBuffer = Buffer + BufferSize * BytesPerOutputCharacter; + // + // Set the tag for the end of the input Buffer. + // + EndBuffer = Buffer + BufferSize * BytesPerOutputCharacter; + } if ((Flags & FORMAT_UNICODE) != 0) { // @@ -411,7 +415,10 @@ BasePrintLibSPrintMarker ( // // Loop until the end of the format string is reached or the output buffer is full // - while (FormatCharacter != 0 && Buffer < EndBuffer) { + while (FormatCharacter != 0) { + if ((Buffer != NULL) && (Buffer >= EndBuffer)) { + break; + } // // Clear all the flag bits except those that may have been passed in //