From 533843e3546cd24c8344eaa899c6b0b681c8d222 Mon Sep 17 00:00:00 2001 From: Eugene Kliuchnikov Date: Fri, 2 Mar 2018 15:49:58 +0100 Subject: Update (#643) Update * make the zopflification aware of `NDIRECT`, `NPOSTFIX` (better compression in `font` mode) * add small and simple decoder tool * fix typo * Java: wrapper: make decoder channel more async-friendly Ramp up version to 1.0.3 / 1.0.3 --- research/BUILD | 7 ++++ research/brotli_decoder.c | 93 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 100 insertions(+) create mode 100644 research/brotli_decoder.c (limited to 'research') diff --git a/research/BUILD b/research/BUILD index 211b3e7..9da08c2 100755 --- a/research/BUILD +++ b/research/BUILD @@ -35,3 +35,10 @@ cc_binary( ":sieve", ], ) + +cc_binary( + name = "brotli_decoder", + srcs = ["brotli_decoder.c"], + linkstatic = 1, + deps = ["//:brotlidec"], +) diff --git a/research/brotli_decoder.c b/research/brotli_decoder.c new file mode 100644 index 0000000..b1d556d --- /dev/null +++ b/research/brotli_decoder.c @@ -0,0 +1,93 @@ +/* Copyright 2018 Google Inc. All Rights Reserved. + + Distributed under MIT license. + See file LICENSE for detail or copy at https://opensource.org/licenses/MIT +*/ + +#include +#include +#include + +#include + +#define BUFFER_SIZE (1u << 20) + +typedef struct Context { + FILE* fin; + FILE* fout; + uint8_t* input_buffer; + uint8_t* output_buffer; + BrotliDecoderState* decoder; +} Context; + +void init(Context* ctx) { + ctx->fin = 0; + ctx->fout = 0; + ctx->input_buffer = 0; + ctx->output_buffer = 0; + ctx->decoder = 0; +} + +void cleanup(Context* ctx) { + if (ctx->decoder) BrotliDecoderDestroyInstance(ctx->decoder); + if (ctx->output_buffer) free(ctx->output_buffer); + if (ctx->input_buffer) free(ctx->input_buffer); + if (ctx->fout) fclose(ctx->fout); + if (ctx->fin) fclose(ctx->fin); +} + +void fail(Context* ctx, const char* message) { + fprintf(stderr, "%s\n", message); + exit(1); +} + +int main(int argc, char** argv) { + Context ctx; + BrotliDecoderResult result = BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT; + size_t available_in; + const uint8_t* next_in; + size_t available_out = BUFFER_SIZE; + uint8_t* next_out; + init(&ctx); + + ctx.fin = fdopen(STDIN_FILENO, "rb"); + if (!ctx.fin) fail(&ctx, "can't open input file"); + ctx.fout = fdopen(STDOUT_FILENO, "wb"); + if (!ctx.fout) fail(&ctx, "can't open output file"); + ctx.input_buffer = (uint8_t*)malloc(BUFFER_SIZE); + if (!ctx.input_buffer) fail(&ctx, "out of memory / input buffer"); + ctx.output_buffer = (uint8_t*)malloc(BUFFER_SIZE); + if (!ctx.output_buffer) fail(&ctx, "out of memory / output buffer"); + ctx.decoder = BrotliDecoderCreateInstance(0, 0, 0); + if (!ctx.decoder) fail(&ctx, "out of memory / decoder"); + BrotliDecoderSetParameter(ctx.decoder, BROTLI_DECODER_PARAM_LARGE_WINDOW, 1); + + next_out = ctx.output_buffer; + while (1) { + if (result == BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT) { + if (feof(ctx.fin)) break; + available_in = fread(ctx.input_buffer, 1, BUFFER_SIZE, ctx.fin); + next_in = ctx.input_buffer; + if (ferror(ctx.fin)) break; + } else if (result == BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT) { + fwrite(ctx.output_buffer, 1, BUFFER_SIZE, ctx.fout); + if (ferror(ctx.fout)) break; + available_out = BUFFER_SIZE; + next_out = ctx.output_buffer; + } else { + break; + } + result = BrotliDecoderDecompressStream( + ctx.decoder, &available_in, &next_in, &available_out, &next_out, 0); + } + if (next_out != ctx.output_buffer) { + fwrite(ctx.output_buffer, 1, next_out - ctx.output_buffer, ctx.fout); + } + if ((result == BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT) || ferror(ctx.fout)) { + fail(&ctx, "failed to write output"); + } else if (result != BROTLI_DECODER_RESULT_SUCCESS) { + fail(&ctx, "corrupt input"); + } + cleanup(&ctx); + return 0; +} -- cgit v1.1