fix callbackmon error

tklengyel · Oct 31, 2023 · 036de14 · 036de14
1 parent 85ae67b
commit 036de14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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;
 };