xref: /aosp_15_r20/external/AFLplusplus/src/afl-common.c (revision 08b48e0b10e97b33e7b60c5b6e2243bd915777f2)
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