1 files changed, 51 insertions, 0 deletions
diff --git a/libnbtplusplus/include/io/zlib_streambuf.h b/libnbtplusplus/include/io/zlib_streambuf.h
new file mode 100644
index 0000000000..91a660b9f2
--- /dev/null
+++ b/libnbtplusplus/include/io/zlib_streambuf.h
@@ -0,0 +1,51 @@
+/*
+ * SPDX-FileCopyrightText: 2013, 2015 ljfa-ag <ljfa-ag@web.de>
+ *
+ * SPDX-License-Identifier: LGPL-3.0-or-later
+ */
+
+#ifndef ZLIB_STREAMBUF_H_INCLUDED
+#define ZLIB_STREAMBUF_H_INCLUDED
+
+#include <stdexcept>
+#include <streambuf>
+#include <vector>
+#include <zlib.h>
+#include "nbt_export.h"
+
+namespace zlib
+{
+
+	/// Exception thrown in case zlib encounters a problem
+	class NBT_EXPORT zlib_error : public std::runtime_error
+	{
+	  public:
+		const int errcode;
+
+		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;
+		}
+	};
+
+} // namespace zlib
+
+#endif // ZLIB_STREAMBUF_H_INCLUDED