2020-08-01 23:01:39 +03:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2020, the SerenityOS developers.
|
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
* modification, are permitted provided that the following conditions are met:
|
|
|
|
*
|
|
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
|
|
* list of conditions and the following disclaimer.
|
|
|
|
*
|
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
|
|
* and/or other materials provided with the distribution.
|
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
#include <AK/TestSuite.h>
|
|
|
|
|
2020-09-05 14:18:01 +03:00
|
|
|
#include <AK/FixedArray.h>
|
2020-08-01 23:01:39 +03:00
|
|
|
#include <LibCompress/Deflate.h>
|
2020-08-28 18:53:57 +03:00
|
|
|
#include <LibCompress/Gzip.h>
|
2020-08-01 23:01:39 +03:00
|
|
|
#include <LibCompress/Zlib.h>
|
|
|
|
|
2020-08-28 11:21:05 +03:00
|
|
|
static bool compare(ReadonlyBytes lhs, ReadonlyBytes rhs)
|
2020-08-01 23:01:39 +03:00
|
|
|
{
|
2020-08-20 19:17:19 +03:00
|
|
|
if (lhs.size() != rhs.size())
|
|
|
|
return false;
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
for (size_t idx = 0; idx < lhs.size(); ++idx) {
|
|
|
|
if (lhs[idx] != rhs[idx])
|
|
|
|
return false;
|
2020-08-01 23:01:39 +03:00
|
|
|
}
|
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
return true;
|
|
|
|
}
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
TEST_CASE(deflate_decompress_compressed_block)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x0B, 0xC9, 0xC8, 0x2C, 0x56, 0x00, 0xA2, 0x44, 0x85, 0xE2, 0xCC, 0xDC,
|
|
|
|
0x82, 0x9C, 0x54, 0x85, 0x92, 0xD4, 0x8A, 0x12, 0x85, 0xB4, 0x4C, 0x20,
|
|
|
|
0xCB, 0x4A, 0x13, 0x00
|
|
|
|
};
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
const u8 uncompressed[] = "This is a simple text file :)";
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-26 15:22:25 +03:00
|
|
|
const auto decompressed = Compress::DeflateDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
2020-08-20 19:17:19 +03:00
|
|
|
}
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-26 15:22:25 +03:00
|
|
|
TEST_CASE(deflate_decompress_uncompressed_block)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x01, 0x0d, 0x00, 0xf2, 0xff, 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2c, 0x20,
|
|
|
|
0x57, 0x6f, 0x72, 0x6c, 0x64, 0x21
|
|
|
|
};
|
|
|
|
|
|
|
|
const u8 uncompressed[] = "Hello, World!";
|
|
|
|
|
|
|
|
const auto decompressed = Compress::DeflateDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE(deflate_decompress_multiple_blocks)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x00, 0x1f, 0x00, 0xe0, 0xff, 0x54, 0x68, 0x65, 0x20, 0x66, 0x69, 0x72,
|
|
|
|
0x73, 0x74, 0x20, 0x62, 0x6c, 0x6f, 0x63, 0x6b, 0x20, 0x69, 0x73, 0x20,
|
|
|
|
0x75, 0x6e, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73, 0x65, 0x64,
|
|
|
|
0x53, 0x48, 0xcc, 0x4b, 0x51, 0x28, 0xc9, 0x48, 0x55, 0x28, 0x4e, 0x4d,
|
|
|
|
0xce, 0x07, 0x32, 0x93, 0x72, 0xf2, 0x93, 0xb3, 0x15, 0x32, 0x8b, 0x15,
|
|
|
|
0x92, 0xf3, 0x73, 0x0b, 0x8a, 0x52, 0x8b, 0x8b, 0x53, 0x53, 0xf4, 0x00
|
|
|
|
};
|
|
|
|
|
|
|
|
const u8 uncompressed[] = "The first block is uncompressed and the second block is compressed.";
|
|
|
|
|
|
|
|
const auto decompressed = Compress::DeflateDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE(deflate_decompress_zeroes)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0xed, 0xc1, 0x01, 0x0d, 0x00, 0x00, 0x00, 0xc2, 0xa0, 0xf7, 0x4f, 0x6d,
|
|
|
|
0x0f, 0x07, 0x14, 0x00, 0x00, 0x00, 0xf0, 0x6e
|
|
|
|
};
|
|
|
|
|
|
|
|
u8 uncompressed[4096];
|
|
|
|
Bytes { uncompressed, sizeof(uncompressed) }.fill(0);
|
|
|
|
|
|
|
|
const auto decompressed = Compress::DeflateDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE(zlib_decompress_simple)
|
2020-08-20 19:17:19 +03:00
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x78, 0x01, 0x01, 0x1D, 0x00, 0xE2, 0xFF, 0x54, 0x68, 0x69, 0x73, 0x20,
|
|
|
|
0x69, 0x73, 0x20, 0x61, 0x20, 0x73, 0x69, 0x6D, 0x70, 0x6C, 0x65, 0x20,
|
|
|
|
0x74, 0x65, 0x78, 0x74, 0x20, 0x66, 0x69, 0x6C, 0x65, 0x20, 0x3A, 0x29,
|
|
|
|
0x99, 0x5E, 0x09, 0xE8
|
|
|
|
};
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
const u8 uncompressed[] = "This is a simple text file :)";
|
2020-08-01 23:01:39 +03:00
|
|
|
|
2020-08-26 15:22:25 +03:00
|
|
|
const auto decompressed = Compress::Zlib::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
2020-08-01 23:01:39 +03:00
|
|
|
}
|
2020-08-20 19:17:19 +03:00
|
|
|
|
2020-08-28 18:53:57 +03:00
|
|
|
TEST_CASE(gzip_decompress_simple)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x1f, 0x8b, 0x08, 0x00, 0x77, 0xff, 0x47, 0x5f, 0x02, 0xff, 0x2b, 0xcf,
|
|
|
|
0x2f, 0x4a, 0x31, 0x54, 0x48, 0x4c, 0x4a, 0x56, 0x28, 0x07, 0xb2, 0x8c,
|
|
|
|
0x00, 0xc2, 0x1d, 0x22, 0x15, 0x0f, 0x00, 0x00, 0x00
|
|
|
|
};
|
|
|
|
|
|
|
|
const u8 uncompressed[] = "word1 abc word2";
|
|
|
|
|
|
|
|
const auto decompressed = Compress::GzipDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
2020-08-31 13:30:16 +03:00
|
|
|
TEST_CASE(gzip_decompress_multiple_members)
|
2020-08-28 18:53:57 +03:00
|
|
|
{
|
|
|
|
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x1f, 0x8b, 0x08, 0x00, 0xe0, 0x03, 0x48, 0x5f, 0x02, 0xff, 0x4b, 0x4c,
|
|
|
|
0x4a, 0x4e, 0x4c, 0x4a, 0x06, 0x00, 0x4c, 0x99, 0x6e, 0x72, 0x06, 0x00,
|
|
|
|
0x00, 0x00, 0x1f, 0x8b, 0x08, 0x00, 0xe0, 0x03, 0x48, 0x5f, 0x02, 0xff,
|
|
|
|
0x4b, 0x4c, 0x4a, 0x4e, 0x4c, 0x4a, 0x06, 0x00, 0x4c, 0x99, 0x6e, 0x72,
|
|
|
|
0x06, 0x00, 0x00, 0x00
|
|
|
|
};
|
|
|
|
|
|
|
|
const u8 uncompressed[] = "abcabcabcabc";
|
|
|
|
|
|
|
|
const auto decompressed = Compress::GzipDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
2020-08-31 13:30:16 +03:00
|
|
|
TEST_CASE(gzip_decompress_zeroes)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x1f, 0x8b, 0x08, 0x00, 0x6e, 0x7a, 0x4b, 0x5f, 0x02, 0xff, 0xed, 0xc1,
|
|
|
|
0x31, 0x01, 0x00, 0x00, 0x00, 0xc2, 0xa0, 0xf5, 0x4f, 0xed, 0x61, 0x0d,
|
|
|
|
0xa0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x6e, 0xcd, 0xcd, 0xe8,
|
|
|
|
0x7e, 0x00, 0x00, 0x02, 0x00
|
|
|
|
};
|
|
|
|
|
|
|
|
const u8 uncompressed[128 * 1024] = { 0 };
|
|
|
|
|
|
|
|
const auto decompressed = Compress::GzipDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare({ uncompressed, sizeof(uncompressed) }, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
2020-09-05 14:18:01 +03:00
|
|
|
TEST_CASE(gzip_decompress_repeat_around_buffer)
|
|
|
|
{
|
|
|
|
const u8 compressed[] = {
|
|
|
|
0x1f, 0x8b, 0x08, 0x00, 0xc6, 0x74, 0x53, 0x5f, 0x02, 0xff, 0xed, 0xc1,
|
|
|
|
0x01, 0x0d, 0x00, 0x00, 0x0c, 0x02, 0xa0, 0xdb, 0xbf, 0xf4, 0x37, 0x6b,
|
|
|
|
0x08, 0x24, 0xdb, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x80, 0xca,
|
|
|
|
0xb8, 0x07, 0xcd, 0xe5, 0x38, 0xfa, 0x00, 0x80, 0x00, 0x00
|
|
|
|
};
|
|
|
|
|
|
|
|
FixedArray<u8> uncompressed { 0x8000 };
|
|
|
|
uncompressed.bytes().slice(0x0, 0x100).fill(1);
|
|
|
|
uncompressed.bytes().slice(0x100, 0x7e00).fill(0);
|
|
|
|
uncompressed.bytes().slice(0x7f00, 0x100).fill(1);
|
|
|
|
|
|
|
|
const auto decompressed = Compress::GzipDecompressor::decompress_all({ compressed, sizeof(compressed) });
|
|
|
|
EXPECT(compare(uncompressed, decompressed.bytes()));
|
|
|
|
}
|
|
|
|
|
2020-08-20 19:17:19 +03:00
|
|
|
TEST_MAIN(Compress)
|