mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-13 16:18:33 +00:00
34d41b897f
* Fix nested command repeaters * Add hardcoded limit for BB analysis (fix infinite analysis bug) - RAnal.Fcn now has a .depth field * Fix some r_cons usage related issues in rcore * Add 'pif' command. as in 'pdf' one resizes the blocksize * Fix '#' command/comment again * Fix r_io_read issue that was causing everything to go wrong
113 lines
2.8 KiB
C
113 lines
2.8 KiB
C
/* radare - LGPL - Copyright 2008-2012 pancake<nopcode.org> */
|
|
|
|
#include "r_io.h"
|
|
#include "r_lib.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
|
|
typedef struct {
|
|
int fd;
|
|
ut8 *buf;
|
|
ut32 size;
|
|
} RIOMalloc;
|
|
|
|
#define RIOMALLOC_FD(x) (((RIOMalloc*)x->data)->fd)
|
|
#define RIOMALLOC_SZ(x) (((RIOMalloc*)x->data)->size)
|
|
#define RIOMALLOC_BUF(x) (((RIOMalloc*)x->data)->buf)
|
|
|
|
static int __write(RIO *io, RIODesc *fd, const ut8 *buf, int count) {
|
|
if (fd == NULL || fd->data == NULL)
|
|
return -1;
|
|
if (io->off+count > RIOMALLOC_SZ (fd))
|
|
return -1;
|
|
memcpy (RIOMALLOC_BUF (fd)+io->off, buf, count);
|
|
return count;
|
|
}
|
|
|
|
static int __read(RIO *io, RIODesc *fd, ut8 *buf, int count) {
|
|
memset (buf, 0xff, count);
|
|
if (fd == NULL || fd->data == NULL)
|
|
return -1;
|
|
if (io->off>= RIOMALLOC_SZ (fd))
|
|
return -1;
|
|
if (io->off+count >= RIOMALLOC_SZ (fd))
|
|
count = RIOMALLOC_SZ (fd) - io->off;
|
|
memcpy (buf, RIOMALLOC_BUF (fd)+io->off, count);
|
|
return count;
|
|
}
|
|
|
|
static int __close(RIODesc *fd) {
|
|
RIOMalloc *riom;
|
|
if (fd == NULL || fd->data == NULL)
|
|
return -1;
|
|
riom = fd->data;
|
|
free (riom->buf);
|
|
riom->buf = NULL;
|
|
free (fd->data);
|
|
fd->data = NULL;
|
|
fd->state = R_IO_DESC_TYPE_CLOSED;
|
|
return 0;
|
|
}
|
|
|
|
static ut64 __lseek(struct r_io_t *io, RIODesc *fd, ut64 offset, int whence) {
|
|
switch (whence) {
|
|
case SEEK_SET: return offset;
|
|
case SEEK_CUR: return io->off + offset;
|
|
case SEEK_END: return RIOMALLOC_SZ (fd);
|
|
}
|
|
return offset;
|
|
}
|
|
|
|
static int __plugin_open(struct r_io_t *io, const char *pathname) {
|
|
return (
|
|
(!memcmp (pathname, "malloc://", 9)) ||
|
|
(!memcmp (pathname, "hex://", 6))
|
|
);
|
|
}
|
|
|
|
static RIODesc *__open(struct r_io_t *io, const char *pathname, int rw, int mode) {
|
|
if (__plugin_open (io, pathname)) {
|
|
RIOMalloc *mal = R_NEW (RIOMalloc);
|
|
mal->fd = -1; /* causes r_io_desc_new() to set the correct fd */
|
|
if (!memcmp (pathname, "hex://", 6)) {
|
|
mal->size = strlen (pathname);
|
|
mal->buf = malloc (mal->size);
|
|
memset (mal->buf, 0, mal->size);
|
|
mal->size = r_hex_str2bin (pathname+6, mal->buf);
|
|
} else {
|
|
mal->size = r_num_math (NULL, pathname+9);
|
|
if ((mal->size)>0) {
|
|
mal->buf = malloc (mal->size);
|
|
memset (mal->buf, '\0', mal->size);
|
|
} else {
|
|
eprintf ("Cannot allocate (%s) 0 bytes\n", pathname+9);
|
|
return NULL;
|
|
}
|
|
}
|
|
if (mal->buf != NULL)
|
|
return r_io_desc_new (&r_io_plugin_malloc, mal->fd, pathname, rw, mode, mal);
|
|
eprintf ("Cannot allocate (%s) %d bytes\n", pathname+9, mal->size);
|
|
free (mal);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
struct r_io_plugin_t r_io_plugin_malloc = {
|
|
.name = "malloc",
|
|
.desc = "memory allocation (malloc://1024 hex://10294505)",
|
|
.open = __open,
|
|
.close = __close,
|
|
.read = __read,
|
|
.plugin_open = __plugin_open,
|
|
.lseek = __lseek,
|
|
.write = __write,
|
|
};
|
|
|
|
#ifndef CORELIB
|
|
struct r_lib_struct_t radare_plugin = {
|
|
.type = R_LIB_TYPE_IO,
|
|
.data = &r_io_plugin_malloc
|
|
};
|
|
#endif
|