10#include <botan/zlib.h>
11#include <botan/internal/compress_utils.h>
12#include <botan/exceptn.h>
19class Zlib_Stream :
public Zlib_Style_Stream<z_stream, Bytef, unsigned int>
24 streamp()->opaque = alloc();
25 streamp()->zalloc = Compression_Alloc_Info::malloc<unsigned int>;
29 uint32_t run_flag()
const override {
return Z_NO_FLUSH; }
30 uint32_t flush_flag()
const override {
return Z_SYNC_FLUSH; }
31 uint32_t finish_flag()
const override {
return Z_FINISH; }
33 int compute_window_bits(
int wbits,
int wbits_offset)
const
35 if(wbits_offset == -1)
38 return wbits + wbits_offset;
42class Zlib_Compression_Stream :
public Zlib_Stream
45 Zlib_Compression_Stream(
size_t level,
int wbits,
int wbits_offset = 0)
47 wbits = compute_window_bits(wbits, wbits_offset);
54 int rc = ::deflateInit2(streamp(),
static_cast<int>(level), Z_DEFLATED, wbits, 8, Z_DEFAULT_STRATEGY);
60 ~Zlib_Compression_Stream()
62 ::deflateEnd(streamp());
65 bool run(uint32_t flags)
override
67 int rc = ::deflate(streamp(), flags);
69 if(rc != Z_OK && rc != Z_STREAM_END && rc != Z_BUF_ERROR)
72 return (rc == Z_STREAM_END);
76class Zlib_Decompression_Stream :
public Zlib_Stream
79 Zlib_Decompression_Stream(
int wbits,
int wbits_offset = 0)
81 int rc = ::inflateInit2(streamp(), compute_window_bits(wbits, wbits_offset));
87 ~Zlib_Decompression_Stream()
89 ::inflateEnd(streamp());
92 bool run(uint32_t flags)
override
94 int rc = ::inflate(streamp(), flags);
96 if(rc != Z_OK && rc != Z_STREAM_END && rc != Z_BUF_ERROR)
99 return (rc == Z_STREAM_END);
103class Deflate_Compression_Stream
final :
public Zlib_Compression_Stream
106 Deflate_Compression_Stream(
size_t level,
int wbits) :
107 Zlib_Compression_Stream(level, wbits, -1) {}
110class Deflate_Decompression_Stream
final :
public Zlib_Decompression_Stream
113 explicit Deflate_Decompression_Stream(
int wbits) : Zlib_Decompression_Stream(wbits, -1) {}
116class Gzip_Compression_Stream
final :
public Zlib_Compression_Stream
119 Gzip_Compression_Stream(
size_t level,
int wbits, uint8_t os_code, uint64_t hdr_time) :
120 Zlib_Compression_Stream(level, wbits, 16)
123 m_header.os = os_code;
124 m_header.time =
static_cast<uLong
>(hdr_time);
126 int rc = deflateSetHeader(streamp(), &m_header);
132 ::gz_header m_header;
135class Gzip_Decompression_Stream
final :
public Zlib_Decompression_Stream
138 explicit Gzip_Decompression_Stream(
int wbits) : Zlib_Decompression_Stream(wbits, 16) {}
143std::unique_ptr<Compression_Stream> Zlib_Compression::make_stream(
size_t level)
const
145 return std::make_unique<Zlib_Compression_Stream>(level, 15);
148std::unique_ptr<Compression_Stream> Zlib_Decompression::make_stream()
const
150 return std::make_unique<Zlib_Decompression_Stream>(15);
153std::unique_ptr<Compression_Stream> Deflate_Compression::make_stream(
size_t level)
const
155 return std::make_unique<Deflate_Compression_Stream>(level, 15);
158std::unique_ptr<Compression_Stream> Deflate_Decompression::make_stream()
const
160 return std::make_unique<Deflate_Decompression_Stream>(15);
163std::unique_ptr<Compression_Stream> Gzip_Compression::make_stream(
size_t level)
const
165 return std::make_unique<Gzip_Compression_Stream>(level, 15, m_os_code, m_hdr_time);
168std::unique_ptr<Compression_Stream> Gzip_Decompression::make_stream()
const
170 return std::make_unique<Gzip_Decompression_Stream>(15);
static void free(void *self, void *ptr)
int(* final)(unsigned char *, CTX *)
constexpr void clear_mem(T *ptr, size_t n)