MdeModulePkg SmmLockBox: Return actual data length in SmmLockBoxRestore

[mirror_edk2.git] / MdeModulePkg / Universal / LockBox / SmmLockBox / SmmLockBox.c

diff --git a/MdeModulePkg/Universal/LockBox/SmmLockBox/SmmLockBox.c b/MdeModulePkg/Universal/LockBox/SmmLockBox/SmmLockBox.c
index 28d5180e011e51dca7614ec0174e3adcb6998bec..5a11743cb921bb2f47e8ba8920a52567b2ee33a8 100644 (file)
--- a/MdeModulePkg/Universal/LockBox/SmmLockBox/SmmLockBox.c
+++ b/MdeModulePkg/Universal/LockBox/SmmLockBox/SmmLockBox.c
@@ -217,7 +217,10 @@ SmmLockBoxRestore (
                (VOID *)(UINTN)TempLockBoxParameterRestore.Buffer,\r
                (UINTN *)&TempLockBoxParameterRestore.Length\r
                );\r
-    if (Status == EFI_BUFFER_TOO_SMALL) {\r
+    if ((Status == EFI_BUFFER_TOO_SMALL) || (Status == EFI_SUCCESS)) {\r
+      //\r
+      // Return the actual Length value.\r
+      //\r
       LockBoxParameterRestore->Length = TempLockBoxParameterRestore.Length;\r
     }\r
   }\r