aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/restricted/boost/iostreams/src/zstd.cpp
diff options
context:
space:
mode:
authorrobot-contrib <robot-contrib@yandex-team.com>2022-09-13 00:02:01 +0300
committerrobot-contrib <robot-contrib@yandex-team.com>2022-09-13 00:02:01 +0300
commitc7de57e4744e7575f21f74bb45f2a417d356ea5e (patch)
tree135cd7d2cf1787441107d727bd501c3d836e0478 /contrib/restricted/boost/iostreams/src/zstd.cpp
parent87c7569ea3c8aa5d371e9e3d15a8668c5cfcc4fc (diff)
downloadydb-c7de57e4744e7575f21f74bb45f2a417d356ea5e.tar.gz
Update contrib/restricted/boost/iostreams to 1.80.0
Diffstat (limited to 'contrib/restricted/boost/iostreams/src/zstd.cpp')
-rw-r--r--contrib/restricted/boost/iostreams/src/zstd.cpp166
1 files changed, 166 insertions, 0 deletions
diff --git a/contrib/restricted/boost/iostreams/src/zstd.cpp b/contrib/restricted/boost/iostreams/src/zstd.cpp
new file mode 100644
index 0000000000..20cc009804
--- /dev/null
+++ b/contrib/restricted/boost/iostreams/src/zstd.cpp
@@ -0,0 +1,166 @@
+// (C) Copyright Reimar Döffinger 2018.
+// Based on zstd.cpp by:
+// (C) Copyright Milan Svoboda 2008.
+// (C) Copyright Jonathan Turkanis 2003.
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
+
+// See http://www.boost.org/libs/iostreams for documentation.
+
+// Define BOOST_IOSTREAMS_SOURCE so that <boost/iostreams/detail/config.hpp>
+// knows that we are building the library (possibly exporting code), rather
+// than using it (possibly importing code).
+#define BOOST_IOSTREAMS_SOURCE
+
+#include <zstd.h>
+
+#include <boost/throw_exception.hpp>
+#include <boost/iostreams/detail/config/dyn_link.hpp>
+#include <boost/iostreams/filter/zstd.hpp>
+
+namespace boost { namespace iostreams {
+
+namespace zstd {
+ // Compression levels
+
+const uint32_t best_speed = 1;
+const uint32_t best_compression = 19;
+const uint32_t default_compression = 3;
+
+ // Status codes
+
+const int okay = 0;
+const int stream_end = 1;
+
+ // Flush codes
+
+const int finish = 0;
+const int flush = 1;
+const int run = 2;
+} // End namespace zstd.
+
+//------------------Implementation of zstd_error------------------------------//
+
+zstd_error::zstd_error(size_t error)
+ : BOOST_IOSTREAMS_FAILURE(ZSTD_getErrorName(error)), error_(error)
+ { }
+
+void zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(size_t error)
+{
+ if (ZSTD_isError(error))
+ boost::throw_exception(zstd_error(error));
+}
+
+//------------------Implementation of zstd_base-------------------------------//
+
+namespace detail {
+
+zstd_base::zstd_base()
+ : cstream_(ZSTD_createCStream()), dstream_(ZSTD_createDStream()), in_(new ZSTD_inBuffer), out_(new ZSTD_outBuffer), eof_(0)
+ { }
+
+zstd_base::~zstd_base()
+{
+ ZSTD_freeCStream(static_cast<ZSTD_CStream *>(cstream_));
+ ZSTD_freeDStream(static_cast<ZSTD_DStream *>(dstream_));
+ delete static_cast<ZSTD_inBuffer*>(in_);
+ delete static_cast<ZSTD_outBuffer*>(out_);
+}
+
+void zstd_base::before( const char*& src_begin, const char* src_end,
+ char*& dest_begin, char* dest_end )
+{
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+ in->src = src_begin;
+ in->size = static_cast<size_t>(src_end - src_begin);
+ in->pos = 0;
+ out->dst = dest_begin;
+ out->size = static_cast<size_t>(dest_end - dest_begin);
+ out->pos = 0;
+}
+
+void zstd_base::after(const char*& src_begin, char*& dest_begin, bool)
+{
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+ src_begin = reinterpret_cast<const char*>(in->src) + in->pos;
+ dest_begin = reinterpret_cast<char*>(out->dst) + out->pos;
+}
+
+int zstd_base::deflate(int action)
+{
+ ZSTD_CStream *s = static_cast<ZSTD_CStream *>(cstream_);
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+ // Ignore spurious extra calls.
+ // Note size > 0 will trigger an error in this case.
+ if (eof_ && in->size == 0) return zstd::stream_end;
+ size_t result = ZSTD_compressStream(s, out, in);
+ zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(result);
+ if (action != zstd::run)
+ {
+ result = action == zstd::finish ? ZSTD_endStream(s, out) : ZSTD_flushStream(s, out);
+ zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(result);
+ eof_ = action == zstd::finish && result == 0;
+ return result == 0 ? zstd::stream_end : zstd::okay;
+ }
+ return zstd::okay;
+}
+
+int zstd_base::inflate(int action)
+{
+ ZSTD_DStream *s = static_cast<ZSTD_DStream *>(dstream_);
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+ // need loop since iostream code cannot handle short reads
+ do {
+ size_t result = ZSTD_decompressStream(s, out, in);
+ zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(result);
+ } while (in->pos < in->size && out->pos < out->size);
+ return action == zstd::finish && in->size == 0 && out->pos == 0 ? zstd::stream_end : zstd::okay;
+}
+
+void zstd_base::reset(bool compress, bool realloc)
+{
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+ if (realloc)
+ {
+ memset(in, 0, sizeof(*in));
+ memset(out, 0, sizeof(*out));
+ eof_ = 0;
+
+ zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(
+ compress ?
+ ZSTD_initCStream(static_cast<ZSTD_CStream *>(cstream_), level) :
+ ZSTD_initDStream(static_cast<ZSTD_DStream *>(dstream_))
+ );
+ }
+}
+
+void zstd_base::do_init
+ ( const zstd_params& p, bool compress,
+ zstd::alloc_func, zstd::free_func,
+ void* )
+{
+ ZSTD_inBuffer *in = static_cast<ZSTD_inBuffer *>(in_);
+ ZSTD_outBuffer *out = static_cast<ZSTD_outBuffer *>(out_);
+
+ memset(in, 0, sizeof(*in));
+ memset(out, 0, sizeof(*out));
+ eof_ = 0;
+
+ level = p.level;
+ zstd_error::check BOOST_PREVENT_MACRO_SUBSTITUTION(
+ compress ?
+ ZSTD_initCStream(static_cast<ZSTD_CStream *>(cstream_), level) :
+ ZSTD_initDStream(static_cast<ZSTD_DStream *>(dstream_))
+ );
+}
+
+} // End namespace detail.
+
+//----------------------------------------------------------------------------//
+
+} } // End namespaces iostreams, boost.