ArmPkg/MmCommunicationDxe: fix logic bug in DXE driver entrypoint

[mirror_edk2.git] / ArmPkg / Drivers / MmCommunicationDxe / MmCommunication.c

diff --git a/ArmPkg/Drivers/MmCommunicationDxe/MmCommunication.c b/ArmPkg/Drivers/MmCommunicationDxe/MmCommunication.c
index 5f52afa0fed8d50044162d37da37f04bdb902f1c..a9e06be1adc2d60e84e4ea9d91f558891c98db53 100644 (file)
--- a/ArmPkg/Drivers/MmCommunicationDxe/MmCommunication.c
+++ b/ArmPkg/Drivers/MmCommunicationDxe/MmCommunication.c
@@ -390,8 +390,16 @@ MmCommunicationInitialize (
                     MmGuidedEventNotify, mGuidedEventGuid[Index],\r
                     mGuidedEventGuid[Index], &mGuidedEvent[Index]);\r
     ASSERT_EFI_ERROR (Status);\r
+    if (EFI_ERROR (Status)) {\r
+      while (Index-- > 0) {\r
+        gBS->CloseEvent (mGuidedEvent[Index]);\r
+      }\r
+      goto UninstallProtocol;\r
+    }\r
   }\r
+  return EFI_SUCCESS;\r
 \r
+UninstallProtocol:\r
   gBS->UninstallProtocolInterface (\r
          mMmCommunicateHandle,\r
          &gEfiMmCommunicationProtocolGuid,\r