radare2/libr/util/zip.c
2017-05-09 23:53:20 +02:00

82 lines
1.8 KiB
C

/* radare - LGPL - Copyright 2014-2015 - pancake */
#include <r_util.h>
#include <zlib.h>
// set a maximum output buffer of 50MB
#define MAXOUT 50000000
static const char *gzerr(int n) {
const char *errors[] = {
"",
"file error", /* Z_ERRNO (-1) */
"stream error", /* Z_STREAM_ERROR (-2) */
"data error", /* Z_DATA_ERROR (-3) */
"insufficient memory", /* Z_MEM_ERROR (-4) */
"buffer error", /* Z_BUF_ERROR (-5) */
"incompatible version",/* Z_VERSION_ERROR (-6) */
};
if (n<1 || n>6) {
return "unknown";
}
return errors[n];
}
R_API ut8 *r_inflate(const ut8 *src, int srcLen, int *srcConsumed, int *dstLen) {
int err = 0;
int out_size = 0;
ut8 *dst = NULL;
z_stream stream;
if (srcLen <= 0) {
return NULL;
}
memset (&stream, 0, sizeof (z_stream));
stream.avail_in = srcLen;
stream.next_in = (Bytef *) src;
stream.zalloc = Z_NULL;
stream.zfree = Z_NULL;
stream.opaque = Z_NULL;
// + 32 tells zlib not to care whether the stream is a zlib or gzip stream
if (inflateInit2 (&stream, MAX_WBITS + 32) != Z_OK) {
return NULL;
}
do {
if (stream.avail_out == 0) {
if (! (dst = realloc (dst, stream.total_out + srcLen*2)))
goto err_exit;
out_size += srcLen*2;
if (out_size > MAXOUT)
goto err_exit;
stream.next_out = dst + stream.total_out;
stream.avail_out = srcLen * 2;
}
err = inflate (&stream, Z_NO_FLUSH);
if (err<0) {
eprintf ("inflate error: %d %s\n",
err, gzerr (-err));
goto err_exit;
}
} while (err != Z_STREAM_END);
if (dstLen) {
*dstLen = stream.total_out;
}
if (srcConsumed) {
*srcConsumed = (const ut8*)stream.next_in-(const ut8*)src;
}
inflateEnd (&stream);
return dst;
err_exit:
inflateEnd (&stream);
free (dst);
return NULL;
}