MdeModulePkg/Variable: Parameterize GetNextVariableInternal () stores

[mirror_edk2.git] / MdeModulePkg / Universal / Variable / RuntimeDxe / VariableExLib.c

diff --git a/MdeModulePkg/Universal/Variable/RuntimeDxe/VariableExLib.c b/MdeModulePkg/Universal/Variable/RuntimeDxe/VariableExLib.c
index dc78f68fa9e05e9580ab043a0f58d866c3428666..c787ddba5bbf7b1216a8f295cb14f9813eae1afd 100644 (file)
--- a/MdeModulePkg/Universal/Variable/RuntimeDxe/VariableExLib.c
+++ b/MdeModulePkg/Universal/Variable/RuntimeDxe/VariableExLib.c
@@ -98,10 +98,16 @@ VariableExLibFindNextVariable (
   EFI_STATUS                    Status;\r
   VARIABLE_HEADER               *VariablePtr;\r
   AUTHENTICATED_VARIABLE_HEADER *AuthVariablePtr;\r
+  VARIABLE_STORE_HEADER         *VariableStoreHeader[VariableStoreTypeMax];\r
+\r
+  VariableStoreHeader[VariableStoreTypeVolatile] = (VARIABLE_STORE_HEADER *) (UINTN) mVariableModuleGlobal->VariableGlobal.VolatileVariableBase;\r
+  VariableStoreHeader[VariableStoreTypeHob]      = (VARIABLE_STORE_HEADER *) (UINTN) mVariableModuleGlobal->VariableGlobal.HobVariableBase;\r
+  VariableStoreHeader[VariableStoreTypeNv]       = mNvVariableCache;\r
 \r
   Status = VariableServiceGetNextVariableInternal (\r
              VariableName,\r
              VendorGuid,\r
+             VariableStoreHeader,\r
              &VariablePtr\r
              );\r
   if (EFI_ERROR (Status)) {\r