Fix memory overflow.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7203 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdePkg/Library/BasePrintLib/PrintLibInternal.c b/MdePkg/Library/BasePrintLib/PrintLibInternal.c
index d87513042e2d1385e2be324b7a3791e20ab4ce6d..05e3ed501dccfbfa5779292190efe688e59300a6 100644 (file)
--- a/MdePkg/Library/BasePrintLib/PrintLibInternal.c
+++ b/MdePkg/Library/BasePrintLib/PrintLibInternal.c
@@ -77,11 +77,17 @@ BasePrintLibFillBuffer (
   )\r
 {\r
   INTN  Index;\r
-\r
-  for (Index = 0; Index < Length && Buffer < EndBuffer; Index++) {\r
-    *Buffer       =  (CHAR8) Character;\r
-    *(Buffer + 1) =  (CHAR8) (Character >> 8);\r
-    Buffer        += Increment;\r
+  if(Increment != 1) {\r
+    for (Index = 0; Index < Length && Buffer < EndBuffer; Index++) {\r
+      *Buffer       =  (CHAR8) Character;\r
+      *(Buffer + 1) =  (CHAR8) (Character >> 8);\r
+    }\r
+    Buffer += Increment;\r
+  } else {\r
+    for (Index = 0; Index < Length && Buffer < EndBuffer; Index++) {\r
+      *Buffer       =  (CHAR8) Character;\r
+    }\r
+    Buffer += Increment;\r
   }\r
   return Buffer;\r
 }\r