diff options
author | Jari Vetoniemi <mailroxas@gmail.com> | 2018-10-21 16:23:23 +0300 |
---|---|---|
committer | Jari Vetoniemi <mailroxas@gmail.com> | 2018-10-21 16:23:23 +0300 |
commit | 75f9922f6d3c1e5bbbe7b90ca170392cc0f5efbc (patch) | |
tree | d79e445e87ae9a55e146c6c3b46fef6f238d23e7 /src/cli/proc-address-rw.c | |
parent | d81411896f140981400e4fbf4aafacdbabad96cd (diff) |
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.
Diffstat (limited to 'src/cli/proc-address-rw.c')
-rw-r--r-- | src/cli/proc-address-rw.c | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/src/cli/proc-address-rw.c b/src/cli/proc-address-rw.c new file mode 100644 index 0000000..6318795 --- /dev/null +++ b/src/cli/proc-address-rw.c @@ -0,0 +1,77 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include "mem/io.h" +#include "mem/io-stream.h" +#include "util.h" + +static void +usage(const char *argv0) +{ + fprintf(stderr, "usage: %s pid write offset [len] < data\n" + " %s pid read offset len\n", argv0, argv0); + exit(EXIT_FAILURE); +} + +struct options { + size_t offset, len; + enum { + MODE_WRITE, + MODE_READ + } mode; + bool has_len; +}; + +static inline void +options_init(struct options *opt, size_t argc, const char *argv[]) +{ + size_t arg = 0; + *opt = (struct options){0}; + + { + bool w = false, r = false; + const char *mode = argv[arg++]; + if (!(w = !strcmp(mode, "write")) && !(r = !strcmp(mode, "read"))) + errx(EXIT_FAILURE, "mode must be write or read"); + + opt->mode = (w ? MODE_WRITE : MODE_READ); + } + + opt->offset = hexdecstrtoull(argv[arg++], NULL); + + if (argc >= arg + 1) { + opt->len = hexdecstrtoull(argv[arg++], NULL); + opt->has_len = true; + } + + if (opt->mode == MODE_READ && !opt->has_len) + usage(argv[0]); +} + +int +proc_address_rw(int argc, const char *argv[], bool (*mem_io_init)(struct mem_io*, const pid_t)) +{ + if (argc < 4) + usage(argv[0]); + + const pid_t pid = strtoull(argv[1], NULL, 10); + + struct options opt; + options_init(&opt, argc - 2, argv + 2); + + struct mem_io io; + if (!mem_io_init(&io, pid)) + return EXIT_FAILURE; + + size_t trw = 0; + if (opt.mode == MODE_WRITE) { + struct mem_io_istream stream = mem_io_istream_from_file(stdin); + trw = mem_io_write_from_stream(&io, &stream, opt.offset, (opt.has_len ? opt.len : (size_t)~0)); + } else { + struct mem_io_ostream stream = mem_io_ostream_from_file(stdout); + trw = mem_io_read_to_stream(&io, &stream, opt.offset, opt.len); + } + + mem_io_release(&io); + return trw; +} |