Support Variable driver (VariableAuthenticatedPei/VariableAuthenticatedRuntimeDxe...
authorniruiyu <niruiyu@6f19259b-4bc3-4df7-8a09-765794883524>
Wed, 19 Oct 2011 12:40:52 +0000 (12:40 +0000)
committerniruiyu <niruiyu@6f19259b-4bc3-4df7-8a09-765794883524>
Wed, 19 Oct 2011 12:40:52 +0000 (12:40 +0000)
Signed-off-by: niruiyu
Reviewed-by: lgao4
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12554 6f19259b-4bc3-4df7-8a09-765794883524

SecurityPkg/SecurityPkg.dsc
SecurityPkg/VariableAuthenticated/Pei/Variable.c
SecurityPkg/VariableAuthenticated/Pei/Variable.h
SecurityPkg/VariableAuthenticated/RuntimeDxe/Variable.c
SecurityPkg/VariableAuthenticated/RuntimeDxe/Variable.h
SecurityPkg/VariableAuthenticated/RuntimeDxe/VariableRuntimeDxe.inf
SecurityPkg/VariableAuthenticated/RuntimeDxe/VariableSmm.inf

index 804da70..0600a5f 100644 (file)
@@ -64,6 +64,7 @@
   ReportStatusCodeLib|MdeModulePkg/Library/DxeReportStatusCodeLib/DxeReportStatusCodeLib.inf\r
 \r
 [LibraryClasses.common.UEFI_DRIVER, LibraryClasses.common.DXE_RUNTIME_DRIVER, LibraryClasses.common.DXE_SAL_DRIVER,]\r
+  HobLib|MdePkg/Library/DxeHobLib/DxeHobLib.inf\r
   MemoryAllocationLib|MdePkg/Library/UefiMemoryAllocationLib/UefiMemoryAllocationLib.inf\r
   DebugLib|MdePkg/Library/UefiDebugLibConOut/UefiDebugLibConOut.inf\r
 \r
@@ -71,6 +72,7 @@
   ExtendedSalLib|MdePkg/Library/DxeRuntimeExtendedSalLib/DxeRuntimeExtendedSalLib.inf\r
 \r
 [LibraryClasses.common.DXE_SMM_DRIVER]\r
+  HobLib|MdePkg/Library/DxeHobLib/DxeHobLib.inf\r
   SmmServicesTableLib|MdePkg/Library/SmmServicesTableLib/SmmServicesTableLib.inf\r
   MemoryAllocationLib|MdePkg/Library/SmmMemoryAllocationLib/SmmMemoryAllocationLib.inf\r
   BaseCryptLib|CryptoPkg/Library/BaseCryptLib/SmmCryptLib.inf\r
index f637675..240bc8a 100644 (file)
@@ -31,33 +31,6 @@ EFI_PEI_PPI_DESCRIPTOR     mPpiListVariable = {
   &mVariablePpi\r
 };\r
 \r
-\r
-/**\r
-  Check if it runs in Recovery mode.\r
-  \r
-  @param  PeiServices  General purpose services available to every PEIM.\r
-\r
-  @retval TRUE         It's in Recovery mode.\r
-  @retval FALSE        It's not in Recovery mode.\r
-\r
-**/\r
-BOOLEAN\r
-IsInRecoveryMode (\r
-  IN CONST EFI_PEI_SERVICES          **PeiServices\r
-  )\r
-{\r
-  EFI_STATUS              Status;\r
-  EFI_BOOT_MODE           BootMode;\r
-\r
-  Status = (*PeiServices)->GetBootMode (PeiServices, &BootMode);\r
-  ASSERT_EFI_ERROR (Status);\r
-  \r
-  if (BootMode == BOOT_IN_RECOVERY_MODE) {\r
-    return TRUE;\r
-  }\r
-  return FALSE;\r
-}\r
-\r
 /**\r
   Provide the functionality of the variable services.\r
   \r
@@ -67,7 +40,6 @@ IsInRecoveryMode (
 \r
   @retval EFI_SUCCESS  If the interface could be successfully installed\r
   @retval Others       Returned from PeiServicesInstallPpi()\r
-\r
 **/\r
 EFI_STATUS\r
 EFIAPI\r
@@ -348,14 +320,84 @@ CompareWithValidVariable (
   return EFI_NOT_FOUND;\r
 }\r
 \r
+/**\r
+  Return the variable store header and the index table based on the Index.\r
+\r
+  @param Index      The index of the variable store.\r
+  @param IndexTable Return the index table.\r
+\r
+  @return  Pointer to the variable store header.\r
+**/\r
+VARIABLE_STORE_HEADER *\r
+GetVariableStore (\r
+  IN VARIABLE_STORE_TYPE         Type,\r
+  OUT VARIABLE_INDEX_TABLE       **IndexTable  OPTIONAL\r
+  )\r
+{\r
+  EFI_HOB_GUID_TYPE           *GuidHob;\r
+  EFI_FIRMWARE_VOLUME_HEADER  *FvHeader;\r
+  VARIABLE_STORE_HEADER       *VariableStoreHeader;\r
+\r
+  if (IndexTable != NULL) {\r
+    *IndexTable       = NULL;\r
+  }\r
+  VariableStoreHeader = NULL;\r
+  switch (Type) {\r
+    case VariableStoreTypeHob:\r
+      GuidHob     = GetFirstGuidHob (&gEfiAuthenticatedVariableGuid);\r
+      if (GuidHob != NULL) {\r
+        VariableStoreHeader = (VARIABLE_STORE_HEADER *) GET_GUID_HOB_DATA (GuidHob);\r
+      }\r
+      break;\r
+\r
+    case VariableStoreTypeNv:\r
+      if (GetBootModeHob () != BOOT_IN_RECOVERY_MODE) {\r
+        //\r
+        // The content of NV storage for variable is not reliable in recovery boot mode.\r
+        //\r
+        FvHeader = (EFI_FIRMWARE_VOLUME_HEADER *) (UINTN) (PcdGet64 (PcdFlashNvStorageVariableBase64) != 0 ? \r
+                                                           PcdGet64 (PcdFlashNvStorageVariableBase64) : \r
+                                                           PcdGet32 (PcdFlashNvStorageVariableBase)\r
+                                                          );\r
+        VariableStoreHeader = (VARIABLE_STORE_HEADER *) ((UINT8 *) FvHeader + FvHeader->HeaderLength);\r
+\r
+        if (IndexTable != NULL) {\r
+          GuidHob = GetFirstGuidHob (&gEfiVariableIndexTableGuid);\r
+          if (GuidHob != NULL) {\r
+            *IndexTable = GET_GUID_HOB_DATA (GuidHob);\r
+          } else {\r
+            //\r
+            // If it's the first time to access variable region in flash, create a guid hob to record\r
+            // VAR_ADDED type variable info.\r
+            // Note that as the resource of PEI phase is limited, only store the limited number of \r
+            // VAR_ADDED type variables to reduce access time.\r
+            //\r
+            *IndexTable = BuildGuidHob (&gEfiVariableIndexTableGuid, sizeof (VARIABLE_INDEX_TABLE));\r
+            (*IndexTable)->Length      = 0;\r
+            (*IndexTable)->StartPtr    = GetStartPointer (VariableStoreHeader);\r
+            (*IndexTable)->EndPtr      = GetEndPointer   (VariableStoreHeader);\r
+            (*IndexTable)->GoneThrough = 0;\r
+          }\r
+        }\r
+      }\r
+      break;\r
+\r
+    default:\r
+      ASSERT (FALSE);\r
+      break;\r
+  }\r
+\r
+  return VariableStoreHeader;\r
+}\r
 \r
 /**\r
-  This code finds variable in storage blocks (Non-Volatile).\r
+  Find the variable in the specified variable store.\r
 \r
-  @param  PeiServices   General purpose services available to every PEIM.\r
-  @param  VariableName  Name of the variable to be found\r
-  @param  VendorGuid    Vendor GUID to be found.\r
-  @param  PtrTrack      Variable Track Pointer structure that contains Variable Information.\r
+  @param  VariableStoreHeader Pointer to the variable store header.\r
+  @param  IndexTable          Pointer to the index table.\r
+  @param  VariableName        Name of the variable to be found\r
+  @param  VendorGuid          Vendor GUID to be found.\r
+  @param  PtrTrack            Variable Track Pointer structure that contains Variable Information.\r
 \r
   @retval  EFI_SUCCESS            Variable found successfully\r
   @retval  EFI_NOT_FOUND          Variable not found\r
@@ -363,134 +405,100 @@ CompareWithValidVariable (
 \r
 **/\r
 EFI_STATUS\r
-FindVariable (\r
-  IN CONST EFI_PEI_SERVICES   **PeiServices,\r
-  IN CONST  CHAR16            *VariableName,\r
-  IN CONST  EFI_GUID          *VendorGuid,\r
-  OUT VARIABLE_POINTER_TRACK  *PtrTrack\r
+FindVariableEx (\r
+  IN VARIABLE_STORE_HEADER       *VariableStoreHeader,\r
+  IN VARIABLE_INDEX_TABLE        *IndexTable,\r
+  IN CONST CHAR16                *VariableName,\r
+  IN CONST EFI_GUID              *VendorGuid,\r
+  OUT VARIABLE_POINTER_TRACK     *PtrTrack\r
   )\r
 {\r
-  EFI_HOB_GUID_TYPE       *GuidHob;\r
-  VARIABLE_STORE_HEADER   *VariableStoreHeader;\r
   VARIABLE_HEADER         *Variable;\r
   VARIABLE_HEADER         *LastVariable;\r
   VARIABLE_HEADER         *MaxIndex;\r
-  VARIABLE_INDEX_TABLE    *IndexTable;\r
-  UINT32                  Count;\r
-  UINT32                  Offset;\r
-  UINT8                   *VariableBase;\r
+  UINTN                   Index;\r
+  UINTN                   Offset;\r
   BOOLEAN                 StopRecord;\r
 \r
-  if (VariableName[0] != 0 && VendorGuid == NULL) {\r
+  if (VariableStoreHeader == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
+\r
+  if (GetVariableStoreStatus (VariableStoreHeader) != EfiValid) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  if (~VariableStoreHeader->Size == 0) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  PtrTrack->StartPtr = GetStartPointer (VariableStoreHeader);\r
+  PtrTrack->EndPtr   = GetEndPointer   (VariableStoreHeader);\r
+\r
   //\r
   // No Variable Address equals zero, so 0 as initial value is safe.\r
   //\r
-  MaxIndex = 0;\r
-  StopRecord = FALSE;\r
+  MaxIndex   = NULL;\r
 \r
-  GuidHob = GetFirstGuidHob (&gEfiVariableIndexTableGuid);\r
-  if (GuidHob == NULL) {\r
+  if (IndexTable != NULL) {\r
     //\r
-    // If it's the first time to access variable region in flash, create a guid hob to record\r
-    // VAR_ADDED type variable info.\r
-    // Note that as the resource of PEI phase is limited, only store the number of \r
-    // VARIABLE_INDEX_TABLE_VOLUME of VAR_ADDED type variables to reduce access time.\r
+    // traverse the variable index table to look for varible.\r
+    // The IndexTable->Index[Index] records the distance of two neighbouring VAR_ADDED type variables.\r
     //\r
-    IndexTable = BuildGuidHob (&gEfiVariableIndexTableGuid, sizeof (VARIABLE_INDEX_TABLE));\r
-    IndexTable->Length      = 0;\r
-    IndexTable->StartPtr    = NULL;\r
-    IndexTable->EndPtr      = NULL;\r
-    IndexTable->GoneThrough = 0;\r
-  } else {\r
-    IndexTable = GET_GUID_HOB_DATA (GuidHob);\r
-    for (Offset = 0, Count = 0; Count < IndexTable->Length; Count++) {\r
-      //\r
-      // traverse the variable info list to look for varible.\r
-      // The IndexTable->Index[Count] records the distance of two neighbouring VAR_ADDED type variables.\r
-      //\r
-      ASSERT (Count < VARIABLE_INDEX_TABLE_VOLUME);\r
-      Offset   += IndexTable->Index[Count];\r
-      MaxIndex  = (VARIABLE_HEADER *)((CHAR8 *)(IndexTable->StartPtr) + Offset);\r
+    for (Offset = 0, Index = 0; Index < IndexTable->Length; Index++) {\r
+      ASSERT (Index < sizeof (IndexTable->Index) / sizeof (IndexTable->Index[0]));\r
+      Offset   += IndexTable->Index[Index];\r
+      MaxIndex  = (VARIABLE_HEADER *) ((UINT8 *) IndexTable->StartPtr + Offset);\r
       if (CompareWithValidVariable (MaxIndex, VariableName, VendorGuid, PtrTrack) == EFI_SUCCESS) {\r
-        PtrTrack->StartPtr  = IndexTable->StartPtr;\r
-        PtrTrack->EndPtr    = IndexTable->EndPtr;\r
-\r
         return EFI_SUCCESS;\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
+      //\r
       return EFI_NOT_FOUND;\r
     }\r
   }\r
-  //\r
-  // If not found in HOB, then let's start from the MaxIndex we've found.\r
-  //\r
+\r
   if (MaxIndex != NULL) {\r
+    //\r
+    // HOB exists but the variable cannot be found in HOB\r
+    // If not found in HOB, then let's start from the MaxIndex we've found.\r
+    //\r
     Variable     = GetNextVariablePtr (MaxIndex);\r
     LastVariable = MaxIndex;\r
   } else {\r
-    if ((IndexTable->StartPtr != NULL) || (IndexTable->EndPtr != NULL)) {\r
-      Variable = IndexTable->StartPtr;\r
-    } else {\r
-      VariableBase = (UINT8 *) (UINTN) PcdGet64 (PcdFlashNvStorageVariableBase64);\r
-      if (VariableBase == NULL) {\r
-        VariableBase = (UINT8 *) (UINTN) PcdGet32 (PcdFlashNvStorageVariableBase);\r
-      }\r
-      \r
-      VariableStoreHeader = (VARIABLE_STORE_HEADER *) (VariableBase + \\r
-                            ((EFI_FIRMWARE_VOLUME_HEADER *) (VariableBase)) -> HeaderLength);\r
-\r
-      if (GetVariableStoreStatus (VariableStoreHeader) != EfiValid) {\r
-        return EFI_UNSUPPORTED;\r
-      }\r
-\r
-      if (~VariableStoreHeader->Size == 0) {\r
-        return EFI_NOT_FOUND;\r
-      }\r
-      //\r
-      // Find the variable by walk through non-volatile variable store\r
-      //\r
-      IndexTable->StartPtr  = GetStartPointer (VariableStoreHeader);\r
-      IndexTable->EndPtr    = GetEndPointer (VariableStoreHeader);\r
-\r
-      //\r
-      // Start Pointers for the variable.\r
-      // Actual Data Pointer where data can be written.\r
-      //\r
-      Variable = IndexTable->StartPtr;\r
-    }\r
-\r
-    LastVariable = IndexTable->StartPtr;\r
+    //\r
+    // Start Pointers for the variable.\r
+    // Actual Data Pointer where data can be written.\r
+    //\r
+    Variable     = PtrTrack->StartPtr;\r
+    LastVariable = PtrTrack->StartPtr;\r
   }\r
+\r
   //\r
   // Find the variable by walk through non-volatile variable store\r
   //\r
-  PtrTrack->StartPtr  = IndexTable->StartPtr;\r
-  PtrTrack->EndPtr    = IndexTable->EndPtr;\r
-\r
-  while ((Variable < IndexTable->EndPtr) && IsValidVariableHeader (Variable)) {\r
+  StopRecord = FALSE;\r
+  while ((Variable < PtrTrack->EndPtr) && IsValidVariableHeader (Variable)) {\r
     if (Variable->State == VAR_ADDED) {\r
       //\r
       // Record Variable in VariableIndex HOB\r
       //\r
-      if (IndexTable->Length < VARIABLE_INDEX_TABLE_VOLUME && !StopRecord) {\r
-        Offset = (UINT32)((UINTN)Variable - (UINTN)LastVariable);\r
-        //\r
-        // The distance of two neighbouring VAR_ADDED variable is larger than 2^16, \r
-        // which is beyond the allowable scope(UINT16) of record. In such case, need not to\r
-        // record the subsequent VAR_ADDED type variables again.\r
-        //\r
-        if ((Offset & 0xFFFF0000UL) != 0) {\r
+      if ((IndexTable != NULL) && !StopRecord) {\r
+        Offset = (UINTN) Variable - (UINTN) LastVariable;\r
+        if ((Offset > 0x0FFFF) || (IndexTable->Length == sizeof (IndexTable->Index) / sizeof (IndexTable->Index[0]))) {\r
+          //\r
+          // Stop to record if the distance of two neighbouring VAR_ADDED variable is larger than the allowable scope(UINT16),\r
+          // or the record buffer is full.\r
+          //\r
           StopRecord = TRUE;\r
-        }\r
-\r
-        if (!StopRecord) {\r
+        } else {\r
           IndexTable->Index[IndexTable->Length++] = (UINT16) Offset;\r
+          LastVariable = Variable;\r
         }\r
-        LastVariable = Variable;\r
       }\r
 \r
       if (CompareWithValidVariable (Variable, VariableName, VendorGuid, PtrTrack) == EFI_SUCCESS) {\r
@@ -503,7 +511,7 @@ FindVariable (
   //\r
   // If gone through the VariableStore, that means we never find in Firmware any more.\r
   //\r
-  if ((IndexTable->Length < VARIABLE_INDEX_TABLE_VOLUME) && (!StopRecord)) {\r
+  if ((IndexTable != NULL) && !StopRecord) {\r
     IndexTable->GoneThrough = 1;\r
   }\r
 \r
@@ -512,6 +520,50 @@ FindVariable (
   return EFI_NOT_FOUND;\r
 }\r
 \r
+/**\r
+  Find the variable in HOB and Non-Volatile variable storages.\r
+\r
+  @param  VariableName  Name of the variable to be found\r
+  @param  VendorGuid    Vendor GUID to be found.\r
+  @param  PtrTrack      Variable Track Pointer structure that contains Variable Information.\r
+\r
+  @retval  EFI_SUCCESS            Variable found successfully\r
+  @retval  EFI_NOT_FOUND          Variable not found\r
+  @retval  EFI_INVALID_PARAMETER  Invalid variable name\r
+**/\r
+EFI_STATUS\r
+FindVariable (\r
+  IN CONST  CHAR16            *VariableName,\r
+  IN CONST  EFI_GUID          *VendorGuid,\r
+  OUT VARIABLE_POINTER_TRACK  *PtrTrack\r
+  )\r
+{\r
+  EFI_STATUS                  Status;\r
+  VARIABLE_STORE_HEADER       *VariableStoreHeader;\r
+  VARIABLE_INDEX_TABLE        *IndexTable;\r
+  VARIABLE_STORE_TYPE         Type;\r
+\r
+  if (VariableName[0] != 0 && VendorGuid == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  for (Type = (VARIABLE_STORE_TYPE) 0; Type < VariableStoreTypeMax; Type++) {\r
+    VariableStoreHeader = GetVariableStore (Type, &IndexTable);\r
+    Status = FindVariableEx (\r
+               VariableStoreHeader,\r
+               IndexTable,\r
+               VariableName,\r
+               VendorGuid, \r
+               PtrTrack\r
+               );\r
+    if (!EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
+  }\r
+\r
+  return EFI_NOT_FOUND;\r
+}\r
+\r
 /**\r
   This service retrieves a variable's value using its name and GUID.\r
 \r
@@ -552,27 +604,16 @@ PeiGetVariable (
   VARIABLE_POINTER_TRACK  Variable;\r
   UINTN                   VarDataSize;\r
   EFI_STATUS              Status;\r
-  CONST EFI_PEI_SERVICES  **PeiServices;\r
 \r
-  PeiServices = GetPeiServicesTablePointer ();\r
   if (VariableName == NULL || VariableGuid == NULL || DataSize == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  //\r
-  // Check if this is recovery boot path.\r
-  // If yes, the content of variable area is not reliable. Therefore we directly\r
-  // return EFI_NOT_FOUND. \r
-  // \r
-  if (IsInRecoveryMode(PeiServices)) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
   //\r
   // Find existing variable\r
   //\r
-  Status = FindVariable (PeiServices, VariableName, VariableGuid, &Variable);\r
-  if (Variable.CurrPtr == NULL || Status != EFI_SUCCESS) {\r
+  Status = FindVariable (VariableName, VariableGuid, &Variable);\r
+  if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
   //\r
@@ -636,26 +677,18 @@ PeiGetNextVariableName (
   IN OUT EFI_GUID                           *VariableGuid\r
   )\r
 {\r
+  VARIABLE_STORE_TYPE     Type;\r
   VARIABLE_POINTER_TRACK  Variable;\r
+  VARIABLE_POINTER_TRACK  VariableInHob;\r
   UINTN                   VarNameSize;\r
   EFI_STATUS              Status;\r
-  CONST EFI_PEI_SERVICES  **PeiServices;\r
+  VARIABLE_STORE_HEADER   *VariableStoreHeader[VariableStoreTypeMax];\r
 \r
-  PeiServices = GetPeiServicesTablePointer ();\r
   if (VariableName == NULL || VariableGuid == NULL || VariableNameSize == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  //\r
-  // Check if this is recovery boot path.\r
-  // If yes, the content of variable area is not reliable. Therefore we directly\r
-  // return EFI_NOT_FOUND. \r
-  //   \r
-  if (IsInRecoveryMode(PeiServices)) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
-  Status = FindVariable (PeiServices, VariableName, VariableGuid, &Variable);\r
+  Status = FindVariable (VariableName, VariableGuid, &Variable);\r
   if (Variable.CurrPtr == NULL || Status != EFI_SUCCESS) {\r
     return Status;\r
   }\r
@@ -667,34 +700,88 @@ PeiGetNextVariableName (
     Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
   }\r
 \r
-  while (!(Variable.CurrPtr >= Variable.EndPtr || Variable.CurrPtr == NULL)) {\r
-    if (IsValidVariableHeader (Variable.CurrPtr)) {\r
-      if (Variable.CurrPtr->State == VAR_ADDED) {\r
-        ASSERT (NameSizeOfVariable (Variable.CurrPtr) != 0);\r
+  VariableStoreHeader[VariableStoreTypeHob] = GetVariableStore (VariableStoreTypeHob, NULL);\r
+  VariableStoreHeader[VariableStoreTypeNv]  = GetVariableStore (VariableStoreTypeNv, NULL);\r
 \r
-        VarNameSize = (UINTN) NameSizeOfVariable (Variable.CurrPtr);\r
-        if (VarNameSize <= *VariableNameSize) {\r
-          CopyMem (VariableName, GetVariableNamePtr (Variable.CurrPtr), VarNameSize);\r
+  while (TRUE) {\r
+    //\r
+    // Switch from HOB to Non-Volatile.\r
+    //\r
+    while ((Variable.CurrPtr >= Variable.EndPtr) ||\r
+           (Variable.CurrPtr == NULL)            ||\r
+           !IsValidVariableHeader (Variable.CurrPtr)\r
+          ) {\r
+      //\r
+      // Find current storage index\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
+      //\r
+      // Switch to next storage\r
+      //\r
+      for (Type++; Type < VariableStoreTypeMax; Type++) {\r
+        if (VariableStoreHeader[Type] != NULL) {\r
+          break;\r
+        }\r
+      }\r
+      //\r
+      // Capture the case that \r
+      // 1. current storage is the last one, or\r
+      // 2. no further storage\r
+      //\r
+      if (Type == VariableStoreTypeMax) {\r
+        return EFI_NOT_FOUND;\r
+      }\r
+      Variable.StartPtr = GetStartPointer (VariableStoreHeader[Type]);\r
+      Variable.EndPtr   = GetEndPointer   (VariableStoreHeader[Type]);\r
+      Variable.CurrPtr  = Variable.StartPtr;\r
+    }\r
 \r
-          CopyMem (VariableGuid, &Variable.CurrPtr->VendorGuid, sizeof (EFI_GUID));\r
+    if (Variable.CurrPtr->State == VAR_ADDED) {\r
 \r
-          Status = EFI_SUCCESS;\r
-        } else {\r
-          Status = EFI_BUFFER_TOO_SMALL;\r
+      //\r
+      // Don't return NV variable when HOB overrides it\r
+      //\r
+      if ((VariableStoreHeader[VariableStoreTypeHob] != NULL) && (VariableStoreHeader[VariableStoreTypeNv] != NULL) &&\r
+          (Variable.StartPtr == GetStartPointer (VariableStoreHeader[VariableStoreTypeNv]))\r
+         ) {\r
+        Status = FindVariableEx (\r
+                   VariableStoreHeader[VariableStoreTypeHob],\r
+                   NULL,\r
+                   GetVariableNamePtr (Variable.CurrPtr),\r
+                   &Variable.CurrPtr->VendorGuid, \r
+                   &VariableInHob\r
+                   );\r
+        if (!EFI_ERROR (Status)) {\r
+          Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
+          continue;\r
         }\r
+      }\r
 \r
-        *VariableNameSize = VarNameSize;\r
-        return Status;\r
-        //\r
-        // Variable is found\r
-        //\r
+      VarNameSize = NameSizeOfVariable (Variable.CurrPtr);\r
+      ASSERT (VarNameSize != 0);\r
+\r
+      if (VarNameSize <= *VariableNameSize) {\r
+        CopyMem (VariableName, GetVariableNamePtr (Variable.CurrPtr), VarNameSize);\r
+\r
+        CopyMem (VariableGuid, &Variable.CurrPtr->VendorGuid, sizeof (EFI_GUID));\r
+\r
+        Status = EFI_SUCCESS;\r
       } else {\r
-        Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
+        Status = EFI_BUFFER_TOO_SMALL;\r
       }\r
+\r
+      *VariableNameSize = VarNameSize;\r
+      //\r
+      // Variable is found\r
+      //\r
+      return Status;\r
     } else {\r
-      break;\r
+      Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
     }\r
   }\r
-\r
-  return EFI_NOT_FOUND;\r
 }\r
index af22687..75d32da 100644 (file)
@@ -29,6 +29,13 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 #include <Guid/AuthenticatedVariableFormat.h>\r
 #include <Guid/VariableIndexTable.h>\r
+\r
+typedef enum {\r
+  VariableStoreTypeHob,\r
+  VariableStoreTypeNv,\r
+  VariableStoreTypeMax\r
+} VARIABLE_STORE_TYPE;\r
+\r
 //\r
 // Functions\r
 //\r
index 0846b2a..32bddbc 100644 (file)
@@ -715,6 +715,67 @@ Reclaim (
   return Status;\r
 }\r
 \r
+/**\r
+  Find the variable in the specified variable store.\r
+\r
+  @param  VariableName        Name of the variable to be found\r
+  @param  VendorGuid          Vendor GUID to be found.\r
+  @param  PtrTrack            Variable Track Pointer structure that contains Variable Information.\r
+\r
+  @retval  EFI_SUCCESS            Variable found successfully\r
+  @retval  EFI_NOT_FOUND          Variable not found\r
+**/\r
+EFI_STATUS\r
+FindVariableEx (\r
+  IN     CHAR16                  *VariableName,\r
+  IN     EFI_GUID                *VendorGuid,\r
+  IN OUT VARIABLE_POINTER_TRACK  *PtrTrack\r
+  )\r
+{\r
+  VARIABLE_HEADER                *InDeletedVariable;\r
+  VOID                           *Point;\r
+\r
+  //\r
+  // Find the variable by walk through HOB, volatile and non-volatile variable store.\r
+  //\r
+  InDeletedVariable  = NULL;\r
+\r
+  for ( PtrTrack->CurrPtr = PtrTrack->StartPtr\r
+      ; (PtrTrack->CurrPtr < PtrTrack->EndPtr) && IsValidVariableHeader (PtrTrack->CurrPtr)\r
+      ; PtrTrack->CurrPtr = GetNextVariablePtr (PtrTrack->CurrPtr)\r
+      ) {\r
+    if (PtrTrack->CurrPtr->State == VAR_ADDED || \r
+        PtrTrack->CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)\r
+       ) {\r
+      if (!AtRuntime () || ((PtrTrack->CurrPtr->Attributes & EFI_VARIABLE_RUNTIME_ACCESS) != 0)) {\r
+        if (VariableName[0] == 0) {\r
+          if (PtrTrack->CurrPtr->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+            InDeletedVariable   = PtrTrack->CurrPtr;\r
+          } else {\r
+            return EFI_SUCCESS;\r
+          }\r
+        } else {\r
+          if (CompareGuid (VendorGuid, &PtrTrack->CurrPtr->VendorGuid)) {\r
+            Point = (VOID *) GetVariableNamePtr (PtrTrack->CurrPtr);\r
+\r
+            ASSERT (NameSizeOfVariable (PtrTrack->CurrPtr) != 0);\r
+            if (CompareMem (VariableName, Point, NameSizeOfVariable (PtrTrack->CurrPtr)) == 0) {\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
+      }\r
+    }\r
+  }\r
+\r
+  PtrTrack->CurrPtr = InDeletedVariable;\r
+  return (PtrTrack->CurrPtr  == NULL) ? EFI_NOT_FOUND : EFI_SUCCESS;\r
+}\r
+\r
 \r
 /**\r
   Finds variable in storage blocks of volatile and non-volatile storage areas.\r
@@ -746,89 +807,40 @@ FindVariable (
   IN  VARIABLE_GLOBAL         *Global\r
   )\r
 {\r
-  VARIABLE_HEADER         *Variable[2];\r
-  VARIABLE_HEADER         *InDeletedVariable;\r
-  VARIABLE_STORE_HEADER   *VariableStoreHeader[2];\r
-  UINTN                   InDeletedStorageIndex;\r
-  UINTN                   Index;\r
-  VOID                    *Point;\r
+  EFI_STATUS              Status;\r
+  VARIABLE_STORE_HEADER   *VariableStoreHeader[VariableStoreTypeMax];\r
+  VARIABLE_STORE_TYPE     Type;\r
+\r
+  if (VariableName[0] != 0 && VendorGuid == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
 \r
   //\r
-  // 0: Volatile, 1: Non-Volatile.\r
+  // 0: Volatile, 1: HOB, 2: Non-Volatile.\r
   // The index and attributes mapping must be kept in this order as RuntimeServiceGetNextVariableName\r
   // make use of this mapping to implement search algorithm.\r
   //\r
-  VariableStoreHeader[0]  = (VARIABLE_STORE_HEADER *) ((UINTN) mVariableModuleGlobal->VariableGlobal.VolatileVariableBase);\r
-  VariableStoreHeader[1]  = mNvVariableCache;\r
+  VariableStoreHeader[VariableStoreTypeVolatile] = (VARIABLE_STORE_HEADER *) (UINTN) Global->VolatileVariableBase;\r
+  VariableStoreHeader[VariableStoreTypeHob]      = (VARIABLE_STORE_HEADER *) (UINTN) Global->HobVariableBase;\r
+  VariableStoreHeader[VariableStoreTypeNv]       = mNvVariableCache;\r
 \r
   //\r
-  // Start Pointers for the variable.\r
-  // Actual Data Pointer where data can be written.\r
+  // Find the variable by walk through HOB, volatile and non-volatile variable store.\r
   //\r
-  Variable[0] = GetStartPointer (VariableStoreHeader[0]);\r
-  Variable[1] = GetStartPointer (VariableStoreHeader[1]);\r
-\r
-  if (VariableName[0] != 0 && VendorGuid == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
+  for (Type = (VARIABLE_STORE_TYPE) 0; Type < VariableStoreTypeMax; Type++) {\r
+    if (VariableStoreHeader[Type] == NULL) {\r
+      continue;\r
+    }\r
 \r
-  //\r
-  // Find the variable by walk through volatile and then non-volatile variable store.\r
-  //\r
-  InDeletedVariable     = NULL;\r
-  InDeletedStorageIndex = 0;\r
-  for (Index = 0; Index < 2; Index++) {\r
-    while ((Variable[Index] < GetEndPointer (VariableStoreHeader[Index])) && IsValidVariableHeader (Variable[Index])) {\r
-      if (Variable[Index]->State == VAR_ADDED || \r
-          Variable[Index]->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)\r
-         ) {\r
-        if (!AtRuntime () || ((Variable[Index]->Attributes & EFI_VARIABLE_RUNTIME_ACCESS) != 0)) {\r
-          if (VariableName[0] == 0) {\r
-            if (Variable[Index]->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
-              InDeletedVariable     = Variable[Index];\r
-              InDeletedStorageIndex = Index;\r
-            } else {\r
-              PtrTrack->StartPtr  = GetStartPointer (VariableStoreHeader[Index]);\r
-              PtrTrack->EndPtr    = GetEndPointer (VariableStoreHeader[Index]);\r
-              PtrTrack->CurrPtr   = Variable[Index];\r
-              PtrTrack->Volatile  = (BOOLEAN)(Index == 0);\r
-\r
-              return EFI_SUCCESS;\r
-            }\r
-          } else {\r
-            if (CompareGuid (VendorGuid, &Variable[Index]->VendorGuid)) {\r
-              Point = (VOID *) GetVariableNamePtr (Variable[Index]);\r
-\r
-              ASSERT (NameSizeOfVariable (Variable[Index]) != 0);\r
-              if (CompareMem (VariableName, Point, NameSizeOfVariable (Variable[Index])) == 0) {\r
-                if (Variable[Index]->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
-                  InDeletedVariable     = Variable[Index];\r
-                  InDeletedStorageIndex = Index;\r
-                } else {\r
-                  PtrTrack->StartPtr  = GetStartPointer (VariableStoreHeader[Index]);\r
-                  PtrTrack->EndPtr    = GetEndPointer (VariableStoreHeader[Index]);\r
-                  PtrTrack->CurrPtr   = Variable[Index];\r
-                  PtrTrack->Volatile  = (BOOLEAN)(Index == 0);\r
-\r
-                  return EFI_SUCCESS;\r
-                }\r
-              }\r
-            }\r
-          }\r
-        }\r
-      }\r
+    PtrTrack->StartPtr = GetStartPointer (VariableStoreHeader[Type]);\r
+    PtrTrack->EndPtr   = GetEndPointer   (VariableStoreHeader[Type]);\r
+    PtrTrack->Volatile = (BOOLEAN) (Type == VariableStoreTypeVolatile);\r
 \r
-      Variable[Index] = GetNextVariablePtr (Variable[Index]);\r
-    }\r
-    if (InDeletedVariable != NULL) {\r
-      PtrTrack->StartPtr  = GetStartPointer (VariableStoreHeader[InDeletedStorageIndex]);\r
-      PtrTrack->EndPtr    = GetEndPointer (VariableStoreHeader[InDeletedStorageIndex]);\r
-      PtrTrack->CurrPtr   = InDeletedVariable;\r
-      PtrTrack->Volatile  = (BOOLEAN)(InDeletedStorageIndex == 0);\r
-      return EFI_SUCCESS;\r
+    Status = FindVariableEx (VariableName, VendorGuid, PtrTrack);\r
+    if (!EFI_ERROR (Status)) {\r
+      return Status;\r
     }\r
   }\r
-  PtrTrack->CurrPtr = NULL;\r
   return EFI_NOT_FOUND;\r
 }\r
 \r
@@ -1226,7 +1238,7 @@ AutoUpdateLangVariable(
     // Update Lang if PlatformLang is already set\r
     // Update PlatformLang if Lang is already set\r
     //\r
-    Status = FindVariable (L"PlatformLang", &gEfiGlobalVariableGuid, &Variable, (VARIABLE_GLOBAL *) mVariableModuleGlobal);\r
+    Status = FindVariable (L"PlatformLang", &gEfiGlobalVariableGuid, &Variable, &mVariableModuleGlobal->VariableGlobal);\r
     if (!EFI_ERROR (Status)) {\r
       //\r
       // Update Lang\r
@@ -1235,7 +1247,7 @@ AutoUpdateLangVariable(
       Data         = GetVariableDataPtr (Variable.CurrPtr);\r
       DataSize     = Variable.CurrPtr->DataSize;\r
     } else {\r
-      Status = FindVariable (L"Lang", &gEfiGlobalVariableGuid, &Variable, (VARIABLE_GLOBAL *) mVariableModuleGlobal);\r
+      Status = FindVariable (L"Lang", &gEfiGlobalVariableGuid, &Variable, &mVariableModuleGlobal->VariableGlobal);\r
       if (!EFI_ERROR (Status)) {\r
         //\r
         // Update PlatformLang\r
@@ -1280,7 +1292,7 @@ AutoUpdateLangVariable(
         //\r
         // Successfully convert PlatformLang to Lang, and set the BestLang value into Lang variable simultaneously.\r
         //\r
-        FindVariable (L"Lang", &gEfiGlobalVariableGuid, &Variable, (VARIABLE_GLOBAL *)mVariableModuleGlobal);\r
+        FindVariable (L"Lang", &gEfiGlobalVariableGuid, &Variable, &mVariableModuleGlobal->VariableGlobal);\r
 \r
         Status = UpdateVariable (L"Lang", &gEfiGlobalVariableGuid, BestLang,\r
                                  ISO_639_2_ENTRY_SIZE + 1, Attributes, 0, 0, &Variable, NULL);\r
@@ -1314,7 +1326,7 @@ AutoUpdateLangVariable(
         //\r
         // Successfully convert Lang to PlatformLang, and set the BestPlatformLang value into PlatformLang variable simultaneously.\r
         //\r
-        FindVariable (L"PlatformLang", &gEfiGlobalVariableGuid, &Variable, (VARIABLE_GLOBAL *)mVariableModuleGlobal);\r
+        FindVariable (L"PlatformLang", &gEfiGlobalVariableGuid, &Variable, &mVariableModuleGlobal->VariableGlobal);\r
 \r
         Status = UpdateVariable (L"PlatformLang", &gEfiGlobalVariableGuid, BestPlatformLang, \r
                                  AsciiStrSize (BestPlatformLang), Attributes, 0, 0, &Variable, NULL);\r
@@ -1935,9 +1947,12 @@ VariableServiceGetNextVariableName (
   IN OUT  EFI_GUID          *VendorGuid\r
   )\r
 {\r
+  VARIABLE_STORE_TYPE     Type;\r
   VARIABLE_POINTER_TRACK  Variable;\r
+  VARIABLE_POINTER_TRACK  VariableInHob;\r
   UINTN                   VarNameSize;\r
   EFI_STATUS              Status;\r
+  VARIABLE_STORE_HEADER   *VariableStoreHeader[VariableStoreTypeMax];\r
 \r
   if (VariableNameSize == NULL || VariableName == NULL || VendorGuid == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -1957,45 +1972,85 @@ VariableServiceGetNextVariableName (
     Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
   }\r
 \r
+  //\r
+  // 0: Volatile, 1: HOB, 2: Non-Volatile.\r
+  // The index and attributes mapping must be kept in this order as FindVariable\r
+  // makes use of this mapping to implement search algorithm.\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
   while (TRUE) {\r
     //\r
-    // If both volatile and non-volatile variable store are parsed,\r
-    // return not found.\r
+    // Switch from Volatile to HOB, to Non-Volatile.\r
     //\r
-    if (Variable.CurrPtr >= Variable.EndPtr || Variable.CurrPtr == NULL) {\r
-      Variable.Volatile = (BOOLEAN) (Variable.Volatile ^ ((BOOLEAN) 0x1));\r
-      if (!Variable.Volatile) {\r
-        Variable.StartPtr = GetStartPointer ((VARIABLE_STORE_HEADER *) (UINTN) mVariableModuleGlobal->VariableGlobal.NonVolatileVariableBase);\r
-        Variable.EndPtr   = GetEndPointer ((VARIABLE_STORE_HEADER *) ((UINTN) mVariableModuleGlobal->VariableGlobal.NonVolatileVariableBase));\r
-      } else {\r
+    while ((Variable.CurrPtr >= Variable.EndPtr) ||\r
+           (Variable.CurrPtr == NULL)            ||\r
+           !IsValidVariableHeader (Variable.CurrPtr)\r
+          ) {\r
+      //\r
+      // Find current storage index\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
+      //\r
+      // Switch to next storage\r
+      //\r
+      for (Type++; Type < VariableStoreTypeMax; Type++) {\r
+        if (VariableStoreHeader[Type] != NULL) {\r
+          break;\r
+        }\r
+      }\r
+      //\r
+      // Capture the case that \r
+      // 1. current storage is the last one, or\r
+      // 2. no further storage\r
+      //\r
+      if (Type == VariableStoreTypeMax) {\r
         Status = EFI_NOT_FOUND;\r
         goto Done;\r
       }\r
-\r
-      Variable.CurrPtr = Variable.StartPtr;\r
-      if (!IsValidVariableHeader (Variable.CurrPtr)) {\r
-        continue;\r
-      }\r
+      Variable.StartPtr = GetStartPointer (VariableStoreHeader[Type]);\r
+      Variable.EndPtr   = GetEndPointer   (VariableStoreHeader[Type]);\r
+      Variable.CurrPtr  = Variable.StartPtr;\r
     }\r
+\r
     //\r
     // Variable is found\r
     //\r
-    if (IsValidVariableHeader (Variable.CurrPtr) && Variable.CurrPtr->State == VAR_ADDED) {\r
+    if (Variable.CurrPtr->State == VAR_ADDED) {\r
       if ((AtRuntime () && ((Variable.CurrPtr->Attributes & EFI_VARIABLE_RUNTIME_ACCESS) == 0)) == 0) {\r
+\r
+        //\r
+        // Don't return NV variable when HOB overrides it\r
+        //\r
+        if ((VariableStoreHeader[VariableStoreTypeHob] != NULL) && (VariableStoreHeader[VariableStoreTypeNv] != NULL) && \r
+            (Variable.StartPtr == GetStartPointer (VariableStoreHeader[VariableStoreTypeNv]))\r
+           ) {\r
+          VariableInHob.StartPtr = GetStartPointer (VariableStoreHeader[VariableStoreTypeHob]);\r
+          VariableInHob.EndPtr   = GetEndPointer   (VariableStoreHeader[VariableStoreTypeHob]);\r
+          Status = FindVariableEx (\r
+                     GetVariableNamePtr (Variable.CurrPtr),\r
+                     &Variable.CurrPtr->VendorGuid,\r
+                     &VariableInHob\r
+                     );\r
+          if (!EFI_ERROR (Status)) {\r
+            Variable.CurrPtr = GetNextVariablePtr (Variable.CurrPtr);\r
+            continue;\r
+          }\r
+        }\r
+\r
         VarNameSize = NameSizeOfVariable (Variable.CurrPtr);\r
         ASSERT (VarNameSize != 0);\r
 \r
         if (VarNameSize <= *VariableNameSize) {\r
-          CopyMem (\r
-            VariableName,\r
-            GetVariableNamePtr (Variable.CurrPtr),\r
-            VarNameSize\r
-            );\r
-          CopyMem (\r
-            VendorGuid,\r
-            &Variable.CurrPtr->VendorGuid,\r
-            sizeof (EFI_GUID)\r
-            );\r
+          CopyMem (VariableName, GetVariableNamePtr (Variable.CurrPtr), VarNameSize);\r
+          CopyMem (VendorGuid, &Variable.CurrPtr->VendorGuid, sizeof (EFI_GUID));\r
           Status = EFI_SUCCESS;\r
         } else {\r
           Status = EFI_BUFFER_TOO_SMALL;\r
@@ -2376,16 +2431,16 @@ VariableWriteServiceInitialize (
   UINTN                           Index;\r
   UINT8                           Data;\r
   EFI_PHYSICAL_ADDRESS            VariableStoreBase;\r
-  UINT64                          VariableStoreLength;\r
+  VARIABLE_HEADER                 *Variable;\r
+  VOID                            *VariableData;\r
 \r
   VariableStoreBase   = mVariableModuleGlobal->VariableGlobal.NonVolatileVariableBase;\r
   VariableStoreHeader = (VARIABLE_STORE_HEADER *)(UINTN)VariableStoreBase;\r
-  VariableStoreLength = VariableStoreHeader->Size;\r
-  \r
\r
   //\r
   // Check if the free area is really free.\r
   //\r
-  for (Index = mVariableModuleGlobal->NonVolatileLastVariableOffset; Index < VariableStoreLength; Index++) {\r
+  for (Index = mVariableModuleGlobal->NonVolatileLastVariableOffset; Index < VariableStoreHeader->Size; Index++) {\r
     Data = ((UINT8 *) mNvVariableCache)[Index];\r
     if (Data != 0xff) {\r
       //\r
@@ -2404,6 +2459,35 @@ VariableWriteServiceInitialize (
     }\r
   }\r
 \r
+  \r
+  //\r
+  // Flush the HOB variable to flash and invalidate HOB variable.\r
+  //\r
+  if (mVariableModuleGlobal->VariableGlobal.HobVariableBase != 0) {\r
+    //\r
+    // Clear the HobVariableBase to avoid SetVariable() updating the variable in HOB\r
+    //\r
+    VariableStoreHeader = (VARIABLE_STORE_HEADER *) (UINTN) mVariableModuleGlobal->VariableGlobal.HobVariableBase;\r
+    mVariableModuleGlobal->VariableGlobal.HobVariableBase = 0;\r
+\r
+    for ( Variable = GetStartPointer (VariableStoreHeader)\r
+        ; (Variable < GetEndPointer (VariableStoreHeader) && IsValidVariableHeader (Variable))\r
+        ; Variable = GetNextVariablePtr (Variable)\r
+        ) {\r
+      ASSERT (Variable->State == VAR_ADDED);\r
+      ASSERT ((Variable->Attributes & EFI_VARIABLE_NON_VOLATILE) != 0);\r
+      VariableData = GetVariableDataPtr (Variable);\r
+      Status = VariableServiceSetVariable (\r
+                 GetVariableNamePtr (Variable),\r
+                 &Variable->VendorGuid,\r
+                 Variable->Attributes,\r
+                 Variable->DataSize,\r
+                 VariableData\r
+                 );\r
+      ASSERT_EFI_ERROR (Status);\r
+    }\r
+  }\r
+\r
   //\r
   // Authenticated variable initialize.\r
   //\r
@@ -2434,6 +2518,7 @@ VariableCommonInitialize (
   UINT64                          VariableStoreLength;\r
   UINTN                           ScratchSize;\r
   UINTN                           VariableSize;\r
+  EFI_HOB_GUID_TYPE               *GuidHob;\r
 \r
   //\r
   // Allocate runtime memory for variable driver global structure.\r
@@ -2453,6 +2538,19 @@ VariableCommonInitialize (
   //\r
   ASSERT (PcdGet32 (PcdHwErrStorageSize) <= PcdGet32 (PcdFlashNvStorageVariableSize));\r
 \r
+  //\r
+  // Get HOB variable store.\r
+  //\r
+  GuidHob = GetFirstGuidHob (&gEfiAuthenticatedVariableGuid);\r
+  if (GuidHob != NULL) {\r
+    VariableStoreHeader = GET_GUID_HOB_DATA (GuidHob);\r
+    if (GetVariableStoreStatus (VariableStoreHeader) == EfiValid) {\r
+      mVariableModuleGlobal->VariableGlobal.HobVariableBase = (EFI_PHYSICAL_ADDRESS) (UINTN) VariableStoreHeader;\r
+    } else {\r
+      DEBUG ((EFI_D_ERROR, "HOB Variable Store header is corrupted!\n"));\r
+    }\r
+  }\r
+\r
   //\r
   // Allocate memory for volatile variable store, note that there is a scratch space to store scratch data.\r
   //\r
@@ -2480,7 +2578,7 @@ VariableCommonInitialize (
   VolatileVariableStore->Reserved1   = 0;\r
 \r
   //\r
-  // Get non-volatile varaible store.\r
+  // Get non-volatile variable store.\r
   //\r
 \r
   TempVariableStoreHeader = (EFI_PHYSICAL_ADDRESS) PcdGet64 (PcdFlashNvStorageVariableBase64);\r
index 6865f0d..91c7b4a 100644 (file)
@@ -22,6 +22,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Protocol/FirmwareVolumeBlock.h>\r
 #include <Protocol/Variable.h>\r
 #include <Library/PcdLib.h>\r
+#include <Library/HobLib.h>\r
 #include <Library/UefiDriverEntryPoint.h>\r
 #include <Library/DxeServicesTableLib.h>\r
 #include <Library/UefiRuntimeLib.h>\r
@@ -46,6 +47,13 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 ///\r
 #define ISO_639_2_ENTRY_SIZE    3\r
 \r
+typedef enum {\r
+  VariableStoreTypeVolatile,\r
+  VariableStoreTypeHob,\r
+  VariableStoreTypeNv,\r
+  VariableStoreTypeMax\r
+} VARIABLE_STORE_TYPE;\r
+\r
 typedef struct {\r
   VARIABLE_HEADER *CurrPtr;\r
   VARIABLE_HEADER *EndPtr;\r
@@ -54,6 +62,7 @@ typedef struct {
 } VARIABLE_POINTER_TRACK;\r
 \r
 typedef struct {\r
+  EFI_PHYSICAL_ADDRESS  HobVariableBase;\r
   EFI_PHYSICAL_ADDRESS  VolatileVariableBase;\r
   EFI_PHYSICAL_ADDRESS  NonVolatileVariableBase;\r
   EFI_LOCK              VariableServicesLock;\r
index 7858084..5b2689e 100644 (file)
@@ -54,7 +54,8 @@
   UefiDriverEntryPoint\r
   PcdLib\r
   BaseCryptLib\r
-  PlatformSecureLib  \r
+  PlatformSecureLib\r
+  HobLib\r
 \r
 [Protocols]\r
   gEfiFirmwareVolumeBlockProtocolGuid           ## SOMETIMES_CONSUMES\r
index 63c34e4..01bda72 100644 (file)
@@ -61,6 +61,7 @@
   DxeServicesTableLib\r
   BaseCryptLib\r
   PlatformSecureLib  \r
+  HobLib\r
 \r
 [Protocols]\r
   gEfiSmmFirmwareVolumeBlockProtocolGuid        ## SOMETIMES_CONSUMES\r