From 75f9922f6d3c1e5bbbe7b90ca170392cc0f5efbc Mon Sep 17 00:00:00 2001 From: Jari Vetoniemi Date: Sun, 21 Oct 2018 16:23:23 +0300 Subject: Refactor io utils, add *-address-rw tools Namespace io_ stuff into mem_io_ to be less likely to collision with anything else. Add io-stream utility for working with streams instead of direct buffers. Add address-rw tools for simple memory read/write, where regions aren't needed. --- src/io/io-ptrace.c | 82 ------------------------------------------------------ 1 file changed, 82 deletions(-) delete mode 100644 src/io/io-ptrace.c (limited to 'src/io/io-ptrace.c') diff --git a/src/io/io-ptrace.c b/src/io/io-ptrace.c deleted file mode 100644 index 50d0766..0000000 --- a/src/io/io-ptrace.c +++ /dev/null @@ -1,82 +0,0 @@ -#include "io.h" -#include -#include -#include -#include - -static size_t -io_ptrace_do(const struct io *io, void *ptr, const size_t offset, const size_t size, size_t (*iofun)(void*, size_t, size_t, FILE*)) -{ - if (fseek(io->backing, offset, SEEK_SET) != 0) { - warn("fseek(/proc/%u/mem, %zu)", io->pid, offset); - return 0; - } - - return iofun(ptr, 1, size, io->backing); -} - -static size_t -io_ptrace_write(const struct io *io, const void *ptr, const size_t offset, const size_t size) -{ - clearerr(io->backing); - const size_t ret = io_ptrace_do(io, (void*)ptr, offset, size, (size_t(*)())fwrite); - - if (ferror(io->backing)) - warn("fwrite(/proc/%u/mem)", io->pid); - - return ret; -} - -static size_t -io_ptrace_read(const struct io *io, void *ptr, const size_t offset, const size_t size) -{ - clearerr(io->backing); - const size_t ret = io_ptrace_do(io, ptr, offset, size, fread); - - if (ferror(io->backing)) - warn("fread(/proc/%u/mem)", io->pid); - - return ret; -} - -static void -io_ptrace_cleanup(struct io *io) -{ - if (io->backing) - fclose(io->backing); - - if (io->pid) - ptrace(PTRACE_DETACH, io->pid, 1, 0); -} - -bool -io_ptrace_init(struct io *io, const pid_t pid) -{ - *io = (struct io){ .pid = pid, .read = io_ptrace_read, .write = io_ptrace_write, .cleanup = io_ptrace_cleanup }; - - if (ptrace(PTRACE_ATTACH, pid, NULL, NULL) == -1L) { - warn("ptrace(PTRACE_ATTACH, %u, NULL, NULL)", pid); - goto fail; - } - - { - int status; - if (waitpid(pid, &status, 0) == -1 || !WIFSTOPPED(status)) { - warn("waitpid(%u) == %d", pid, status); - goto fail; - } - } - - char path[128]; - snprintf(path, sizeof(path), "/proc/%u/mem", pid); - if (!(io->backing = fopen(path, "w+b"))) { - warn("fopen(%s)", path); - goto fail; - } - - return true; - -fail: - io->cleanup(io); - return false; -} -- cgit v1.2.3