Index: llvm/trunk/tools/sancov/sancov.cc =================================================================== --- llvm/trunk/tools/sancov/sancov.cc +++ llvm/trunk/tools/sancov/sancov.cc @@ -207,21 +207,30 @@ return result; } -// Locate __sanitizer_cov function address. -static uint64_t findSanitizerCovFunction(const object::ObjectFile &O) { +// Locate __sanitizer_cov* function addresses that are used for coverage +// reporting. +static std::set +findSanitizerCovFunctions(const object::ObjectFile &O) { + std::set Result; + for (const object::SymbolRef &Symbol : O.symbols()) { ErrorOr AddressOrErr = Symbol.getAddress(); FailIfError(AddressOrErr); - ErrorOr Name = Symbol.getName(); - FailIfError(Name); - - if (Name.get() == "__sanitizer_cov") { - return AddressOrErr.get(); + ErrorOr NameOrErr = Symbol.getName(); + FailIfError(NameOrErr); + StringRef Name = NameOrErr.get(); + + if (Name == "__sanitizer_cov" || Name == "__sanitizer_cov_with_check" || + Name == "__sanitizer_cov_trace_func_enter") { + Result.insert(AddressOrErr.get()); } } - FailIfNotEmpty("__sanitizer_cov not found"); - return 0; // not reachable. + + if (Result.empty()) + FailIfNotEmpty("__sanitizer_cov* functions not found"); + + return Result; } // Locate addresses of all coverage points in a file. Coverage point @@ -262,7 +271,7 @@ TheTarget->createMCInstrAnalysis(MII.get())); FailIfEmpty(MIA, "no instruction analysis info for target " + TripleName); - uint64_t SanCovAddr = findSanitizerCovFunction(O); + auto SanCovAddrs = findSanitizerCovFunctions(O); for (const auto Section : O.sections()) { if (Section.isVirtual() || !Section.isText()) // llvm-objdump does the same. @@ -292,7 +301,7 @@ uint64_t Target; if (MIA->isCall(Inst) && MIA->evaluateBranch(Inst, SectionAddr + Index, Size, Target)) { - if (Target == SanCovAddr) { + if (SanCovAddrs.find(Target) != SanCovAddrs.end()) { // Sanitizer coverage uses the address of the next instruction - 1. Addrs->insert(Index + SectionAddr + Size - 1); }