blob: 72a890b3b258609d8eacadba342da16f3ef0beeb [file] [log] [blame]
robert.swiecki3bb518c2010-10-14 00:48:24 +00001/*
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +00002 *
robert.swiecki@gmail.com90e99112015-02-15 02:05:14 +00003 * honggfuzz - fuzzing routines
4 * -----------------------------------------
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +00005 *
robert.swiecki@gmail.com8531f692015-02-17 12:25:36 +00006 * Author:
7 * Robert Swiecki <swiecki@google.com>
8 * Felix Gröbert <groebert@google.com>
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +00009 *
robert.swiecki@gmail.com772b33d2015-02-14 20:35:00 +000010 * Copyright 2010-2015 by Google Inc. All Rights Reserved.
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +000011 *
12 * Licensed under the Apache License, Version 2.0 (the "License"); you may
13 * not use this file except in compliance with the License. You may obtain
robert.swiecki@gmail.com772b33d2015-02-14 20:35:00 +000014 * a copy of the License at
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +000015 *
robert.swiecki@gmail.com772b33d2015-02-14 20:35:00 +000016 * http://www.apache.org/licenses/LICENSE-2.0
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +000017 *
robert.swiecki@gmail.com772b33d2015-02-14 20:35:00 +000018 * Unless required by applicable law or agreed to in writing, software
19 * distributed under the License is distributed on an "AS IS" BASIS,
20 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
21 * implied. See the License for the specific language governing
22 * permissions and limitations under the License.
robert.swiecki@gmail.com3b630b42015-02-16 10:53:53 +000023 *
robert.swiecki@gmail.com772b33d2015-02-14 20:35:00 +000024 */
robert.swiecki3bb518c2010-10-14 00:48:24 +000025
robert.swiecki@gmail.comba85c3e2015-02-02 14:55:16 +000026#include "common.h"
27#include "fuzz.h"
28
29#include <errno.h>
30#include <fcntl.h>
robert.swiecki@gmail.com90e99112015-02-15 02:05:14 +000031#include <inttypes.h>
robert.swiecki@gmail.com882900b2015-02-11 13:56:22 +000032#include <pthread.h>
robert.swiecki@gmail.comba85c3e2015-02-02 14:55:16 +000033#include <signal.h>
34#include <stddef.h>
35#include <stdint.h>
36#include <stdio.h>
37#include <stdlib.h>
38#include <string.h>
robert.swiecki3bb518c2010-10-14 00:48:24 +000039#include <sys/param.h>
Robert Swiecki1e1b3532016-03-10 18:43:28 +010040#include <sys/resource.h>
Jagger08816fd2016-03-11 01:32:38 +010041#include <sys/mman.h>
robert.swiecki3bb518c2010-10-14 00:48:24 +000042#include <sys/stat.h>
robert.swiecki@gmail.comba85c3e2015-02-02 14:55:16 +000043#include <sys/time.h>
44#include <sys/types.h>
45#include <sys/wait.h>
robert.swiecki3bb518c2010-10-14 00:48:24 +000046#include <time.h>
robert.swiecki@gmail.comba85c3e2015-02-02 14:55:16 +000047#include <unistd.h>
robert.swiecki3bb518c2010-10-14 00:48:24 +000048
robert.swiecki3bb518c2010-10-14 00:48:24 +000049#include "arch.h"
robert.swiecki3bb518c2010-10-14 00:48:24 +000050#include "files.h"
robert.swiecki@gmail.come7190b92015-02-14 23:05:42 +000051#include "log.h"
robert.swiecki@gmail.com36700b52015-02-22 05:03:16 +000052#include "mangle.h"
robert.swiecki@gmail.come7190b92015-02-14 23:05:42 +000053#include "report.h"
Jagger00265602016-03-10 02:36:27 +010054#include "sancov.h"
robert.swiecki@gmail.come7190b92015-02-14 23:05:42 +000055#include "util.h"
robert.swiecki3bb518c2010-10-14 00:48:24 +000056
Jagger1e05f742016-03-10 21:36:27 +010057extern char **environ;
58
Jagger6113c432015-09-24 05:00:28 +020059static pthread_t fuzz_mainThread;
60
robert.swiecki3bb518c2010-10-14 00:48:24 +000061static void fuzz_getFileName(honggfuzz_t * hfuzz, char *fileName)
62{
robert.swieckiba512632011-01-28 11:57:26 +000063 struct timeval tv;
64 gettimeofday(&tv, NULL);
65
Anestis Bechtsoudisd9680532015-09-06 17:37:05 +030066 snprintf(fileName, PATH_MAX, "%s/.honggfuzz.%d.%lu.%llx.%s", hfuzz->workDir, (int)getpid(),
robert.swiecki@gmail.com6f319912015-02-28 05:01:37 +000067 (unsigned long int)tv.tv_sec, (unsigned long long int)util_rndGet(0, 1ULL << 62),
robert.swiecki@gmail.combb5d2642015-02-25 20:00:00 +000068 hfuzz->fileExtn);
robert.swiecki3bb518c2010-10-14 00:48:24 +000069}
70
Robert Swiecki1e1b3532016-03-10 18:43:28 +010071static bool fuzz_prepareExecve(honggfuzz_t * hfuzz, const char *fileName)
72{
73 /*
74 * Set timeout (prof), real timeout (2*prof), and rlimit_cpu (2*prof)
75 */
Robert Swiecki4038c9b2016-03-31 18:58:52 +020076 if (hfuzz->persistent == false && hfuzz->tmOut) {
Robert Swiecki1e1b3532016-03-10 18:43:28 +010077 struct itimerval it;
78
79 /*
80 * The hfuzz->tmOut is real CPU usage time...
81 */
82 it.it_value.tv_sec = hfuzz->tmOut;
83 it.it_value.tv_usec = 0;
84 it.it_interval.tv_sec = 0;
85 it.it_interval.tv_usec = 0;
86 if (setitimer(ITIMER_PROF, &it, NULL) == -1) {
Robert Swieckic6a48b12016-03-11 17:41:57 +010087 PLOG_D("Couldn't set the ITIMER_PROF timer");
Robert Swiecki1e1b3532016-03-10 18:43:28 +010088 }
89
90 /*
91 * ...so, if a process sleeps, this one should
92 * trigger a signal...
93 */
94 it.it_value.tv_sec = hfuzz->tmOut;
95 it.it_value.tv_usec = 0;
96 it.it_interval.tv_sec = 0;
97 it.it_interval.tv_usec = 0;
98 if (setitimer(ITIMER_REAL, &it, NULL) == -1) {
99 PLOG_E("Couldn't set the ITIMER_REAL timer");
100 return false;
101 }
102
103 /*
104 * ..if a process sleeps and catches SIGPROF/SIGALRM
105 * rlimits won't help either. However, arch_checkTimeLimit
106 * will send a SIGKILL at tmOut + 2 seconds. That should
107 * do it :)
108 */
109 struct rlimit rl;
110
111 rl.rlim_cur = hfuzz->tmOut + 1;
112 rl.rlim_max = hfuzz->tmOut + 1;
113 if (setrlimit(RLIMIT_CPU, &rl) == -1) {
Robert Swieckic6a48b12016-03-11 17:41:57 +0100114 PLOG_D("Couldn't enforce the RLIMIT_CPU resource limit");
Robert Swiecki1e1b3532016-03-10 18:43:28 +0100115 }
116 }
117
118 /*
119 * The address space limit. If big enough - roughly the size of RAM used
120 */
121 if (hfuzz->asLimit) {
Jagger06a4bbc2016-03-10 21:19:23 +0100122 struct rlimit rl = {
Robert Swiecki1e1b3532016-03-10 18:43:28 +0100123 .rlim_cur = hfuzz->asLimit * 1024ULL * 1024ULL,
124 .rlim_max = hfuzz->asLimit * 1024ULL * 1024ULL,
125 };
Jagger06a4bbc2016-03-10 21:19:23 +0100126 if (setrlimit(RLIMIT_AS, &rl) == -1) {
Robert Swiecki1e1b3532016-03-10 18:43:28 +0100127 PLOG_D("Couldn't enforce the RLIMIT_AS resource limit, ignoring");
128 }
129 }
130
131 if (hfuzz->nullifyStdio) {
132 util_nullifyStdio();
133 }
134
135 if (hfuzz->fuzzStdin) {
136 /*
137 * Uglyyyyyy ;)
138 */
139 if (!util_redirectStdin(fileName)) {
140 return false;
141 }
142 }
143
Jagger80041fe2016-03-10 21:32:35 +0100144 if (hfuzz->clearEnv) {
Jagger1e05f742016-03-10 21:36:27 +0100145 environ = NULL;
Jagger80041fe2016-03-10 21:32:35 +0100146 }
Robert Swiecki1e1b3532016-03-10 18:43:28 +0100147 if (sancov_prepareExecve(hfuzz) == false) {
148 LOG_E("sancov_prepareExecve() failed");
149 return false;
150 }
151 for (size_t i = 0; i < ARRAYSIZE(hfuzz->envs) && hfuzz->envs[i]; i++) {
152 putenv(hfuzz->envs[i]);
153 }
154 setsid();
155
156 return true;
157}
158
Robert Swieckia96d78d2016-03-14 16:50:50 +0100159static bool fuzz_prepareFileDynamically(honggfuzz_t * hfuzz, fuzzer_t * fuzzer)
robert.swiecki@gmail.com6d6f7562015-02-17 22:18:51 +0000160{
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100161 struct dynfile_t *dynfile;
Robert Swieckia08ab1c2016-03-14 18:29:21 +0100162
163 {
Robert Swiecki76ecd5e2016-03-16 14:57:03 +0100164 MX_SCOPED_LOCK(&hfuzz->dynfileq_mutex);
Robert Swiecki4b0390a2016-03-14 18:44:05 +0100165
Jaggerb070b812016-03-15 05:31:30 +0100166 if (hfuzz->dynfileqCnt == 0) {
Jaggere4496512016-03-16 01:55:46 +0100167 LOG_F("The dynamic file corpus is empty. Apparently, the initial fuzzing of the "
168 "provided file corpus (-f) has not produced any follow-up files with positive "
169 "coverage and/or CPU counters");
Jaggerb070b812016-03-15 05:31:30 +0100170 }
171
Robert Swiecki4b0390a2016-03-14 18:44:05 +0100172 size_t i = 0U;
Robert Swieckia08ab1c2016-03-14 18:29:21 +0100173 size_t dynFilePos = util_rndGet(0, hfuzz->dynfileqCnt - 1);
174 TAILQ_FOREACH(dynfile, &hfuzz->dynfileq, pointers) {
175 if (i++ == dynFilePos) {
176 break;
177 }
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100178 }
179 }
180
181 memcpy(fuzzer->dynamicFile, dynfile->data, dynfile->size);
Robert Swieckia08ab1c2016-03-14 18:29:21 +0100182 fuzzer->dynamicFileSz = dynfile->size;
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100183
Anestis Bechtsoudis6ef24e42016-01-13 13:05:53 +0200184 mangle_Resize(hfuzz, fuzzer->dynamicFile, &fuzzer->dynamicFileSz);
Robert Swieckia96d78d2016-03-14 16:50:50 +0100185 mangle_mangleContent(hfuzz, fuzzer);
robert.swiecki@gmail.com6d6f7562015-02-17 22:18:51 +0000186
Jagger1d743382016-04-02 02:09:41 +0200187 if (hfuzz->persistent == false && files_writeBufToFile
robert.swiecki@gmail.com62e34ae2015-03-05 03:39:32 +0000188 (fuzzer->fileName, fuzzer->dynamicFile, fuzzer->dynamicFileSz,
Jagger5a3c4c32016-04-16 19:27:47 +0200189 O_WRONLY | O_CREAT | O_EXCL | O_TRUNC | O_CLOEXEC) == false) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200190 LOG_E("Couldn't write buffer to file '%s'", fuzzer->fileName);
robert.swiecki@gmail.com3b6c6292015-02-26 11:48:46 +0000191 return false;
192 }
robert.swiecki@gmail.com6d6f7562015-02-17 22:18:51 +0000193
robert.swiecki@gmail.com6d6f7562015-02-17 22:18:51 +0000194 return true;
robert.swiecki@gmail.comd7aed312015-02-03 21:26:37 +0000195}
196
robert.swiecki@gmail.com4a7a9d82015-03-01 01:25:16 +0000197static bool fuzz_prepareFile(honggfuzz_t * hfuzz, fuzzer_t * fuzzer, int rnd_index)
robert.swiecki3bb518c2010-10-14 00:48:24 +0000198{
Jagger1aa94d72016-04-02 02:37:35 +0200199 ssize_t fileSz =
robert.swiecki@gmail.com4a7a9d82015-03-01 01:25:16 +0000200 files_readFileToBufMax(hfuzz->files[rnd_index], fuzzer->dynamicFile, hfuzz->maxFileSz);
Jagger1aa94d72016-04-02 02:37:35 +0200201 if (fileSz < 0) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200202 LOG_E("Couldn't read contents of '%s'", hfuzz->files[rnd_index]);
robert.swiecki@gmail.combb5d2642015-02-25 20:00:00 +0000203 return false;
robert.swiecki3bb518c2010-10-14 00:48:24 +0000204 }
Robert Swieckia96d78d2016-03-14 16:50:50 +0100205 fuzzer->dynamicFileSz = fileSz;
robert.swiecki3bb518c2010-10-14 00:48:24 +0000206
Anestis Bechtsoudis25262b32015-11-07 22:39:47 +0200207 /* If flip rate is 0.0, early abort file mangling */
Robert Swieckia96d78d2016-03-14 16:50:50 +0100208 if (fuzzer->flipRate != 0.0L) {
209 mangle_Resize(hfuzz, fuzzer->dynamicFile, &fuzzer->dynamicFileSz);
210 mangle_mangleContent(hfuzz, fuzzer);
Anestis Bechtsoudis25262b32015-11-07 22:39:47 +0200211 }
robert.swiecki@gmail.comc070b942015-02-25 18:29:19 +0000212
Jagger1d743382016-04-02 02:09:41 +0200213 if (hfuzz->persistent == false && files_writeBufToFile
Robert Swieckia96d78d2016-03-14 16:50:50 +0100214 (fuzzer->fileName, fuzzer->dynamicFile, fuzzer->dynamicFileSz,
Jagger5a3c4c32016-04-16 19:27:47 +0200215 O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC) == false) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200216 LOG_E("Couldn't write buffer to file '%s'", fuzzer->fileName);
robert.swiecki@gmail.come7680522015-02-22 22:22:37 +0000217 return false;
218 }
219
robert.swiecki3bb518c2010-10-14 00:48:24 +0000220 return true;
221}
222
Jaggera34b3022016-04-02 02:20:40 +0200223static bool fuzz_prepareFileExternally(honggfuzz_t * hfuzz, fuzzer_t * fuzzer)
robert.swiecki3bb518c2010-10-14 00:48:24 +0000224{
Jagger5a3c4c32016-04-16 19:27:47 +0200225 int dstfd = open(fuzzer->fileName, O_CREAT | O_EXCL | O_RDWR | O_CLOEXEC, 0644);
Jaggera34b3022016-04-02 02:20:40 +0200226 if (dstfd == -1) {
227 PLOG_E("Couldn't create a temporary file '%s'", fuzzer->fileName);
228 return false;
Robert Swieckie48811e2016-03-09 18:31:19 +0100229 }
Jaggera34b3022016-04-02 02:20:40 +0200230 close(dstfd);
231
232 LOG_D("Created '%s' as an input file", fuzzer->fileName);
robert.swiecki3d505e22010-10-14 01:17:17 +0000233
Robert Swiecki12800cd2016-03-31 15:38:10 +0200234 pid_t pid = fork();
robert.swiecki3bb518c2010-10-14 00:48:24 +0000235 if (pid == -1) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200236 PLOG_E("Couldn't fork");
robert.swiecki3bb518c2010-10-14 00:48:24 +0000237 return false;
238 }
239
240 if (!pid) {
241 /*
robert.swiecki@gmail.comcdf18f92015-02-11 22:22:18 +0000242 * child performs the external file modifications
robert.swiecki3bb518c2010-10-14 00:48:24 +0000243 */
robert.swiecki@gmail.com20851202015-03-01 01:48:15 +0000244 execl(hfuzz->externalCommand, hfuzz->externalCommand, fuzzer->fileName, NULL);
Robert Swieckic8c32db2015-10-09 18:06:22 +0200245 PLOG_F("Couldn't execute '%s %s'", hfuzz->externalCommand, fuzzer->fileName);
robert.swiecki3bb518c2010-10-14 00:48:24 +0000246 return false;
robert.swiecki@gmail.com757ee192015-02-13 16:54:02 +0000247 }
robert.swiecki@gmail.com8a9df0e2015-02-13 17:08:06 +0000248
robert.swiecki@gmail.com757ee192015-02-13 16:54:02 +0000249 /*
250 * parent waits until child is done fuzzing the input file
251 */
robert.swiecki@gmail.com757ee192015-02-13 16:54:02 +0000252 int childStatus;
253 int flags = 0;
robert.swiecki@gmail.coma2291182015-02-13 16:56:27 +0000254#if defined(__WNOTHREAD)
255 flags |= __WNOTHREAD;
robert.swiecki@gmail.com8a9df0e2015-02-13 17:08:06 +0000256#endif /* defined(__WNOTHREAD) */
robert.swiecki@gmail.com757ee192015-02-13 16:54:02 +0000257 while (wait4(pid, &childStatus, flags, NULL) != pid) ;
robert.swiecki@gmail.com757ee192015-02-13 16:54:02 +0000258 if (WIFSIGNALED(childStatus)) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200259 LOG_E("External command terminated with signal %d", WTERMSIG(childStatus));
robert.swiecki3d505e22010-10-14 01:17:17 +0000260 return false;
robert.swiecki3bb518c2010-10-14 00:48:24 +0000261 }
Jagger1d743382016-04-02 02:09:41 +0200262 if (!WIFEXITED(childStatus)) {
263 LOG_F("External command terminated abnormally, status: %d", childStatus);
264 return false;
265 }
266 LOG_D("External command exited with status %d", WEXITSTATUS(childStatus));
267
268 ssize_t rsz = files_readFileToBufMax(fuzzer->fileName, fuzzer->dynamicFile, hfuzz->maxFileSz);
Jagger1aa94d72016-04-02 02:37:35 +0200269 if (rsz < 0) {
Jagger1d743382016-04-02 02:09:41 +0200270 LOG_W("Couldn't read back '%s' to the buffer", fuzzer->fileName);
271 return false;
272 }
273 fuzzer->dynamicFileSz = rsz;
274
275 if (hfuzz->persistent) {
276 unlink(fuzzer->fileName);
277 }
278
279 return true;
robert.swiecki3bb518c2010-10-14 00:48:24 +0000280}
281
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300282static bool fuzz_runVerifier(honggfuzz_t * hfuzz, fuzzer_t * crashedFuzzer)
283{
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300284 int crashFd = -1;
285 uint8_t *crashBuf = NULL;
286 off_t crashFileSz = 0;
287
288 crashBuf = files_mapFile(crashedFuzzer->crashFileName, &crashFileSz, &crashFd, false);
289 if (crashBuf == NULL) {
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700290 LOG_E("Couldn't open and map '%s' in R/O mode", crashedFuzzer->crashFileName);
Robert Swieckie48811e2016-03-09 18:31:19 +0100291 return false;
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300292 }
Jagger28aebbd2016-03-10 22:58:51 +0100293 DEFER(munmap(crashBuf, crashFileSz));
294 DEFER(close(crashFd));
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300295
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700296 LOG_I("Launching verifier for %" PRIx64 " hash", crashedFuzzer->backtrace);
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300297 for (int i = 0; i < _HF_VERIFIER_ITER; i++) {
298 fuzzer_t vFuzzer = {
299 .pid = 0,
300 .timeStartedMillis = util_timeNowMillis(),
Anestis Bechtsoudisecb0a662015-09-27 18:19:46 +0300301 .crashFileName = {0},
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300302 .pc = 0ULL,
303 .backtrace = 0ULL,
304 .access = 0ULL,
305 .exception = 0,
306 .dynamicFileSz = 0,
307 .dynamicFile = NULL,
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200308 .sanCovCnts = {
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200309 .hitBBCnt = 0ULL,
310 .totalBBCnt = 0ULL,
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200311 .dsoCnt = 0ULL,
312 .iDsoCnt = 0ULL,
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200313 .newBBCnt = 0ULL,
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200314 .crashesCnt = 0ULL,
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200315 },
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300316 .report = {'\0'},
Jaggerb2c1fad2016-03-17 23:37:10 +0100317 .mainWorker = false,
318
Jagger2381ef42016-03-20 23:32:05 +0100319 .linux = {
320 .hwCnts = {
321 .cpuInstrCnt = 0ULL,
322 .cpuBranchCnt = 0ULL,
323 .customCnt = 0ULL,
324 .bbCnt = 0ULL,
325 .newBBCnt = 0ULL,
326 },
327 .perfMmapBuf = NULL,
328 .perfMmapAux = NULL,
329 },
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300330 };
331
332 fuzz_getFileName(hfuzz, vFuzzer.fileName);
333 if (files_writeBufToFile
Jagger5a3c4c32016-04-16 19:27:47 +0200334 (vFuzzer.fileName, crashBuf, crashFileSz, O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC) == false) {
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700335 LOG_E("Couldn't write buffer to file '%s'", vFuzzer.fileName);
Robert Swieckie48811e2016-03-09 18:31:19 +0100336 return false;
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300337 }
338
Robert Swiecki12800cd2016-03-31 15:38:10 +0200339 vFuzzer.pid = arch_fork(hfuzz, &vFuzzer);
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300340 if (vFuzzer.pid == -1) {
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700341 PLOG_F("Couldn't fork");
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300342 return false;
343 }
344
345 if (!vFuzzer.pid) {
Robert Swiecki1e1b3532016-03-10 18:43:28 +0100346 if (fuzz_prepareExecve(hfuzz, crashedFuzzer->crashFileName) == false) {
347 LOG_E("fuzz_prepareExecve() failed");
Jagger2b97dc92016-03-10 04:28:18 +0100348 return false;
349 }
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300350 if (!arch_launchChild(hfuzz, crashedFuzzer->crashFileName)) {
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700351 LOG_E("Error launching verifier child process");
Robert Swieckie48811e2016-03-09 18:31:19 +0100352 return false;
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300353 }
354 }
355
356 arch_reapChild(hfuzz, &vFuzzer);
357 unlink(vFuzzer.fileName);
358
Anestis Bechtsoudisecb0a662015-09-27 18:19:46 +0300359 /* If stack hash doesn't match skip name tag and exit */
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300360 if (crashedFuzzer->backtrace != vFuzzer.backtrace) {
Anestis Bechtsoudis0cde66f2015-10-11 19:37:11 -0700361 LOG_D("Verifier stack hash mismatch");
Robert Swieckie48811e2016-03-09 18:31:19 +0100362 return false;
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300363 }
364 }
365
366 /* Workspace is inherited, just append a extra suffix */
367 char verFile[PATH_MAX] = { 0 };
368 snprintf(verFile, sizeof(verFile), "%s.verified", crashedFuzzer->crashFileName);
Anestis Bechtsoudisecb0a662015-09-27 18:19:46 +0300369
Anestis Bechtsoudisd86e6602015-11-07 18:34:06 +0200370 /* Copy file with new suffix & remove original copy */
371 bool dstFileExists = false;
372 if (files_copyFile(crashedFuzzer->crashFileName, verFile, &dstFileExists)) {
373 LOG_I("Successfully verified, saving as (%s)", verFile);
Jaggerd34417d2016-03-16 01:26:54 +0100374 ATOMIC_POST_INC(hfuzz->verifiedCrashesCnt);
Anestis Bechtsoudisd86e6602015-11-07 18:34:06 +0200375 unlink(crashedFuzzer->crashFileName);
376 } else {
377 if (dstFileExists) {
378 LOG_I("It seems that '%s' already exists, skipping", verFile);
379 } else {
380 LOG_E("Couldn't copy '%s' to '%s'", crashedFuzzer->crashFileName, verFile);
Robert Swieckie48811e2016-03-09 18:31:19 +0100381 return false;
Anestis Bechtsoudisd86e6602015-11-07 18:34:06 +0200382 }
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300383 }
Anestis Bechtsoudisecb0a662015-09-27 18:19:46 +0300384
Robert Swieckie48811e2016-03-09 18:31:19 +0100385 return true;
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300386}
387
Robert Swieckia4a68052016-03-17 16:51:01 +0100388static void fuzz_addFileToFileQLocked(honggfuzz_t * hfuzz, uint8_t * data, size_t size,
389 uint64_t cov)
Robert Swieckif3534bb2016-03-14 18:55:10 +0100390{
Robert Swiecki7084e652016-03-14 19:47:00 +0100391 struct dynfile_t *dynfile = (struct dynfile_t *)util_Malloc(sizeof(struct dynfile_t));
Robert Swieckif3534bb2016-03-14 18:55:10 +0100392 dynfile->size = size;
Robert Swiecki7084e652016-03-14 19:47:00 +0100393 dynfile->data = (uint8_t *) util_Malloc(size);
Robert Swieckif3534bb2016-03-14 18:55:10 +0100394 memcpy(dynfile->data, data, size);
395 TAILQ_INSERT_TAIL(&hfuzz->dynfileq, dynfile, pointers);
396 hfuzz->dynfileqCnt++;
397
398 char fname[PATH_MAX];
Robert Swieckia4a68052016-03-17 16:51:01 +0100399 snprintf(fname, sizeof(fname),
Robert Swiecki4fc3d572016-03-17 16:55:07 +0100400 "%s/COV.RANK.%06" PRIu64 ".PID.%d.COVBB.%07" PRIu64 ".TIME.%" PRIu64 ".RND.%" PRIx64,
401 hfuzz->workDir, (uint64_t) 999999ULL - cov, getpid(), cov, (uint64_t) time(NULL),
402 util_rndGet(0, 0xFFFFFFFFFFFF));
Jagger5a3c4c32016-04-16 19:27:47 +0200403 if (files_writeBufToFile(fname, data, size, O_WRONLY | O_CREAT | O_EXCL | O_TRUNC | O_CLOEXEC) == false) {
Robert Swieckif3534bb2016-03-14 18:55:10 +0100404 LOG_W("Couldn't write buffer to file '%s'", fname);
405 }
406}
407
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200408static void fuzz_perfFeedback(honggfuzz_t * hfuzz, fuzzer_t * fuzzer)
409{
410 LOG_D
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100411 ("New file size: %zu, Perf feedback new/cur (instr,branch): %" PRIu64 "/%" PRIu64 ",%"
412 PRIu64 "/%" PRIu64 ", BBcnt new/total: %" PRIu64 "/%" PRIu64, fuzzer->dynamicFileSz,
Jagger247c3b42016-03-21 23:24:05 +0100413 fuzzer->linux.hwCnts.cpuInstrCnt, hfuzz->linux.hwCnts.cpuInstrCnt,
414 fuzzer->linux.hwCnts.cpuBranchCnt, hfuzz->linux.hwCnts.cpuBranchCnt,
415 fuzzer->linux.hwCnts.newBBCnt, hfuzz->linux.hwCnts.bbCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200416
Robert Swiecki76ecd5e2016-03-16 14:57:03 +0100417 MX_SCOPED_LOCK(&hfuzz->dynfileq_mutex);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200418
Jagger247c3b42016-03-21 23:24:05 +0100419 int64_t diff0 = hfuzz->linux.hwCnts.cpuInstrCnt - fuzzer->linux.hwCnts.cpuInstrCnt;
420 int64_t diff1 = hfuzz->linux.hwCnts.cpuBranchCnt - fuzzer->linux.hwCnts.cpuBranchCnt;
421 int64_t diff2 = hfuzz->linux.hwCnts.customCnt - fuzzer->linux.hwCnts.customCnt;
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200422
Jagger2381ef42016-03-20 23:32:05 +0100423 if (diff0 < 0 || diff1 < 0 || diff2 < 0 || fuzzer->linux.hwCnts.newBBCnt > 0) {
Robert Swieckif3534bb2016-03-14 18:55:10 +0100424 LOG_I
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100425 ("New file size: %zu, Perf feedback new/cur (instr,branch): %" PRIu64 "/%" PRIu64 ",%"
426 PRIu64 "/%" PRIu64 ", BBcnt new/total: %" PRIu64 "/%" PRIu64, fuzzer->dynamicFileSz,
Jagger247c3b42016-03-21 23:24:05 +0100427 fuzzer->linux.hwCnts.cpuInstrCnt, hfuzz->linux.hwCnts.cpuInstrCnt,
428 fuzzer->linux.hwCnts.cpuBranchCnt, hfuzz->linux.hwCnts.cpuBranchCnt,
429 fuzzer->linux.hwCnts.newBBCnt, hfuzz->linux.hwCnts.bbCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200430
Jagger247c3b42016-03-21 23:24:05 +0100431 hfuzz->linux.hwCnts.cpuInstrCnt = fuzzer->linux.hwCnts.cpuInstrCnt;
432 hfuzz->linux.hwCnts.cpuBranchCnt = fuzzer->linux.hwCnts.cpuBranchCnt;
433 hfuzz->linux.hwCnts.customCnt = fuzzer->linux.hwCnts.customCnt;
434 hfuzz->linux.hwCnts.bbCnt += fuzzer->linux.hwCnts.newBBCnt;
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200435
Robert Swieckia4a68052016-03-17 16:51:01 +0100436 fuzz_addFileToFileQLocked(hfuzz, fuzzer->dynamicFile, fuzzer->dynamicFileSz,
Jagger2381ef42016-03-20 23:32:05 +0100437 fuzzer->linux.hwCnts.newBBCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200438 }
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200439}
440
441static void fuzz_sanCovFeedback(honggfuzz_t * hfuzz, fuzzer_t * fuzzer)
442{
443 LOG_D
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100444 ("File size (Best/New): %zu, SanCov feedback (bb,dso): Best: [%" PRIu64
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200445 ",%" PRIu64 "] / New: [%" PRIu64 ",%" PRIu64 "], newBBs:%" PRIu64,
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100446 fuzzer->dynamicFileSz, hfuzz->sanCovCnts.hitBBCnt,
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200447 hfuzz->sanCovCnts.iDsoCnt, fuzzer->sanCovCnts.hitBBCnt, fuzzer->sanCovCnts.iDsoCnt,
448 fuzzer->sanCovCnts.newBBCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200449
Robert Swiecki76ecd5e2016-03-16 14:57:03 +0100450 MX_SCOPED_LOCK(&hfuzz->dynfileq_mutex);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200451
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200452 /*
453 * Keep mutated seed if:
Robert Swiecki142f9412016-03-14 19:22:01 +0100454 * a) Newly discovered (not met before) BBs
455 * b) More instrumented DSOs loaded
Robert Swiecki23ec02a2016-01-19 18:47:45 +0100456 *
Anestis Bechtsoudisb78cf602016-01-07 13:10:50 +0200457 * TODO: (a) method can significantly assist to further improvements in interesting areas
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200458 * discovery if combined with seeds pool/queue support. If a runtime queue is maintained
459 * more interesting seeds can be saved between runs instead of instantly discarded
460 * based on current absolute elitism (only one mutated seed is promoted).
461 */
Robert Swiecki142f9412016-03-14 19:22:01 +0100462 if (fuzzer->sanCovCnts.newBBCnt > 0 || hfuzz->sanCovCnts.iDsoCnt < fuzzer->sanCovCnts.iDsoCnt) {
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100463 LOG_I("SanCov Update: file size (Cur): %zu, newBBs:%" PRIu64
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200464 ", counters (Cur,New): %" PRIu64 "/%" PRIu64 ",%" PRIu64 "/%" PRIu64,
Robert Swiecki3bfc33c2016-03-14 18:12:41 +0100465 fuzzer->dynamicFileSz, fuzzer->sanCovCnts.newBBCnt,
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200466 hfuzz->sanCovCnts.hitBBCnt, hfuzz->sanCovCnts.iDsoCnt, fuzzer->sanCovCnts.hitBBCnt,
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200467 fuzzer->sanCovCnts.iDsoCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200468
Robert Swiecki142f9412016-03-14 19:22:01 +0100469 hfuzz->sanCovCnts.hitBBCnt += fuzzer->sanCovCnts.newBBCnt;
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200470 hfuzz->sanCovCnts.dsoCnt = fuzzer->sanCovCnts.dsoCnt;
471 hfuzz->sanCovCnts.iDsoCnt = fuzzer->sanCovCnts.iDsoCnt;
472 hfuzz->sanCovCnts.crashesCnt += fuzzer->sanCovCnts.crashesCnt;
Robert Swiecki142f9412016-03-14 19:22:01 +0100473 hfuzz->sanCovCnts.newBBCnt = fuzzer->sanCovCnts.newBBCnt;
Anestis Bechtsoudisb78cf602016-01-07 13:10:50 +0200474
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200475 if (hfuzz->sanCovCnts.totalBBCnt < fuzzer->sanCovCnts.totalBBCnt) {
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200476 /* Keep only the max value (for dlopen cases) to measure total target coverage */
Anestis Bechtsoudis56e360f2016-01-11 14:29:17 +0200477 hfuzz->sanCovCnts.totalBBCnt = fuzzer->sanCovCnts.totalBBCnt;
Anestis Bechtsoudisa16f70f2016-01-03 13:03:21 +0200478 }
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200479
Robert Swieckia4a68052016-03-17 16:51:01 +0100480 fuzz_addFileToFileQLocked(hfuzz, fuzzer->dynamicFile, fuzzer->dynamicFileSz,
481 fuzzer->sanCovCnts.hitBBCnt);
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200482 }
Anestis Bechtsoudisbe0ac7b2015-12-26 15:38:47 +0200483}
484
Robert Swiecki6d01e822016-03-15 20:21:46 +0100485static fuzzState_t fuzz_getState(honggfuzz_t * hfuzz)
486{
Jaggerd34417d2016-03-16 01:26:54 +0100487 return ATOMIC_GET(hfuzz->state);
Robert Swiecki6d01e822016-03-15 20:21:46 +0100488}
489
490static void fuzz_setState(honggfuzz_t * hfuzz, fuzzState_t state)
491{
Jaggerd34417d2016-03-16 01:26:54 +0100492 ATOMIC_SET(hfuzz->state, state);
Robert Swiecki6d01e822016-03-15 20:21:46 +0100493}
494
Robert Swieckidecf14b2016-03-31 15:09:28 +0200495static void fuzz_fuzzLoop(honggfuzz_t * hfuzz, fuzzer_t * fuzzer)
Jagger190f0dc2015-09-05 16:41:22 +0200496{
Robert Swieckidecf14b2016-03-31 15:09:28 +0200497 fuzzer->pid = 0;
498 fuzzer->timeStartedMillis = util_timeNowMillis();
499 fuzzer->crashFileName[0] = '\0';
500 fuzzer->pc = 0ULL;
501 fuzzer->backtrace = 0ULL;
502 fuzzer->access = 0ULL;
503 fuzzer->exception = 0;
504 fuzzer->report[0] = '\0';
505 fuzzer->mainWorker = true;
506 fuzzer->origFileName = "DYNAMIC";
507 fuzzer->flipRate = hfuzz->origFlipRate;
508 fuzzer->dynamicFileSz = 0;
Robert Swieckia96d78d2016-03-14 16:50:50 +0100509
Robert Swieckidecf14b2016-03-31 15:09:28 +0200510 fuzzer->sanCovCnts.hitBBCnt = 0ULL;
511 fuzzer->sanCovCnts.totalBBCnt = 0ULL;
512 fuzzer->sanCovCnts.dsoCnt = 0ULL;
513 fuzzer->sanCovCnts.newBBCnt = 0ULL;
514 fuzzer->sanCovCnts.crashesCnt = 0ULL;
Robert Swieckia96d78d2016-03-14 16:50:50 +0100515
Robert Swieckidecf14b2016-03-31 15:09:28 +0200516 fuzzer->linux.hwCnts.cpuInstrCnt = 0ULL;
517 fuzzer->linux.hwCnts.cpuBranchCnt = 0ULL;
518 fuzzer->linux.hwCnts.customCnt = 0ULL;
519 fuzzer->linux.hwCnts.bbCnt = 0ULL;
520 fuzzer->linux.hwCnts.newBBCnt = 0ULL;
521 fuzzer->linux.perfMmapBuf = NULL;
522 fuzzer->linux.perfMmapAux = NULL;
Jagger190f0dc2015-09-05 16:41:22 +0200523
Anestis Bechtsoudis2013f252015-11-07 22:03:31 +0200524 size_t rnd_index = util_rndGet(0, hfuzz->fileCnt - 1);
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200525
526 /* If dry run mode, pick the next file and not a random one */
Robert Swieckidecf14b2016-03-31 15:09:28 +0200527 if (fuzzer->flipRate == 0.0L && hfuzz->useVerifier) {
Jaggerd34417d2016-03-16 01:26:54 +0100528 rnd_index = ATOMIC_POST_INC(hfuzz->lastFileIndex);
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200529 }
Robert Swiecki129afe82016-03-15 20:03:20 +0100530
Robert Swiecki6d01e822016-03-15 20:21:46 +0100531 if (fuzz_getState(hfuzz) == _HF_STATE_DYNAMIC_PRE) {
Jaggerd34417d2016-03-16 01:26:54 +0100532 rnd_index = ATOMIC_POST_INC(hfuzz->lastFileIndex);
Robert Swieckia96d78d2016-03-14 16:50:50 +0100533 if (rnd_index >= hfuzz->fileCnt) {
Jagger33fd9522016-03-15 22:13:57 +0100534 /*
535 * The waiting logic (for the DYNAMIC_PRE phase to finish) should be based on cond-waits
536 * or mutexes, but it'd complicate code too much
537 */
Robert Swiecki6d01e822016-03-15 20:21:46 +0100538 while (fuzz_getState(hfuzz) == _HF_STATE_DYNAMIC_PRE) {
539 sleep(1);
Robert Swiecki129afe82016-03-15 20:03:20 +0100540 }
Robert Swieckia96d78d2016-03-14 16:50:50 +0100541 }
542 }
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200543
Jagger33fd9522016-03-15 22:13:57 +0100544 fuzzState_t state = fuzz_getState(hfuzz);
Robert Swieckidecf14b2016-03-31 15:09:28 +0200545 if (state != _HF_STATE_DYNAMIC_MAIN) {
546 fuzzer->origFileName = files_basename(hfuzz->files[rnd_index]);
Robert Swieckia08ab1c2016-03-14 18:29:21 +0100547 }
Robert Swiecki6d01e822016-03-15 20:21:46 +0100548
Robert Swieckidecf14b2016-03-31 15:09:28 +0200549 fuzz_getFileName(hfuzz, fuzzer->fileName);
Jagger190f0dc2015-09-05 16:41:22 +0200550
Jagger30384ec2016-04-02 02:30:32 +0200551 if (hfuzz->externalCommand) {
552 if (!fuzz_prepareFileExternally(hfuzz, fuzzer)) {
553 LOG_F("fuzz_prepareFileExternally() failed");
554 }
555 } else if (state == _HF_STATE_DYNAMIC_MAIN) {
Robert Swieckidecf14b2016-03-31 15:09:28 +0200556 if (!fuzz_prepareFileDynamically(hfuzz, fuzzer)) {
Jagger1d8e0c62016-04-02 02:22:40 +0200557 LOG_F("fuzz_prepareFileDynamically() failed");
Robert Swieckia96d78d2016-03-14 16:50:50 +0100558 }
Robert Swiecki6d01e822016-03-15 20:21:46 +0100559 } else if (state == _HF_STATE_DYNAMIC_PRE) {
Robert Swieckidecf14b2016-03-31 15:09:28 +0200560 fuzzer->flipRate = 0.0f;
561 if (!fuzz_prepareFile(hfuzz, fuzzer, rnd_index)) {
Jagger1d8e0c62016-04-02 02:22:40 +0200562 LOG_F("fuzz_prepareFile() failed");
Jagger190f0dc2015-09-05 16:41:22 +0200563 }
Jagger190f0dc2015-09-05 16:41:22 +0200564 } else {
Robert Swieckidecf14b2016-03-31 15:09:28 +0200565 if (!fuzz_prepareFile(hfuzz, fuzzer, rnd_index)) {
Jagger1d8e0c62016-04-02 02:22:40 +0200566 LOG_F("fuzz_prepareFile() failed");
Jagger190f0dc2015-09-05 16:41:22 +0200567 }
568 }
569
Robert Swieckidecf14b2016-03-31 15:09:28 +0200570 fuzzer->pid = fuzzer->persistentPid;
571 if (fuzzer->pid == 0) {
Robert Swiecki12800cd2016-03-31 15:38:10 +0200572 fuzzer->pid = arch_fork(hfuzz, fuzzer);
Robert Swieckidecf14b2016-03-31 15:09:28 +0200573 if (fuzzer->pid == -1) {
Robert Swiecki0f937af2016-03-30 18:19:16 +0200574 PLOG_F("Couldn't fork");
Jaggerbb83e7c2016-03-10 04:33:40 +0100575 }
Robert Swiecki0f937af2016-03-30 18:19:16 +0200576
Robert Swieckidecf14b2016-03-31 15:09:28 +0200577 if (!fuzzer->pid) {
578 if (!fuzz_prepareExecve(hfuzz, fuzzer->fileName)) {
Robert Swiecki0f937af2016-03-30 18:19:16 +0200579 LOG_E("fuzz_prepareExecve() failed");
580 exit(EXIT_FAILURE);
581 }
Robert Swieckidecf14b2016-03-31 15:09:28 +0200582 if (!arch_launchChild(hfuzz, fuzzer->fileName)) {
Robert Swiecki0f937af2016-03-30 18:19:16 +0200583 LOG_E("Error launching child process");
584 exit(EXIT_FAILURE);
585 }
586 }
Robert Swiecki1ae5eab2016-03-31 15:17:46 +0200587
588 if (hfuzz->persistent) {
Robert Swiecki0e673882016-03-31 18:50:29 +0200589 LOG_I("Persistent mode: Launched new persistent PID: %d", (int)fuzzer->pid);
Robert Swiecki12800cd2016-03-31 15:38:10 +0200590 fuzzer->persistentPid = fuzzer->pid;
Robert Swiecki1ae5eab2016-03-31 15:17:46 +0200591 }
Jagger190f0dc2015-09-05 16:41:22 +0200592 }
593
Robert Swieckidecf14b2016-03-31 15:09:28 +0200594 LOG_D("Launched new process, pid: %d, (concurrency: %zd)", fuzzer->pid, hfuzz->threadsMax);
Jagger190f0dc2015-09-05 16:41:22 +0200595
Robert Swieckidecf14b2016-03-31 15:09:28 +0200596 arch_reapChild(hfuzz, fuzzer);
Jaggera34b3022016-04-02 02:20:40 +0200597 if (hfuzz->persistent == false) {
598 unlink(fuzzer->fileName);
599 }
Jagger190f0dc2015-09-05 16:41:22 +0200600
601 if (hfuzz->dynFileMethod != _HF_DYNFILE_NONE) {
Robert Swieckidecf14b2016-03-31 15:09:28 +0200602 fuzz_perfFeedback(hfuzz, fuzzer);
Jaggerabb6ae02016-03-18 01:55:51 +0100603 }
604 if (hfuzz->useSanCov) {
Robert Swieckidecf14b2016-03-31 15:09:28 +0200605 fuzz_sanCovFeedback(hfuzz, fuzzer);
Robert Swiecki0f937af2016-03-30 18:19:16 +0200606 }
Jagger190f0dc2015-09-05 16:41:22 +0200607
Robert Swieckidecf14b2016-03-31 15:09:28 +0200608 if (hfuzz->useVerifier && (fuzzer->crashFileName[0] != 0) && fuzzer->backtrace) {
609 if (!fuzz_runVerifier(hfuzz, fuzzer)) {
610 LOG_I("Failed to verify %s", fuzzer->crashFileName);
Anestis Bechtsoudis3085f222015-09-27 18:35:26 +0300611 }
Anestis Bechtsoudis5c86ebc2015-09-27 18:06:43 +0300612 }
613
Robert Swieckidecf14b2016-03-31 15:09:28 +0200614 report_Report(hfuzz, fuzzer->report);
Robert Swiecki05354ca2016-03-15 19:10:23 +0100615
Jaggerd34417d2016-03-16 01:26:54 +0100616 if (state == _HF_STATE_DYNAMIC_PRE && ATOMIC_PRE_INC(hfuzz->doneFileIndex) >= hfuzz->fileCnt) {
Robert Swiecki6d01e822016-03-15 20:21:46 +0100617 fuzz_setState(hfuzz, _HF_STATE_DYNAMIC_MAIN);
Robert Swiecki05354ca2016-03-15 19:10:23 +0100618 }
Jagger190f0dc2015-09-05 16:41:22 +0200619}
620
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000621static void *fuzz_threadNew(void *arg)
robert.swiecki3bb518c2010-10-14 00:48:24 +0000622{
Robert Swiecki0e673882016-03-31 18:50:29 +0200623 LOG_I("Launched new fuzzing thread");
624
robert.swiecki@gmail.com882900b2015-02-11 13:56:22 +0000625 honggfuzz_t *hfuzz = (honggfuzz_t *) arg;
Anestis Bechtsoudis02b99be2015-12-27 11:53:01 +0200626
Robert Swieckidecf14b2016-03-31 15:09:28 +0200627 fuzzer_t fuzzer = {
628 .pid = 0,
629 .persistentPid = 0,
630 .dynamicFile = util_Malloc(hfuzz->maxFileSz),
Robert Swiecki12800cd2016-03-31 15:38:10 +0200631
Robert Swiecki9f9968a2016-04-01 14:39:51 +0200632#if defined(_HF_ARCH_LINUX)
Robert Swieckie33c90b2016-03-31 17:32:56 +0200633 .linux.timerId = (timer_t) 0,
Robert Swiecki9f9968a2016-04-01 14:39:51 +0200634#endif // defined(_HF_ARCH_LINUX)
635 .linux.attachedPid = 0,.linux.persistentSock = -1,
Robert Swieckidecf14b2016-03-31 15:09:28 +0200636 };
637 DEFER(free(fuzzer.dynamicFile));
638
639 if (arch_archThreadInit(hfuzz, &fuzzer) == false) {
Robert Swiecki0f937af2016-03-30 18:19:16 +0200640 LOG_F("Could not initialize the thread");
641 }
642
Robert Swieckia96d78d2016-03-14 16:50:50 +0100643 for (;;) {
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200644 /* Check if dry run mode with verifier enabled */
Robert Swieckia96d78d2016-03-14 16:50:50 +0100645 if (hfuzz->origFlipRate == 0.0L && hfuzz->useVerifier) {
Jaggerd34417d2016-03-16 01:26:54 +0100646 if (ATOMIC_POST_INC(hfuzz->mutationsCnt) >= hfuzz->fileCnt) {
647 ATOMIC_POST_INC(hfuzz->threadsFinished);
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200648 // All files checked, weak-up the main process
649 pthread_kill(fuzz_mainThread, SIGALRM);
650 return NULL;
651 }
652 }
653 /* Check for max iterations limit if set */
Jaggerd34417d2016-03-16 01:26:54 +0100654 else if ((ATOMIC_POST_INC(hfuzz->mutationsCnt) >= hfuzz->mutationsMax)
Anestis Bechtsoudis46ea10e2015-11-07 18:16:25 +0200655 && hfuzz->mutationsMax) {
Jaggerd34417d2016-03-16 01:26:54 +0100656 ATOMIC_POST_INC(hfuzz->threadsFinished);
Robert Swiecki81c6a0d2015-09-08 15:43:20 +0200657 // Wake-up the main process
Jagger6113c432015-09-24 05:00:28 +0200658 pthread_kill(fuzz_mainThread, SIGALRM);
Jaggerea39a8f2015-09-05 00:57:22 +0200659 return NULL;
660 }
robert.swiecki3bb518c2010-10-14 00:48:24 +0000661
Robert Swieckidecf14b2016-03-31 15:09:28 +0200662 fuzz_fuzzLoop(hfuzz, &fuzzer);
robert.swiecki@gmail.comd4dd4df2015-02-18 00:50:12 +0000663 }
robert.swiecki@gmail.com882900b2015-02-11 13:56:22 +0000664}
665
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000666static void fuzz_runThread(honggfuzz_t * hfuzz, void *(*thread) (void *))
667{
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000668 pthread_attr_t attr;
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000669
robert.swiecki@gmail.com6f5c2392015-02-16 18:13:09 +0000670 pthread_attr_init(&attr);
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000671 pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
robert.swiecki@gmail.com441089a2015-02-23 13:14:07 +0000672 pthread_attr_setstacksize(&attr, _HF_PTHREAD_STACKSIZE);
robert.swiecki@gmail.com011981f2015-02-17 19:06:44 +0000673 pthread_attr_setguardsize(&attr, (size_t) sysconf(_SC_PAGESIZE));
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000674
robert.swiecki@gmail.com01b6dd42015-02-16 18:11:28 +0000675 pthread_t t;
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000676 if (pthread_create(&t, &attr, thread, (void *)hfuzz) < 0) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200677 PLOG_F("Couldn't create a new thread");
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000678 }
679
680 return;
681}
682
Jagger08816fd2016-03-11 01:32:38 +0100683void fuzz_threads(honggfuzz_t * hfuzz)
robert.swiecki3bb518c2010-10-14 00:48:24 +0000684{
Jagger6113c432015-09-24 05:00:28 +0200685 fuzz_mainThread = pthread_self();
686
robert.swiecki@gmail.com956276a2015-04-16 16:51:52 +0000687 if (!arch_archInit(hfuzz)) {
Robert Swieckic8c32db2015-10-09 18:06:22 +0200688 LOG_F("Couldn't prepare arch for fuzzing");
robert.swiecki@gmail.comef829fa2011-06-22 13:51:57 +0000689 }
Jagger00265602016-03-10 02:36:27 +0100690 if (!sancov_Init(hfuzz)) {
691 LOG_F("Couldn't prepare sancov options");
692 }
robert.swiecki@gmail.comef829fa2011-06-22 13:51:57 +0000693
Robert Swieckia96d78d2016-03-14 16:50:50 +0100694 if (hfuzz->useSanCov || hfuzz->dynFileMethod != _HF_DYNFILE_NONE) {
Jaggerebe19292016-03-17 00:11:44 +0100695 fuzz_setState(hfuzz, _HF_STATE_DYNAMIC_PRE);
Robert Swieckia96d78d2016-03-14 16:50:50 +0100696 } else {
Jaggerebe19292016-03-17 00:11:44 +0100697 fuzz_setState(hfuzz, _HF_STATE_STATIC);
Robert Swieckia96d78d2016-03-14 16:50:50 +0100698 }
699
Jaggerea39a8f2015-09-05 00:57:22 +0200700 for (size_t i = 0; i < hfuzz->threadsMax; i++) {
robert.swiecki@gmail.comc8443142015-02-13 13:46:40 +0000701 fuzz_runThread(hfuzz, fuzz_threadNew);
robert.swiecki3bb518c2010-10-14 00:48:24 +0000702 }
robert.swiecki3bb518c2010-10-14 00:48:24 +0000703}