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

[mirror_edk2.git] / SecurityPkg / VariableAuthenticated / Pei / Variable.c

diff --git a/SecurityPkg/VariableAuthenticated/Pei/Variable.c b/SecurityPkg/VariableAuthenticated/Pei/Variable.c
index d27f679073827937fb6a57d763edd8903e0fc1d3..38b9170aa245ca0365642c9b491cdbd222aabde5 100644 (file)
--- a/SecurityPkg/VariableAuthenticated/Pei/Variable.c
+++ b/SecurityPkg/VariableAuthenticated/Pei/Variable.c
@@ -3,7 +3,7 @@
   ReadOnly Varaiable2 PPI. These services operates the non-volatile \r
   storage space.\r
 \r
-Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2009 - 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
@@ -428,6 +428,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
@@ -444,6 +445,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
@@ -459,15 +462,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
@@ -488,11 +496,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
@@ -511,7 +519,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
@@ -524,9 +536,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
@@ -689,6 +701,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
@@ -750,7 +764,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