tdlight/tdutils/td/utils/GzipByteFlow.cpp

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#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 {

bool GzipByteFlow::loop() {
  if (gzip_.need_input()) {
    auto slice = input_->prepare_read();
    if (slice.empty()) {
      if (!is_input_active_) {
        gzip_.close_input();
      } else {
        return false;
      }
    } 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) {
    if (output_size > max_output_size_ || total_output_size_ > max_output_size_ - output_size) {
      finish(Status::Error("Max output size limit exceeded"));
      return false;
    }
    total_output_size_ += output_size;
    output_.confirm_append(output_size);
  }

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

}  // namespace td

#endif