diff options
author | Kurt Roeckx <kurt@roeckx.be> | 2016-05-07 22:09:13 +0200 |
---|---|---|
committer | Kurt Roeckx <kurt@roeckx.be> | 2016-06-04 14:39:24 +0200 |
commit | f59d0131cb6fc224aee0a0a92de1f04cdebe97c8 (patch) | |
tree | 58b37d06e30988507c0b9eb59bc66ff8747a58af /fuzz/driver.c | |
parent | 255cf605d67e72b73e6130d4c7bbe68c0eb22d8e (diff) |
Add support for fuzzing with AFL
Reviewed-by: Ben Laurie <ben@links.org>
MR: #2740
Diffstat (limited to 'fuzz/driver.c')
-rw-r--r-- | fuzz/driver.c | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/fuzz/driver.c b/fuzz/driver.c new file mode 100644 index 0000000000..de515748ce --- /dev/null +++ b/fuzz/driver.c @@ -0,0 +1,51 @@ +/* + * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved. + * + * Licensed under the OpenSSL licenses, (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * https://www.openssl.org/source/license.html + * or in the file LICENSE in the source distribution. + */ +#include <stdint.h> +#include <unistd.h> +#include <openssl/opensslconf.h> +#include "fuzzer.h" + +#ifndef OPENSSL_NO_FUZZ_LIBFUZZER + +int LLVMFuzzerInitialize(int *argc, char ***argv) +{ + if (FuzzerInitialize) + return FuzzerInitialize(argc, argv); + return 0; +} + +int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len) { + return FuzzerTestOneInput(buf, len); +} + +#elif !defined(OPENSSL_NO_FUZZ_AFL) + +#define BUF_SIZE 65536 + +int main(int argc, char** argv) +{ + if (FuzzerInitialize) + FuzzerInitialize(&argc, &argv); + + while (__AFL_LOOP(10000)) { + uint8_t *buf = malloc(BUF_SIZE); + size_t size = read(0, buf, BUF_SIZE); + + FuzzerTestOneInput(buf, size); + free(buf); + } + return 0; +} + +#else + +#error "Unsupported fuzzer" + +#endif |