IPEFDylibObject.h

Go to the documentation of this file.

/*
 *      ========================================================
 *
 *      Kernel
 *      Copyright (C) 2024-2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
 *
 *      ========================================================
 */
 
#ifndef __KERNELKIT_PEF_SHARED_OBJECT_H__
#define __KERNELKIT_PEF_SHARED_OBJECT_H__
 
#include <KernelKit/IDylibObject.h>
#include <KernelKit/PEF.h>
#include <KernelKit/PEFCodeMgr.h>
#include <KernelKit/ProcessScheduler.h>
#include <NeKit/Config.h>
 
namespace Kernel {
class IPEFDylibObject final NE_DYLIB_OBJECT {
 public:
  explicit IPEFDylibObject() = default;
  ~IPEFDylibObject()         = default;
 
 public:
  NE_COPY_DEFAULT(IPEFDylibObject)
 
 private:
  DylibTraits* fMounted{nullptr};
 
 public:
  DylibTraits** GetAddressOf() { return &fMounted; }
 
  DylibTraits* Get() { return fMounted; }
 
 public:
  void Mount(DylibTraits* 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, const SizeT& len, const UInt32& 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).Leak().Leak());
 
    if (!ret) {
      if (kind == kPefCode) return (VoidPtr) &__ne_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
 
#endif /* ifndef __KERNELKIT_PEF_SHARED_OBJECT_H__ */