[libFuzzer] make caller-callee feedback work with trace-pc-guard
llvm-svn: 281667
diff --git a/llvm/lib/Fuzzer/FuzzerTracePC.cpp b/llvm/lib/Fuzzer/FuzzerTracePC.cpp
index b017692..f729a5e 100644
--- a/llvm/lib/Fuzzer/FuzzerTracePC.cpp
+++ b/llvm/lib/Fuzzer/FuzzerTracePC.cpp
@@ -73,6 +73,12 @@
return Delta;
}
+void TracePC::HandleCallerCallee(uintptr_t Caller, uintptr_t Callee) {
+ const uintptr_t kBits = 12;
+ const uintptr_t kMask = (1 << kBits) - 1;
+ CounterMap.AddValue((Caller & kMask) | ((Callee & kMask) << kBits));
+}
+
} // namespace fuzzer
extern "C" {
@@ -86,4 +92,10 @@
void __sanitizer_cov_trace_pc_guard_init(uint8_t *Start, uint8_t *Stop) {
fuzzer::TPC.HandleInit(Start, Stop);
}
+
+__attribute__((visibility("default")))
+void __sanitizer_cov_trace_pc_indir(uintptr_t Callee) {
+ uintptr_t PC = (uintptr_t)__builtin_return_address(0);
+ fuzzer::TPC.HandleCallerCallee(PC, Callee);
+}
}