blob: 20230d496e9c506a7e395da110ceb45c85021a4b [file] [log] [blame]
george.karpenkov29efa6d2017-08-21 23:25:50 +00001//===- FuzzerTracePC.cpp - PC tracing--------------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9// Trace PCs.
10// This module implements __sanitizer_cov_trace_pc_guard[_init],
11// the callback required for -fsanitize-coverage=trace-pc-guard instrumentation.
12//
13//===----------------------------------------------------------------------===//
14
15#include "FuzzerTracePC.h"
16#include "FuzzerCorpus.h"
17#include "FuzzerDefs.h"
18#include "FuzzerDictionary.h"
19#include "FuzzerExtFunctions.h"
20#include "FuzzerIO.h"
21#include "FuzzerUtil.h"
22#include "FuzzerValueBitMap.h"
23#include <set>
24
25// The coverage counters and PCs.
26// These are declared as global variables named "__sancov_*" to simplify
27// experiments with inlined instrumentation.
28alignas(64) ATTRIBUTE_INTERFACE
29uint8_t __sancov_trace_pc_guard_8bit_counters[fuzzer::TracePC::kNumPCs];
30
31ATTRIBUTE_INTERFACE
32uintptr_t __sancov_trace_pc_pcs[fuzzer::TracePC::kNumPCs];
33
kcc1c0379f2017-08-22 01:28:32 +000034// Used by -fsanitize-coverage=stack-depth to track stack depth
morehouse398297f2017-08-22 21:28:29 +000035ATTRIBUTE_INTERFACE __attribute__((tls_model("initial-exec")))
36thread_local uintptr_t __sancov_lowest_stack;
kcc1c0379f2017-08-22 01:28:32 +000037
george.karpenkov29efa6d2017-08-21 23:25:50 +000038namespace fuzzer {
39
40TracePC TPC;
41
42int ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr;
43
44uint8_t *TracePC::Counters() const {
45 return __sancov_trace_pc_guard_8bit_counters;
46}
47
48uintptr_t *TracePC::PCs() const {
49 return __sancov_trace_pc_pcs;
50}
51
52size_t TracePC::GetTotalPCCoverage() {
53 if (ObservedPCs.size())
54 return ObservedPCs.size();
55 size_t Res = 0;
56 for (size_t i = 1, N = GetNumPCs(); i < N; i++)
57 if (PCs()[i])
58 Res++;
59 return Res;
60}
61
62
63void TracePC::HandleInline8bitCountersInit(uint8_t *Start, uint8_t *Stop) {
64 if (Start == Stop) return;
65 if (NumModulesWithInline8bitCounters &&
66 ModuleCounters[NumModulesWithInline8bitCounters-1].Start == Start) return;
67 assert(NumModulesWithInline8bitCounters <
68 sizeof(ModuleCounters) / sizeof(ModuleCounters[0]));
69 ModuleCounters[NumModulesWithInline8bitCounters++] = {Start, Stop};
70 NumInline8bitCounters += Stop - Start;
71}
72
kcc98957a12017-08-25 19:29:47 +000073void TracePC::HandlePCsInit(const uintptr_t *Start, const uintptr_t *Stop) {
74 const PCTableEntry *B = reinterpret_cast<const PCTableEntry *>(Start);
75 const PCTableEntry *E = reinterpret_cast<const PCTableEntry *>(Stop);
george.karpenkov29efa6d2017-08-21 23:25:50 +000076 if (NumPCTables && ModulePCTable[NumPCTables - 1].Start == B) return;
77 assert(NumPCTables < sizeof(ModulePCTable) / sizeof(ModulePCTable[0]));
78 ModulePCTable[NumPCTables++] = {B, E};
79 NumPCsInPCTables += E - B;
80}
81
82void TracePC::HandleInit(uint32_t *Start, uint32_t *Stop) {
83 if (Start == Stop || *Start) return;
84 assert(NumModules < sizeof(Modules) / sizeof(Modules[0]));
85 for (uint32_t *P = Start; P < Stop; P++) {
86 NumGuards++;
87 if (NumGuards == kNumPCs) {
88 RawPrint(
89 "WARNING: The binary has too many instrumented PCs.\n"
90 " You may want to reduce the size of the binary\n"
91 " for more efficient fuzzing and precise coverage data\n");
92 }
93 *P = NumGuards % kNumPCs;
94 }
95 Modules[NumModules].Start = Start;
96 Modules[NumModules].Stop = Stop;
97 NumModules++;
98}
99
100void TracePC::PrintModuleInfo() {
101 if (NumGuards) {
102 Printf("INFO: Loaded %zd modules (%zd guards): ", NumModules, NumGuards);
103 for (size_t i = 0; i < NumModules; i++)
104 Printf("%zd [%p, %p), ", Modules[i].Stop - Modules[i].Start,
105 Modules[i].Start, Modules[i].Stop);
106 Printf("\n");
107 }
108 if (NumModulesWithInline8bitCounters) {
109 Printf("INFO: Loaded %zd modules (%zd inline 8-bit counters): ",
110 NumModulesWithInline8bitCounters, NumInline8bitCounters);
111 for (size_t i = 0; i < NumModulesWithInline8bitCounters; i++)
112 Printf("%zd [%p, %p), ", ModuleCounters[i].Stop - ModuleCounters[i].Start,
113 ModuleCounters[i].Start, ModuleCounters[i].Stop);
114 Printf("\n");
115 }
116 if (NumPCTables) {
117 Printf("INFO: Loaded %zd PC tables (%zd PCs): ", NumPCTables,
118 NumPCsInPCTables);
119 for (size_t i = 0; i < NumPCTables; i++) {
120 Printf("%zd [%p,%p), ", ModulePCTable[i].Stop - ModulePCTable[i].Start,
121 ModulePCTable[i].Start, ModulePCTable[i].Stop);
122 }
123 Printf("\n");
124
125 if ((NumGuards && NumGuards != NumPCsInPCTables) ||
126 (NumInline8bitCounters && NumInline8bitCounters != NumPCsInPCTables)) {
kcce220ebb2017-10-14 00:07:11 +0000127 Printf("ERROR: The size of coverage PC tables does not match the\n"
128 "number of instrumented PCs. This might be a compiler bug,\n"
129 "please contact the libFuzzer developers.\n"
130 "Also check https://bugs.llvm.org/show_bug.cgi?id=34636\n"
131 "for possible workarounds (tl;dr: don't use the old GNU ld)\n");
george.karpenkov29efa6d2017-08-21 23:25:50 +0000132 _Exit(1);
133 }
134 }
delcypherfcd19a82018-04-20 06:46:19 +0000135 if (size_t NumExtraCounters = ExtraCountersEnd() - ExtraCountersBegin())
136 Printf("INFO: %zd Extra Counters\n", NumExtraCounters);
george.karpenkov29efa6d2017-08-21 23:25:50 +0000137}
138
139ATTRIBUTE_NO_SANITIZE_ALL
140void TracePC::HandleCallerCallee(uintptr_t Caller, uintptr_t Callee) {
141 const uintptr_t kBits = 12;
142 const uintptr_t kMask = (1 << kBits) - 1;
143 uintptr_t Idx = (Caller & kMask) | ((Callee & kMask) << kBits);
144 ValueProfileMap.AddValueModPrime(Idx);
145}
146
147void TracePC::UpdateObservedPCs() {
kccec9da662017-08-28 22:52:22 +0000148 Vector<uintptr_t> CoveredFuncs;
kcc00da6482017-08-25 20:09:25 +0000149 auto ObservePC = [&](uintptr_t PC) {
150 if (ObservedPCs.insert(PC).second && DoPrintNewPCs)
kcc1c0379f2017-08-22 01:28:32 +0000151 PrintPC("\tNEW_PC: %p %F %L\n", "\tNEW_PC: %p\n", PC + 1);
152 };
kcc00da6482017-08-25 20:09:25 +0000153
154 auto Observe = [&](const PCTableEntry &TE) {
155 if (TE.PCFlags & 1)
kccec9da662017-08-28 22:52:22 +0000156 if (ObservedFuncs.insert(TE.PC).second && NumPrintNewFuncs)
157 CoveredFuncs.push_back(TE.PC);
kcc00da6482017-08-25 20:09:25 +0000158 ObservePC(TE.PC);
159 };
160
george.karpenkov29efa6d2017-08-21 23:25:50 +0000161 if (NumPCsInPCTables) {
george.karpenkov29efa6d2017-08-21 23:25:50 +0000162 if (NumInline8bitCounters == NumPCsInPCTables) {
163 for (size_t i = 0; i < NumModulesWithInline8bitCounters; i++) {
164 uint8_t *Beg = ModuleCounters[i].Start;
165 size_t Size = ModuleCounters[i].Stop - Beg;
166 assert(Size ==
167 (size_t)(ModulePCTable[i].Stop - ModulePCTable[i].Start));
168 for (size_t j = 0; j < Size; j++)
169 if (Beg[j])
kcc00da6482017-08-25 20:09:25 +0000170 Observe(ModulePCTable[i].Start[j]);
george.karpenkov29efa6d2017-08-21 23:25:50 +0000171 }
172 } else if (NumGuards == NumPCsInPCTables) {
173 size_t GuardIdx = 1;
174 for (size_t i = 0; i < NumModules; i++) {
175 uint32_t *Beg = Modules[i].Start;
176 size_t Size = Modules[i].Stop - Beg;
177 assert(Size ==
178 (size_t)(ModulePCTable[i].Stop - ModulePCTable[i].Start));
179 for (size_t j = 0; j < Size; j++, GuardIdx++)
180 if (Counters()[GuardIdx])
kcc00da6482017-08-25 20:09:25 +0000181 Observe(ModulePCTable[i].Start[j]);
george.karpenkov29efa6d2017-08-21 23:25:50 +0000182 }
183 }
184 }
kccec9da662017-08-28 22:52:22 +0000185
186 for (size_t i = 0, N = Min(CoveredFuncs.size(), NumPrintNewFuncs); i < N; i++) {
187 Printf("\tNEW_FUNC[%zd/%zd]: ", i, CoveredFuncs.size());
188 PrintPC("%p %F %L\n", "%p\n", CoveredFuncs[i] + 1);
189 }
george.karpenkov29efa6d2017-08-21 23:25:50 +0000190}
191
192inline ALWAYS_INLINE uintptr_t GetPreviousInstructionPc(uintptr_t PC) {
193 // TODO: this implementation is x86 only.
194 // see sanitizer_common GetPreviousInstructionPc for full implementation.
195 return PC - 1;
196}
197
198inline ALWAYS_INLINE uintptr_t GetNextInstructionPc(uintptr_t PC) {
199 // TODO: this implementation is x86 only.
200 // see sanitizer_common GetPreviousInstructionPc for full implementation.
201 return PC + 1;
202}
203
204static std::string GetModuleName(uintptr_t PC) {
205 char ModulePathRaw[4096] = ""; // What's PATH_MAX in portable C++?
206 void *OffsetRaw = nullptr;
207 if (!EF->__sanitizer_get_module_and_offset_for_pc(
208 reinterpret_cast<void *>(PC), ModulePathRaw,
209 sizeof(ModulePathRaw), &OffsetRaw))
210 return "";
211 return ModulePathRaw;
212}
213
kcc85cad3d2018-05-11 01:17:52 +0000214template<class CallBack>
215void TracePC::IterateCoveredFunctions(CallBack CB) {
216 for (size_t i = 0; i < NumPCTables; i++) {
217 auto &M = ModulePCTable[i];
218 assert(M.Start < M.Stop);
219 auto ModuleName = GetModuleName(M.Start->PC);
220 for (auto NextFE = M.Start; NextFE < M.Stop; ) {
221 auto FE = NextFE;
222 assert((FE->PCFlags & 1) && "Not a function entry point");
223 do {
224 NextFE++;
225 } while (NextFE < M.Stop && !(NextFE->PCFlags & 1));
226 if (ObservedFuncs.count(FE->PC))
227 CB(FE, NextFE);
228 }
229 }
230}
231
george.karpenkov29efa6d2017-08-21 23:25:50 +0000232void TracePC::PrintCoverage() {
233 if (!EF->__sanitizer_symbolize_pc ||
234 !EF->__sanitizer_get_module_and_offset_for_pc) {
235 Printf("INFO: __sanitizer_symbolize_pc or "
236 "__sanitizer_get_module_and_offset_for_pc is not available,"
237 " not printing coverage\n");
238 return;
239 }
240 Printf("COVERAGE:\n");
kcc85cad3d2018-05-11 01:17:52 +0000241 auto CoveredFunctionCallback = [&](const PCTableEntry *First, const PCTableEntry *Last) {
242 assert(First < Last);
243 auto VisualizePC = GetNextInstructionPc(First->PC);
244 std::string FileStr = DescribePC("%s", VisualizePC);
245 if (!IsInterestingCoverageFile(FileStr)) return;
246 std::string FunctionStr = DescribePC("%F", VisualizePC);
247 std::string LineStr = DescribePC("%l", VisualizePC);
248 size_t Line = std::stoul(LineStr);
249 std::vector<uintptr_t> UncoveredPCs;
250 for (auto TE = First; TE < Last; TE++)
251 if (!ObservedPCs.count(TE->PC))
252 UncoveredPCs.push_back(TE->PC);
253 Printf("COVERED_FUNC: ");
254 UncoveredPCs.empty()
255 ? Printf("all")
256 : Printf("%zd/%zd", (Last - First) - UncoveredPCs.size(), Last - First);
257 Printf(" PCs covered %s %s:%zd\n", FunctionStr.c_str(), FileStr.c_str(),
258 Line);
259 for (auto PC: UncoveredPCs) {
260 Printf(" UNCOVERED_PC: %s\n",
261 DescribePC("%s:%l", GetNextInstructionPc(PC)).c_str());
george.karpenkov29efa6d2017-08-21 23:25:50 +0000262 }
263 };
264
kcc85cad3d2018-05-11 01:17:52 +0000265 IterateCoveredFunctions(CoveredFunctionCallback);
george.karpenkov29efa6d2017-08-21 23:25:50 +0000266}
267
george.karpenkov29efa6d2017-08-21 23:25:50 +0000268// Value profile.
269// We keep track of various values that affect control flow.
270// These values are inserted into a bit-set-based hash map.
271// Every new bit in the map is treated as a new coverage.
272//
273// For memcmp/strcmp/etc the interesting value is the length of the common
274// prefix of the parameters.
275// For cmp instructions the interesting value is a XOR of the parameters.
276// The interesting value is mixed up with the PC and is then added to the map.
277
278ATTRIBUTE_NO_SANITIZE_ALL
279void TracePC::AddValueForMemcmp(void *caller_pc, const void *s1, const void *s2,
280 size_t n, bool StopAtZero) {
281 if (!n) return;
282 size_t Len = std::min(n, Word::GetMaxSize());
283 const uint8_t *A1 = reinterpret_cast<const uint8_t *>(s1);
284 const uint8_t *A2 = reinterpret_cast<const uint8_t *>(s2);
285 uint8_t B1[Word::kMaxSize];
286 uint8_t B2[Word::kMaxSize];
287 // Copy the data into locals in this non-msan-instrumented function
288 // to avoid msan complaining further.
289 size_t Hash = 0; // Compute some simple hash of both strings.
290 for (size_t i = 0; i < Len; i++) {
291 B1[i] = A1[i];
292 B2[i] = A2[i];
293 size_t T = B1[i];
294 Hash ^= (T << 8) | B2[i];
295 }
296 size_t I = 0;
297 for (; I < Len; I++)
298 if (B1[I] != B2[I] || (StopAtZero && B1[I] == 0))
299 break;
300 size_t PC = reinterpret_cast<size_t>(caller_pc);
301 size_t Idx = (PC & 4095) | (I << 12);
302 ValueProfileMap.AddValue(Idx);
303 TORCW.Insert(Idx ^ Hash, Word(B1, Len), Word(B2, Len));
304}
305
306template <class T>
307ATTRIBUTE_TARGET_POPCNT ALWAYS_INLINE
308ATTRIBUTE_NO_SANITIZE_ALL
309void TracePC::HandleCmp(uintptr_t PC, T Arg1, T Arg2) {
310 uint64_t ArgXor = Arg1 ^ Arg2;
311 uint64_t ArgDistance = __builtin_popcountll(ArgXor) + 1; // [1,65]
312 uintptr_t Idx = ((PC & 4095) + 1) * ArgDistance;
313 if (sizeof(T) == 4)
314 TORC4.Insert(ArgXor, Arg1, Arg2);
315 else if (sizeof(T) == 8)
316 TORC8.Insert(ArgXor, Arg1, Arg2);
317 ValueProfileMap.AddValue(Idx);
318}
319
320static size_t InternalStrnlen(const char *S, size_t MaxLen) {
321 size_t Len = 0;
322 for (; Len < MaxLen && S[Len]; Len++) {}
323 return Len;
324}
325
326// Finds min of (strlen(S1), strlen(S2)).
327// Needed bacause one of these strings may actually be non-zero terminated.
328static size_t InternalStrnlen2(const char *S1, const char *S2) {
329 size_t Len = 0;
330 for (; S1[Len] && S2[Len]; Len++) {}
331 return Len;
332}
333
334void TracePC::ClearInlineCounters() {
335 for (size_t i = 0; i < NumModulesWithInline8bitCounters; i++) {
336 uint8_t *Beg = ModuleCounters[i].Start;
337 size_t Size = ModuleCounters[i].Stop - Beg;
338 memset(Beg, 0, Size);
339 }
340}
341
kcc0f3c0312017-08-22 01:50:00 +0000342ATTRIBUTE_NO_SANITIZE_ALL
kcc1c0379f2017-08-22 01:28:32 +0000343void TracePC::RecordInitialStack() {
kcc0f3c0312017-08-22 01:50:00 +0000344 int stack;
345 __sancov_lowest_stack = InitialStack = reinterpret_cast<uintptr_t>(&stack);
kcc1c0379f2017-08-22 01:28:32 +0000346}
347
348uintptr_t TracePC::GetMaxStackOffset() const {
349 return InitialStack - __sancov_lowest_stack; // Stack grows down
350}
351
george.karpenkov29efa6d2017-08-21 23:25:50 +0000352} // namespace fuzzer
353
354extern "C" {
355ATTRIBUTE_INTERFACE
356ATTRIBUTE_NO_SANITIZE_ALL
357void __sanitizer_cov_trace_pc_guard(uint32_t *Guard) {
358 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
359 uint32_t Idx = *Guard;
360 __sancov_trace_pc_pcs[Idx] = PC;
361 __sancov_trace_pc_guard_8bit_counters[Idx]++;
george.karpenkov29efa6d2017-08-21 23:25:50 +0000362}
363
364// Best-effort support for -fsanitize-coverage=trace-pc, which is available
365// in both Clang and GCC.
366ATTRIBUTE_INTERFACE
367ATTRIBUTE_NO_SANITIZE_ALL
368void __sanitizer_cov_trace_pc() {
369 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
370 uintptr_t Idx = PC & (((uintptr_t)1 << fuzzer::TracePC::kTracePcBits) - 1);
371 __sancov_trace_pc_pcs[Idx] = PC;
372 __sancov_trace_pc_guard_8bit_counters[Idx]++;
373}
374
375ATTRIBUTE_INTERFACE
376void __sanitizer_cov_trace_pc_guard_init(uint32_t *Start, uint32_t *Stop) {
377 fuzzer::TPC.HandleInit(Start, Stop);
378}
379
380ATTRIBUTE_INTERFACE
381void __sanitizer_cov_8bit_counters_init(uint8_t *Start, uint8_t *Stop) {
382 fuzzer::TPC.HandleInline8bitCountersInit(Start, Stop);
383}
384
385ATTRIBUTE_INTERFACE
kcc98957a12017-08-25 19:29:47 +0000386void __sanitizer_cov_pcs_init(const uintptr_t *pcs_beg,
387 const uintptr_t *pcs_end) {
george.karpenkov29efa6d2017-08-21 23:25:50 +0000388 fuzzer::TPC.HandlePCsInit(pcs_beg, pcs_end);
389}
390
391ATTRIBUTE_INTERFACE
392ATTRIBUTE_NO_SANITIZE_ALL
393void __sanitizer_cov_trace_pc_indir(uintptr_t Callee) {
394 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
395 fuzzer::TPC.HandleCallerCallee(PC, Callee);
396}
397
398ATTRIBUTE_INTERFACE
399ATTRIBUTE_NO_SANITIZE_ALL
400ATTRIBUTE_TARGET_POPCNT
401void __sanitizer_cov_trace_cmp8(uint64_t Arg1, uint64_t Arg2) {
402 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
403 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
404}
405
406ATTRIBUTE_INTERFACE
407ATTRIBUTE_NO_SANITIZE_ALL
408ATTRIBUTE_TARGET_POPCNT
409// Now the __sanitizer_cov_trace_const_cmp[1248] callbacks just mimic
410// the behaviour of __sanitizer_cov_trace_cmp[1248] ones. This, however,
411// should be changed later to make full use of instrumentation.
412void __sanitizer_cov_trace_const_cmp8(uint64_t Arg1, uint64_t Arg2) {
413 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
414 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
415}
416
417ATTRIBUTE_INTERFACE
418ATTRIBUTE_NO_SANITIZE_ALL
419ATTRIBUTE_TARGET_POPCNT
420void __sanitizer_cov_trace_cmp4(uint32_t Arg1, uint32_t Arg2) {
421 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
422 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
423}
424
425ATTRIBUTE_INTERFACE
426ATTRIBUTE_NO_SANITIZE_ALL
427ATTRIBUTE_TARGET_POPCNT
428void __sanitizer_cov_trace_const_cmp4(uint32_t Arg1, uint32_t Arg2) {
429 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
430 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
431}
432
433ATTRIBUTE_INTERFACE
434ATTRIBUTE_NO_SANITIZE_ALL
435ATTRIBUTE_TARGET_POPCNT
436void __sanitizer_cov_trace_cmp2(uint16_t Arg1, uint16_t Arg2) {
437 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
438 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
439}
440
441ATTRIBUTE_INTERFACE
442ATTRIBUTE_NO_SANITIZE_ALL
443ATTRIBUTE_TARGET_POPCNT
444void __sanitizer_cov_trace_const_cmp2(uint16_t Arg1, uint16_t Arg2) {
445 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
446 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
447}
448
449ATTRIBUTE_INTERFACE
450ATTRIBUTE_NO_SANITIZE_ALL
451ATTRIBUTE_TARGET_POPCNT
452void __sanitizer_cov_trace_cmp1(uint8_t Arg1, uint8_t Arg2) {
453 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
454 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
455}
456
457ATTRIBUTE_INTERFACE
458ATTRIBUTE_NO_SANITIZE_ALL
459ATTRIBUTE_TARGET_POPCNT
460void __sanitizer_cov_trace_const_cmp1(uint8_t Arg1, uint8_t Arg2) {
461 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
462 fuzzer::TPC.HandleCmp(PC, Arg1, Arg2);
463}
464
465ATTRIBUTE_INTERFACE
466ATTRIBUTE_NO_SANITIZE_ALL
467ATTRIBUTE_TARGET_POPCNT
468void __sanitizer_cov_trace_switch(uint64_t Val, uint64_t *Cases) {
469 uint64_t N = Cases[0];
470 uint64_t ValSizeInBits = Cases[1];
471 uint64_t *Vals = Cases + 2;
472 // Skip the most common and the most boring case.
473 if (Vals[N - 1] < 256 && Val < 256)
474 return;
475 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
476 size_t i;
477 uint64_t Token = 0;
478 for (i = 0; i < N; i++) {
479 Token = Val ^ Vals[i];
480 if (Val < Vals[i])
481 break;
482 }
483
484 if (ValSizeInBits == 16)
485 fuzzer::TPC.HandleCmp(PC + i, static_cast<uint16_t>(Token), (uint16_t)(0));
486 else if (ValSizeInBits == 32)
487 fuzzer::TPC.HandleCmp(PC + i, static_cast<uint32_t>(Token), (uint32_t)(0));
488 else
489 fuzzer::TPC.HandleCmp(PC + i, Token, (uint64_t)(0));
490}
491
492ATTRIBUTE_INTERFACE
493ATTRIBUTE_NO_SANITIZE_ALL
494ATTRIBUTE_TARGET_POPCNT
495void __sanitizer_cov_trace_div4(uint32_t Val) {
496 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
497 fuzzer::TPC.HandleCmp(PC, Val, (uint32_t)0);
498}
499
500ATTRIBUTE_INTERFACE
501ATTRIBUTE_NO_SANITIZE_ALL
502ATTRIBUTE_TARGET_POPCNT
503void __sanitizer_cov_trace_div8(uint64_t Val) {
504 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
505 fuzzer::TPC.HandleCmp(PC, Val, (uint64_t)0);
506}
507
508ATTRIBUTE_INTERFACE
509ATTRIBUTE_NO_SANITIZE_ALL
510ATTRIBUTE_TARGET_POPCNT
511void __sanitizer_cov_trace_gep(uintptr_t Idx) {
512 uintptr_t PC = reinterpret_cast<uintptr_t>(__builtin_return_address(0));
513 fuzzer::TPC.HandleCmp(PC, Idx, (uintptr_t)0);
514}
515
516ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
517void __sanitizer_weak_hook_memcmp(void *caller_pc, const void *s1,
518 const void *s2, size_t n, int result) {
519 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
520 if (result == 0) return; // No reason to mutate.
521 if (n <= 1) return; // Not interesting.
522 fuzzer::TPC.AddValueForMemcmp(caller_pc, s1, s2, n, /*StopAtZero*/false);
523}
524
525ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
526void __sanitizer_weak_hook_strncmp(void *caller_pc, const char *s1,
527 const char *s2, size_t n, int result) {
528 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
529 if (result == 0) return; // No reason to mutate.
530 size_t Len1 = fuzzer::InternalStrnlen(s1, n);
531 size_t Len2 = fuzzer::InternalStrnlen(s2, n);
532 n = std::min(n, Len1);
533 n = std::min(n, Len2);
534 if (n <= 1) return; // Not interesting.
535 fuzzer::TPC.AddValueForMemcmp(caller_pc, s1, s2, n, /*StopAtZero*/true);
536}
537
538ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
539void __sanitizer_weak_hook_strcmp(void *caller_pc, const char *s1,
540 const char *s2, int result) {
541 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
542 if (result == 0) return; // No reason to mutate.
543 size_t N = fuzzer::InternalStrnlen2(s1, s2);
544 if (N <= 1) return; // Not interesting.
545 fuzzer::TPC.AddValueForMemcmp(caller_pc, s1, s2, N, /*StopAtZero*/true);
546}
547
548ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
549void __sanitizer_weak_hook_strncasecmp(void *called_pc, const char *s1,
550 const char *s2, size_t n, int result) {
551 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
552 return __sanitizer_weak_hook_strncmp(called_pc, s1, s2, n, result);
553}
554
555ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
556void __sanitizer_weak_hook_strcasecmp(void *called_pc, const char *s1,
557 const char *s2, int result) {
558 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
559 return __sanitizer_weak_hook_strcmp(called_pc, s1, s2, result);
560}
561
562ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
563void __sanitizer_weak_hook_strstr(void *called_pc, const char *s1,
564 const char *s2, char *result) {
565 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
566 fuzzer::TPC.MMT.Add(reinterpret_cast<const uint8_t *>(s2), strlen(s2));
567}
568
569ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
570void __sanitizer_weak_hook_strcasestr(void *called_pc, const char *s1,
571 const char *s2, char *result) {
572 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
573 fuzzer::TPC.MMT.Add(reinterpret_cast<const uint8_t *>(s2), strlen(s2));
574}
575
576ATTRIBUTE_INTERFACE ATTRIBUTE_NO_SANITIZE_MEMORY
577void __sanitizer_weak_hook_memmem(void *called_pc, const void *s1, size_t len1,
578 const void *s2, size_t len2, void *result) {
579 if (fuzzer::ScopedDoingMyOwnMemOrStr::DoingMyOwnMemOrStr) return;
580 fuzzer::TPC.MMT.Add(reinterpret_cast<const uint8_t *>(s2), len2);
581}
582} // extern "C"