aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2019-09-27 14:05:20 +0200
committerDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2019-09-27 14:14:23 +0200
commitacce940aa48678632e58d0befacdc2787f1e43bf (patch)
tree78804b67679b1142a1c7fc9c5d1caeeab90e3b7c
parentb0d718a4b6628fc9673e2a1e2d26792aae3d0a0d (diff)
Add xz-utils based lzma compressor implementation
Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>
-rw-r--r--lib/sqfs/Makemodule.am1
-rw-r--r--lib/sqfs/comp/compressor.c1
-rw-r--r--lib/sqfs/comp/internal.h4
-rw-r--r--lib/sqfs/comp/lzma.c180
4 files changed, 186 insertions, 0 deletions
diff --git a/lib/sqfs/Makemodule.am b/lib/sqfs/Makemodule.am
index b4744d4..fe6101a 100644
--- a/lib/sqfs/Makemodule.am
+++ b/lib/sqfs/Makemodule.am
@@ -43,6 +43,7 @@ endif
if WITH_XZ
libsquashfs_la_SOURCES += lib/sqfs/comp/xz.c
+libsquashfs_la_SOURCES += lib/sqfs/comp/lzma.c
libsquashfs_la_CPPFLAGS += -DWITH_XZ
endif
diff --git a/lib/sqfs/comp/compressor.c b/lib/sqfs/comp/compressor.c
index 765558f..37f852c 100644
--- a/lib/sqfs/comp/compressor.c
+++ b/lib/sqfs/comp/compressor.c
@@ -22,6 +22,7 @@ static compressor_fun_t compressors[SQFS_COMP_MAX + 1] = {
#endif
#ifdef WITH_XZ
[SQFS_COMP_XZ] = xz_compressor_create,
+ [SQFS_COMP_LZMA] = lzma_compressor_create,
#endif
#ifdef WITH_LZO
[SQFS_COMP_LZO] = lzo_compressor_create,
diff --git a/lib/sqfs/comp/internal.h b/lib/sqfs/comp/internal.h
index f57f2ce..94b4c8c 100644
--- a/lib/sqfs/comp/internal.h
+++ b/lib/sqfs/comp/internal.h
@@ -12,6 +12,7 @@
#include "sqfs/predef.h"
#include "sqfs/compress.h"
#include "sqfs/error.h"
+#include "sqfs/data.h"
#include "sqfs/io.h"
#include "util.h"
@@ -37,4 +38,7 @@ sqfs_compressor_t *lz4_compressor_create(const sqfs_compressor_config_t *cfg);
SQFS_INTERNAL
sqfs_compressor_t *zstd_compressor_create(const sqfs_compressor_config_t *cfg);
+SQFS_INTERNAL
+sqfs_compressor_t *lzma_compressor_create(const sqfs_compressor_config_t *cfg);
+
#endif /* INTERNAL_H */
diff --git a/lib/sqfs/comp/lzma.c b/lib/sqfs/comp/lzma.c
new file mode 100644
index 0000000..82590c7
--- /dev/null
+++ b/lib/sqfs/comp/lzma.c
@@ -0,0 +1,180 @@
+/* SPDX-License-Identifier: LGPL-3.0-or-later */
+/*
+ * lzma.c
+ *
+ * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
+ */
+#define SQFS_BUILDING_DLL
+#include "config.h"
+
+#include <stdbool.h>
+#include <stdlib.h>
+#include <string.h>
+#include <lzma.h>
+
+#include "internal.h"
+
+#define LZMA_SIZE_OFFSET (5)
+#define LZMA_SIZE_BYTES (8)
+#define LZMA_HEADER_SIZE (13)
+
+#define LZMA_DEFAULT_LEVEL (5)
+#define MEMLIMIT (32 * 1024 * 1024)
+
+typedef struct {
+ sqfs_compressor_t base;
+ size_t block_size;
+} lzma_compressor_t;
+
+static int lzma_write_options(sqfs_compressor_t *base, sqfs_file_t *file)
+{
+ (void)base; (void)file;
+ return 0;
+}
+
+static int lzma_read_options(sqfs_compressor_t *base, sqfs_file_t *file)
+{
+ (void)base;
+ (void)file;
+ return SQFS_ERROR_UNSUPPORTED;
+}
+
+static ssize_t lzma_comp_block(sqfs_compressor_t *base, const uint8_t *in,
+ size_t size, uint8_t *out, size_t outsize)
+{
+ lzma_compressor_t *lzma = (lzma_compressor_t *)base;
+ lzma_stream strm = LZMA_STREAM_INIT;
+ lzma_options_lzma opt;
+ int ret;
+
+ if (outsize < LZMA_HEADER_SIZE)
+ return 0;
+
+ lzma_lzma_preset(&opt, LZMA_DEFAULT_LEVEL);
+ opt.dict_size = lzma->block_size;
+
+ if (lzma_alone_encoder(&strm, &opt) != LZMA_OK) {
+ lzma_end(&strm);
+ return SQFS_ERROR_COMRPESSOR;
+ }
+
+ strm.next_out = out;
+ strm.avail_out = outsize;
+ strm.next_in = in;
+ strm.avail_in = size;
+
+ ret = lzma_code(&strm, LZMA_FINISH);
+ lzma_end(&strm);
+
+ if (ret != LZMA_STREAM_END)
+ return ret == LZMA_OK ? 0 : SQFS_ERROR_COMRPESSOR;
+
+ out[LZMA_SIZE_OFFSET ] = size & 0xFF;
+ out[LZMA_SIZE_OFFSET + 1] = (size >> 8) & 0xFF;
+ out[LZMA_SIZE_OFFSET + 2] = (size >> 16) & 0xFF;
+ out[LZMA_SIZE_OFFSET + 3] = (size >> 24) & 0xFF;
+ out[LZMA_SIZE_OFFSET + 4] = 0;
+ out[LZMA_SIZE_OFFSET + 5] = 0;
+ out[LZMA_SIZE_OFFSET + 6] = 0;
+ out[LZMA_SIZE_OFFSET + 7] = 0;
+ return strm.total_out;
+}
+
+static ssize_t lzma_uncomp_block(sqfs_compressor_t *base, const uint8_t *in,
+ size_t size, uint8_t *out, size_t outsize)
+{
+ uint8_t lzma_header[LZMA_HEADER_SIZE];
+ lzma_stream strm = LZMA_STREAM_INIT;
+ size_t hdrsize;
+ int ret;
+ (void)base;
+
+ if (size < sizeof(lzma_header))
+ return SQFS_ERROR_CORRUPTED;
+
+ hdrsize = in[LZMA_SIZE_OFFSET] |
+ (in[LZMA_SIZE_OFFSET + 1] << 8) |
+ (in[LZMA_SIZE_OFFSET + 2] << 16) |
+ (in[LZMA_SIZE_OFFSET + 3] << 24);
+
+ if (hdrsize > outsize)
+ return 0;
+
+ if (lzma_alone_decoder(&strm, MEMLIMIT) != LZMA_OK) {
+ lzma_end(&strm);
+ return SQFS_ERROR_COMRPESSOR;
+ }
+
+ memcpy(lzma_header, in, sizeof(lzma_header));
+ memset(lzma_header + LZMA_SIZE_OFFSET, 0xFF, LZMA_SIZE_BYTES);
+
+ strm.next_out = out;
+ strm.avail_out = outsize;
+ strm.next_in = lzma_header;
+ strm.avail_in = sizeof(lzma_header);
+
+ ret = lzma_code(&strm, LZMA_RUN);
+
+ if (ret != LZMA_OK || strm.avail_in != 0) {
+ lzma_end(&strm);
+ return SQFS_ERROR_COMRPESSOR;
+ }
+
+ strm.next_in = in + sizeof(lzma_header);
+ strm.avail_in = size - sizeof(lzma_header);
+
+ ret = lzma_code(&strm, LZMA_FINISH);
+ lzma_end(&strm);
+
+ if (ret != LZMA_STREAM_END && ret != LZMA_OK)
+ return SQFS_ERROR_COMRPESSOR;
+
+ if (ret == LZMA_OK) {
+ if (strm.total_out < hdrsize || strm.avail_in != 0)
+ return 0;
+ }
+
+ return hdrsize;
+}
+
+static sqfs_compressor_t *lzma_create_copy(sqfs_compressor_t *cmp)
+{
+ lzma_compressor_t *copy = malloc(sizeof(*copy));
+
+ if (copy != NULL)
+ memcpy(copy, cmp, sizeof(*copy));
+
+ return (sqfs_compressor_t *)copy;
+}
+
+static void lzma_destroy(sqfs_compressor_t *base)
+{
+ free(base);
+}
+
+sqfs_compressor_t *lzma_compressor_create(const sqfs_compressor_config_t *cfg)
+{
+ sqfs_compressor_t *base;
+ lzma_compressor_t *lzma;
+
+ if (cfg->flags & ~SQFS_COMP_FLAG_GENERIC_ALL)
+ return NULL;
+
+ lzma = calloc(1, sizeof(*lzma));
+ base = (sqfs_compressor_t *)lzma;
+ if (lzma == NULL)
+ return NULL;
+
+ lzma->block_size = cfg->block_size;
+
+ if (lzma->block_size < SQFS_META_BLOCK_SIZE)
+ lzma->block_size = SQFS_META_BLOCK_SIZE;
+
+ base->destroy = lzma_destroy;
+ base->do_block = (cfg->flags & SQFS_COMP_FLAG_UNCOMPRESS) ?
+ lzma_uncomp_block : lzma_comp_block;
+ base->write_options = lzma_write_options;
+ base->read_options = lzma_read_options;
+ base->create_copy = lzma_create_copy;
+ return base;
+}