ryantimwilson / rpms / systemd

Forked from rpms/systemd a month ago
Clone
52b84b
From fc1e6209f622ff96c24259a50d98ca6f57a55426 Mon Sep 17 00:00:00 2001
52b84b
From: =?UTF-8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= <zbyszek@in.waw.pl>
52b84b
Date: Mon, 29 Oct 2018 22:21:28 +0100
52b84b
Subject: [PATCH] test-compress: add test for short decompress_startswith calls
52b84b
52b84b
I thought this might fail with lz4 < 1.8.3, but it seems that because of
52b84b
greedy_realloc, we always use a buffer that is large enough, and it always
52b84b
passes.
52b84b
52b84b
(cherry picked from commit ba17efce44e6a1e139c1671205e9a6ed3824af1b)
52b84b
Resolves: #1843871
52b84b
---
52b84b
 src/journal/test-compress.c | 32 ++++++++++++++++++++++++++++++++
52b84b
 1 file changed, 32 insertions(+)
52b84b
52b84b
diff --git a/src/journal/test-compress.c b/src/journal/test-compress.c
52b84b
index eb3dc3eb6b..65cd3fbfeb 100644
52b84b
--- a/src/journal/test-compress.c
52b84b
+++ b/src/journal/test-compress.c
52b84b
@@ -131,6 +131,32 @@ static void test_decompress_startswith(int compression,
52b84b
         assert_se(r > 0);
52b84b
 }
52b84b
 
52b84b
+static void test_decompress_startswith_short(int compression,
52b84b
+                                             compress_blob_t compress,
52b84b
+                                             decompress_sw_t decompress_sw) {
52b84b
+
52b84b
+#define TEXT "HUGE=xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
52b84b
+
52b84b
+        char buf[1024];
52b84b
+        size_t i, csize;
52b84b
+        int r;
52b84b
+
52b84b
+        log_info("/* %s with %s */", __func__, object_compressed_to_string(compression));
52b84b
+
52b84b
+        r = compress(TEXT, sizeof TEXT, buf, sizeof buf, &csize);
52b84b
+        assert_se(r == 0);
52b84b
+
52b84b
+        for (i = 1; i < strlen(TEXT); i++) {
52b84b
+                size_t alloc_size = i;
52b84b
+                _cleanup_free_ void *buf2 = NULL;
52b84b
+
52b84b
+                assert_se(buf2 = malloc(i));
52b84b
+
52b84b
+                assert_se(decompress_sw(buf, csize, &buf2, &alloc_size, TEXT, i, TEXT[i]) == 1);
52b84b
+                assert_se(decompress_sw(buf, csize, &buf2, &alloc_size, TEXT, i, 'y') == 0);
52b84b
+        }
52b84b
+}
52b84b
+
52b84b
 static void test_compress_stream(int compression,
52b84b
                                  const char* cat,
52b84b
                                  compress_stream_t compress,
52b84b
@@ -271,6 +297,9 @@ int main(int argc, char *argv[]) {
52b84b
 
52b84b
         test_compress_stream(OBJECT_COMPRESSED_XZ, "xzcat",
52b84b
                              compress_stream_xz, decompress_stream_xz, srcfile);
52b84b
+
52b84b
+        test_decompress_startswith_short(OBJECT_COMPRESSED_XZ, compress_blob_xz, decompress_startswith_xz);
52b84b
+
52b84b
 #else
52b84b
         log_info("/* XZ test skipped */");
52b84b
 #endif
52b84b
@@ -295,6 +324,9 @@ int main(int argc, char *argv[]) {
52b84b
                              compress_stream_lz4, decompress_stream_lz4, srcfile);
52b84b
 
52b84b
         test_lz4_decompress_partial();
52b84b
+
52b84b
+        test_decompress_startswith_short(OBJECT_COMPRESSED_LZ4, compress_blob_lz4, decompress_startswith_lz4);
52b84b
+
52b84b
 #else
52b84b
         log_info("/* LZ4 test skipped */");
52b84b
 #endif