ton/tdutils/td/utils/GzipByteFlow.cpp

/*
    This file is part of TON Blockchain Library.

    TON Blockchain Library is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    TON Blockchain Library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.

    Copyright 2017-2020 Telegram Systems LLP
*/
#include "td/utils/GzipByteFlow.h"

char disable_linker_warning_about_empty_file_gzipbyteflow_cpp TD_UNUSED;

#if TD_HAVE_ZLIB
#include "td/utils/common.h"
#include "td/utils/Status.h"

namespace td {

void GzipByteFlow::loop() {
  while (true) {
    if (gzip_.need_input()) {
      auto slice = input_->prepare_read();
      if (slice.empty()) {
        if (!is_input_active_) {
          gzip_.close_input();
        } else {
          break;
        }
      } else {
        gzip_.set_input(input_->prepare_read());
      }
    }
    if (gzip_.need_output()) {
      auto slice = output_.prepare_append();
      CHECK(!slice.empty());
      gzip_.set_output(slice);
    }
    auto r_state = gzip_.run();
    auto output_size = gzip_.flush_output();
    if (output_size) {
      uncommited_size_ += output_size;
      total_output_size_ += output_size;
      if (total_output_size_ > max_output_size_) {
        return finish(Status::Error("Max output size limit exceeded"));
      }
      output_.confirm_append(output_size);
    }

    auto input_size = gzip_.flush_input();
    if (input_size) {
      input_->confirm_read(input_size);
    }
    if (r_state.is_error()) {
      return finish(r_state.move_as_error());
    }
    auto state = r_state.ok();
    if (state == Gzip::State::Done) {
      on_output_updated();
      return consume_input();
    }
  }
  if (uncommited_size_ >= MIN_UPDATE_SIZE) {
    uncommited_size_ = 0;
    on_output_updated();
  }
}

constexpr size_t GzipByteFlow::MIN_UPDATE_SIZE;

}  // namespace td

#endif