ArmPkg: Fixed GetEnvironmentVariable() when the UEFI Variable did not exist

[mirror_edk2.git] / ArmPkg / Library / BdsLib / BdsHelper.c

diff --git a/ArmPkg/Library/BdsLib/BdsHelper.c b/ArmPkg/Library/BdsLib/BdsHelper.c
index fb2ac9b8062ea5df9ef391ad519197664c52f88e..1d4aa3572840d7d3169949549069f9a330eefbdb 100644 (file)
--- a/ArmPkg/Library/BdsLib/BdsHelper.c
+++ b/ArmPkg/Library/BdsLib/BdsHelper.c
@@ -331,7 +331,7 @@ GetEnvironmentVariable (
                     *Size,\r
                     DefaultValue\r
                     );\r
-      *Value = DefaultValue;\r
+      *Value = AllocateCopyPool (*Size, DefaultValue);\r
     } else {\r
       return EFI_NOT_FOUND;\r
     }\r
@@ -352,7 +352,7 @@ GetEnvironmentVariable (
       *Size = VariableSize;\r
     }\r
   } else {\r
-    *Value = DefaultValue;\r
+    *Value = AllocateCopyPool (*Size, DefaultValue);\r
     return Status;\r
   }\r
 \r