From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received-SPF: Pass (sender SPF authorized) identity=mailfrom; client-ip=192.55.52.43; helo=mga05.intel.com; envelope-from=hong-chihx.hsueh@intel.com; receiver=edk2-devel@lists.01.org Received: from mga05.intel.com (mga05.intel.com [192.55.52.43]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by ml01.01.org (Postfix) with ESMTPS id 98AF12194D3B3 for ; Tue, 29 Jan 2019 17:19:39 -0800 (PST) X-Amp-Result: SKIPPED(no attachment in message) X-Amp-File-Uploaded: False Received: from fmsmga003.fm.intel.com ([10.253.24.29]) by fmsmga105.fm.intel.com with ESMTP/TLS/DHE-RSA-AES256-GCM-SHA384; 29 Jan 2019 17:19:38 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="5.56,538,1539673200"; d="scan'208";a="129612703" Received: from hhsuehx-desk.gar.corp.intel.com ([10.9.67.173]) by FMSMGA003.fm.intel.com with ESMTP; 29 Jan 2019 17:19:38 -0800 From: Neo Hsueh To: edk2-devel@lists.01.org Cc: Michael D Kinney , Liming Gao , Dandan Bi , Laszlo Ersek Date: Tue, 29 Jan 2019 17:19:36 -0800 Message-Id: <20190130011936.5804-1-hong-chihx.hsueh@intel.com> X-Mailer: git-send-email 2.16.2.windows.1 Subject: [PATCH] MdePkg/BasePeCoffLib: skip runtime relocation if reloc info is invalid X-BeenThere: edk2-devel@lists.01.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: EDK II Development List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Wed, 30 Jan 2019 01:19:39 -0000 Skip runtime relocation for PE images that provide invalid relocation infomation (ex: RelocDir->Size = 0) to fix a hang observed while booting Windows. Contributed-under: TianoCore Contribution Agreement 1.1 Signed-off-by: Neo Hsueh Cc: Michael D Kinney Cc: Liming Gao Cc: Dandan Bi Cc: Laszlo Ersek --- MdePkg/Library/BasePeCoffLib/BasePeCoff.c | 30 ++++++++++++++++++++---------- 1 file changed, 20 insertions(+), 10 deletions(-) diff --git a/MdePkg/Library/BasePeCoffLib/BasePeCoff.c b/MdePkg/Library/BasePeCoffLib/BasePeCoff.c index 1bd079ad6a..e2c62e1932 100644 --- a/MdePkg/Library/BasePeCoffLib/BasePeCoff.c +++ b/MdePkg/Library/BasePeCoffLib/BasePeCoff.c @@ -1002,7 +1002,7 @@ PeCoffLoaderRelocateImage ( RelocDir->VirtualAddress + RelocDir->Size - 1, TeStrippedOffset ); - if (RelocBase == NULL || RelocBaseEnd == NULL || RelocBaseEnd < RelocBase) { + if (RelocBase == NULL || RelocBaseEnd == NULL || (UINTN) RelocBaseEnd < (UINTN) RelocBase) { ImageContext->ImageError = IMAGE_ERROR_FAILED_RELOCATION; return RETURN_LOAD_ERROR; } @@ -1022,7 +1022,7 @@ PeCoffLoaderRelocateImage ( // Run the relocation information and apply the fixups // FixupData = ImageContext->FixupData; - while (RelocBase < RelocBaseEnd) { + while ((UINTN) RelocBase < (UINTN) RelocBaseEnd) { Reloc = (UINT16 *) ((CHAR8 *) RelocBase + sizeof (EFI_IMAGE_BASE_RELOCATION)); // @@ -1051,7 +1051,7 @@ PeCoffLoaderRelocateImage ( // // Run this relocation record // - while (Reloc < RelocEnd) { + while ((UINTN) Reloc < (UINTN) RelocEnd) { Fixup = PeCoffLoaderImageAddress (ImageContext, RelocBase->VirtualAddress + (*Reloc & 0xFFF), TeStrippedOffset); if (Fixup == NULL) { ImageContext->ImageError = IMAGE_ERROR_FAILED_RELOCATION; @@ -1739,13 +1739,23 @@ PeCoffLoaderRelocateImageForRuntime ( // is present in the image. You have to check the NumberOfRvaAndSizes in // the optional header to verify a desired directory entry is there. // + RelocBase = NULL; + RelocBaseEnd = NULL; if (NumberOfRvaAndSizes > EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC) { RelocDir = DataDirectory + EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC; - RelocBase = (EFI_IMAGE_BASE_RELOCATION *) PeCoffLoaderImageAddress (&ImageContext, RelocDir->VirtualAddress, 0); - RelocBaseEnd = (EFI_IMAGE_BASE_RELOCATION *) PeCoffLoaderImageAddress (&ImageContext, - RelocDir->VirtualAddress + RelocDir->Size - 1, - 0 - ); + if ((RelocDir != NULL) && (RelocDir->Size > 0)) { + RelocBase = (EFI_IMAGE_BASE_RELOCATION *) PeCoffLoaderImageAddress (&ImageContext, RelocDir->VirtualAddress, 0); + RelocBaseEnd = (EFI_IMAGE_BASE_RELOCATION *) PeCoffLoaderImageAddress (&ImageContext, + RelocDir->VirtualAddress + RelocDir->Size - 1, + 0 + ); + } + if (RelocBase == NULL || RelocBaseEnd == NULL || (UINTN) RelocBaseEnd < (UINTN) RelocBase) { + // + // relocation block is not valid, just return + // + return; + } } else { // // Cannot find relocations, cannot continue to relocate the image, ASSERT for this invalid image. @@ -1769,7 +1779,7 @@ PeCoffLoaderRelocateImageForRuntime ( // FixupData = RelocationData; RelocBaseOrig = RelocBase; - while (RelocBase < RelocBaseEnd) { + while ((UINTN) RelocBase < (UINTN) RelocBaseEnd) { // // Add check for RelocBase->SizeOfBlock field. // @@ -1794,7 +1804,7 @@ PeCoffLoaderRelocateImageForRuntime ( // // Run this relocation record // - while (Reloc < RelocEnd) { + while ((UINTN) Reloc < (UINTN) RelocEnd) { Fixup = PeCoffLoaderImageAddress (&ImageContext, RelocBase->VirtualAddress + (*Reloc & 0xFFF), 0); if (Fixup == NULL) { -- 2.16.2.windows.1