Add generic uncompress function
uncompress() has the same prototype as the various kernel decompress functions. It automatically detects the compression type and selects the correct uncompress function. Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
This commit is contained in:
parent
83d0904980
commit
eb1c2e7de6
|
@ -0,0 +1,19 @@
|
|||
#ifndef __UNCOMPRESS_H
|
||||
#define __UNCOMPRESS_H
|
||||
|
||||
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));
|
||||
|
||||
int uncompress_fd_to_fd(int infd, int outfd,
|
||||
void(*error_fn)(char *x));
|
||||
|
||||
int uncompress_fd_to_buf(int infd, void *output,
|
||||
void(*error_fn)(char *x));
|
||||
|
||||
void uncompress_err_stdout(char *);
|
||||
|
||||
#endif /* __UNCOMPRESS_H */
|
|
@ -33,3 +33,4 @@ obj-$(CONFIG_MD5) += md5.o
|
|||
obj-$(CONFIG_SHA1) += sha1.o
|
||||
obj-$(CONFIG_SHA256) += sha256.o
|
||||
obj-$(CONFIG_FDT) += fdt/
|
||||
obj-y += uncompress.o
|
||||
|
|
|
@ -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);
|
||||
}
|
Loading…
Reference in New Issue