From 5e44d0572ff6985045e295d30fba76125a3fca1b Mon Sep 17 00:00:00 2001
From: ALittlePatate <maxime@patate.dev>
Date: Sun, 22 Oct 2023 15:37:52 +0200
Subject: [PATCH] add: WPM

---
 src/TaxiDriver.c  | 43 +++++++++++++++++++++++++++++++++++--------
 src/client/main.c | 12 ++++++------
 2 files changed, 41 insertions(+), 14 deletions(-)

diff --git a/src/TaxiDriver.c b/src/TaxiDriver.c
index af40d71..02281fe 100644
--- a/src/TaxiDriver.c
+++ b/src/TaxiDriver.c
@@ -27,22 +27,51 @@ int RPM(t_RPM args) {
     struct mm_struct *mm;
     unsigned long value = 0;
 
-    printk(KERN_ALERT "TaxiDriver: RPM --> addr : 0x%lx, size : %ld\n", rpm_args.addr, rpm_args.size);
+    printk(KERN_INFO "TaxiDriver: RPM --> addr : 0x%lx, size : %ld\n", args.addr, args.size);
     if (args.addr == 0)
 	return -1;
+    if (task == NULL) {
+        return -EINVAL;  // Invalid argument
+    }
     mm = get_task_mm(task);
     if (mm != NULL) {
         if (access_process_vm(task, args.addr, &value, args.size, 0) == args.size) {
 	    printk(KERN_INFO "TaxiDriver: Value at 0x%lx: %lu\n", args.addr, value);
 	    mmput(mm);
 	} else {
-	    printk(KERN_INFO "TaxiDriver: Failed to read value at 0x%lx\n", args.addr);
+	    printk(KERN_ALERT "TaxiDriver: Failed to read value at 0x%lx\n", args.addr);
 	    return -1;
 	}
     }
     return (int)value;
 }
 
+int WPM(t_WPM args) {
+    struct mm_struct *mm;
+    int ret = 0;
+
+    printk(KERN_INFO "TaxiDriver: WPM --> addr : 0x%lx, size : %ld, value : %ld\n",
+	   args.addr, args.size, args.value);
+    if (args.addr == 0)
+        return -1;
+    if (task == NULL) {
+        return -EINVAL;  // Invalid argument
+    }
+    mm = get_task_mm(task);
+    if (mm != NULL) {
+        if (access_process_vm(task, args.addr, &args.value, args.size, 1) == args.size) {
+            printk(KERN_INFO "TaxiDriver: Successfully wrote value %lu to 0x%lx\n", args.value, args.addr);
+            mmput(mm);
+        } else {
+            printk(KERN_ALERT "TaxiDriver: Failed to write value at 0x%lx\n", args.addr);
+            ret = -1;
+        }
+    } else {
+        ret = -EINVAL;
+    }
+    return ret;
+}
+
 static int init_process_by_pid(int target_pid) {
     printk(KERN_INFO "TaxiDriver: Accessing process with PID: %d\n", target_pid);
 
@@ -91,9 +120,7 @@ static long device_ioctl(struct file *file, unsigned int ioctl_num, unsigned lon
         case IOCTL_WPM:
             if (copy_from_user(&wpm_args, (int *)arg, sizeof(t_WPM)))
                 return -EFAULT;
-	    printk(KERN_ALERT "TaxiDriver: WPM --> addr : %ld, size : %ld, value : %ld\n",
-		   wpm_args.addr, wpm_args.size, wpm_args.value);
-	    return_value = 1337;
+	    return_value = WPM(wpm_args);
             break;
 
         default:
@@ -111,7 +138,7 @@ static struct file_operations fops = {
 
 static int __init driver_init(void)
 {
-    printk(KERN_ALERT "TaxiDriver: Loaded\n");
+    printk(KERN_INFO "TaxiDriver: Loaded\n");
 
     // Dynamically allocate the major number
     major_number = register_chrdev(0, DRIVER_NAME, &fops);
@@ -121,7 +148,7 @@ static int __init driver_init(void)
         return major_number;
     }
 
-    printk(KERN_ALERT "TaxiDriver: Registered %s with major number %d\n", DRIVER_NAME, major_number);
+    printk(KERN_INFO "TaxiDriver: Registered %s with major number %d\n", DRIVER_NAME, major_number);
 
     return 0;
 }
@@ -129,7 +156,7 @@ static int __init driver_init(void)
 static void __exit driver_exit(void)
 {
     unregister_chrdev(major_number, DRIVER_NAME);
-    printk(KERN_ALERT "TaxiDriver: Unloaded\n");
+    printk(KERN_INFO "TaxiDriver: Unloaded\n");
 }
 
 module_init(driver_init);
diff --git a/src/client/main.c b/src/client/main.c
index 57f7779..b98e2d9 100644
--- a/src/client/main.c
+++ b/src/client/main.c
@@ -8,17 +8,17 @@ int main() {
     if (!open_process(pid))
 	return -1;
     
+    struct s_WPM args_wpm;
+    args_wpm.addr = 0x55a813479d70;
+    args_wpm.size = 4;
+    args_wpm.value = (uintptr_t)667;
+    WPM(args_wpm);
+    
     struct s_RPM args;
     args.addr = 0x55a813479d70;
     args.size = sizeof(int);
     int val = (int)RPM(args);
     printf("Value from RPM: %d\n", val);
-
-    struct s_WPM args_wpm;
-    args_wpm.addr = 0x55a813479d7;
-    args_wpm.size = 4;
-    args_wpm.value = (uintptr_t)667;
-    WPM(args_wpm);
     
     close_device();
     return 0;