From d89deb3d7ec0f41af85abe5e0bd7403ecf2809fa Mon Sep 17 00:00:00 2001 From: Qiu Shumin Date: Wed, 1 Jul 2015 08:03:50 +0000 Subject: [PATCH] MdeModulePkg/Application/MemoryProfileInfo: Use safe string functions to refine code. Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Qiu Shumin Reviewed-by: Star Zeng git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@17777 6f19259b-4bc3-4df7-8a09-765794883524 --- .../Application/MemoryProfileInfo/MemoryProfileInfo.c | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/MdeModulePkg/Application/MemoryProfileInfo/MemoryProfileInfo.c b/MdeModulePkg/Application/MemoryProfileInfo/MemoryProfileInfo.c index 6304b7fc6c..b425af17c2 100644 --- a/MdeModulePkg/Application/MemoryProfileInfo/MemoryProfileInfo.c +++ b/MdeModulePkg/Application/MemoryProfileInfo/MemoryProfileInfo.c @@ -121,10 +121,10 @@ GetShortPdbFileName ( UINTN StartIndex; UINTN EndIndex; - ZeroMem (UnicodeBuffer, PROFILE_NAME_STRING_LENGTH * sizeof (CHAR16)); + ZeroMem (UnicodeBuffer, (PROFILE_NAME_STRING_LENGTH + 1) * sizeof (CHAR16)); if (PdbFileName == NULL) { - StrnCpy (UnicodeBuffer, L" ", 1); + StrnCpyS (UnicodeBuffer, PROFILE_NAME_STRING_LENGTH + 1, L" ", 1); } else { StartIndex = 0; for (EndIndex = 0; PdbFileName[EndIndex] != 0; EndIndex++); @@ -201,7 +201,7 @@ GetDriverNameString ( // // Method 2: Get the name string from FFS UI section // - StrnCpy (mNameString, NameString, PROFILE_NAME_STRING_LENGTH); + StrCpyS (mNameString, PROFILE_NAME_STRING_LENGTH + 1, NameString); mNameString[PROFILE_NAME_STRING_LENGTH] = 0; FreePool (NameString); return; -- 2.39.2