mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-12 15:38:09 +00:00
741 lines
17 KiB
C
741 lines
17 KiB
C
/* radare - LGPL - Copyright 2009-2016 - pancake */
|
|
|
|
#include <r_types.h>
|
|
#include <r_util.h>
|
|
#include <r_io.h>
|
|
|
|
// TODO: Optimize to use memcpy when buffers are not in range..
|
|
// check buf boundaries and offsets and use memcpy or memmove
|
|
|
|
// copied from riocacheread
|
|
// ret # of bytes copied
|
|
static int sparse_read(RList *list, ut64 addr, ut8 *buf, int len) {
|
|
int l, ret, da, db;
|
|
RListIter *iter;
|
|
RBufferSparse *c;
|
|
|
|
r_list_foreach (list, iter, c) {
|
|
if (r_range_overlap (addr, addr+len-1, c->from, c->to, &ret)) {
|
|
if (ret > 0) {
|
|
da = ret;
|
|
db = 0;
|
|
l = c->size;
|
|
} else if (ret < 0) {
|
|
da = 0;
|
|
db = -ret;
|
|
l = c->size-db;
|
|
} else {
|
|
da = 0;
|
|
db = 0;
|
|
l = c->size;
|
|
}
|
|
// say hello to integer overflow, but this won't happen in
|
|
// realistic scenarios because malloc will fail befor
|
|
if ((l + da) > len) {
|
|
l = len - da;
|
|
}
|
|
if (l > 0) {
|
|
memcpy (buf + da, c->data + db, l);
|
|
}
|
|
}
|
|
}
|
|
return len;
|
|
}
|
|
|
|
static RBufferSparse *sparse_append(RList *l, ut64 addr, const ut8 *data, int len) {
|
|
if (l && data && len > 0) {
|
|
RBufferSparse *s = R_NEW0 (RBufferSparse);
|
|
if (s) {
|
|
s->data = calloc (1, len);
|
|
if (s->data) {
|
|
s->from = addr;
|
|
s->to = addr + len;
|
|
s->size = len;
|
|
s->odata = NULL;
|
|
memcpy (s->data, data, len);
|
|
return r_list_append (l, s)? s: NULL;
|
|
}
|
|
free (s);
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
//ret -1 if failed; # of bytes copied if success
|
|
static int sparse_write(RList *l, ut64 addr, const ut8 *data, int len) {
|
|
RBufferSparse *s;
|
|
RListIter *iter;
|
|
|
|
r_list_foreach (l, iter, s) {
|
|
if (addr >= s->from && addr < s->to) {
|
|
int newlen = addr + len - s->to;
|
|
int delta = addr - s->from;
|
|
if (newlen > 0) {
|
|
// must realloc
|
|
ut8 *ndata = realloc (s->data, len + newlen);
|
|
if (ndata) {
|
|
s->data = ndata;
|
|
} else {
|
|
eprintf ("sparse write fail\n");
|
|
return -1;
|
|
}
|
|
}
|
|
memcpy (s->data + delta, data, len);
|
|
/* write here */
|
|
return len;
|
|
}
|
|
}
|
|
if (!sparse_append (l, addr, data, len)) {
|
|
return -1;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
static bool sparse_limits(RList *l, ut64 *min, ut64 *max) {
|
|
bool set = false;
|
|
RBufferSparse *s;
|
|
RListIter *iter;
|
|
|
|
if (min) {
|
|
*min = UT64_MAX;
|
|
}
|
|
r_list_foreach (l, iter, s) {
|
|
if (set) {
|
|
if (min && s->from < *min) {
|
|
*min = s->from;
|
|
}
|
|
if (max && s->to > *max) {
|
|
*max = s->to;
|
|
}
|
|
} else {
|
|
set = true;
|
|
if (min) {
|
|
*min = s->from;
|
|
}
|
|
if (max) {
|
|
*max = s->to;
|
|
}
|
|
}
|
|
}
|
|
return set;
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new_with_pointers (const ut8 *bytes, ut64 len) {
|
|
RBuffer *b = r_buf_new ();
|
|
if (b && bytes && len > 0 && len != UT64_MAX) {
|
|
b->buf = (ut8*)bytes;
|
|
b->length = len;
|
|
b->empty = false;
|
|
b->ro = true;
|
|
}
|
|
return b;
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new_with_bytes (const ut8 *bytes, ut64 len) {
|
|
RBuffer *b = r_buf_new ();
|
|
if (b && bytes && (len > 0 && len != UT64_MAX)) {
|
|
r_buf_set_bytes (b, bytes, len);
|
|
}
|
|
return b;
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new_with_buf(RBuffer *b) {
|
|
return r_buf_new_with_bytes (b->buf, b->length);
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new_sparse() {
|
|
RBuffer *b = r_buf_new ();
|
|
if (!b) {
|
|
return NULL;
|
|
}
|
|
b->sparse = r_list_newf ((RListFree)free);
|
|
return b;
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new() {
|
|
RBuffer *b = R_NEW0 (RBuffer);
|
|
if (b) {
|
|
b->fd = -1;
|
|
}
|
|
return b;
|
|
}
|
|
|
|
R_API const ut8 *r_buf_buffer (RBuffer *b) {
|
|
return (b && !b->sparse)? b->buf: NULL;
|
|
}
|
|
|
|
R_API ut64 r_buf_size (RBuffer *b) {
|
|
if (!b) return 0LL;
|
|
if (b->fd != -1) {
|
|
return b->length;
|
|
}
|
|
if (b->sparse) {
|
|
ut64 max = 0LL;
|
|
if (sparse_limits (b->sparse, NULL, &max)) {
|
|
return max; // -min
|
|
}
|
|
return 0LL;
|
|
}
|
|
return b->empty? 0: b->length;
|
|
}
|
|
|
|
// rename to new?
|
|
R_API RBuffer *r_buf_mmap (const char *file, int flags) {
|
|
int rw = flags & R_IO_WRITE ? true : false;
|
|
RBuffer *b = r_buf_new ();
|
|
if (!b) return NULL;
|
|
b->mmap = r_file_mmap (file, rw, 0);
|
|
if (b->mmap) {
|
|
b->buf = b->mmap->buf;
|
|
b->length = b->mmap->len;
|
|
if (b->length == 0) b->empty = 1;
|
|
return b;
|
|
}
|
|
r_buf_free (b);
|
|
return NULL; /* we just freed b, don't return it */
|
|
}
|
|
|
|
R_API RBuffer *r_buf_new_file(const char *file, bool newFile) {
|
|
const int mode = 0644;
|
|
int flags = O_RDWR;
|
|
if (newFile) {
|
|
flags |= O_CREAT;
|
|
}
|
|
int fd = r_sandbox_open (file, flags, mode);
|
|
if (fd != -1) {
|
|
RBuffer *b = r_buf_new ();
|
|
if (!b) {
|
|
r_sandbox_close (fd);
|
|
return NULL;
|
|
}
|
|
b->fd = fd;
|
|
return b;
|
|
}
|
|
return NULL; /* we just freed b, don't return it */
|
|
}
|
|
|
|
// TODO: rename to new_from_file ?
|
|
R_API RBuffer *r_buf_new_slurp(const char *file) {
|
|
int len;
|
|
RBuffer *b = r_buf_new ();
|
|
if (!b) return NULL;
|
|
b->buf = (ut8*)r_file_slurp (file, &len);
|
|
b->length = len;
|
|
if (b->buf) {
|
|
return b;
|
|
}
|
|
r_buf_free (b);
|
|
return NULL; /* we just freed b, don't return it */
|
|
}
|
|
|
|
R_API bool r_buf_dump(RBuffer *b, const char *file) {
|
|
if (!b || !file) {
|
|
return false;
|
|
}
|
|
return r_file_dump (file, r_buf_get_at (b, 0, NULL), r_buf_size (b), 0);
|
|
}
|
|
|
|
R_API int r_buf_seek (RBuffer *b, st64 addr, int whence) {
|
|
ut64 min = 0LL, max = 0LL;
|
|
if (b->fd != -1) {
|
|
if (r_sandbox_lseek (b->fd, addr, whence) == -1) {
|
|
// seek failed - print error here?
|
|
return -1;
|
|
}
|
|
} else if (b->sparse) {
|
|
sparse_limits (b->sparse, &min, &max);
|
|
switch (whence) {
|
|
case R_IO_SEEK_SET: b->cur = addr; break;
|
|
case R_IO_SEEK_CUR: b->cur = b->cur + addr; break;
|
|
case R_IO_SEEK_END:
|
|
if (sparse_limits (b->sparse, NULL, &max)) {
|
|
return max; // -min
|
|
}
|
|
b->cur = max + addr; break; //b->base + b->length + addr; break;
|
|
}
|
|
} else {
|
|
min = b->base;
|
|
max = b->base + b->length;
|
|
switch (whence) {
|
|
//case 0: b->cur = b->base + addr; break;
|
|
case R_IO_SEEK_SET: b->cur = addr; break;
|
|
case R_IO_SEEK_CUR: b->cur = b->cur + addr; break;
|
|
case R_IO_SEEK_END: b->cur = b->base + b->length + addr; break;
|
|
}
|
|
}
|
|
/* avoid out-of-bounds */
|
|
if (b->cur < min) {
|
|
b->cur = min;
|
|
}
|
|
if (b->cur >= max) {
|
|
b->cur = max;
|
|
}
|
|
return (int)b->cur;
|
|
}
|
|
|
|
R_API int r_buf_set_bits(RBuffer *b, int bitoff, int bitsize, ut64 value) {
|
|
// TODO: implement r_buf_set_bits
|
|
// TODO: get the implementation from reg/value.c ?
|
|
return false;
|
|
}
|
|
|
|
R_API int r_buf_set_bytes(RBuffer *b, const ut8 *buf, ut64 length) {
|
|
if (length <= 0 || !buf) return false;
|
|
free (b->buf);
|
|
if (!(b->buf = malloc (length)))
|
|
return false;
|
|
memmove (b->buf, buf, length);
|
|
b->length = length;
|
|
b->empty = 0;
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_prepend_bytes(RBuffer *b, const ut8 *buf, int length) {
|
|
if ((b->buf = realloc (b->buf, b->length+length))) {
|
|
memmove (b->buf+length, b->buf, b->length);
|
|
memmove (b->buf, buf, length);
|
|
b->length += length;
|
|
b->empty = 0;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// TODO: R_API void r_buf_insert_bytes() // with shift
|
|
// TODO: R_API void r_buf_write_bytes() // overwrite
|
|
|
|
R_API char *r_buf_to_string(RBuffer *b) {
|
|
char *s;
|
|
if (!b) {
|
|
return strdup ("");
|
|
}
|
|
s = malloc (b->length + 1);
|
|
if (s) {
|
|
memmove (s, b->buf, b->length);
|
|
s[b->length] = 0;
|
|
}
|
|
return s;
|
|
}
|
|
|
|
R_API bool r_buf_append_bytes(RBuffer *b, const ut8 *buf, int length) {
|
|
if (!b) {
|
|
return false;
|
|
}
|
|
if (b->fd != -1) {
|
|
r_sandbox_lseek (b->fd, 0, SEEK_END);
|
|
r_sandbox_write (b->fd, buf, length);
|
|
return true;
|
|
}
|
|
if (b->empty) {
|
|
b->length = b->empty = 0;
|
|
}
|
|
if (!(b->buf = realloc (b->buf, 1 + b->length + length))) {
|
|
return false;
|
|
}
|
|
memmove (b->buf+b->length, buf, length);
|
|
b->buf[b->length+length] = 0;
|
|
b->length += length;
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_append_nbytes(RBuffer *b, int length) {
|
|
if (!b) return false;
|
|
if (b->fd != -1) {
|
|
ut8 *buf = calloc (1, length);
|
|
if (buf) {
|
|
r_sandbox_lseek (b->fd, 0, SEEK_END);
|
|
r_sandbox_write (b->fd, buf, length);
|
|
free (buf);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
if (b->empty) b->length = b->empty = 0;
|
|
if (!(b->buf = realloc (b->buf, b->length+length)))
|
|
return false;
|
|
memset (b->buf+b->length, 0, length);
|
|
b->length += length;
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_append_ut16(RBuffer *b, ut16 n) {
|
|
if (!b) return false;
|
|
if (b->fd != -1) {
|
|
return r_buf_append_bytes (b, (const ut8*)&n, sizeof (n));
|
|
}
|
|
if (b->empty) b->length = b->empty = 0;
|
|
if (!(b->buf = realloc (b->buf, b->length + sizeof (n))))
|
|
return false;
|
|
memmove (b->buf+b->length, &n, sizeof (n));
|
|
b->length += sizeof (n);
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_append_ut32(RBuffer *b, ut32 n) {
|
|
if (b->empty) b->length = b->empty = 0;
|
|
if (b->fd != -1) {
|
|
return r_buf_append_bytes (b, (const ut8*)&n, sizeof (n));
|
|
}
|
|
if (!(b->buf = realloc (b->buf, b->length+sizeof (n))))
|
|
return false;
|
|
memmove (b->buf+b->length, &n, sizeof (n));
|
|
b->length += sizeof (n);
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_append_ut64(RBuffer *b, ut64 n) {
|
|
if (!b) return false;
|
|
if (b->fd != -1) {
|
|
return r_buf_append_bytes (b, (const ut8*)&n, sizeof (n));
|
|
}
|
|
if (b->empty) b->length = b->empty = 0;
|
|
if (!(b->buf = realloc (b->buf, b->length+sizeof (n))))
|
|
return false;
|
|
memmove (b->buf+b->length, &n, sizeof (n));
|
|
b->length += sizeof (n);
|
|
return true;
|
|
}
|
|
|
|
R_API bool r_buf_append_buf(RBuffer *b, RBuffer *a) {
|
|
if (!b) return false;
|
|
if (b->fd != -1) {
|
|
r_buf_append_bytes (b, a->buf, a->length);
|
|
return true;
|
|
}
|
|
if (b->empty) {
|
|
b->length = 0;
|
|
b->empty = 0;
|
|
}
|
|
if ((b->buf = realloc (b->buf, b->length + a->length))) {
|
|
memmove (b->buf+b->length, a->buf, a->length);
|
|
b->length += a->length;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// ret copied length if successful, -1 if failed
|
|
static int r_buf_cpy(RBuffer *b, ut64 addr, ut8 *dst, const ut8 *src, int len, int write) {
|
|
int end;
|
|
if (!b || b->empty) {
|
|
return 0;
|
|
}
|
|
if (b->fd != -1) {
|
|
if (r_sandbox_lseek (b->fd, addr, SEEK_SET) == -1) {
|
|
// seek failed - print error here?
|
|
// return 0;
|
|
}
|
|
if (write) {
|
|
return r_sandbox_write (b->fd, src, len);
|
|
}
|
|
memset (dst, 0, len);
|
|
return r_sandbox_read (b->fd, dst, len);
|
|
}
|
|
if (b->sparse) {
|
|
if (write) {
|
|
// create new with src + len
|
|
if (sparse_write (b->sparse, addr, src, len) < 0) {
|
|
return -1;
|
|
}
|
|
} else {
|
|
// read from sparse and write into dst
|
|
memset (dst, 0xff, len);
|
|
if (sparse_read (b->sparse, addr, dst, len) < 0) {
|
|
return -1;
|
|
}
|
|
}
|
|
return len;
|
|
}
|
|
addr = (addr == R_BUF_CUR) ? b->cur : addr - b->base;
|
|
if (len < 1 || !dst || addr > b->length) {
|
|
return -1;
|
|
}
|
|
end = (int)(addr + len);
|
|
if (end > b->length) {
|
|
len -= end-b->length;
|
|
}
|
|
if (write) {
|
|
dst += addr;
|
|
} else {
|
|
src += addr;
|
|
}
|
|
memmove (dst, src, len);
|
|
b->cur = addr + len;
|
|
return len;
|
|
}
|
|
|
|
static int r_buf_fcpy_at (RBuffer *b, ut64 addr, ut8 *buf, const char *fmt, int n, int write) {
|
|
ut64 len, check_len;
|
|
int i, j, k, tsize, bigendian, m = 1;
|
|
if (!b || b->empty) return 0;
|
|
if (b->fd != -1) {
|
|
eprintf ("r_buf_fcpy_at not supported yet for r_buf_new_file\n");
|
|
return 0;
|
|
}
|
|
if (addr == R_BUF_CUR)
|
|
addr = b->cur;
|
|
else addr -= b->base;
|
|
if (addr == UT64_MAX || addr > b->length)
|
|
return -1;
|
|
tsize = 2;
|
|
for (i = len = 0; i < n; i++)
|
|
for (j = 0; fmt[j]; j++) {
|
|
switch (fmt[j]) {
|
|
case '0'...'9':
|
|
if (m == 1)
|
|
m = r_num_get (NULL, &fmt[j]);
|
|
continue;
|
|
case 's': tsize = 2; bigendian = 0; break;
|
|
case 'S': tsize = 2; bigendian = 1; break;
|
|
case 'i': tsize = 4; bigendian = 0; break;
|
|
case 'I': tsize = 4; bigendian = 1; break;
|
|
case 'l': tsize = 8; bigendian = 0; break;
|
|
case 'L': tsize = 8; bigendian = 1; break;
|
|
case 'c': tsize = 1; bigendian = 0; break;
|
|
default: return -1;
|
|
}
|
|
|
|
/* Avoid read/write out of bound.
|
|
tsize and m are not user controled, then don't
|
|
need to check possible overflow.
|
|
*/
|
|
if (!UT64_ADD (&check_len, len, tsize*m))
|
|
return -1;
|
|
if (!UT64_ADD (&check_len, check_len, addr))
|
|
return -1;
|
|
if (check_len > b->length) {
|
|
return check_len;
|
|
// return -1;
|
|
}
|
|
|
|
for (k = 0; k < m; k++) {
|
|
ut8* src1 = &b->buf[len+(k*tsize)];
|
|
ut8* src2 = &b->buf[addr+len+(k*tsize)];
|
|
void* dest1 = &buf[addr+len+(k*tsize)];
|
|
void* dest2 = &buf[len+(k*tsize)];
|
|
ut8* dest1_8 = (ut8*)dest1;
|
|
ut16* dest1_16 = (ut16*)dest1;
|
|
ut32* dest1_32 = (ut32*)dest1;
|
|
ut64* dest1_64 = (ut64*)dest1;
|
|
ut8* dest2_8 = (ut8*)dest2;
|
|
ut16* dest2_16 = (ut16*)dest2;
|
|
ut32* dest2_32 = (ut32*)dest2;
|
|
ut64* dest2_64 = (ut64*)dest2;
|
|
if (write) {
|
|
switch (tsize) {
|
|
case 1:
|
|
*dest1_8 = r_read_ble8 (src1);
|
|
break;
|
|
case 2:
|
|
*dest1_16 = r_read_ble16 (src1, bigendian);
|
|
break;
|
|
case 4:
|
|
*dest1_32 = r_read_ble32 (src1, bigendian);
|
|
break;
|
|
case 8:
|
|
*dest1_64 = r_read_ble64 (src1, bigendian);
|
|
break;
|
|
}
|
|
} else {
|
|
switch (tsize) {
|
|
case 1:
|
|
*dest2_8 = r_read_ble8 (src2);
|
|
break;
|
|
case 2:
|
|
*dest2_16 = r_read_ble16 (src2, bigendian);
|
|
break;
|
|
case 4:
|
|
*dest2_32 = r_read_ble32 (src2, bigendian);
|
|
break;
|
|
case 8:
|
|
*dest2_64 = r_read_ble64 (src2, bigendian);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
len += tsize * m;
|
|
m = 1;
|
|
}
|
|
b->cur = addr + len;
|
|
return len;
|
|
}
|
|
|
|
R_API ut8 *r_buf_get_at (RBuffer *b, ut64 addr, int *left) {
|
|
if (b->empty) return 0;
|
|
if (b->fd != -1) {
|
|
eprintf ("r_buf_get_at not supported for r_buf_new_file\n");
|
|
return 0;
|
|
}
|
|
if (addr == R_BUF_CUR) {
|
|
addr = b->cur;
|
|
} else {
|
|
addr -= b->base;
|
|
}
|
|
if (addr == UT64_MAX || addr > b->length) {
|
|
return NULL;
|
|
}
|
|
if (left) {
|
|
*left = b->length - addr;
|
|
}
|
|
return b->buf+addr;
|
|
}
|
|
|
|
//ret 0 if failed; ret copied length if successful
|
|
R_API int r_buf_read_at(RBuffer *b, ut64 addr, ut8 *buf, int len) {
|
|
st64 pa;
|
|
if (!b || !buf || len < 1) {
|
|
return 0;
|
|
}
|
|
#if R_BUF_CUR != UT64_MAX
|
|
#error R_BUF_CUR must be UT64_MAX
|
|
#endif
|
|
if (addr == R_BUF_CUR) {
|
|
addr = b->cur;
|
|
}
|
|
if (b->fd != -1) {
|
|
if (r_sandbox_lseek (b->fd, addr, SEEK_SET) == -1) {
|
|
return 0;
|
|
}
|
|
return r_sandbox_read (b->fd, buf, len);
|
|
}
|
|
if (!b->sparse) {
|
|
if (addr < b->base || len < 1) {
|
|
return 0;
|
|
}
|
|
pa = addr - b->base;
|
|
if (pa + len > b->length) {
|
|
memset (buf, 0xff, len);
|
|
len = b->length - pa;
|
|
if (len < 0) {
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
// must be +pa, but maybe its missused?
|
|
//return r_buf_cpy (b, addr, buf, b->buf+pa, len, false);
|
|
return r_buf_cpy (b, addr, buf, b->buf, len, false);
|
|
}
|
|
|
|
R_API int r_buf_fread_at (RBuffer *b, ut64 addr, ut8 *buf, const char *fmt, int n) {
|
|
return r_buf_fcpy_at (b, addr, buf, fmt, n, false);
|
|
}
|
|
|
|
//ret 0 or -1 if failed; ret copied length if success
|
|
R_API int r_buf_write_at(RBuffer *b, ut64 addr, const ut8 *buf, int len) {
|
|
if (!b || !buf || len < 1) {
|
|
return 0;
|
|
}
|
|
if (b->fd != -1) {
|
|
ut64 newlen = addr + len;
|
|
if (r_sandbox_lseek (b->fd, addr, SEEK_SET) == -1) {
|
|
return 0;
|
|
}
|
|
if (newlen > b->length) {
|
|
b->length = newlen;
|
|
ftruncate (b->fd, newlen);
|
|
}
|
|
return r_sandbox_write (b->fd, buf, len);
|
|
}
|
|
if (b->sparse) {
|
|
return (sparse_write (b->sparse, addr, buf, len) < 0) ? -1 : len;
|
|
}
|
|
if (b->empty) {
|
|
b->empty = 0;
|
|
free (b->buf);
|
|
b->buf = (ut8 *) malloc (addr + len);
|
|
}
|
|
return r_buf_cpy (b, addr, b->buf, buf, len, true);
|
|
}
|
|
|
|
R_API int r_buf_fwrite_at (RBuffer *b, ut64 addr, ut8 *buf, const char *fmt, int n) {
|
|
return r_buf_fcpy_at (b, addr, buf, fmt, n, true);
|
|
}
|
|
|
|
R_API void r_buf_deinit(RBuffer *b) {
|
|
if (!b) return;
|
|
if (b->fd != -1) {
|
|
r_sandbox_close (b->fd);
|
|
b->fd = -1;
|
|
return;
|
|
}
|
|
if (b->sparse) {
|
|
r_list_free (b->sparse);
|
|
b->sparse = NULL;
|
|
}
|
|
if (b->mmap) {
|
|
r_file_mmap_free (b->mmap);
|
|
b->mmap = NULL;
|
|
} else R_FREE (b->buf);
|
|
}
|
|
|
|
R_API void r_buf_free(RBuffer *b) {
|
|
if (!b) {
|
|
return;
|
|
}
|
|
if (!b->ro) {
|
|
r_buf_deinit (b);
|
|
}
|
|
R_FREE (b);
|
|
}
|
|
|
|
R_API int r_buf_append_string (RBuffer *b, const char *str) {
|
|
return r_buf_append_bytes (b, (const ut8*)str, strlen (str));
|
|
}
|
|
|
|
R_API char *r_buf_free_to_string(RBuffer *b) {
|
|
char *p;
|
|
if (!b) {
|
|
return NULL;
|
|
}
|
|
if (b->mmap) {
|
|
p = r_buf_to_string (b);
|
|
} else {
|
|
r_buf_append_bytes (b, (const ut8*)"", 1);
|
|
p = malloc (b->length + 1);
|
|
if (!p) {
|
|
return NULL;
|
|
}
|
|
memmove (p, b->buf, b->length);
|
|
p[b->length] = 0;
|
|
}
|
|
r_buf_free (b);
|
|
return p;
|
|
}
|
|
|
|
R_API bool r_buf_resize (RBuffer *b, ut64 newsize) {
|
|
if (b->mmap) {
|
|
return false;
|
|
}
|
|
if ((!b->sparse && !b->buf) || newsize < 1) {
|
|
return false;
|
|
}
|
|
if (b->sparse) {
|
|
ut64 last_addr = 0;
|
|
sparse_limits (b->sparse, 0, &last_addr);
|
|
int buf_len = newsize - last_addr;
|
|
if (buf_len > 0) {
|
|
ut8 *buf = malloc (buf_len);
|
|
if (buf) {
|
|
memset (buf, 0xff, buf_len);
|
|
sparse_write (b->sparse, last_addr, buf, buf_len);
|
|
free (buf);
|
|
return true;
|
|
}
|
|
}
|
|
eprintf ("Invalid resize for an sparse RBuffer\n");
|
|
return false;
|
|
}
|
|
ut8 *buf = calloc (newsize, 1);
|
|
if (buf) {
|
|
ut32 len = R_MIN (newsize, b->length);
|
|
memcpy (buf, b->buf, len);
|
|
memset (buf + len, 0xff, newsize - len);
|
|
/* commit */
|
|
free (b->buf);
|
|
b->buf = buf;
|
|
b->length = newsize;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|