diff options
Diffstat (limited to 'include/io/zlib_streambuf.h')
| -rw-r--r-- | include/io/zlib_streambuf.h | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/include/io/zlib_streambuf.h b/include/io/zlib_streambuf.h new file mode 100644 index 0000000000..36062192b7 --- /dev/null +++ b/include/io/zlib_streambuf.h @@ -0,0 +1,45 @@ +#ifndef ZLIB_STREAMBUF_H_INCLUDED +#define ZLIB_STREAMBUF_H_INCLUDED + +#include <stdexcept> +#include <streambuf> +#include <vector> +#include <zlib.h> + +namespace zlib +{ + +///Exception thrown in case zlib encounters a problem +class zlib_error : public std::runtime_error +{ +public: + const int errcode; + + explicit zlib_error(const char* msg, int errcode): + std::runtime_error(msg + ? std::string(zError(errcode)) + ": " + msg + : zError(errcode)), + errcode(errcode) + {} +}; + +///Base class for deflate_streambuf and inflate_streambuf +class zlib_streambuf : public std::streambuf +{ +protected: + std::vector<char> in; + std::vector<char> out; + z_stream zstr; + + explicit zlib_streambuf(size_t bufsize): + in(bufsize), out(bufsize) + { + zstr.zalloc = Z_NULL; + zstr.zfree = Z_NULL; + zstr.opaque = Z_NULL; + } +}; + +} + +#endif // ZLIB_STREAMBUF_H_INCLUDED |
