KernelCache rewrite

1 files changed, 49 insertions, 0 deletions

diff --git a/view/kernelcache/core/KernelCacheController.h b/view/kernelcache/core/KernelCacheController.h
new file mode 100644
index 00000000..71f5e5a9
--- /dev/null
+++ b/view/kernelcache/core/KernelCacheController.h
@@ -0,0 +1,49 @@
+#pragma once
+
+#include <regex>
+
+#include <shared_mutex>
+#include "KernelCache.h"
+#include "refcountobject.h"
+#include "ffi_global.h"
+
+DECLARE_KC_API_OBJECT(BNKernelCacheController, KernelCacheController);
+
+void RegisterKernelCacheControllerDestructor();
+
+namespace BinaryNinja::KC {
+	// Represents the view state for a given `DSCache`
+	class KernelCacheController : public KCRefCountObject
+	{
+		IMPLEMENT_KC_API_OBJECT(BNKernelCacheController);
+		Ref<Logger> m_logger;
+		KernelCache m_cache;
+
+		// Locks on load attempts (region or image).
+		std::shared_mutex m_loadMutex;
+
+		// Store the open images.
+		std::unordered_set<uint64_t> m_loadedImages;
+
+		// Settings from the view.
+		std::regex m_regionFilter;
+
+		explicit KernelCacheController(KernelCache&& cache, Ref<Logger> logger);
+
+	public:
+		// Initialize the DSCacheView, this should be called from the view initialize function only!
+		static KCRef<KernelCacheController> Initialize(BinaryView& view, KernelCache&& cache);
+
+		// NOTE: This will not create one if it does not exist. To create one for the view call `Initialize`.
+		static KCRef<KernelCacheController> FromView(const BinaryView& view);
+
+		KernelCache& GetCache() { return m_cache; };
+		const std::unordered_set<uint64_t>& GetLoadedImages() { return m_loadedImages; };
+
+		// Loads the relevant image info into the view. This does not update analysis so if you
+		// call this make sure at some point you update analysis and likely with linear sweep.
+		bool ApplyImage(BinaryView& view, const CacheImage& image);
+
+		bool IsImageLoaded(const CacheImage& image);
+	};
+}  // namespace BinaryNinja::KC