]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Variable/Pei/Variable.c
UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm
[mirror_edk2.git] / MdeModulePkg / Universal / Variable / Pei / Variable.c
index 97c86f2febe71cd41075f3d099c88ec17a13c27e..26a4c73b45a5995dc233e593a83d034e67ffee82 100644 (file)
@@ -1,41 +1,33 @@
 /** @file\r
-\r
   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 - 2014, 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
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) Microsoft Corporation.<BR>\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
-\r
 #include "Variable.h"\r
 \r
 //\r
 // Module globals\r
 //\r
-EFI_PEI_READ_ONLY_VARIABLE2_PPI mVariablePpi = {\r
+EFI_PEI_READ_ONLY_VARIABLE2_PPI  mVariablePpi = {\r
   PeiGetVariable,\r
   PeiGetNextVariableName\r
 };\r
 \r
-EFI_PEI_PPI_DESCRIPTOR     mPpiListVariable = {\r
+EFI_PEI_PPI_DESCRIPTOR  mPpiListVariable = {\r
   (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
   &gEfiPeiReadOnlyVariable2PpiGuid,\r
   &mVariablePpi\r
 };\r
 \r
-\r
 /**\r
   Provide the functionality of the variable services.\r
-  \r
-  @param  FileHandle   Handle of the file being invoked. \r
+\r
+  @param  FileHandle   Handle of the file being invoked.\r
                        Type EFI_PEI_FILE_HANDLE is defined in FfsFindNextFile().\r
   @param  PeiServices  General purpose services available to every PEIM.\r
 \r
@@ -45,8 +37,8 @@ EFI_PEI_PPI_DESCRIPTOR     mPpiListVariable = {
 EFI_STATUS\r
 EFIAPI\r
 PeimInitializeVariableServices (\r
-  IN       EFI_PEI_FILE_HANDLE       FileHandle,\r
-  IN CONST EFI_PEI_SERVICES          **PeiServices\r
+  IN       EFI_PEI_FILE_HANDLE  FileHandle,\r
+  IN CONST EFI_PEI_SERVICES     **PeiServices\r
   )\r
 {\r
   return PeiServicesInstallPpi (&mPpiListVariable);\r
@@ -58,41 +50,43 @@ PeimInitializeVariableServices (
 \r
   @param VarStoreHeader  Pointer to the Variable Store Header.\r
 \r
-  @return Pointer to the first variable header\r
+  @return Pointer to the first variable header.\r
 \r
 **/\r
 VARIABLE_HEADER *\r
 GetStartPointer (\r
-  IN VARIABLE_STORE_HEADER       *VarStoreHeader\r
+  IN VARIABLE_STORE_HEADER  *VarStoreHeader\r
   )\r
 {\r
   //\r
-  // The end of variable store\r
+  // The start of variable store\r
   //\r
-  return (VARIABLE_HEADER *) HEADER_ALIGN (VarStoreHeader + 1);\r
+  return (VARIABLE_HEADER *)HEADER_ALIGN (VarStoreHeader + 1);\r
 }\r
 \r
-\r
 /**\r
-  This code gets the pointer to the last variable memory pointer byte.\r
 \r
-  @param  VarStoreHeader  Pointer to the Variable Store Header.\r
+  Gets the pointer to the end of the variable storage area.\r
+\r
+  This function gets pointer to the end of the variable storage\r
+  area, according to the input variable store header.\r
+\r
+  @param VarStoreHeader  Pointer to the Variable Store Header.\r
 \r
-  @return VARIABLE_HEADER* pointer to last unavailable Variable Header.\r
+  @return Pointer to the end of the variable storage area.\r
 \r
 **/\r
 VARIABLE_HEADER *\r
 GetEndPointer (\r
-  IN VARIABLE_STORE_HEADER       *VarStoreHeader\r
+  IN VARIABLE_STORE_HEADER  *VarStoreHeader\r
   )\r
 {\r
   //\r
   // The end of variable store\r
   //\r
-  return (VARIABLE_HEADER *) HEADER_ALIGN ((UINTN) VarStoreHeader + VarStoreHeader->Size);\r
+  return (VARIABLE_HEADER *)HEADER_ALIGN ((UINTN)VarStoreHeader + VarStoreHeader->Size);\r
 }\r
 \r
