summaryrefslogtreecommitdiffhomepage
path: root/dev/Kernel/src/IPEFDylibObject.cc
blob: a28f2a945c0ca93ce69acb3cb7230d1cfa6b4dcf (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
/*
 * ========================================================
 *
 * minoskrnl
 * Copyright (C) 2024, Amlal EL Mahrouss, all rights reserved., all rights reserved.
 *
 *  ========================================================
 */

#include <KernelKit/DebugOutput.h>
#include <KernelKit/PEF.h>
#include <KernelKit/IPEFDylibObject.h>
#include <KernelKit/UserProcessScheduler.h>
#include <KernelKit/ThreadLocalStorage.h>
#include <NewKit/Defines.h>

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

 Revision History:

	 01/02/24: Reworked dll ABI, expect a rtl_init_dylib and
 rtl_fini_dylib (amlel) 15/02/24: Breaking changes, changed the name of the
 routines. (amlel)

	07/28/24: Replace rt_library_free with rtl_fini_dylib

	10/8/24: FIX: Fix log comment.

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

using namespace Kernel;

/***********************************************************************************/
/// @file IPEFDylibObject.cc
/// @brief PEF's Dylib runtime.
/***********************************************************************************/

/***********************************************************************************/
/** @brief Library initializer. */
/***********************************************************************************/

EXTERN_C IDylibRef rtl_init_dylib(UserProcess& thread)
{
	IDylibRef dll_obj = tls_new_class<IPEFDylibObject>();

	if (!dll_obj)
	{
		thread.Crash();
		return nullptr;
	}

	dll_obj->Mount(new IPEFDylibObject::DLL_TRAITS());

	if (!dll_obj->Get())
	{
		tls_delete_class(dll_obj);
		thread.Crash();

		return nullptr;
	}

	dll_obj->Get()->ImageObject =
		thread.Image.fBlob;

	if (!dll_obj->Get()->ImageObject)
	{
		tls_delete_class(dll_obj);
		thread.Crash();

		return nullptr;
	}

	dll_obj->Get()->ImageEntrypointOffset =
		dll_obj->Load<VoidPtr>(kPefStart, rt_string_len(kPefStart, 0), kPefCode);

	return dll_obj;
}

/***********************************************************************************/
/** @brief Frees the dll_obj. */
/** @note Please check if the dll_obj got freed! */
/** @param dll_obj The dll_obj to free. */
/** @param successful Reports if successful or not. */
/***********************************************************************************/

EXTERN_C Void rtl_fini_dylib(UserProcess& thread, IDylibRef dll_obj, Bool* successful)
{
	MUST_PASS(successful);

	// sanity check (will also trigger a bug check if this fails)
	if (dll_obj == nullptr)
	{
		*successful = false;
		thread.Crash();
	}

	delete dll_obj->Get();
	delete dll_obj;

	dll_obj = nullptr;

	*successful = true;
}