aboutsummaryrefslogtreecommitdiff
path: root/lib/io/compress/ostream_compressor.c
diff options
context:
space:
mode:
authorDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2022-06-26 22:51:27 +0200
committerDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2022-07-08 19:17:35 +0200
commit359d71e90050a8b83f7bc7d2ecd4ff29c477cc22 (patch)
treedf2b7a585b78e6776446d4ca9be62a22db27f0a7 /lib/io/compress/ostream_compressor.c
parent2087cc237cd0fe1ed29ebf891648bacb46f4833b (diff)
Cleanup: rename libfstream to libio, split headers
Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>
Diffstat (limited to 'lib/io/compress/ostream_compressor.c')
-rw-r--r--lib/io/compress/ostream_compressor.c108
1 files changed, 108 insertions, 0 deletions
diff --git a/lib/io/compress/ostream_compressor.c b/lib/io/compress/ostream_compressor.c
new file mode 100644
index 0000000..314ce6b
--- /dev/null
+++ b/lib/io/compress/ostream_compressor.c
@@ -0,0 +1,108 @@
+/* SPDX-License-Identifier: GPL-3.0-or-later */
+/*
+ * ostream_compressor.c
+ *
+ * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
+ */
+#include "../internal.h"
+
+static int comp_append(ostream_t *strm, const void *data, size_t size)
+{
+ ostream_comp_t *comp = (ostream_comp_t *)strm;
+ size_t diff;
+
+ while (size > 0) {
+ if (comp->inbuf_used >= BUFSZ) {
+ if (comp->flush_inbuf(comp, false))
+ return -1;
+ }
+
+ diff = BUFSZ - comp->inbuf_used;
+
+ if (diff > size)
+ diff = size;
+
+ memcpy(comp->inbuf + comp->inbuf_used, data, diff);
+
+ comp->inbuf_used += diff;
+ data = (const char *)data + diff;
+ size -= diff;
+ }
+
+ return 0;
+}
+
+static int comp_flush(ostream_t *strm)
+{
+ ostream_comp_t *comp = (ostream_comp_t *)strm;
+
+ if (comp->inbuf_used > 0) {
+ if (comp->flush_inbuf(comp, true))
+ return -1;
+ }
+
+ return comp->wrapped->flush(comp->wrapped);
+}
+
+static const char *comp_get_filename(ostream_t *strm)
+{
+ ostream_comp_t *comp = (ostream_comp_t *)strm;
+
+ return comp->wrapped->get_filename(comp->wrapped);
+}
+
+static void comp_destroy(sqfs_object_t *obj)
+{
+ ostream_comp_t *comp = (ostream_comp_t *)obj;
+
+ comp->cleanup(comp);
+ sqfs_destroy(comp->wrapped);
+ free(comp);
+}
+
+ostream_t *ostream_compressor_create(ostream_t *strm, int comp_id)
+{
+ ostream_comp_t *comp = NULL;
+ sqfs_object_t *obj;
+ ostream_t *base;
+
+ switch (comp_id) {
+ case IO_COMPRESSOR_GZIP:
+#ifdef WITH_GZIP
+ comp = ostream_gzip_create(strm->get_filename(strm));
+#endif
+ break;
+ case IO_COMPRESSOR_XZ:
+#ifdef WITH_XZ
+ comp = ostream_xz_create(strm->get_filename(strm));
+#endif
+ break;
+ case IO_COMPRESSOR_ZSTD:
+#if defined(WITH_ZSTD) && defined(HAVE_ZSTD_STREAM)
+ comp = ostream_zstd_create(strm->get_filename(strm));
+#endif
+ break;
+ case IO_COMPRESSOR_BZIP2:
+#ifdef WITH_BZIP2
+ comp = ostream_bzip2_create(strm->get_filename(strm));
+#endif
+ break;
+ default:
+ break;
+ }
+
+ if (comp == NULL)
+ return NULL;
+
+ comp->wrapped = strm;
+ comp->inbuf_used = 0;
+
+ base = (ostream_t *)comp;
+ base->append = comp_append;
+ base->flush = comp_flush;
+ base->get_filename = comp_get_filename;
+
+ obj = (sqfs_object_t *)comp;
+ obj->destroy = comp_destroy;
+ return base;
+}