/src/libressl.fuzzers/driver.c
Line  | Count  | Source  | 
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  |  |  | 
18  |  | int LLVMFuzzerInitialize(int *argc, char ***argv);  | 
19  |  | int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len);  | 
20  |  |  | 
21  |  | int LLVMFuzzerInitialize(int *argc, char ***argv)  | 
22  | 2  | { | 
23  | 2  |     return FuzzerInitialize(argc, argv);  | 
24  | 2  | }  | 
25  |  |  | 
26  |  | int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len)  | 
27  | 14.3k  | { | 
28  | 14.3k  |     return FuzzerTestOneInput(buf, len);  | 
29  | 14.3k  | }  | 
30  |  |  | 
31  |  | #elif !defined(OPENSSL_NO_FUZZ_AFL)  | 
32  |  |  | 
33  |  | #define BUF_SIZE 65536  | 
34  |  |  | 
35  |  | int 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  |