1*3117ece4Schristos /* 2*3117ece4Schristos * Copyright (c) Meta Platforms, Inc. and affiliates. 3*3117ece4Schristos * All rights reserved. 4*3117ece4Schristos * 5*3117ece4Schristos * This source code is licensed under both the BSD-style license (found in the 6*3117ece4Schristos * LICENSE file in the root directory of this source tree) and the GPLv2 (found 7*3117ece4Schristos * in the COPYING file in the root directory of this source tree). 8*3117ece4Schristos * You may select, at your option, one of the above-listed licenses. 9*3117ece4Schristos */ 10*3117ece4Schristos /** 11*3117ece4Schristos * Helper functions for fuzzing. 12*3117ece4Schristos */ 13*3117ece4Schristos 14*3117ece4Schristos #ifndef ZSTD_HELPERS_H 15*3117ece4Schristos #define ZSTD_HELPERS_H 16*3117ece4Schristos 17*3117ece4Schristos #define ZSTD_STATIC_LINKING_ONLY 18*3117ece4Schristos 19*3117ece4Schristos #include "zstd.h" 20*3117ece4Schristos #include "zstd_errors.h" 21*3117ece4Schristos #include "fuzz_data_producer.h" 22*3117ece4Schristos #include <stdint.h> 23*3117ece4Schristos 24*3117ece4Schristos #ifdef __cplusplus 25*3117ece4Schristos extern "C" { 26*3117ece4Schristos #endif 27*3117ece4Schristos 28*3117ece4Schristos extern const int kMinClevel; 29*3117ece4Schristos extern const int kMaxClevel; 30*3117ece4Schristos 31*3117ece4Schristos void FUZZ_setRandomParameters(ZSTD_CCtx *cctx, size_t srcSize, FUZZ_dataProducer_t *producer); 32*3117ece4Schristos 33*3117ece4Schristos ZSTD_compressionParameters FUZZ_randomCParams(size_t srcSize, FUZZ_dataProducer_t *producer); 34*3117ece4Schristos ZSTD_frameParameters FUZZ_randomFParams(FUZZ_dataProducer_t *producer); 35*3117ece4Schristos ZSTD_parameters FUZZ_randomParams(size_t srcSize, FUZZ_dataProducer_t *producer); 36*3117ece4Schristos 37*3117ece4Schristos typedef struct { 38*3117ece4Schristos void* buff; 39*3117ece4Schristos size_t size; 40*3117ece4Schristos } FUZZ_dict_t; 41*3117ece4Schristos 42*3117ece4Schristos /* Quickly train a dictionary from a source for fuzzing. 43*3117ece4Schristos * NOTE: Don't use this to train production dictionaries, it is only optimized 44*3117ece4Schristos * for speed, and doesn't care about dictionary quality. 45*3117ece4Schristos */ 46*3117ece4Schristos FUZZ_dict_t FUZZ_train(void const* src, size_t srcSize, FUZZ_dataProducer_t *producer); 47*3117ece4Schristos 48*3117ece4Schristos #ifdef FUZZ_THIRD_PARTY_SEQ_PROD 49*3117ece4Schristos extern void* FUZZ_seqProdState; 50*3117ece4Schristos #endif 51*3117ece4Schristos 52*3117ece4Schristos #ifdef __cplusplus 53*3117ece4Schristos } 54*3117ece4Schristos #endif 55*3117ece4Schristos 56*3117ece4Schristos #endif /* ZSTD_HELPERS_H */ 57