mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-08 10:07:02 -06:00
Add options to brotli and implement clean brotli .br support
- allow to specify brotli window size - parameter -m0=brotli:long=n, BROTLI_MAX_WINDOW_BITS (24) used by default in brotli-mt, smaller == faster - note that :long can be set up to BROTLI_LARGE_MAX_WINDOW_BITS (30), whereas :wlog can be set up to BROTLI_MAX_WINDOW_BITS (24) only... - todo: check whether set of BROTLI_PARAM_LARGE_WINDOW to BROTLI_TRUE is needed if (lgwin > BROTLI_MAX_WINDOW_BITS) - implementation of single-threaded brotli compression / decompression for .br data Signed-off-by: Sergey G. Brester <info@sebres.de> Reviewed-by: Tino Reichardt <milky-7zip@mcmilk.de>
This commit is contained in:
@@ -50,7 +50,7 @@ struct writelist {
|
||||
|
||||
struct BROTLIMT_DCtx_s {
|
||||
|
||||
/* threads: 1..BROTLIMT_THREAD_MAX */
|
||||
/* threads: 0, 1..BROTLIMT_THREAD_MAX */
|
||||
int threads;
|
||||
|
||||
/* should be used for read from input */
|
||||
@@ -96,7 +96,7 @@ BROTLIMT_DCtx *BROTLIMT_createDCtx(int threads, int inputsize)
|
||||
return 0;
|
||||
|
||||
/* check threads value */
|
||||
if (threads < 1 || threads > BROTLIMT_THREAD_MAX)
|
||||
if (threads < 0 || threads > BROTLIMT_THREAD_MAX)
|
||||
return 0;
|
||||
|
||||
/* setup ctx */
|
||||
@@ -112,12 +112,16 @@ BROTLIMT_DCtx *BROTLIMT_createDCtx(int threads, int inputsize)
|
||||
else
|
||||
ctx->inputsize = 1024 * 64; /* 64K buffer */
|
||||
|
||||
pthread_mutex_init(&ctx->read_mutex, NULL);
|
||||
pthread_mutex_init(&ctx->write_mutex, NULL);
|
||||
if (threads) {
|
||||
pthread_mutex_init(&ctx->read_mutex, NULL);
|
||||
pthread_mutex_init(&ctx->write_mutex, NULL);
|
||||
|
||||
INIT_LIST_HEAD(&ctx->writelist_free);
|
||||
INIT_LIST_HEAD(&ctx->writelist_busy);
|
||||
INIT_LIST_HEAD(&ctx->writelist_done);
|
||||
INIT_LIST_HEAD(&ctx->writelist_free);
|
||||
INIT_LIST_HEAD(&ctx->writelist_busy);
|
||||
INIT_LIST_HEAD(&ctx->writelist_done);
|
||||
} else {
|
||||
threads = 1;
|
||||
}
|
||||
|
||||
ctx->cwork = (cwork_t *) malloc(sizeof(cwork_t) * threads);
|
||||
if (!ctx->cwork)
|
||||
@@ -376,6 +380,90 @@ static void *pt_decompress(void *arg)
|
||||
return (void *)result;
|
||||
}
|
||||
|
||||
/* single threaded (standard brotli stream, without header/mt-frames) */
|
||||
static size_t st_decompress(void *arg)
|
||||
{
|
||||
BROTLIMT_DCtx *ctx = (BROTLIMT_DCtx *) arg;
|
||||
BrotliDecoderResult bres = BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT;
|
||||
cwork_t *w = &ctx->cwork[0];
|
||||
BROTLIMT_Buffer Out;
|
||||
BROTLIMT_Buffer *out = &Out;
|
||||
BROTLIMT_Buffer *in = &w->in;
|
||||
BrotliDecoderState *state;
|
||||
const uint8_t* next_in;
|
||||
uint8_t* next_out;
|
||||
int rv;
|
||||
size_t retval = 0;
|
||||
|
||||
/* allocate space for input buffer */
|
||||
in->allocated = in->size = ctx->inputsize;
|
||||
in->buf = malloc(in->size);
|
||||
if (!in->buf)
|
||||
return MT_ERROR(memory_allocation);
|
||||
next_in = in->buf;
|
||||
|
||||
/* allocate space for output buffer */
|
||||
out->allocated = out->size = ctx->inputsize * 4;
|
||||
out->buf = malloc(out->size);
|
||||
if (!out->buf) {
|
||||
free(in->buf);
|
||||
return MT_ERROR(memory_allocation);
|
||||
}
|
||||
next_out = out->buf;
|
||||
|
||||
state = BrotliDecoderCreateInstance(NULL, NULL, NULL);
|
||||
if (!state) {
|
||||
free(in->buf);
|
||||
free(out->buf);
|
||||
return MT_ERROR(memory_allocation);
|
||||
}
|
||||
BrotliDecoderSetParameter(state, BROTLI_DECODER_PARAM_LARGE_WINDOW, 1);
|
||||
|
||||
while (1) {
|
||||
if (bres == BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT) {
|
||||
in->size = in->allocated;
|
||||
rv = ctx->fn_read(ctx->arg_read, in);
|
||||
if (in->size == 0) break;
|
||||
if (rv != 0) {
|
||||
retval = mt_error(rv);
|
||||
goto done;
|
||||
}
|
||||
next_in = in->buf;
|
||||
} else if (bres == BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT) {
|
||||
out->size = next_out - (uint8_t*)out->buf;
|
||||
rv = ctx->fn_write(ctx->arg_write, out);
|
||||
if (rv != 0) {
|
||||
retval = mt_error(rv);
|
||||
goto done;
|
||||
}
|
||||
next_out = out->buf;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
|
||||
bres = BrotliDecoderDecompressStream(state, &in->size, &next_in, &out->size, &next_out, 0);
|
||||
}
|
||||
|
||||
if (bres != BROTLI_DECODER_RESULT_SUCCESS) {
|
||||
retval = MT_ERROR(data_error); // corrupt input
|
||||
goto done;
|
||||
}
|
||||
out->size = next_out - (uint8_t*)out->buf;
|
||||
if (out->size != 0) {
|
||||
rv = ctx->fn_write(ctx->arg_write, out);
|
||||
if (rv != 0) {
|
||||
retval = mt_error(rv);
|
||||
goto done;
|
||||
}
|
||||
}
|
||||
|
||||
done:
|
||||
free(in->buf);
|
||||
free(out->buf);
|
||||
BrotliDecoderDestroyInstance(state);
|
||||
return retval;
|
||||
}
|
||||
|
||||
size_t BROTLIMT_decompressDCtx(BROTLIMT_DCtx * ctx, BROTLIMT_RdWr_t * rdwr)
|
||||
{
|
||||
unsigned char buf[4];
|
||||
@@ -393,6 +481,14 @@ size_t BROTLIMT_decompressDCtx(BROTLIMT_DCtx * ctx, BROTLIMT_RdWr_t * rdwr)
|
||||
ctx->arg_read = rdwr->arg_read;
|
||||
ctx->arg_write = rdwr->arg_write;
|
||||
|
||||
/* For single threaded brotli (no header, no mt-frames), don't check for
|
||||
BROTLIMT_MAGIC_SKIPPABLE here, because stdandard brotli stream may also
|
||||
start with it, so will be mistakenly considered as brotli-mt stream. */
|
||||
if (ctx->threads == 0) {
|
||||
/* decompress single threaded */
|
||||
return st_decompress(ctx);
|
||||
}
|
||||
|
||||
/* check for BROTLIMT_MAGIC_SKIPPABLE */
|
||||
in->buf = buf;
|
||||
in->size = 4;
|
||||
@@ -402,10 +498,6 @@ size_t BROTLIMT_decompressDCtx(BROTLIMT_DCtx * ctx, BROTLIMT_RdWr_t * rdwr)
|
||||
if (in->size != 4)
|
||||
return MT_ERROR(data_error);
|
||||
|
||||
/* single threaded with unknown sizes */
|
||||
if (MEM_readLE32(buf) != BROTLIMT_MAGIC_SKIPPABLE)
|
||||
return MT_ERROR(data_error);
|
||||
|
||||
/* mark unused */
|
||||
in->buf = 0;
|
||||
in->size = 0;
|
||||
@@ -485,8 +577,10 @@ void BROTLIMT_freeDCtx(BROTLIMT_DCtx * ctx)
|
||||
if (!ctx)
|
||||
return;
|
||||
|
||||
pthread_mutex_destroy(&ctx->read_mutex);
|
||||
pthread_mutex_destroy(&ctx->write_mutex);
|
||||
if (ctx->threads) {
|
||||
pthread_mutex_destroy(&ctx->read_mutex);
|
||||
pthread_mutex_destroy(&ctx->write_mutex);
|
||||
}
|
||||
free(ctx->cwork);
|
||||
free(ctx);
|
||||
ctx = 0;
|
||||
|
||||
Reference in New Issue
Block a user