blob: 683e39154dda25f85146ab95b0ab548a7f1d57e1 [file] [log] [blame]
// Copyright 2017 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "bsdiff/brotli_decompressor.h"
#include "bsdiff/logging.h"
using std::endl;
namespace bsdiff {
bool BrotliDecompressor::SetInputData(const uint8_t* input_data, size_t size) {
brotli_decoder_state_ =
BrotliDecoderCreateInstance(nullptr, nullptr, nullptr);
if (brotli_decoder_state_ == nullptr) {
LOG(ERROR) << "Failed to initialize brotli decoder." << endl;
return false;
}
next_in_ = input_data;
available_in_ = size;
return true;
}
bool BrotliDecompressor::Read(uint8_t* output_data, size_t bytes_to_output) {
auto next_out = output_data;
size_t available_out = bytes_to_output;
while (available_out > 0) {
// The brotli decoder will update |available_in_|, |available_in_|,
// |next_out| and |available_out|.
BrotliDecoderResult result = BrotliDecoderDecompressStream(
brotli_decoder_state_, &available_in_, &next_in_, &available_out,
&next_out, nullptr);
if (result == BROTLI_DECODER_RESULT_ERROR) {
LOG(ERROR) << "Decompression failed with "
<< BrotliDecoderErrorString(
BrotliDecoderGetErrorCode(brotli_decoder_state_))
<< endl;
return false;
} else if (result == BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT) {
LOG(ERROR) << "Decompressor reached EOF while reading from input stream."
<< endl;
return false;
}
}
return true;
}
bool BrotliDecompressor::Close() {
if (!BrotliDecoderIsFinished(brotli_decoder_state_)) {
LOG(ERROR) << "Unfinished brotli decoder." << endl;
return false;
}
BrotliDecoderDestroyInstance(brotli_decoder_state_);
return true;
}
} // namespace bsdiff