bzip2: port bzip2 1.0.4 to busybox. note: bzip2 code resides

in separate directory (archival/bz/*)
and is covered by BSD-style license.
code size: 13k

1 files changed, 166 insertions, 0 deletions

diff --git a/archival/bzip2.c b/archival/bzip2.c
new file mode 100644
index 000000000..04478eecc
--- /dev/null
+++ b/archival/bzip2.c
@@ -0,0 +1,166 @@
+/*
+ * Copyright (C) 2007 Denys Vlasenko <vda.linux@googlemail.com>
+ *
+ * This file uses bzip2 library code which is written
+ * by Julian Seward <jseward@bzip.org>.
+ * See README and LICENSE files in bz/ directory for more information
+ * about bzip2 library code.
+ */
+
+#include "libbb.h"
+
+/* This buys 6% speed for nearly 4k code */
+/*#define FAST_GROUP6 1*/
+
+#include "bz/bzlib.h"
+
+#include "bz/bzlib_private.h"
+
+#include "bz/blocksort.c"
+#include "bz/bzlib.c"
+#include "bz/compress.c"
+#include "bz/crctable.c"
+#include "bz/huffman.c"
+#include "bz/randtable.c"
+
+/* No point in being shy and having very small buffer here.
+ * bzip2 internal buffers are much bigger anyway, hundreds of kbytes.
+ * If iobuf is several pages long, malloc() may use mmap,
+ * making iobuf is page aligned and thus (maybe) have one memcpy less
+ * if kernel is clever enough.
+ */
+enum {
+	IOBUF_SIZE = 8 * 1024
+};
+
+/* Returns:
+ * <0 on write errors (examine errno),
+ * >0 on short writes (errno == 0)
+ * 0  no error (entire input consume, gimme more)
+ * on "impossible" errors (internal bzip2 compressor bug) dies
+ */
+static
+ssize_t bz_write(bz_stream *strm, void* rbuf, ssize_t rlen, void *wbuf)
+{
+	int n, n2, ret;
+
+	/* if (len == 0) return 0; */
+
+	strm->avail_in = rlen;
+	strm->next_in  = rbuf;
+	while (1) {
+		strm->avail_out = IOBUF_SIZE;
+		strm->next_out = wbuf;
+
+		ret = BZ2_bzCompress(strm, BZ_RUN);
+		if (ret != BZ_RUN_OK)
+			bb_error_msg_and_die("internal error %d", ret);
+
+		n = IOBUF_SIZE - strm->avail_out;
+		if (n) {
+			/* short reads must have errno == 0 */
+			errno = 0;
+			n2 = full_write(STDOUT_FILENO, wbuf, n);
+			if (n2 != n)
+				return n2 ? n2 : 1;
+		}
+
+		if (strm->avail_in == 0)
+			return 0;
+	}
+}
+
+
+/*---------------------------------------------------*/
+static
+USE_DESKTOP(long long) int bz_write_tail(bz_stream *strm, void *wbuf)
+{
+	int n, n2, ret;
+	USE_DESKTOP(long long) int total;
+
+	total = -1;
+	while (1) {
+		strm->avail_out = IOBUF_SIZE;
+		strm->next_out = wbuf;
+
+		ret = BZ2_bzCompress(strm, BZ_FINISH);
+		if (ret != BZ_FINISH_OK && ret != BZ_STREAM_END)
+			bb_error_msg_and_die("internal error %d", ret);
+
+		n = IOBUF_SIZE - strm->avail_out;
+		if (n) {
+			n2 = full_write(STDOUT_FILENO, wbuf, n);
+			if (n2 != n)
+				goto err;
+		}
+
+		if (ret == BZ_STREAM_END)
+			break;
+	}
+
+	total = 0 USE_DESKTOP( + strm->total_out );
+ err:
+	BZ2_bzCompressEnd(strm);
+	return total;
+}
+
+
+static
+USE_DESKTOP(long long) int compressStream(void)
+{
+	USE_DESKTOP(long long) int total;
+	ssize_t count;
+	bz_stream bzs; /* it's small */
+#define strm (&bzs)
+	char *iobuf;
+#define rbuf iobuf
+#define wbuf (iobuf + IOBUF_SIZE)
+
+	iobuf = xmalloc(2 * IOBUF_SIZE);
+
+	BZ2_bzCompressInit(strm, 9 /*blockSize100k*/);
+
+	while (1) {
+		count = full_read(STDIN_FILENO, rbuf, IOBUF_SIZE);
+		if (count < 0)
+			bb_perror_msg("read error");
+		if (count <= 0)
+			break;
+		count = bz_write(strm, rbuf, count, wbuf);
+		if (count) {
+			bb_perror_msg(count < 0 ? "write error" : "short write");
+			break;
+		}
+	}
+
+	total = bz_write_tail(strm, wbuf);
+	free(iobuf);
+	/* we had no error _only_ if count == 0 */
+	return count == 0 ? total : -1;
+}
+
+static
+char* make_new_name_bzip2(char *filename)
+{
+	return xasprintf("%s.bz2", filename);
+}
+
+int bzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int bzip2_main(int argc, char **argv)
+{
+	unsigned opt;
+
+	/* Must match bbunzip's constants OPT_STDOUT, OPT_FORCE! */
+	opt = getopt32(argv, "cfv" USE_BUNZIP2("d") "q123456789" );
+#if ENABLE_BUNZIP2 /* bunzip2_main may not be visible... */
+	if (opt & 0x8) // -d
+		return bunzip2_main(argc, argv);
+#endif
+	option_mask32 &= 0x7; /* ignore -q, -0..9 */
+	//if (opt & 0x1) // -c
+	//if (opt & 0x2) // -f
+	//if (opt & 0x4) // -v
+	argv += optind;
+
+	return bbunpack(argv, make_new_name_bzip2, compressStream);
+}