Files
TaxiDriver/src/client/memory.c
2023-10-22 19:48:55 +02:00

117 lines
2.3 KiB
C

#include "memory.h"
#include "communication_struct.h"
#include <stdio.h>
#include <sys/ioctl.h>
#include <stdint.h>
#include <fcntl.h>
#include <unistd.h>
#define DEVICE_FILE "/dev/TaxiDriver"
static int file_desc;
int open_device(void)
{
file_desc = open(DEVICE_FILE, O_RDWR);
if (file_desc < 0) {
perror("Revird: Failed to open the device.");
return -1;
}
return 1;
}
void close_device(void)
{
close(file_desc);
}
int get_pid(const char *program_name) {
FILE *fp;
char command[128];
char buffer[128];
int pid = -1;
// Create a command to run 'pidof' for the specified program
snprintf(command, sizeof(command), "pidof %s", program_name);
// Open a pipe to execute the command and read the output
fp = popen(command, "r");
if (fp == NULL) {
perror("popen");
return -1;
}
// Read the output (should be a space-separated list of PIDs)
if (fgets(buffer, sizeof(buffer), fp) != NULL) {
// Extract the first PID from the list
if (sscanf(buffer, "%d", &pid) == 1) {
}
}
// Close the pipe and check for errors
if (pclose(fp) == -1) {
perror("pclose");
return -1;
}
return pid;
}
void *RPM(uintptr_t address, ssize_t size)
{
struct s_RPM args;
args.addr = address;
args.size = size;
args.out = 0;
args.out_addr = &args.out;
int ret = ioctl(file_desc, IOCTL_RPM, &args);
if (ret < 0) {
perror("Revird: RPM failed.");
close(file_desc);
return 0;
}
return (void *)args.out;
}
void WPM(uintptr_t addr, uintptr_t value, ssize_t size)
{
struct s_WPM args_wpm;
args_wpm.addr = addr;
args_wpm.size = size;
args_wpm.value = value;
int ret;
ret = ioctl(file_desc, IOCTL_WPM, &args_wpm);
if (ret < 0) {
perror("Revird: WPM failed.");
close(file_desc);
return;
}
return;
}
int open_process(int pid)
{
int ret;
ret = ioctl(file_desc, IOCTL_OPENPROC, &pid);
if (ret < 0) {
perror("Revird: openprocess failed.");
close(file_desc);
return -1;
}
return 1;
}
uintptr_t get_module(const char *mod)
{
int ret = ioctl(file_desc, IOCTL_GETMODULE, mod);
if (ret < 0) {
perror("Revird: getmodule failed.");
close(file_desc);
return -1;
}
return ret;
}