summaryrefslogtreecommitdiffhomepage
path: root/Private/KernelKit/PEFSharedObject.hxx
blob: 263f08f47f7e9c1f4026ac9806085817a6e1b1f6 (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
/*
 *      ========================================================
 *
 *      HCore
 *      Copyright Mahrouss Logic, all rights reserved.
 *
 *      ========================================================
 */

#ifndef __KERNELKIT_SHARED_OBJECT_HXX__
#define __KERNELKIT_SHARED_OBJECT_HXX__

#include <KernelKit/Loader.hpp>
#include <KernelKit/PEF.hpp>
#include <KernelKit/PEFCodeManager.hxx>
#include <NewKit/Defines.hpp>

namespace HCore {
/// @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 SharedObjectTraits final {
    VoidPtr fImageObject;
    VoidPtr fImageEntrypointOffset;
  };

 public:
  explicit SharedObject() = default;
  ~SharedObject() = default;

 public:
  HCORE_COPY_DEFAULT(SharedObject);

 private:
  SharedObjectTraits *fMounted{nullptr};

 public:
  SharedObjectTraits **GetAddressOf() { return &fMounted; }

  SharedObjectTraits *Get() { return fMounted; }

 public:
  void Mount(SharedObjectTraits *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 HCore

#endif /* ifndef __KERNELKIT_SHARED_OBJECT_HXX__ */