Merge pull request #70 from juntyr/patch-1

Fix UB for small array compression
szcompressor · Dec 10, 2024 · 780870a · 780870a
2 parents 10ebed0 + 0f3dde7
commit 780870a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/SZ3/api/sz.hpp b/include/SZ3/api/sz.hpp
@@ -92,7 +92,8 @@ template <class T>
 char *SZ_compress(const SZ3::Config &config, const T *data, size_t &cmpSize) {
     using namespace SZ3;
 
-    size_t bufferLen = 2 * config.num * sizeof(T);
+    // Ensure that the buffer can always hold the config and the lossless fallback
+    size_t bufferLen = config.size_est() + ZSTD_compressBound(config.num * sizeof(T));
     auto buffer = new char[bufferLen];
     cmpSize = SZ_compress(config, data, buffer, bufferLen);