1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
/*
* Zlib Compressor
* (C) 2001 Peter J Jones
* 2001-2007,2014 Jack Lloyd
* 2006 Matt Johnston
*
* Distributed under the terms of the Botan license
*/
#include <botan/zlib.h>
#include <botan/internal/comp_util.h>
#include <ctime>
#include <zlib.h>
namespace Botan {
namespace {
class Zlib_Stream : public Zlib_Style_Stream<z_stream, Bytef>
{
public:
Zlib_Stream()
{
streamp()->opaque = alloc();
streamp()->zalloc = Compression_Alloc_Info::malloc<unsigned int>;
streamp()->zfree = Compression_Alloc_Info::free;
}
u32bit run_flag() const override { return Z_NO_FLUSH; }
u32bit flush_flag() const override { return Z_FULL_FLUSH; }
u32bit finish_flag() const override { return Z_FINISH; }
int compute_window_bits(int wbits, int wbits_offset) const
{
if(wbits_offset == -1)
return -wbits;
else
return wbits + wbits_offset;
}
};
class Zlib_Compression_Stream : public Zlib_Stream
{
public:
Zlib_Compression_Stream(size_t level, int wbits, int wbits_offset = 0)
{
int rc = deflateInit2(streamp(), level, Z_DEFLATED,
compute_window_bits(wbits, wbits_offset),
8, Z_DEFAULT_STRATEGY);
if(rc != Z_OK)
throw std::runtime_error("zlib deflate initialization failed");
}
~Zlib_Compression_Stream()
{
deflateEnd(streamp());
}
bool run(u32bit flags) override
{
int rc = deflate(streamp(), flags);
if(rc == Z_MEM_ERROR)
throw std::bad_alloc();
else if(rc != Z_OK && rc != Z_STREAM_END)
throw std::runtime_error("zlib deflate error");
return (rc == Z_STREAM_END);
}
};
class Zlib_Decompression_Stream : public Zlib_Stream
{
public:
Zlib_Decompression_Stream(int wbits, int wbits_offset = 0)
{
int rc = inflateInit2(streamp(), compute_window_bits(wbits, wbits_offset));
if(rc == Z_MEM_ERROR)
throw std::bad_alloc();
else if(rc != Z_OK)
throw std::runtime_error("zlib inflate initialization failed");
}
~Zlib_Decompression_Stream()
{
inflateEnd(streamp());
}
bool run(u32bit flags) override
{
int rc = inflate(streamp(), flags);
if(rc == Z_MEM_ERROR)
throw std::bad_alloc();
else if(rc != Z_OK && rc != Z_STREAM_END)
throw std::runtime_error("zlib deflate error");
return (rc == Z_STREAM_END);
}
};
class Deflate_Compression_Stream : public Zlib_Compression_Stream
{
public:
Deflate_Compression_Stream(size_t level, int wbits) :
Zlib_Compression_Stream(level, wbits, -1) {}
};
class Deflate_Decompression_Stream : public Zlib_Decompression_Stream
{
public:
Deflate_Decompression_Stream(int wbits) : Zlib_Decompression_Stream(wbits, -1) {}
};
class Gzip_Compression_Stream : public Zlib_Compression_Stream
{
public:
Gzip_Compression_Stream(size_t level, int wbits, byte os_code) :
Zlib_Compression_Stream(level, wbits, 16)
{
std::memset(&m_header, 0, sizeof(m_header));
m_header.os = os_code;
m_header.time = std::time(nullptr);
int rc = deflateSetHeader(streamp(), &m_header);
if(rc != Z_OK)
throw std::runtime_error("setting gzip header failed");
}
private:
::gz_header m_header;
};
class Gzip_Decompression_Stream : public Zlib_Decompression_Stream
{
public:
Gzip_Decompression_Stream(int wbits) : Zlib_Decompression_Stream(wbits, 16) {}
};
}
Compression_Stream* Zlib_Compression::make_stream() const
{
if(m_raw_deflate)
return new Deflate_Compression_Stream(m_level, 15);
else
return new Zlib_Compression_Stream(m_level, 15);
}
Compression_Stream* Zlib_Decompression::make_stream() const
{
if(m_raw_deflate)
return new Deflate_Decompression_Stream(15);
else
return new Zlib_Decompression_Stream(15);
}
Compression_Stream* Gzip_Compression::make_stream() const
{
return new Gzip_Compression_Stream(m_level, 15, m_os_code);
}
Compression_Stream* Gzip_Decompression::make_stream() const
{
return new Gzip_Decompression_Stream(15);
}
}
|
