diff options
Diffstat (limited to 'Core/CORE_DXE/Partition/Mbr.c')
| -rw-r--r-- | Core/CORE_DXE/Partition/Mbr.c | 407 |
1 files changed, 407 insertions, 0 deletions
diff --git a/Core/CORE_DXE/Partition/Mbr.c b/Core/CORE_DXE/Partition/Mbr.c new file mode 100644 index 0000000..7ccdae8 --- /dev/null +++ b/Core/CORE_DXE/Partition/Mbr.c @@ -0,0 +1,407 @@ +//********************************************************************** +//********************************************************************** +//** ** +//** (C)Copyright 1985-2009, American Megatrends, Inc. ** +//** ** +//** All Rights Reserved. ** +//** ** +//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 ** +//** ** +//** Phone: (770)-246-8600 ** +//** ** +//********************************************************************** +//********************************************************************** + +//********************************************************************** +// $Header: /Alaska/SOURCE/Core/CORE_DXE/Partition/Mbr.c 12 4/16/10 4:24p Pats $ +// +// $Revision: 12 $ +// +// $Date: 4/16/10 4:24p $ +//********************************************************************** +// Revision History +// ---------------- +// $Log: /Alaska/SOURCE/Core/CORE_DXE/Partition/Mbr.c $ +// +// 12 4/16/10 4:24p Pats +// EIP 30719: Support for the HDD with sector size more than 512bytes. +// +// 11 2/04/10 3:03p Oleksiyy +// EIP 31633 Minor Fix +// +// 10 7/09/09 5:01p Oleksiyy +// Files clean-up +// +// 9 11/13/08 6:10p Felixp +// Bug fix in MbrCreateChildHandle: unsed part of the signature (12 bytes) +// in hard driver device path was never initialized. +// +// 8 1/07/08 4:21p Robert +// Updated for coding standard +// +// 7 4/25/07 11:52a Robert +// Updated for CHM and coding standard commpliance +// +// 5 5/19/06 10:47p Felixp +// Device Path code updated to use NEXT_NODE/NODE_LENGTH/SET_NODE_LENGTH +// macros to remove direct access to the Length field +// +// 3 7/12/05 5:01p Robert +// The Free BSD operating system created a partition that referenced +// itself. This was not checked for in the partition driver so an +// infinite loop was created. +// this reference now causes an error condition.. +// +// 2 2/11/05 6:13p Felixp +// Code optimized by using DPAddNode instead of DPAdd +// +// 6 12/21/04 4:54p Markw +// Modified device path defines for consistency. +// +// 5 12/02/04 5:54p Robert +// Partition driver version works +// +//********************************************************************** +//<AMI_FHDR_START> +// +// Name: Mbr.c +// +// Description: EFI Mbr Partition Generic Driver. This file finds all +// the partitions defined in the MBR found on a Device +// +//<AMI_FHDR_END> +//********************************************************************** +//----------------------------------------------------------------------------- +// Includes +#include "partition.h" +#include "mbr.h" + + +//----------------------------------------------------------------------------- +// Global Variables +EFI_GUID gMbrPartition = EFI_PART_TYPE_LEGACY_MBR_GUID; + + + +//----------------------------------------------------------------------------- +// Function Prototypes + + + +//----------------------------------------------------------------------------- +// Function Definitions + +//----------------------------------------------------------------------------- +//<AMI_PHDR_START> +// Procedure: CheckValidMbr +// +// Description: Makes sure the Master Boot Record has valid partitions +// +// Input: +// *BlockIo - Pointer to the parent blockIo Protocol +// *Mbr - pointer to a data structure that contains the MBR for +// the current drive +// +// Output: None +// +// Returns: +// TRUE - MBR is valid +// FALSE - MBR is invalid +// +// Referrals: ReadBlocks +// +//<AMI_PHDR_END> +//----------------------------------------------------------------------------- + +BOOLEAN CheckValidMbr( + IN EFI_BLOCK_IO_PROTOCOL *BlockIo, + IN MASTER_BOOT_RECORD *Mbr +) +{ + EFI_STATUS Status; + EFI_LBA EndLba; + UINT8 PartitionFound = 0; + UINT8 Idx; + UINT8 Loop2; + + // read the first block of the Harddrive + Status = BlockIo->ReadBlocks(BlockIo, BlockIo->Media->MediaId, 0, + BlockIo->Media->BlockSize, Mbr); + + if (Mbr->Sig != MBR_SIGNATURE) + return FALSE; + + // check for valid LBA range for all the partitions + for (Idx = 0; Idx < NUM_MBR_PARTITIONS; Idx++) + { + // ignore partitions whose OSType or Size is zero + if ((Mbr->PartRec[Idx].OSType == 0) + || (Mbr->PartRec[Idx].SizeInLba == 0)) + continue; + + if (Mbr->PartRec[Idx].StartingLba == 0) + continue; + + PartitionFound++; + + EndLba = Mbr->PartRec[Idx].StartingLba + + Mbr->PartRec[Idx].SizeInLba - 1; + + // check to make sure the partition does not exceed the length + // of the disk + if (EndLba > BlockIo->Media->LastBlock) + return FALSE; + + // Now loop through all partitions to make sure they don't overlap + for (Loop2 = Idx+1; Loop2 < NUM_MBR_PARTITIONS; Loop2++) + { + // ignore partitions whose OSType or Size is zero + if ((Mbr->PartRec[Loop2].OSType == 0) + || (Mbr->PartRec[Loop2].SizeInLba == 0)) + continue; + + // check for overlapping partitions + if ( EndLba >= Mbr->PartRec[Loop2].StartingLba ) + return FALSE; + } + } + + if (PartitionFound) + return TRUE; + else + return FALSE; +} + + + +//----------------------------------------------------------------------------- +//<AMI_PHDR_START> +// Procedure: MbrCreateChildHandle +// +// Description: Searches Master Boot Record for partitions +// +// Input: +// *This - Pointer to the instance of this blockIo protocol +// *BlockIo - Pointer to the parent blockIo Protocol +// *DiskIo - Pointer to the parent DiskIo protocol +// *DevicePath - Pointer to the Parent's device path +// ControllerHandle - the parent's controller handle +// +// Output: None +// +// Returns: +// EFI_SUCCESS - Driver loaded +// other - Driver not loaded +// +// Referrals: AllocatePool FreePool CheckValidMbr SetMem ReadBlocks +// CopyMem CreateChildHandle SET_NODE_LENGTH +//<AMI_PHDR_END> +//----------------------------------------------------------------------------- + +EFI_STATUS MbrCreateChildHandle( + IN EFI_DRIVER_BINDING_PROTOCOL *This, + IN EFI_BLOCK_IO_PROTOCOL *BlockIo, + IN EFI_DISK_IO_PROTOCOL *DiskIo, + IN EFI_DEVICE_PATH_PROTOCOL *DevicePath, + IN EFI_HANDLE ControllerHandle +) +{ + EFI_STATUS Status; + MASTER_BOOT_RECORD *Mbr = NULL; + MASTER_BOOT_RECORD *ExtPart = NULL; + UINT8 Idx; + UINT32 PartitionExists = 0; + PARTITION_DATA PData; + EFI_GUID *PartType = NULL; + EFI_LBA Lba; + HARDDRIVE_DEVICE_PATH HdDp; + + + // allocate memory for the MBR and then read the data from the disk + Status = gBS->AllocatePool( EfiBootServicesData, + BlockIo->Media->BlockSize, &Mbr); + + if (EFI_ERROR(Status)) + return EFI_OUT_OF_RESOURCES; + + // check to see if the current drive has a valid MBR and a valid + // partition on the drive + if (CheckValidMbr(BlockIo, Mbr) == FALSE) + { + gBS->FreePool(Mbr); + return EFI_INVALID_PARAMETER; + } + + // To get here there is a valid partition + // create a handle for the partitions + + // set up the device path header for the partition + HdDp.Header.Type = MEDIA_DEVICE_PATH; + HdDp.Header.SubType = MEDIA_HARDDRIVE_DP; + SET_NODE_LENGTH(&HdDp.Header,sizeof(HARDDRIVE_DEVICE_PATH)); + + // Loop through the partitions + for (Idx=0; Idx < NUM_MBR_PARTITIONS; Idx++) + { + // skip over empty partitions + if ((Mbr->PartRec[Idx].OSType == 0) + || (Mbr->PartRec[Idx].SizeInLba == 0)) + continue; + + // Take care of repetative data assignment + // clear the data structure + gBS->SetMem(&PData, sizeof(PARTITION_DATA), 0); + + // set up the device path for the partition + HdDp.MBRType = MBR_TYPE_MASTER_BOOT_RECORD; + HdDp.SignatureType = SIGNATURE_TYPE_MBR; + pBS->SetMem(&HdDp.Signature,sizeof(HdDp.Signature),0); + PData.Handle = NULL; + PData.ParentBlockIo = BlockIo; + PData.ParentDiskIo = DiskIo; + + + // Now that a valid partition is found process it + // TODO: are there any other types of extended partitions + // if so, add them to this if statement + if ((Mbr->PartRec[Idx].OSType == EXTENDED_PARTITION) || + (Mbr->PartRec[Idx].OSType == WIN95_EXTENDED_PARTITION)) + { + // process the logical partitions that may or may not exist + Status = gBS->AllocatePool( EfiBootServicesData, + BlockIo->Media->BlockSize, &ExtPart); + + if (EFI_ERROR(Status)) + { + gBS->FreePool(Mbr); + return EFI_OUT_OF_RESOURCES; + } + + // defines where to start reading the next MBR/partition + // table from + Lba = Mbr->PartRec[Idx].StartingLba; + + // loop through logical partitions: any number of + // possible partitions + while (TRUE) + { + // if the table points back to itself, exit + if ( Lba == 0) + break; + + // get Partition table from the first block of the device + Status = BlockIo->ReadBlocks(BlockIo, BlockIo->Media->MediaId, + Lba, BlockIo->Media->BlockSize, ExtPart); + + // check to see if the disk was read + // Also check for All logical partitions having an OSType + // of 0x05 or EXTENDED_PARTITION + if (EFI_ERROR(Status)) + { + gBS->FreePool(Mbr); + gBS->FreePool(ExtPart); + return Status; + } + + // make sure this is a valid partition + if (ExtPart->PartRec[0].OSType == NO_PARTITION) + break; + + // check for problems that make the partition invalid + if ((Lba + ExtPart->PartRec[0].SizeInLba) > + (Mbr->PartRec[Idx].StartingLba + + Mbr->PartRec[Idx].SizeInLba)) + break; + + // register the Partition + HdDp.PartitionNumber = ++PartitionExists; + + // notice that the start of the partition is after the + // partition table + HdDp.PartitionStart = ExtPart->PartRec[0].StartingLba + Lba; + HdDp.PartitionSize = ExtPart->PartRec[0].SizeInLba; + gBS->CopyMem(&HdDp.Signature, &ExtPart->UniqueMbrSig, + sizeof(UINT32)); + PData.StartingLba = HdDp.PartitionStart; + PData.EndingLba = HdDp.PartitionStart + + ExtPart->PartRec[0].SizeInLba - 1; + + // create device path + PData.DevPath = DPAddNode(DevicePath, &HdDp.Header); + + // Store Parent Handle + PData.ParentHandle = ControllerHandle; + + // if the OSType member is 0xEF this variable set. + // otherwise it is left as 0 + if (ExtPart->PartRec[0].OSType == EFI_SYSTEM_PARTITION) + gBS->CopyMem(&PData.PartGuid, &gMbrPartition, + sizeof (EFI_GUID)); + + // This is a valid GPT partition: Add a child instance + Status = CreateChildHandle(This, &PData, BlockIo->Media->BlockSize); + + // check to see if the next partition is an extended partition + if ((ExtPart->PartRec[1].OSType != EXTENDED_PARTITION) && + (ExtPart->PartRec[1].OSType != WIN95_EXTENDED_PARTITION) ) + break; + + // get set up for the next partition. The Starting Lba + // address is a relative address inside the extended + // partition add the starting address of the extended + // partition to get the actual LBA where it starts + Lba = ExtPart->PartRec[1].StartingLba + + Mbr->PartRec[Idx].StartingLba; + } + } + + else // process primary partition + { + // set up the device path for the partition + HdDp.PartitionNumber = ++PartitionExists; + HdDp.PartitionStart = Mbr->PartRec[Idx].StartingLba; + HdDp.PartitionSize = Mbr->PartRec[Idx].SizeInLba; + gBS->CopyMem(HdDp.Signature, &Mbr->UniqueMbrSig, sizeof(UINT32)); + + // create device path + PData.DevPath = DPAddNode(DevicePath, &HdDp.Header); + + // save pertinent info + PData.StartingLba = Mbr->PartRec[Idx].StartingLba; + PData.EndingLba = Mbr->PartRec[Idx].StartingLba + + Mbr->PartRec[Idx].SizeInLba - 1; + + // Store Parent Handle + PData.ParentHandle = ControllerHandle; + + // if the OSType member is 0xEF is this variable set. + // otherwise it is left as 0 + if (Mbr->PartRec[Idx].OSType == 0xEF) + gBS->CopyMem(&PData.PartGuid, &gMbrPartition, + sizeof (EFI_GUID)); + + // This is a valid GPT partition: Add a child instance + Status = CreateChildHandle(This, &PData, BlockIo->Media->BlockSize); + + } + } + + gBS->FreePool(Mbr); + gBS->FreePool(ExtPart); + + return EFI_SUCCESS; +} + +//********************************************************************** +//********************************************************************** +//** ** +//** (C)Copyright 1985-2009, American Megatrends, Inc. ** +//** ** +//** All Rights Reserved. ** +//** ** +//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 ** +//** ** +//** Phone: (770)-246-8600 ** +//** ** +//********************************************************************** +//********************************************************************** |