src/trace_processor/util/zstd_compressor.h

/*
 * Copyright (C) 2026 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef SRC_TRACE_PROCESSOR_UTIL_ZSTD_COMPRESSOR_H_
#define SRC_TRACE_PROCESSOR_UTIL_ZSTD_COMPRESSOR_H_

#include <cstddef>
#include <cstdint>
#include <memory>

#include "perfetto/ext/base/utils.h"

namespace perfetto::trace_processor::util {

// One-shot zstd compression, the counterpart of ZstdDecompressor. Check
// IsZstdSupported() (decompressor.h) before relying on this.
class ZstdCompressor {
 public:
  // Compresses |data| into a single zstd frame, returning a malloc()-allocated
  // buffer (caller frees; the frame length is written to |*out_size|) or
  // nullptr on error or when zstd is not compiled in.
  static std::unique_ptr<uint8_t, base::FreeDeleter>
  CompressFully(const uint8_t* data, size_t len, size_t* out_size, int level);
};

}  // namespace perfetto::trace_processor::util

#endif  // SRC_TRACE_PROCESSOR_UTIL_ZSTD_COMPRESSOR_H_