170 lines
3.4 KiB
C
170 lines
3.4 KiB
C
/*
|
|
* 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.
|
|
*
|
|
*/
|
|
#include <common.h>
|
|
#include <uncompress.h>
|
|
#include <bunzip2.h>
|
|
#include <gunzip.h>
|
|
#include <lzo.h>
|
|
#include <linux/decompress/unlz4.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;
|
|
|
|
if (inbuf) {
|
|
ft = file_detect_type(inbuf, len);
|
|
uncompress_buf = NULL;
|
|
uncompress_size = 0;
|
|
} 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, 32);
|
|
}
|
|
|
|
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
|
|
#ifdef CONFIG_LZ4_DECOMPRESS
|
|
case filetype_lz4_compressed:
|
|
compfn = decompress_unlz4;
|
|
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);
|
|
|
|
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);
|
|
}
|
|
|
|
int uncompress_fd_to_buf(int infd, void *output,
|
|
void(*error_fn)(char *x))
|
|
{
|
|
uncompress_infd = infd;
|
|
|
|
return uncompress(NULL, 0, fill_fd, NULL, output, NULL, error_fn);
|
|
}
|