diff options
Diffstat (limited to 'fuzz/wget_options_fuzzer.c')
-rw-r--r-- | fuzz/wget_options_fuzzer.c | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/fuzz/wget_options_fuzzer.c b/fuzz/wget_options_fuzzer.c new file mode 100644 index 0000000..08a3760 --- /dev/null +++ b/fuzz/wget_options_fuzzer.c @@ -0,0 +1,113 @@ +/* + * Copyright (c) 2017-2020 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 int do_jump; +static jmp_buf jmpbuf; +#ifdef FUZZING +void exit_wget(int status) +{ + longjmp(jmpbuf, 1); +} +#elif defined HAVE_DLFCN_H +#include <dlfcn.h> // dlsym +#ifndef RTLD_NEXT +#define RTLD_NEXT RTLD_GLOBAL +#endif +void exit(int status) +{ + if (do_jump) { + longjmp(jmpbuf, 1); + } else { + void (*libc_exit)(int) = (void(*)(int)) dlsym (RTLD_NEXT, "exit"); + libc_exit(status); + } +} +#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; + + CLOSE_STDERR + + do_jump = 1; + + if (setjmp(jmpbuf)) + goto done; + + main_wget(sizeof(argv)/sizeof(argv[0]), argv); + +done: + cleanup(); + + do_jump = 0; + + RESTORE_STDERR + + return 0; +} |