MdeModulePkg/Universal/CapsulePei: Use safe string functions to refine code.

[mirror_edk2.git] / MdeModulePkg / Universal / CapsulePei / UefiCapsule.c

diff --git a/MdeModulePkg/Universal/CapsulePei/UefiCapsule.c b/MdeModulePkg/Universal/CapsulePei/UefiCapsule.c
index 6b934c9e3c91408b39b41d50eb8fecef2a6effd0..3184eb1a6249fa7be8f3f07080c1fb5f9913365e 100644 (file)
--- a/MdeModulePkg/Universal/CapsulePei/UefiCapsule.c
+++ b/MdeModulePkg/Universal/CapsulePei/UefiCapsule.c
@@ -583,7 +583,7 @@ GetCapsuleDescriptors (
               (VOID **) &PPIVariableServices\r
               );\r
   if (Status == EFI_SUCCESS) {\r
-    StrCpy (CapsuleVarName, EFI_CAPSULE_VARIABLE_NAME);\r
+    StrCpyS (CapsuleVarName, sizeof(CapsuleVarName)/sizeof(CHAR16), EFI_CAPSULE_VARIABLE_NAME);\r
     TempVarName = CapsuleVarName + StrLen (CapsuleVarName);\r
     Size = sizeof (CapsuleDataPtr64);\r
     while (1) {\r
@@ -773,7 +773,7 @@ CapsuleCoalesce (
     goto Done;\r
   }\r
   Size = sizeof (CapsuleDataPtr64);\r
-  StrCpy (CapsuleVarName, EFI_CAPSULE_VARIABLE_NAME);\r
+  StrCpyS (CapsuleVarName, sizeof(CapsuleVarName)/sizeof(CHAR16), EFI_CAPSULE_VARIABLE_NAME);\r
   TempVarName = CapsuleVarName + StrLen (CapsuleVarName);\r
   while (TRUE) {\r
     if (Index > 0) {\r