radare2/libr/io/iobuf.c
2017-08-19 20:05:21 +02:00

71 lines
1.4 KiB
C

/* radare - LGPL - Copyright 2013-2017 - pancake */
#include "r_io.h"
R_API void r_io_buffer_close(RIO* io) {
r_cache_flush (io->buffer);
io->buffer_enabled = 0;
}
R_API int r_io_buffer_load(RIO* io, ut64 addr, int len) {
ut64 at;
int i, r;
ut8 buf[512];
if (len < 1) {
return false;
}
io->buffer_enabled = 0;
for (i = 0; i < len; i += sizeof (buf)) {
at = addr + i;
(void) r_io_seek (io, at, R_IO_SEEK_SET);
memset (buf, 0xff, sizeof (buf));
r = r_io_read (io, buf, sizeof (buf));
if (r < 1) {
break;
}
r_cache_set (io->buffer, at, buf, sizeof (buf));
}
io->buffer_enabled = 1;
return true;
}
R_API const ut8* r_io_buffer_get (RIO *io, ut64 addr, int *len) {
return r_cache_get (io->buffer, addr, len);
}
R_API int r_io_buffer_read (RIO *io, ut64 addr, ut8* buf, int len) {
const ut8 *ret;
int next, l = 0;
// align addr if out of buffer if its mapped on io //
ret = r_cache_get (io->buffer, addr, &l);
if (!ret) {
if (l < 1) {
return 0; // no next block in buffer cache
}
if (l > len) {
return 0; // next block too far
}
next = l;
ret = r_cache_get (io->buffer, addr + next + 1, &l);
if (!ret) {
return 0;
}
if (l < len) {
memset (buf + l, 0xff, len - l);
}
if (l > len) {
l = len;
}
memset (buf, 0xff, next);
memcpy (buf + next, ret, len - next);
return len;
}
if (l > len) {
l = len;
} else if (l < len) {
memset (buf + l, 0xff, len - l);
}
memcpy (buf, ret, l);
return l;
}