blob: dbe9cda80850ce5447dd96cb5262a93d8318aca3 (
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
|
/*
* ========================================================
*
* NewOS
* Copyright Zeta Electronics Corporation, all rights reserved.
*
* ========================================================
*/
#ifndef __KERNELKIT_SHARED_OBJECT_HXX__
#define __KERNELKIT_SHARED_OBJECT_HXX__
#include <KernelKit/LoaderInterface.hpp>
#include <KernelKit/PEF.hpp>
#include <KernelKit/PEFCodeManager.hxx>
#include <NewKit/Defines.hpp>
namespace NewOS
{
/// @brief Pure implementation, missing method/function handler.
extern "C" void __mh_purecall(void);
/**
* @brief Shared Library class
* Load library from this class
*/
class SharedObject final
{
public:
struct SharedObjectTrait final
{
VoidPtr fImageObject;
VoidPtr fImageEntrypointOffset;
};
public:
explicit SharedObject() = default;
~SharedObject() = default;
public:
NEWOS_COPY_DEFAULT(SharedObject);
private:
SharedObjectTrait* fMounted{nullptr};
public:
SharedObjectTrait** GetAddressOf()
{
return &fMounted;
}
SharedObjectTrait* Get()
{
return fMounted;
}
public:
void Mount(SharedObjectTrait* 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 SharedObject* SharedObjectPtr;
} // namespace NewOS
#endif /* ifndef __KERNELKIT_SHARED_OBJECT_HXX__ */
|
