blob: 48b705081a2656684b375cf87dae47498b4e8918 (
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* ========================================================
*
* Kernel
* Copyright ZKA Technologies., all rights reserved.
*
* ========================================================
*/
#ifndef __KERNELKIT_SHARED_OBJECT_HXX__
#define __KERNELKIT_SHARED_OBJECT_HXX__
#include <KernelKit/LoaderInterface.hxx>
#include <KernelKit/PEF.hxx>
#include <KernelKit/PEFCodeManager.hxx>
#include <NewKit/Defines.hxx>
namespace Kernel
{
/// @brief Pure implementation, missing method/function handler.
extern "C" void __mh_purecall(void);
/**
* @brief Shared Library class
* Load library from this class
*/
class PEFSharedObjectInterface final
{
public:
struct PEF_SHARED_OBJECT_TRAITS final
{
VoidPtr fImageObject{nullptr};
VoidPtr fImageEntrypointOffset{nullptr};
Bool IsValid() { return fImageObject && fImageEntrypointOffset; }
};
public:
explicit PEFSharedObjectInterface() = default;
~PEFSharedObjectInterface() = default;
public:
ZKA_COPY_DEFAULT(PEFSharedObjectInterface);
private:
PEF_SHARED_OBJECT_TRAITS* fMounted{nullptr};
public:
PEF_SHARED_OBJECT_TRAITS** GetAddressOf()
{
return &fMounted;
}
PEF_SHARED_OBJECT_TRAITS* Get()
{
return fMounted;
}
public:
void Mount(PEF_SHARED_OBJECT_TRAITS* to_mount)
{
if (!to_mount || !to_mount->fImageObject)
return;
fMounted = to_mount;
if (fLoader && to_mount)
{
delete fLoader;
fLoader = nullptr;
}
if (!fLoader)
{
fLoader = new PEFLoader(fMounted->fImageObject);
}
}
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)__mh_purecall;
return nullptr;
}
return ret;
}
private:
PEFLoader* fLoader{nullptr};
};
typedef PEFSharedObjectInterface* SharedObjectPtr;
EXTERN_C SharedObjectPtr rtl_init_shared_object(PROCESS_HEADER_BLOCK* header);
EXTERN_C Void rtl_fini_shared_object(PROCESS_HEADER_BLOCK* header, SharedObjectPtr lib, Bool* successful);
} // namespace Kernel
#endif /* ifndef __KERNELKIT_SHARED_OBJECT_HXX__ */
|
