> -----Original Message----- > From: Bi, Dandan > Sent: Monday, April 22, 2019 9:56 AM > To: devel@edk2.groups.io > Cc: Wu, Hao A; Ni, Ray; Gao, Liming > Subject: [patch] MdeModulePkg/HiiDatabaseDxe: Release lock on all error > return path > > REF: https://bugzilla.tianocore.org/show_bug.cgi?id=1652 > > Commit ffe5f7a6b4e9 > "MdeModulePkg/HiiDatabase: Fix potential integer overflow " > added some new error paths, but it missed releasing the > mHiiDatabaseLock lock on those paths. > This patch releases mHiiDatabaseLock on those paths.
Reviewed-by: Hao Wu <hao.a...@intel.com> Best Regards, Hao Wu > > Cc: Hao Wu <hao.a...@intel.com> > Cc: Ray Ni <ray...@intel.com> > Cc: Liming Gao <liming....@intel.com> > Signed-off-by: Dandan Bi <dandan...@intel.com> > --- > MdeModulePkg/Universal/HiiDatabaseDxe/Image.c | 7 ++++++- > 1 file changed, 6 insertions(+), 1 deletion(-) > > diff --git a/MdeModulePkg/Universal/HiiDatabaseDxe/Image.c > b/MdeModulePkg/Universal/HiiDatabaseDxe/Image.c > index bd623cae15..a108fc6157 100644 > --- a/MdeModulePkg/Universal/HiiDatabaseDxe/Image.c > +++ b/MdeModulePkg/Universal/HiiDatabaseDxe/Image.c > @@ -1,10 +1,10 @@ > /** @file > Implementation for EFI_HII_IMAGE_PROTOCOL. > > > -Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.<BR> > +Copyright (c) 2007 - 2019, Intel Corporation. All rights reserved.<BR> > SPDX-License-Identifier: BSD-2-Clause-Patent > > **/ > > > @@ -651,10 +651,11 @@ HiiNewImage ( > // Make sure the size doesn't overflow UINT32. > // Note: 24Bit BMP occpuies 3 bytes per pixel. > // > NewBlockSize = (UINT32)Image->Width * Image->Height; > if (NewBlockSize > (MAX_UINT32 - (sizeof (EFI_HII_IIBT_IMAGE_24BIT_BLOCK) > - sizeof (EFI_HII_RGB_PIXEL))) / 3) { > + EfiReleaseLock (&mHiiDatabaseLock); > return EFI_OUT_OF_RESOURCES; > } > NewBlockSize = NewBlockSize * 3 + (sizeof > (EFI_HII_IIBT_IMAGE_24BIT_BLOCK) - sizeof (EFI_HII_RGB_PIXEL)); > > // > @@ -678,10 +679,11 @@ HiiNewImage ( > // > // Make sure the final package length doesn't overflow. > // Length of the package header is represented using 24 bits. So MAX > length > is MAX_UINT24. > // > if (NewBlockSize > MAX_UINT24 - ImagePackage- > >ImagePkgHdr.Header.Length) { > + EfiReleaseLock (&mHiiDatabaseLock); > return EFI_OUT_OF_RESOURCES; > } > // > // Because ImagePackage->ImageBlockSize < ImagePackage- > >ImagePkgHdr.Header.Length, > // So (ImagePackage->ImageBlockSize + NewBlockSize) <= MAX_UINT24 > @@ -719,10 +721,11 @@ HiiNewImage ( > // > // Make sure the final package length doesn't overflow. > // Length of the package header is represented using 24 bits. So MAX > length > is MAX_UINT24. > // > if (NewBlockSize > MAX_UINT24 - (sizeof (EFI_HII_IMAGE_PACKAGE_HDR) + > sizeof (EFI_HII_IIBT_END_BLOCK))) { > + EfiReleaseLock (&mHiiDatabaseLock); > return EFI_OUT_OF_RESOURCES; > } > // > // The specified package list does not contain image package. > // Create one to add this image block. > @@ -1159,16 +1162,18 @@ HiiSetImage ( > // Length of the package header is represented using 24 bits. So MAX length > is MAX_UINT24. > // 24Bit BMP occpuies 3 bytes per pixel. > // > NewBlockSize = (UINT32)Image->Width * Image->Height; > if (NewBlockSize > (MAX_UINT32 - (sizeof (EFI_HII_IIBT_IMAGE_24BIT_BLOCK) > - sizeof (EFI_HII_RGB_PIXEL))) / 3) { > + EfiReleaseLock (&mHiiDatabaseLock); > return EFI_OUT_OF_RESOURCES; > } > NewBlockSize = NewBlockSize * 3 + (sizeof > (EFI_HII_IIBT_IMAGE_24BIT_BLOCK) - sizeof (EFI_HII_RGB_PIXEL)); > if ((NewBlockSize > OldBlockSize) && > (NewBlockSize - OldBlockSize > MAX_UINT24 - ImagePackage- > >ImagePkgHdr.Header.Length) > ) { > + EfiReleaseLock (&mHiiDatabaseLock); > return EFI_OUT_OF_RESOURCES; > } > > // > // Adjust the image package to remove the original block firstly then add > the > new block. > -- > 2.18.0.windows.1 -=-=-=-=-=-=-=-=-=-=-=- Groups.io Links: You receive all messages sent to this group. View/Reply Online (#39363): https://edk2.groups.io/g/devel/message/39363 Mute This Topic: https://groups.io/mt/31269775/21656 Group Owner: devel+ow...@edk2.groups.io Unsubscribe: https://edk2.groups.io/g/devel/unsub [arch...@mail-archive.com] -=-=-=-=-=-=-=-=-=-=-=-