Magisk/jni/magiskboot/compress.c

118 lines
2.5 KiB
C
Raw Normal View History

2017-02-28 14:56:13 +01:00
#include <zlib.h>
#include <lzma.h>
2017-02-27 22:37:47 +01:00
#include "magiskboot.h"
void gzip(int dec, const char* filename, unsigned char* buf, size_t size) {
int ret, flush, have, pos = 0;
z_stream strm;
unsigned char out[CHUNK];
int fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
if (fd < 0)
error(1, "Unable to create %s", filename);
strm.zalloc = Z_NULL;
strm.zfree = Z_NULL;
strm.opaque = Z_NULL;
if (dec) {
ret = inflateInit2(&strm, windowBits | ZLIB_GZIP);
} else {
ret = deflateInit2(&strm, 9, Z_DEFLATED, windowBits | ZLIB_GZIP, memLevel, Z_DEFAULT_STRATEGY);
}
if (ret != Z_OK)
error(1, "Unable to init zlib");
do {
strm.next_in = buf + pos;
2017-02-28 14:56:13 +01:00
if (pos + CHUNK >= size) {
2017-02-27 22:37:47 +01:00
strm.avail_in = size - pos;
pos = size;
flush = Z_FINISH;
} else {
strm.avail_in = CHUNK;
pos += CHUNK;
flush = Z_NO_FLUSH;
}
do {
strm.avail_out = CHUNK;
strm.next_out = out;
if (dec) {
inflate(&strm, flush);
} else {
deflate(&strm, flush);
}
have = CHUNK - strm.avail_out;
if (write(fd, out, have) != have)
error(1, "Error in writing %s", filename);
} while (strm.avail_out == 0);
} while(pos < size);
if (dec) {
inflateEnd(&strm);
} else {
deflateEnd(&strm);
}
close(fd);
}
2017-02-28 14:56:13 +01:00
void lzma(int dec, const char* filename, unsigned char* buf, size_t size) {
int have, pos = 0;
lzma_ret ret;
lzma_stream strm = LZMA_STREAM_INIT;
lzma_options_lzma opt;
lzma_action action = LZMA_RUN;
unsigned char out[BUFSIZ];
int fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
if (fd < 0)
error(1, "Unable to create %s", filename);
if (dec) {
ret = lzma_auto_decoder(&strm, UINT64_MAX, 0);
} else {
lzma_lzma_preset(&opt, LZMA_PRESET_DEFAULT);
lzma_filter filters[] = {
{ .id = LZMA_FILTER_LZMA2, .options = &opt },
{ .id = LZMA_VLI_UNKNOWN, .options = NULL },
};
ret = lzma_stream_encoder(&strm, filters, LZMA_CHECK_CRC64);
}
if (ret != LZMA_OK)
error(1, "Unable to init lzma");
do {
strm.next_in = buf + pos;
if (pos + BUFSIZ >= size) {
strm.avail_in = size - pos;
pos = size;
action = LZMA_FINISH;
} else {
strm.avail_in = BUFSIZ;
pos += BUFSIZ;
}
do {
strm.avail_out = BUFSIZ;
strm.next_out = out;
ret = lzma_code(&strm, action);
have = BUFSIZ - strm.avail_out;
if (write(fd, out, have) != have)
error(1, "Error in writing %s", filename);
} while (strm.avail_out == 0 && ret == LZMA_OK);
if (ret != LZMA_OK && ret != LZMA_STREAM_END)
error(1, "LZMA error %d!", ret);
} while (pos < size);
lzma_end(&strm);
}