fuzz/driver.c

   1 /*
   2  * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL licenses, (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 {
  20     if (FuzzerInitialize)
  21         return FuzzerInitialize(argc, argv);
  22     return 0;
  23 }
  24
  25 int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len)
  26 {
  27     return FuzzerTestOneInput(buf, len);
  28 }
  29
  30 #elif !defined(OPENSSL_NO_FUZZ_AFL)
  31
  32 #define BUF_SIZE 65536
  33
  34 int main(int argc, char** argv)
  35 {
  36     if (FuzzerInitialize)
  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     return 0;
  47 }
  48
  49 #else
  50
  51 #error "Unsupported fuzzer"
  52
  53 #endif