Initial commit

wiipax/client/lzma.c

@@ -0,0 +1,123 @@
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+
+#include "common.h"
+#include "lzma.h"
+
+#include "LzmaDec.h"
+
+static void *lz_malloc(void *p, size_t size) {
+	(void) p;
+	return malloc(size);
+}
+
+static void lz_free(void *p, void *address) {
+	(void) p;
+	free(address);
+}
+
+static ISzAlloc lz_alloc = { lz_malloc, lz_free };
+
+static lzma_t *lzma_alloc(const u32 len) {
+	lzma_t *lzma;
+
+	lzma = (lzma_t *) calloc(1, sizeof(lzma_t));
+
+	if (!lzma)
+		die("Failed to alloc 0x%x bytes", (u32) sizeof(lzma_t));
+
+	lzma->data = calloc(1, len);
+	if (!lzma->data)
+		die("Failed to alloc 0x%x bytes", len);
+
+	return lzma;
+}
+
+void lzma_free(lzma_t *lzma) {
+	free(lzma->data);
+	free(lzma);
+}
+
+lzma_t *lzma_compress(const u8 *src, const u32 len) {
+	lzma_t *lzma;
+	CLzmaEncProps props;
+	size_t len_out;
+	size_t len_props = LZMA_PROPS_SIZE;
+	SRes res;
+
+	printf("Compressing...");
+	fflush(stdout);
+
+	lzma = lzma_alloc(2*len);
+
+	LzmaEncProps_Init(&props);
+	props.level = 7;
+	LzmaEncProps_Normalize(&props);
+
+	len_out = 2*len;
+	res = LzmaEncode(lzma->data, &len_out, src, len, &props, lzma->props,
+					&len_props, 1, NULL, &lz_alloc, &lz_alloc);
+
+	if (res != SZ_OK)
+		die(" failed (%d)", res);
+
+	if (len_props != LZMA_PROPS_SIZE)
+		die(" failed: encoder propsize %u != %u", (u32) len_props,
+			LZMA_PROPS_SIZE);
+
+	printf(" %u -> %u = %3.2f%%\n", len, (u32) len_out,
+			100.0 * (float) len_out / (float) len);
+
+	lzma->len_in = len;
+	lzma->len_out = len_out;
+
+	return lzma;
+}
+
+void lzma_decode(const lzma_t *lzma, u8 *dst) {
+	SizeT len_in;
+	SizeT len_out;
+	ELzmaStatus status;
+	SRes res;
+
+	len_in = lzma->len_out;
+	len_out = lzma->len_in;
+
+	res = LzmaDecode(dst, &len_out, lzma->data, &len_in, lzma->props,
+					LZMA_PROPS_SIZE, LZMA_FINISH_END, &status, &lz_alloc);
+
+	if (res != SZ_OK)
+		die("Error decoding %d (%u)\n", res, status);
+
+	*dst = len_out;
+}
+
+void lzma_write(const char *filename, const lzma_t *lzma) {
+	int fd;
+	int i;
+
+	fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0755);
+	if (fd < 0)
+		perrordie("Could not open file");
+
+	if (write(fd, lzma->props, LZMA_PROPS_SIZE) != LZMA_PROPS_SIZE)
+		perrordie("Could not write to file");
+
+	u64 size = lzma->len_in;
+	for (i = 0; i < 8; ++i) {
+		u8 j = size & 0xff;
+		size >>= 8;
+		if (write(fd, &j, 1) != 1)
+			perrordie("Could not write to file");
+	}
+
+	if (write(fd, lzma->data, lzma->len_out) != lzma->len_out)
+		perrordie("Could not write to file");
+
+	close(fd);
+}
+