summaryrefslogtreecommitdiffhomepage
path: root/dev/kernel/KernelKit/PECodeMgr.h
blob: 860f3426545b4515882c0bb7826831977c0dda4a (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
/* -------------------------------------------

  Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.

  File: PECodeMgr.h
  Purpose: PE32+ Code Mgr and Dylib mgr.

  Revision History:

  12/02/24: Added file (amlel)

------------------------------------------- */

#pragma once

////////////////////////////////////////////////////

// LAST REV: Mon Feb 12 13:52:01 CET 2024

////////////////////////////////////////////////////

#include <KernelKit/FileMgr.h>
#include <KernelKit/LoaderInterface.h>
#include <KernelKit/PE.h>
#include <NewKit/ErrorOr.h>
#include <NewKit/KString.h>

#ifndef INC_PROCESS_SCHEDULER_H
#include <KernelKit/ProcessScheduler.h>
#endif

#define kPeApplicationMime "application/vnd-portable-executable"

namespace Kernel {
///
/// \name PE32Loader
/// \brief PE32+ loader class.
///
class PE32Loader : public LoaderInterface {
 private:
  explicit PE32Loader() = delete;

 public:
  explicit PE32Loader(const VoidPtr blob);
  explicit PE32Loader(const Char* path);
  ~PE32Loader() override;

 public:
  NE_COPY_DEFAULT(PE32Loader)

 public:
  const Char* Path() override;
  const Char* AsString() override;
  const Char* MIME() override;

 public:
  ErrorOr<VoidPtr> FindStart() override;
  VoidPtr          FindSymbol(const Char* name, Int32 kind) override;
  ErrorOr<VoidPtr> GetBlob() override;

 public:
  bool IsLoaded() noexcept;

 private:
#ifdef __FSKIT_INCLUDES_NEFS__
  OwnPtr<FileStream<Char, NeFileSystemMgr>> fFile;
#elif defined(__FSKIT_INCLUDES_HEFS__)
  OwnPtr<FileStream<Char, HeFileSystemMgr>> fFile;
#else
  OwnPtr<FileStream<Char>> fFile;
#endif  // __FSKIT_INCLUDES_NEFS__

  Ref<KString> fPath;
  VoidPtr      fCachedBlob;
  bool         fBad;
};
}  // namespace Kernel