1 files changed, 101 insertions, 0 deletions
diff --git a/fuzz/wget_options_fuzzer.c b/fuzz/wget_options_fuzzer.c
new file mode 100644
index 0000000..d3dcaff
--- /dev/null
+++ b/fuzz/wget_options_fuzzer.c
@@ -0,0 +1,101 @@
+/*
+ * Copyright(c) 2017-2018 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Wget.
+ *
+ * GNU Wget is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GNU Wget is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Wget.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include <config.h>
+
+#include <sys/types.h>
+#include <dirent.h> // opendir, readdir
+#include <stdint.h> // uint8_t
+#include <stdio.h>  // fmemopen
+#include <string.h>  // strncmp
+#include <fcntl.h>  // open flags
+#include <unistd.h>  // close
+#include <setjmp.h> // longjmp, setjmp
+
+#ifdef  __cplusplus
+  extern "C" {
+#endif
+  // declarations for wget internal functions
+  int main_wget(int argc, const char **argv);
+  void cleanup(void);
+  FILE *fopen_wget(const char *pathname, const char *mode);
+  FILE *fopen_wgetrc(const char *pathname, const char *mode);
+  void exit_wget(int status);
+#ifdef __cplusplus
+  }
+#endif
+
+#include "fuzzer.h"
+
+static const uint8_t *g_data;
+static size_t g_size;
+
+FILE *fopen_wget(const char *pathname, const char *mode)
+{
+	return fopen("/dev/null", mode);
+}
+
+FILE *fopen_wgetrc(const char *pathname, const char *mode)
+{
+#ifdef HAVE_FMEMOPEN
+	return fmemopen((void *) g_data, g_size, mode);
+#else
+	return NULL;
+#endif
+}
+
+static jmp_buf jmpbuf;
+#ifdef FUZZING
+void exit_wget(int status)
+{
+	longjmp(jmpbuf, 1);
+}
+#else
+void exit(int status)
+{
+	longjmp(jmpbuf, 1);
+}
+#endif
+
+int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
+{
+	static const char *argv[] = { "wget", "-q" };
+
+	if (size > 2048) // same as max_len = ... in .options file
+		return 0;
+
+	g_data = data;
+	g_size = size;
+
+	FILE *bak = stderr;
+	stderr = fopen("/dev/null", "w");
+
+	if (setjmp(jmpbuf))
+		goto done;
+
+	main_wget(sizeof(argv)/sizeof(argv[0]), argv);
+
+done:
+	cleanup();
+
+	fclose(stderr);
+	stderr = bak;
+
+	return 0;
+}