1 #ifndef OSMIUM_IO_GZIP_COMPRESSION_HPP 2 #define OSMIUM_IO_GZIP_COMPRESSION_HPP 46 #include <osmium/io/detail/read_write.hpp> 78 gzip_error(
const std::string& what,
const int error_code) :
80 gzip_error_code(error_code) {
81 if (error_code == Z_ERRNO) {
92 [[noreturn]]
inline void throw_gzip_error(gzFile gzfile,
const char* msg) {
93 std::string error{
"gzip error: "};
98 error += ::gzerror(gzfile, &error_code);
116 osmium::detail::disable_invalid_parameter_handler diph;
118 m_gzfile = ::gzdopen(fd,
"wb");
120 throw gzip_error{
"gzip error: write initialization failed"};
138 void write(
const std::string& data)
final {
140 osmium::detail::disable_invalid_parameter_handler diph;
143 assert(data.size() < std::numeric_limits<unsigned int>::max());
145 const int nwrite = ::gzwrite(m_gzfile, data.data(),
static_cast<unsigned int>(data.size()));
147 detail::throw_gzip_error(m_gzfile,
"write failed");
155 osmium::detail::disable_invalid_parameter_handler diph;
157 const int result = ::gzclose_w(m_gzfile);
159 if (result != Z_OK) {
160 throw gzip_error{
"gzip error: write close failed", result};
163 osmium::io::detail::reliable_fsync(m_fd);
165 osmium::io::detail::reliable_close(m_fd);
173 gzFile m_gzfile =
nullptr;
179 osmium::detail::disable_invalid_parameter_handler diph;
181 m_gzfile = ::gzdopen(fd,
"rb");
184 osmium::io::detail::reliable_close(fd);
187 throw gzip_error{
"gzip error: read initialization failed"};
207 osmium::detail::disable_invalid_parameter_handler diph;
211 assert(buffer.size() < std::numeric_limits<unsigned int>::max());
212 int nread = ::gzread(m_gzfile, &*buffer.begin(),
static_cast<unsigned int>(buffer.size()));
214 detail::throw_gzip_error(m_gzfile,
"read failed");
216 buffer.resize(static_cast<std::string::size_type>(nread));
217 #if ZLIB_VERNUM >= 0x1240 218 set_offset(static_cast<std::size_t>(::gzoffset(m_gzfile)));
226 osmium::detail::disable_invalid_parameter_handler diph;
228 const int result = ::gzclose_r(m_gzfile);
230 if (result != Z_OK) {
231 throw gzip_error{
"gzip error: read close failed", result};
250 m_zstream.next_in =
reinterpret_cast<unsigned char*
>(
const_cast<char*
>(buffer));
251 assert(size < std::numeric_limits<unsigned int>::max());
252 m_zstream.avail_in =
static_cast<unsigned int>(size);
253 const int result = inflateInit2(&m_zstream, MAX_WBITS | 32);
254 if (result != Z_OK) {
255 std::string message{
"gzip error: decompression init failed: "};
257 message.append(m_zstream.msg);
281 const std::size_t buffer_size = 10240;
282 output.append(buffer_size,
'\0');
283 m_zstream.next_out =
reinterpret_cast<unsigned char*
>(&*output.begin());
284 m_zstream.avail_out = buffer_size;
285 const int result = inflate(&m_zstream, Z_SYNC_FLUSH);
287 if (result != Z_OK) {
292 if (result != Z_OK && result != Z_STREAM_END) {
293 std::string message{
"gzip error: inflate failed: "};
295 message.append(m_zstream.msg);
300 output.resize(static_cast<std::size_t>(m_zstream.next_out - reinterpret_cast<const unsigned char*>(output.data())));
307 inflateEnd(&m_zstream);
323 inline bool get_registered_gzip_compression() noexcept {
324 return registered_gzip_compression;
333 #endif // OSMIUM_IO_GZIP_COMPRESSION_HPP gzFile m_gzfile
Definition: gzip_compression.hpp:108
Definition: compression.hpp:95
Definition: gzip_compression.hpp:238
Definition: gzip_compression.hpp:105
std::size_t m_buffer_size
Definition: gzip_compression.hpp:241
z_stream m_zstream
Definition: gzip_compression.hpp:242
gzip_error(const std::string &what, const int error_code)
Definition: gzip_compression.hpp:78
static CompressionFactory & instance()
Definition: compression.hpp:180
Definition: gzip_compression.hpp:171
bool register_compression(osmium::io::file_compression compression, const create_compressor_type &create_compressor, const create_decompressor_type_fd &create_decompressor_fd, const create_decompressor_type_buffer &create_decompressor_buffer)
Definition: compression.hpp:185
~GzipDecompressor() noexcept final
Definition: gzip_compression.hpp:197
Definition: compression.hpp:87
int m_fd
Definition: gzip_compression.hpp:107
Namespace for everything in the Osmium library.
Definition: assembler.hpp:53
GzipCompressor(const int fd, const fsync sync)
Definition: gzip_compression.hpp:112
const char * m_buffer
Definition: gzip_compression.hpp:240
void close() final
Definition: gzip_compression.hpp:306
fsync
Definition: writer_options.hpp:51
GzipDecompressor(const int fd)
Definition: gzip_compression.hpp:177
void close() final
Definition: gzip_compression.hpp:152
int system_errno
Definition: gzip_compression.hpp:72
gzip_error(const std::string &what)
Definition: gzip_compression.hpp:74
Definition: gzip_compression.hpp:69
std::string read() final
Definition: gzip_compression.hpp:205
int gzip_error_code
Definition: gzip_compression.hpp:71
Definition: compression.hpp:57
void write(const std::string &data) final
Definition: gzip_compression.hpp:138
~GzipCompressor() noexcept final
Definition: gzip_compression.hpp:130
~GzipBufferDecompressor() noexcept final
Definition: gzip_compression.hpp:269
GzipBufferDecompressor(const char *buffer, const std::size_t size)
Definition: gzip_compression.hpp:246
std::string read() final
Definition: gzip_compression.hpp:277
void close() final
Definition: gzip_compression.hpp:223