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;
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);

12
src/client/main.c
View File

@@ -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;