From 036de146370820e38354aa0643076d4d90b6111c Mon Sep 17 00:00:00 2001
From: archercreat <archercreat@gmail.com>
Date: Tue, 31 Oct 2023 10:17:15 +0000
Subject: [PATCH] fix callbackmon error

---
 src/plugins/callbackmon/callbackmon.cpp | 3 +--
 src/plugins/callbackmon/callbackmon.h   | 1 +
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/plugins/callbackmon/callbackmon.cpp b/src/plugins/callbackmon/callbackmon.cpp
index 04e9cc439..5abf96297 100644
--- a/src/plugins/callbackmon/callbackmon.cpp
+++ b/src/plugins/callbackmon/callbackmon.cpp
@@ -721,8 +721,7 @@ callbackmon::callbackmon(drakvuf_t drakvuf, const callbackmon_config* config, ou
         });
         return true;
     }, this);
-
-    createSyscallHook("MiProcessLoaderEntry", &callbackmon::load_unload_cb);
+    this->driver_hook = createSyscallHook("MiProcessLoaderEntry", &callbackmon::load_unload_cb);
 }
 
 bool callbackmon::stop_impl()
diff --git a/src/plugins/callbackmon/callbackmon.h b/src/plugins/callbackmon/callbackmon.h
index 6ccbad91a..f71b4d519 100644
--- a/src/plugins/callbackmon/callbackmon.h
+++ b/src/plugins/callbackmon/callbackmon.h
@@ -161,6 +161,7 @@ class callbackmon : public pluginex
     std::unordered_map<addr_t, protocol_cb_t> ndis_protocol_cb;
 
     std::vector<callbackmon_ns::callbackmon_module_t> drivers;
+    std::unique_ptr<libhook::SyscallHook> driver_hook;
 
     virtual bool stop_impl() override;
 };