Reviewed-by: [email protected]

> -----Original Message-----
> From: Zeng, Star
> Sent: Tuesday, June 20, 2017 9:28 AM
> To: Yao, Jiewen <[email protected]>; Gao, Liming <[email protected]>
> Cc: Zeng, Star <[email protected]>; Shi, Steven <[email protected]>
> Subject: RE: [edk2] [PATCH] MdeModulePkg: Fix use-after-free error in
> InstallConfigurationTable()
> 
> Help review this patch.
> 
> Thanks,
> Star
> -----Original Message-----
> From: edk2-devel [mailto:[email protected]] On Behalf Of Star
> Zeng
> Sent: Monday, June 19, 2017 6:24 PM
> To: [email protected]
> Cc: Yao, Jiewen <[email protected]>; Zeng, Star <[email protected]>; Gao,
> Liming <[email protected]>
> Subject: [edk2] [PATCH] MdeModulePkg: Fix use-after-free error in
> InstallConfigurationTable()
> 
> From: "Shi, Steven" <[email protected]>
> 
> When installing configuration table and the original
> gDxeCoreST->ConfigurationTable[] buffer happen to be not big enough to
> add a new table, the CoreInstallConfigurationTable() enter the branch of line 
> 113
> in InstallConfigurationTable.c to free the old
> gDxeCoreST->ConfigurationTable[] buffer and allocate a new bigger one.
> The problem happens at line 139 CoreFreePool(), which is to free the old
> gDxeCoreST->ConfigurationTable[] buffer. The CoreFreePool()'s behavior is to
> free the buffer firstly, then call the
> InstallMemoryAttributesTableOnMemoryAllocation (PoolType) to update the
> EfiRuntimeServices type memory info, the
> CoreInstallConfigurationTable() will be re-entered by CoreFreePool() in its 
> calling
> stack, then use-after-free read error will happen at line 59 of
> InstallConfigurationTable.c and use-after-free write error will happen at 
> line 151
> and 152 of InstallConfigurationTable.c.
> 
> The patch is to update System table to the new table pointer before calling
> CoreFreePool() to free the old table.
> 
> The case above is in DxeCore, but not in PiSmmCore.
> The change in PiSmmCore is to be consistent with DxeCore.
> 
> Cc: Jiewen Yao <[email protected]>
> Cc: Liming Gao <[email protected]>
> Contributed-under: TianoCore Contribution Agreement 1.0
> Signed-off-by: Steven Shi <[email protected]>
> Signed-off-by: Star Zeng <[email protected]>
> ---
>  .../Core/Dxe/Misc/InstallConfigurationTable.c      | 34
> ++++++++++++++++------
>  .../Core/PiSmmCore/InstallConfigurationTable.c     | 34
> ++++++++++++++++------
>  2 files changed, 50 insertions(+), 18 deletions(-)  mode change 100644 =>
> 100755 MdeModulePkg/Core/Dxe/Misc/InstallConfigurationTable.c
> 
> diff --git a/MdeModulePkg/Core/Dxe/Misc/InstallConfigurationTable.c
> b/MdeModulePkg/Core/Dxe/Misc/InstallConfigurationTable.c
> old mode 100644
> new mode 100755
> index e4735db7ba45..dcdeb7f45803
> --- a/MdeModulePkg/Core/Dxe/Misc/InstallConfigurationTable.c
> +++ b/MdeModulePkg/Core/Dxe/Misc/InstallConfigurationTable.c
> @@ -1,7 +1,7 @@
>  /** @file
>    UEFI Miscellaneous boot Services InstallConfigurationTable service
> 
> -Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>
> +Copyright (c) 2006 - 2017, 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 @@ -42,6 +42,7 @@
> CoreInstallConfigurationTable (  {
>    UINTN                   Index;
>    EFI_CONFIGURATION_TABLE *EfiConfigurationTable;
> +  EFI_CONFIGURATION_TABLE *OldTable;
> 
>    //
>    // If Guid is NULL, then this operation cannot be performed @@ -68,7 +69,7
> @@ CoreInstallConfigurationTable (
>      if (Table != NULL) {
>        //
>        // If Table is not NULL, then this is a modify operation.
> -      // Modify the table enty and return.
> +      // Modify the table entry and return.
>        //
>        gDxeCoreST->ConfigurationTable[Index].VendorTable = Table;
> 
> @@ -134,15 +135,30 @@ CoreInstallConfigurationTable (
>            );
> 
>          //
> -        // Free Old Table
> +        // Record the old table pointer.
>          //
> -        CoreFreePool (gDxeCoreST->ConfigurationTable);
> -      }
> +        OldTable = gDxeCoreST->ConfigurationTable;
> 
> -      //
> -      // Update System Table
> -      //
> -      gDxeCoreST->ConfigurationTable = EfiConfigurationTable;
> +        //
> +        // As the CoreInstallConfigurationTable() may be re-entered by
> CoreFreePool()
> +        // in its calling stack, updating System table to the new table 
> pointer
> must
> +        // be done before calling CoreFreePool() to free the old table.
> +        // It can make sure the gDxeCoreST->ConfigurationTable point to the
> new table
> +        // and avoid the errors of use-after-free to the old table by the
> reenter of
> +        // CoreInstallConfigurationTable() in CoreFreePool()'s calling stack.
> +        //
> +        gDxeCoreST->ConfigurationTable = EfiConfigurationTable;
> +
> +        //
> +        // Free the old table after updating System Table to the new table
> pointer.
> +        //
> +        CoreFreePool (OldTable);
> +      } else {
> +        //
> +        // Update System Table
> +        //
> +        gDxeCoreST->ConfigurationTable = EfiConfigurationTable;
> +      }
>      }
> 
>      //
> diff --git a/MdeModulePkg/Core/PiSmmCore/InstallConfigurationTable.c
> b/MdeModulePkg/Core/PiSmmCore/InstallConfigurationTable.c
> index b2f6769c109f..2b6eef9a0e3e 100644
> --- a/MdeModulePkg/Core/PiSmmCore/InstallConfigurationTable.c
> +++ b/MdeModulePkg/Core/PiSmmCore/InstallConfigurationTable.c
> @@ -1,7 +1,7 @@
>  /** @file
>    System Management System Table Services SmmInstallConfigurationTable
> service
> 
> -  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
> +  Copyright (c) 2009 - 2017, 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
> @@ -46,6 +46,7 @@ SmmInstallConfigurationTable (  {
>    UINTN                    Index;
>    EFI_CONFIGURATION_TABLE  *ConfigurationTable;
> +  EFI_CONFIGURATION_TABLE  *OldTable;
> 
>    //
>    // If Guid is NULL, then this operation cannot be performed @@ -72,7 +73,7
> @@ SmmInstallConfigurationTable (
>      if (Table != NULL) {
>        //
>        // If Table is not NULL, then this is a modify operation.
> -      // Modify the table enty and return.
> +      // Modify the table entry and return.
>        //
>        ConfigurationTable[Index].VendorTable = Table;
>        return EFI_SUCCESS;
> @@ -130,15 +131,30 @@ SmmInstallConfigurationTable (
>            );
> 
>          //
> -        // Free Old Table
> +        // Record the old table pointer.
>          //
> -        FreePool (gSmmCoreSmst.SmmConfigurationTable);
> -      }
> +        OldTable = gSmmCoreSmst.SmmConfigurationTable;
> 
> -      //
> -      // Update System Table
> -      //
> -      gSmmCoreSmst.SmmConfigurationTable = ConfigurationTable;
> +        //
> +        // As the SmmInstallConfigurationTable() may be re-entered by
> FreePool() in
> +        // its calling stack, updating System table to the new table pointer
> must
> +        // be done before calling FreePool() to free the old table.
> +        // It can make sure the gSmmCoreSmst->SmmConfigurationTable
> point to the new
> +        // table and avoid the errors of use-after-free to the old table by 
> the
> +        // reenter of SmmInstallConfigurationTable() in FreePool()'s calling
> stack.
> +        //
> +        gSmmCoreSmst.SmmConfigurationTable = ConfigurationTable;
> +
> +        //
> +        // Free the old table after updating System Table to the new table
> pointer.
> +        //
> +        FreePool (OldTable);
> +      } else {
> +        //
> +        // Update System Table
> +        //
> +        gSmmCoreSmst.SmmConfigurationTable = ConfigurationTable;
> +      }
>      }
> 
>      //
> --
> 2.7.0.windows.1
> 
> _______________________________________________
> edk2-devel mailing list
> [email protected]
> https://lists.01.org/mailman/listinfo/edk2-devel
_______________________________________________
edk2-devel mailing list
[email protected]
https://lists.01.org/mailman/listinfo/edk2-devel

Reply via email to