-\r
 /**\r
   This code checks if variable header is valid or not.\r
 \r
@@ -104,107 +98,194 @@ GetEndPointer (
 **/\r
 BOOLEAN\r
 IsValidVariableHeader (\r
-  IN  VARIABLE_HEADER   *Variable\r
+  IN  VARIABLE_HEADER  *Variable\r
   )\r
 {\r
-  if (Variable == NULL || Variable->StartId != VARIABLE_DATA ) {\r
+  if ((Variable == NULL) || (Variable->StartId != VARIABLE_DATA)) {\r
     return FALSE;\r
   }\r
 \r
   return TRUE;\r
 }\r
 \r
+/**\r
+  This code gets the size of variable header.\r
+\r
+  @param AuthFlag   Authenticated variable flag.\r
+\r
+  @return Size of variable header in bytes in type UINTN.\r
+\r
+**/\r
+UINTN\r
+GetVariableHeaderSize (\r
+  IN  BOOLEAN  AuthFlag\r
+  )\r
+{\r
+  UINTN  Value;\r
+\r
+  if (AuthFlag) {\r
+    Value = sizeof (AUTHENTICATED_VARIABLE_HEADER);\r
+  } else {\r
+    Value = sizeof (VARIABLE_HEADER);\r
+  }\r
+\r
+  return Value;\r
+}\r
 \r
 /**\r
   This code gets the size of name of variable.\r
 \r
   @param  Variable  Pointer to the Variable Header.\r
+  @param  AuthFlag  Authenticated variable flag.\r
 \r
   @return Size of variable in bytes in type UINTN.\r
 \r
 **/\r
 UINTN\r
 NameSizeOfVariable (\r
-  IN  VARIABLE_HEADER   *Variable\r
+  IN  VARIABLE_HEADER  *Variable,\r
+  IN  BOOLEAN          AuthFlag\r
   )\r
 {\r
-  if (Variable->State    == (UINT8) (-1) ||\r
-      Variable->DataSize == (UINT32) (-1) ||\r
-      Variable->NameSize == (UINT32) (-1) ||\r
-      Variable->Attributes == (UINT32) (-1)) {\r
-    return 0;\r
+  AUTHENTICATED_VARIABLE_HEADER  *AuthVariable;\r
+\r
+  AuthVariable = (AUTHENTICATED_VARIABLE_HEADER *)Variable;\r
+  if (AuthFlag) {\r
+    if ((AuthVariable->State == (UINT8)(-1)) ||\r
+        (AuthVariable->DataSize == (UINT32)(-1)) ||\r
+        (AuthVariable->NameSize == (UINT32)(-1)) ||\r
+        (AuthVariable->Attributes == (UINT32)(-1)))\r
+    {\r
+      return 0;\r
+    }\r
+\r
+    return (UINTN)AuthVariable->NameSize;\r
+  } else {\r
+    if ((Variable->State == (UINT8)(-1)) ||\r
+        (Variable->DataSize == (UINT32)(-1)) ||\r
+        (Variable->NameSize == (UINT32)(-1)) ||\r
+        (Variable->Attributes == (UINT32)(-1)))\r
+    {\r
+      return 0;\r
+    }\r
+\r
+    return (UINTN)Variable->NameSize;\r
   }\r
-  return (UINTN) Variable->NameSize;\r
 }\r
 \r
-\r
 /**\r
   This code gets the size of data of variable.\r
 \r
   @param  Variable  Pointer to the Variable Header.\r
+  @param  AuthFlag  Authenticated variable flag.\r
 \r
   @return Size of variable in bytes in type UINTN.\r
 \r
 **/\r
 UINTN\r
 DataSizeOfVariable (\r
-  IN  VARIABLE_HEADER   *Variable\r
+  IN  VARIABLE_HEADER  *Variable,\r
+  IN  BOOLEAN          AuthFlag\r
   )\r
 {\r
-  if (Variable->State    == (UINT8)  (-1) ||\r
-      Variable->DataSize == (UINT32) (-1) ||\r
-      Variable->NameSize == (UINT32) (-1) ||\r
-      Variable->Attributes == (UINT32) (-1)) {\r
-    return 0;\r
+  AUTHENTICATED_VARIABLE_HEADER  *AuthVariable;\r
+\r
+  AuthVariable = (AUTHENTICATED_VARIABLE_HEADER *)Variable;\r
+  if (AuthFlag) {\r
+    if ((AuthVariable->State == (UINT8)(-1)) ||\r
+        (AuthVariable->DataSize == (UINT32)(-1)) ||\r
+        (AuthVariable->NameSize == (UINT32)(-1)) ||\r
+        (AuthVariable->Attributes == (UINT32)(-1)))\r
+    {\r
+      return 0;\r
+    }\r
+\r
+    return (UINTN)AuthVariable->DataSize;\r
+  } else {\r
+    if ((Variable->State == (UINT8)(-1)) ||\r
+        (Variable->DataSize == (UINT32)(-1)) ||\r
+        (Variable->NameSize == (UINT32)(-1)) ||\r
+        (Variable->Attributes == (UINT32)(-1)))\r
+    {\r
+      return 0;\r
+    }\r
+\r
+    return (UINTN)Variable->DataSize;\r
   }\r
-  return (UINTN) Variable->DataSize;\r
 }\r
 \r
 /**\r
   This code gets the pointer to the variable name.\r
 \r
   @param   Variable  Pointer to the Variable Header.\r
+  @param   AuthFlag  Authenticated variable flag.\r
 \r
   @return  A CHAR16* pointer to Variable Name.\r
 \r
 **/\r
 CHAR16 *\r
 GetVariableNamePtr (\r
-  IN  VARIABLE_HEADER   *Variable\r
+  IN VARIABLE_HEADER  *Variable,\r
+  IN BOOLEAN          AuthFlag\r
   )\r
 {\r
-  return (CHAR16 *) (Variable + 1);\r
+  return (CHAR16 *)((UINTN)Variable + GetVariableHeaderSize (AuthFlag));\r
 }\r
 \r
+/**\r
+  This code gets the pointer to the variable guid.\r
+\r
+  @param Variable   Pointer to the Variable Header.\r
+  @param AuthFlag   Authenticated variable flag.\r
+\r
+  @return A EFI_GUID* pointer to Vendor Guid.\r
+\r
+**/\r
+EFI_GUID *\r
+GetVendorGuidPtr (\r
+  IN VARIABLE_HEADER  *Variable,\r
+  IN BOOLEAN          AuthFlag\r
+  )\r
+{\r
+  AUTHENTICATED_VARIABLE_HEADER  *AuthVariable;\r
+\r
+  AuthVariable = (AUTHENTICATED_VARIABLE_HEADER *)Variable;\r
+  if (AuthFlag) {\r
+    return &AuthVariable->VendorGuid;\r
+  } else {\r
+    return &Variable->VendorGuid;\r
+  }\r
+}\r
 \r
 /**\r
   This code gets the pointer to the variable data.\r
 \r
   @param   Variable         Pointer to the Variable Header.\r
   @param   VariableHeader   Pointer to the Variable Header that has consecutive content.\r
+  @param   AuthFlag         Authenticated variable flag.\r
 \r
   @return  A UINT8* pointer to Variable Data.\r
 \r
 **/\r
 UINT8 *\r
 GetVariableDataPtr (\r
-  IN  VARIABLE_HEADER   *Variable,\r
-  IN  VARIABLE_HEADER   *VariableHeader\r
+  IN  VARIABLE_HEADER  *Variable,\r
+  IN  VARIABLE_HEADER  *VariableHeader,\r
+  IN  BOOLEAN          AuthFlag\r
   )\r
 {\r
-  UINTN Value;\r
-  \r
+  UINTN  Value;\r
+\r
   //\r
   // Be careful about pad size for alignment\r
   //\r
-  Value =  (UINTN) GetVariableNamePtr (Variable);\r
-  Value += NameSizeOfVariable (VariableHeader);\r
-  Value += GET_PAD_SIZE (NameSizeOfVariable (VariableHeader));\r
+  Value  =  (UINTN)GetVariableNamePtr (Variable, AuthFlag);\r
+  Value += NameSizeOfVariable (VariableHeader, AuthFlag);\r
+  Value += GET_PAD_SIZE (NameSizeOfVariable (VariableHeader, AuthFlag));\r
 \r
-  return (UINT8 *) Value;\r
+  return (UINT8 *)Value;\r
 }\r
 \r
-\r
 /**\r
   This code gets the pointer to the next variable header.\r
 \r
@@ -217,18 +298,18 @@ GetVariableDataPtr (
 **/\r
 VARIABLE_HEADER *\r
 GetNextVariablePtr (\r
-  IN  VARIABLE_STORE_INFO   *StoreInfo,\r
-  IN  VARIABLE_HEADER       *Variable,\r
-  IN  VARIABLE_HEADER       *VariableHeader\r
+  IN  VARIABLE_STORE_INFO  *StoreInfo,\r
+  IN  VARIABLE_HEADER      *Variable,\r
+  IN  VARIABLE_HEADER      *VariableHeader\r
   )\r
 {\r
   EFI_PHYSICAL_ADDRESS  TargetAddress;\r
   EFI_PHYSICAL_ADDRESS  SpareAddress;\r
   UINTN                 Value;\r
 \r
-  Value =  (UINTN) GetVariableDataPtr (Variable, VariableHeader);\r
-  Value += DataSizeOfVariable (VariableHeader);\r
-  Value += GET_PAD_SIZE (DataSizeOfVariable (VariableHeader));\r
+  Value  =  (UINTN)GetVariableDataPtr (Variable, VariableHeader, StoreInfo->AuthFlag);\r
+  Value += DataSizeOfVariable (VariableHeader, StoreInfo->AuthFlag);\r
+  Value += GET_PAD_SIZE (DataSizeOfVariable (VariableHeader, StoreInfo->AuthFlag));\r
   //\r
   // Be careful about pad size for alignment\r
   //\r
@@ -236,16 +317,16 @@ GetNextVariablePtr (
 \r
   if (StoreInfo->FtwLastWriteData != NULL) {\r
     TargetAddress = StoreInfo->FtwLastWriteData->TargetAddress;\r
-    SpareAddress = StoreInfo->FtwLastWriteData->SpareAddress;\r
-    if (((UINTN) Variable < (UINTN) TargetAddress) && (Value >= (UINTN) TargetAddress)) {\r
+    SpareAddress  = StoreInfo->FtwLastWriteData->SpareAddress;\r
+    if (((UINTN)Variable < (UINTN)TargetAddress) && (Value >= (UINTN)TargetAddress)) {\r
       //\r
       // Next variable is in spare block.\r
       //\r
-      Value = (UINTN) SpareAddress + (Value - (UINTN) TargetAddress);\r
+      Value = (UINTN)SpareAddress + (Value - (UINTN)TargetAddress);\r
     }\r
   }\r
 \r
-  return (VARIABLE_HEADER *) Value;\r
+  return (VARIABLE_HEADER *)Value;\r
 }\r
 \r
 /**\r
@@ -260,26 +341,27 @@ GetNextVariablePtr (
 **/\r
 VARIABLE_STORE_STATUS\r
 GetVariableStoreStatus (\r
-  IN VARIABLE_STORE_HEADER *VarStoreHeader\r
+  IN VARIABLE_STORE_HEADER  *VarStoreHeader\r
   )\r
 {\r
-  if (CompareGuid (&VarStoreHeader->Signature, &gEfiVariableGuid) &&\r
-      VarStoreHeader->Format == VARIABLE_STORE_FORMATTED &&\r
-      VarStoreHeader->State == VARIABLE_STORE_HEALTHY\r
-      ) {\r
-\r
+  if ((CompareGuid (&VarStoreHeader->Signature, &gEfiAuthenticatedVariableGuid) ||\r
+       CompareGuid (&VarStoreHeader->Signature, &gEfiVariableGuid)) &&\r
+      (VarStoreHeader->Format == VARIABLE_STORE_FORMATTED) &&\r
+      (VarStoreHeader->State == VARIABLE_STORE_HEALTHY)\r
+      )\r
+  {\r
     return EfiValid;\r
   }\r
 \r
-  if (((UINT32 *)(&VarStoreHeader->Signature))[0] == 0xffffffff &&\r
-      ((UINT32 *)(&VarStoreHeader->Signature))[1] == 0xffffffff &&\r
-      ((UINT32 *)(&VarStoreHeader->Signature))[2] == 0xffffffff &&\r
-      ((UINT32 *)(&VarStoreHeader->Signature))[3] == 0xffffffff &&\r
-      VarStoreHeader->Size == 0xffffffff &&\r
-      VarStoreHeader->Format == 0xff &&\r
-      VarStoreHeader->State == 0xff\r
-      ) {\r
-\r
+  if ((((UINT32 *)(&VarStoreHeader->Signature))[0] == 0xffffffff) &&\r
+      (((UINT32 *)(&VarStoreHeader->Signature))[1] == 0xffffffff) &&\r
+      (((UINT32 *)(&VarStoreHeader->Signature))[2] == 0xffffffff) &&\r
+      (((UINT32 *)(&VarStoreHeader->Signature))[3] == 0xffffffff) &&\r
+      (VarStoreHeader->Size == 0xffffffff) &&\r
+      (VarStoreHeader->Format == 0xff) &&\r
+      (VarStoreHeader->State == 0xff)\r
+      )\r
+  {\r
     return EfiRaw;\r
   } else {\r
     return EfiInvalid;\r
@@ -300,10 +382,10 @@ GetVariableStoreStatus (
 **/\r
 BOOLEAN\r
 CompareVariableName (\r
-  IN VARIABLE_STORE_INFO    *StoreInfo,\r
-  IN CONST CHAR16           *Name1,\r
-  IN CONST CHAR16           *Name2,\r
-  IN UINTN                  NameSize\r
+  IN VARIABLE_STORE_INFO  *StoreInfo,\r
+  IN CONST CHAR16         *Name1,\r
+  IN CONST CHAR16         *Name2,\r
+  IN UINTN                NameSize\r
   )\r
 {\r
   EFI_PHYSICAL_ADDRESS  TargetAddress;\r
@@ -312,40 +394,42 @@ CompareVariableName (
 \r
   if (StoreInfo->FtwLastWriteData != NULL) {\r
     TargetAddress = StoreInfo->FtwLastWriteData->TargetAddress;\r
-    SpareAddress = StoreInfo->FtwLastWriteData->SpareAddress;\r
-    if (((UINTN) Name1 < (UINTN) TargetAddress) && (((UINTN) Name1 + NameSize) > (UINTN) TargetAddress)) {\r
+    SpareAddress  = StoreInfo->FtwLastWriteData->SpareAddress;\r
+    if (((UINTN)Name1 < (UINTN)TargetAddress) && (((UINTN)Name1 + NameSize) > (UINTN)TargetAddress)) {\r
       //\r
       // Name1 is inconsecutive.\r
       //\r
-      PartialNameSize = (UINTN) TargetAddress - (UINTN) Name1;\r
+      PartialNameSize = (UINTN)TargetAddress - (UINTN)Name1;\r
       //\r
       // Partial content is in NV storage.\r
       //\r
-      if (CompareMem ((UINT8 *) Name1, (UINT8 *) Name2, PartialNameSize) == 0) {\r
+      if (CompareMem ((UINT8 *)Name1, (UINT8 *)Name2, PartialNameSize) == 0) {\r
         //\r
         // Another partial content is in spare block.\r
         //\r
-        if (CompareMem ((UINT8 *) (UINTN) SpareAddress, (UINT8 *) Name2 + PartialNameSize, NameSize - PartialNameSize) == 0) {\r
+        if (CompareMem ((UINT8 *)(UINTN)SpareAddress, (UINT8 *)Name2 + PartialNameSize, NameSize - PartialNameSize) == 0) {\r
           return TRUE;\r
         }\r
       }\r
+\r
       return FALSE;\r
-    } else if (((UINTN) Name2 < (UINTN) TargetAddress) && (((UINTN) Name2 + NameSize) > (UINTN) TargetAddress)) {\r
+    } else if (((UINTN)Name2 < (UINTN)TargetAddress) && (((UINTN)Name2 + NameSize) > (UINTN)TargetAddress)) {\r
       //\r
       // Name2 is inconsecutive.\r
       //\r
-      PartialNameSize = (UINTN) TargetAddress - (UINTN) Name2;\r
+      PartialNameSize = (UINTN)TargetAddress - (UINTN)Name2;\r
       //\r
       // Partial content is in NV storage.\r
       //\r
-      if (CompareMem ((UINT8 *) Name2, (UINT8 *) Name1, PartialNameSize) == 0) {\r
+      if (CompareMem ((UINT8 *)Name2, (UINT8 *)Name1, PartialNameSize) == 0) {\r
         //\r
         // Another partial content is in spare block.\r
         //\r
-        if (CompareMem ((UINT8 *) (UINTN) SpareAddress, (UINT8 *) Name1 + PartialNameSize, NameSize - PartialNameSize) == 0) {\r
+        if (CompareMem ((UINT8 *)(UINTN)SpareAddress, (UINT8 *)Name1 + PartialNameSize, NameSize - PartialNameSize) == 0) {\r
           return TRUE;\r
         }\r
       }\r
+\r
       return FALSE;\r
     }\r
   }\r
@@ -353,9 +437,10 @@ CompareVariableName (
   //\r
   // Both Name1 and Name2 are consecutive.\r
   //\r
-  if (CompareMem ((UINT8 *) Name1, (UINT8 *) Name2, NameSize) == 0) {\r
+  if (CompareMem ((UINT8 *)Name1, (UINT8 *)Name2, NameSize) == 0) {\r
     return TRUE;\r
   }\r
+\r
   return FALSE;\r
 }\r
 \r
@@ -375,15 +460,18 @@ CompareVariableName (
 **/\r
 EFI_STATUS\r
 CompareWithValidVariable (\r
-  IN  VARIABLE_STORE_INFO           *StoreInfo,\r
-  IN  VARIABLE_HEADER               *Variable,\r
-  IN  VARIABLE_HEADER               *VariableHeader,\r
-  IN  CONST CHAR16                  *VariableName,\r
-  IN  CONST EFI_GUID                *VendorGuid,\r
-  OUT VARIABLE_POINTER_TRACK        *PtrTrack\r
+  IN  VARIABLE_STORE_INFO     *StoreInfo,\r
+  IN  VARIABLE_HEADER         *Variable,\r
+  IN  VARIABLE_HEADER         *VariableHeader,\r
+  IN  CONST CHAR16            *VariableName,\r
+  IN  CONST EFI_GUID          *VendorGuid,\r
+  OUT VARIABLE_POINTER_TRACK  *PtrTrack\r
   )\r
 {\r
-  VOID  *Point;\r
+  VOID      *Point;\r
+  EFI_GUID  *TempVendorGuid;\r
+\r
+  TempVendorGuid = GetVendorGuidPtr (VariableHeader, StoreInfo->AuthFlag);\r
 \r
   if (VariableName[0] == 0) {\r
     PtrTrack->CurrPtr = Variable;\r
@@ -394,14 +482,15 @@ CompareWithValidVariable (
     // Instead we compare the GUID a UINT32 at a time and branch\r
     // on the first failed comparison.\r
     //\r
-    if ((((INT32 *) VendorGuid)[0] == ((INT32 *) &VariableHeader->VendorGuid)[0]) &&\r
-        (((INT32 *) VendorGuid)[1] == ((INT32 *) &VariableHeader->VendorGuid)[1]) &&\r
-        (((INT32 *) VendorGuid)[2] == ((INT32 *) &VariableHeader->VendorGuid)[2]) &&\r
-        (((INT32 *) VendorGuid)[3] == ((INT32 *) &VariableHeader->VendorGuid)[3])\r
-        ) {\r
-      ASSERT (NameSizeOfVariable (VariableHeader) != 0);\r
-      Point = (VOID *) GetVariableNamePtr (Variable);\r
-      if (CompareVariableName (StoreInfo, VariableName, Point, NameSizeOfVariable (VariableHeader))) {\r
+    if ((((INT32 *)VendorGuid)[0] == ((INT32 *)TempVendorGuid)[0]) &&\r
+        (((INT32 *)VendorGuid)[1] == ((INT32 *)TempVendorGuid)[1]) &&\r
+        (((INT32 *)VendorGuid)[2] == ((INT32 *)TempVendorGuid)[2]) &&\r
+        (((INT32 *)VendorGuid)[3] == ((INT32 *)TempVendorGuid)[3])\r
+        )\r
+    {\r
+      ASSERT (NameSizeOfVariable (VariableHeader, StoreInfo->AuthFlag) != 0);\r
+      Point = (VOID *)GetVariableNamePtr (Variable, StoreInfo->AuthFlag);\r
+      if (CompareVariableName (StoreInfo, VariableName, Point, NameSizeOfVariable (VariableHeader, StoreInfo->AuthFlag))) {\r
         PtrTrack->CurrPtr = Variable;\r
         return EFI_SUCCESS;\r
       }\r
@@ -411,6 +500,59 @@ CompareWithValidVariable (
   return EFI_NOT_FOUND;\r
 }\r
 \r
+/**\r
+  Get HOB variable store.\r
+\r
+  @param[out] StoreInfo             Return the store info.\r
+  @param[out] VariableStoreHeader   Return variable store header.\r
+\r
+**/\r
+VOID\r
+GetHobVariableStore (\r
+  OUT VARIABLE_STORE_INFO    *StoreInfo,\r
+  OUT VARIABLE_STORE_HEADER  **VariableStoreHeader\r
+  )\r
+{\r
+  EFI_HOB_GUID_TYPE  *GuidHob;\r
+\r
+  //\r
+  // Make sure there is no more than one Variable HOB.\r
+  //\r
+  DEBUG_CODE_BEGIN ();\r
+  GuidHob = GetFirstGuidHob (&gEfiAuthenticatedVariableGuid);\r
+  if (GuidHob != NULL) {\r
+    if ((GetNextGuidHob (&gEfiAuthenticatedVariableGuid, GET_NEXT_HOB (GuidHob)) != NULL)) {\r
+      DEBUG ((DEBUG_ERROR, "ERROR: Found two Auth Variable HOBs\n"));\r
+      ASSERT (FALSE);\r
+    } else if (GetFirstGuidHob (&gEfiVariableGuid) != NULL) {\r
+      DEBUG ((DEBUG_ERROR, "ERROR: Found one Auth + one Normal Variable HOBs\n"));\r
+      ASSERT (FALSE);\r
+    }\r
+  } else {\r
+    GuidHob = GetFirstGuidHob (&gEfiVariableGuid);\r
+    if (GuidHob != NULL) {\r
+      if ((GetNextGuidHob (&gEfiVariableGuid, GET_NEXT_HOB (GuidHob)) != NULL)) {\r
+        DEBUG ((DEBUG_ERROR, "ERROR: Found two Normal Variable HOBs\n"));\r
+        ASSERT (FALSE);\r
+      }\r
+    }\r
+  }\r
+\r
+  DEBUG_CODE_END ();\r
+\r
+  GuidHob = GetFirstGuidHob (&gEfiAuthenticatedVariableGuid);\r
+  if (GuidHob != NULL) {\r
+    *VariableStoreHeader = (VARIABLE_STORE_HEADER *)GET_GUID_HOB_DATA (GuidHob);\r
+    StoreInfo->AuthFlag  = TRUE;\r
+  } else {\r
+    GuidHob = GetFirstGuidHob (&gEfiVariableGuid);\r
+    if (GuidHob != NULL) {\r
+      *VariableStoreHeader = (VARIABLE_STORE_HEADER *)GET_GUID_HOB_DATA (GuidHob);\r
+      StoreInfo->AuthFlag  = FALSE;\r
+    }\r
+  }\r
+}\r
+\r
 /**\r
   Return the variable store header and the store info based on the Index.\r
 \r
@@ -421,65 +563,70 @@ CompareWithValidVariable (
 **/\r
 VARIABLE_STORE_HEADER *\r
 GetVariableStore (\r
-  IN VARIABLE_STORE_TYPE         Type,\r
-  OUT VARIABLE_STORE_INFO        *StoreInfo\r
+  IN VARIABLE_STORE_TYPE   Type,\r
+  OUT VARIABLE_STORE_INFO  *StoreInfo\r
   )\r
 {\r
+  EFI_STATUS                            Status;\r
   EFI_HOB_GUID_TYPE                     *GuidHob;\r
   EFI_FIRMWARE_VOLUME_HEADER            *FvHeader;\r
   VARIABLE_STORE_HEADER                 *VariableStoreHeader;\r
   EFI_PHYSICAL_ADDRESS                  NvStorageBase;\r
   UINT32                                NvStorageSize;\r
+  UINT64                                NvStorageSize64;\r
   FAULT_TOLERANT_WRITE_LAST_WRITE_DATA  *FtwLastWriteData;\r
   UINT32                                BackUpOffset;\r
 \r
-  StoreInfo->IndexTable = NULL;\r
+  StoreInfo->IndexTable       = NULL;\r
   StoreInfo->FtwLastWriteData = NULL;\r
-  VariableStoreHeader = NULL;\r
+  StoreInfo->AuthFlag         = FALSE;\r
+  VariableStoreHeader         = NULL;\r
   switch (Type) {\r
     case VariableStoreTypeHob:\r
-      GuidHob     = GetFirstGuidHob (&gEfiVariableGuid);\r
-      if (GuidHob != NULL) {\r
-        VariableStoreHeader = (VARIABLE_STORE_HEADER *) GET_GUID_HOB_DATA (GuidHob);\r
-      }\r
+      GetHobVariableStore (StoreInfo, &VariableStoreHeader);\r
+\r
       break;\r
 \r
     case VariableStoreTypeNv:\r
-      if (GetBootModeHob () != BOOT_IN_RECOVERY_MODE) {\r
+      if (!PcdGetBool (PcdEmuVariableNvModeEnable)) {\r
         //\r
-        // The content of NV storage for variable is not reliable in recovery boot mode.\r
+        // Emulated non-volatile variable mode is not enabled.\r
         //\r
 \r
-        NvStorageSize = PcdGet32 (PcdFlashNvStorageVariableSize);\r
-        NvStorageBase = (EFI_PHYSICAL_ADDRESS) (PcdGet64 (PcdFlashNvStorageVariableBase64) != 0 ? \r
-                                                PcdGet64 (PcdFlashNvStorageVariableBase64) : \r
-                                                PcdGet32 (PcdFlashNvStorageVariableBase)\r
-                                               );\r
+        Status = GetVariableFlashNvStorageInfo (&NvStorageBase, &NvStorageSize64);\r
+        ASSERT_EFI_ERROR (Status);\r
+\r
+        Status = SafeUint64ToUint32 (NvStorageSize64, &NvStorageSize);\r
+        // This driver currently assumes the size will be UINT32 so assert the value is safe for now.\r
+        ASSERT_EFI_ERROR (Status);\r
+\r
+        ASSERT (NvStorageBase != 0);\r
+\r
         //\r
         // First let FvHeader point to NV storage base.\r
         //\r
-        FvHeader = (EFI_FIRMWARE_VOLUME_HEADER *) (UINTN) NvStorageBase;\r
+        FvHeader = (EFI_FIRMWARE_VOLUME_HEADER *)(UINTN)NvStorageBase;\r
 \r
         //\r
         // Check the FTW last write data hob.\r
         //\r
         BackUpOffset = 0;\r
-        GuidHob = GetFirstGuidHob (&gEdkiiFaultTolerantWriteGuid);\r
+        GuidHob      = GetFirstGuidHob (&gEdkiiFaultTolerantWriteGuid);\r
         if (GuidHob != NULL) {\r
-          FtwLastWriteData = (FAULT_TOLERANT_WRITE_LAST_WRITE_DATA *) GET_GUID_HOB_DATA (GuidHob);\r
+          FtwLastWriteData = (FAULT_TOLERANT_WRITE_LAST_WRITE_DATA *)GET_GUID_HOB_DATA (GuidHob);\r
           if (FtwLastWriteData->TargetAddress == NvStorageBase) {\r
             //\r
             // Let FvHeader point to spare block.\r
             //\r
-            FvHeader = (EFI_FIRMWARE_VOLUME_HEADER *) (UINTN) FtwLastWriteData->SpareAddress;\r
-            DEBUG ((EFI_D_INFO, "PeiVariable: NV storage is backed up in spare block: 0x%x\n", (UINTN) FtwLastWriteData->SpareAddress));\r
+            FvHeader = (EFI_FIRMWARE_VOLUME_HEADER *)(UINTN)FtwLastWriteData->SpareAddress;\r
+            DEBUG ((DEBUG_INFO, "PeiVariable: NV storage is backed up in spare block: 0x%x\n", (UINTN)FtwLastWriteData->SpareAddress));\r
           } else if ((FtwLastWriteData->TargetAddress > NvStorageBase) && (FtwLastWriteData->TargetAddress < (NvStorageBase + NvStorageSize))) {\r
             StoreInfo->FtwLastWriteData = FtwLastWriteData;\r
             //\r
             // Flash NV storage from the offset is backed up in spare block.\r
             //\r
-            BackUpOffset = (UINT32) (FtwLastWriteData->TargetAddress - NvStorageBase);\r
-            DEBUG ((EFI_D_INFO, "PeiVariable: High partial NV storage from offset: %x is backed up in spare block: 0x%x\n", BackUpOffset, (UINTN) FtwLastWriteData->SpareAddress));\r
+            BackUpOffset = (UINT32)(FtwLastWriteData->TargetAddress - NvStorageBase);\r
+            DEBUG ((DEBUG_INFO, "PeiVariable: High partial NV storage from offset: %x is backed up in spare block: 0x%x\n", BackUpOffset, (UINTN)FtwLastWriteData->SpareAddress));\r
             //\r
             // At least one block data in flash NV storage is still valid, so still leave FvHeader point to NV storage base.\r
             //\r
@@ -490,11 +637,13 @@ GetVariableStore (
         // Check if the Firmware Volume is not corrupted\r
         //\r
         if ((FvHeader->Signature != EFI_FVH_SIGNATURE) || (!CompareGuid (&gEfiSystemNvDataFvGuid, &FvHeader->FileSystemGuid))) {\r
-          DEBUG ((EFI_D_ERROR, "Firmware Volume for Variable Store is corrupted\n"));\r
+          DEBUG ((DEBUG_ERROR, "Firmware Volume for Variable Store is corrupted\n"));\r
           break;\r
         }\r
 \r
-        VariableStoreHeader = (VARIABLE_STORE_HEADER *) ((UINT8 *) FvHeader + FvHeader->HeaderLength);\r
+        VariableStoreHeader = (VARIABLE_STORE_HEADER *)((UINT8 *)FvHeader + FvHeader->HeaderLength);\r
+\r
+        StoreInfo->AuthFlag = (BOOLEAN)(CompareGuid (&VariableStoreHeader->Signature, &gEfiAuthenticatedVariableGuid));\r
 \r
         GuidHob = GetFirstGuidHob (&gEfiVariableIndexTableGuid);\r
         if (GuidHob != NULL) {\r
@@ -503,16 +652,17 @@ GetVariableStore (
           //\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
+          // 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
-          StoreInfo->IndexTable = (VARIABLE_INDEX_TABLE *) BuildGuidHob (&gEfiVariableIndexTableGuid, sizeof (VARIABLE_INDEX_TABLE));\r
+          StoreInfo->IndexTable              = (VARIABLE_INDEX_TABLE *)BuildGuidHob (&gEfiVariableIndexTableGuid, sizeof (VARIABLE_INDEX_TABLE));\r
           StoreInfo->IndexTable->Length      = 0;\r
           StoreInfo->IndexTable->StartPtr    = GetStartPointer (VariableStoreHeader);\r
-          StoreInfo->IndexTable->EndPtr      = GetEndPointer   (VariableStoreHeader);\r
+          StoreInfo->IndexTable->EndPtr      = GetEndPointer (VariableStoreHeader);\r
           StoreInfo->IndexTable->GoneThrough = 0;\r
         }\r
       }\r
+\r
       break;\r
 \r
     default:\r
@@ -537,9 +687,9 @@ GetVariableStore (
 **/\r
 BOOLEAN\r
 GetVariableHeader (\r
-  IN VARIABLE_STORE_INFO    *StoreInfo,\r
-  IN VARIABLE_HEADER        *Variable,\r
-  OUT VARIABLE_HEADER       **VariableHeader\r
+  IN VARIABLE_STORE_INFO  *StoreInfo,\r
+  IN VARIABLE_HEADER      *Variable,\r
+  OUT VARIABLE_HEADER     **VariableHeader\r
   )\r
 {\r
   EFI_PHYSICAL_ADDRESS  TargetAddress;\r
@@ -551,40 +701,42 @@ GetVariableHeader (
     return FALSE;\r
   }\r
 \r
-   //\r
-   // First assume variable header pointed by Variable is consecutive.\r
-   //\r
+  //\r
+  // First assume variable header pointed by Variable is consecutive.\r
+  //\r
   *VariableHeader = Variable;\r
 \r
   if (StoreInfo->FtwLastWriteData != NULL) {\r
     TargetAddress = StoreInfo->FtwLastWriteData->TargetAddress;\r
-    SpareAddress = StoreInfo->FtwLastWriteData->SpareAddress;\r
-    if (((UINTN) Variable > (UINTN) SpareAddress) &&\r
-        (((UINTN) Variable - (UINTN) SpareAddress + (UINTN) TargetAddress) >= (UINTN) GetEndPointer (StoreInfo->VariableStoreHeader))) {\r
+    SpareAddress  = StoreInfo->FtwLastWriteData->SpareAddress;\r
+    if (((UINTN)Variable > (UINTN)SpareAddress) &&\r
+        (((UINTN)Variable - (UINTN)SpareAddress + (UINTN)TargetAddress) >= (UINTN)GetEndPointer (StoreInfo->VariableStoreHeader)))\r
+    {\r
       //\r
       // Reach the end of variable store.\r
       //\r
       return FALSE;\r
     }\r
-    if (((UINTN) Variable < (UINTN) TargetAddress) && (((UINTN) Variable + sizeof (VARIABLE_HEADER)) > (UINTN) TargetAddress)) {\r
+\r
+    if (((UINTN)Variable < (UINTN)TargetAddress) && (((UINTN)Variable + GetVariableHeaderSize (StoreInfo->AuthFlag)) > (UINTN)TargetAddress)) {\r
       //\r
       // Variable header pointed by Variable is inconsecutive,\r
       // create a guid hob to combine the two partial variable header content together.\r
       //\r
       GuidHob = GetFirstGuidHob (&gEfiCallerIdGuid);\r
       if (GuidHob != NULL) {\r
-        *VariableHeader = (VARIABLE_HEADER *) GET_GUID_HOB_DATA (GuidHob);\r
+        *VariableHeader = (VARIABLE_HEADER *)GET_GUID_HOB_DATA (GuidHob);\r
       } else {\r
-        *VariableHeader = (VARIABLE_HEADER *) BuildGuidHob (&gEfiCallerIdGuid, sizeof (VARIABLE_HEADER));\r
-        PartialHeaderSize = (UINTN) TargetAddress - (UINTN) Variable;\r
+        *VariableHeader   = (VARIABLE_HEADER *)BuildGuidHob (&gEfiCallerIdGuid, GetVariableHeaderSize (StoreInfo->AuthFlag));\r
+        PartialHeaderSize = (UINTN)TargetAddress - (UINTN)Variable;\r
         //\r
         // Partial content is in NV storage.\r
         //\r
-        CopyMem ((UINT8 *) *VariableHeader, (UINT8 *) Variable, PartialHeaderSize);\r
+        CopyMem ((UINT8 *)*VariableHeader, (UINT8 *)Variable, PartialHeaderSize);\r
         //\r
         // Another partial content is in spare block.\r
         //\r
-        CopyMem ((UINT8 *) *VariableHeader + PartialHeaderSize, (UINT8 *) (UINTN) SpareAddress, sizeof (VARIABLE_HEADER) - PartialHeaderSize);\r
+        CopyMem ((UINT8 *)*VariableHeader + PartialHeaderSize, (UINT8 *)(UINTN)SpareAddress, GetVariableHeaderSize (StoreInfo->AuthFlag) - PartialHeaderSize);\r
       }\r
     }\r
   } else {\r
@@ -610,24 +762,24 @@ GetVariableHeader (
 **/\r
 VOID\r
 GetVariableNameOrData (\r
-  IN VARIABLE_STORE_INFO    *StoreInfo,\r
-  IN UINT8                  *NameOrData,\r
-  IN UINTN                  Size,\r
-  OUT UINT8                 *Buffer\r
+  IN VARIABLE_STORE_INFO  *StoreInfo,\r
+  IN UINT8                *NameOrData,\r
+  IN UINTN                Size,\r
+  OUT UINT8               *Buffer\r
   )\r
 {\r
   EFI_PHYSICAL_ADDRESS  TargetAddress;\r
   EFI_PHYSICAL_ADDRESS  SpareAddress;\r
   UINTN                 PartialSize;\r
\r
+\r
   if (StoreInfo->FtwLastWriteData != NULL) {\r
     TargetAddress = StoreInfo->FtwLastWriteData->TargetAddress;\r
-    SpareAddress = StoreInfo->FtwLastWriteData->SpareAddress;\r
-    if (((UINTN) NameOrData < (UINTN) TargetAddress) && (((UINTN) NameOrData + Size) > (UINTN) TargetAddress)) {\r
+    SpareAddress  = StoreInfo->FtwLastWriteData->SpareAddress;\r
+    if (((UINTN)NameOrData < (UINTN)TargetAddress) && (((UINTN)NameOrData + Size) > (UINTN)TargetAddress)) {\r
       //\r
       // Variable name/data is inconsecutive.\r
       //\r
-      PartialSize = (UINTN) TargetAddress - (UINTN) NameOrData;\r
+      PartialSize = (UINTN)TargetAddress - (UINTN)NameOrData;\r
       //\r
       // Partial content is in NV storage.\r
       //\r
@@ -635,7 +787,7 @@ GetVariableNameOrData (
       //\r
       // Another partial content is in spare block.\r
       //\r
-      CopyMem (Buffer + PartialSize, (UINT8 *) (UINTN) SpareAddress, Size - PartialSize);\r
+      CopyMem (Buffer + PartialSize, (UINT8 *)(UINTN)SpareAddress, Size - PartialSize);\r
       return;\r
     }\r
   }\r
@@ -661,22 +813,22 @@ GetVariableNameOrData (
 **/\r
 EFI_STATUS\r
 FindVariableEx (\r
-  IN VARIABLE_STORE_INFO         *StoreInfo,\r
-  IN CONST CHAR16                *VariableName,\r
-  IN CONST EFI_GUID              *VendorGuid,\r
-  OUT VARIABLE_POINTER_TRACK     *PtrTrack\r
+  IN VARIABLE_STORE_INFO      *StoreInfo,\r
+  IN CONST CHAR16             *VariableName,\r
+  IN CONST EFI_GUID           *VendorGuid,\r
+  OUT VARIABLE_POINTER_TRACK  *PtrTrack\r
   )\r
 {\r
-  VARIABLE_HEADER         *Variable;\r
-  VARIABLE_HEADER         *LastVariable;\r
-  VARIABLE_HEADER         *MaxIndex;\r
-  UINTN                   Index;\r
-  UINTN                   Offset;\r
-  BOOLEAN                 StopRecord;\r
-  VARIABLE_HEADER         *InDeletedVariable;\r
-  VARIABLE_STORE_HEADER   *VariableStoreHeader;\r
-  VARIABLE_INDEX_TABLE    *IndexTable;\r
-  VARIABLE_HEADER         *VariableHeader;\r
+  VARIABLE_HEADER        *Variable;\r
+  VARIABLE_HEADER        *LastVariable;\r
+  VARIABLE_HEADER        *MaxIndex;\r
+  UINTN                  Index;\r
+  UINTN                  Offset;\r
+  BOOLEAN                StopRecord;\r
+  VARIABLE_HEADER        *InDeletedVariable;\r
+  VARIABLE_STORE_HEADER  *VariableStoreHeader;\r
+  VARIABLE_INDEX_TABLE   *IndexTable;\r
+  VARIABLE_HEADER        *VariableHeader;\r
 \r
   VariableStoreHeader = StoreInfo->VariableStoreHeader;\r
 \r
@@ -692,16 +844,16 @@ FindVariableEx (
     return EFI_NOT_FOUND;\r
   }\r
 \r
-  IndexTable = StoreInfo->IndexTable;\r
+  IndexTable         = StoreInfo->IndexTable;\r
   PtrTrack->StartPtr = GetStartPointer (VariableStoreHeader);\r
-  PtrTrack->EndPtr   = GetEndPointer   (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
-  MaxIndex   = NULL;\r
+  MaxIndex       = NULL;\r
   VariableHeader = NULL;\r
 \r
   if (IndexTable != NULL) {\r
@@ -711,8 +863,8 @@ FindVariableEx (
     //\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
+      Offset  += IndexTable->Index[Index];\r
+      MaxIndex = (VARIABLE_HEADER *)((UINT8 *)IndexTable->StartPtr + Offset);\r
       GetVariableHeader (StoreInfo, MaxIndex, &VariableHeader);\r
       if (CompareWithValidVariable (StoreInfo, MaxIndex, VariableHeader, VariableName, VendorGuid, PtrTrack) == EFI_SUCCESS) {\r
         if (VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
@@ -753,21 +905,21 @@ FindVariableEx (
   //\r
   StopRecord = FALSE;\r
   while (GetVariableHeader (StoreInfo, Variable, &VariableHeader)) {\r
-    if (VariableHeader->State == VAR_ADDED || VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+    if ((VariableHeader->State == VAR_ADDED) || (VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED))) {\r
       //\r
       // Record Variable in VariableIndex HOB\r
       //\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
+        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
         } else {\r
-          IndexTable->Index[IndexTable->Length++] = (UINT16) Offset;\r
-          LastVariable = Variable;\r
+          IndexTable->Index[IndexTable->Length++] = (UINT16)Offset;\r
+          LastVariable                            = Variable;\r
         }\r
       }\r
 \r
@@ -782,6 +934,7 @@ FindVariableEx (
 \r
     Variable = GetNextVariablePtr (StoreInfo, Variable, VariableHeader);\r
   }\r
+\r
   //\r
   // If gone through the VariableStore, that means we never find in Firmware any more.\r
   //\r
@@ -814,19 +967,19 @@ FindVariable (
   OUT VARIABLE_STORE_INFO     *StoreInfo\r
   )\r
 {\r
-  EFI_STATUS                  Status;\r
-  VARIABLE_STORE_TYPE         Type;\r
+  EFI_STATUS           Status;\r
+  VARIABLE_STORE_TYPE  Type;\r
 \r
-  if (VariableName[0] != 0 && VendorGuid == NULL) {\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
+  for (Type = (VARIABLE_STORE_TYPE)0; Type < VariableStoreTypeMax; Type++) {\r
     GetVariableStore (Type, StoreInfo);\r
     Status = FindVariableEx (\r
                StoreInfo,\r
                VariableName,\r
-               VendorGuid, \r
+               VendorGuid,\r
                PtrTrack\r
                );\r
     if (!EFI_ERROR (Status)) {\r
@@ -840,7 +993,7 @@ FindVariable (
 /**\r
   This service retrieves a variable's value using its name and GUID.\r
 \r
-  Read the specified variable from the UEFI variable store. If the Data \r
+  Read the specified variable from the UEFI variable store. If the Data\r
   buffer is too small to hold the contents of the variable, the error\r
   EFI_BUFFER_TOO_SMALL is returned and DataSize is set to the required buffer\r
   size to obtain the data.\r
@@ -853,11 +1006,12 @@ FindVariable (
   @param  DataSize              On entry, points to the size in bytes of the Data buffer.\r
                                 On return, points to the size of the data returned in Data.\r
   @param  Data                  Points to the buffer which will hold the returned variable value.\r
+                                May be NULL with a zero DataSize in order to determine the size of the buffer needed.\r
 \r
   @retval EFI_SUCCESS           The variable was read successfully.\r
-  @retval EFI_NOT_FOUND         The variable could not be found.\r
-  @retval EFI_BUFFER_TOO_SMALL  The DataSize is too small for the resulting data. \r
-                                DataSize is updated with the size required for \r
+  @retval EFI_NOT_FOUND         The variable was be found.\r
+  @retval EFI_BUFFER_TOO_SMALL  The DataSize is too small for the resulting data.\r
+                                DataSize is updated with the size required for\r
                                 the specified variable.\r
   @retval EFI_INVALID_PARAMETER VariableName, VariableGuid, DataSize or Data is NULL.\r
   @retval EFI_DEVICE_ERROR      The variable could not be retrieved because of a device error.\r
@@ -866,12 +1020,12 @@ FindVariable (
 EFI_STATUS\r
 EFIAPI\r
 PeiGetVariable (\r
-  IN CONST  EFI_PEI_READ_ONLY_VARIABLE2_PPI *This,\r
-  IN CONST  CHAR16                          *VariableName,\r
-  IN CONST  EFI_GUID                        *VariableGuid,\r
-  OUT       UINT32                          *Attributes,\r
-  IN OUT    UINTN                           *DataSize,\r
-  OUT       VOID                            *Data\r
+  IN CONST  EFI_PEI_READ_ONLY_VARIABLE2_PPI  *This,\r
+  IN CONST  CHAR16                           *VariableName,\r
+  IN CONST  EFI_GUID                         *VariableGuid,\r
+  OUT       UINT32                           *Attributes,\r
+  IN OUT    UINTN                            *DataSize,\r
+  OUT       VOID                             *Data OPTIONAL\r
   )\r
 {\r
   VARIABLE_POINTER_TRACK  Variable;\r
@@ -880,10 +1034,14 @@ PeiGetVariable (
   VARIABLE_STORE_INFO     StoreInfo;\r
   VARIABLE_HEADER         *VariableHeader;\r
 \r
-  if (VariableName == NULL || VariableGuid == NULL || DataSize == NULL) {\r
+  if ((VariableName == NULL) || (VariableGuid == NULL) || (DataSize == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
+  if (VariableName[0] == 0) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
   VariableHeader = NULL;\r
 \r
   //\r
@@ -893,39 +1051,41 @@ PeiGetVariable (
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
+\r
   GetVariableHeader (&StoreInfo, Variable.CurrPtr, &VariableHeader);\r
 \r
   //\r
   // Get data size\r
   //\r
-  VarDataSize = DataSizeOfVariable (VariableHeader);\r
+  VarDataSize = DataSizeOfVariable (VariableHeader, StoreInfo.AuthFlag);\r
   if (*DataSize >= VarDataSize) {\r
     if (Data == NULL) {\r
       return EFI_INVALID_PARAMETER;\r
     }\r
 \r
-    GetVariableNameOrData (&StoreInfo, GetVariableDataPtr (Variable.CurrPtr, VariableHeader), VarDataSize, Data);\r
-\r
-    if (Attributes != NULL) {\r
-      *Attributes = VariableHeader->Attributes;\r
-    }\r
-\r
-    *DataSize = VarDataSize;\r
-    return EFI_SUCCESS;\r
+    GetVariableNameOrData (&StoreInfo, GetVariableDataPtr (Variable.CurrPtr, VariableHeader, StoreInfo.AuthFlag), VarDataSize, Data);\r
+    Status = EFI_SUCCESS;\r
   } else {\r
-    *DataSize = VarDataSize;\r
-    return EFI_BUFFER_TOO_SMALL;\r
+    Status = EFI_BUFFER_TOO_SMALL;\r
   }\r
+\r
+  if (Attributes != NULL) {\r
+    *Attributes = VariableHeader->Attributes;\r
+  }\r
+\r
+  *DataSize = VarDataSize;\r
+\r
+  return Status;\r
 }\r
 \r
 /**\r
   Return the next variable name and GUID.\r
 \r
-  This function is called multiple times to retrieve the VariableName \r
-  and VariableGuid of all variables currently available in the system. \r
-  On each call, the previous results are passed into the interface, \r
-  and, on return, the interface returns the data for the next \r
-  interface. When the entire variable list has been returned, \r
+  This function is called multiple times to retrieve the VariableName\r
+  and VariableGuid of all variables currently available in the system.\r
+  On each call, the previous results are passed into the interface,\r
+  and, on return, the interface returns the data for the next\r
+  interface. When the entire variable list has been returned,\r
   EFI_NOT_FOUND is returned.\r
 \r
   @param  This              A pointer to this instance of the EFI_PEI_READ_ONLY_VARIABLE2_PPI.\r
@@ -934,7 +1094,7 @@ PeiGetVariable (
                             On return, the size of the variable name buffer.\r
   @param  VariableName      On entry, a pointer to a null-terminated string that is the variable's name.\r
                             On return, points to the next variable's null-terminated name string.\r
-  @param  VariableGuid      On entry, a pointer to an EFI_GUID that is the variable's GUID. \r
+  @param  VariableGuid      On entry, a pointer to an EFI_GUID that is the variable's GUID.\r
                             On return, a pointer to the next variable's GUID.\r
 \r
   @retval EFI_SUCCESS           The variable was read successfully.\r
@@ -950,10 +1110,10 @@ PeiGetVariable (
 EFI_STATUS\r
 EFIAPI\r
 PeiGetNextVariableName (\r
-  IN CONST  EFI_PEI_READ_ONLY_VARIABLE2_PPI *This,\r
-  IN OUT UINTN                              *VariableNameSize,\r
-  IN OUT CHAR16                             *VariableName,\r
-  IN OUT EFI_GUID                           *VariableGuid\r
+  IN CONST  EFI_PEI_READ_ONLY_VARIABLE2_PPI  *This,\r
+  IN OUT UINTN                               *VariableNameSize,\r
+  IN OUT CHAR16                              *VariableName,\r
+  IN OUT EFI_GUID                            *VariableGuid\r
   )\r
 {\r
   VARIABLE_STORE_TYPE     Type;\r
@@ -968,14 +1128,14 @@ PeiGetNextVariableName (
   VARIABLE_STORE_INFO     StoreInfoForNv;\r
   VARIABLE_STORE_INFO     StoreInfoForHob;\r
 \r
-  if (VariableName == NULL || VariableGuid == NULL || VariableNameSize == NULL) {\r
+  if ((VariableName == NULL) || (VariableGuid == NULL) || (VariableNameSize == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
   VariableHeader = NULL;\r
 \r
   Status = FindVariable (VariableName, VariableGuid, &Variable, &StoreInfo);\r
-  if (Variable.CurrPtr == NULL || Status != EFI_SUCCESS) {\r
+  if ((Variable.CurrPtr == NULL) || (Status != EFI_SUCCESS)) {\r
     return Status;\r
   }\r
 \r
@@ -998,11 +1158,12 @@ PeiGetNextVariableName (
       //\r
       // Find current storage index\r
       //\r
-      for (Type = (VARIABLE_STORE_TYPE) 0; Type < VariableStoreTypeMax; Type++) {\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
+\r
       ASSERT (Type < VariableStoreTypeMax);\r
       //\r
       // Switch to next storage\r
@@ -1012,21 +1173,23 @@ PeiGetNextVariableName (
           break;\r
         }\r
       }\r
+\r
       //\r
-      // Capture the case that \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
+\r
       Variable.StartPtr = GetStartPointer (VariableStoreHeader[Type]);\r
-      Variable.EndPtr   = GetEndPointer   (VariableStoreHeader[Type]);\r
+      Variable.EndPtr   = GetEndPointer (VariableStoreHeader[Type]);\r
       Variable.CurrPtr  = Variable.StartPtr;\r
       GetVariableStore (Type, &StoreInfo);\r
     }\r
 \r
-    if (VariableHeader->State == VAR_ADDED || VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
+    if ((VariableHeader->State == VAR_ADDED) || (VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED))) {\r
       if (VariableHeader->State == (VAR_IN_DELETED_TRANSITION & VAR_ADDED)) {\r
         //\r
         // If it is a IN_DELETED_TRANSITION variable,\r
@@ -1035,11 +1198,11 @@ PeiGetNextVariableName (
         //\r
         Status = FindVariableEx (\r
                    &StoreInfo,\r
-                   GetVariableNamePtr (Variable.CurrPtr),\r
-                   &VariableHeader->VendorGuid,\r
+                   GetVariableNamePtr (Variable.CurrPtr, StoreInfo.AuthFlag),\r
+                   GetVendorGuidPtr (VariableHeader, StoreInfo.AuthFlag),\r
                    &VariablePtrTrack\r
                    );\r
-        if (!EFI_ERROR (Status) && VariablePtrTrack.CurrPtr != Variable.CurrPtr) {\r
+        if (!EFI_ERROR (Status) && (VariablePtrTrack.CurrPtr != Variable.CurrPtr)) {\r
           Variable.CurrPtr = GetNextVariablePtr (&StoreInfo, Variable.CurrPtr, VariableHeader);\r
           continue;\r
         }\r
@@ -1050,11 +1213,12 @@ PeiGetNextVariableName (
       //\r
       if ((VariableStoreHeader[VariableStoreTypeHob] != NULL) && (VariableStoreHeader[VariableStoreTypeNv] != NULL) &&\r
           (Variable.StartPtr == GetStartPointer (VariableStoreHeader[VariableStoreTypeNv]))\r
-         ) {\r
+          )\r
+      {\r
         Status = FindVariableEx (\r
                    &StoreInfoForHob,\r
-                   GetVariableNamePtr (Variable.CurrPtr),\r
-                   &VariableHeader->VendorGuid, \r
+                   GetVariableNamePtr (Variable.CurrPtr, StoreInfo.AuthFlag),\r
+                   GetVendorGuidPtr (VariableHeader, StoreInfo.AuthFlag),\r
                    &VariableInHob\r
                    );\r
         if (!EFI_ERROR (Status)) {\r
@@ -1063,13 +1227,13 @@ PeiGetNextVariableName (
         }\r
       }\r
 \r
-      VarNameSize = NameSizeOfVariable (VariableHeader);\r
+      VarNameSize = NameSizeOfVariable (VariableHeader, StoreInfo.AuthFlag);\r
       ASSERT (VarNameSize != 0);\r
 \r
       if (VarNameSize <= *VariableNameSize) {\r
-        GetVariableNameOrData (&StoreInfo, (UINT8 *) GetVariableNamePtr (Variable.CurrPtr), VarNameSize, (UINT8 *) VariableName);\r
+        GetVariableNameOrData (&StoreInfo, (UINT8 *)GetVariableNamePtr (Variable.CurrPtr, StoreInfo.AuthFlag), VarNameSize, (UINT8 *)VariableName);\r
 \r
-        CopyMem (VariableGuid, &VariableHeader->VendorGuid, sizeof (EFI_GUID));\r
+        CopyMem (VariableGuid, GetVendorGuidPtr (VariableHeader, StoreInfo.AuthFlag), sizeof (EFI_GUID));\r
 \r
         Status = EFI_SUCCESS;\r
       } else {\r