Variables with state VAR_ADDED&VAR_IN_DELETED_TRANSITION should be considered as...

[mirror_edk2.git] / MdeModulePkg / Universal / Variable / Pei / Variable.c

diff --git a/MdeModulePkg/Universal/Variable/Pei/Variable.c b/MdeModulePkg/Universal/Variable/Pei/Variable.c
index 5040f8208d2edb98803800dcb03e387c7f9c6e6d..10ad3f51d079b80a207db2ff7ce74c2f02b75a54 100644 (file)
--- a/MdeModulePkg/Universal/Variable/Pei/Variable.c
+++ b/MdeModulePkg/Universal/Variable/Pei/Variable.c
@@ -3,7 +3,7 @@
   Implement ReadOnly Variable Services required by PEIM and install\r
   PEI ReadOnly Varaiable2 PPI. These services operates the non volatile storage space.\r
 \r
-Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2013, Intel Corporation. All rights reserved.<BR>\r
 This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
@@ -430,6 +430,7 @@ FindVariableEx (
   UINTN                   Index;\r
   UINTN                   Offset;\r
   BOOLEAN                 StopRecord;\r
+  VARIABLE_HEADER         *InDeletedVariable;\r
 \r
   if (VariableStoreHeader == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -446,6 +447,8 @@ FindVariableEx (
   PtrTrack->StartPtr = GetStartPointer (VariableStoreHeader);\r
   PtrTrack->EndPtr   = GetEndPointer   (VariableStoreHeader);\r
 \r
+  InDeletedVariable = NULL;\r
+\r
   //\r
   // No Variable Address equals zero, so 0 as initial value is safe.\r
   //\r
@@ -461,15 +464,20 @@ FindVariableEx (
       Offset   += IndexTable->Index[Index];\r
       MaxIndex  = (VARIABLE_HEADER *) ((UINT8 *) IndexTable->StartPtr + Offset);\r
       if (CompareWithValidVariable (MaxIndex, VariableName, VendorGuid, PtrTrack) == EFI_SUCCESS) {\r
-        return EFI_SUCCESS;\r
+        if (PtrTrack->CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+          InDeletedVariable = PtrTrack->CurrPtr;\r
+        } else {\r
+          return EFI_SUCCESS;\r
+        }\r
       }\r
     }\r
 \r
     if (IndexTable->GoneThrough != 0) {\r
       //\r
-      // If the table has all the existing variables indexed and we still cannot find it.\r
+      // If the table has all the existing variables indexed, return.\r
       //\r
-      return EFI_NOT_FOUND;\r
+      PtrTrack->CurrPtr = InDeletedVariable;\r
+      return (PtrTrack->CurrPtr == NULL) ? EFI_NOT_FOUND : EFI_SUCCESS;\r
     }\r
   }\r
 \r
@@ -490,11 +498,11 @@ FindVariableEx (
   }\r
 \r
   //\r
-  // Find the variable by walk through non-volatile variable store\r
+  // Find the variable by walk through variable store\r
   //\r
   StopRecord = FALSE;\r
   while ((Variable < PtrTrack->EndPtr) && IsValidVariableHeader (Variable)) {\r
-    if (Variable->State == VAR_ADDED) {\r
+    if (Variable->State == VAR_ADDED || Variable->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
       //\r
       // Record Variable in VariableIndex HOB\r
       //\r
@@ -513,7 +521,11 @@ FindVariableEx (
       }\r
 \r
       if (CompareWithValidVariable (Variable, VariableName, VendorGuid, PtrTrack) == EFI_SUCCESS) {\r
-        return EFI_SUCCESS;\r
+        if (PtrTrack->CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+          InDeletedVariable = PtrTrack->CurrPtr;\r
+        } else {\r
+          return EFI_SUCCESS;\r
+        }\r
       }\r
     }\r
 \r
@@ -526,9 +538,9 @@ FindVariableEx (
     IndexTable->GoneThrough = 1;\r
   }\r
 \r
-  PtrTrack->CurrPtr = NULL;\r
+  PtrTrack->CurrPtr = InDeletedVariable;\r
 \r
-  return EFI_NOT_FOUND;\r
+  return (PtrTrack->CurrPtr == NULL) ? EFI_NOT_FOUND : EFI_SUCCESS;\r
 }\r
 \r
 /**\r
@@ -691,6 +703,8 @@ PeiGetNextVariableName (
   VARIABLE_STORE_TYPE     Type;\r
   VARIABLE_POINTER_TRACK  Variable;\r
   VARIABLE_POINTER_TRACK  VariableInHob;\r
+  VARIABLE_POINTER_TRACK  VariablePtrTrack;\r
+  VARIABLE_INDEX_TABLE    *IndexTable;\r
   UINTN                   VarNameSize;\r
   EFI_STATUS              Status;\r
   VARIABLE_STORE_HEADER   *VariableStoreHeader[VariableStoreTypeMax];\r
@@ -752,7 +766,32 @@ PeiGetNextVariableName (
       Variable.CurrPtr  = Variable.StartPtr;\r
     }\r
 \r
-    if (Variable.CurrPtr->State == VAR_ADDED) {\r
+    if (Variable.CurrPtr->State == VAR_ADDED || Variable.CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+      if (Variable.CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+        //\r
+        // If it is a IN_DELETED_TRANSITION variable,\r
+        // and there is also a same ADDED one at the same time,\r
+        // don't return it.\r
+        //\r
+        for (Type = (VARIABLE_STORE_TYPE) 0; Type < VariableStoreTypeMax; Type++) {\r
+          if ((VariableStoreHeader[Type] != NULL) && (Variable.StartPtr == GetStartPointer (VariableStoreHeader[Type]))) {\r
+            break;\r
+          }\r
+        }\r
+        ASSERT (Type < VariableStoreTypeMax);\r
+        GetVariableStore (Type, &IndexTable);\r
+        Status = FindVariableEx (\r
+                   VariableStoreHeader[Type],\r
+                   IndexTable,\r
+                   GetVariableNamePtr (Variable.CurrPtr),\r
+                   &Variable.CurrPtr->VendorGuid,\r
+                   &VariablePtrTrack\r
+                   );\r
+        if (!EFI_ERROR (Status) && VariablePtrTrack.CurrPtr->State == VAR_ADDED) {\r
+          Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
+          continue;\r
+        }\r
+      }\r
 \r
       //\r
       // Don't return NV variable when HOB overrides it\r