Initial import of the io.gzip plugin

libr/include/r_io.h

@@ -445,6 +445,7 @@ extern RIOPlugin r_io_plugin_mmap;
 extern RIOPlugin r_io_plugin_default;
 extern RIOPlugin r_io_plugin_ihex;
 extern RIOPlugin r_io_plugin_self;
+extern RIOPlugin r_io_plugin_gzip;
 #endif
 
 #ifdef __cplusplus

libr/io/p/gzip.mk

@@ -0,0 +1,16 @@
+OBJ_GZIP=io_gzip.o
+
+STATIC_OBJ+=${OBJ_GZIP}
+TARGET_GZIP=io_gzip.${EXT_SO}
+ALL_TARGETS+=${TARGET_GZIP}
+
+ifeq (${WITHPIC},0)
+LINKFLAGS+=../../util/libr_util.a
+LINKFLAGS+=../../io/libr_io.a
+else
+LINKFLAGS+=-L../../util -lr_util
+LINKFLAGS+=-L.. -lr_io
+endif
+
+${TARGET_GZIP}: ${OBJ_GZIP}
+	${CC_LIB} $(call libname,io_gzip) ${CFLAGS} -o ${TARGET_GZIP} ${OBJ_GZIP} ${LINKFLAGS}

libr/io/p/io_gzip.c

@@ -0,0 +1,143 @@
+/* radare - LGPL - Copyright 2008-2014 - pancake */
+
+#include "r_io.h"
+#include "r_lib.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/types.h>
+
+// copypasta from RIOMalloc... we can heavily simplify this with RBuffer
+
+typedef struct {
+	int fd;
+	ut8 *buf;
+	ut32 size;
+	ut64 offset;
+} RIOGzip;
+
+#define RIOMALLOC_FD(x) (((RIOGzip*)x->data)->fd)
+#define RIOMALLOC_SZ(x) (((RIOGzip*)x->data)->size)
+#define RIOMALLOC_BUF(x) (((RIOGzip*)x->data)->buf)
+#define RIOMALLOC_OFF(x) (((RIOGzip*)x->data)->offset)
+
+static int __write(RIO *io, RIODesc *fd, const ut8 *buf, int count) {
+	if (fd == NULL || fd->data == NULL)
+		return -1;
+	if (RIOMALLOC_OFF (fd) > RIOMALLOC_SZ (fd))
+		return -1;
+	if (RIOMALLOC_OFF (fd) + count > RIOMALLOC_SZ (fd))
+		count -= (RIOMALLOC_OFF (fd) + count-(RIOMALLOC_SZ (fd)));
+
+	if (count > 0) {
+		memcpy (RIOMALLOC_BUF (fd) + RIOMALLOC_OFF (fd), buf, count);
+		RIOMALLOC_OFF (fd) += count;
+		return count;
+	}
+	return -1;
+}
+
+
+static int __resize(RIO *io, RIODesc *fd, ut64 count) {
+	ut8 * new_buf = NULL;
+	if (fd == NULL || fd->data == NULL || count == 0)
+		return -1;
+	if (RIOMALLOC_OFF (fd) > RIOMALLOC_SZ (fd))
+		return -1;
+	new_buf = malloc (count);
+	if (!new_buf) return -1;
+	memcpy (new_buf, RIOMALLOC_BUF (fd), R_MIN(count, RIOMALLOC_SZ (fd)));
+	if (count > RIOMALLOC_SZ (fd) )
+		memset (new_buf+RIOMALLOC_SZ (fd), 0, count-RIOMALLOC_SZ (fd));
+
+	free (RIOMALLOC_BUF (fd));
+	RIOMALLOC_BUF (fd) = new_buf;
+	RIOMALLOC_SZ (fd) = 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 (RIOMALLOC_OFF (fd) > RIOMALLOC_SZ (fd))
+		return -1;
+	if (RIOMALLOC_OFF (fd) + count >= RIOMALLOC_SZ (fd))
+		count = RIOMALLOC_SZ (fd) - RIOMALLOC_OFF (fd);
+	memcpy (buf, RIOMALLOC_BUF (fd) + RIOMALLOC_OFF (fd), count);
+	return count;
+}
+
+static int __close(RIODesc *fd) {
+	RIOGzip *riom;
+	if (fd == NULL || fd->data == NULL)
+		return -1;
+	eprintf ("TODO: Writing changes into gzipped files is not yet supported\n");
+	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(RIO* io, RIODesc *fd, ut64 offset, int whence) {
+	ut64 r_offset = offset;
+	if (!fd->data)
+		return offset;
+	switch (whence) {
+	case SEEK_SET:
+		r_offset = (offset <= RIOMALLOC_SZ (fd)) ? offset : RIOMALLOC_SZ (fd);
+		break;
+	case SEEK_CUR:
+		r_offset = (RIOMALLOC_OFF (fd) + offset <= RIOMALLOC_SZ (fd)) ? RIOMALLOC_OFF (fd) + offset : RIOMALLOC_SZ (fd);
+		break;
+	case SEEK_END:
+		r_offset = RIOMALLOC_SZ (fd);
+		break;
+	}
+	RIOMALLOC_OFF (fd) = r_offset;
+	return RIOMALLOC_OFF (fd);
+}
+
+static int __plugin_open(struct r_io_t *io, const char *pathname, ut8 many) {
+	return (!strncmp (pathname, "gzip://", 7));
+}
+
+static RIODesc *__open(RIO *io, const char *pathname, int rw, int mode) {
+	if (__plugin_open (io, pathname, 0)) {
+		RIOGzip *mal = R_NEW0 (RIOGzip);
+		int len;
+		char *file = r_file_slurp (pathname+7, &len);
+		mal->buf = r_gunzip (file, len, &mal->size);
+		if (mal->buf) {
+			RETURN_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_gzip = {
+	.name = "gzip",
+	.desc = "read/write gzipped files",
+	.license = "LGPL3",
+	.open = __open,
+	.close = __close,
+	.read = __read,
+	.plugin_open = __plugin_open,
+	.lseek = __lseek,
+	.write = __write,
+	.resize = __resize,
+};
+
+#ifndef CORELIB
+struct r_lib_struct_t radare_plugin = {
+	.type = R_LIB_TYPE_IO,
+	.data = &r_io_plugin_gzip
+};
+#endif

plugins.def.cfg

@@ -137,6 +137,7 @@ fs.sfs
 io.debug
 io.rap
 io.ewf
+io.gzip
 io.http
 io.bfdbg
 io.gdb