test: internal: gzip: Add testcases for payloads of concatenated gzip

Signed-off-by: Hiroshi Hatake <[email protected]>

tests/internal/gzip.c

@@ -40,7 +40,69 @@ void test_compress()
     flb_free(str);
 }
 
+void test_concatenated_gzip_count()
+{
+     int ret;
+    int sample_len;
+    char *in_data = morpheus;
+    size_t in_len;
+    void *str;
+    size_t len;
+    flb_sds_t payload = NULL;
+    flb_sds_t payload2 = NULL;
+    size_t border_count = 0;
+
+    sample_len = strlen(morpheus);
+    in_len = sample_len;
+    ret = flb_gzip_compress(in_data, in_len, &str, &len);
+    TEST_CHECK(ret == 0);
+
+    payload = flb_sds_create_len((char *)str, len);
+    payload2 = flb_sds_create_len((char *)str, len);
+    ret = flb_sds_cat_safe(&payload, payload2, flb_sds_len(payload2));
+    TEST_CHECK(ret == 0);
+
+    border_count = flb_gzip_count(payload, flb_sds_len(payload), NULL, 0);
+    TEST_CHECK(border_count == 1);
+
+    flb_free(str);
+    flb_sds_destroy(payload);
+    flb_sds_destroy(payload2);
+}
+
+void test_not_overflow_for_concatenated_gzip()
+{
+    const char data[] = {
+        0x00, 0x00, /* Initial padding */
+        0x1F, 0x8B, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, /* First gzip header (valid header) */
+        0x1F, 0x8B, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, /* Second gzip header (valid header) */
+        0x1F, 0x8B, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, /* Third gzip header (valid header) */
+    };
+    size_t len = sizeof(data);
+    size_t *borders = NULL;
+    size_t border_count = 0;
+    size_t count = 0;
+
+    /* Vaild gzip payloads have to 18 bytes lentgh at least.
+     * So, we get only 2 of vaild parts.
+     */
+    border_count = flb_gzip_count(data, len, NULL, 0);
+    TEST_CHECK(border_count == 2);
+
+    borders = (size_t *)flb_calloc(1, sizeof(size_t) * (border_count + 1));
+    TEST_CHECK(borders != NULL);
+
+    count = flb_gzip_count(data, len, &borders, border_count);
+    TEST_CHECK(count == 2);
+
+    if (borders != NULL) {
+        free(borders);
+    }
+}
+
 TEST_LIST = {
     {"compress", test_compress},
+    {"count",  test_concatenated_gzip_count},
+    {"not_overflow", test_not_overflow_for_concatenated_gzip},
     { 0 }
 };