Re: [Patch] MdeModulePkg/DxeCapsuleLibFmp: Verify nested capsule with FMP

["Yao, Jiewen" <jiewen.yao@intel.com>]

Reviewed-by: Jiewen.yao@intel.com

> -----Original Message-----
> From: Kinney, Michael D
> Sent: Thursday, February 8, 2018 2:50 AM
> To: edk2-devel@lists.01.org
> Cc: Yao, Jiewen <jiewen.yao@intel.com>; Zeng, Star <star.zeng@intel.com>;
> Dong, Eric <eric.dong@intel.com>; Kinney, Michael D
> <michael.d.kinney@intel.com>
> Subject: [Patch] MdeModulePkg/DxeCapsuleLibFmp: Verify nested capsule with
> FMP
> 
> https://bugzilla.tianocore.org/show_bug.cgi?id=873
> 
> Update IsNestedFmpCapsule() to verify the CapsuleGuid in
> the CapsuleHeader against the installed Firmware Management
> Protocol instances.  The current logic that uses the ESRT
> Table does not work because capsules are processed before
> the ESRT Table is published at the Ready To Boot event.
> 
> Cc: Jiewen Yao <jiewen.yao@intel.com>
> Cc: Star Zeng <star.zeng@intel.com>
> Cc: Eric Dong <eric.dong@intel.com>
> Contributed-under: TianoCore Contribution Agreement 1.1
> Signed-off-by: Michael D Kinney <michael.d.kinney@intel.com>
> ---
>  .../Library/DxeCapsuleLibFmp/DxeCapsuleLib.c       | 27
> ++++++++++++----------
>  .../Library/DxeCapsuleLibFmp/DxeCapsuleLib.inf     |  3 +--
>  2 files changed, 16 insertions(+), 14 deletions(-)
> 
> diff --git a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
> b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
> index 2f397789b5..87e1deec03 100644
> --- a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
> +++ b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
> @@ -10,7 +10,7 @@
>    ValidateFmpCapsule(), DisplayCapsuleImage(), ConvertBmpToGopBlt() will
>    receive untrusted input and do basic validation.
> 
> -  Copyright (c) 2016 - 2017, Intel Corporation. All rights reserved.<BR>
> +  Copyright (c) 2016 - 2018, Intel Corporation. All rights reserved.<BR>
>    This program and the accompanying materials
>    are licensed and made available under the terms and conditions of the BSD
> License
>    which accompanies this distribution.  The full text of the license may be
> found at
> @@ -1446,7 +1446,6 @@ IsNestedFmpCapsule (
>    )
>  {
>    EFI_STATUS                 Status;
> -  EFI_SYSTEM_RESOURCE_TABLE  *Esrt;
>    EFI_SYSTEM_RESOURCE_ENTRY  *EsrtEntry;
>    UINTN                      Index;
>    BOOLEAN                    EsrtGuidFound;
> @@ -1454,6 +1453,8 @@ IsNestedFmpCapsule (
>    UINTN                      NestedCapsuleSize;
>    ESRT_MANAGEMENT_PROTOCOL   *EsrtProtocol;
>    EFI_SYSTEM_RESOURCE_ENTRY  Entry;
> +  EFI_HANDLE                 *HandleBuffer;
> +  UINTN                      NumberOfHandles;
> 
>    EsrtGuidFound = FALSE;
>    if (mIsVirtualAddrConverted) {
> @@ -1479,19 +1480,21 @@ IsNestedFmpCapsule (
>      }
> 
>      //
> -    // Check ESRT configuration table
> +    // Check Firmware Management Protocols
>      //
>      if (!EsrtGuidFound) {
> -      Status = EfiGetSystemConfigurationTable(&gEfiSystemResourceTableGuid,
> (VOID **)&Esrt);
> +      HandleBuffer = NULL;
> +      Status = GetFmpHandleBufferByType (
> +                 &CapsuleHeader->CapsuleGuid,
> +                 0,
> +                 &NumberOfHandles,
> +                 &HandleBuffer
> +                 );
>        if (!EFI_ERROR(Status)) {
> -        ASSERT (Esrt != NULL);
> -        EsrtEntry = (VOID *)(Esrt + 1);
> -        for (Index = 0; Index < Esrt->FwResourceCount; Index++, EsrtEntry++)
> {
> -          if (CompareGuid(&EsrtEntry->FwClass,
> &CapsuleHeader->CapsuleGuid)) {
> -            EsrtGuidFound = TRUE;
> -            break;
> -          }
> -        }
> +        EsrtGuidFound = TRUE;
> +      }
> +      if (HandleBuffer != NULL) {
> +        FreePool (HandleBuffer);
>        }
>      }
>    }
> diff --git a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.inf
> b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.inf
> index a7c36993c4..90edc52ee0 100644
> --- a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.inf
> +++ b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.inf
> @@ -3,7 +3,7 @@
>  #
>  #  Capsule library instance for DXE_DRIVER module types.
>  #
> -#  Copyright (c) 2016 - 2017, Intel Corporation. All rights reserved.<BR>
> +#  Copyright (c) 2016 - 2018, Intel Corporation. All rights reserved.<BR>
>  #  This program and the accompanying materials
>  #  are licensed and made available under the terms and conditions of the BSD
> License
>  #  which accompanies this distribution.  The full text of the license may be
> found at
> @@ -72,7 +72,6 @@
>  [Guids]
>    gEfiFmpCapsuleGuid                      ## SOMETIMES_CONSUMES
> ## GUID
>    gWindowsUxCapsuleGuid                   ## SOMETIMES_CONSUMES
> ## GUID
> -  gEfiSystemResourceTableGuid             ## SOMETIMES_CONSUMES ##
> GUID
>    ## SOMETIMES_CONSUMES ## Variable:L"CapsuleMax"
>    ## SOMETIMES_PRODUCES ## Variable:L"CapsuleMax"
>    gEfiCapsuleReportGuid
> --
> 2.14.2.windows.3