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) :
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;
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);
179 osmium::detail::disable_invalid_parameter_handler diph;
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 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: "};
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());
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: "};
300 output.resize(static_cast<std::size_t>(
m_zstream.next_out - reinterpret_cast<const unsigned char*>(output.data())));
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
GzipBufferDecompressor & operator=(const GzipBufferDecompressor &)=delete
Definition: gzip_compression.hpp:238
gzFile m_gzfile
Definition: gzip_compression.hpp:173
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
GzipDecompressor & operator=(const GzipDecompressor &)=delete
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
void set_offset(const std::size_t offset) noexcept
Definition: compression.hpp:124
int m_fd
Definition: gzip_compression.hpp:107
bool do_fsync() const noexcept
Definition: compression.hpp:63
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
GzipCompressor & operator=(const GzipCompressor &)=delete
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