summaryrefslogtreecommitdiffstats
path: root/tests/test_memlimit.c
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 21:12:04 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 21:12:04 +0000
commiteac54b7c4aec25060d7bd856f7cdc290943d6aae (patch)
tree9a6d81c9f88df4698e746d63d14ddafeddd918b8 /tests/test_memlimit.c
parentInitial commit. (diff)
downloadxz-utils-eac54b7c4aec25060d7bd856f7cdc290943d6aae.tar.xz
xz-utils-eac54b7c4aec25060d7bd856f7cdc290943d6aae.zip
Adding upstream version 5.4.1.upstream/5.4.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r--tests/test_memlimit.c173
1 files changed, 173 insertions, 0 deletions
diff --git a/tests/test_memlimit.c b/tests/test_memlimit.c
new file mode 100644
index 0000000..c45a44b
--- /dev/null
+++ b/tests/test_memlimit.c
@@ -0,0 +1,173 @@
+///////////////////////////////////////////////////////////////////////////////
+//
+/// \file test_memlimit.c
+/// \brief Tests memory usage limit in decoders
+//
+// Author: Lasse Collin
+//
+// This file has been put into the public domain.
+// You can do whatever you want with this file.
+//
+///////////////////////////////////////////////////////////////////////////////
+
+#include "tests.h"
+#include "mythread.h"
+
+
+#define MEMLIMIT_TOO_LOW 1234U
+#define MEMLIMIT_HIGH_ENOUGH (2U << 20)
+
+
+static uint8_t *in;
+static size_t in_size;
+
+#ifdef HAVE_DECODERS
+static uint8_t out[8192];
+#endif
+
+
+static void
+test_memlimit_stream_decoder(void)
+{
+#ifndef HAVE_DECODERS
+ assert_skip("Decoder support disabled");
+#else
+ lzma_stream strm = LZMA_STREAM_INIT;
+ assert_lzma_ret(lzma_stream_decoder(&strm, MEMLIMIT_TOO_LOW, 0),
+ LZMA_OK);
+
+ strm.next_in = in;
+ strm.avail_in = in_size;
+ strm.next_out = out;
+ strm.avail_out = sizeof(out);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_MEMLIMIT_ERROR);
+
+ assert_uint_eq(lzma_memlimit_get(&strm), MEMLIMIT_TOO_LOW);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_TOO_LOW + 1),
+ LZMA_MEMLIMIT_ERROR);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_HIGH_ENOUGH),
+ LZMA_OK);
+
+ // This fails before commit 660739f99ab211edec4071de98889fb32ed04e98
+ // (liblzma <= 5.2.6, liblzma <= 5.3.3alpha). It was fixed in 5.2.7.
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+
+ lzma_end(&strm);
+#endif
+}
+
+
+static void
+test_memlimit_stream_decoder_mt(void)
+{
+#ifndef MYTHREAD_ENABLED
+ assert_skip("Threading support disabled");
+#elif !defined(HAVE_DECODERS)
+ assert_skip("Decoder support disabled");
+#else
+ lzma_stream strm = LZMA_STREAM_INIT;
+ lzma_mt mt = {
+ .flags = 0,
+ .threads = 1,
+ .timeout = 0,
+ .memlimit_threading = 0,
+ .memlimit_stop = MEMLIMIT_TOO_LOW,
+ };
+
+ assert_lzma_ret(lzma_stream_decoder_mt(&strm, &mt), LZMA_OK);
+
+ strm.next_in = in;
+ strm.avail_in = in_size;
+ strm.next_out = out;
+ strm.avail_out = sizeof(out);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_MEMLIMIT_ERROR);
+
+ assert_uint_eq(lzma_memlimit_get(&strm), MEMLIMIT_TOO_LOW);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_TOO_LOW + 1),
+ LZMA_MEMLIMIT_ERROR);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_HIGH_ENOUGH),
+ LZMA_OK);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+ lzma_end(&strm);
+#endif
+}
+
+
+static void
+test_memlimit_alone_decoder(void)
+{
+#ifndef HAVE_DECODERS
+ assert_skip("Decoder support disabled");
+#else
+ size_t alone_size;
+ uint8_t *alone_buf = tuktest_file_from_srcdir(
+ "files/good-unknown_size-with_eopm.lzma", &alone_size);
+
+ lzma_stream strm = LZMA_STREAM_INIT;
+ assert_lzma_ret(lzma_alone_decoder(&strm, MEMLIMIT_TOO_LOW), LZMA_OK);
+
+ strm.next_in = alone_buf;
+ strm.avail_in = alone_size;
+ strm.next_out = out;
+ strm.avail_out = sizeof(out);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_MEMLIMIT_ERROR);
+
+ assert_uint_eq(lzma_memlimit_get(&strm), MEMLIMIT_TOO_LOW);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_TOO_LOW + 1),
+ LZMA_MEMLIMIT_ERROR);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_HIGH_ENOUGH),
+ LZMA_OK);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+ lzma_end(&strm);
+#endif
+}
+
+
+static void
+test_memlimit_auto_decoder(void)
+{
+#ifndef HAVE_DECODERS
+ assert_skip("Decoder support disabled");
+#else
+ lzma_stream strm = LZMA_STREAM_INIT;
+ assert_lzma_ret(lzma_auto_decoder(&strm, MEMLIMIT_TOO_LOW, 0),
+ LZMA_OK);
+
+ strm.next_in = in;
+ strm.avail_in = in_size;
+ strm.next_out = out;
+ strm.avail_out = sizeof(out);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_MEMLIMIT_ERROR);
+
+ assert_uint_eq(lzma_memlimit_get(&strm), MEMLIMIT_TOO_LOW);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_TOO_LOW + 1),
+ LZMA_MEMLIMIT_ERROR);
+ assert_lzma_ret(lzma_memlimit_set(&strm, MEMLIMIT_HIGH_ENOUGH),
+ LZMA_OK);
+
+ assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+ lzma_end(&strm);
+#endif
+}
+
+
+extern int
+main(int argc, char **argv)
+{
+ tuktest_start(argc, argv);
+
+ in = tuktest_file_from_srcdir("files/good-1-check-crc32.xz", &in_size);
+
+ tuktest_run(test_memlimit_stream_decoder);
+ tuktest_run(test_memlimit_stream_decoder_mt);
+ tuktest_run(test_memlimit_alone_decoder);
+ tuktest_run(test_memlimit_auto_decoder);
+
+ return tuktest_end();
+}