From 25952ff9d8810bd37db8dad39322967bab0f1a25 Mon Sep 17 00:00:00 2001
From: Mykola Grymalyuk <khronokernel@icloud.com>
Date: Fri, 7 Oct 2022 18:18:24 -0600
Subject: [PATCH] kdk_handler: Verify checksum after downloading

---
 resources/kdk_handler.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/resources/kdk_handler.py b/resources/kdk_handler.py
index a2c10d6f2..413ac083d 100644
--- a/resources/kdk_handler.py
+++ b/resources/kdk_handler.py
@@ -180,6 +180,13 @@ class kernel_debug_kit_handler:
             return False, msg, ""
 
         if utilities.download_apple_developer_portal(download_link, self.constants.kdk_download_path):
+            result = subprocess.run(["hdiutil", "verify", self.constants.kdk_download_path], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            if result.returncode != 0:
+                print(f"Error: Kernel Debug Kit checksum verification failed!")
+                print(f"Output: {result.stderr}")
+                msg = "Kernel Debug Kit checksum verification failed, please try again.\n\nIf this continues to fail, ensure you're downloading on a stable network connection (ie. Ethernet)"
+                print(f"- {msg}")
+                return False, msg, ""
             self.remove_unused_kdks(detected_build)
             return True, "", detected_build
         msg = "Failed to download KDK"