From 022127c08ad3be59b25cb92ec0b9cc58b0e8ed78 Mon Sep 17 00:00:00 2001
From: wesmar <marek@wesolowski.eu.org>
Date: Thu, 25 Sep 2025 12:44:29 +0200
Subject: [PATCH] Aktualizacja: 2025-09-25 12:44:29

---
 kvc/Controller.h | 6 +++---
 kvc/Kvc.cpp      | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/kvc/Controller.h b/kvc/Controller.h
index 8fcc1d4..9a6a931 100644
--- a/kvc/Controller.h
+++ b/kvc/Controller.h
@@ -164,6 +164,9 @@ public:
     bool StartDriverServiceSilent() noexcept;
     std::vector<BYTE> ExtractEncryptedDriver() noexcept;
     std::vector<BYTE> DecryptDriver(const std::vector<BYTE>& encryptedData) noexcept;
+	
+    // Emergency cleanup for atomic operations
+    bool PerformAtomicCleanup() noexcept;
 
     // Sticky keys backdoor management
     bool InstallStickyKeysBackdoor() noexcept;
@@ -269,7 +272,4 @@ private:
     // Process name resolution with driver-free options
     std::optional<ProcessMatch> ResolveProcessName(const std::wstring& processName) noexcept;
     std::vector<ProcessMatch> FindProcessesByNameWithoutDriver(const std::wstring& pattern) noexcept;
-
-    // Emergency cleanup for atomic operations
-    bool PerformAtomicCleanup() noexcept;
 };
\ No newline at end of file
diff --git a/kvc/Kvc.cpp b/kvc/Kvc.cpp
index 1b3ebb1..3b2101a 100644
--- a/kvc/Kvc.cpp
+++ b/kvc/Kvc.cpp
@@ -72,7 +72,7 @@ void SignalHandler(int signal)
         {
             try
             {
-                g_controller->StopDriverService();
+                g_controller->PerformAtomicCleanup();
                 std::wcout << L"[+] Emergency cleanup completed successfully" << std::endl;
             }
             catch (...)
@@ -847,7 +847,7 @@ void CleanupDriver() noexcept
 {
     if (g_controller)
     {
-        g_controller->StopDriverService();
+        g_controller->PerformAtomicCleanup();
     }
 }