summaryrefslogtreecommitdiffstats
path: root/lib/uncompress.c
diff options
context:
space:
mode:
Diffstat (limited to 'lib/uncompress.c')
-rw-r--r--lib/uncompress.c159
1 files changed, 159 insertions, 0 deletions
diff --git a/lib/uncompress.c b/lib/uncompress.c
new file mode 100644
index 0000000000..80982f34d6
--- /dev/null
+++ b/lib/uncompress.c
@@ -0,0 +1,159 @@
+/*
+ * uncompress.c - uncompress files
+ *
+ * Copyright (c) 2011 Sascha Hauer <s.hauer@pengutronix.de>, Pengutronix
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+#include <common.h>
+#include <uncompress.h>
+#include <bunzip2.h>
+#include <gunzip.h>
+#include <lzo.h>
+#include <errno.h>
+#include <filetype.h>
+#include <malloc.h>
+#include <fs.h>
+
+static void *uncompress_buf;
+static unsigned int uncompress_size;
+
+void uncompress_err_stdout(char *x)
+{
+ printf("%s\n", x);
+}
+
+static int (*uncompress_fill_fn)(void*, unsigned int);
+
+static int uncompress_fill(void *buf, unsigned int len)
+{
+ int total = 0;
+
+ if (uncompress_size) {
+ int now = min(len, uncompress_size);
+
+ memcpy(buf, uncompress_buf, now);
+ uncompress_size -= now;
+ len -= now;
+ total = now;
+ buf += now;
+ }
+
+ if (len) {
+ int ret = uncompress_fill_fn(buf, len);
+ if (ret < 0)
+ return ret;
+ total += ret;
+ }
+
+ return total;
+}
+
+int uncompress(unsigned char *inbuf, int len,
+ int(*fill)(void*, unsigned int),
+ int(*flush)(void*, unsigned int),
+ unsigned char *output,
+ int *pos,
+ void(*error_fn)(char *x))
+{
+ enum filetype ft;
+ int (*compfn)(unsigned char *inbuf, int len,
+ int(*fill)(void*, unsigned int),
+ int(*flush)(void*, unsigned int),
+ unsigned char *output,
+ int *pos,
+ void(*error)(char *x));
+ int ret;
+ char *err;
+
+ BUG_ON(uncompress_size);
+
+ if (inbuf) {
+ ft = file_detect_type(inbuf);
+ } else {
+ if (!fill)
+ return -EINVAL;
+
+ uncompress_fill_fn = fill;
+ uncompress_buf = xzalloc(32);
+ uncompress_size = 32;
+
+ ret = fill(uncompress_buf, 32);
+ if (ret < 0)
+ goto err;
+
+ ft = file_detect_type(uncompress_buf);
+ }
+
+ switch (ft) {
+#ifdef CONFIG_BZLIB
+ case filetype_bzip2:
+ compfn = bunzip2;
+ break;
+#endif
+#ifdef CONFIG_ZLIB
+ case filetype_gzip:
+ compfn = gunzip;
+ break;
+#endif
+#ifdef CONFIG_LZO_DECOMPRESS
+ case filetype_lzo_compressed:
+ compfn = decompress_unlzo;
+ break;
+#endif
+ default:
+ err = asprintf("cannot handle filetype %s", file_type_to_string(ft));
+ error_fn(err);
+ free(err);
+ ret = -ENOSYS;
+ goto err;
+ }
+
+ ret = compfn(inbuf, len, fill ? uncompress_fill : NULL,
+ flush, output, pos, error_fn);
+err:
+ free(uncompress_buf);
+ uncompress_size = 0;
+
+ return ret;
+}
+
+static int uncompress_infd, uncompress_outfd;
+
+static int fill_fd(void *buf, unsigned int len)
+{
+ return read(uncompress_infd, buf, len);
+}
+
+static int flush_fd(void *buf, unsigned int len)
+{
+ return write(uncompress_outfd, buf, len);
+}
+
+int uncompress_fd_to_fd(int infd, int outfd,
+ void(*error_fn)(char *x))
+{
+ uncompress_infd = infd;
+ uncompress_outfd = outfd;
+
+ return uncompress(NULL, 0,
+ fill_fd,
+ flush_fd,
+ NULL,
+ NULL,
+ error_fn);
+}