Add the missing assignment of EFI_DISK_IO2_TOKEN::Token in FlushDiskEx() implementati...

[mirror_edk2.git] / MdeModulePkg / Universal / Disk / DiskIoDxe / DiskIo.c

diff --git a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
index f53c3d91d030b64bed1e0437f1786d90e7a0aaf5..eb3edeb53cfddc2783bc4ecaa935c6902477c2f8 100644 (file)
--- a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
+++ b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
@@ -1111,6 +1111,7 @@ DiskIo2FlushDiskEx (
       return Status;\r
     }\r
     Task->Signature = DISK_IO2_FLUSH_TASK_SIGNATURE;\r
+    Task->Token     = Token;\r
     Status = Private->BlockIo2->FlushBlocksEx (Private->BlockIo2, &Task->BlockIo2Token);\r
     if (EFI_ERROR (Status)) {\r
       gBS->CloseEvent (Task->BlockIo2Token.Event);\r