PerformancePkg Dp_App: Fix ASSERT in GetNameFromHandle

[mirror_edk2.git] / PerformancePkg / Dp_App / DpUtilities.c

diff --git a/PerformancePkg/Dp_App/DpUtilities.c b/PerformancePkg/Dp_App/DpUtilities.c
index 38d7aa37f1f792311df182f5b2389565e4581e95..ec5a524475795652f838cec1936b2e215eb19265 100644 (file)
--- a/PerformancePkg/Dp_App/DpUtilities.c
+++ b/PerformancePkg/Dp_App/DpUtilities.c
@@ -290,10 +290,11 @@ GetNameFromHandle (
                                );\r
     SafeFreePool (BestLanguage);\r
     if (!EFI_ERROR (Status)) {\r
-      StrCpyS (\r
+      StrnCpyS (\r
         mGaugeString,\r
         DP_GAUGE_STRING_LENGTH + 1,\r
-        StringPtr\r
+        StringPtr,\r
+        DP_GAUGE_STRING_LENGTH\r
         );\r
       return;\r
     }\r
@@ -337,10 +338,11 @@ GetNameFromHandle (
         //\r
         // Method 3. Get the name string from FFS UI section\r
         //\r
-        StrCpyS (\r
+        StrnCpyS (\r
           mGaugeString,\r
           DP_GAUGE_STRING_LENGTH + 1,\r
-          NameString\r
+          NameString,\r
+          DP_GAUGE_STRING_LENGTH\r
           );\r
         FreePool (NameString);\r
       } else {\r
@@ -356,10 +358,11 @@ GetNameFromHandle (
       //\r
       NameString = ConvertDevicePathToText (LoadedImageDevicePath, TRUE, FALSE);\r
       if (NameString != NULL) {\r
-        StrCpyS (\r
+        StrnCpyS (\r
           mGaugeString,\r
           DP_GAUGE_STRING_LENGTH + 1,\r
-          NameString\r
+          NameString,\r
+          DP_GAUGE_STRING_LENGTH\r
           );\r
         FreePool (NameString);\r
         return;\r