hak5
/
wifipineapple-openwrt
mirror of https://github.com/hak5/wifipineapple-openwrt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
wifipineapple-openwrt/tools/squashfs4/patches/110-lzma.patch

445 lines
14 KiB
Diff
Raw Blame History

--- a/squashfs-tools/mksquashfs.c
+++ b/squashfs-tools/mksquashfs.c
@@ -64,6 +64,18 @@
#include "global.h"
#include "sort.h"
#include "pseudo.h"
+#include "uncompress.h"
+
+#ifdef USE_LZMA
+#include <LzmaEnc.h>
+#include <LzmaDec.h>
+#define LZMA_DEFAULT_LEVEL 5
+#define LZMA_DEFAULT_DICT 0
+#define LZMA_DEFAULT_LC 1
+#define LZMA_DEFAULT_LP 2
+#define LZMA_DEFAULT_PB 2
+#define LZMA_DEFAULT_FB 32
+#endif
#ifdef SQUASHFS_TRACE
#define TRACE(s, args...) do { \
@@ -830,6 +842,19 @@ void sigalrm_handler()
rotate = (rotate + 1) % 4;
}
+#ifdef USE_LZMA
+static void *lzma_malloc(void *p, size_t size)
+{
+ (void)p;
+ return malloc(size);
+}
+static void lzma_free(void *p, void *addr)
+{
+ (void)p;
+ free(addr);
+}
+static ISzAlloc lzma_alloc = { lzma_malloc, lzma_free };
+#endif
unsigned int mangle2(z_stream **strm, char *d, char *s, int size,
int block_size, int uncompressed, int data_block)
@@ -841,6 +866,48 @@ unsigned int mangle2(z_stream **strm, ch
if(uncompressed)
goto notcompressed;
+#ifdef USE_LZMA
+ if (compression == LZMA_COMPRESSION) {
+ size_t outsize = block_size - LZMA_PROPS_SIZE;
+ size_t propsize = LZMA_PROPS_SIZE;
+ CLzmaEncProps props;
+
+ LzmaEncProps_Init(&props);
+ props.level = LZMA_DEFAULT_LEVEL;
+ props.dictSize = LZMA_DEFAULT_DICT;
+ props.lc = LZMA_DEFAULT_LC;
+ props.lp = LZMA_DEFAULT_LP;
+ props.pb = LZMA_DEFAULT_PB;
+ props.fb = LZMA_DEFAULT_FB;
+ props.numThreads = 1;
+
+ res = LzmaEncode((unsigned char *) d + LZMA_PROPS_SIZE, &outsize,
+ (unsigned char *) s, size,
+ &props, (unsigned char *) d, &propsize,
+ 1, NULL, &lzma_alloc, &lzma_alloc);
+ switch(res) {
+ case SZ_OK:
+ outsize += LZMA_PROPS_SIZE;
+ break;
+ case SZ_ERROR_DATA:
+ BAD_ERROR("lzma::compress failed, data error\n");
+ break;
+ case SZ_ERROR_MEM:
+ BAD_ERROR("lzma::compress failed, memory allocation error\n");
+ break;
+ case SZ_ERROR_PARAM:
+ BAD_ERROR("lzma::compress failed, invalid parameters\n");
+ break;
+ /* should not happen */
+ default:
+ BAD_ERROR("lzma::compress failed, unknown error\n");
+ break;
+ }
+
+ return outsize;
+ }
+#endif
+
if(stream == NULL) {
if((stream = *strm = malloc(sizeof(z_stream))) == NULL)
BAD_ERROR("mangle::compress failed, not enough "
@@ -1669,17 +1736,17 @@ struct file_buffer *get_fragment(struct
else
data = read_from_disk(start_block, size);
- res = uncompress((unsigned char *) buffer->data, &bytes,
+ res = uncompress_wrapper((unsigned char *) buffer->data, &bytes,
(const unsigned char *) data, size);
if(res != Z_OK) {
if(res == Z_MEM_ERROR)
- BAD_ERROR("zlib::uncompress failed, not enough "
+ BAD_ERROR("uncompress failed, not enough "
"memory\n");
else if(res == Z_BUF_ERROR)
- BAD_ERROR("zlib::uncompress failed, not enough "
+ BAD_ERROR("uncompress failed, not enough "
"room in output buffer\n");
else
- BAD_ERROR("zlib::uncompress failed,"
+ BAD_ERROR("uncompress failed,"
" unknown error %d\n", res);
}
} else if(compressed_buffer)
@@ -4282,6 +4349,10 @@ int main(int argc, char *argv[])
argv[0]);
exit(1);
}
+#ifdef USE_LZMA
+ } else if(strcmp(argv[i], "-lzma") == 0) {
+ compression = LZMA_COMPRESSION;
+#endif
} else if(strcmp(argv[i], "-ef") == 0) {
if(++i == argc) {
ERROR("%s: -ef missing filename\n", argv[0]);
@@ -4410,6 +4481,9 @@ printOptions:
ERROR("-b <block_size>\t\tset data block to "
"<block_size>. Default %d bytes\n",
SQUASHFS_FILE_SIZE);
+#ifdef USE_LZMA
+ ERROR("-lzma Enable LZMA compression\n");
+#endif
ERROR("-processors <number>\tUse <number> processors."
" By default will use number of\n");
ERROR("\t\t\tprocessors available\n");
@@ -4804,7 +4878,7 @@ restore_filesystem:
sBlk.bytes_used = bytes;
/* Only compression supported */
- sBlk.compression = ZLIB_COMPRESSION;
+ sBlk.compression = compression;
/* Xattrs are not currently supported */
sBlk.xattr_table_start = SQUASHFS_INVALID_BLK;
--- a/squashfs-tools/squashfs_fs.h
+++ b/squashfs-tools/squashfs_fs.h
@@ -229,6 +229,7 @@ typedef long long squashfs_block_t;
typedef long long squashfs_inode_t;
#define ZLIB_COMPRESSION 1
+#define LZMA_COMPRESSION 2
struct squashfs_super_block {
unsigned int s_magic;
--- a/squashfs-tools/Makefile
+++ b/squashfs-tools/Makefile
@@ -4,14 +4,20 @@ INCLUDEDIR = .
CFLAGS := -I$(INCLUDEDIR) -D_FILE_OFFSET_BITS=64 -D_LARGEFILE_SOURCE -D_GNU_SOURCE -O2
+ifdef USE_LZMA
+ LZMA_CFLAGS = -DUSE_LZMA
+ LZMA_LIB = -llzma
+ CFLAGS += $(LZMA_CFLAGS)
+endif
+
all: mksquashfs unsquashfs
-mksquashfs: mksquashfs.o read_fs.o sort.o swap.o pseudo.o
- $(CC) mksquashfs.o read_fs.o sort.o swap.o pseudo.o -lz -lpthread -lm -o $@
+mksquashfs: mksquashfs.o read_fs.o sort.o swap.o pseudo.o uncompress.o
+ $(CC) mksquashfs.o read_fs.o sort.o swap.o pseudo.o uncompress.o -lz -lpthread -lm $(LZMA_LIB) -o $@
-mksquashfs.o: mksquashfs.c squashfs_fs.h mksquashfs.h global.h sort.h squashfs_swap.h Makefile
+mksquashfs.o: mksquashfs.c squashfs_fs.h mksquashfs.h global.h sort.h squashfs_swap.h uncompress.h Makefile
-read_fs.o: read_fs.c squashfs_fs.h read_fs.h global.h squashfs_swap.h Makefile
+read_fs.o: read_fs.c squashfs_fs.h read_fs.h global.h squashfs_swap.h uncompress.h Makefile
sort.o: sort.c squashfs_fs.h global.h sort.h Makefile
@@ -19,18 +25,20 @@ swap.o: swap.c Makefile
pseudo.o: pseudo.c pseudo.h Makefile
-unsquashfs: unsquashfs.o unsquash-1.o unsquash-2.o unsquash-3.o unsquash-4.o swap.o
- $(CC) unsquashfs.o unsquash-1.o unsquash-2.o unsquash-3.o unsquash-4.o swap.o -lz -lpthread -lm -o $@
+uncompress.o: uncompress.c uncompress.h
+
+unsquashfs: unsquashfs.o unsquash-1.o unsquash-2.o unsquash-3.o unsquash-4.o swap.o uncompress.o
+ $(CC) unsquashfs.o unsquash-1.o unsquash-2.o unsquash-3.o unsquash-4.o swap.o uncompress.o -lz -lpthread -lm $(LZMA_LIB) -o $@
-unsquashfs.o: unsquashfs.h unsquashfs.c squashfs_fs.h squashfs_swap.h squashfs_compat.h global.h Makefile
+unsquashfs.o: unsquashfs.h unsquashfs.c squashfs_fs.h squashfs_swap.h squashfs_compat.h global.h uncompress.h Makefile
-unsquash-1.o: unsquashfs.h unsquash-1.c squashfs_fs.h squashfs_compat.h global.h Makefile
+unsquash-1.o: unsquashfs.h unsquash-1.c squashfs_fs.h squashfs_compat.h global.h uncompress.h Makefile
-unsquash-2.o: unsquashfs.h unsquash-2.c unsquashfs.h squashfs_fs.h squashfs_compat.h global.h Makefile
+unsquash-2.o: unsquashfs.h unsquash-2.c unsquashfs.h squashfs_fs.h squashfs_compat.h global.h uncompress.h Makefile
-unsquash-3.o: unsquashfs.h unsquash-3.c squashfs_fs.h squashfs_compat.h global.h Makefile
+unsquash-3.o: unsquashfs.h unsquash-3.c squashfs_fs.h squashfs_compat.h global.h uncompress.h Makefile
-unsquash-4.o: unsquashfs.h unsquash-4.c squashfs_fs.h squashfs_swap.h global.h Makefile
+unsquash-4.o: unsquashfs.h unsquash-4.c squashfs_fs.h squashfs_swap.h global.h uncompress.h Makefile
clean:
-rm -f *.o mksquashfs unsquashfs
--- a/squashfs-tools/read_fs.c
+++ b/squashfs-tools/read_fs.c
@@ -51,6 +51,7 @@ extern unsigned int get_guid(unsigned in
#include "squashfs_swap.h"
#include "read_fs.h"
#include "global.h"
+#include "uncompress.h"
#include <stdlib.h>
@@ -83,17 +84,17 @@ int read_block(int fd, long long start,
c_byte = SQUASHFS_COMPRESSED_SIZE(c_byte);
read_destination(fd, start + offset, c_byte, buffer);
- res = uncompress(block, &bytes, (const unsigned char *) buffer,
- c_byte);
+ res = uncompress_wrapper(block, &bytes,
+ (const unsigned char *) buffer, c_byte);
if(res != Z_OK) {
if(res == Z_MEM_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"memory\n");
else if(res == Z_BUF_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"room in output buffer\n");
else
- ERROR("zlib::uncompress failed, unknown error "
+ ERROR("uncompress failed, unknown error "
"%d\n", res);
return 0;
}
--- a/squashfs-tools/unsquashfs.c
+++ b/squashfs-tools/unsquashfs.c
@@ -24,6 +24,7 @@
#include "unsquashfs.h"
#include "squashfs_swap.h"
#include "squashfs_compat.h"
+#include "uncompress.h"
#include "read_fs.h"
struct cache *fragment_cache, *data_cache;
@@ -597,18 +598,17 @@ int read_block(long long start, long lon
if(read_bytes(start + offset, c_byte, buffer) == FALSE)
goto failed;
- res = uncompress((unsigned char *) block, &bytes,
+ res = uncompress_wrapper((unsigned char *) block, &bytes,
(const unsigned char *) buffer, c_byte);
-
if(res != Z_OK) {
if(res == Z_MEM_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"memory\n");
else if(res == Z_BUF_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"room in output buffer\n");
else
- ERROR("zlib::uncompress failed, unknown error "
+ ERROR("uncompress failed, unknown error "
"%d\n", res);
goto failed;
}
@@ -645,18 +645,17 @@ int read_data_block(long long start, uns
if(read_bytes(start, c_byte, data) == FALSE)
goto failed;
- res = uncompress((unsigned char *) block, &bytes,
+ res = uncompress_wrapper((unsigned char *) block, &bytes,
(const unsigned char *) data, c_byte);
-
if(res != Z_OK) {
if(res == Z_MEM_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"memory\n");
else if(res == Z_BUF_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"room in output buffer\n");
else
- ERROR("zlib::uncompress failed, unknown error "
+ ERROR("uncompress failed, unknown error "
"%d\n", res);
goto failed;
}
@@ -1459,7 +1458,7 @@ int read_super(char *source)
s_ops.read_inode = read_inode_4;
s_ops.read_uids_guids = read_uids_guids_4;
memcpy(&sBlk, &sBlk_4, sizeof(sBlk_4));
- return TRUE;
+ goto done;
}
/*
@@ -1548,6 +1547,9 @@ int read_super(char *source)
goto failed_mount;
}
+done:
+ compression = sBlk.compression;
+
return TRUE;
failed_mount:
@@ -1710,19 +1712,19 @@ void *deflator(void *arg)
int res;
unsigned long bytes = block_size;
- res = uncompress((unsigned char *) tmp, &bytes,
+ res = uncompress_wrapper((unsigned char *) tmp, &bytes,
(const unsigned char *) entry->data,
SQUASHFS_COMPRESSED_SIZE_BLOCK(entry->size));
if(res != Z_OK) {
if(res == Z_MEM_ERROR)
- ERROR("zlib::uncompress failed, not enough"
+ ERROR("uncompress failed, not enough"
"memory\n");
else if(res == Z_BUF_ERROR)
- ERROR("zlib::uncompress failed, not enough "
+ ERROR("uncompress failed, not enough "
"room in output buffer\n");
else
- ERROR("zlib::uncompress failed, unknown error "
+ ERROR("uncompress failed, unknown error "
"%d\n", res);
} else
memcpy(entry->data, tmp, bytes);
--- a/squashfs-tools/mksquashfs.h
+++ b/squashfs-tools/mksquashfs.h
@@ -41,4 +41,9 @@
#define SQUASHFS_SWAP_LONG_LONGS(s, d, n) \
memcpy(d, s, n * sizeof(long long))
#endif
+
+extern int uncompress_wrapper(unsigned char *dest, unsigned long *dest_len,
+ const unsigned char *src, unsigned long src_len);
+
+
#endif
--- /dev/null
+++ b/squashfs-tools/uncompress.c
@@ -0,0 +1,58 @@
+/*
+ * Copyright (c) 2009 Felix Fietkau <nbd@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2,
+ * or (at your option) any later version.
+ *
+ * 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, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ *
+ * uncompress.c
+ */
+
+
+
+#ifdef USE_LZMA
+#include <LzmaLib.h>
+#endif
+#include <zlib.h>
+#include "squashfs_fs.h"
+
+/* compression algorithm */
+int compression = ZLIB_COMPRESSION;
+
+
+int uncompress_wrapper(unsigned char *dest, unsigned long *dest_len,
+ const unsigned char *src, unsigned long src_len)
+{
+ int res;
+
+#ifdef USE_LZMA
+ if (compression == LZMA_COMPRESSION) {
+ size_t slen = src_len - LZMA_PROPS_SIZE;
+ res = LzmaUncompress((unsigned char *)dest, dest_len,
+ (const unsigned char *) src + LZMA_PROPS_SIZE, &slen,
+ (const unsigned char *) src, LZMA_PROPS_SIZE);
+ switch(res) {
+ case SZ_OK:
+ res = Z_OK;
+ break;
+ case SZ_ERROR_MEM:
+ res = Z_MEM_ERROR;
+ break;
+ }
+ } else
+#endif
+ res = uncompress(dest, dest_len, src, src_len);
+ return res;
+}
+
+
--- /dev/null
+++ b/squashfs-tools/uncompress.h
@@ -0,0 +1,29 @@
+/*
+ * Copyright (c) 2009 Felix Fietkau <nbd@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2,
+ * or (at your option) any later version.
+ *
+ * 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, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ *
+ * uncompress.h
+ */
+
+#ifdef USE_LZMA
+#include <LzmaLib.h>
+#endif
+
+extern int compression;
+extern int uncompress_wrapper(unsigned char *dest, unsigned long *dest_len,
+ const unsigned char *src, unsigned long src_len);
+
+
Reference in New Issue View Git Blame Copy Permalink