Reviewed-by: Steven Shi <[email protected]>
Steven Shi Intel\SSG\STO\UEFI Firmware Tel: +86 021-61166522 iNet: 821-6522 > -----Original Message----- > From: Zeng, Star > Sent: Monday, June 19, 2017 6:24 PM > To: [email protected] > Cc: Shi, Steven <[email protected]>; Yao, Jiewen <[email protected]>; > Gao, Liming <[email protected]>; Zeng, Star <[email protected]> > Subject: [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

