1*08b48e0bSAndroid Build Coastguard Worker /*
2*08b48e0bSAndroid Build Coastguard Worker american fuzzy lop++ - common routines
3*08b48e0bSAndroid Build Coastguard Worker --------------------------------------
4*08b48e0bSAndroid Build Coastguard Worker
5*08b48e0bSAndroid Build Coastguard Worker Originally written by Michal Zalewski
6*08b48e0bSAndroid Build Coastguard Worker
7*08b48e0bSAndroid Build Coastguard Worker Now maintained by Marc Heuse <[email protected]>,
8*08b48e0bSAndroid Build Coastguard Worker Heiko Eißfeldt <[email protected]> and
9*08b48e0bSAndroid Build Coastguard Worker Andrea Fioraldi <[email protected]>
10*08b48e0bSAndroid Build Coastguard Worker
11*08b48e0bSAndroid Build Coastguard Worker Copyright 2016, 2017 Google Inc. All rights reserved.
12*08b48e0bSAndroid Build Coastguard Worker Copyright 2019-2024 AFLplusplus Project. All rights reserved.
13*08b48e0bSAndroid Build Coastguard Worker
14*08b48e0bSAndroid Build Coastguard Worker Licensed under the Apache License, Version 2.0 (the "License");
15*08b48e0bSAndroid Build Coastguard Worker you may not use this file except in compliance with the License.
16*08b48e0bSAndroid Build Coastguard Worker You may obtain a copy of the License at:
17*08b48e0bSAndroid Build Coastguard Worker
18*08b48e0bSAndroid Build Coastguard Worker https://www.apache.org/licenses/LICENSE-2.0
19*08b48e0bSAndroid Build Coastguard Worker
20*08b48e0bSAndroid Build Coastguard Worker Gather some functions common to multiple executables
21*08b48e0bSAndroid Build Coastguard Worker
22*08b48e0bSAndroid Build Coastguard Worker - detect_file_args
23*08b48e0bSAndroid Build Coastguard Worker
24*08b48e0bSAndroid Build Coastguard Worker */
25*08b48e0bSAndroid Build Coastguard Worker
26*08b48e0bSAndroid Build Coastguard Worker #include <stdlib.h>
27*08b48e0bSAndroid Build Coastguard Worker #include <stdio.h>
28*08b48e0bSAndroid Build Coastguard Worker #include "forkserver.h"
29*08b48e0bSAndroid Build Coastguard Worker #ifndef _GNU_SOURCE
30*08b48e0bSAndroid Build Coastguard Worker #define _GNU_SOURCE
31*08b48e0bSAndroid Build Coastguard Worker #endif
32*08b48e0bSAndroid Build Coastguard Worker #ifndef __USE_GNU
33*08b48e0bSAndroid Build Coastguard Worker #define __USE_GNU
34*08b48e0bSAndroid Build Coastguard Worker #endif
35*08b48e0bSAndroid Build Coastguard Worker #include <string.h>
36*08b48e0bSAndroid Build Coastguard Worker #include <strings.h>
37*08b48e0bSAndroid Build Coastguard Worker #include <math.h>
38*08b48e0bSAndroid Build Coastguard Worker #include <sys/mman.h>
39*08b48e0bSAndroid Build Coastguard Worker
40*08b48e0bSAndroid Build Coastguard Worker #include "debug.h"
41*08b48e0bSAndroid Build Coastguard Worker #include "alloc-inl.h"
42*08b48e0bSAndroid Build Coastguard Worker #include "envs.h"
43*08b48e0bSAndroid Build Coastguard Worker #include "common.h"
44*08b48e0bSAndroid Build Coastguard Worker
45*08b48e0bSAndroid Build Coastguard Worker /* Detect @@ in args. */
46*08b48e0bSAndroid Build Coastguard Worker #include <unistd.h>
47*08b48e0bSAndroid Build Coastguard Worker #include <limits.h>
48*08b48e0bSAndroid Build Coastguard Worker #include <sys/types.h>
49*08b48e0bSAndroid Build Coastguard Worker #include <sys/stat.h>
50*08b48e0bSAndroid Build Coastguard Worker #include <fcntl.h>
51*08b48e0bSAndroid Build Coastguard Worker #include <signal.h>
52*08b48e0bSAndroid Build Coastguard Worker
53*08b48e0bSAndroid Build Coastguard Worker u8 be_quiet = 0;
54*08b48e0bSAndroid Build Coastguard Worker u8 *doc_path = "";
55*08b48e0bSAndroid Build Coastguard Worker u8 last_intr = 0;
56*08b48e0bSAndroid Build Coastguard Worker
57*08b48e0bSAndroid Build Coastguard Worker #ifndef AFL_PATH
58*08b48e0bSAndroid Build Coastguard Worker #define AFL_PATH "/usr/local/lib/afl/"
59*08b48e0bSAndroid Build Coastguard Worker #endif
60*08b48e0bSAndroid Build Coastguard Worker
afl_memmem(const void * haystack,size_t haystacklen,const void * needle,size_t needlelen)61*08b48e0bSAndroid Build Coastguard Worker void *afl_memmem(const void *haystack, size_t haystacklen, const void *needle,
62*08b48e0bSAndroid Build Coastguard Worker size_t needlelen) {
63*08b48e0bSAndroid Build Coastguard Worker
64*08b48e0bSAndroid Build Coastguard Worker if (unlikely(needlelen > haystacklen)) { return NULL; }
65*08b48e0bSAndroid Build Coastguard Worker
66*08b48e0bSAndroid Build Coastguard Worker for (u32 i = 0; i <= haystacklen - needlelen; ++i) {
67*08b48e0bSAndroid Build Coastguard Worker
68*08b48e0bSAndroid Build Coastguard Worker if (unlikely(memcmp(haystack + i, needle, needlelen) == 0)) {
69*08b48e0bSAndroid Build Coastguard Worker
70*08b48e0bSAndroid Build Coastguard Worker return (void *)(haystack + i);
71*08b48e0bSAndroid Build Coastguard Worker
72*08b48e0bSAndroid Build Coastguard Worker }
73*08b48e0bSAndroid Build Coastguard Worker
74*08b48e0bSAndroid Build Coastguard Worker }
75*08b48e0bSAndroid Build Coastguard Worker
76*08b48e0bSAndroid Build Coastguard Worker return (void *)NULL;
77*08b48e0bSAndroid Build Coastguard Worker
78*08b48e0bSAndroid Build Coastguard Worker }
79*08b48e0bSAndroid Build Coastguard Worker
set_sanitizer_defaults()80*08b48e0bSAndroid Build Coastguard Worker void set_sanitizer_defaults() {
81*08b48e0bSAndroid Build Coastguard Worker
82*08b48e0bSAndroid Build Coastguard Worker /* Set sane defaults for ASAN if nothing else is specified. */
83*08b48e0bSAndroid Build Coastguard Worker u8 *have_asan_options = getenv("ASAN_OPTIONS");
84*08b48e0bSAndroid Build Coastguard Worker u8 *have_ubsan_options = getenv("UBSAN_OPTIONS");
85*08b48e0bSAndroid Build Coastguard Worker u8 *have_msan_options = getenv("MSAN_OPTIONS");
86*08b48e0bSAndroid Build Coastguard Worker u8 *have_lsan_options = getenv("LSAN_OPTIONS");
87*08b48e0bSAndroid Build Coastguard Worker u8 have_san_options = 0;
88*08b48e0bSAndroid Build Coastguard Worker u8 default_options[1024] =
89*08b48e0bSAndroid Build Coastguard Worker "detect_odr_violation=0:abort_on_error=1:symbolize=0:allocator_may_"
90*08b48e0bSAndroid Build Coastguard Worker "return_null=1:handle_segv=0:handle_sigbus=0:handle_abort=0:handle_"
91*08b48e0bSAndroid Build Coastguard Worker "sigfpe=0:handle_sigill=0:";
92*08b48e0bSAndroid Build Coastguard Worker
93*08b48e0bSAndroid Build Coastguard Worker if (have_asan_options || have_ubsan_options || have_msan_options ||
94*08b48e0bSAndroid Build Coastguard Worker have_lsan_options) {
95*08b48e0bSAndroid Build Coastguard Worker
96*08b48e0bSAndroid Build Coastguard Worker have_san_options = 1;
97*08b48e0bSAndroid Build Coastguard Worker
98*08b48e0bSAndroid Build Coastguard Worker }
99*08b48e0bSAndroid Build Coastguard Worker
100*08b48e0bSAndroid Build Coastguard Worker /* LSAN does not support abort_on_error=1. (is this still true??) */
101*08b48e0bSAndroid Build Coastguard Worker u8 should_detect_leaks = 0;
102*08b48e0bSAndroid Build Coastguard Worker
103*08b48e0bSAndroid Build Coastguard Worker if (!have_lsan_options) {
104*08b48e0bSAndroid Build Coastguard Worker
105*08b48e0bSAndroid Build Coastguard Worker u8 buf[2048] = "";
106*08b48e0bSAndroid Build Coastguard Worker if (!have_san_options) { strcpy(buf, default_options); }
107*08b48e0bSAndroid Build Coastguard Worker if (have_asan_options) {
108*08b48e0bSAndroid Build Coastguard Worker
109*08b48e0bSAndroid Build Coastguard Worker if (NULL != strstr(have_asan_options, "detect_leaks=0")) {
110*08b48e0bSAndroid Build Coastguard Worker
111*08b48e0bSAndroid Build Coastguard Worker strcat(buf, "exitcode=" STRINGIFY(LSAN_ERROR) ":fast_unwind_on_malloc=0:print_suppressions=0:detect_leaks=0:malloc_context_size=0:");
112*08b48e0bSAndroid Build Coastguard Worker
113*08b48e0bSAndroid Build Coastguard Worker } else {
114*08b48e0bSAndroid Build Coastguard Worker
115*08b48e0bSAndroid Build Coastguard Worker should_detect_leaks = 1;
116*08b48e0bSAndroid Build Coastguard Worker strcat(buf, "exitcode=" STRINGIFY(LSAN_ERROR) ":fast_unwind_on_malloc=0:print_suppressions=0:detect_leaks=1:malloc_context_size=30:");
117*08b48e0bSAndroid Build Coastguard Worker
118*08b48e0bSAndroid Build Coastguard Worker }
119*08b48e0bSAndroid Build Coastguard Worker
120*08b48e0bSAndroid Build Coastguard Worker }
121*08b48e0bSAndroid Build Coastguard Worker
122*08b48e0bSAndroid Build Coastguard Worker setenv("LSAN_OPTIONS", buf, 1);
123*08b48e0bSAndroid Build Coastguard Worker
124*08b48e0bSAndroid Build Coastguard Worker }
125*08b48e0bSAndroid Build Coastguard Worker
126*08b48e0bSAndroid Build Coastguard Worker /* for everything not LSAN we disable detect_leaks */
127*08b48e0bSAndroid Build Coastguard Worker
128*08b48e0bSAndroid Build Coastguard Worker if (!have_lsan_options) {
129*08b48e0bSAndroid Build Coastguard Worker
130*08b48e0bSAndroid Build Coastguard Worker if (should_detect_leaks) {
131*08b48e0bSAndroid Build Coastguard Worker
132*08b48e0bSAndroid Build Coastguard Worker strcat(default_options, "detect_leaks=1:malloc_context_size=30:");
133*08b48e0bSAndroid Build Coastguard Worker
134*08b48e0bSAndroid Build Coastguard Worker } else {
135*08b48e0bSAndroid Build Coastguard Worker
136*08b48e0bSAndroid Build Coastguard Worker strcat(default_options, "detect_leaks=0:malloc_context_size=0:");
137*08b48e0bSAndroid Build Coastguard Worker
138*08b48e0bSAndroid Build Coastguard Worker }
139*08b48e0bSAndroid Build Coastguard Worker
140*08b48e0bSAndroid Build Coastguard Worker }
141*08b48e0bSAndroid Build Coastguard Worker
142*08b48e0bSAndroid Build Coastguard Worker /* Set sane defaults for ASAN if nothing else is specified. */
143*08b48e0bSAndroid Build Coastguard Worker
144*08b48e0bSAndroid Build Coastguard Worker if (!have_san_options) { setenv("ASAN_OPTIONS", default_options, 1); }
145*08b48e0bSAndroid Build Coastguard Worker
146*08b48e0bSAndroid Build Coastguard Worker /* Set sane defaults for UBSAN if nothing else is specified. */
147*08b48e0bSAndroid Build Coastguard Worker
148*08b48e0bSAndroid Build Coastguard Worker if (!have_san_options) { setenv("UBSAN_OPTIONS", default_options, 1); }
149*08b48e0bSAndroid Build Coastguard Worker
150*08b48e0bSAndroid Build Coastguard Worker /* MSAN is tricky, because it doesn't support abort_on_error=1 at this
151*08b48e0bSAndroid Build Coastguard Worker point. So, we do this in a very hacky way. */
152*08b48e0bSAndroid Build Coastguard Worker
153*08b48e0bSAndroid Build Coastguard Worker if (!have_msan_options) {
154*08b48e0bSAndroid Build Coastguard Worker
155*08b48e0bSAndroid Build Coastguard Worker u8 buf[2048] = "";
156*08b48e0bSAndroid Build Coastguard Worker if (!have_san_options) { strcpy(buf, default_options); }
157*08b48e0bSAndroid Build Coastguard Worker strcat(buf, "exit_code=" STRINGIFY(MSAN_ERROR) ":msan_track_origins=0:");
158*08b48e0bSAndroid Build Coastguard Worker setenv("MSAN_OPTIONS", buf, 1);
159*08b48e0bSAndroid Build Coastguard Worker
160*08b48e0bSAndroid Build Coastguard Worker }
161*08b48e0bSAndroid Build Coastguard Worker
162*08b48e0bSAndroid Build Coastguard Worker /* Envs for QASan */
163*08b48e0bSAndroid Build Coastguard Worker setenv("QASAN_MAX_CALL_STACK", "0", 0);
164*08b48e0bSAndroid Build Coastguard Worker setenv("QASAN_SYMBOLIZE", "0", 0);
165*08b48e0bSAndroid Build Coastguard Worker
166*08b48e0bSAndroid Build Coastguard Worker }
167*08b48e0bSAndroid Build Coastguard Worker
check_binary_signatures(u8 * fn)168*08b48e0bSAndroid Build Coastguard Worker u32 check_binary_signatures(u8 *fn) {
169*08b48e0bSAndroid Build Coastguard Worker
170*08b48e0bSAndroid Build Coastguard Worker int ret = 0, fd = open(fn, O_RDONLY);
171*08b48e0bSAndroid Build Coastguard Worker if (fd < 0) { PFATAL("Unable to open '%s'", fn); }
172*08b48e0bSAndroid Build Coastguard Worker struct stat st;
173*08b48e0bSAndroid Build Coastguard Worker if (fstat(fd, &st) < 0) { PFATAL("Unable to fstat '%s'", fn); }
174*08b48e0bSAndroid Build Coastguard Worker u32 f_len = st.st_size;
175*08b48e0bSAndroid Build Coastguard Worker u8 *f_data = mmap(0, f_len, PROT_READ, MAP_PRIVATE, fd, 0);
176*08b48e0bSAndroid Build Coastguard Worker if (f_data == MAP_FAILED) { PFATAL("Unable to mmap file '%s'", fn); }
177*08b48e0bSAndroid Build Coastguard Worker close(fd);
178*08b48e0bSAndroid Build Coastguard Worker
179*08b48e0bSAndroid Build Coastguard Worker if (afl_memmem(f_data, f_len, PERSIST_SIG, strlen(PERSIST_SIG) + 1)) {
180*08b48e0bSAndroid Build Coastguard Worker
181*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) { OKF(cPIN "Persistent mode binary detected."); }
182*08b48e0bSAndroid Build Coastguard Worker setenv(PERSIST_ENV_VAR, "1", 1);
183*08b48e0bSAndroid Build Coastguard Worker ret = 1;
184*08b48e0bSAndroid Build Coastguard Worker
185*08b48e0bSAndroid Build Coastguard Worker } else if (getenv("AFL_PERSISTENT")) {
186*08b48e0bSAndroid Build Coastguard Worker
187*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) { OKF(cPIN "Persistent mode enforced."); }
188*08b48e0bSAndroid Build Coastguard Worker setenv(PERSIST_ENV_VAR, "1", 1);
189*08b48e0bSAndroid Build Coastguard Worker ret = 1;
190*08b48e0bSAndroid Build Coastguard Worker
191*08b48e0bSAndroid Build Coastguard Worker } else if (getenv("AFL_FRIDA_PERSISTENT_ADDR")) {
192*08b48e0bSAndroid Build Coastguard Worker
193*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) {
194*08b48e0bSAndroid Build Coastguard Worker
195*08b48e0bSAndroid Build Coastguard Worker OKF("FRIDA Persistent mode configuration options detected.");
196*08b48e0bSAndroid Build Coastguard Worker
197*08b48e0bSAndroid Build Coastguard Worker }
198*08b48e0bSAndroid Build Coastguard Worker
199*08b48e0bSAndroid Build Coastguard Worker setenv(PERSIST_ENV_VAR, "1", 1);
200*08b48e0bSAndroid Build Coastguard Worker ret = 1;
201*08b48e0bSAndroid Build Coastguard Worker
202*08b48e0bSAndroid Build Coastguard Worker }
203*08b48e0bSAndroid Build Coastguard Worker
204*08b48e0bSAndroid Build Coastguard Worker if (afl_memmem(f_data, f_len, DEFER_SIG, strlen(DEFER_SIG) + 1)) {
205*08b48e0bSAndroid Build Coastguard Worker
206*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) { OKF(cPIN "Deferred forkserver binary detected."); }
207*08b48e0bSAndroid Build Coastguard Worker setenv(DEFER_ENV_VAR, "1", 1);
208*08b48e0bSAndroid Build Coastguard Worker ret += 2;
209*08b48e0bSAndroid Build Coastguard Worker
210*08b48e0bSAndroid Build Coastguard Worker } else if (getenv("AFL_DEFER_FORKSRV")) {
211*08b48e0bSAndroid Build Coastguard Worker
212*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) { OKF(cPIN "Deferred forkserver enforced."); }
213*08b48e0bSAndroid Build Coastguard Worker setenv(DEFER_ENV_VAR, "1", 1);
214*08b48e0bSAndroid Build Coastguard Worker ret += 2;
215*08b48e0bSAndroid Build Coastguard Worker
216*08b48e0bSAndroid Build Coastguard Worker }
217*08b48e0bSAndroid Build Coastguard Worker
218*08b48e0bSAndroid Build Coastguard Worker if (munmap(f_data, f_len)) { PFATAL("unmap() failed"); }
219*08b48e0bSAndroid Build Coastguard Worker
220*08b48e0bSAndroid Build Coastguard Worker return ret;
221*08b48e0bSAndroid Build Coastguard Worker
222*08b48e0bSAndroid Build Coastguard Worker }
223*08b48e0bSAndroid Build Coastguard Worker
detect_file_args(char ** argv,u8 * prog_in,bool * use_stdin)224*08b48e0bSAndroid Build Coastguard Worker void detect_file_args(char **argv, u8 *prog_in, bool *use_stdin) {
225*08b48e0bSAndroid Build Coastguard Worker
226*08b48e0bSAndroid Build Coastguard Worker u32 i = 0;
227*08b48e0bSAndroid Build Coastguard Worker u8 cwd[PATH_MAX];
228*08b48e0bSAndroid Build Coastguard Worker if (getcwd(cwd, (size_t)sizeof(cwd)) == NULL) { PFATAL("getcwd() failed"); }
229*08b48e0bSAndroid Build Coastguard Worker
230*08b48e0bSAndroid Build Coastguard Worker /* we are working with libc-heap-allocated argvs. So do not mix them with
231*08b48e0bSAndroid Build Coastguard Worker * other allocation APIs like ck_alloc. That would disturb the free() calls.
232*08b48e0bSAndroid Build Coastguard Worker */
233*08b48e0bSAndroid Build Coastguard Worker while (argv[i]) {
234*08b48e0bSAndroid Build Coastguard Worker
235*08b48e0bSAndroid Build Coastguard Worker u8 *aa_loc = strstr(argv[i], "@@");
236*08b48e0bSAndroid Build Coastguard Worker
237*08b48e0bSAndroid Build Coastguard Worker if (aa_loc) {
238*08b48e0bSAndroid Build Coastguard Worker
239*08b48e0bSAndroid Build Coastguard Worker if (!prog_in) { FATAL("@@ syntax is not supported by this tool."); }
240*08b48e0bSAndroid Build Coastguard Worker
241*08b48e0bSAndroid Build Coastguard Worker *use_stdin = false;
242*08b48e0bSAndroid Build Coastguard Worker
243*08b48e0bSAndroid Build Coastguard Worker /* Be sure that we're always using fully-qualified paths. */
244*08b48e0bSAndroid Build Coastguard Worker
245*08b48e0bSAndroid Build Coastguard Worker *aa_loc = 0;
246*08b48e0bSAndroid Build Coastguard Worker
247*08b48e0bSAndroid Build Coastguard Worker /* Construct a replacement argv value. */
248*08b48e0bSAndroid Build Coastguard Worker u8 *n_arg;
249*08b48e0bSAndroid Build Coastguard Worker
250*08b48e0bSAndroid Build Coastguard Worker if (prog_in[0] == '/') {
251*08b48e0bSAndroid Build Coastguard Worker
252*08b48e0bSAndroid Build Coastguard Worker n_arg = alloc_printf("%s%s%s", argv[i], prog_in, aa_loc + 2);
253*08b48e0bSAndroid Build Coastguard Worker
254*08b48e0bSAndroid Build Coastguard Worker } else {
255*08b48e0bSAndroid Build Coastguard Worker
256*08b48e0bSAndroid Build Coastguard Worker n_arg = alloc_printf("%s%s/%s%s", argv[i], cwd, prog_in, aa_loc + 2);
257*08b48e0bSAndroid Build Coastguard Worker
258*08b48e0bSAndroid Build Coastguard Worker }
259*08b48e0bSAndroid Build Coastguard Worker
260*08b48e0bSAndroid Build Coastguard Worker ck_free(argv[i]);
261*08b48e0bSAndroid Build Coastguard Worker argv[i] = n_arg;
262*08b48e0bSAndroid Build Coastguard Worker
263*08b48e0bSAndroid Build Coastguard Worker }
264*08b48e0bSAndroid Build Coastguard Worker
265*08b48e0bSAndroid Build Coastguard Worker i++;
266*08b48e0bSAndroid Build Coastguard Worker
267*08b48e0bSAndroid Build Coastguard Worker }
268*08b48e0bSAndroid Build Coastguard Worker
269*08b48e0bSAndroid Build Coastguard Worker /* argvs are automatically freed at exit. */
270*08b48e0bSAndroid Build Coastguard Worker
271*08b48e0bSAndroid Build Coastguard Worker }
272*08b48e0bSAndroid Build Coastguard Worker
273*08b48e0bSAndroid Build Coastguard Worker /* duplicate the system argv so that
274*08b48e0bSAndroid Build Coastguard Worker we can edit (and free!) it later */
275*08b48e0bSAndroid Build Coastguard Worker
argv_cpy_dup(int argc,char ** argv)276*08b48e0bSAndroid Build Coastguard Worker char **argv_cpy_dup(int argc, char **argv) {
277*08b48e0bSAndroid Build Coastguard Worker
278*08b48e0bSAndroid Build Coastguard Worker int i = 0;
279*08b48e0bSAndroid Build Coastguard Worker
280*08b48e0bSAndroid Build Coastguard Worker char **ret = ck_alloc((argc + 1) * sizeof(char *));
281*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!ret)) { FATAL("Amount of arguments specified is too high"); }
282*08b48e0bSAndroid Build Coastguard Worker
283*08b48e0bSAndroid Build Coastguard Worker for (i = 0; i < argc; i++) {
284*08b48e0bSAndroid Build Coastguard Worker
285*08b48e0bSAndroid Build Coastguard Worker ret[i] = ck_strdup(argv[i]);
286*08b48e0bSAndroid Build Coastguard Worker
287*08b48e0bSAndroid Build Coastguard Worker }
288*08b48e0bSAndroid Build Coastguard Worker
289*08b48e0bSAndroid Build Coastguard Worker ret[i] = NULL;
290*08b48e0bSAndroid Build Coastguard Worker
291*08b48e0bSAndroid Build Coastguard Worker return ret;
292*08b48e0bSAndroid Build Coastguard Worker
293*08b48e0bSAndroid Build Coastguard Worker }
294*08b48e0bSAndroid Build Coastguard Worker
295*08b48e0bSAndroid Build Coastguard Worker /* frees all args in the given argv,
296*08b48e0bSAndroid Build Coastguard Worker previously created by argv_cpy_dup */
297*08b48e0bSAndroid Build Coastguard Worker
argv_cpy_free(char ** argv)298*08b48e0bSAndroid Build Coastguard Worker void argv_cpy_free(char **argv) {
299*08b48e0bSAndroid Build Coastguard Worker
300*08b48e0bSAndroid Build Coastguard Worker u32 i = 0;
301*08b48e0bSAndroid Build Coastguard Worker while (argv[i]) {
302*08b48e0bSAndroid Build Coastguard Worker
303*08b48e0bSAndroid Build Coastguard Worker ck_free(argv[i]);
304*08b48e0bSAndroid Build Coastguard Worker argv[i] = NULL;
305*08b48e0bSAndroid Build Coastguard Worker i++;
306*08b48e0bSAndroid Build Coastguard Worker
307*08b48e0bSAndroid Build Coastguard Worker }
308*08b48e0bSAndroid Build Coastguard Worker
309*08b48e0bSAndroid Build Coastguard Worker ck_free(argv);
310*08b48e0bSAndroid Build Coastguard Worker
311*08b48e0bSAndroid Build Coastguard Worker }
312*08b48e0bSAndroid Build Coastguard Worker
313*08b48e0bSAndroid Build Coastguard Worker /* Rewrite argv for CoreSight process tracer. */
314*08b48e0bSAndroid Build Coastguard Worker
get_cs_argv(u8 * own_loc,u8 ** target_path_p,int argc,char ** argv)315*08b48e0bSAndroid Build Coastguard Worker char **get_cs_argv(u8 *own_loc, u8 **target_path_p, int argc, char **argv) {
316*08b48e0bSAndroid Build Coastguard Worker
317*08b48e0bSAndroid Build Coastguard Worker if (unlikely(getenv("AFL_CS_CUSTOM_BIN"))) {
318*08b48e0bSAndroid Build Coastguard Worker
319*08b48e0bSAndroid Build Coastguard Worker WARNF(
320*08b48e0bSAndroid Build Coastguard Worker "AFL_CS_CUSTOM_BIN is enabled. "
321*08b48e0bSAndroid Build Coastguard Worker "You must run your target under afl-cs-proxy on your own!");
322*08b48e0bSAndroid Build Coastguard Worker return argv;
323*08b48e0bSAndroid Build Coastguard Worker
324*08b48e0bSAndroid Build Coastguard Worker }
325*08b48e0bSAndroid Build Coastguard Worker
326*08b48e0bSAndroid Build Coastguard Worker char **new_argv = ck_alloc(sizeof(char *) * (argc + 4));
327*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!new_argv)) { FATAL("Illegal amount of arguments specified"); }
328*08b48e0bSAndroid Build Coastguard Worker
329*08b48e0bSAndroid Build Coastguard Worker memcpy(&new_argv[3], &argv[1], (int)(sizeof(char *)) * (argc - 1));
330*08b48e0bSAndroid Build Coastguard Worker new_argv[argc + 3] = NULL;
331*08b48e0bSAndroid Build Coastguard Worker
332*08b48e0bSAndroid Build Coastguard Worker new_argv[2] = *target_path_p;
333*08b48e0bSAndroid Build Coastguard Worker new_argv[1] = "--";
334*08b48e0bSAndroid Build Coastguard Worker
335*08b48e0bSAndroid Build Coastguard Worker /* Now we need to actually find the cs-proxy binary to put in argv[0]. */
336*08b48e0bSAndroid Build Coastguard Worker
337*08b48e0bSAndroid Build Coastguard Worker *target_path_p = new_argv[0] = find_afl_binary(own_loc, "afl-cs-proxy");
338*08b48e0bSAndroid Build Coastguard Worker return new_argv;
339*08b48e0bSAndroid Build Coastguard Worker
340*08b48e0bSAndroid Build Coastguard Worker }
341*08b48e0bSAndroid Build Coastguard Worker
342*08b48e0bSAndroid Build Coastguard Worker /* Rewrite argv for QEMU. */
343*08b48e0bSAndroid Build Coastguard Worker
get_qemu_argv(u8 * own_loc,u8 ** target_path_p,int argc,char ** argv)344*08b48e0bSAndroid Build Coastguard Worker char **get_qemu_argv(u8 *own_loc, u8 **target_path_p, int argc, char **argv) {
345*08b48e0bSAndroid Build Coastguard Worker
346*08b48e0bSAndroid Build Coastguard Worker if (unlikely(getenv("AFL_QEMU_CUSTOM_BIN"))) {
347*08b48e0bSAndroid Build Coastguard Worker
348*08b48e0bSAndroid Build Coastguard Worker WARNF(
349*08b48e0bSAndroid Build Coastguard Worker "AFL_QEMU_CUSTOM_BIN is enabled. "
350*08b48e0bSAndroid Build Coastguard Worker "You must run your target under afl-qemu-trace on your own!");
351*08b48e0bSAndroid Build Coastguard Worker return argv;
352*08b48e0bSAndroid Build Coastguard Worker
353*08b48e0bSAndroid Build Coastguard Worker }
354*08b48e0bSAndroid Build Coastguard Worker
355*08b48e0bSAndroid Build Coastguard Worker char **new_argv = ck_alloc(sizeof(char *) * (argc + 3));
356*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!new_argv)) { FATAL("Illegal amount of arguments specified"); }
357*08b48e0bSAndroid Build Coastguard Worker
358*08b48e0bSAndroid Build Coastguard Worker memcpy(&new_argv[3], &argv[1], (int)(sizeof(char *)) * (argc - 1));
359*08b48e0bSAndroid Build Coastguard Worker
360*08b48e0bSAndroid Build Coastguard Worker new_argv[2] = *target_path_p;
361*08b48e0bSAndroid Build Coastguard Worker new_argv[1] = "--";
362*08b48e0bSAndroid Build Coastguard Worker
363*08b48e0bSAndroid Build Coastguard Worker /* Now we need to actually find the QEMU binary to put in argv[0]. */
364*08b48e0bSAndroid Build Coastguard Worker
365*08b48e0bSAndroid Build Coastguard Worker *target_path_p = new_argv[0] = find_afl_binary(own_loc, "afl-qemu-trace");
366*08b48e0bSAndroid Build Coastguard Worker return new_argv;
367*08b48e0bSAndroid Build Coastguard Worker
368*08b48e0bSAndroid Build Coastguard Worker }
369*08b48e0bSAndroid Build Coastguard Worker
370*08b48e0bSAndroid Build Coastguard Worker /* Rewrite argv for Wine+QEMU. */
371*08b48e0bSAndroid Build Coastguard Worker
get_wine_argv(u8 * own_loc,u8 ** target_path_p,int argc,char ** argv)372*08b48e0bSAndroid Build Coastguard Worker char **get_wine_argv(u8 *own_loc, u8 **target_path_p, int argc, char **argv) {
373*08b48e0bSAndroid Build Coastguard Worker
374*08b48e0bSAndroid Build Coastguard Worker char **new_argv = ck_alloc(sizeof(char *) * (argc + 2));
375*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!new_argv)) { FATAL("Illegal amount of arguments specified"); }
376*08b48e0bSAndroid Build Coastguard Worker
377*08b48e0bSAndroid Build Coastguard Worker memcpy(&new_argv[2], &argv[1], (int)(sizeof(char *)) * (argc - 1));
378*08b48e0bSAndroid Build Coastguard Worker
379*08b48e0bSAndroid Build Coastguard Worker new_argv[1] = *target_path_p;
380*08b48e0bSAndroid Build Coastguard Worker
381*08b48e0bSAndroid Build Coastguard Worker /* Now we need to actually find the QEMU binary to put in argv[0]. */
382*08b48e0bSAndroid Build Coastguard Worker
383*08b48e0bSAndroid Build Coastguard Worker u8 *tmp = find_afl_binary(own_loc, "afl-qemu-trace");
384*08b48e0bSAndroid Build Coastguard Worker ck_free(tmp);
385*08b48e0bSAndroid Build Coastguard Worker *target_path_p = new_argv[0] = find_afl_binary(own_loc, "afl-wine-trace");
386*08b48e0bSAndroid Build Coastguard Worker return new_argv;
387*08b48e0bSAndroid Build Coastguard Worker
388*08b48e0bSAndroid Build Coastguard Worker }
389*08b48e0bSAndroid Build Coastguard Worker
390*08b48e0bSAndroid Build Coastguard Worker /* Find binary, used by analyze, showmap, tmin
391*08b48e0bSAndroid Build Coastguard Worker @returns the path, allocating the string */
392*08b48e0bSAndroid Build Coastguard Worker
find_binary(u8 * fname)393*08b48e0bSAndroid Build Coastguard Worker u8 *find_binary(u8 *fname) {
394*08b48e0bSAndroid Build Coastguard Worker
395*08b48e0bSAndroid Build Coastguard Worker // TODO: Merge this function with check_binary of afl-fuzz-init.c
396*08b48e0bSAndroid Build Coastguard Worker
397*08b48e0bSAndroid Build Coastguard Worker u8 *env_path = NULL;
398*08b48e0bSAndroid Build Coastguard Worker u8 *target_path = NULL;
399*08b48e0bSAndroid Build Coastguard Worker
400*08b48e0bSAndroid Build Coastguard Worker struct stat st;
401*08b48e0bSAndroid Build Coastguard Worker
402*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!fname)) { FATAL("No binary supplied"); }
403*08b48e0bSAndroid Build Coastguard Worker
404*08b48e0bSAndroid Build Coastguard Worker if (strchr(fname, '/') || !(env_path = getenv("PATH"))) {
405*08b48e0bSAndroid Build Coastguard Worker
406*08b48e0bSAndroid Build Coastguard Worker target_path = ck_strdup(fname);
407*08b48e0bSAndroid Build Coastguard Worker
408*08b48e0bSAndroid Build Coastguard Worker if (stat(target_path, &st) || !S_ISREG(st.st_mode) ||
409*08b48e0bSAndroid Build Coastguard Worker !(st.st_mode & 0111) || st.st_size < 4) {
410*08b48e0bSAndroid Build Coastguard Worker
411*08b48e0bSAndroid Build Coastguard Worker ck_free(target_path);
412*08b48e0bSAndroid Build Coastguard Worker FATAL("Program '%s' not found or not executable", fname);
413*08b48e0bSAndroid Build Coastguard Worker
414*08b48e0bSAndroid Build Coastguard Worker }
415*08b48e0bSAndroid Build Coastguard Worker
416*08b48e0bSAndroid Build Coastguard Worker } else {
417*08b48e0bSAndroid Build Coastguard Worker
418*08b48e0bSAndroid Build Coastguard Worker while (env_path) {
419*08b48e0bSAndroid Build Coastguard Worker
420*08b48e0bSAndroid Build Coastguard Worker u8 *cur_elem, *delim = strchr(env_path, ':');
421*08b48e0bSAndroid Build Coastguard Worker
422*08b48e0bSAndroid Build Coastguard Worker if (delim) {
423*08b48e0bSAndroid Build Coastguard Worker
424*08b48e0bSAndroid Build Coastguard Worker cur_elem = ck_alloc(delim - env_path + 1);
425*08b48e0bSAndroid Build Coastguard Worker if (unlikely(!cur_elem)) {
426*08b48e0bSAndroid Build Coastguard Worker
427*08b48e0bSAndroid Build Coastguard Worker FATAL(
428*08b48e0bSAndroid Build Coastguard Worker "Unexpected overflow when processing ENV. This should never "
429*08b48e0bSAndroid Build Coastguard Worker "had happened.");
430*08b48e0bSAndroid Build Coastguard Worker
431*08b48e0bSAndroid Build Coastguard Worker }
432*08b48e0bSAndroid Build Coastguard Worker
433*08b48e0bSAndroid Build Coastguard Worker memcpy(cur_elem, env_path, delim - env_path);
434*08b48e0bSAndroid Build Coastguard Worker delim++;
435*08b48e0bSAndroid Build Coastguard Worker
436*08b48e0bSAndroid Build Coastguard Worker } else {
437*08b48e0bSAndroid Build Coastguard Worker
438*08b48e0bSAndroid Build Coastguard Worker cur_elem = ck_strdup(env_path);
439*08b48e0bSAndroid Build Coastguard Worker
440*08b48e0bSAndroid Build Coastguard Worker }
441*08b48e0bSAndroid Build Coastguard Worker
442*08b48e0bSAndroid Build Coastguard Worker env_path = delim;
443*08b48e0bSAndroid Build Coastguard Worker
444*08b48e0bSAndroid Build Coastguard Worker if (cur_elem[0]) {
445*08b48e0bSAndroid Build Coastguard Worker
446*08b48e0bSAndroid Build Coastguard Worker target_path = alloc_printf("%s/%s", cur_elem, fname);
447*08b48e0bSAndroid Build Coastguard Worker
448*08b48e0bSAndroid Build Coastguard Worker } else {
449*08b48e0bSAndroid Build Coastguard Worker
450*08b48e0bSAndroid Build Coastguard Worker target_path = ck_strdup(fname);
451*08b48e0bSAndroid Build Coastguard Worker
452*08b48e0bSAndroid Build Coastguard Worker }
453*08b48e0bSAndroid Build Coastguard Worker
454*08b48e0bSAndroid Build Coastguard Worker ck_free(cur_elem);
455*08b48e0bSAndroid Build Coastguard Worker
456*08b48e0bSAndroid Build Coastguard Worker if (!stat(target_path, &st) && S_ISREG(st.st_mode) &&
457*08b48e0bSAndroid Build Coastguard Worker (st.st_mode & 0111) && st.st_size >= 4) {
458*08b48e0bSAndroid Build Coastguard Worker
459*08b48e0bSAndroid Build Coastguard Worker break;
460*08b48e0bSAndroid Build Coastguard Worker
461*08b48e0bSAndroid Build Coastguard Worker }
462*08b48e0bSAndroid Build Coastguard Worker
463*08b48e0bSAndroid Build Coastguard Worker ck_free(target_path);
464*08b48e0bSAndroid Build Coastguard Worker target_path = NULL;
465*08b48e0bSAndroid Build Coastguard Worker
466*08b48e0bSAndroid Build Coastguard Worker }
467*08b48e0bSAndroid Build Coastguard Worker
468*08b48e0bSAndroid Build Coastguard Worker if (!target_path) {
469*08b48e0bSAndroid Build Coastguard Worker
470*08b48e0bSAndroid Build Coastguard Worker FATAL("Program '%s' not found or not executable", fname);
471*08b48e0bSAndroid Build Coastguard Worker
472*08b48e0bSAndroid Build Coastguard Worker }
473*08b48e0bSAndroid Build Coastguard Worker
474*08b48e0bSAndroid Build Coastguard Worker }
475*08b48e0bSAndroid Build Coastguard Worker
476*08b48e0bSAndroid Build Coastguard Worker return target_path;
477*08b48e0bSAndroid Build Coastguard Worker
478*08b48e0bSAndroid Build Coastguard Worker }
479*08b48e0bSAndroid Build Coastguard Worker
find_afl_binary(u8 * own_loc,u8 * fname)480*08b48e0bSAndroid Build Coastguard Worker u8 *find_afl_binary(u8 *own_loc, u8 *fname) {
481*08b48e0bSAndroid Build Coastguard Worker
482*08b48e0bSAndroid Build Coastguard Worker u8 *afl_path = NULL, *target_path, *own_copy, *tmp;
483*08b48e0bSAndroid Build Coastguard Worker int perm = X_OK;
484*08b48e0bSAndroid Build Coastguard Worker
485*08b48e0bSAndroid Build Coastguard Worker if ((tmp = strrchr(fname, '.'))) {
486*08b48e0bSAndroid Build Coastguard Worker
487*08b48e0bSAndroid Build Coastguard Worker if (!strcasecmp(tmp, ".so") || !strcasecmp(tmp, ".dylib")) { perm = R_OK; }
488*08b48e0bSAndroid Build Coastguard Worker
489*08b48e0bSAndroid Build Coastguard Worker }
490*08b48e0bSAndroid Build Coastguard Worker
491*08b48e0bSAndroid Build Coastguard Worker if ((afl_path = getenv("AFL_PATH"))) {
492*08b48e0bSAndroid Build Coastguard Worker
493*08b48e0bSAndroid Build Coastguard Worker target_path = alloc_printf("%s/%s", afl_path, fname);
494*08b48e0bSAndroid Build Coastguard Worker if (!access(target_path, perm)) {
495*08b48e0bSAndroid Build Coastguard Worker
496*08b48e0bSAndroid Build Coastguard Worker return target_path;
497*08b48e0bSAndroid Build Coastguard Worker
498*08b48e0bSAndroid Build Coastguard Worker } else {
499*08b48e0bSAndroid Build Coastguard Worker
500*08b48e0bSAndroid Build Coastguard Worker ck_free(target_path);
501*08b48e0bSAndroid Build Coastguard Worker
502*08b48e0bSAndroid Build Coastguard Worker }
503*08b48e0bSAndroid Build Coastguard Worker
504*08b48e0bSAndroid Build Coastguard Worker }
505*08b48e0bSAndroid Build Coastguard Worker
506*08b48e0bSAndroid Build Coastguard Worker if (own_loc) {
507*08b48e0bSAndroid Build Coastguard Worker
508*08b48e0bSAndroid Build Coastguard Worker own_copy = ck_strdup(own_loc);
509*08b48e0bSAndroid Build Coastguard Worker u8 *rsl = strrchr(own_copy, '/');
510*08b48e0bSAndroid Build Coastguard Worker
511*08b48e0bSAndroid Build Coastguard Worker if (rsl) {
512*08b48e0bSAndroid Build Coastguard Worker
513*08b48e0bSAndroid Build Coastguard Worker *rsl = 0;
514*08b48e0bSAndroid Build Coastguard Worker
515*08b48e0bSAndroid Build Coastguard Worker target_path = alloc_printf("%s/%s", own_copy, fname);
516*08b48e0bSAndroid Build Coastguard Worker ck_free(own_copy);
517*08b48e0bSAndroid Build Coastguard Worker
518*08b48e0bSAndroid Build Coastguard Worker if (!access(target_path, perm)) {
519*08b48e0bSAndroid Build Coastguard Worker
520*08b48e0bSAndroid Build Coastguard Worker return target_path;
521*08b48e0bSAndroid Build Coastguard Worker
522*08b48e0bSAndroid Build Coastguard Worker } else {
523*08b48e0bSAndroid Build Coastguard Worker
524*08b48e0bSAndroid Build Coastguard Worker ck_free(target_path);
525*08b48e0bSAndroid Build Coastguard Worker
526*08b48e0bSAndroid Build Coastguard Worker }
527*08b48e0bSAndroid Build Coastguard Worker
528*08b48e0bSAndroid Build Coastguard Worker } else {
529*08b48e0bSAndroid Build Coastguard Worker
530*08b48e0bSAndroid Build Coastguard Worker ck_free(own_copy);
531*08b48e0bSAndroid Build Coastguard Worker
532*08b48e0bSAndroid Build Coastguard Worker }
533*08b48e0bSAndroid Build Coastguard Worker
534*08b48e0bSAndroid Build Coastguard Worker }
535*08b48e0bSAndroid Build Coastguard Worker
536*08b48e0bSAndroid Build Coastguard Worker if (perm == X_OK) {
537*08b48e0bSAndroid Build Coastguard Worker
538*08b48e0bSAndroid Build Coastguard Worker target_path = alloc_printf("%s/%s", BIN_PATH, fname);
539*08b48e0bSAndroid Build Coastguard Worker
540*08b48e0bSAndroid Build Coastguard Worker } else {
541*08b48e0bSAndroid Build Coastguard Worker
542*08b48e0bSAndroid Build Coastguard Worker target_path = alloc_printf("%s/%s", AFL_PATH, fname);
543*08b48e0bSAndroid Build Coastguard Worker
544*08b48e0bSAndroid Build Coastguard Worker }
545*08b48e0bSAndroid Build Coastguard Worker
546*08b48e0bSAndroid Build Coastguard Worker if (!access(target_path, perm)) {
547*08b48e0bSAndroid Build Coastguard Worker
548*08b48e0bSAndroid Build Coastguard Worker return target_path;
549*08b48e0bSAndroid Build Coastguard Worker
550*08b48e0bSAndroid Build Coastguard Worker } else {
551*08b48e0bSAndroid Build Coastguard Worker
552*08b48e0bSAndroid Build Coastguard Worker ck_free(target_path);
553*08b48e0bSAndroid Build Coastguard Worker
554*08b48e0bSAndroid Build Coastguard Worker }
555*08b48e0bSAndroid Build Coastguard Worker
556*08b48e0bSAndroid Build Coastguard Worker if (perm == X_OK) {
557*08b48e0bSAndroid Build Coastguard Worker
558*08b48e0bSAndroid Build Coastguard Worker return find_binary(fname);
559*08b48e0bSAndroid Build Coastguard Worker
560*08b48e0bSAndroid Build Coastguard Worker } else {
561*08b48e0bSAndroid Build Coastguard Worker
562*08b48e0bSAndroid Build Coastguard Worker FATAL("Library '%s' not found", fname);
563*08b48e0bSAndroid Build Coastguard Worker
564*08b48e0bSAndroid Build Coastguard Worker }
565*08b48e0bSAndroid Build Coastguard Worker
566*08b48e0bSAndroid Build Coastguard Worker }
567*08b48e0bSAndroid Build Coastguard Worker
parse_afl_kill_signal(u8 * numeric_signal_as_str,int default_signal)568*08b48e0bSAndroid Build Coastguard Worker int parse_afl_kill_signal(u8 *numeric_signal_as_str, int default_signal) {
569*08b48e0bSAndroid Build Coastguard Worker
570*08b48e0bSAndroid Build Coastguard Worker if (numeric_signal_as_str && numeric_signal_as_str[0]) {
571*08b48e0bSAndroid Build Coastguard Worker
572*08b48e0bSAndroid Build Coastguard Worker char *endptr;
573*08b48e0bSAndroid Build Coastguard Worker u8 signal_code;
574*08b48e0bSAndroid Build Coastguard Worker signal_code = (u8)strtoul(numeric_signal_as_str, &endptr, 10);
575*08b48e0bSAndroid Build Coastguard Worker /* Did we manage to parse the full string? */
576*08b48e0bSAndroid Build Coastguard Worker if (*endptr != '\0' || endptr == (char *)numeric_signal_as_str) {
577*08b48e0bSAndroid Build Coastguard Worker
578*08b48e0bSAndroid Build Coastguard Worker FATAL("Invalid signal name: %s", numeric_signal_as_str);
579*08b48e0bSAndroid Build Coastguard Worker
580*08b48e0bSAndroid Build Coastguard Worker } else {
581*08b48e0bSAndroid Build Coastguard Worker
582*08b48e0bSAndroid Build Coastguard Worker return signal_code;
583*08b48e0bSAndroid Build Coastguard Worker
584*08b48e0bSAndroid Build Coastguard Worker }
585*08b48e0bSAndroid Build Coastguard Worker
586*08b48e0bSAndroid Build Coastguard Worker }
587*08b48e0bSAndroid Build Coastguard Worker
588*08b48e0bSAndroid Build Coastguard Worker return default_signal;
589*08b48e0bSAndroid Build Coastguard Worker
590*08b48e0bSAndroid Build Coastguard Worker }
591*08b48e0bSAndroid Build Coastguard Worker
configure_afl_kill_signals(afl_forkserver_t * fsrv,char * afl_kill_signal_env,char * afl_fsrv_kill_signal_env,int default_server_kill_signal)592*08b48e0bSAndroid Build Coastguard Worker void configure_afl_kill_signals(afl_forkserver_t *fsrv,
593*08b48e0bSAndroid Build Coastguard Worker char *afl_kill_signal_env,
594*08b48e0bSAndroid Build Coastguard Worker char *afl_fsrv_kill_signal_env,
595*08b48e0bSAndroid Build Coastguard Worker int default_server_kill_signal) {
596*08b48e0bSAndroid Build Coastguard Worker
597*08b48e0bSAndroid Build Coastguard Worker afl_kill_signal_env =
598*08b48e0bSAndroid Build Coastguard Worker afl_kill_signal_env ? afl_kill_signal_env : getenv("AFL_KILL_SIGNAL");
599*08b48e0bSAndroid Build Coastguard Worker afl_fsrv_kill_signal_env = afl_fsrv_kill_signal_env
600*08b48e0bSAndroid Build Coastguard Worker ? afl_fsrv_kill_signal_env
601*08b48e0bSAndroid Build Coastguard Worker : getenv("AFL_FORK_SERVER_KILL_SIGNAL");
602*08b48e0bSAndroid Build Coastguard Worker
603*08b48e0bSAndroid Build Coastguard Worker fsrv->child_kill_signal = parse_afl_kill_signal(afl_kill_signal_env, SIGKILL);
604*08b48e0bSAndroid Build Coastguard Worker
605*08b48e0bSAndroid Build Coastguard Worker if (afl_kill_signal_env && !afl_fsrv_kill_signal_env) {
606*08b48e0bSAndroid Build Coastguard Worker
607*08b48e0bSAndroid Build Coastguard Worker /*
608*08b48e0bSAndroid Build Coastguard Worker Set AFL_FORK_SERVER_KILL_SIGNAL to the value of AFL_KILL_SIGNAL for
609*08b48e0bSAndroid Build Coastguard Worker backwards compatibility. However, if AFL_FORK_SERVER_KILL_SIGNAL is set, is
610*08b48e0bSAndroid Build Coastguard Worker takes precedence.
611*08b48e0bSAndroid Build Coastguard Worker */
612*08b48e0bSAndroid Build Coastguard Worker afl_fsrv_kill_signal_env = afl_kill_signal_env;
613*08b48e0bSAndroid Build Coastguard Worker
614*08b48e0bSAndroid Build Coastguard Worker }
615*08b48e0bSAndroid Build Coastguard Worker
616*08b48e0bSAndroid Build Coastguard Worker fsrv->fsrv_kill_signal = parse_afl_kill_signal(afl_fsrv_kill_signal_env,
617*08b48e0bSAndroid Build Coastguard Worker default_server_kill_signal);
618*08b48e0bSAndroid Build Coastguard Worker
619*08b48e0bSAndroid Build Coastguard Worker }
620*08b48e0bSAndroid Build Coastguard Worker
helper_min3(unsigned int a,unsigned int b,unsigned int c)621*08b48e0bSAndroid Build Coastguard Worker static inline unsigned int helper_min3(unsigned int a, unsigned int b,
622*08b48e0bSAndroid Build Coastguard Worker unsigned int c) {
623*08b48e0bSAndroid Build Coastguard Worker
624*08b48e0bSAndroid Build Coastguard Worker return a < b ? (a < c ? a : c) : (b < c ? b : c);
625*08b48e0bSAndroid Build Coastguard Worker
626*08b48e0bSAndroid Build Coastguard Worker }
627*08b48e0bSAndroid Build Coastguard Worker
628*08b48e0bSAndroid Build Coastguard Worker // from
629*08b48e0bSAndroid Build Coastguard Worker // https://en.wikibooks.org/wiki/Algorithm_Implementation/Strings/Levenshtein_distance#C
string_distance_levenshtein(char * s1,char * s2)630*08b48e0bSAndroid Build Coastguard Worker static int string_distance_levenshtein(char *s1, char *s2) {
631*08b48e0bSAndroid Build Coastguard Worker
632*08b48e0bSAndroid Build Coastguard Worker unsigned int s1len, s2len, x, y, lastdiag, olddiag;
633*08b48e0bSAndroid Build Coastguard Worker s1len = strlen(s1);
634*08b48e0bSAndroid Build Coastguard Worker s2len = strlen(s2);
635*08b48e0bSAndroid Build Coastguard Worker unsigned int column[s1len + 1];
636*08b48e0bSAndroid Build Coastguard Worker column[s1len] = 1;
637*08b48e0bSAndroid Build Coastguard Worker
638*08b48e0bSAndroid Build Coastguard Worker for (y = 1; y <= s1len; y++)
639*08b48e0bSAndroid Build Coastguard Worker column[y] = y;
640*08b48e0bSAndroid Build Coastguard Worker for (x = 1; x <= s2len; x++) {
641*08b48e0bSAndroid Build Coastguard Worker
642*08b48e0bSAndroid Build Coastguard Worker column[0] = x;
643*08b48e0bSAndroid Build Coastguard Worker for (y = 1, lastdiag = x - 1; y <= s1len; y++) {
644*08b48e0bSAndroid Build Coastguard Worker
645*08b48e0bSAndroid Build Coastguard Worker olddiag = column[y];
646*08b48e0bSAndroid Build Coastguard Worker column[y] = helper_min3(column[y] + 1, column[y - 1] + 1,
647*08b48e0bSAndroid Build Coastguard Worker lastdiag + (s1[y - 1] == s2[x - 1] ? 0 : 1));
648*08b48e0bSAndroid Build Coastguard Worker lastdiag = olddiag;
649*08b48e0bSAndroid Build Coastguard Worker
650*08b48e0bSAndroid Build Coastguard Worker }
651*08b48e0bSAndroid Build Coastguard Worker
652*08b48e0bSAndroid Build Coastguard Worker }
653*08b48e0bSAndroid Build Coastguard Worker
654*08b48e0bSAndroid Build Coastguard Worker return column[s1len];
655*08b48e0bSAndroid Build Coastguard Worker
656*08b48e0bSAndroid Build Coastguard Worker }
657*08b48e0bSAndroid Build Coastguard Worker
658*08b48e0bSAndroid Build Coastguard Worker #define ENV_SIMILARITY_TRESHOLD 3
659*08b48e0bSAndroid Build Coastguard Worker
print_suggested_envs(char * mispelled_env)660*08b48e0bSAndroid Build Coastguard Worker void print_suggested_envs(char *mispelled_env) {
661*08b48e0bSAndroid Build Coastguard Worker
662*08b48e0bSAndroid Build Coastguard Worker size_t env_name_len =
663*08b48e0bSAndroid Build Coastguard Worker strcspn(mispelled_env, "=") - 4; // remove the AFL_prefix
664*08b48e0bSAndroid Build Coastguard Worker char *env_name = ck_alloc(env_name_len + 1);
665*08b48e0bSAndroid Build Coastguard Worker memcpy(env_name, mispelled_env + 4, env_name_len);
666*08b48e0bSAndroid Build Coastguard Worker
667*08b48e0bSAndroid Build Coastguard Worker char *seen = ck_alloc(sizeof(afl_environment_variables) / sizeof(char *));
668*08b48e0bSAndroid Build Coastguard Worker int found = 0;
669*08b48e0bSAndroid Build Coastguard Worker
670*08b48e0bSAndroid Build Coastguard Worker int j;
671*08b48e0bSAndroid Build Coastguard Worker for (j = 0; afl_environment_variables[j] != NULL; ++j) {
672*08b48e0bSAndroid Build Coastguard Worker
673*08b48e0bSAndroid Build Coastguard Worker char *afl_env = afl_environment_variables[j] + 4;
674*08b48e0bSAndroid Build Coastguard Worker int distance = string_distance_levenshtein(afl_env, env_name);
675*08b48e0bSAndroid Build Coastguard Worker if (distance < ENV_SIMILARITY_TRESHOLD && seen[j] == 0) {
676*08b48e0bSAndroid Build Coastguard Worker
677*08b48e0bSAndroid Build Coastguard Worker SAYF("Did you mean %s?\n", afl_environment_variables[j]);
678*08b48e0bSAndroid Build Coastguard Worker seen[j] = 1;
679*08b48e0bSAndroid Build Coastguard Worker found = 1;
680*08b48e0bSAndroid Build Coastguard Worker
681*08b48e0bSAndroid Build Coastguard Worker }
682*08b48e0bSAndroid Build Coastguard Worker
683*08b48e0bSAndroid Build Coastguard Worker }
684*08b48e0bSAndroid Build Coastguard Worker
685*08b48e0bSAndroid Build Coastguard Worker if (found) goto cleanup;
686*08b48e0bSAndroid Build Coastguard Worker
687*08b48e0bSAndroid Build Coastguard Worker for (j = 0; afl_environment_variables[j] != NULL; ++j) {
688*08b48e0bSAndroid Build Coastguard Worker
689*08b48e0bSAndroid Build Coastguard Worker char *afl_env = afl_environment_variables[j] + 4;
690*08b48e0bSAndroid Build Coastguard Worker size_t afl_env_len = strlen(afl_env);
691*08b48e0bSAndroid Build Coastguard Worker char *reduced = ck_alloc(afl_env_len + 1);
692*08b48e0bSAndroid Build Coastguard Worker
693*08b48e0bSAndroid Build Coastguard Worker size_t start = 0;
694*08b48e0bSAndroid Build Coastguard Worker while (start < afl_env_len) {
695*08b48e0bSAndroid Build Coastguard Worker
696*08b48e0bSAndroid Build Coastguard Worker size_t end = start + strcspn(afl_env + start, "_") + 1;
697*08b48e0bSAndroid Build Coastguard Worker memcpy(reduced, afl_env, start);
698*08b48e0bSAndroid Build Coastguard Worker if (end < afl_env_len) {
699*08b48e0bSAndroid Build Coastguard Worker
700*08b48e0bSAndroid Build Coastguard Worker memcpy(reduced + start, afl_env + end, afl_env_len - end);
701*08b48e0bSAndroid Build Coastguard Worker
702*08b48e0bSAndroid Build Coastguard Worker }
703*08b48e0bSAndroid Build Coastguard Worker
704*08b48e0bSAndroid Build Coastguard Worker if (afl_env_len + start >= end) {
705*08b48e0bSAndroid Build Coastguard Worker
706*08b48e0bSAndroid Build Coastguard Worker reduced[afl_env_len - end + start] = 0;
707*08b48e0bSAndroid Build Coastguard Worker
708*08b48e0bSAndroid Build Coastguard Worker }
709*08b48e0bSAndroid Build Coastguard Worker
710*08b48e0bSAndroid Build Coastguard Worker int distance = string_distance_levenshtein(reduced, env_name);
711*08b48e0bSAndroid Build Coastguard Worker if (distance < ENV_SIMILARITY_TRESHOLD && seen[j] == 0) {
712*08b48e0bSAndroid Build Coastguard Worker
713*08b48e0bSAndroid Build Coastguard Worker SAYF("Did you mean %s?\n", afl_environment_variables[j]);
714*08b48e0bSAndroid Build Coastguard Worker seen[j] = 1;
715*08b48e0bSAndroid Build Coastguard Worker found = 1;
716*08b48e0bSAndroid Build Coastguard Worker
717*08b48e0bSAndroid Build Coastguard Worker }
718*08b48e0bSAndroid Build Coastguard Worker
719*08b48e0bSAndroid Build Coastguard Worker start = end;
720*08b48e0bSAndroid Build Coastguard Worker
721*08b48e0bSAndroid Build Coastguard Worker };
722*08b48e0bSAndroid Build Coastguard Worker
723*08b48e0bSAndroid Build Coastguard Worker ck_free(reduced);
724*08b48e0bSAndroid Build Coastguard Worker
725*08b48e0bSAndroid Build Coastguard Worker }
726*08b48e0bSAndroid Build Coastguard Worker
727*08b48e0bSAndroid Build Coastguard Worker if (found) goto cleanup;
728*08b48e0bSAndroid Build Coastguard Worker
729*08b48e0bSAndroid Build Coastguard Worker char *reduced = ck_alloc(env_name_len + 1);
730*08b48e0bSAndroid Build Coastguard Worker size_t start = 0;
731*08b48e0bSAndroid Build Coastguard Worker while (start < env_name_len) {
732*08b48e0bSAndroid Build Coastguard Worker
733*08b48e0bSAndroid Build Coastguard Worker size_t end = start + strcspn(env_name + start, "_") + 1;
734*08b48e0bSAndroid Build Coastguard Worker memcpy(reduced, env_name, start);
735*08b48e0bSAndroid Build Coastguard Worker if (end < env_name_len)
736*08b48e0bSAndroid Build Coastguard Worker memcpy(reduced + start, env_name + end, env_name_len - end);
737*08b48e0bSAndroid Build Coastguard Worker reduced[env_name_len - end + start] = 0;
738*08b48e0bSAndroid Build Coastguard Worker
739*08b48e0bSAndroid Build Coastguard Worker for (j = 0; afl_environment_variables[j] != NULL; ++j) {
740*08b48e0bSAndroid Build Coastguard Worker
741*08b48e0bSAndroid Build Coastguard Worker int distance = string_distance_levenshtein(
742*08b48e0bSAndroid Build Coastguard Worker afl_environment_variables[j] + 4, reduced);
743*08b48e0bSAndroid Build Coastguard Worker if (distance < ENV_SIMILARITY_TRESHOLD && seen[j] == 0) {
744*08b48e0bSAndroid Build Coastguard Worker
745*08b48e0bSAndroid Build Coastguard Worker SAYF("Did you mean %s?\n", afl_environment_variables[j]);
746*08b48e0bSAndroid Build Coastguard Worker seen[j] = 1;
747*08b48e0bSAndroid Build Coastguard Worker
748*08b48e0bSAndroid Build Coastguard Worker }
749*08b48e0bSAndroid Build Coastguard Worker
750*08b48e0bSAndroid Build Coastguard Worker }
751*08b48e0bSAndroid Build Coastguard Worker
752*08b48e0bSAndroid Build Coastguard Worker start = end;
753*08b48e0bSAndroid Build Coastguard Worker
754*08b48e0bSAndroid Build Coastguard Worker };
755*08b48e0bSAndroid Build Coastguard Worker
756*08b48e0bSAndroid Build Coastguard Worker ck_free(reduced);
757*08b48e0bSAndroid Build Coastguard Worker
758*08b48e0bSAndroid Build Coastguard Worker cleanup:
759*08b48e0bSAndroid Build Coastguard Worker ck_free(env_name);
760*08b48e0bSAndroid Build Coastguard Worker ck_free(seen);
761*08b48e0bSAndroid Build Coastguard Worker
762*08b48e0bSAndroid Build Coastguard Worker }
763*08b48e0bSAndroid Build Coastguard Worker
check_environment_vars(char ** envp)764*08b48e0bSAndroid Build Coastguard Worker void check_environment_vars(char **envp) {
765*08b48e0bSAndroid Build Coastguard Worker
766*08b48e0bSAndroid Build Coastguard Worker if (be_quiet) { return; }
767*08b48e0bSAndroid Build Coastguard Worker
768*08b48e0bSAndroid Build Coastguard Worker int index = 0, issue_detected = 0;
769*08b48e0bSAndroid Build Coastguard Worker char *env, *val, *ignore = getenv("AFL_IGNORE_UNKNOWN_ENVS");
770*08b48e0bSAndroid Build Coastguard Worker while ((env = envp[index++]) != NULL) {
771*08b48e0bSAndroid Build Coastguard Worker
772*08b48e0bSAndroid Build Coastguard Worker if (strncmp(env, "ALF_", 4) == 0 || strncmp(env, "_ALF", 4) == 0 ||
773*08b48e0bSAndroid Build Coastguard Worker strncmp(env, "__ALF", 5) == 0 || strncmp(env, "_AFL", 4) == 0 ||
774*08b48e0bSAndroid Build Coastguard Worker strncmp(env, "__AFL", 5) == 0) {
775*08b48e0bSAndroid Build Coastguard Worker
776*08b48e0bSAndroid Build Coastguard Worker WARNF("Potentially mistyped AFL environment variable: %s", env);
777*08b48e0bSAndroid Build Coastguard Worker issue_detected = 1;
778*08b48e0bSAndroid Build Coastguard Worker
779*08b48e0bSAndroid Build Coastguard Worker } else if (strncmp(env, "AFL_", 4) == 0) {
780*08b48e0bSAndroid Build Coastguard Worker
781*08b48e0bSAndroid Build Coastguard Worker int i = 0, match = 0;
782*08b48e0bSAndroid Build Coastguard Worker while (match == 0 && afl_environment_variables[i] != NULL) {
783*08b48e0bSAndroid Build Coastguard Worker
784*08b48e0bSAndroid Build Coastguard Worker if (strncmp(env, afl_environment_variables[i],
785*08b48e0bSAndroid Build Coastguard Worker strlen(afl_environment_variables[i])) == 0 &&
786*08b48e0bSAndroid Build Coastguard Worker env[strlen(afl_environment_variables[i])] == '=') {
787*08b48e0bSAndroid Build Coastguard Worker
788*08b48e0bSAndroid Build Coastguard Worker match = 1;
789*08b48e0bSAndroid Build Coastguard Worker
790*08b48e0bSAndroid Build Coastguard Worker if ((val = getenv(afl_environment_variables[i])) && !*val) {
791*08b48e0bSAndroid Build Coastguard Worker
792*08b48e0bSAndroid Build Coastguard Worker WARNF(
793*08b48e0bSAndroid Build Coastguard Worker "AFL environment variable %s defined but is empty, this can "
794*08b48e0bSAndroid Build Coastguard Worker "lead to unexpected consequences",
795*08b48e0bSAndroid Build Coastguard Worker afl_environment_variables[i]);
796*08b48e0bSAndroid Build Coastguard Worker issue_detected = 1;
797*08b48e0bSAndroid Build Coastguard Worker
798*08b48e0bSAndroid Build Coastguard Worker }
799*08b48e0bSAndroid Build Coastguard Worker
800*08b48e0bSAndroid Build Coastguard Worker } else {
801*08b48e0bSAndroid Build Coastguard Worker
802*08b48e0bSAndroid Build Coastguard Worker i++;
803*08b48e0bSAndroid Build Coastguard Worker
804*08b48e0bSAndroid Build Coastguard Worker }
805*08b48e0bSAndroid Build Coastguard Worker
806*08b48e0bSAndroid Build Coastguard Worker }
807*08b48e0bSAndroid Build Coastguard Worker
808*08b48e0bSAndroid Build Coastguard Worker i = 0;
809*08b48e0bSAndroid Build Coastguard Worker while (match == 0 && afl_environment_deprecated[i] != NULL) {
810*08b48e0bSAndroid Build Coastguard Worker
811*08b48e0bSAndroid Build Coastguard Worker if (strncmp(env, afl_environment_deprecated[i],
812*08b48e0bSAndroid Build Coastguard Worker strlen(afl_environment_deprecated[i])) == 0 &&
813*08b48e0bSAndroid Build Coastguard Worker env[strlen(afl_environment_deprecated[i])] == '=') {
814*08b48e0bSAndroid Build Coastguard Worker
815*08b48e0bSAndroid Build Coastguard Worker match = 1;
816*08b48e0bSAndroid Build Coastguard Worker
817*08b48e0bSAndroid Build Coastguard Worker WARNF("AFL environment variable %s is deprecated!",
818*08b48e0bSAndroid Build Coastguard Worker afl_environment_deprecated[i]);
819*08b48e0bSAndroid Build Coastguard Worker issue_detected = 1;
820*08b48e0bSAndroid Build Coastguard Worker
821*08b48e0bSAndroid Build Coastguard Worker } else {
822*08b48e0bSAndroid Build Coastguard Worker
823*08b48e0bSAndroid Build Coastguard Worker i++;
824*08b48e0bSAndroid Build Coastguard Worker
825*08b48e0bSAndroid Build Coastguard Worker }
826*08b48e0bSAndroid Build Coastguard Worker
827*08b48e0bSAndroid Build Coastguard Worker }
828*08b48e0bSAndroid Build Coastguard Worker
829*08b48e0bSAndroid Build Coastguard Worker if (match == 0 && !ignore) {
830*08b48e0bSAndroid Build Coastguard Worker
831*08b48e0bSAndroid Build Coastguard Worker WARNF("Mistyped AFL environment variable: %s", env);
832*08b48e0bSAndroid Build Coastguard Worker issue_detected = 1;
833*08b48e0bSAndroid Build Coastguard Worker
834*08b48e0bSAndroid Build Coastguard Worker print_suggested_envs(env);
835*08b48e0bSAndroid Build Coastguard Worker
836*08b48e0bSAndroid Build Coastguard Worker }
837*08b48e0bSAndroid Build Coastguard Worker
838*08b48e0bSAndroid Build Coastguard Worker }
839*08b48e0bSAndroid Build Coastguard Worker
840*08b48e0bSAndroid Build Coastguard Worker }
841*08b48e0bSAndroid Build Coastguard Worker
842*08b48e0bSAndroid Build Coastguard Worker if (issue_detected) { sleep(2); }
843*08b48e0bSAndroid Build Coastguard Worker
844*08b48e0bSAndroid Build Coastguard Worker }
845*08b48e0bSAndroid Build Coastguard Worker
get_afl_env(char * env)846*08b48e0bSAndroid Build Coastguard Worker char *get_afl_env(char *env) {
847*08b48e0bSAndroid Build Coastguard Worker
848*08b48e0bSAndroid Build Coastguard Worker char *val;
849*08b48e0bSAndroid Build Coastguard Worker
850*08b48e0bSAndroid Build Coastguard Worker if ((val = getenv(env))) {
851*08b48e0bSAndroid Build Coastguard Worker
852*08b48e0bSAndroid Build Coastguard Worker if (*val) {
853*08b48e0bSAndroid Build Coastguard Worker
854*08b48e0bSAndroid Build Coastguard Worker if (!be_quiet) {
855*08b48e0bSAndroid Build Coastguard Worker
856*08b48e0bSAndroid Build Coastguard Worker OKF("Enabled environment variable %s with value %s", env, val);
857*08b48e0bSAndroid Build Coastguard Worker
858*08b48e0bSAndroid Build Coastguard Worker }
859*08b48e0bSAndroid Build Coastguard Worker
860*08b48e0bSAndroid Build Coastguard Worker return val;
861*08b48e0bSAndroid Build Coastguard Worker
862*08b48e0bSAndroid Build Coastguard Worker }
863*08b48e0bSAndroid Build Coastguard Worker
864*08b48e0bSAndroid Build Coastguard Worker }
865*08b48e0bSAndroid Build Coastguard Worker
866*08b48e0bSAndroid Build Coastguard Worker return NULL;
867*08b48e0bSAndroid Build Coastguard Worker
868*08b48e0bSAndroid Build Coastguard Worker }
869*08b48e0bSAndroid Build Coastguard Worker
extract_and_set_env(u8 * env_str)870*08b48e0bSAndroid Build Coastguard Worker bool extract_and_set_env(u8 *env_str) {
871*08b48e0bSAndroid Build Coastguard Worker
872*08b48e0bSAndroid Build Coastguard Worker if (!env_str) { return false; }
873*08b48e0bSAndroid Build Coastguard Worker
874*08b48e0bSAndroid Build Coastguard Worker bool ret = false; // return false by default
875*08b48e0bSAndroid Build Coastguard Worker
876*08b48e0bSAndroid Build Coastguard Worker u8 *p = ck_strdup(env_str);
877*08b48e0bSAndroid Build Coastguard Worker u8 *end = p + strlen((char *)p);
878*08b48e0bSAndroid Build Coastguard Worker u8 *rest = p;
879*08b48e0bSAndroid Build Coastguard Worker
880*08b48e0bSAndroid Build Coastguard Worker u8 closing_sym = ' ';
881*08b48e0bSAndroid Build Coastguard Worker u8 c;
882*08b48e0bSAndroid Build Coastguard Worker
883*08b48e0bSAndroid Build Coastguard Worker size_t num_pairs = 0;
884*08b48e0bSAndroid Build Coastguard Worker
885*08b48e0bSAndroid Build Coastguard Worker while (rest < end) {
886*08b48e0bSAndroid Build Coastguard Worker
887*08b48e0bSAndroid Build Coastguard Worker while (*rest == ' ') {
888*08b48e0bSAndroid Build Coastguard Worker
889*08b48e0bSAndroid Build Coastguard Worker rest++;
890*08b48e0bSAndroid Build Coastguard Worker
891*08b48e0bSAndroid Build Coastguard Worker }
892*08b48e0bSAndroid Build Coastguard Worker
893*08b48e0bSAndroid Build Coastguard Worker if (rest + 1 >= end) break;
894*08b48e0bSAndroid Build Coastguard Worker
895*08b48e0bSAndroid Build Coastguard Worker u8 *key = rest;
896*08b48e0bSAndroid Build Coastguard Worker // env variable names may not start with numbers or '='
897*08b48e0bSAndroid Build Coastguard Worker if (*key == '=' || (*key >= '0' && *key <= '9')) { goto free_and_return; }
898*08b48e0bSAndroid Build Coastguard Worker
899*08b48e0bSAndroid Build Coastguard Worker while (rest < end && *rest != '=' && *rest != ' ') {
900*08b48e0bSAndroid Build Coastguard Worker
901*08b48e0bSAndroid Build Coastguard Worker c = *rest;
902*08b48e0bSAndroid Build Coastguard Worker // lowercase is bad but we may still allow it
903*08b48e0bSAndroid Build Coastguard Worker if ((c < 'A' || c > 'Z') && (c < 'a' || c > 'z') &&
904*08b48e0bSAndroid Build Coastguard Worker (c < '0' || c > '9') && c != '_') {
905*08b48e0bSAndroid Build Coastguard Worker
906*08b48e0bSAndroid Build Coastguard Worker goto free_and_return;
907*08b48e0bSAndroid Build Coastguard Worker
908*08b48e0bSAndroid Build Coastguard Worker }
909*08b48e0bSAndroid Build Coastguard Worker
910*08b48e0bSAndroid Build Coastguard Worker rest++;
911*08b48e0bSAndroid Build Coastguard Worker
912*08b48e0bSAndroid Build Coastguard Worker }
913*08b48e0bSAndroid Build Coastguard Worker
914*08b48e0bSAndroid Build Coastguard Worker if (*rest != '=') { goto free_and_return; }
915*08b48e0bSAndroid Build Coastguard Worker
916*08b48e0bSAndroid Build Coastguard Worker *rest = '\0'; // done with variable name
917*08b48e0bSAndroid Build Coastguard Worker
918*08b48e0bSAndroid Build Coastguard Worker rest += 1;
919*08b48e0bSAndroid Build Coastguard Worker if (rest >= end || *rest == ' ') { goto free_and_return; }
920*08b48e0bSAndroid Build Coastguard Worker
921*08b48e0bSAndroid Build Coastguard Worker u8 *val = rest;
922*08b48e0bSAndroid Build Coastguard Worker if (*val == '\'' || *val == '"') {
923*08b48e0bSAndroid Build Coastguard Worker
924*08b48e0bSAndroid Build Coastguard Worker closing_sym = *val;
925*08b48e0bSAndroid Build Coastguard Worker val += 1;
926*08b48e0bSAndroid Build Coastguard Worker rest += 1;
927*08b48e0bSAndroid Build Coastguard Worker if (rest >= end) { goto free_and_return; }
928*08b48e0bSAndroid Build Coastguard Worker
929*08b48e0bSAndroid Build Coastguard Worker } else {
930*08b48e0bSAndroid Build Coastguard Worker
931*08b48e0bSAndroid Build Coastguard Worker closing_sym = ' ';
932*08b48e0bSAndroid Build Coastguard Worker
933*08b48e0bSAndroid Build Coastguard Worker }
934*08b48e0bSAndroid Build Coastguard Worker
935*08b48e0bSAndroid Build Coastguard Worker while (rest < end && *rest != closing_sym) {
936*08b48e0bSAndroid Build Coastguard Worker
937*08b48e0bSAndroid Build Coastguard Worker rest++;
938*08b48e0bSAndroid Build Coastguard Worker
939*08b48e0bSAndroid Build Coastguard Worker }
940*08b48e0bSAndroid Build Coastguard Worker
941*08b48e0bSAndroid Build Coastguard Worker if (closing_sym != ' ' && *rest != closing_sym) { goto free_and_return; }
942*08b48e0bSAndroid Build Coastguard Worker
943*08b48e0bSAndroid Build Coastguard Worker *rest = '\0'; // done with variable value
944*08b48e0bSAndroid Build Coastguard Worker
945*08b48e0bSAndroid Build Coastguard Worker rest += 1;
946*08b48e0bSAndroid Build Coastguard Worker num_pairs++;
947*08b48e0bSAndroid Build Coastguard Worker setenv(key, val, 1);
948*08b48e0bSAndroid Build Coastguard Worker
949*08b48e0bSAndroid Build Coastguard Worker }
950*08b48e0bSAndroid Build Coastguard Worker
951*08b48e0bSAndroid Build Coastguard Worker if (num_pairs) { ret = true; }
952*08b48e0bSAndroid Build Coastguard Worker
953*08b48e0bSAndroid Build Coastguard Worker free_and_return:
954*08b48e0bSAndroid Build Coastguard Worker ck_free(p);
955*08b48e0bSAndroid Build Coastguard Worker return ret;
956*08b48e0bSAndroid Build Coastguard Worker
957*08b48e0bSAndroid Build Coastguard Worker }
958*08b48e0bSAndroid Build Coastguard Worker
959*08b48e0bSAndroid Build Coastguard Worker /* Read mask bitmap from file. This is for the -B option. */
960*08b48e0bSAndroid Build Coastguard Worker
read_bitmap(u8 * fname,u8 * map,size_t len)961*08b48e0bSAndroid Build Coastguard Worker void read_bitmap(u8 *fname, u8 *map, size_t len) {
962*08b48e0bSAndroid Build Coastguard Worker
963*08b48e0bSAndroid Build Coastguard Worker s32 fd = open(fname, O_RDONLY);
964*08b48e0bSAndroid Build Coastguard Worker
965*08b48e0bSAndroid Build Coastguard Worker if (fd < 0) { PFATAL("Unable to open '%s'", fname); }
966*08b48e0bSAndroid Build Coastguard Worker
967*08b48e0bSAndroid Build Coastguard Worker ck_read(fd, map, len, fname);
968*08b48e0bSAndroid Build Coastguard Worker
969*08b48e0bSAndroid Build Coastguard Worker close(fd);
970*08b48e0bSAndroid Build Coastguard Worker
971*08b48e0bSAndroid Build Coastguard Worker }
972*08b48e0bSAndroid Build Coastguard Worker
973*08b48e0bSAndroid Build Coastguard Worker /* Get unix time in milliseconds */
974*08b48e0bSAndroid Build Coastguard Worker
get_cur_time(void)975*08b48e0bSAndroid Build Coastguard Worker inline u64 get_cur_time(void) {
976*08b48e0bSAndroid Build Coastguard Worker
977*08b48e0bSAndroid Build Coastguard Worker struct timeval tv;
978*08b48e0bSAndroid Build Coastguard Worker struct timezone tz;
979*08b48e0bSAndroid Build Coastguard Worker
980*08b48e0bSAndroid Build Coastguard Worker gettimeofday(&tv, &tz);
981*08b48e0bSAndroid Build Coastguard Worker
982*08b48e0bSAndroid Build Coastguard Worker return (tv.tv_sec * 1000ULL) + (tv.tv_usec / 1000);
983*08b48e0bSAndroid Build Coastguard Worker
984*08b48e0bSAndroid Build Coastguard Worker }
985*08b48e0bSAndroid Build Coastguard Worker
986*08b48e0bSAndroid Build Coastguard Worker /* Get unix time in microseconds */
987*08b48e0bSAndroid Build Coastguard Worker
get_cur_time_us(void)988*08b48e0bSAndroid Build Coastguard Worker u64 get_cur_time_us(void) {
989*08b48e0bSAndroid Build Coastguard Worker
990*08b48e0bSAndroid Build Coastguard Worker struct timeval tv;
991*08b48e0bSAndroid Build Coastguard Worker struct timezone tz;
992*08b48e0bSAndroid Build Coastguard Worker
993*08b48e0bSAndroid Build Coastguard Worker gettimeofday(&tv, &tz);
994*08b48e0bSAndroid Build Coastguard Worker
995*08b48e0bSAndroid Build Coastguard Worker return (tv.tv_sec * 1000000ULL) + tv.tv_usec;
996*08b48e0bSAndroid Build Coastguard Worker
997*08b48e0bSAndroid Build Coastguard Worker }
998*08b48e0bSAndroid Build Coastguard Worker
999*08b48e0bSAndroid Build Coastguard Worker /* Describe integer. The buf should be
1000*08b48e0bSAndroid Build Coastguard Worker at least 6 bytes to fit all ints we randomly see.
1001*08b48e0bSAndroid Build Coastguard Worker Will return buf for convenience. */
1002*08b48e0bSAndroid Build Coastguard Worker
stringify_int(u8 * buf,size_t len,u64 val)1003*08b48e0bSAndroid Build Coastguard Worker u8 *stringify_int(u8 *buf, size_t len, u64 val) {
1004*08b48e0bSAndroid Build Coastguard Worker \
1005*08b48e0bSAndroid Build Coastguard Worker #define CHK_FORMAT(_divisor, _limit_mult, _fmt, _cast) \
1006*08b48e0bSAndroid Build Coastguard Worker do { \
1007*08b48e0bSAndroid Build Coastguard Worker \
1008*08b48e0bSAndroid Build Coastguard Worker if (val < (_divisor) * (_limit_mult)) { \
1009*08b48e0bSAndroid Build Coastguard Worker \
1010*08b48e0bSAndroid Build Coastguard Worker snprintf(buf, len, _fmt, ((_cast)val) / (_divisor)); \
1011*08b48e0bSAndroid Build Coastguard Worker return buf; \
1012*08b48e0bSAndroid Build Coastguard Worker \
1013*08b48e0bSAndroid Build Coastguard Worker } \
1014*08b48e0bSAndroid Build Coastguard Worker \
1015*08b48e0bSAndroid Build Coastguard Worker } while (0)
1016*08b48e0bSAndroid Build Coastguard Worker
1017*08b48e0bSAndroid Build Coastguard Worker /* 0-9999 */
1018*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1, 10000, "%llu", u64);
1019*08b48e0bSAndroid Build Coastguard Worker
1020*08b48e0bSAndroid Build Coastguard Worker /* 10.0k - 99.9k */
1021*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000, 99.95, "%0.01fk", double);
1022*08b48e0bSAndroid Build Coastguard Worker
1023*08b48e0bSAndroid Build Coastguard Worker /* 100k - 999k */
1024*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000, 1000, "%lluk", u64);
1025*08b48e0bSAndroid Build Coastguard Worker
1026*08b48e0bSAndroid Build Coastguard Worker /* 1.00M - 9.99M */
1027*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 9.995, "%0.02fM", double);
1028*08b48e0bSAndroid Build Coastguard Worker
1029*08b48e0bSAndroid Build Coastguard Worker /* 10.0M - 99.9M */
1030*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 99.95, "%0.01fM", double);
1031*08b48e0bSAndroid Build Coastguard Worker
1032*08b48e0bSAndroid Build Coastguard Worker /* 100M - 999M */
1033*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 1000, "%lluM", u64);
1034*08b48e0bSAndroid Build Coastguard Worker
1035*08b48e0bSAndroid Build Coastguard Worker /* 1.00G - 9.99G */
1036*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 9.995, "%0.02fG", double);
1037*08b48e0bSAndroid Build Coastguard Worker
1038*08b48e0bSAndroid Build Coastguard Worker /* 10.0G - 99.9G */
1039*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 99.95, "%0.01fG", double);
1040*08b48e0bSAndroid Build Coastguard Worker
1041*08b48e0bSAndroid Build Coastguard Worker /* 100G - 999G */
1042*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 1000, "%lluG", u64);
1043*08b48e0bSAndroid Build Coastguard Worker
1044*08b48e0bSAndroid Build Coastguard Worker /* 1.00T - 9.99G */
1045*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000 * 1000, 9.995, "%0.02fT", double);
1046*08b48e0bSAndroid Build Coastguard Worker
1047*08b48e0bSAndroid Build Coastguard Worker /* 10.0T - 99.9T */
1048*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000 * 1000, 99.95, "%0.01fT", double);
1049*08b48e0bSAndroid Build Coastguard Worker
1050*08b48e0bSAndroid Build Coastguard Worker /* 100T+ */
1051*08b48e0bSAndroid Build Coastguard Worker strncpy(buf, "infty", len);
1052*08b48e0bSAndroid Build Coastguard Worker buf[len - 1] = '\0';
1053*08b48e0bSAndroid Build Coastguard Worker
1054*08b48e0bSAndroid Build Coastguard Worker return buf;
1055*08b48e0bSAndroid Build Coastguard Worker
1056*08b48e0bSAndroid Build Coastguard Worker }
1057*08b48e0bSAndroid Build Coastguard Worker
1058*08b48e0bSAndroid Build Coastguard Worker /* Describe float. Similar as int. */
1059*08b48e0bSAndroid Build Coastguard Worker
stringify_float(u8 * buf,size_t len,double val)1060*08b48e0bSAndroid Build Coastguard Worker u8 *stringify_float(u8 *buf, size_t len, double val) {
1061*08b48e0bSAndroid Build Coastguard Worker
1062*08b48e0bSAndroid Build Coastguard Worker if (val < 99.995) {
1063*08b48e0bSAndroid Build Coastguard Worker
1064*08b48e0bSAndroid Build Coastguard Worker snprintf(buf, len, "%0.02f", val);
1065*08b48e0bSAndroid Build Coastguard Worker
1066*08b48e0bSAndroid Build Coastguard Worker } else if (val < 999.95) {
1067*08b48e0bSAndroid Build Coastguard Worker
1068*08b48e0bSAndroid Build Coastguard Worker snprintf(buf, len, "%0.01f", val);
1069*08b48e0bSAndroid Build Coastguard Worker
1070*08b48e0bSAndroid Build Coastguard Worker } else if (unlikely(isnan(val) || isinf(val))) {
1071*08b48e0bSAndroid Build Coastguard Worker
1072*08b48e0bSAndroid Build Coastguard Worker strcpy(buf, "inf");
1073*08b48e0bSAndroid Build Coastguard Worker
1074*08b48e0bSAndroid Build Coastguard Worker } else {
1075*08b48e0bSAndroid Build Coastguard Worker
1076*08b48e0bSAndroid Build Coastguard Worker stringify_int(buf, len, (u64)val);
1077*08b48e0bSAndroid Build Coastguard Worker
1078*08b48e0bSAndroid Build Coastguard Worker }
1079*08b48e0bSAndroid Build Coastguard Worker
1080*08b48e0bSAndroid Build Coastguard Worker return buf;
1081*08b48e0bSAndroid Build Coastguard Worker
1082*08b48e0bSAndroid Build Coastguard Worker }
1083*08b48e0bSAndroid Build Coastguard Worker
1084*08b48e0bSAndroid Build Coastguard Worker /* Describe integer as memory size. */
1085*08b48e0bSAndroid Build Coastguard Worker
stringify_mem_size(u8 * buf,size_t len,u64 val)1086*08b48e0bSAndroid Build Coastguard Worker u8 *stringify_mem_size(u8 *buf, size_t len, u64 val) {
1087*08b48e0bSAndroid Build Coastguard Worker
1088*08b48e0bSAndroid Build Coastguard Worker /* 0-9999 */
1089*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1, 10000, "%llu B", u64);
1090*08b48e0bSAndroid Build Coastguard Worker
1091*08b48e0bSAndroid Build Coastguard Worker /* 10.0k - 99.9k */
1092*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024, 99.95, "%0.01f kB", double);
1093*08b48e0bSAndroid Build Coastguard Worker
1094*08b48e0bSAndroid Build Coastguard Worker /* 100k - 999k */
1095*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024, 1000, "%llu kB", u64);
1096*08b48e0bSAndroid Build Coastguard Worker
1097*08b48e0bSAndroid Build Coastguard Worker /* 1.00M - 9.99M */
1098*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 9.995, "%0.02f MB", double);
1099*08b48e0bSAndroid Build Coastguard Worker
1100*08b48e0bSAndroid Build Coastguard Worker /* 10.0M - 99.9M */
1101*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 99.95, "%0.01f MB", double);
1102*08b48e0bSAndroid Build Coastguard Worker
1103*08b48e0bSAndroid Build Coastguard Worker /* 100M - 999M */
1104*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 1000, "%llu MB", u64);
1105*08b48e0bSAndroid Build Coastguard Worker
1106*08b48e0bSAndroid Build Coastguard Worker /* 1.00G - 9.99G */
1107*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 9.995, "%0.02f GB", double);
1108*08b48e0bSAndroid Build Coastguard Worker
1109*08b48e0bSAndroid Build Coastguard Worker /* 10.0G - 99.9G */
1110*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 99.95, "%0.01f GB", double);
1111*08b48e0bSAndroid Build Coastguard Worker
1112*08b48e0bSAndroid Build Coastguard Worker /* 100G - 999G */
1113*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 1000, "%llu GB", u64);
1114*08b48e0bSAndroid Build Coastguard Worker
1115*08b48e0bSAndroid Build Coastguard Worker /* 1.00T - 9.99G */
1116*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024 * 1024, 9.995, "%0.02f TB", double);
1117*08b48e0bSAndroid Build Coastguard Worker
1118*08b48e0bSAndroid Build Coastguard Worker /* 10.0T - 99.9T */
1119*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024 * 1024, 99.95, "%0.01f TB", double);
1120*08b48e0bSAndroid Build Coastguard Worker
1121*08b48e0bSAndroid Build Coastguard Worker #undef CHK_FORMAT
1122*08b48e0bSAndroid Build Coastguard Worker
1123*08b48e0bSAndroid Build Coastguard Worker /* 100T+ */
1124*08b48e0bSAndroid Build Coastguard Worker strncpy(buf, "infty", len - 1);
1125*08b48e0bSAndroid Build Coastguard Worker buf[len - 1] = '\0';
1126*08b48e0bSAndroid Build Coastguard Worker
1127*08b48e0bSAndroid Build Coastguard Worker return buf;
1128*08b48e0bSAndroid Build Coastguard Worker
1129*08b48e0bSAndroid Build Coastguard Worker }
1130*08b48e0bSAndroid Build Coastguard Worker
1131*08b48e0bSAndroid Build Coastguard Worker /* Describe time delta as string.
1132*08b48e0bSAndroid Build Coastguard Worker Returns a pointer to buf for convenience. */
1133*08b48e0bSAndroid Build Coastguard Worker
stringify_time_diff(u8 * buf,size_t len,u64 cur_ms,u64 event_ms)1134*08b48e0bSAndroid Build Coastguard Worker u8 *stringify_time_diff(u8 *buf, size_t len, u64 cur_ms, u64 event_ms) {
1135*08b48e0bSAndroid Build Coastguard Worker
1136*08b48e0bSAndroid Build Coastguard Worker if (!event_ms) {
1137*08b48e0bSAndroid Build Coastguard Worker
1138*08b48e0bSAndroid Build Coastguard Worker snprintf(buf, len, "none seen yet");
1139*08b48e0bSAndroid Build Coastguard Worker
1140*08b48e0bSAndroid Build Coastguard Worker } else {
1141*08b48e0bSAndroid Build Coastguard Worker
1142*08b48e0bSAndroid Build Coastguard Worker u64 delta;
1143*08b48e0bSAndroid Build Coastguard Worker s32 t_d, t_h, t_m, t_s;
1144*08b48e0bSAndroid Build Coastguard Worker u8 val_buf[STRINGIFY_VAL_SIZE_MAX];
1145*08b48e0bSAndroid Build Coastguard Worker
1146*08b48e0bSAndroid Build Coastguard Worker delta = cur_ms - event_ms;
1147*08b48e0bSAndroid Build Coastguard Worker
1148*08b48e0bSAndroid Build Coastguard Worker t_d = delta / 1000 / 60 / 60 / 24;
1149*08b48e0bSAndroid Build Coastguard Worker t_h = (delta / 1000 / 60 / 60) % 24;
1150*08b48e0bSAndroid Build Coastguard Worker t_m = (delta / 1000 / 60) % 60;
1151*08b48e0bSAndroid Build Coastguard Worker t_s = (delta / 1000) % 60;
1152*08b48e0bSAndroid Build Coastguard Worker
1153*08b48e0bSAndroid Build Coastguard Worker stringify_int(val_buf, sizeof(val_buf), t_d);
1154*08b48e0bSAndroid Build Coastguard Worker snprintf(buf, len, "%s days, %d hrs, %d min, %d sec", val_buf, t_h, t_m,
1155*08b48e0bSAndroid Build Coastguard Worker t_s);
1156*08b48e0bSAndroid Build Coastguard Worker
1157*08b48e0bSAndroid Build Coastguard Worker }
1158*08b48e0bSAndroid Build Coastguard Worker
1159*08b48e0bSAndroid Build Coastguard Worker return buf;
1160*08b48e0bSAndroid Build Coastguard Worker
1161*08b48e0bSAndroid Build Coastguard Worker }
1162*08b48e0bSAndroid Build Coastguard Worker
1163*08b48e0bSAndroid Build Coastguard Worker /* Unsafe Describe integer. The buf sizes are not checked.
1164*08b48e0bSAndroid Build Coastguard Worker This is unsafe but fast.
1165*08b48e0bSAndroid Build Coastguard Worker Will return buf for convenience. */
1166*08b48e0bSAndroid Build Coastguard Worker
u_stringify_int(u8 * buf,u64 val)1167*08b48e0bSAndroid Build Coastguard Worker u8 *u_stringify_int(u8 *buf, u64 val) {
1168*08b48e0bSAndroid Build Coastguard Worker \
1169*08b48e0bSAndroid Build Coastguard Worker #define CHK_FORMAT(_divisor, _limit_mult, _fmt, _cast) \
1170*08b48e0bSAndroid Build Coastguard Worker do { \
1171*08b48e0bSAndroid Build Coastguard Worker \
1172*08b48e0bSAndroid Build Coastguard Worker if (val < (_divisor) * (_limit_mult)) { \
1173*08b48e0bSAndroid Build Coastguard Worker \
1174*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, _fmt, ((_cast)val) / (_divisor)); \
1175*08b48e0bSAndroid Build Coastguard Worker return buf; \
1176*08b48e0bSAndroid Build Coastguard Worker \
1177*08b48e0bSAndroid Build Coastguard Worker } \
1178*08b48e0bSAndroid Build Coastguard Worker \
1179*08b48e0bSAndroid Build Coastguard Worker } while (0)
1180*08b48e0bSAndroid Build Coastguard Worker
1181*08b48e0bSAndroid Build Coastguard Worker /* 0-9999 */
1182*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1, 10000, "%llu", u64);
1183*08b48e0bSAndroid Build Coastguard Worker
1184*08b48e0bSAndroid Build Coastguard Worker /* 10.0k - 99.9k */
1185*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000, 99.95, "%0.01fk", double);
1186*08b48e0bSAndroid Build Coastguard Worker
1187*08b48e0bSAndroid Build Coastguard Worker /* 100k - 999k */
1188*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000, 1000, "%lluk", u64);
1189*08b48e0bSAndroid Build Coastguard Worker
1190*08b48e0bSAndroid Build Coastguard Worker /* 1.00M - 9.99M */
1191*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 9.995, "%0.02fM", double);
1192*08b48e0bSAndroid Build Coastguard Worker
1193*08b48e0bSAndroid Build Coastguard Worker /* 10.0M - 99.9M */
1194*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 99.95, "%0.01fM", double);
1195*08b48e0bSAndroid Build Coastguard Worker
1196*08b48e0bSAndroid Build Coastguard Worker /* 100M - 999M */
1197*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000 * 1000, 1000, "%lluM", u64);
1198*08b48e0bSAndroid Build Coastguard Worker
1199*08b48e0bSAndroid Build Coastguard Worker /* 1.00G - 9.99G */
1200*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 9.995, "%0.02fG", double);
1201*08b48e0bSAndroid Build Coastguard Worker
1202*08b48e0bSAndroid Build Coastguard Worker /* 10.0G - 99.9G */
1203*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 99.95, "%0.01fG", double);
1204*08b48e0bSAndroid Build Coastguard Worker
1205*08b48e0bSAndroid Build Coastguard Worker /* 100G - 999G */
1206*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000, 1000, "%lluG", u64);
1207*08b48e0bSAndroid Build Coastguard Worker
1208*08b48e0bSAndroid Build Coastguard Worker /* 1.00T - 9.99G */
1209*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000 * 1000, 9.995, "%0.02fT", double);
1210*08b48e0bSAndroid Build Coastguard Worker
1211*08b48e0bSAndroid Build Coastguard Worker /* 10.0T - 99.9T */
1212*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1000LL * 1000 * 1000 * 1000, 99.95, "%0.01fT", double);
1213*08b48e0bSAndroid Build Coastguard Worker
1214*08b48e0bSAndroid Build Coastguard Worker /* 100T+ */
1215*08b48e0bSAndroid Build Coastguard Worker strcpy(buf, "infty");
1216*08b48e0bSAndroid Build Coastguard Worker
1217*08b48e0bSAndroid Build Coastguard Worker return buf;
1218*08b48e0bSAndroid Build Coastguard Worker
1219*08b48e0bSAndroid Build Coastguard Worker }
1220*08b48e0bSAndroid Build Coastguard Worker
1221*08b48e0bSAndroid Build Coastguard Worker /* Unsafe describe float. Similar as unsafe int. */
1222*08b48e0bSAndroid Build Coastguard Worker
u_stringify_float(u8 * buf,double val)1223*08b48e0bSAndroid Build Coastguard Worker u8 *u_stringify_float(u8 *buf, double val) {
1224*08b48e0bSAndroid Build Coastguard Worker
1225*08b48e0bSAndroid Build Coastguard Worker if (val < 99.995) {
1226*08b48e0bSAndroid Build Coastguard Worker
1227*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "%0.02f", val);
1228*08b48e0bSAndroid Build Coastguard Worker
1229*08b48e0bSAndroid Build Coastguard Worker } else if (val < 999.95) {
1230*08b48e0bSAndroid Build Coastguard Worker
1231*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "%0.01f", val);
1232*08b48e0bSAndroid Build Coastguard Worker
1233*08b48e0bSAndroid Build Coastguard Worker } else if (unlikely(isnan(val) || isinf(val))) {
1234*08b48e0bSAndroid Build Coastguard Worker
1235*08b48e0bSAndroid Build Coastguard Worker strcpy(buf, "infinite");
1236*08b48e0bSAndroid Build Coastguard Worker
1237*08b48e0bSAndroid Build Coastguard Worker } else {
1238*08b48e0bSAndroid Build Coastguard Worker
1239*08b48e0bSAndroid Build Coastguard Worker return u_stringify_int(buf, (u64)val);
1240*08b48e0bSAndroid Build Coastguard Worker
1241*08b48e0bSAndroid Build Coastguard Worker }
1242*08b48e0bSAndroid Build Coastguard Worker
1243*08b48e0bSAndroid Build Coastguard Worker return buf;
1244*08b48e0bSAndroid Build Coastguard Worker
1245*08b48e0bSAndroid Build Coastguard Worker }
1246*08b48e0bSAndroid Build Coastguard Worker
1247*08b48e0bSAndroid Build Coastguard Worker /* Unsafe describe integer as memory size. */
1248*08b48e0bSAndroid Build Coastguard Worker
u_stringify_mem_size(u8 * buf,u64 val)1249*08b48e0bSAndroid Build Coastguard Worker u8 *u_stringify_mem_size(u8 *buf, u64 val) {
1250*08b48e0bSAndroid Build Coastguard Worker
1251*08b48e0bSAndroid Build Coastguard Worker /* 0-9999 */
1252*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1, 10000, "%llu B", u64);
1253*08b48e0bSAndroid Build Coastguard Worker
1254*08b48e0bSAndroid Build Coastguard Worker /* 10.0k - 99.9k */
1255*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024, 99.95, "%0.01f kB", double);
1256*08b48e0bSAndroid Build Coastguard Worker
1257*08b48e0bSAndroid Build Coastguard Worker /* 100k - 999k */
1258*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024, 1000, "%llu kB", u64);
1259*08b48e0bSAndroid Build Coastguard Worker
1260*08b48e0bSAndroid Build Coastguard Worker /* 1.00M - 9.99M */
1261*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 9.995, "%0.02f MB", double);
1262*08b48e0bSAndroid Build Coastguard Worker
1263*08b48e0bSAndroid Build Coastguard Worker /* 10.0M - 99.9M */
1264*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 99.95, "%0.01f MB", double);
1265*08b48e0bSAndroid Build Coastguard Worker
1266*08b48e0bSAndroid Build Coastguard Worker /* 100M - 999M */
1267*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024 * 1024, 1000, "%llu MB", u64);
1268*08b48e0bSAndroid Build Coastguard Worker
1269*08b48e0bSAndroid Build Coastguard Worker /* 1.00G - 9.99G */
1270*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 9.995, "%0.02f GB", double);
1271*08b48e0bSAndroid Build Coastguard Worker
1272*08b48e0bSAndroid Build Coastguard Worker /* 10.0G - 99.9G */
1273*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 99.95, "%0.01f GB", double);
1274*08b48e0bSAndroid Build Coastguard Worker
1275*08b48e0bSAndroid Build Coastguard Worker /* 100G - 999G */
1276*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024, 1000, "%llu GB", u64);
1277*08b48e0bSAndroid Build Coastguard Worker
1278*08b48e0bSAndroid Build Coastguard Worker /* 1.00T - 9.99G */
1279*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024 * 1024, 9.995, "%0.02f TB", double);
1280*08b48e0bSAndroid Build Coastguard Worker
1281*08b48e0bSAndroid Build Coastguard Worker /* 10.0T - 99.9T */
1282*08b48e0bSAndroid Build Coastguard Worker CHK_FORMAT(1024LL * 1024 * 1024 * 1024, 99.95, "%0.01f TB", double);
1283*08b48e0bSAndroid Build Coastguard Worker
1284*08b48e0bSAndroid Build Coastguard Worker #undef CHK_FORMAT
1285*08b48e0bSAndroid Build Coastguard Worker
1286*08b48e0bSAndroid Build Coastguard Worker /* 100T+ */
1287*08b48e0bSAndroid Build Coastguard Worker strcpy(buf, "infty");
1288*08b48e0bSAndroid Build Coastguard Worker
1289*08b48e0bSAndroid Build Coastguard Worker return buf;
1290*08b48e0bSAndroid Build Coastguard Worker
1291*08b48e0bSAndroid Build Coastguard Worker }
1292*08b48e0bSAndroid Build Coastguard Worker
1293*08b48e0bSAndroid Build Coastguard Worker /* Unsafe describe time delta as string.
1294*08b48e0bSAndroid Build Coastguard Worker Returns a pointer to buf for convenience. */
1295*08b48e0bSAndroid Build Coastguard Worker
u_stringify_time_diff(u8 * buf,u64 cur_ms,u64 event_ms)1296*08b48e0bSAndroid Build Coastguard Worker u8 *u_stringify_time_diff(u8 *buf, u64 cur_ms, u64 event_ms) {
1297*08b48e0bSAndroid Build Coastguard Worker
1298*08b48e0bSAndroid Build Coastguard Worker if (!event_ms) {
1299*08b48e0bSAndroid Build Coastguard Worker
1300*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "none seen yet");
1301*08b48e0bSAndroid Build Coastguard Worker
1302*08b48e0bSAndroid Build Coastguard Worker } else {
1303*08b48e0bSAndroid Build Coastguard Worker
1304*08b48e0bSAndroid Build Coastguard Worker u64 delta;
1305*08b48e0bSAndroid Build Coastguard Worker s32 t_d, t_h, t_m, t_s;
1306*08b48e0bSAndroid Build Coastguard Worker u8 val_buf[STRINGIFY_VAL_SIZE_MAX];
1307*08b48e0bSAndroid Build Coastguard Worker
1308*08b48e0bSAndroid Build Coastguard Worker delta = cur_ms - event_ms;
1309*08b48e0bSAndroid Build Coastguard Worker
1310*08b48e0bSAndroid Build Coastguard Worker t_d = delta / 1000 / 60 / 60 / 24;
1311*08b48e0bSAndroid Build Coastguard Worker t_h = (delta / 1000 / 60 / 60) % 24;
1312*08b48e0bSAndroid Build Coastguard Worker t_m = (delta / 1000 / 60) % 60;
1313*08b48e0bSAndroid Build Coastguard Worker t_s = (delta / 1000) % 60;
1314*08b48e0bSAndroid Build Coastguard Worker
1315*08b48e0bSAndroid Build Coastguard Worker u_stringify_int(val_buf, t_d);
1316*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "%s days, %d hrs, %d min, %d sec", val_buf, t_h, t_m, t_s);
1317*08b48e0bSAndroid Build Coastguard Worker
1318*08b48e0bSAndroid Build Coastguard Worker }
1319*08b48e0bSAndroid Build Coastguard Worker
1320*08b48e0bSAndroid Build Coastguard Worker return buf;
1321*08b48e0bSAndroid Build Coastguard Worker
1322*08b48e0bSAndroid Build Coastguard Worker }
1323*08b48e0bSAndroid Build Coastguard Worker
1324*08b48e0bSAndroid Build Coastguard Worker /* Unsafe describe time delta as simple string.
1325*08b48e0bSAndroid Build Coastguard Worker Returns a pointer to buf for convenience. */
1326*08b48e0bSAndroid Build Coastguard Worker
u_simplestring_time_diff(u8 * buf,u64 cur_ms,u64 event_ms)1327*08b48e0bSAndroid Build Coastguard Worker u8 *u_simplestring_time_diff(u8 *buf, u64 cur_ms, u64 event_ms) {
1328*08b48e0bSAndroid Build Coastguard Worker
1329*08b48e0bSAndroid Build Coastguard Worker if (!event_ms) {
1330*08b48e0bSAndroid Build Coastguard Worker
1331*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "00:00:00");
1332*08b48e0bSAndroid Build Coastguard Worker
1333*08b48e0bSAndroid Build Coastguard Worker } else {
1334*08b48e0bSAndroid Build Coastguard Worker
1335*08b48e0bSAndroid Build Coastguard Worker u64 delta;
1336*08b48e0bSAndroid Build Coastguard Worker s32 t_d, t_h, t_m, t_s;
1337*08b48e0bSAndroid Build Coastguard Worker
1338*08b48e0bSAndroid Build Coastguard Worker delta = cur_ms - event_ms;
1339*08b48e0bSAndroid Build Coastguard Worker
1340*08b48e0bSAndroid Build Coastguard Worker t_d = delta / 1000 / 60 / 60 / 24;
1341*08b48e0bSAndroid Build Coastguard Worker t_h = (delta / 1000 / 60 / 60) % 24;
1342*08b48e0bSAndroid Build Coastguard Worker t_m = (delta / 1000 / 60) % 60;
1343*08b48e0bSAndroid Build Coastguard Worker t_s = (delta / 1000) % 60;
1344*08b48e0bSAndroid Build Coastguard Worker
1345*08b48e0bSAndroid Build Coastguard Worker sprintf(buf, "%d:%02d:%02d:%02d", t_d, t_h, t_m, t_s);
1346*08b48e0bSAndroid Build Coastguard Worker
1347*08b48e0bSAndroid Build Coastguard Worker }
1348*08b48e0bSAndroid Build Coastguard Worker
1349*08b48e0bSAndroid Build Coastguard Worker return buf;
1350*08b48e0bSAndroid Build Coastguard Worker
1351*08b48e0bSAndroid Build Coastguard Worker }
1352*08b48e0bSAndroid Build Coastguard Worker
1353*08b48e0bSAndroid Build Coastguard Worker /* Reads the map size from ENV */
get_map_size(void)1354*08b48e0bSAndroid Build Coastguard Worker u32 get_map_size(void) {
1355*08b48e0bSAndroid Build Coastguard Worker
1356*08b48e0bSAndroid Build Coastguard Worker uint32_t map_size = DEFAULT_SHMEM_SIZE;
1357*08b48e0bSAndroid Build Coastguard Worker char *ptr;
1358*08b48e0bSAndroid Build Coastguard Worker
1359*08b48e0bSAndroid Build Coastguard Worker if ((ptr = getenv("AFL_MAP_SIZE")) || (ptr = getenv("AFL_MAPSIZE"))) {
1360*08b48e0bSAndroid Build Coastguard Worker
1361*08b48e0bSAndroid Build Coastguard Worker map_size = atoi(ptr);
1362*08b48e0bSAndroid Build Coastguard Worker if (!map_size || map_size > (1 << 29)) {
1363*08b48e0bSAndroid Build Coastguard Worker
1364*08b48e0bSAndroid Build Coastguard Worker FATAL("illegal AFL_MAP_SIZE %u, must be between %u and %u", map_size, 64U,
1365*08b48e0bSAndroid Build Coastguard Worker 1U << 29);
1366*08b48e0bSAndroid Build Coastguard Worker
1367*08b48e0bSAndroid Build Coastguard Worker }
1368*08b48e0bSAndroid Build Coastguard Worker
1369*08b48e0bSAndroid Build Coastguard Worker if (map_size % 64) { map_size = (((map_size >> 6) + 1) << 6); }
1370*08b48e0bSAndroid Build Coastguard Worker
1371*08b48e0bSAndroid Build Coastguard Worker } else if (getenv("AFL_SKIP_BIN_CHECK")) {
1372*08b48e0bSAndroid Build Coastguard Worker
1373*08b48e0bSAndroid Build Coastguard Worker map_size = MAP_SIZE;
1374*08b48e0bSAndroid Build Coastguard Worker
1375*08b48e0bSAndroid Build Coastguard Worker }
1376*08b48e0bSAndroid Build Coastguard Worker
1377*08b48e0bSAndroid Build Coastguard Worker return map_size;
1378*08b48e0bSAndroid Build Coastguard Worker
1379*08b48e0bSAndroid Build Coastguard Worker }
1380*08b48e0bSAndroid Build Coastguard Worker
1381*08b48e0bSAndroid Build Coastguard Worker /* Create a stream file */
1382*08b48e0bSAndroid Build Coastguard Worker
create_ffile(u8 * fn)1383*08b48e0bSAndroid Build Coastguard Worker FILE *create_ffile(u8 *fn) {
1384*08b48e0bSAndroid Build Coastguard Worker
1385*08b48e0bSAndroid Build Coastguard Worker s32 fd;
1386*08b48e0bSAndroid Build Coastguard Worker FILE *f;
1387*08b48e0bSAndroid Build Coastguard Worker
1388*08b48e0bSAndroid Build Coastguard Worker fd = open(fn, O_WRONLY | O_CREAT | O_TRUNC, DEFAULT_PERMISSION);
1389*08b48e0bSAndroid Build Coastguard Worker
1390*08b48e0bSAndroid Build Coastguard Worker if (fd < 0) { PFATAL("Unable to create '%s'", fn); }
1391*08b48e0bSAndroid Build Coastguard Worker
1392*08b48e0bSAndroid Build Coastguard Worker f = fdopen(fd, "w");
1393*08b48e0bSAndroid Build Coastguard Worker
1394*08b48e0bSAndroid Build Coastguard Worker if (!f) { PFATAL("fdopen() failed"); }
1395*08b48e0bSAndroid Build Coastguard Worker
1396*08b48e0bSAndroid Build Coastguard Worker return f;
1397*08b48e0bSAndroid Build Coastguard Worker
1398*08b48e0bSAndroid Build Coastguard Worker }
1399*08b48e0bSAndroid Build Coastguard Worker
1400*08b48e0bSAndroid Build Coastguard Worker /* Create a file */
1401*08b48e0bSAndroid Build Coastguard Worker
create_file(u8 * fn)1402*08b48e0bSAndroid Build Coastguard Worker s32 create_file(u8 *fn) {
1403*08b48e0bSAndroid Build Coastguard Worker
1404*08b48e0bSAndroid Build Coastguard Worker s32 fd;
1405*08b48e0bSAndroid Build Coastguard Worker
1406*08b48e0bSAndroid Build Coastguard Worker fd = open(fn, O_WRONLY | O_CREAT | O_TRUNC, DEFAULT_PERMISSION);
1407*08b48e0bSAndroid Build Coastguard Worker
1408*08b48e0bSAndroid Build Coastguard Worker if (fd < 0) { PFATAL("Unable to create '%s'", fn); }
1409*08b48e0bSAndroid Build Coastguard Worker
1410*08b48e0bSAndroid Build Coastguard Worker return fd;
1411*08b48e0bSAndroid Build Coastguard Worker
1412*08b48e0bSAndroid Build Coastguard Worker }
1413*08b48e0bSAndroid Build Coastguard Worker
1414*08b48e0bSAndroid Build Coastguard Worker #ifdef __linux__
1415*08b48e0bSAndroid Build Coastguard Worker
1416*08b48e0bSAndroid Build Coastguard Worker /* Nyx requires a tmp workdir to access specific files (such as mmapped files,
1417*08b48e0bSAndroid Build Coastguard Worker * etc.). This helper function basically creates both a path to a tmp workdir
1418*08b48e0bSAndroid Build Coastguard Worker * and the workdir itself. If the environment variable TMPDIR is set, we use
1419*08b48e0bSAndroid Build Coastguard Worker * that as the base directory, otherwise we use /tmp. */
create_nyx_tmp_workdir(void)1420*08b48e0bSAndroid Build Coastguard Worker char *create_nyx_tmp_workdir(void) {
1421*08b48e0bSAndroid Build Coastguard Worker
1422*08b48e0bSAndroid Build Coastguard Worker char *tmpdir = getenv("TMPDIR");
1423*08b48e0bSAndroid Build Coastguard Worker
1424*08b48e0bSAndroid Build Coastguard Worker if (!tmpdir) { tmpdir = "/tmp"; }
1425*08b48e0bSAndroid Build Coastguard Worker
1426*08b48e0bSAndroid Build Coastguard Worker char *nyx_out_dir_path =
1427*08b48e0bSAndroid Build Coastguard Worker alloc_printf("%s/.nyx_tmp_%d/", tmpdir, (u32)getpid());
1428*08b48e0bSAndroid Build Coastguard Worker
1429*08b48e0bSAndroid Build Coastguard Worker if (mkdir(nyx_out_dir_path, 0700)) { PFATAL("Unable to create nyx workdir"); }
1430*08b48e0bSAndroid Build Coastguard Worker
1431*08b48e0bSAndroid Build Coastguard Worker return nyx_out_dir_path;
1432*08b48e0bSAndroid Build Coastguard Worker
1433*08b48e0bSAndroid Build Coastguard Worker }
1434*08b48e0bSAndroid Build Coastguard Worker
1435*08b48e0bSAndroid Build Coastguard Worker /* Vice versa, we remove the tmp workdir for nyx with this helper function. */
remove_nyx_tmp_workdir(afl_forkserver_t * fsrv,char * nyx_out_dir_path)1436*08b48e0bSAndroid Build Coastguard Worker void remove_nyx_tmp_workdir(afl_forkserver_t *fsrv, char *nyx_out_dir_path) {
1437*08b48e0bSAndroid Build Coastguard Worker
1438*08b48e0bSAndroid Build Coastguard Worker char *workdir_path = alloc_printf("%s/workdir", nyx_out_dir_path);
1439*08b48e0bSAndroid Build Coastguard Worker
1440*08b48e0bSAndroid Build Coastguard Worker if (access(workdir_path, R_OK) == 0) {
1441*08b48e0bSAndroid Build Coastguard Worker
1442*08b48e0bSAndroid Build Coastguard Worker if (fsrv->nyx_handlers->nyx_remove_work_dir(workdir_path) != true) {
1443*08b48e0bSAndroid Build Coastguard Worker
1444*08b48e0bSAndroid Build Coastguard Worker WARNF("Unable to remove nyx workdir (%s)", workdir_path);
1445*08b48e0bSAndroid Build Coastguard Worker
1446*08b48e0bSAndroid Build Coastguard Worker }
1447*08b48e0bSAndroid Build Coastguard Worker
1448*08b48e0bSAndroid Build Coastguard Worker }
1449*08b48e0bSAndroid Build Coastguard Worker
1450*08b48e0bSAndroid Build Coastguard Worker if (rmdir(nyx_out_dir_path)) {
1451*08b48e0bSAndroid Build Coastguard Worker
1452*08b48e0bSAndroid Build Coastguard Worker WARNF("Unable to remove nyx workdir (%s)", nyx_out_dir_path);
1453*08b48e0bSAndroid Build Coastguard Worker
1454*08b48e0bSAndroid Build Coastguard Worker }
1455*08b48e0bSAndroid Build Coastguard Worker
1456*08b48e0bSAndroid Build Coastguard Worker ck_free(workdir_path);
1457*08b48e0bSAndroid Build Coastguard Worker ck_free(nyx_out_dir_path);
1458*08b48e0bSAndroid Build Coastguard Worker
1459*08b48e0bSAndroid Build Coastguard Worker }
1460*08b48e0bSAndroid Build Coastguard Worker
1461*08b48e0bSAndroid Build Coastguard Worker #endif
1462*08b48e0bSAndroid Build Coastguard Worker
1463