[IMP] BootJump has been fixed, LoaderUtils API for CFKit. (Kernel's CoreFoundation like API)

[IMP] Add Write for UChar* types. (BTextWriter)

Signed-off-by: Amlal <amlal@zka.com>

1 files changed, 54 insertions, 0 deletions

diff --git a/Kernel/CFKit/LoaderUtils.hxx b/Kernel/CFKit/LoaderUtils.hxx
new file mode 100644
index 00000000..3edacc67
--- /dev/null
+++ b/Kernel/CFKit/LoaderUtils.hxx
@@ -0,0 +1,54 @@
+#ifndef __CFKIT_LOADER_UTILS_HXX__
+#define __CFKIT_LOADER_UTILS_HXX__
+
+#include <KernelKit/PE.hxx>
+#include <KernelKit/MSDOS.hxx>
+
+namespace Kernel
+{
+	/// @brief Find the PE header inside the blob.
+	inline auto ldr_find_exec_header(DosHeaderPtr ptrDos) -> ExecHeaderPtr
+	{
+		if (!ptrDos)
+			return nullptr;
+
+		if (ptrDos->eMagic[0] != kMagMz0)
+			return nullptr;
+
+		if (ptrDos->eMagic[1] != kMagMz1)
+			return nullptr;
+
+		return (ExecHeaderPtr)(VoidPtr)(&ptrDos->eLfanew + 1);
+	}
+
+	/// @brief Find the PE optional header inside the blob.
+	inline auto ldr_find_opt_exec_header(DosHeaderPtr ptrDos) -> ExecOptionalHeaderPtr
+	{
+		if (!ptrDos)
+			return nullptr;
+
+        auto exec = ldr_find_exec_header(ptrDos);
+
+        if (!exec)
+            return nullptr;
+
+		return (ExecOptionalHeaderPtr)(VoidPtr)(&exec->mCharacteristics + 1);
+	}
+
+	/// @brief Find the PE header inside the blob.
+	/// @note overloaded function.
+	inline auto ldr_find_exec_header(const Char* ptrDos) -> ExecHeaderPtr
+	{
+	   return ldr_find_exec_header((DosHeaderPtr)ptrDos);
+	}
+
+
+	/// @brief Find the PE header inside the blob.
+	/// @note overloaded function.
+	inline auto ldr_find_opt_exec_header(const Char* ptrDos) -> ExecOptionalHeaderPtr
+	{
+	   return ldr_find_opt_exec_header((DosHeaderPtr)ptrDos);
+	}
+} // namespace Kernel
+
+#endif // ifndef __CFKIT_LOADER_UTILS_HXX__