MdeModulePkg/PartitionDxe: Add impl of Partition Information Protocol

[mirror_edk2.git] / MdeModulePkg / Universal / Disk / PartitionDxe / Mbr.c

diff --git a/MdeModulePkg/Universal/Disk/PartitionDxe/Mbr.c b/MdeModulePkg/Universal/Disk/PartitionDxe/Mbr.c
index 377fb1931998cce0e9f2e8fd6ed66e8a48f7559d..55e9d26bae5c8bd0ccae1e005e5b8257f274ad9b 100644 (file)
--- a/MdeModulePkg/Universal/Disk/PartitionDxe/Mbr.c
+++ b/MdeModulePkg/Universal/Disk/PartitionDxe/Mbr.c
@@ -12,7 +12,7 @@
         the legacy boot strap code.\r
 \r
 Copyright (c) 2014, Hewlett-Packard Development Company, L.P.<BR>\r
-Copyright (c) 2006 - 2016, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2017, Intel Corporation. All rights reserved.<BR>\r
 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
@@ -129,18 +129,19 @@ PartitionInstallMbrChildHandles (
   IN  EFI_DEVICE_PATH_PROTOCOL     *DevicePath\r
   )\r
 {\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                   LastBlock;\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                      LastBlock;\r
+  EFI_PARTITION_INFO_PROTOCOL  PartitionInfo;\r
 \r
   Found           = EFI_NOT_FOUND;\r
 \r
@@ -225,6 +226,14 @@ PartitionInstallMbrChildHandles (
       HdDev.PartitionSize   = UNPACK_UINT32 (Mbr->Partition[Index].SizeInLBA);\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
+      CopyMem (&PartitionInfo.Info.Mbr, &Mbr->Partition[Index], sizeof (MBR_PARTITION_RECORD));\r
+\r
       Status = PartitionInstallChildHandle (\r
                 This,\r
                 Handle,\r
@@ -234,10 +243,10 @@ PartitionInstallMbrChildHandles (
                 BlockIo2,\r
                 DevicePath,\r
                 (EFI_DEVICE_PATH_PROTOCOL *) &HdDev,\r
+                &PartitionInfo,\r
                 HdDev.PartitionStart,\r
                 HdDev.PartitionStart + HdDev.PartitionSize - 1,\r
-                MBR_SIZE,\r
-                (BOOLEAN) (Mbr->Partition[Index].OSIndicator == EFI_PARTITION)\r
+                MBR_SIZE\r
                 );\r
 \r
       if (!EFI_ERROR (Status)) {\r
@@ -288,6 +297,14 @@ PartitionInstallMbrChildHandles (
       //\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
+      CopyMem (&PartitionInfo.Info.Mbr, &Mbr->Partition[0], sizeof (MBR_PARTITION_RECORD));\r
+\r
       Status = PartitionInstallChildHandle (\r
                  This,\r
                  Handle,\r
@@ -297,10 +314,10 @@ PartitionInstallMbrChildHandles (
                  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
-                 (BOOLEAN) (Mbr->Partition[0].OSIndicator == EFI_PARTITION)\r
+                 MBR_SIZE\r
                  );\r
       if (!EFI_ERROR (Status)) {\r
         Found = EFI_SUCCESS;\r