static BROTLI_BOOL DecompressFiles(Context* context) {
while (NextFile(context)) {
BROTLI_BOOL is_ok = BROTLI_TRUE;
BrotliDecoderState* s = BrotliDecoderCreateInstance(BrotliAllocFunc, BrotliFreeFunc, &scratch_buffer_size);
if (!s) {
fprintf(stderr, "out of memory\n");
return BROTLI_FALSE;
}
/* This allows decoding "large-window" streams. Though it creates
fragmentation (new builds decode streams that old builds don't),
it is better from used experience perspective. */
BrotliDecoderSetParameter(s, BROTLI_DECODER_PARAM_LARGE_WINDOW, 1u);
is_ok = OpenFiles(context);
if (is_ok && !context->current_input_path &&
!context->force_overwrite && isatty(STDIN_FILENO)) {
fprintf(stderr, "Use -h help. Use -f to force input from a terminal.\n");
is_ok = BROTLI_FALSE;
}
if (is_ok) is_ok = DecompressFile(context, s);
BrotliDecoderDestroyInstance(s);
if (!CloseFiles(context, is_ok)) is_ok = BROTLI_FALSE;
if (!is_ok) return BROTLI_FALSE;
}
return BROTLI_TRUE;
}
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;
}
