1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// SPDX-License-Identifier: GPL-2.0
/*
* Copyright (C) 2018-2020 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
*/
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
const char *__asan_default_options()
{
return "verbosity=1";
}
int wg_main(int argc, char *argv[]);
static FILE *devnull;
int LLVMFuzzerTestOneInput(const char *data, size_t data_len)
{
char *argv[8192] = { 0 }, *args;
size_t argc = 0;
FILE *fake_stdin = NULL;
if (!devnull) {
assert((devnull = fopen("/dev/null", "r+")));
stdin = stdout = stderr = devnull;
}
assert((args = malloc(data_len)));
memcpy(args, data, data_len);
if (data_len)
args[data_len - 1] = '\0';
for (const char *arg = args; argc < 8192 && arg - args < data_len; arg += strlen(arg) + 1) {
if (arg[0])
assert((argv[argc++] = strdup(arg)));
}
if (!argc)
assert((argv[argc++] = strdup("no argv[0]!")));
if (argc > 2 && (!strcmp(argv[1], "show") || !strcmp(argv[1], "showconf") || !strcmp(argv[1], "set") || !strcmp(argv[1], "setconf") || !strcmp(argv[1], "addconf") || !strcmp(argv[1], "syncconf"))) {
free(argv[2]);
assert((argv[2] = strdup("wg0")));
}
if (argc >= 2 && !strcmp(argv[1], "pubkey")) {
char *arg;
size_t len;
for (size_t i = 2; i < argc; ++i)
free(argv[i]);
argc = 2;
arg = args;
for (; !arg[0]; ++arg);
arg += strlen(arg) + 1;
for (; !arg[0]; ++arg);
arg += strlen(arg) + 1;
len = data_len - (arg - args);
if (len <= 1)
goto done;
assert((fake_stdin = fmemopen(arg, len - 1, "r")));
stdin = fake_stdin;
}
wg_main(argc, argv);
done:
for (size_t i = 0; i < argc; ++i)
free(argv[i]);
free(args);
if (fake_stdin)
fclose(fake_stdin);
return 0;
}
|
