From d7d203c6401ddbabb3582498e31de1857d0bb379 Mon Sep 17 00:00:00 2001
From: Mattiwatti <mattiwatti@gmail.com>
Date: Mon, 6 May 2019 20:24:05 +0200
Subject: Call driver unload if a non-Windows OS is being booted

---
 EfiGuardDxe/EfiGuardDxe.c | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/EfiGuardDxe/EfiGuardDxe.c b/EfiGuardDxe/EfiGuardDxe.c
index df2528f..6d9dd2c 100644
--- a/EfiGuardDxe/EfiGuardDxe.c
+++ b/EfiGuardDxe/EfiGuardDxe.c
@@ -15,6 +15,15 @@ EFI_DRIVER_SUPPORTED_EFI_VERSION_PROTOCOL gEfiGuardSupportedEfiVersion =
 	EFI_2_10_SYSTEM_TABLE_REVISION
 };
 
+//
+// Driver unload
+//
+EFI_STATUS
+EFIAPI
+EfiGuardUnload(
+	IN EFI_HANDLE ImageHandle
+	);
+
 //
 // EfiGuard driver protocol
 //
@@ -205,6 +214,11 @@ HookedLoadImage(
 								LoadedImage->ImageBase,
 								LoadedImage->ImageSize);
 			}
+			else
+			{
+				// A non-Windows OS is being booted. Unload ourselves
+				EfiGuardUnload(gImageHandle);
+			}
 		}
 	}
 
-- 
cgit v1.2.3