fuzz/driver.c

   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 {
  23     return FuzzerInitialize(argc, argv);
  24 }
  25
  26 int 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
  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