X-Git-Url: https://git.openssl.org/gitweb/?a=blobdiff_plain;f=fuzz%2Fdriver.c;h=21bbb255afe51066f16af31dff214dee347bba28;hb=0282aeb690d63fab73a07191b63300a2fe30d212;hp=de515748cefc1137021f02cd7663c4e93c8b62c8;hpb=f59d0131cb6fc224aee0a0a92de1f04cdebe97c8;p=openssl.git

diff --git a/fuzz/driver.c b/fuzz/driver.c
index de515748ce..21bbb255af 100644
--- a/fuzz/driver.c
+++ b/fuzz/driver.c
@@ -9,6 +9,7 @@
  */
 #include <stdint.h>
 #include <unistd.h>
+#include <stdlib.h>
 #include <openssl/opensslconf.h>
 #include "fuzzer.h"
 
@@ -16,12 +17,11 @@
 
 int LLVMFuzzerInitialize(int *argc, char ***argv)
 {
-    if (FuzzerInitialize)
-        return FuzzerInitialize(argc, argv);
-    return 0;
+    return FuzzerInitialize(argc, argv);
 }
 
-int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len) {
+int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len)
+{
     return FuzzerTestOneInput(buf, len);
 }
 
@@ -31,8 +31,7 @@ int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t len) {
 
 int main(int argc, char** argv)
 {
-    if (FuzzerInitialize)
-        FuzzerInitialize(&argc, &argv);
+    FuzzerInitialize(&argc, &argv);
 
     while (__AFL_LOOP(10000)) {
         uint8_t *buf = malloc(BUF_SIZE);
@@ -41,6 +40,8 @@ int main(int argc, char** argv)
         FuzzerTestOneInput(buf, size);
         free(buf);
     }
+
+    FuzzerCleanup();
     return 0;
 }