]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Disk/PartitionDxe/Mbr.c
UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm
[mirror_edk2.git] / MdeModulePkg / Universal / Disk / PartitionDxe / Mbr.c
index 21d1163ab7819c3238b8be0b62a1aba41bcced42..531b3b45ea0f7f6eb0eafd4dafb4ddf99030c379 100644 (file)
@@ -3,22 +3,18 @@
 \r
   MBR - Master Boot Record is in the first sector of a partitioned hard disk.\r
         The MBR supports four partitions per disk. The MBR also contains legacy\r
-        code that is not run on an EFI system. The legacy code reads the \r
-        first sector of the active partition into memory and \r
+        code that is not run on an EFI system. The legacy code reads the\r
+        first sector of the active partition into memory and\r
 \r
-  BPB - Boot(?) Parameter Block is in the first sector of a FAT file system. \r
-        The BPB contains information about the FAT file system. The BPB is \r
+  BPB - BIOS Parameter Block is in the first sector of a FAT file system.\r
+        The BPB contains information about the FAT file system. The BPB is\r
         always on the first sector of a media. The first sector also contains\r
         the legacy boot strap code.\r
 \r
-Copyright (c) 2006 - 2008, Intel Corporation. <BR>\r
-All rights reserved. This program and the accompanying materials\r
-are licensed and made available under the terms and conditions of the BSD License\r
-which accompanies this distribution.  The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+Copyright (c) 2018 Qualcomm Datacenter Technologies, Inc.\r
+Copyright (c) 2014, Hewlett-Packard Development Company, L.P.<BR>\r
+Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
@@ -29,33 +25,34 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
   @param  Mbr         Parent Handle.\r
   @param  LastLba     Last Lba address on the device.\r
