1/*
2 * Copyright 2016-2018 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 * https://www.openssl.org/source/license.html
8 * or in the file LICENSE in the source distribution.
9 */
10#include <stdint.h>
11#include <unistd.h>
12#include <stdlib.h>
13#include <openssl/opensslconf.h>
14#include "fuzzer.h"
15
16#ifndef OPENSSL_NO_FUZZ_LIBFUZZER
17
18int LLVMFuzzerInitialize(int *argc, char ***argv);
19int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len);
20
21int LLVMFuzzerInitialize(int *argc, char ***argv)
22{
23    return FuzzerInitialize(argc, argv);
24}
25
26int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len)
27{
28    return FuzzerTestOneInput(buf, len);
29}
30
31#elif !defined(OPENSSL_NO_FUZZ_AFL)
32
33#define BUF_SIZE 65536
34
35int main(int argc, char** argv)
36{
37    FuzzerInitialize(&argc, &argv);
38
39    while (__AFL_LOOP(10000)) {
40        uint8_t *buf = malloc(BUF_SIZE);
41        size_t size = read(0, buf, BUF_SIZE);
42
43        FuzzerTestOneInput(buf, size);
44        free(buf);
45    }
46
47    FuzzerCleanup();
48    return 0;
49}
50
51#else
52
53#error "Unsupported fuzzer"
54
55#endif
56