mirror of
https://github.com/intel/isa-l.git
synced 2024-12-13 01:42:56 +01:00
4f59eeda90
Moved the afl fuzz test and added llvm fuzz tests including inflate and round trip compress and inflate. Currently only works with clang, std makefile and libFuzzer installed. Need to add checking and support later when libfuzzer is more tightly integrated into the compiler. Change-Id: I2db9ad2335d6c5ed846886703b58225f67bcc935 Signed-off-by: Greg Tucker <greg.b.tucker@intel.com>
70 lines
2.1 KiB
C
70 lines
2.1 KiB
C
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h>
|
|
#include <stddef.h>
|
|
#include <zlib.h>
|
|
#include <assert.h>
|
|
#include "igzip_lib.h"
|
|
|
|
int LLVMFuzzerTestOneInput(const uint8_t * data, size_t size)
|
|
{
|
|
struct inflate_state state;
|
|
z_stream zstate;
|
|
size_t out_buf_size = 2 * size;
|
|
int zret, iret;
|
|
char z_msg_invalid_code_set[] = "invalid code lengths set";
|
|
char z_msg_invalid_dist_set[] = "invalid distances set";
|
|
char z_msg_invalid_lit_len_set[] = "invalid literal/lengths set";
|
|
|
|
uint8_t *isal_out_buf = (uint8_t *) malloc(size * 2);
|
|
uint8_t *zlib_out_buf = (uint8_t *) malloc(size * 2);
|
|
|
|
assert(NULL != isal_out_buf && NULL != zlib_out_buf);
|
|
|
|
/* Inflate data with isal_inflate */
|
|
memset(&state, 0xff, sizeof(struct inflate_state));
|
|
|
|
isal_inflate_init(&state);
|
|
state.next_in = (uint8_t *) data;
|
|
state.avail_in = size;
|
|
state.next_out = isal_out_buf;
|
|
state.avail_out = out_buf_size;
|
|
|
|
iret = isal_inflate_stateless(&state);
|
|
|
|
/* Inflate data with zlib */
|
|
zstate.zalloc = Z_NULL;
|
|
zstate.zfree = Z_NULL;
|
|
zstate.opaque = Z_NULL;
|
|
zstate.avail_in = size;
|
|
zstate.next_in = (Bytef *) data;
|
|
zstate.avail_out = out_buf_size;
|
|
zstate.next_out = zlib_out_buf;
|
|
inflateInit2(&zstate, -15);
|
|
|
|
zret = inflate(&zstate, Z_FINISH);
|
|
|
|
if (zret == Z_STREAM_END) {
|
|
/* If zlib finished, assert isal finished with the same answer */
|
|
assert(state.block_state == ISAL_BLOCK_FINISH);
|
|
assert(zstate.total_out == state.total_out);
|
|
assert(memcmp(isal_out_buf, zlib_out_buf, state.total_out) == 0);
|
|
} else if (zret < 0) {
|
|
if (zret != Z_BUF_ERROR)
|
|
/* If zlib errors, assert isal errors, excluding a few
|
|
* cases where zlib is overzealous */
|
|
assert(iret < 0 || strcmp(zstate.msg, z_msg_invalid_code_set) == 0
|
|
|| strcmp(zstate.msg, z_msg_invalid_dist_set) == 0
|
|
|| strcmp(zstate.msg, z_msg_invalid_lit_len_set) == 0);
|
|
} else
|
|
/* If zlib did not finish or error, assert isal did not finish
|
|
* or that isal found an invalid header since isal notices the
|
|
* error faster than zlib */
|
|
assert(iret > 0 || iret == ISAL_INVALID_BLOCK);
|
|
|
|
inflateEnd(&zstate);
|
|
free(isal_out_buf);
|
|
free(zlib_out_buf);
|
|
return 0;
|
|
}
|