patate/TaxiDriver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add: WPM

Browse Source
This commit is contained in:
ALittlePatate
2023-10-22 15:37:52 +02:00
parent 52c8b2a49f
commit 5e44d0572f
2 changed files with 41 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/TaxiDriver.c
View File

@@ -27,22 +27,51 @@ int RPM(t_RPM args) {
struct mm_struct *mm; struct mm_struct *mm;
unsigned long value = 0; 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) if (args.addr == 0)
return -1; return -1;
if (task == NULL) {
return -EINVAL; // Invalid argument
}
mm = get_task_mm(task); mm = get_task_mm(task);
if (mm != NULL) { if (mm != NULL) {
if (access_process_vm(task, args.addr, &value, args.size, 0) == args.size) { 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); printk(KERN_INFO "TaxiDriver: Value at 0x%lx: %lu\n", args.addr, value);
mmput(mm); mmput(mm);
} else { } 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 -1;
} }
} }
return (int)value; 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) { static int init_process_by_pid(int target_pid) {
printk(KERN_INFO "TaxiDriver: Accessing process with PID: %d\n", 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: case IOCTL_WPM:
if (copy_from_user(&wpm_args, (int *)arg, sizeof(t_WPM))) if (copy_from_user(&wpm_args, (int *)arg, sizeof(t_WPM)))
return -EFAULT; return -EFAULT;
printk(KERN_ALERT "TaxiDriver: WPM --> addr : %ld, size : %ld, value : %ld\n", return_value = WPM(wpm_args);
wpm_args.addr, wpm_args.size, wpm_args.value);
return_value = 1337;
break; break;
default: default:
@@ -111,7 +138,7 @@ static struct file_operations fops = {
static int __init driver_init(void) static int __init driver_init(void)
{ {
printk(KERN_ALERT "TaxiDriver: Loaded\n"); printk(KERN_INFO "TaxiDriver: Loaded\n");
// Dynamically allocate the major number // Dynamically allocate the major number
major_number = register_chrdev(0, DRIVER_NAME, &fops); major_number = register_chrdev(0, DRIVER_NAME, &fops);
@@ -121,7 +148,7 @@ static int __init driver_init(void)
return major_number; 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; return 0;
} }
@@ -129,7 +156,7 @@ static int __init driver_init(void)
static void __exit driver_exit(void) static void __exit driver_exit(void)
{ {
unregister_chrdev(major_number, DRIVER_NAME); unregister_chrdev(major_number, DRIVER_NAME);
printk(KERN_ALERT "TaxiDriver: Unloaded\n"); printk(KERN_INFO "TaxiDriver: Unloaded\n");
} }
module_init(driver_init); module_init(driver_init);

12
src/client/main.c
View File

@@ -8,18 +8,18 @@ int main() {
if (!open_process(pid)) if (!open_process(pid))
return -1; 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; struct s_RPM args;
args.addr = 0x55a813479d70; args.addr = 0x55a813479d70;
args.size = sizeof(int); args.size = sizeof(int);
int val = (int)RPM(args); int val = (int)RPM(args);
printf("Value from RPM: %d\n", val); 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(); close_device();
return 0; return 0;
} }