mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-13 16:18:33 +00:00
110 lines
2.5 KiB
C
110 lines
2.5 KiB
C
/* radare - LGPL - Copyright 2008-2012 - pancake */
|
|
|
|
#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 RIOHTTP_FD(x) (((RIOMalloc*)x->data)->fd)
|
|
#define RIOHTTP_SZ(x) (((RIOMalloc*)x->data)->size)
|
|
#define RIOHTTP_BUF(x) (((RIOMalloc*)x->data)->buf)
|
|
|
|
static int __write(struct r_io_t *io, RIODesc *fd, const ut8 *buf, int count) {
|
|
if (fd == NULL || fd->data == NULL)
|
|
return -1;
|
|
if (io->off+count >= RIOHTTP_SZ (fd))
|
|
return -1;
|
|
memcpy (RIOHTTP_BUF (fd)+io->off, buf, count);
|
|
return count;
|
|
}
|
|
|
|
static int __read(struct r_io_t *io, RIODesc *fd, ut8 *buf, int count) {
|
|
int sz = RIOHTTP_SZ (fd);
|
|
if (fd == NULL || fd->data == NULL)
|
|
return -1;
|
|
if (io->off >= sz)
|
|
return -1;
|
|
if (io->off+count >= sz)
|
|
count = sz-io->off;
|
|
memcpy (buf, RIOHTTP_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 RIOHTTP_SZ (fd);
|
|
}
|
|
return offset;
|
|
}
|
|
|
|
static int __plugin_open(struct r_io_t *io, const char *pathname) {
|
|
return (!memcmp (pathname, "http://", 7));
|
|
}
|
|
|
|
static inline int getmalfd (RIOMalloc *mal) {
|
|
return 0xfffffff & (int)(size_t)mal->buf;
|
|
}
|
|
|
|
static RIODesc *__open(struct r_io_t *io, const char *pathname, int rw, int mode) {
|
|
char *out;
|
|
int rlen, code;
|
|
if (__plugin_open (io, pathname)) {
|
|
RIOMalloc *mal = R_NEW0 (RIOMalloc);
|
|
out = r_socket_http_get (pathname, &code, &rlen);
|
|
if (!out || rlen<1)
|
|
return NULL;
|
|
mal->size = rlen;
|
|
mal->buf = malloc (mal->size+1);
|
|
if (mal->buf != NULL) {
|
|
mal->fd = getmalfd (mal);
|
|
memcpy (mal->buf, out, rlen);
|
|
free (out);
|
|
return r_io_desc_new (&r_io_plugin_http, mal->fd, pathname, rw, mode, mal);
|
|
}
|
|
eprintf ("Cannot allocate (%s) %d bytes\n", pathname+9, mal->size);
|
|
free (mal);
|
|
free (out);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
struct r_io_plugin_t r_io_plugin_http = {
|
|
.name = "http",
|
|
.desc = "http get (http://radare.org/)",
|
|
.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_http
|
|
};
|
|
#endif
|