admin/hypervisor-kli
1
0
Fork 0
mirror of https://github.com/hypervisor/kli synced 2026-08-09 12:40:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1 from hypervisor/caching

Browse Source 
Add function caching
This commit is contained in:
Adrian
2022-09-19 22:22:36 +02:00
committed by GitHub
parent d820850a8d ab0be12bc9
commit e8fd9e9db0
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kli.hpp
+18 -3
View File
@@ -374,7 +374,8 @@ namespace kli {
}
}
KLI_FORCEINLINE uintptr_t find_kernel_export(uint64_t export_hash)
template <uint64_t ExportHash>
KLI_FORCEINLINE uintptr_t find_kernel_export()
{
if (!cache::kernel_base)
cache::kernel_base = detail::find_kernel_base();
@@ -397,13 +398,27 @@ namespace kli {
// address_of_functions is indexed through an ordinal
// address_of_name_ordinals gets the ordinal through our own index - i.
//
if (export_entry_hash == export_hash)
if (export_entry_hash == ExportHash)
return cache::kernel_base + address_of_functions[address_of_name_ordinals[i]];
}
__debugbreak();
return { };
}
template <uint64_t ExportHash>
KLI_FORCEINLINE uintptr_t find_kernel_export_cached()
{
static uintptr_t address = 0;
if (!address)
address = find_kernel_export<ExportHash>();
return address;
}
}
#define KLI_FN(name) ((decltype(&##name))(::kli::find_kernel_export(KLI_HASH_STR(#name))))
#ifdef KLI_DISABLE_CACHE
#define KLI_FN(name) ((decltype(&##name))(::kli::find_kernel_export<KLI_HASH_STR(#name)>()))
#else
#define KLI_FN(name) ((decltype(&##name))(::kli::find_kernel_export_cached<KLI_HASH_STR(#name)>()))
#endif