-   \r
+\r
   @retval TRUE        Mbr is a Valid MBR.\r
   @retval FALSE       Mbr is not a Valid MBR.\r
 \r
 **/\r
 BOOLEAN\r
 PartitionValidMbr (\r
-  IN  MASTER_BOOT_RECORD      *Mbr,\r
-  IN  EFI_LBA                 LastLba\r
+  IN  MASTER_BOOT_RECORD  *Mbr,\r
+  IN  EFI_LBA             LastLba\r
   )\r
 {\r
-  UINT32  StartingLBA;\r
-  UINT32  EndingLBA;\r
-  UINT32  NewEndingLBA;\r
-  INTN    Index1;\r
-  INTN    Index2;\r
-  BOOLEAN MbrValid;\r
+  UINT32   StartingLBA;\r
+  UINT32   EndingLBA;\r
+  UINT32   NewEndingLBA;\r
+  INTN     Index1;\r
+  INTN     Index2;\r
+  BOOLEAN  MbrValid;\r
 \r
   if (Mbr->Signature != MBR_SIGNATURE) {\r
     return FALSE;\r
   }\r
+\r
   //\r
   // The BPB also has this signature, so it can not be used alone.\r
   //\r
   MbrValid = FALSE;\r
   for (Index1 = 0; Index1 < MAX_MBR_PARTITIONS; Index1++) {\r
-    if (Mbr->Partition[Index1].OSIndicator == 0x00 || UNPACK_UINT32 (Mbr->Partition[Index1].SizeInLBA) == 0) {\r
+    if ((Mbr->Partition[Index1].OSIndicator == 0x00) || (UNPACK_UINT32 (Mbr->Partition[Index1].SizeInLBA) == 0)) {\r
       continue;\r
     }\r
 \r
@@ -74,16 +71,19 @@ PartitionValidMbr (
       // return FALSE since no block devices on a system are implemented\r
       // with INT 13h\r
       //\r
+\r
+      DEBUG ((DEBUG_INFO, "PartitionValidMbr: Bad MBR partition size EndingLBA(%1x) > LastLBA(%1x)\n", EndingLBA, LastLba));\r
+\r
       return FALSE;\r
     }\r
 \r
     for (Index2 = Index1 + 1; Index2 < MAX_MBR_PARTITIONS; Index2++) {\r
-      if (Mbr->Partition[Index2].OSIndicator == 0x00 || UNPACK_UINT32 (Mbr->Partition[Index2].SizeInLBA) == 0) {\r
+      if ((Mbr->Partition[Index2].OSIndicator == 0x00) || (UNPACK_UINT32 (Mbr->Partition[Index2].SizeInLBA) == 0)) {\r
         continue;\r
       }\r
 \r
       NewEndingLBA = UNPACK_UINT32 (Mbr->Partition[Index2].StartingLBA) + UNPACK_UINT32 (Mbr->Partition[Index2].SizeInLBA) - 1;\r
-      if (NewEndingLBA >= StartingLBA && UNPACK_UINT32 (Mbr->Partition[Index2].StartingLBA) <= EndingLBA) {\r
+      if ((NewEndingLBA >= StartingLBA) && (UNPACK_UINT32 (Mbr->Partition[Index2].StartingLBA) <= EndingLBA)) {\r
         //\r
         // This region overlaps with the Index1'th region\r
         //\r
@@ -91,22 +91,24 @@ PartitionValidMbr (
       }\r
     }\r
   }\r
+\r
   //\r
   // None of the regions overlapped so MBR is O.K.\r
   //\r
   return MbrValid;\r
 }\r
 \r
-\r
 /**\r
   Install child handles if the Handle supports MBR format.\r
 \r
-  @param  This              Calling context.\r
-  @param  Handle            Parent Handle.\r
-  @param  DiskIo            Parent DiskIo interface.\r
-  @param  BlockIo           Parent BlockIo interface.\r
-  @param  DevicePath        Parent Device Path.\r
-   \r
+  @param[in]  This              Calling context.\r
+  @param[in]  Handle            Parent Handle.\r
+  @param[in]  DiskIo            Parent DiskIo interface.\r
+  @param[in]  DiskIo2           Parent DiskIo2 interface.\r
+  @param[in]  BlockIo           Parent BlockIo interface.\r
+  @param[in]  BlockIo2          Parent BlockIo2 interface.\r
+  @param[in]  DevicePath        Parent Device Path.\r
+\r
   @retval EFI_SUCCESS       A child handle was added.\r
   @retval EFI_MEDIA_CHANGED Media change was detected.\r
   @retval Others            MBR partition was not found.\r
@@ -117,43 +119,63 @@ PartitionInstallMbrChildHandles (
   IN  EFI_DRIVER_BINDING_PROTOCOL  *This,\r
   IN  EFI_HANDLE                   Handle,\r
   IN  EFI_DISK_IO_PROTOCOL         *DiskIo,\r
+  IN  EFI_DISK_IO2_PROTOCOL        *DiskIo2,\r
   IN  EFI_BLOCK_IO_PROTOCOL        *BlockIo,\r
+  IN  EFI_BLOCK_IO2_PROTOCOL       *BlockIo2,\r
   IN  EFI_DEVICE_PATH_PROTOCOL     *DevicePath\r
   )\r
 {\r
-  EFI_STATUS                Status;\r
-  MASTER_BOOT_RECORD        *Mbr;\r
-  UINT32                    ExtMbrStartingLba;\r
-  UINTN                     Index;\r
-  HARDDRIVE_DEVICE_PATH     HdDev;\r
-  HARDDRIVE_DEVICE_PATH     ParentHdDev;\r
-  EFI_STATUS                Found;\r
-  UINT32                    PartitionNumber;\r
-  EFI_DEVICE_PATH_PROTOCOL  *DevicePathNode;\r
-  EFI_DEVICE_PATH_PROTOCOL  *LastDevicePathNode;\r
-\r
-  Mbr             = NULL;\r
-  Found           = EFI_NOT_FOUND;\r
-\r
-  Mbr             = AllocatePool (BlockIo->Media->BlockSize);\r
+  EFI_STATUS                   Status;\r
+  MASTER_BOOT_RECORD           *Mbr;\r
+  UINT32                       ExtMbrStartingLba;\r
+  UINT32                       Index;\r
+  HARDDRIVE_DEVICE_PATH        HdDev;\r
+  HARDDRIVE_DEVICE_PATH        ParentHdDev;\r
+  EFI_STATUS                   Found;\r
+  EFI_DEVICE_PATH_PROTOCOL     *DevicePathNode;\r
+  EFI_DEVICE_PATH_PROTOCOL     *LastDevicePathNode;\r
+  UINT32                       BlockSize;\r
+  UINT32                       MediaId;\r
+  EFI_LBA                      LastSector;\r
+  EFI_PARTITION_INFO_PROTOCOL  PartitionInfo;\r
+\r
+  Found = EFI_NOT_FOUND;\r
+\r
+  BlockSize  = BlockIo->Media->BlockSize;\r
+  MediaId    = BlockIo->Media->MediaId;\r
+  LastSector = DivU64x32 (\r
+                 MultU64x32 (BlockIo->Media->LastBlock + 1, BlockSize),\r
+                 MBR_SIZE\r
+                 ) - 1;\r
+\r
+  //\r
+  // Ensure the block size can hold the MBR\r
+  //\r
+  if (BlockSize < sizeof (MASTER_BOOT_RECORD)) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  Mbr = AllocatePool (BlockSize);\r
   if (Mbr == NULL) {\r
-    goto Done;\r
+    return Found;\r
   }\r
 \r
-  Status = BlockIo->ReadBlocks (\r
-                      BlockIo,\r
-                      BlockIo->Media->MediaId,\r
-                      0,\r
-                      BlockIo->Media->BlockSize,\r
-                      Mbr\r
-                      );\r
+  Status = DiskIo->ReadDisk (\r
+                     DiskIo,\r
+                     MediaId,\r
+                     0,\r
+                     BlockSize,\r
+                     Mbr\r
+                     );\r
   if (EFI_ERROR (Status)) {\r
     Found = Status;\r
     goto Done;\r
   }\r
-  if (!PartitionValidMbr (Mbr, BlockIo->Media->LastBlock)) {\r
+\r
+  if (!PartitionValidMbr (Mbr, LastSector)) {\r
     goto Done;\r
   }\r
+\r
   //\r
   // We have a valid mbr - add each partition\r
   //\r
@@ -164,35 +186,34 @@ PartitionInstallMbrChildHandles (
   ZeroMem (&ParentHdDev, sizeof (ParentHdDev));\r
   DevicePathNode = DevicePath;\r
   while (!IsDevicePathEnd (DevicePathNode)) {\r
-    LastDevicePathNode  = DevicePathNode;\r
-    DevicePathNode      = NextDevicePathNode (DevicePathNode);\r
+    LastDevicePathNode = DevicePathNode;\r
+    DevicePathNode     = NextDevicePathNode (DevicePathNode);\r
   }\r
 \r
   if (LastDevicePathNode != NULL) {\r
-    if (DevicePathType (LastDevicePathNode) == MEDIA_DEVICE_PATH &&\r
-        DevicePathSubType (LastDevicePathNode) == MEDIA_HARDDRIVE_DP\r
-        ) {\r
+    if ((DevicePathType (LastDevicePathNode) == MEDIA_DEVICE_PATH) &&\r
+        (DevicePathSubType (LastDevicePathNode) == MEDIA_HARDDRIVE_DP)\r
+        )\r
+    {\r
       CopyMem (&ParentHdDev, LastDevicePathNode, sizeof (ParentHdDev));\r
     } else {\r
       LastDevicePathNode = NULL;\r
     }\r
   }\r
 \r
-  PartitionNumber = 1;\r
-\r
   ZeroMem (&HdDev, sizeof (HdDev));\r
-  HdDev.Header.Type     = MEDIA_DEVICE_PATH;\r
-  HdDev.Header.SubType  = MEDIA_HARDDRIVE_DP;\r
+  HdDev.Header.Type    = MEDIA_DEVICE_PATH;\r
+  HdDev.Header.SubType = MEDIA_HARDDRIVE_DP;\r
   SetDevicePathNodeLength (&HdDev.Header, sizeof (HdDev));\r
-  HdDev.MBRType         = MBR_TYPE_PCAT;\r
-  HdDev.SignatureType   = SIGNATURE_TYPE_MBR;\r
+  HdDev.MBRType       = MBR_TYPE_PCAT;\r
+  HdDev.SignatureType = SIGNATURE_TYPE_MBR;\r
 \r
   if (LastDevicePathNode == NULL) {\r
     //\r
     // This is a MBR, add each partition\r
     //\r
     for (Index = 0; Index < MAX_MBR_PARTITIONS; Index++) {\r
-      if (Mbr->Partition[Index].OSIndicator == 0x00 || UNPACK_UINT32 (Mbr->Partition[Index].SizeInLBA) == 0) {\r
+      if ((Mbr->Partition[Index].OSIndicator == 0x00) || (UNPACK_UINT32 (Mbr->Partition[Index].SizeInLBA) == 0)) {\r
         //\r
         // Don't use null MBR entries\r
         //\r
@@ -202,30 +223,42 @@ PartitionInstallMbrChildHandles (
       if (Mbr->Partition[Index].OSIndicator == PMBR_GPT_PARTITION) {\r
         //\r
         // This is the guard MBR for the GPT. If you ever see a GPT disk with zero partitions you can get here.\r
-        //  We can not produce an MBR BlockIo for this device as the MBR spans the GPT headers. So formating \r
+        //  We can not produce an MBR BlockIo for this device as the MBR spans the GPT headers. So formating\r
         //  this BlockIo would corrupt the GPT structures and require a recovery that would corrupt the format\r
-        //  that corrupted the GPT partition. \r
+        //  that corrupted the GPT partition.\r
         //\r
         continue;\r
       }\r
 \r
-      HdDev.PartitionNumber = PartitionNumber ++;\r
+      HdDev.PartitionNumber = Index + 1;\r
       HdDev.PartitionStart  = UNPACK_UINT32 (Mbr->Partition[Index].StartingLBA);\r
       HdDev.PartitionSize   = UNPACK_UINT32 (Mbr->Partition[Index].SizeInLBA);\r
-      CopyMem (HdDev.Signature, &(Mbr->UniqueMbrSignature[0]), sizeof (UINT32));\r
+      CopyMem (HdDev.Signature, &(Mbr->UniqueMbrSignature[0]), sizeof (Mbr->UniqueMbrSignature));\r
+\r
+      ZeroMem (&PartitionInfo, sizeof (EFI_PARTITION_INFO_PROTOCOL));\r
+      PartitionInfo.Revision = EFI_PARTITION_INFO_PROTOCOL_REVISION;\r
+      PartitionInfo.Type     = PARTITION_TYPE_MBR;\r
+      if (Mbr->Partition[Index].OSIndicator == EFI_PARTITION) {\r
+        PartitionInfo.System = 1;\r
+      }\r
+\r
+      CopyMem (&PartitionInfo.Info.Mbr, &Mbr->Partition[Index], sizeof (MBR_PARTITION_RECORD));\r
 \r
       Status = PartitionInstallChildHandle (\r
-                This,\r
-                Handle,\r
-                DiskIo,\r
-                BlockIo,\r
-                DevicePath,\r
-                (EFI_DEVICE_PATH_PROTOCOL *) &HdDev,\r
-                HdDev.PartitionStart,\r
-                HdDev.PartitionStart + HdDev.PartitionSize - 1,\r
-                MBR_SIZE,\r
-                (BOOLEAN) (Mbr->Partition[Index].OSIndicator == EFI_PARTITION)\r
-                );\r
+                 This,\r
+                 Handle,\r
+                 DiskIo,\r
+                 DiskIo2,\r
+                 BlockIo,\r
+                 BlockIo2,\r
+                 DevicePath,\r
+                 (EFI_DEVICE_PATH_PROTOCOL *)&HdDev,\r
+                 &PartitionInfo,\r
+                 HdDev.PartitionStart,\r
+                 HdDev.PartitionStart + HdDev.PartitionSize - 1,\r
+                 MBR_SIZE,\r
+                 ((Mbr->Partition[Index].OSIndicator == EFI_PARTITION) ? &gEfiPartTypeSystemPartGuid : NULL)\r
+                 );\r
 \r
       if (!EFI_ERROR (Status)) {\r
         Found = EFI_SUCCESS;\r
@@ -236,17 +269,17 @@ PartitionInstallMbrChildHandles (
     // It's an extended partition. Follow the extended partition\r
     // chain to get all the logical drives\r
     //\r
+    Index             = 0;\r
     ExtMbrStartingLba = 0;\r
 \r
     do {\r
-\r
-      Status = BlockIo->ReadBlocks (\r
-                          BlockIo,\r
-                          BlockIo->Media->MediaId,\r
-                          ExtMbrStartingLba,\r
-                          BlockIo->Media->BlockSize,\r
-                          Mbr\r
-                          );\r
+      Status = DiskIo->ReadDisk (\r
+                         DiskIo,\r
+                         MediaId,\r
+                         MultU64x32 (ExtMbrStartingLba, BlockSize),\r
+                         BlockSize,\r
+                         Mbr\r
+                         );\r
       if (EFI_ERROR (Status)) {\r
         Found = Status;\r
         goto Done;\r
@@ -257,42 +290,65 @@ PartitionInstallMbrChildHandles (
       }\r
 \r
       if ((Mbr->Partition[0].OSIndicator == EXTENDED_DOS_PARTITION) ||\r
-          (Mbr->Partition[0].OSIndicator == EXTENDED_WINDOWS_PARTITION)) {\r
+          (Mbr->Partition[0].OSIndicator == EXTENDED_WINDOWS_PARTITION))\r
+      {\r
         ExtMbrStartingLba = UNPACK_UINT32 (Mbr->Partition[0].StartingLBA);\r
+        //\r
+        // A value of 0 is invalid for StartingLBA\r
+        //\r
+        if (ExtMbrStartingLba == 0) {\r
+          break;\r
+        }\r
+\r
         continue;\r
       }\r
-      HdDev.PartitionNumber = PartitionNumber ++;\r
+\r
+      HdDev.PartitionNumber = ++Index;\r
       HdDev.PartitionStart  = UNPACK_UINT32 (Mbr->Partition[0].StartingLBA) + ExtMbrStartingLba + ParentHdDev.PartitionStart;\r
       HdDev.PartitionSize   = UNPACK_UINT32 (Mbr->Partition[0].SizeInLBA);\r
       if ((HdDev.PartitionStart + HdDev.PartitionSize - 1 >= ParentHdDev.PartitionStart + ParentHdDev.PartitionSize) ||\r
-          (HdDev.PartitionStart <= ParentHdDev.PartitionStart)) {\r
+          (HdDev.PartitionStart <= ParentHdDev.PartitionStart))\r
+      {\r
         break;\r
       }\r
 \r
       //\r
       // The signature in EBR(Extended Boot Record) should always be 0.\r
       //\r
-      *((UINT32 *) &HdDev.Signature[0]) = 0;\r
+      *((UINT32 *)&HdDev.Signature[0]) = 0;\r
+\r
+      ZeroMem (&PartitionInfo, sizeof (EFI_PARTITION_INFO_PROTOCOL));\r
+      PartitionInfo.Revision = EFI_PARTITION_INFO_PROTOCOL_REVISION;\r
+      PartitionInfo.Type     = PARTITION_TYPE_MBR;\r
+      if (Mbr->Partition[0].OSIndicator == EFI_PARTITION) {\r
+        PartitionInfo.System = 1;\r
+      }\r
+\r
+      CopyMem (&PartitionInfo.Info.Mbr, &Mbr->Partition[0], sizeof (MBR_PARTITION_RECORD));\r
 \r
       Status = PartitionInstallChildHandle (\r
-                This,\r
-                Handle,\r
-                DiskIo,\r
-                BlockIo,\r
-                DevicePath,\r
-                (EFI_DEVICE_PATH_PROTOCOL *) &HdDev,\r
-                HdDev.PartitionStart - ParentHdDev.PartitionStart,\r
-                HdDev.PartitionStart - ParentHdDev.PartitionStart + HdDev.PartitionSize - 1,\r
-                MBR_SIZE,\r
-                (BOOLEAN) (Mbr->Partition[0].OSIndicator == EFI_PARTITION)\r
-                );\r
+                 This,\r
+                 Handle,\r
+                 DiskIo,\r
+                 DiskIo2,\r
+                 BlockIo,\r
+                 BlockIo2,\r
+                 DevicePath,\r
+                 (EFI_DEVICE_PATH_PROTOCOL *)&HdDev,\r
+                 &PartitionInfo,\r
+                 HdDev.PartitionStart - ParentHdDev.PartitionStart,\r
+                 HdDev.PartitionStart - ParentHdDev.PartitionStart + HdDev.PartitionSize - 1,\r
+                 MBR_SIZE,\r
+                 ((Mbr->Partition[0].OSIndicator == EFI_PARTITION) ? &gEfiPartTypeSystemPartGuid : NULL)\r
+                 );\r
       if (!EFI_ERROR (Status)) {\r
         Found = EFI_SUCCESS;\r
       }\r
 \r
       if ((Mbr->Partition[1].OSIndicator != EXTENDED_DOS_PARTITION) &&\r
           (Mbr->Partition[1].OSIndicator != EXTENDED_WINDOWS_PARTITION)\r
-          ) {\r
+          )\r
+      {\r
         break;\r
       }\r
 \r