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

Signed-off-by: Ruiyu Ni <ruiyu.ni@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@14991 6f19259b-4bc3-4df7-8a09-765794883524

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