From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received-SPF: Pass (sender SPF authorized) identity=mailfrom; client-ip=134.134.136.65; helo=mga03.intel.com; envelope-from=hao.a.wu@intel.com; receiver=edk2-devel@lists.01.org Received: from mga03.intel.com (mga03.intel.com [134.134.136.65]) (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 2B9DA211B1112 for ; Tue, 22 Jan 2019 17:25:15 -0800 (PST) X-Amp-Result: SKIPPED(no attachment in message) X-Amp-File-Uploaded: False Received: from orsmga002.jf.intel.com ([10.7.209.21]) by orsmga103.jf.intel.com with ESMTP/TLS/DHE-RSA-AES256-GCM-SHA384; 22 Jan 2019 17:25:15 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="5.56,509,1539673200"; d="scan'208";a="128093766" Received: from fmsmsx105.amr.corp.intel.com ([10.18.124.203]) by orsmga002.jf.intel.com with ESMTP; 22 Jan 2019 17:25:14 -0800 Received: from fmsmsx157.amr.corp.intel.com (10.18.116.73) by FMSMSX105.amr.corp.intel.com (10.18.124.203) with Microsoft SMTP Server (TLS) id 14.3.408.0; Tue, 22 Jan 2019 17:25:14 -0800 Received: from shsmsx102.ccr.corp.intel.com (10.239.4.154) by FMSMSX157.amr.corp.intel.com (10.18.116.73) with Microsoft SMTP Server (TLS) id 14.3.408.0; Tue, 22 Jan 2019 17:25:14 -0800 Received: from shsmsx101.ccr.corp.intel.com ([169.254.1.196]) by shsmsx102.ccr.corp.intel.com ([169.254.2.63]) with mapi id 14.03.0415.000; Wed, 23 Jan 2019 09:25:12 +0800 From: "Wu, Hao A" To: "Ni, Ray" , "Chen, Chen A" , "edk2-devel@lists.01.org" CC: "Zhang, Chao B" Thread-Topic: [PATCH 3/3] FatPkg: Add GPT check in FatPei to support Capsule-on-Disk feature. Thread-Index: AQHUrheqYYgH/GYBB0SreS88FVfnS6W8GEkw Date: Wed, 23 Jan 2019 01:25:11 +0000 Message-ID: References: <20190117020303.10244-1-chen.a.chen@intel.com> <734D49CCEBEEF84792F5B80ED585239D5BFC0EC2@SHSMSX103.ccr.corp.intel.com> In-Reply-To: <734D49CCEBEEF84792F5B80ED585239D5BFC0EC2@SHSMSX103.ccr.corp.intel.com> Accept-Language: zh-CN, en-US X-MS-Has-Attach: X-MS-TNEF-Correlator: x-originating-ip: [10.239.127.40] MIME-Version: 1.0 Subject: Re: [PATCH 3/3] FatPkg: Add GPT check in FatPei to support Capsule-on-Disk feature. 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, 23 Jan 2019 01:25:16 -0000 Content-Language: en-US Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: quoted-printable > -----Original Message----- > From: Ni, Ray > Sent: Thursday, January 17, 2019 11:50 AM > To: Chen, Chen A; edk2-devel@lists.01.org; Wu, Hao A > Cc: Zhang, Chao B > Subject: RE: [PATCH 3/3] FatPkg: Add GPT check in FatPei to support Capsu= le- > on-Disk feature. >=20 > Adding Wu Hao in the reviewer list. Sorry for the delayed response. I will try my best to give my comments as soon as possible. Please help to ping if there is reply within the next two weeks. Best Regards, Hao Wu >=20 > > -----Original Message----- > > From: Chen, Chen A > > Sent: Thursday, January 17, 2019 10:03 AM > > To: edk2-devel@lists.01.org > > Cc: Chen, Chen A ; Ni, Ray ; > > Zhang, Chao B > > Subject: [PATCH 3/3] FatPkg: Add GPT check in FatPei to support Capsule= - > on- > > Disk feature. > > > > BZ: https://bugzilla.tianocore.org/show_bug.cgi?id=3D1470 > > This feature is used for finding GPT partition, follow the following st= ep to > > check. > > 1) Check Protective MBR. > > 2) Check GPT primary/backup header. > > 3) Check GPT primary/backup entry array. > > > > Cc: Ruiyu Ni > > Cc: Zhang Chao B > > Contributed-under: TianoCore Contribution Agreement 1.1 > > Signed-off-by: Chen A Chen > > --- > > FatPkg/FatPei/FatLitePeim.h | 1 + > > FatPkg/FatPei/FatPei.inf | 3 + > > FatPkg/FatPei/Gpt.c | 546 > > ++++++++++++++++++++++++++++++++++++++++++++ > > 3 files changed, 550 insertions(+) > > create mode 100644 FatPkg/FatPei/Gpt.c > > > > diff --git a/FatPkg/FatPei/FatLitePeim.h b/FatPkg/FatPei/FatLitePeim.h > index > > fbf887da5f..afb429c56e 100644 > > --- a/FatPkg/FatPei/FatLitePeim.h > > +++ b/FatPkg/FatPei/FatLitePeim.h > > @@ -27,6 +27,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF > ANY > > KIND, EITHER EXPRESS OR IMPLIED. > > #include > > #include > > #include > > +#include > > #include > > #include > > #include > > diff --git a/FatPkg/FatPei/FatPei.inf b/FatPkg/FatPei/FatPei.inf index > > 829e87fe92..dd0869f7cd 100644 > > --- a/FatPkg/FatPei/FatPei.inf > > +++ b/FatPkg/FatPei/FatPei.inf > > @@ -31,6 +31,7 @@ > > > > [Sources] > > Mbr.c > > + Gpt.c > > Eltorito.c > > Part.c > > FatLiteApi.c > > @@ -49,6 +50,7 @@ > > [LibraryClasses] > > PcdLib > > BaseMemoryLib > > + MemoryAllocationLib > > PeimEntryPoint > > BaseLib > > DebugLib > > @@ -61,6 +63,7 @@ > > gRecoveryOnFatIdeDiskGuid ## SOMETIMES_CONSUMES = ## > > UNDEFINED > > gRecoveryOnFatFloppyDiskGuid ## SOMETIMES_CONSUMES = ## > > UNDEFINED > > gRecoveryOnFatNvmeDiskGuid ## SOMETIMES_CONSUMES = ## > > UNDEFINED > > + gEfiPartTypeUnusedGuid ## SOMETIMES_CONSUMES = ## > > UNDEFINED > > > > > > [Ppis] > > diff --git a/FatPkg/FatPei/Gpt.c b/FatPkg/FatPei/Gpt.c new file mode > 100644 > > index 0000000000..d1f4c1c8b5 > > --- /dev/null > > +++ b/FatPkg/FatPei/Gpt.c > > @@ -0,0 +1,546 @@ > > +/** @file > > + Routines supporting partition discovery and > > + logical device reading > > + > > +Copyright (c) 2006 - 2018, Intel Corporation. All rights reserved.
> > + > > +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 http://opensource.org/licenses/bsd-license.php > > + > > +THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" > > BASIS, > > +WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER > > EXPRESS OR IMPLIED. > > + > > +**/ > > + > > +#include > > +#include > > +#include > > +#include "FatLitePeim.h" > > + > > +// > > +// Assumption: 'a' and 'blocksize' are all UINT32 or UINT64. > > +// If 'a' and 'blocksize' are not the same type, should use DivU64xU32= to > > calculate. > > +// > > +#define EFI_SIZE_TO_BLOCKS(a, blocksize) (((a) / (blocksize)) + (((a) > > +% (blocksize)) ? 1 : 0)) > > + > > +// > > +// GPT Partition Entry Status > > +// > > +typedef struct { > > + BOOLEAN OutOfRange; > > + BOOLEAN Overlap; > > + BOOLEAN OsSpecific; > > +} EFI_PARTITION_ENTRY_STATUS; > > + > > +/** > > + Check if the CRC field in the Partition table header is valid > > + > > + @param[in] BlockIo Parent BlockIo interface > > + @param[in] DiskIo Disk Io Protocol. > > + @param[in] PartHeader Partition table header structure > > + > > + @retval TRUE the CRC is valid > > + @retval FALSE the CRC is invalid > > + > > +**/ > > +BOOLEAN > > +PartitionCheckGptHeaderCRC ( > > + IN EFI_PARTITION_TABLE_HEADER *PartHeader > > + ) > > +{ > > + UINT32 GptHdrCrc; > > + UINT32 Crc; > > + > > + GptHdrCrc =3D PartHeader->Header.CRC32; > > + > > + // > > + // Set CRC field to zero when doing calcuation // > > + PartHeader->Header.CRC32 =3D 0; > > + > > + Crc =3D CalculateCrc32 (PartHeader, PartHeader->Header.HeaderSize); > > + > > + // > > + // Restore Header CRC > > + // > > + PartHeader->Header.CRC32 =3D GptHdrCrc; > > + > > + return (GptHdrCrc =3D=3D Crc); > > +} > > + > > + > > +/** > > + Check if the CRC field in the Partition table header is valid > > + for Partition entry array. > > + > > + @param[in] BlockIo Parent BlockIo interface > > + @param[in] DiskIo Disk Io Protocol. > > + @param[in] PartHeader Partition table header structure > > + > > + @retval TRUE the CRC is valid > > + @retval FALSE the CRC is invalid > > + > > +**/ > > +BOOLEAN > > +PartitionCheckGptEntryArrayCRC ( > > + IN EFI_PARTITION_TABLE_HEADER *PartHeader, > > + IN EFI_PARTITION_ENTRY *PartEntry > > + ) > > +{ > > + UINT32 Crc; > > + UINTN Size; > > + > > + Size =3D (UINTN)MultU64x32(PartHeader->NumberOfPartitionEntries, > > + PartHeader->SizeOfPartitionEntry); > > + Crc =3D CalculateCrc32 (PartEntry, Size); > > + > > + return (BOOLEAN) (PartHeader->PartitionEntryArrayCRC32 =3D=3D Crc); = } > > + > > +/** > > + The function is used for valid GPT table. Both for Primary and Backu= p GPT > > header. > > + > > + @param[in] PrivateData The global memory map > > + @param[in] ParentBlockDevNo The parent block device > > + @param[in] IsPrimaryHeader Indicate to which header will be check= ed. > > + @param[in] PartHdr Stores the partition table that is rea= d > > + > > + @retval TRUE The partition table is valid > > + @retval FALSE The partition table is not valid > > + > > +**/ > > +BOOLEAN > > +PartitionCheckGptHeader ( > > + IN PEI_FAT_PRIVATE_DATA *PrivateData, > > + IN UINTN ParentBlockDevNo, > > + IN BOOLEAN IsPrimaryHeader, > > + IN EFI_PARTITION_TABLE_HEADER *PartHdr > > + ) > > +{ > > + PEI_FAT_BLOCK_DEVICE *ParentBlockDev; > > + EFI_PEI_LBA Lba; > > + EFI_PEI_LBA AlternateLba; > > + EFI_PEI_LBA EntryArrayLastLba; > > + > > + UINT64 PartitionEntryArraySize; > > + UINT64 PartitionEntryBlockNumb; > > + UINT32 EntryArraySizeRemainder; > > + > > + ParentBlockDev =3D &(PrivateData->BlockDevice[ParentBlockDevNo]); > > + > > + if (IsPrimaryHeader) { > > + Lba =3D PRIMARY_PART_HEADER_LBA; > > + AlternateLba =3D ParentBlockDev->LastBlock; } else { > > + Lba =3D ParentBlockDev->LastBlock; > > + AlternateLba =3D PRIMARY_PART_HEADER_LBA; } > > + > > + if ( (PartHdr->Header.Signature !=3D EFI_PTAB_HEADER_ID) || > > + (PartHdr->Header.Revision !=3D 0x00010000) || > > + (PartHdr->Header.HeaderSize < 92) || > > + (PartHdr->Header.HeaderSize > ParentBlockDev->BlockSize) || > > + (!PartitionCheckGptHeaderCRC (PartHdr)) || > > + (PartHdr->Header.Reserved !=3D 0) > > + ) { > > + DEBUG ((DEBUG_ERROR, "Invalid efi partition table header\n")); > > + return FALSE; > > + } > > + > > + // > > + // | Block0 | Block1 |Block2 ~ FirstUsableLBA - > > 1|FirstUsableLBA, ... ,LastUsableLBA|LastUsableLBA+1 ~ LastBlock-1| > > LastBlock | > > + // |Protective MBR|Primary Header|Entry Array(At Least 16384)| > > Partition | Entry Array(At Least 16384) |BackUp Header| > > + // > > + // 1. Protective MBR is fixed at Block 0. > > + // 2. Primary Header is fixed at Block 1. > > + // 3. Backup Header is fixed at LastBlock. > > + // 4. Must be remain 128*128 bytes for primary entry array. > > + // 5. Must be remain 128*128 bytes for backup entry array. > > + // 6. SizeOfPartitionEntry must be equals to 128 * 2^n. > > + // > > + if ( (PartHdr->MyLBA !=3D Lba) || > > + (PartHdr->AlternateLBA !=3D AlternateLba) || > > + (PartHdr->FirstUsableLBA < 2 + EFI_SIZE_TO_BLOCKS > > (GPT_PART_ENTRY_MIN_SIZE, ParentBlockDev->BlockSize)) || > > + (PartHdr->LastUsableLBA > ParentBlockDev->LastBlock - 1 - > > EFI_SIZE_TO_BLOCKS (GPT_PART_ENTRY_MIN_SIZE, ParentBlockDev- > > >BlockSize)) || > > + (PartHdr->FirstUsableLBA > PartHdr->LastUsableLBA) || > > + (PartHdr->PartitionEntryLBA < 2) || > > + (PartHdr->PartitionEntryLBA > ParentBlockDev->LastBlock - 1) || > > + (PartHdr->PartitionEntryLBA >=3D PartHdr->FirstUsableLBA && Par= tHdr- > > >PartitionEntryLBA <=3D PartHdr->LastUsableLBA) || > > + (PartHdr->SizeOfPartitionEntry%128 !=3D 0) || > > + (PartHdr->SizeOfPartitionEntry !=3D sizeof (EFI_PARTITION_ENTRY= )) > > + ) { > > + DEBUG ((DEBUG_ERROR, "Invalid efi partition table header\n")); > > + return FALSE; > > + } > > + > > + PartitionEntryArraySize =3D MultU64x32 > > + (PartHdr->NumberOfPartitionEntries, PartHdr->SizeOfPartitionEntry); > > + EntryArraySizeRemainder =3D 0; PartitionEntryBlockNumb =3D > > + DivU64x32Remainder (PartitionEntryArraySize, ParentBlockDev- > >BlockSize, > > &EntryArraySizeRemainder); if (EntryArraySizeRemainder !=3D 0) { > > + PartitionEntryBlockNumb++; > > + } > > + > > + if (IsPrimaryHeader) { > > + EntryArrayLastLba =3D PartHdr->FirstUsableLBA; } else { > > + EntryArrayLastLba =3D ParentBlockDev->LastBlock; } > > + > > + // > > + // Make sure partition entry array not overlaps with partition area = or the > > LastBlock. > > + // > > + if (PartHdr->PartitionEntryLBA + PartitionEntryBlockNumb > > > EntryArrayLastLba) { > > + DEBUG ((DEBUG_ERROR, "GPT Partition Entry Array Error!\n")); > > + DEBUG ((DEBUG_ERROR, "PartitionEntryArraySize =3D %lu.\n", > > PartitionEntryArraySize)); > > + DEBUG ((DEBUG_ERROR, "PartitionEntryLBA =3D %lu.\n", PartHdr- > > >PartitionEntryLBA)); > > + DEBUG ((DEBUG_ERROR, "PartitionEntryBlockNumb =3D %lu.\n", > > PartitionEntryBlockNumb)); > > + DEBUG ((DEBUG_ERROR, "EntryArrayLastLba =3D %lu.\n", > > EntryArrayLastLba)); > > + return FALSE; > > + } > > + > > + return TRUE; > > +} > > + > > +/** > > + This function is used to verify each partition in block device. > > + > > + @param[in] PrivateData The global memory map > > + @param[in] ParentBlockDevNo The parent block device > > + @param[in] PartHdr Stores the partition table that is rea= d > > + > > + @retval TRUE The partition is valid > > + @retval FALSE The partition is not valid > > + > > +**/ > > + > > +BOOLEAN > > +PartitionCheckGptEntryArray ( > > + IN PEI_FAT_PRIVATE_DATA *PrivateData, > > + IN UINTN ParentBlockDevNo, > > + IN EFI_PARTITION_TABLE_HEADER *PartHdr > > + ) > > +{ > > + EFI_STATUS Status; > > + PEI_FAT_BLOCK_DEVICE *ParentBlockDev; > > + PEI_FAT_BLOCK_DEVICE *BlockDevPtr; > > + > > + UINT64 PartitionEntryArraySize; > > + UINT64 PartitionEntryBlockNumb; > > + UINT32 EntryArraySizeRemainder; > > + > > + EFI_PARTITION_ENTRY *PartitionEntryBuffer; > > + EFI_PARTITION_ENTRY_STATUS *PartitionEntryStatus; > > + > > + BOOLEAN Found; > > + EFI_LBA StartingLBA; > > + EFI_LBA EndingLBA; > > + UINTN Index; > > + UINTN Index1; > > + UINTN Index2; > > + EFI_PARTITION_ENTRY *Entry; > > + > > + ParentBlockDev =3D &(PrivateData->BlockDevice[ParentBlockDevNo]); > > + Found =3D FALSE; > > + > > + PartitionEntryArraySize =3D MultU64x32 > > + (PartHdr->NumberOfPartitionEntries, PartHdr->SizeOfPartitionEntry); > > + EntryArraySizeRemainder =3D 0; PartitionEntryBlockNumb =3D > > + DivU64x32Remainder (PartitionEntryArraySize, ParentBlockDev- > >BlockSize, > > &EntryArraySizeRemainder); if (EntryArraySizeRemainder !=3D 0) { > > + PartitionEntryBlockNumb++; > > + } > > + PartitionEntryArraySize =3D MultU64x32 (PartitionEntryBlockNumb, > > + ParentBlockDev->BlockSize); > > + > > + PartitionEntryBuffer =3D (EFI_PARTITION_ENTRY *) AllocatePages > > + (EFI_SIZE_TO_PAGES ((UINTN)PartitionEntryArraySize)); > > + if (PartitionEntryBuffer =3D=3D NULL) { > > + DEBUG ((DEBUG_ERROR, "Allocate memory error!\n")); > > + goto EXIT; > > + } > > + > > + PartitionEntryStatus =3D (EFI_PARTITION_ENTRY_STATUS *) AllocatePage= s > > + (EFI_SIZE_TO_PAGES (PartHdr->NumberOfPartitionEntries * sizeof > > (EFI_PARTITION_ENTRY_STATUS))); if (PartitionEntryStatus =3D=3D NULL) = { > > + DEBUG ((DEBUG_ERROR, "Allocate memory error!\n")); > > + goto EXIT; > > + } > > + ZeroMem (PartitionEntryStatus, PartHdr->NumberOfPartitionEntries * > > + sizeof (EFI_PARTITION_ENTRY_STATUS)); > > + > > + Status =3D FatReadBlock ( > > + PrivateData, > > + ParentBlockDevNo, > > + PartHdr->PartitionEntryLBA, > > + (UINTN)PartitionEntryArraySize, > > + PartitionEntryBuffer > > + ); > > + if (EFI_ERROR (Status)) { > > + DEBUG ((DEBUG_ERROR, "Read partition entry array error!\n")); > > + goto EXIT; > > + } > > + > > + if (!PartitionCheckGptEntryArrayCRC (PartHdr, PartitionEntryBuffer))= { > > + DEBUG ((EFI_D_ERROR, "Partition entries CRC check fail\n")); > > + goto EXIT; > > + } > > + > > + for (Index1 =3D 0; Index1 < PartHdr->NumberOfPartitionEntries; Index= 1++) > { > > + Entry =3D (EFI_PARTITION_ENTRY *) ((UINT8 *) PartitionEntryBuffer = + > > Index1 * PartHdr->SizeOfPartitionEntry); > > + if (CompareGuid (&Entry->PartitionTypeGUID, > > &gEfiPartTypeUnusedGuid)) { > > + continue; > > + } > > + > > + StartingLBA =3D Entry->StartingLBA; > > + EndingLBA =3D Entry->EndingLBA; > > + if (StartingLBA > EndingLBA || > > + StartingLBA < PartHdr->FirstUsableLBA || > > + StartingLBA > PartHdr->LastUsableLBA || > > + EndingLBA < PartHdr->FirstUsableLBA || > > + EndingLBA > PartHdr->LastUsableLBA > > + ) { > > + PartitionEntryStatus[Index1].OutOfRange =3D TRUE; > > + continue; > > + } > > + > > + if ((Entry->Attributes & BIT1) !=3D 0) { > > + // > > + // If Bit 1 is set, this indicate that this is an OS specific GU= ID partition. > > + // > > + PartitionEntryStatus[Index1].OsSpecific =3D TRUE; > > + } > > + > > + for (Index2 =3D Index1 + 1; Index2 < PartHdr->NumberOfPartitionEnt= ries; > > Index2++) { > > + Entry =3D (EFI_PARTITION_ENTRY *) ((UINT8 *) PartitionEntryBuffe= r + > > Index2 * PartHdr->SizeOfPartitionEntry); > > + if (CompareGuid (&Entry->PartitionTypeGUID, > > &gEfiPartTypeUnusedGuid)) { > > + continue; > > + } > > + > > + if (Entry->EndingLBA >=3D StartingLBA && Entry->StartingLBA <=3D > EndingLBA) > > { > > + // > > + // This region overlaps with the Index1'th region > > + // > > + PartitionEntryStatus[Index1].Overlap =3D TRUE; > > + PartitionEntryStatus[Index2].Overlap =3D TRUE; > > + continue; > > + } > > + } > > + } > > + > > + for (Index =3D 0; Index < PartHdr->NumberOfPartitionEntries; Index++= ) { > > + if (CompareGuid (&PartitionEntryBuffer[Index].PartitionTypeGUID, > > &gEfiPartTypeUnusedGuid)|| > > + PartitionEntryStatus[Index].OutOfRange || > > + PartitionEntryStatus[Index].Overlap || > > + PartitionEntryStatus[Index].OsSpecific) { > > + // > > + // Don't use null EFI Partition Entries, Invalid Partition Entri= es or OS > > specific > > + // partition Entries > > + // > > + continue; > > + } > > + > > + if (PrivateData->BlockDeviceCount >=3D PEI_FAT_MAX_BLOCK_DEVICE) { > > + break; > > + } > > + > > + Found =3D TRUE; > > + BlockDevPtr =3D &(PrivateData->BlockDevice[Priva= teData- > > >BlockDeviceCount]); > > + > > + BlockDevPtr->BlockSize =3D ParentBlockDev->BlockSize; > > + BlockDevPtr->LastBlock =3D PartitionEntryBuffer[Index].Endi= ngLBA; > > + BlockDevPtr->IoAlign =3D ParentBlockDev->IoAlign; > > + BlockDevPtr->Logical =3D TRUE; > > + BlockDevPtr->PartitionChecked =3D FALSE; > > + BlockDevPtr->StartingPos =3D MultU64x32 ( > > + PartitionEntryBuffer[Index].Star= tingLBA, > > + ParentBlockDev->BlockSize > > + ); > > + BlockDevPtr->ParentDevNo =3D ParentBlockDevNo; > > + > > + PrivateData->BlockDeviceCount++; > > + > > + DEBUG ((DEBUG_INFO, "Find GPT Partition [0x%lx", > > PartitionEntryBuffer[Index].StartingLBA, BlockDevPtr->LastBlock)); > > + DEBUG ((DEBUG_INFO, ", 0x%lx]\n", BlockDevPtr->LastBlock)); > > + DEBUG ((DEBUG_INFO, " BlockSize %x\n", BlockDevPtr- > >BlockSize)); > > + } > > + > > +EXIT: > > + if (PartitionEntryBuffer !=3D NULL) { > > + FreePages (PartitionEntryBuffer, EFI_SIZE_TO_PAGES > > +((UINTN)PartitionEntryArraySize)); > > + } > > + > > + if (PartitionEntryStatus !=3D NULL) { > > + FreePages (PartitionEntryStatus, EFI_SIZE_TO_PAGES > > + (PartHdr->NumberOfPartitionEntries * sizeof > > + (EFI_PARTITION_ENTRY_STATUS))); } > > + > > + return Found; > > +} > > + > > +/** > > + The function is used to check GPT structure, include GPT header and = GPT > > entry array. > > + > > + 1. Check GPT header. > > + 2. Check partition entry array. > > + 3. Check each partitions. > > + > > + @param PrivateData The global memory map > > + @param ParentBlockDevNo The parent block device > > + @param IsPrimary Indicate primary or backup to be check > > + > > + @retval TRUE Primary or backup GPT structure is valid. > > + @retval FALSE Both primary and backup are invalid. > > + > > +**/ > > +BOOLEAN > > +PartitionCheckGptStructure ( > > + IN PEI_FAT_PRIVATE_DATA *PrivateData, > > + IN UINTN ParentBlockDevNo, > > + IN BOOLEAN IsPrimary > > + ) > > +{ > > + EFI_STATUS Status; > > + PEI_FAT_BLOCK_DEVICE *ParentBlockDev; > > + EFI_PARTITION_TABLE_HEADER *PartHdr; > > + EFI_PEI_LBA GptHeaderLBA; > > + > > + ParentBlockDev =3D &(PrivateData->BlockDevice[ParentBlockDevNo]); > > + PartHdr =3D (EFI_PARTITION_TABLE_HEADER *) PrivateData- > >BlockData; > > + > > + if (IsPrimary) { > > + GptHeaderLBA =3D PRIMARY_PART_HEADER_LBA; } else { > > + GptHeaderLBA =3D ParentBlockDev->LastBlock; } > > + > > + Status =3D FatReadBlock ( > > + PrivateData, > > + ParentBlockDevNo, > > + GptHeaderLBA, > > + ParentBlockDev->BlockSize, > > + PartHdr > > + ); > > + if (EFI_ERROR (Status)) { > > + return FALSE; > > + } > > + > > + if (!PartitionCheckGptHeader (PrivateData, ParentBlockDevNo, IsPrima= ry, > > PartHdr)) { > > + return FALSE; > > + } > > + > > + if (!PartitionCheckGptEntryArray (PrivateData, ParentBlockDevNo, > > PartHdr)) { > > + return FALSE; > > + } > > + > > + return TRUE; > > +} > > + > > +/** > > + This function is used to check protective MBR structure before check= ing > > GPT. > > + > > + @param PrivateData The global memory map > > + @param ParentBlockDevNo The parent block device > > + > > + @retval TRUE Valid protective MBR > > + @retval FALSE Invalid MBR > > +**/ > > +BOOLEAN > > +PartitionCheckProtectiveMbr ( > > + IN PEI_FAT_PRIVATE_DATA *PrivateData, > > + IN UINTN ParentBlockDevNo > > + ) > > +{ > > + EFI_STATUS Status; > > + MASTER_BOOT_RECORD *ProtectiveMbr; > > + MBR_PARTITION_ENTRY *MbrPartition; > > + PEI_FAT_BLOCK_DEVICE *ParentBlockDev; > > + UINTN Index; > > + > > + ProtectiveMbr =3D (MASTER_BOOT_RECORD *) PrivateData->BlockData; > > + ParentBlockDev =3D &(PrivateData->BlockDevice[ParentBlockDevNo]); > > + > > + // > > + // Read Protective MBR > > + // > > + Status =3D FatReadBlock ( > > + PrivateData, > > + ParentBlockDevNo, > > + 0, > > + ParentBlockDev->BlockSize, > > + ProtectiveMbr > > + ); > > + if (EFI_ERROR (Status)) { > > + DEBUG ((DEBUG_ERROR, "GPT Error When Read Protective Mbr From > > Partition!\n")); > > + return FALSE; > > + } > > + > > + if (ProtectiveMbr->Signature !=3D MBR_SIGNATURE) { > > + DEBUG ((DEBUG_ERROR, "Protective Mbr Signature is invalid!\n")); > > + return FALSE; > > + } > > + > > + // > > + // The partition define in UEFI Spec Table 17. > > + // Boot Code, Unique MBR Disk Signature, Unknown. > > + // These parst will not used by UEFI, so we skip to check them. > > + // > > + for (Index =3D 0; Index < MAX_MBR_PARTITIONS; Index++) { > > + MbrPartition =3D (MBR_PARTITION_ENTRY *)&ProtectiveMbr- > > >Partition[Index]; > > + if (MbrPartition->BootIndicator =3D=3D 0x00 && > > + MbrPartition->StartingCHS[0] =3D=3D 0x00 && > > + MbrPartition->StartingCHS[1] =3D=3D 0x02 && > > + MbrPartition->StartingCHS[2] =3D=3D 0x00 && > > + MbrPartition->OSType =3D=3D PMBR_GPT_PARTITION && > > + MbrPartition->StartingLBA =3D=3D 0x1 > > + ) { > > + return TRUE; > > + } > > + } > > + > > + DEBUG ((DEBUG_ERROR, "Protective Mbr, All Partition Entry Are > > +Empty!\n")); > > + return FALSE; > > +} > > + > > +/** > > + This function is used for findg GPT partition on block device. > > + As follow UEFI spec We should check protecive MBR first and then > > + try to check both primary/backup GPT structures. > > + > > + @param PrivateData The global memory map > > + @param ParentBlockDevNo The parent block device > > + > > + @retval TRUE New partitions are detected and logical bl= ock > devices > > + are added to block device array > > + @retval FALSE No New partitions are added; > > + > > +**/ > > +BOOLEAN > > +FatFindGptPartitions ( > > + IN PEI_FAT_PRIVATE_DATA *PrivateData, > > + IN UINTN ParentBlockDevNo > > + ) > > +{ > > + BOOLEAN Found; > > + PEI_FAT_BLOCK_DEVICE *ParentBlockDev; > > + > > + if (ParentBlockDevNo > PEI_FAT_MAX_BLOCK_DEVICE - 1) { > > + return FALSE; > > + } > > + > > + ParentBlockDev =3D &(PrivateData->BlockDevice[ParentBlockDevNo]); > > + if (ParentBlockDev->BlockSize > PEI_FAT_MAX_BLOCK_SIZE) { > > + DEBUG ((DEBUG_ERROR, "Device BlockSize %x exceed > > FAT_MAX_BLOCK_SIZE\n", ParentBlockDev->BlockSize)); > > + return FALSE; > > + } > > + > > + if (!PartitionCheckProtectiveMbr (PrivateData, ParentBlockDevNo)) { > > + return FALSE; > > + } > > + > > + Found =3D PartitionCheckGptStructure (PrivateData, ParentBlockDevNo, > > + TRUE); if (!Found) { > > + DEBUG ((DEBUG_ERROR, "Primary GPT Header Error, Try to Check > Backup > > GPT Header!\n")); > > + Found =3D PartitionCheckGptStructure (PrivateData, ParentBlockDevN= o, > > + FALSE); } > > + > > + if (Found) { > > + ParentBlockDev->PartitionChecked =3D TRUE; } > > + > > + return Found; > > +} > > -- > > 2.16.2.windows.1