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/mem/io-stream.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/mem/io-stream.c')
-rw-r--r-- | src/mem/io-stream.c | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/src/mem/io-stream.c b/src/mem/io-stream.c new file mode 100644 index 0000000..f7c7834 --- /dev/null +++ b/src/mem/io-stream.c @@ -0,0 +1,53 @@ +#include <stdio.h> +#include "io-stream.h" +#include "io.h" + +static size_t +file_istream_read(const struct mem_io_istream *stream, void *ptr, const size_t size) +{ + return fread(ptr, 1, size, stream->backing); +} + +struct mem_io_istream +mem_io_istream_from_file(FILE *file) +{ + return (struct mem_io_istream){ + .read = file_istream_read, + .backing = file + }; +} + +size_t +mem_io_write_from_stream(const struct mem_io *io, const struct mem_io_istream *stream, const size_t offset, const size_t size) +{ + size_t trw = 0; + unsigned char buf[4096]; + for (size_t rd, trd = 0; (rd = stream->read(stream, buf, (trd + sizeof(buf) > size ? size - trd : sizeof(buf)))) && trd < size; trd += rd) + trw += io->write(io, buf, offset + trd, rd); + return trw; +} + +static size_t +file_ostream_write(const struct mem_io_ostream *stream, const void *ptr, const size_t size) +{ + return fwrite(ptr, 1, size, stream->backing); +} + +struct mem_io_ostream +mem_io_ostream_from_file(FILE *file) +{ + return (struct mem_io_ostream){ + .write = file_ostream_write, + .backing = file + }; +} + +size_t +mem_io_read_to_stream(const struct mem_io *io, const struct mem_io_ostream *stream, const size_t offset, const size_t size) +{ + size_t trw = 0; + unsigned char buf[4096]; + for (size_t rd, trd = 0; (rd = io->read(io, buf, offset + trd, (trd + sizeof(buf) > size ? size - trd : sizeof(buf)))) && trd < size; trd += rd) + trw += stream->write(stream, buf, rd); + return trw; +} |