Merged in MHR-5 (pull request #1)

MHR-5

10 files changed, 242 insertions, 0 deletions

diff --git a/Private/DriverKit/KernelDev.c b/Private/DriverKit/KernelDev.c
new file mode 100644
index 00000000..e8c041af
--- /dev/null
+++ b/Private/DriverKit/KernelDev.c
@@ -0,0 +1,31 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Text I/O.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelDev.h>
+
+/// @brief Open a new binary device from path.
+DK_EXTERN kernelDeviceRef kernelOpenBinaryDevice(const char* devicePath) {
+    if (!devicePath) return NIL;
+
+    return kernelCall("OpenBinaryDevice", 1, devicePath);
+}
+
+/// @brief Open a new character device from path.
+DK_EXTERN kernelDeviceRef kernelOpenCharDevice(const char* devicePath) {
+    if (!devicePath) return NIL;
+
+    return kernelCall("OpenCharDevice", 1, devicePath);
+}
+
+/// @brief Close any device.
+/// @param device valid device.
+DK_EXTERN void kernelCloseDevice(kernelDeviceRef device) {
+    if (!device) return;
+
+    kernelCall("CloseDevice", 1, device);
+}
diff --git a/Private/DriverKit/KernelDev.h b/Private/DriverKit/KernelDev.h
new file mode 100644
index 00000000..daa60a7c
--- /dev/null
+++ b/Private/DriverKit/KernelDev.h
@@ -0,0 +1,28 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Devices.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <DriverKit/KernelStd.h>
+
+typedef struct _kernelDevice {
+    int32_t(*read)(); // read from device.
+    int32_t(*write)(); // write to device.
+    struct _kernelDevice* (*open)(const char* path); // open device.
+    void (*close)(struct _kernelDevice* dev); // close device.
+} kernelDevice,* kernelDeviceRef;
+
+/// @brief Open a new binary device from path.
+DK_EXTERN kernelDeviceRef kernelOpenBinaryDevice(const char* devicePath);
+
+/// @brief Open a new character device from path.
+DK_EXTERN kernelDeviceRef kernelOpenCharDevice(const char* devicePath);
+
+/// @brief Close any device.
+/// @param device valid device.
+DK_EXTERN void kernelCloseDevice(kernelDeviceRef device);
diff --git a/Private/DriverKit/KernelDispatchCall.S b/Private/DriverKit/KernelDispatchCall.S
new file mode 100644
index 00000000..4eec8ae0
--- /dev/null
+++ b/Private/DriverKit/KernelDispatchCall.S
@@ -0,0 +1,21 @@
+.globl __kernelDispatchCall
+
+.section .text
+
+/* Really simple function, takes our va-list,
+    and brings it to the trap handler in the kernel. */
+
+#ifdef __x86_64__
+
+__kernelDispatchCall:
+    int $0x33
+    ret
+
+#elif defined(__powerpc64__)
+
+__kernelDispatchCall:
+    mflr 3
+    sc
+    blr
+
+#endif
diff --git a/Private/DriverKit/KernelPrint.c b/Private/DriverKit/KernelPrint.c
new file mode 100644
index 00000000..5ca09ccf
--- /dev/null
+++ b/Private/DriverKit/KernelPrint.c
@@ -0,0 +1,28 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Text I/O.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelPrint.h>
+
+DK_EXTERN void kernelPrintChar(const char ch) {
+    kernelCall("WriteCharacter", 1, ch);
+}
+
+/// @brief print string to UART.
+/// @param message UART to transmit.
+DK_EXTERN void kernelPrintStr(const char* message) {
+    if (!message) return;
+    if (*message == 0) return;
+
+    size_t index = 0;
+    size_t len = kernelStringLength(message);
+
+    while (index < len) {
+      kernelPrintChar(message[index]);
+      ++index;
+    }
+}
diff --git a/Private/DriverKit/KernelPrint.h b/Private/DriverKit/KernelPrint.h
new file mode 100644
index 00000000..36e55442
--- /dev/null
+++ b/Private/DriverKit/KernelPrint.h
@@ -0,0 +1,18 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Text I/O.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <DriverKit/KernelString.h>
+
+/// @brief print character into UART.
+DK_EXTERN void kernelPrintChar(const char ch);
+
+/// @brief print string to UART.
+/// @param message UART to transmit.
+DK_EXTERN void kernelPrintStr(const char* message);
diff --git a/Private/DriverKit/KernelStd.c b/Private/DriverKit/KernelStd.c
new file mode 100644
index 00000000..494ac0bc
--- /dev/null
+++ b/Private/DriverKit/KernelStd.c
@@ -0,0 +1,24 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Definitions.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelStd.h>
+#include <stdarg.h>
+
+DK_EXTERN __attribute__((naked)) void __kernelDispatchCall(int32_t cnt, ...);
+
+DK_EXTERN void* kernelCall(const char* kernelRpcName, int32_t cnt, ...) {
+    if (!kernelRpcName || cnt == 0) return NIL;
+
+    va_list arg;
+    va_start(arg, cnt);
+
+    __kernelDispatchCall(cnt, arg);
+
+    va_end(arg);
+
+}
diff --git a/Private/DriverKit/KernelStd.h b/Private/DriverKit/KernelStd.h
new file mode 100644
index 00000000..3344630e
--- /dev/null
+++ b/Private/DriverKit/KernelStd.h
@@ -0,0 +1,22 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Definitions.
+
+------------------------------------------- */
+
+#pragma once
+
+#if defined(__cplusplus)
+#define DK_EXTERN extern "C"
+#define NIL nullptr
+#else
+#define DK_EXTERN extern
+#define NIL NULL
+#endif // defined(__cplusplus)
+
+#include <stdint.h>
+#include <stddef.h>
+
+DK_EXTERN void* kernelCall(const char* kernelRpcName, int32_t cnt, ...);
diff --git a/Private/DriverKit/KernelStdCxx.cc b/Private/DriverKit/KernelStdCxx.cc
new file mode 100644
index 00000000..79d41a72
--- /dev/null
+++ b/Private/DriverKit/KernelStdCxx.cc
@@ -0,0 +1,24 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Driver C++ Definitions.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelStd.h>
+
+void* operator new(size_t sz) {
+    if (!sz) ++sz;
+
+    auto ptr = kernelCall("NewKernelHeap", 1, sz);
+    kernelCall("KernelHeapProtect", 1, ptr);
+
+    return ptr;
+}
+
+void operator delete(void* ptr) noexcept {
+    if (!ptr) return;
+
+    kernelCall("DeleteKernelHeap", 1,ptr);
+}
diff --git a/Private/DriverKit/KernelString.c b/Private/DriverKit/KernelString.c
new file mode 100644
index 00000000..0bb45b16
--- /dev/null
+++ b/Private/DriverKit/KernelString.c
@@ -0,0 +1,30 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Strings.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelString.h>
+
+DK_EXTERN size_t kernelStringLength(const char* str) {
+    size_t index = 0;
+
+    while (str[index] != 0) {
+        ++index;
+    }
+
+    return index;
+}
+
+DK_EXTERN int kernelStringCopy(char* dst, const char* src, size_t len) {
+    size_t index = 0;
+
+    while (index != len) {
+        dst[index] = src[index];
+        ++index;
+    }
+
+    return index;
+}
diff --git a/Private/DriverKit/KernelString.h b/Private/DriverKit/KernelString.h
new file mode 100644
index 00000000..8e189d26
--- /dev/null
+++ b/Private/DriverKit/KernelString.h
@@ -0,0 +1,16 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+    Purpose: Kernel Strings.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <DriverKit/KernelStd.h>
+
+/// @brief DriverKit equivalent of POSIX's string.h.
+
+DK_EXTERN size_t kernelStringLength(const char* str);
+DK_EXTERN int kernelStringCopy(char* dst, const char* src, size_t len);