1 files changed, 34 insertions, 7 deletions

diff --git a/src/kernel/user/syscall.c b/src/kernel/user/syscall.c
index 26032ad..ca5a65c 100644
--- a/src/kernel/user/syscall.c
+++ b/src/kernel/user/syscall.c
@@ -4,7 +4,11 @@
 
 #include <syscall.h>
 
-typedef uint32_t (*syscall_handler_t)(uint32_t, uint32_t, uint32_t, uint32_t, uint32_t);
+typedef struct {
+	uint32_t sc_id, a, b, c, d, e;		// a: ebx, b: ecx, c: edx, d: esi, e: edi
+} sc_args_t;
+
+typedef uint32_t (*syscall_handler_t)(sc_args_t);
 
 static syscall_handler_t sc_handlers[SC_MAX] = { 0 };
 
@@ -25,21 +29,22 @@ static char* sc_copy_string(uint32_t s, uint32_t slen) {
 // THE SYSCALLS CODE !! //
 // ==================== //
 
-static uint32_t exit_sc(uint32_t code, uint32_t uu1, uint32_t uu2, uint32_t uu3, uint32_t uu4) {
-	dbg_printf("Proc %d exit with code %d\n", current_process()->pid, code);
+
+static uint32_t exit_sc(sc_args_t args) {
+	dbg_printf("Proc %d exit with code %d\n", current_process()->pid, args.a);
 	// TODO : use code... and process exiting is not supposed to be done this way
 
 	exit();
 	return 0;
 }
 
-static uint32_t yield_sc() {
+static uint32_t yield_sc(sc_args_t args) {
 	yield();
 	return 0;
 }
 
-static uint32_t dbg_print_sc(uint32_t s, uint32_t slen, uint32_t uu2, uint32_t uu3, uint32_t uu4) {
-	char* msg = sc_copy_string(s, slen);
+static uint32_t dbg_print_sc(sc_args_t args) {
+	char* msg = sc_copy_string(args.a, args.b);
 	if (msg == 0) return -1;
 
 	dbg_print(msg);
@@ -48,6 +53,18 @@ static uint32_t dbg_print_sc(uint32_t s, uint32_t slen, uint32_t uu2, uint32_t u
 	return 0;
 }
 
+static uint32_t mmap_sc(sc_args_t args) {
+	return mmap(current_process(), (void*)args.a, args.b, args.c);
+}
+
+static uint32_t mchmap_sc(sc_args_t args) {
+	return mchmap(current_process(), (void*)args.a, args.b);
+}
+
+static uint32_t munmap_sc(sc_args_t args) {
+	return munmap(current_process(), (void*)args.a);
+}
+
 // ====================== //
 // SYSCALLS SETUP ROUTINE //
 // ====================== //
@@ -56,6 +73,10 @@ void setup_syscalls() {
 	sc_handlers[SC_EXIT] = exit_sc;
 	sc_handlers[SC_YIELD] = yield_sc;
 	sc_handlers[SC_DBG_PRINT] = dbg_print_sc;
+
+	sc_handlers[SC_MMAP] = mmap_sc;
+	sc_handlers[SC_MCHMAP] = mchmap_sc;
+	sc_handlers[SC_MUNMAP] = munmap_sc;
 }
 
 void syscall_handler(registers_t *regs) {
@@ -64,7 +85,13 @@ void syscall_handler(registers_t *regs) {
 	if (regs->eax < SC_MAX) {
 		syscall_handler_t h = sc_handlers[regs->eax];
 		if (h != 0) {
-			regs->eax = h(regs->ebx, regs->ecx, regs->edx, regs->esi, regs->edi);
+			sc_args_t args = {
+				.a = regs->ebx,
+				.b = regs->ecx,
+				.c = regs->edx,
+				.d = regs->esi,
+				.e = regs->edi};
+			regs->eax = h(args);
 		} else {
 			regs->eax = -1;
 		}