xref: /aosp_15_r20/external/lz4/ossfuzz/round_trip_hc_fuzzer.c (revision 27162e4e17433d5aa7cb38e7b6a433a09405fc7f)
1*27162e4eSAndroid Build Coastguard Worker /**
2*27162e4eSAndroid Build Coastguard Worker  * This fuzz target performs a lz4 round-trip test (compress & decompress),
3*27162e4eSAndroid Build Coastguard Worker  * compares the result with the original, and calls abort() on corruption.
4*27162e4eSAndroid Build Coastguard Worker  */
5*27162e4eSAndroid Build Coastguard Worker 
6*27162e4eSAndroid Build Coastguard Worker #include <stddef.h>
7*27162e4eSAndroid Build Coastguard Worker #include <stdint.h>
8*27162e4eSAndroid Build Coastguard Worker #include <stdlib.h>
9*27162e4eSAndroid Build Coastguard Worker #include <string.h>
10*27162e4eSAndroid Build Coastguard Worker 
11*27162e4eSAndroid Build Coastguard Worker #include "fuzz_helpers.h"
12*27162e4eSAndroid Build Coastguard Worker #include "fuzz_data_producer.h"
13*27162e4eSAndroid Build Coastguard Worker #include "lz4.h"
14*27162e4eSAndroid Build Coastguard Worker #include "lz4hc.h"
15*27162e4eSAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)16*27162e4eSAndroid Build Coastguard Worker int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
17*27162e4eSAndroid Build Coastguard Worker {
18*27162e4eSAndroid Build Coastguard Worker     FUZZ_dataProducer_t *producer = FUZZ_dataProducer_create(data, size);
19*27162e4eSAndroid Build Coastguard Worker     int const level = FUZZ_dataProducer_range32(producer,
20*27162e4eSAndroid Build Coastguard Worker         LZ4HC_CLEVEL_MIN, LZ4HC_CLEVEL_MAX);
21*27162e4eSAndroid Build Coastguard Worker     size = FUZZ_dataProducer_remainingBytes(producer);
22*27162e4eSAndroid Build Coastguard Worker 
23*27162e4eSAndroid Build Coastguard Worker     size_t const dstCapacity = LZ4_compressBound(size);
24*27162e4eSAndroid Build Coastguard Worker     char* const dst = (char*)malloc(dstCapacity);
25*27162e4eSAndroid Build Coastguard Worker     char* const rt = (char*)malloc(size);
26*27162e4eSAndroid Build Coastguard Worker 
27*27162e4eSAndroid Build Coastguard Worker     FUZZ_ASSERT(dst);
28*27162e4eSAndroid Build Coastguard Worker     FUZZ_ASSERT(rt);
29*27162e4eSAndroid Build Coastguard Worker 
30*27162e4eSAndroid Build Coastguard Worker     /* Compression must succeed and round trip correctly. */
31*27162e4eSAndroid Build Coastguard Worker     int const dstSize = LZ4_compress_HC((const char*)data, dst, size,
32*27162e4eSAndroid Build Coastguard Worker                                         dstCapacity, level);
33*27162e4eSAndroid Build Coastguard Worker     FUZZ_ASSERT(dstSize > 0);
34*27162e4eSAndroid Build Coastguard Worker 
35*27162e4eSAndroid Build Coastguard Worker     int const rtSize = LZ4_decompress_safe(dst, rt, dstSize, size);
36*27162e4eSAndroid Build Coastguard Worker     FUZZ_ASSERT_MSG(rtSize == size, "Incorrect size");
37*27162e4eSAndroid Build Coastguard Worker     FUZZ_ASSERT_MSG(!memcmp(data, rt, size), "Corruption!");
38*27162e4eSAndroid Build Coastguard Worker 
39*27162e4eSAndroid Build Coastguard Worker     free(dst);
40*27162e4eSAndroid Build Coastguard Worker     free(rt);
41*27162e4eSAndroid Build Coastguard Worker     FUZZ_dataProducer_free(producer);
42*27162e4eSAndroid Build Coastguard Worker 
43*27162e4eSAndroid Build Coastguard Worker     return 0;
44*27162e4eSAndroid Build Coastguard Worker }
45