MdeModulePkg: Fixed memory corruption when BlockIo2 is not present

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

diff --git a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
index dd621717490d59a51dfe3300bd6f2ea55458dbe1..ed564e5cf3bfcd432dbd49958bff1d005ca9d619 100644 (file)
--- a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
+++ b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
@@ -1053,13 +1053,13 @@ PartitionInstallChildHandle (
   //  for logical partitions.\r
   //\r
   if (Private->BlockIo.Revision >= EFI_BLOCK_IO_PROTOCOL_REVISION2) {\r
-    Private->BlockIo.Media->LowestAlignedLba               = 0;\r
-    Private->BlockIo.Media->LogicalBlocksPerPhysicalBlock  = 0;\r
-    Private->BlockIo2.Media->LowestAlignedLba              = 0;\r
-    Private->BlockIo2.Media->LogicalBlocksPerPhysicalBlock = 0;\r
+    Private->Media.LowestAlignedLba               = 0;\r
+    Private->Media.LogicalBlocksPerPhysicalBlock  = 0;\r
+    Private->Media2.LowestAlignedLba              = 0;\r
+    Private->Media2.LogicalBlocksPerPhysicalBlock = 0;\r
     if (Private->BlockIo.Revision >= EFI_BLOCK_IO_PROTOCOL_REVISION3) {\r
-      Private->BlockIo.Media->OptimalTransferLengthGranularity  = 0;\r
-      Private->BlockIo2.Media->OptimalTransferLengthGranularity = 0;\r
+      Private->Media.OptimalTransferLengthGranularity  = 0;\r
+      Private->Media2.OptimalTransferLengthGranularity = 0;\r
     }\r
   }\r
 \r