summaryrefslogtreecommitdiffhomepage
path: root/dev/kernel/KernelKit/IPEFDylibObject.h
diff options
context:
space:
mode:
Diffstat (limited to 'dev/kernel/KernelKit/IPEFDylibObject.h')
-rw-r--r--dev/kernel/KernelKit/IPEFDylibObject.h156
1 files changed, 68 insertions, 88 deletions
diff --git a/dev/kernel/KernelKit/IPEFDylibObject.h b/dev/kernel/KernelKit/IPEFDylibObject.h
index a708fb6c..4031bd85 100644
--- a/dev/kernel/KernelKit/IPEFDylibObject.h
+++ b/dev/kernel/KernelKit/IPEFDylibObject.h
@@ -10,97 +10,77 @@
#ifndef __KERNELKIT_SHARED_OBJECT_H__
#define __KERNELKIT_SHARED_OBJECT_H__
+#include <KernelKit/IDylibObject.h>
#include <KernelKit/PEF.h>
-#include <NewKit/Defines.h>
#include <KernelKit/PEFCodeMgr.h>
#include <KernelKit/ProcessScheduler.h>
-#include <KernelKit/IDylibObject.h>
+#include <NewKit/Defines.h>
+
+namespace Kernel {
+/**
+ * @brief Shared Library class
+ * Load library from this class
+ */
+class IPEFDylibObject final NE_DYLIB_OBJECT {
+ public:
+ explicit IPEFDylibObject() = default;
+ ~IPEFDylibObject() = default;
+
+ public:
+ NE_COPY_DEFAULT(IPEFDylibObject)
+
+ private:
+ DLL_TRAITS* fMounted{nullptr};
+
+ public:
+ DLL_TRAITS** GetAddressOf() { return &fMounted; }
+
+ DLL_TRAITS* Get() { return fMounted; }
+
+ public:
+ void Mount(DLL_TRAITS* to_mount) {
+ if (!to_mount || !to_mount->ImageObject) return;
+
+ fMounted = to_mount;
+
+ if (fLoader && to_mount) {
+ delete fLoader;
+ fLoader = nullptr;
+ }
+
+ if (!fLoader) {
+ fLoader = new PEFLoader(fMounted->ImageObject);
+ }
+ }
+
+ void Unmount() {
+ if (fMounted) fMounted = nullptr;
+ };
+
+ template <typename SymbolType>
+ SymbolType Load(const Char* symbol_name, SizeT len, Int32 kind) {
+ if (symbol_name == nullptr || *symbol_name == 0) return nullptr;
+ if (len > kPathLen || len < 1) return nullptr;
+
+ auto ret = reinterpret_cast<SymbolType>(fLoader->FindSymbol(symbol_name, kind));
+
+ if (!ret) {
+ if (kind == kPefCode) return (VoidPtr) &__zka_pure_call;
+
+ return nullptr;
+ }
+
+ return ret;
+ }
+
+ private:
+ PEFLoader* fLoader{nullptr};
+};
+
+typedef IPEFDylibObject* IDylibRef;
-namespace Kernel
-{
- /**
- * @brief Shared Library class
- * Load library from this class
- */
- class IPEFDylibObject final NE_DYLIB_OBJECT
- {
- public:
- explicit IPEFDylibObject() = default;
- ~IPEFDylibObject() = default;
-
- public:
- NE_COPY_DEFAULT(IPEFDylibObject)
-
- private:
- DLL_TRAITS* fMounted{nullptr};
-
- public:
- DLL_TRAITS** GetAddressOf()
- {
- return &fMounted;
- }
-
- DLL_TRAITS* Get()
- {
- return fMounted;
- }
-
- public:
- void Mount(DLL_TRAITS* to_mount)
- {
- if (!to_mount || !to_mount->ImageObject)
- return;
-
- fMounted = to_mount;
-
- if (fLoader && to_mount)
- {
- delete fLoader;
- fLoader = nullptr;
- }
-
- if (!fLoader)
- {
- fLoader = new PEFLoader(fMounted->ImageObject);
- }
- }
-
- void Unmount()
- {
- if (fMounted)
- fMounted = nullptr;
- };
-
- template <typename SymbolType>
- SymbolType Load(const Char* symbol_name, SizeT len, Int32 kind)
- {
- if (symbol_name == nullptr || *symbol_name == 0)
- return nullptr;
- if (len > kPathLen || len < 1)
- return nullptr;
-
- auto ret =
- reinterpret_cast<SymbolType>(fLoader->FindSymbol(symbol_name, kind));
-
- if (!ret)
- {
- if (kind == kPefCode)
- return (VoidPtr)&__zka_pure_call;
-
- return nullptr;
- }
-
- return ret;
- }
-
- private:
- PEFLoader* fLoader{nullptr};
- };
-
- typedef IPEFDylibObject* IDylibRef;
-
- EXTERN_C IDylibRef rtl_init_dylib_pef(USER_PROCESS& header);
- EXTERN_C Void rtl_fini_dylib_pef(USER_PROCESS& header, IDylibRef lib, Bool* successful);
-} // namespace Kernel
+EXTERN_C IDylibRef rtl_init_dylib_pef(USER_PROCESS& header);
+EXTERN_C Void rtl_fini_dylib_pef(USER_PROCESS& header, IDylibRef lib, Bool* successful);
+} // namespace Kernel
#endif /* ifndef __KERNELKIT_SHARED_OBJECT_H__ */