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; }