blob: 71f5e5a98e4aa70e47a21098b5ecb90a3a8565df (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
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
|
