[Test] Add unit tests for the compression functions

1 files changed, 95 insertions, 23 deletions

diff --git a/test/rspamd_cxx_unit_utils.hxx b/test/rspamd_cxx_unit_utils.hxx
index 602b32a7e..f563f4409 100644
--- a/test/rspamd_cxx_unit_utils.hxx
+++ b/test/rspamd_cxx_unit_utils.hxx
@@ -23,6 +23,8 @@
 #include "doctest/doctest.h"
 
 #include "libmime/mime_headers.h"
+#include "contrib/libottery/ottery.h"
+#include "libcryptobox/cryptobox.h"
 
 #include <vector>
 #include <utility>
@@ -35,21 +37,21 @@ TEST_SUITE("rspamd_utils") {
 TEST_CASE("rspamd_strip_smtp_comments_inplace")
 {
 	std::vector<std::pair<std::string, std::string>> cases{
-			{"abc",                    "abc"},
-			{"abc(foo)",               "abc"},
-			{"abc(foo()",              "abc"},
-			{"abc(foo))",              "abc)"},
-			{"abc(foo(bar))",          "abc"},
-			{"(bar)abc(foo)",          "abc"},
-			{"ab(ololo)c(foo)",        "abc"},
-			{"ab(olo\\)lo)c(foo)",     "abc"},
-			{"ab(trol\\\1lo)c(foo)",   "abc"},
-			{"\\ab(trol\\\1lo)c(foo)", "abc"},
-			{"",                       ""},
-			{"<test_id@example.net> (added by postmaster@example.net)", "<test_id@example.net> "}
+		{"abc",                                                     "abc"},
+		{"abc(foo)",                                                "abc"},
+		{"abc(foo()",                                               "abc"},
+		{"abc(foo))",                                               "abc)"},
+		{"abc(foo(bar))",                                           "abc"},
+		{"(bar)abc(foo)",                                           "abc"},
+		{"ab(ololo)c(foo)",                                         "abc"},
+		{"ab(olo\\)lo)c(foo)",                                      "abc"},
+		{"ab(trol\\\1lo)c(foo)",                                    "abc"},
+		{"\\ab(trol\\\1lo)c(foo)",                                  "abc"},
+		{"",                                                        ""},
+		{"<test_id@example.net> (added by postmaster@example.net)", "<test_id@example.net> "}
 	};
 
-	for (const auto &c : cases) {
+	for (const auto &c: cases) {
 		SUBCASE (("strip comments in " + c.first).c_str()) {
 			auto *cpy = new char[c.first.size()];
 			memcpy(cpy, c.first.data(), c.first.size());
@@ -62,18 +64,18 @@ TEST_CASE("rspamd_strip_smtp_comments_inplace")
 
 TEST_CASE("rspamd_http_parse_keepalive_timeout")
 {
-	std::vector<std::pair<std::string, long>> cases {
-			{"timeout=5, max=1000", 5},
-			{"max=1000, timeout=5", 5},
-			{"max=1000, timeout=", -1},
-			{"max=1000, timeout=0", 0},
-			{"max=1000, timeout=-5", -1},
-			{"timeout=5", 5},
-			{"    timeout=5;    ", 5},
-			{"timeout  =   5", 5},
+	std::vector<std::pair<std::string, long>> cases{
+		{"timeout=5, max=1000",  5},
+		{"max=1000, timeout=5",  5},
+		{"max=1000, timeout=",   -1},
+		{"max=1000, timeout=0",  0},
+		{"max=1000, timeout=-5", -1},
+		{"timeout=5",            5},
+		{"    timeout=5;    ",   5},
+		{"timeout  =   5",       5},
 	};
 
-	for (const auto &c : cases) {
+	for (const auto &c: cases) {
 		SUBCASE (("parse http keepalive header " + c.first).c_str()) {
 			rspamd_ftok_t t;
 			t.begin = c.first.data();
@@ -84,6 +86,76 @@ TEST_CASE("rspamd_http_parse_keepalive_timeout")
 	}
 }
 
+TEST_CASE("rspamd_fstring_gzip tests")
+{
+	rspamd_fstring_t *fstr;
+
+	// Test empty data compression
+	SUBCASE("Empty data") {
+		fstr = rspamd_fstring_new_init("", 0);
+		gboolean result = rspamd_fstring_gzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(fstr->len == 20);
+		result = rspamd_fstring_gunzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(fstr->len == 0);
+		rspamd_fstring_free(fstr);
+	}
+
+	SUBCASE("Non empty data") {
+		fstr = RSPAMD_FSTRING_LIT("helohelo");
+		gboolean result = rspamd_fstring_gzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(fstr->len == 26);
+		result = rspamd_fstring_gunzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(memcmp(fstr->str, "helohelo", fstr->len) == 0);
+		CHECK(fstr->len == sizeof("helohelo") - 1);
+		rspamd_fstring_free(fstr);
+	}
+
+	SUBCASE("Some real compression") {
+		fstr = rspamd_fstring_sized_new(sizeof("helohelo") * 1024);
+		for (int i = 0; i < 1024; i ++) {
+			fstr = rspamd_fstring_append(fstr, "helohelo", sizeof("helohelo") - 1);
+		}
+		gboolean result = rspamd_fstring_gzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(fstr->len == 49);
+		result = rspamd_fstring_gunzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(memcmp(fstr->str, "helohelo", sizeof("helohelo") - 1) == 0);
+		CHECK(fstr->len == (sizeof("helohelo") - 1) * 1024);
+		rspamd_fstring_free(fstr);
+	}
+
+	SUBCASE("Random data compression") {
+		rspamd_cryptobox_fast_hash_state_t hst;
+		rspamd_cryptobox_fast_hash_init(&hst, 0);
+		fstr = rspamd_fstring_sized_new(30 * 1024 * 1024);
+		for (int i = 0; i < 30 * 1024; i ++) {
+			char tmp[1024];
+			ottery_rand_bytes(tmp, sizeof(tmp));
+			fstr = rspamd_fstring_append(fstr, tmp, sizeof(tmp));
+			rspamd_cryptobox_fast_hash_update(&hst, tmp, sizeof(tmp));
+		}
+		auto crc = rspamd_cryptobox_fast_hash(fstr->str, fstr->len, 0);
+		CHECK(crc == rspamd_cryptobox_fast_hash_final(&hst));
+		gboolean result = rspamd_fstring_gzip(&fstr);
+		CHECK(result == TRUE);
+		// Assuming there are no miracles
+		CHECK(fstr->len >= 30 * 1024 * 1024);
+		result = rspamd_fstring_gunzip(&fstr);
+		CHECK(result == TRUE);
+		CHECK(fstr->len == 30 * 1024 * 1024);
+		auto final_crc = rspamd_cryptobox_fast_hash(fstr->str, fstr->len, 0);
+		CHECK(crc == final_crc);
+		rspamd_fstring_free(fstr);
+	}
+
+}
+
+
 }
 
 #endif