[sanitizer-coverage] nuke more stale code
llvm-svn: 304504
This commit is contained in:
parent
c1a56baa3e
commit
1800814b41
|
@ -19,8 +19,6 @@
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Initialize coverage.
|
|
||||||
void __sanitizer_cov_init();
|
|
||||||
// Record and dump coverage info.
|
// Record and dump coverage info.
|
||||||
void __sanitizer_cov_dump();
|
void __sanitizer_cov_dump();
|
||||||
|
|
||||||
|
|
|
@ -106,7 +106,6 @@ void AsanDeactivate() {
|
||||||
// Deactivate the runtime.
|
// Deactivate the runtime.
|
||||||
SetCanPoisonMemory(false);
|
SetCanPoisonMemory(false);
|
||||||
SetMallocContextSize(1);
|
SetMallocContextSize(1);
|
||||||
ReInitializeCoverage(false, nullptr);
|
|
||||||
|
|
||||||
AllocatorOptions disabled = asan_deactivated_flags.allocator_options;
|
AllocatorOptions disabled = asan_deactivated_flags.allocator_options;
|
||||||
disabled.quarantine_size_mb = 0;
|
disabled.quarantine_size_mb = 0;
|
||||||
|
@ -130,8 +129,6 @@ void AsanActivate() {
|
||||||
|
|
||||||
SetCanPoisonMemory(asan_deactivated_flags.poison_heap);
|
SetCanPoisonMemory(asan_deactivated_flags.poison_heap);
|
||||||
SetMallocContextSize(asan_deactivated_flags.malloc_context_size);
|
SetMallocContextSize(asan_deactivated_flags.malloc_context_size);
|
||||||
ReInitializeCoverage(asan_deactivated_flags.coverage,
|
|
||||||
asan_deactivated_flags.coverage_dir);
|
|
||||||
ReInitializeAllocator(asan_deactivated_flags.allocator_options);
|
ReInitializeAllocator(asan_deactivated_flags.allocator_options);
|
||||||
|
|
||||||
asan_is_deactivated = false;
|
asan_is_deactivated = false;
|
||||||
|
|
|
@ -722,9 +722,7 @@ static void AfterFork() {
|
||||||
INTERCEPTOR(int, fork, void) {
|
INTERCEPTOR(int, fork, void) {
|
||||||
ENSURE_ASAN_INITED();
|
ENSURE_ASAN_INITED();
|
||||||
BeforeFork();
|
BeforeFork();
|
||||||
if (common_flags()->coverage) CovBeforeFork();
|
|
||||||
int pid = REAL(fork)();
|
int pid = REAL(fork)();
|
||||||
if (common_flags()->coverage) CovAfterFork(pid);
|
|
||||||
AfterFork();
|
AfterFork();
|
||||||
return pid;
|
return pid;
|
||||||
}
|
}
|
||||||
|
|
|
@ -54,7 +54,6 @@ set(SANITIZER_NOLIBC_SOURCES
|
||||||
set(SANITIZER_LIBCDEP_SOURCES
|
set(SANITIZER_LIBCDEP_SOURCES
|
||||||
sanitizer_common_libcdep.cc
|
sanitizer_common_libcdep.cc
|
||||||
sancov_flags.cc
|
sancov_flags.cc
|
||||||
sanitizer_coverage_libcdep.cc
|
|
||||||
sanitizer_coverage_libcdep_new.cc
|
sanitizer_coverage_libcdep_new.cc
|
||||||
sanitizer_coverage_win_sections.cc
|
sanitizer_coverage_win_sections.cc
|
||||||
sanitizer_linux_libcdep.cc
|
sanitizer_linux_libcdep.cc
|
||||||
|
|
|
@ -317,14 +317,9 @@ bool AddressSpaceIsUnlimited();
|
||||||
void SetAddressSpaceUnlimited();
|
void SetAddressSpaceUnlimited();
|
||||||
void AdjustStackSize(void *attr);
|
void AdjustStackSize(void *attr);
|
||||||
void PrepareForSandboxing(__sanitizer_sandbox_arguments *args);
|
void PrepareForSandboxing(__sanitizer_sandbox_arguments *args);
|
||||||
void CovPrepareForSandboxing(__sanitizer_sandbox_arguments *args);
|
|
||||||
void SetSandboxingCallback(void (*f)());
|
void SetSandboxingCallback(void (*f)());
|
||||||
|
|
||||||
void CovBeforeFork();
|
|
||||||
void CovAfterFork(int child_pid);
|
|
||||||
|
|
||||||
void InitializeCoverage(bool enabled, const char *coverage_dir);
|
void InitializeCoverage(bool enabled, const char *coverage_dir);
|
||||||
void ReInitializeCoverage(bool enabled, const char *coverage_dir);
|
|
||||||
|
|
||||||
void InitTlsSize();
|
void InitTlsSize();
|
||||||
uptr GetTlsSize();
|
uptr GetTlsSize();
|
||||||
|
|
|
@ -1,610 +0,0 @@
|
||||||
//===-- sanitizer_coverage.cc ---------------------------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// Sanitizer Coverage.
|
|
||||||
// This file implements run-time support for a poor man's coverage tool.
|
|
||||||
//
|
|
||||||
// Compiler instrumentation:
|
|
||||||
// For every interesting basic block the compiler injects the following code:
|
|
||||||
// if (Guard < 0) {
|
|
||||||
// __sanitizer_cov(&Guard);
|
|
||||||
// }
|
|
||||||
// At the module start up time __sanitizer_cov_module_init sets the guards
|
|
||||||
// to consecutive negative numbers (-1, -2, -3, ...).
|
|
||||||
// It's fine to call __sanitizer_cov more than once for a given block.
|
|
||||||
//
|
|
||||||
// Run-time:
|
|
||||||
// - __sanitizer_cov(): record that we've executed the PC (GET_CALLER_PC).
|
|
||||||
// and atomically set Guard to -Guard.
|
|
||||||
// - __sanitizer_cov_dump: dump the coverage data to disk.
|
|
||||||
// For every module of the current process that has coverage data
|
|
||||||
// this will create a file module_name.PID.sancov.
|
|
||||||
//
|
|
||||||
// The file format is simple: the first 8 bytes is the magic,
|
|
||||||
// one of 0xC0BFFFFFFFFFFF64 and 0xC0BFFFFFFFFFFF32. The last byte of the
|
|
||||||
// magic defines the size of the following offsets.
|
|
||||||
// The rest of the data is the offsets in the module.
|
|
||||||
//
|
|
||||||
// Eventually, this coverage implementation should be obsoleted by a more
|
|
||||||
// powerful general purpose Clang/LLVM coverage instrumentation.
|
|
||||||
// Consider this implementation as prototype.
|
|
||||||
//
|
|
||||||
// FIXME: support (or at least test with) dlclose.
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "sanitizer_allocator_internal.h"
|
|
||||||
#include "sanitizer_common.h"
|
|
||||||
#include "sanitizer_libc.h"
|
|
||||||
#include "sanitizer_mutex.h"
|
|
||||||
#include "sanitizer_procmaps.h"
|
|
||||||
#include "sanitizer_stacktrace.h"
|
|
||||||
#include "sanitizer_symbolizer.h"
|
|
||||||
#include "sanitizer_flags.h"
|
|
||||||
|
|
||||||
using namespace __sanitizer;
|
|
||||||
|
|
||||||
static const u64 kMagic64 = 0xC0BFFFFFFFFFFF64ULL;
|
|
||||||
static const u64 kMagic32 = 0xC0BFFFFFFFFFFF32ULL;
|
|
||||||
static const uptr kNumWordsForMagic = SANITIZER_WORDSIZE == 64 ? 1 : 2;
|
|
||||||
static const u64 kMagic = SANITIZER_WORDSIZE == 64 ? kMagic64 : kMagic32;
|
|
||||||
|
|
||||||
static atomic_uint32_t dump_once_guard; // Ensure that CovDump runs only once.
|
|
||||||
|
|
||||||
static atomic_uintptr_t coverage_counter;
|
|
||||||
|
|
||||||
// pc_array is the array containing the covered PCs.
|
|
||||||
// To make the pc_array thread- and async-signal-safe it has to be large enough.
|
|
||||||
// 128M counters "ought to be enough for anybody" (4M on 32-bit).
|
|
||||||
|
|
||||||
static bool cov_sandboxed = false;
|
|
||||||
static fd_t cov_fd = kInvalidFd;
|
|
||||||
static unsigned int cov_max_block_size = 0;
|
|
||||||
static bool coverage_enabled = false;
|
|
||||||
static const char *coverage_dir;
|
|
||||||
|
|
||||||
namespace __sanitizer {
|
|
||||||
|
|
||||||
class CoverageData {
|
|
||||||
public:
|
|
||||||
void Init();
|
|
||||||
void Enable();
|
|
||||||
void Disable();
|
|
||||||
void ReInit();
|
|
||||||
void BeforeFork();
|
|
||||||
void AfterFork(int child_pid);
|
|
||||||
void Extend(uptr npcs);
|
|
||||||
void Add(uptr pc, u32 *guard);
|
|
||||||
void DumpOffsets();
|
|
||||||
void DumpAll();
|
|
||||||
|
|
||||||
void InitializeGuardArray(s32 *guards);
|
|
||||||
void InitializeGuards(s32 *guards, uptr n, const char *module_name,
|
|
||||||
uptr caller_pc);
|
|
||||||
void ReinitializeGuards();
|
|
||||||
|
|
||||||
uptr *data();
|
|
||||||
uptr size() const;
|
|
||||||
|
|
||||||
private:
|
|
||||||
struct NamedPcRange {
|
|
||||||
const char *copied_module_name;
|
|
||||||
uptr beg, end; // elements [beg,end) in pc_array.
|
|
||||||
};
|
|
||||||
|
|
||||||
void DirectOpen();
|
|
||||||
void UpdateModuleNameVec(uptr caller_pc, uptr range_beg, uptr range_end);
|
|
||||||
void GetRangeOffsets(const NamedPcRange& r, Symbolizer* s,
|
|
||||||
InternalMmapVector<uptr>* offsets) const;
|
|
||||||
|
|
||||||
// Maximal size pc array may ever grow.
|
|
||||||
// We MmapNoReserve this space to ensure that the array is contiguous.
|
|
||||||
static const uptr kPcArrayMaxSize =
|
|
||||||
FIRST_32_SECOND_64(1 << (SANITIZER_ANDROID ? 24 : 26), 1 << 27);
|
|
||||||
// The amount file mapping for the pc array is grown by.
|
|
||||||
static const uptr kPcArrayMmapSize = 64 * 1024;
|
|
||||||
|
|
||||||
// pc_array is allocated with MmapNoReserveOrDie and so it uses only as
|
|
||||||
// much RAM as it really needs.
|
|
||||||
uptr *pc_array;
|
|
||||||
// Index of the first available pc_array slot.
|
|
||||||
atomic_uintptr_t pc_array_index;
|
|
||||||
// Array size.
|
|
||||||
atomic_uintptr_t pc_array_size;
|
|
||||||
// Current file mapped size of the pc array.
|
|
||||||
uptr pc_array_mapped_size;
|
|
||||||
// Descriptor of the file mapped pc array.
|
|
||||||
fd_t pc_fd;
|
|
||||||
|
|
||||||
// Vector of coverage guard arrays, protected by mu.
|
|
||||||
InternalMmapVectorNoCtor<s32*> guard_array_vec;
|
|
||||||
|
|
||||||
// Vector of module and compilation unit pc ranges.
|
|
||||||
InternalMmapVectorNoCtor<NamedPcRange> comp_unit_name_vec;
|
|
||||||
InternalMmapVectorNoCtor<NamedPcRange> module_name_vec;
|
|
||||||
|
|
||||||
StaticSpinMutex mu;
|
|
||||||
};
|
|
||||||
|
|
||||||
static CoverageData coverage_data;
|
|
||||||
|
|
||||||
void CoverageData::DirectOpen() {
|
|
||||||
InternalScopedString path(kMaxPathLength);
|
|
||||||
internal_snprintf((char *)path.data(), path.size(), "%s/%zd.sancov.raw",
|
|
||||||
coverage_dir, internal_getpid());
|
|
||||||
pc_fd = OpenFile(path.data(), RdWr);
|
|
||||||
if (pc_fd == kInvalidFd) {
|
|
||||||
Report("Coverage: failed to open %s for reading/writing\n", path.data());
|
|
||||||
Die();
|
|
||||||
}
|
|
||||||
|
|
||||||
pc_array_mapped_size = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::Init() {
|
|
||||||
pc_fd = kInvalidFd;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::Enable() {
|
|
||||||
if (pc_array)
|
|
||||||
return;
|
|
||||||
pc_array = reinterpret_cast<uptr *>(
|
|
||||||
MmapNoReserveOrDie(sizeof(uptr) * kPcArrayMaxSize, "CovInit"));
|
|
||||||
atomic_store(&pc_array_index, 0, memory_order_relaxed);
|
|
||||||
if (common_flags()->coverage_direct) {
|
|
||||||
Report("coverage_direct=1 is deprecated, don't use it.\n");
|
|
||||||
Die();
|
|
||||||
atomic_store(&pc_array_size, 0, memory_order_relaxed);
|
|
||||||
} else {
|
|
||||||
atomic_store(&pc_array_size, kPcArrayMaxSize, memory_order_relaxed);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::InitializeGuardArray(s32 *guards) {
|
|
||||||
Enable(); // Make sure coverage is enabled at this point.
|
|
||||||
s32 n = guards[0];
|
|
||||||
for (s32 j = 1; j <= n; j++) {
|
|
||||||
uptr idx = atomic_load_relaxed(&pc_array_index);
|
|
||||||
atomic_store_relaxed(&pc_array_index, idx + 1);
|
|
||||||
guards[j] = -static_cast<s32>(idx + 1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::Disable() {
|
|
||||||
if (pc_array) {
|
|
||||||
UnmapOrDie(pc_array, sizeof(uptr) * kPcArrayMaxSize);
|
|
||||||
pc_array = nullptr;
|
|
||||||
}
|
|
||||||
if (pc_fd != kInvalidFd) {
|
|
||||||
CloseFile(pc_fd);
|
|
||||||
pc_fd = kInvalidFd;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::ReinitializeGuards() {
|
|
||||||
// Assuming single thread.
|
|
||||||
atomic_store(&pc_array_index, 0, memory_order_relaxed);
|
|
||||||
for (uptr i = 0; i < guard_array_vec.size(); i++)
|
|
||||||
InitializeGuardArray(guard_array_vec[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::ReInit() {
|
|
||||||
Disable();
|
|
||||||
if (coverage_enabled) {
|
|
||||||
if (common_flags()->coverage_direct) {
|
|
||||||
// In memory-mapped mode we must extend the new file to the known array
|
|
||||||
// size.
|
|
||||||
uptr size = atomic_load(&pc_array_size, memory_order_relaxed);
|
|
||||||
uptr npcs = size / sizeof(uptr);
|
|
||||||
Enable();
|
|
||||||
if (size) Extend(npcs);
|
|
||||||
} else {
|
|
||||||
Enable();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// Re-initialize the guards.
|
|
||||||
// We are single-threaded now, no need to grab any lock.
|
|
||||||
CHECK_EQ(atomic_load(&pc_array_index, memory_order_relaxed), 0);
|
|
||||||
ReinitializeGuards();
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::BeforeFork() {
|
|
||||||
mu.Lock();
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::AfterFork(int child_pid) {
|
|
||||||
// We are single-threaded so it's OK to release the lock early.
|
|
||||||
mu.Unlock();
|
|
||||||
if (child_pid == 0) ReInit();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Extend coverage PC array to fit additional npcs elements.
|
|
||||||
void CoverageData::Extend(uptr npcs) {
|
|
||||||
if (!common_flags()->coverage_direct) return;
|
|
||||||
SpinMutexLock l(&mu);
|
|
||||||
|
|
||||||
uptr size = atomic_load(&pc_array_size, memory_order_relaxed);
|
|
||||||
size += npcs * sizeof(uptr);
|
|
||||||
|
|
||||||
if (coverage_enabled && size > pc_array_mapped_size) {
|
|
||||||
if (pc_fd == kInvalidFd) DirectOpen();
|
|
||||||
CHECK_NE(pc_fd, kInvalidFd);
|
|
||||||
|
|
||||||
uptr new_mapped_size = pc_array_mapped_size;
|
|
||||||
while (size > new_mapped_size) new_mapped_size += kPcArrayMmapSize;
|
|
||||||
CHECK_LE(new_mapped_size, sizeof(uptr) * kPcArrayMaxSize);
|
|
||||||
|
|
||||||
// Extend the file and map the new space at the end of pc_array.
|
|
||||||
uptr res = internal_ftruncate(pc_fd, new_mapped_size);
|
|
||||||
int err;
|
|
||||||
if (internal_iserror(res, &err)) {
|
|
||||||
Printf("failed to extend raw coverage file: %d\n", err);
|
|
||||||
Die();
|
|
||||||
}
|
|
||||||
|
|
||||||
uptr next_map_base = ((uptr)pc_array) + pc_array_mapped_size;
|
|
||||||
void *p = MapWritableFileToMemory((void *)next_map_base,
|
|
||||||
new_mapped_size - pc_array_mapped_size,
|
|
||||||
pc_fd, pc_array_mapped_size);
|
|
||||||
CHECK_EQ((uptr)p, next_map_base);
|
|
||||||
pc_array_mapped_size = new_mapped_size;
|
|
||||||
}
|
|
||||||
|
|
||||||
atomic_store(&pc_array_size, size, memory_order_release);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::UpdateModuleNameVec(uptr caller_pc, uptr range_beg,
|
|
||||||
uptr range_end) {
|
|
||||||
auto sym = Symbolizer::GetOrInit();
|
|
||||||
if (!sym)
|
|
||||||
return;
|
|
||||||
const char *module_name = sym->GetModuleNameForPc(caller_pc);
|
|
||||||
if (!module_name) return;
|
|
||||||
if (module_name_vec.empty() ||
|
|
||||||
module_name_vec.back().copied_module_name != module_name)
|
|
||||||
module_name_vec.push_back({module_name, range_beg, range_end});
|
|
||||||
else
|
|
||||||
module_name_vec.back().end = range_end;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::InitializeGuards(s32 *guards, uptr n,
|
|
||||||
const char *comp_unit_name,
|
|
||||||
uptr caller_pc) {
|
|
||||||
// The array 'guards' has n+1 elements, we use the element zero
|
|
||||||
// to store 'n'.
|
|
||||||
CHECK_LT(n, 1 << 30);
|
|
||||||
guards[0] = static_cast<s32>(n);
|
|
||||||
InitializeGuardArray(guards);
|
|
||||||
SpinMutexLock l(&mu);
|
|
||||||
uptr range_end = atomic_load(&pc_array_index, memory_order_relaxed);
|
|
||||||
uptr range_beg = range_end - n;
|
|
||||||
comp_unit_name_vec.push_back({comp_unit_name, range_beg, range_end});
|
|
||||||
guard_array_vec.push_back(guards);
|
|
||||||
UpdateModuleNameVec(caller_pc, range_beg, range_end);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const uptr kBundleCounterBits = 16;
|
|
||||||
|
|
||||||
// When coverage_order_pcs==true and SANITIZER_WORDSIZE==64
|
|
||||||
// we insert the global counter into the first 16 bits of the PC.
|
|
||||||
uptr BundlePcAndCounter(uptr pc, uptr counter) {
|
|
||||||
if (SANITIZER_WORDSIZE != 64 || !common_flags()->coverage_order_pcs)
|
|
||||||
return pc;
|
|
||||||
static const uptr kMaxCounter = (1 << kBundleCounterBits) - 1;
|
|
||||||
if (counter > kMaxCounter)
|
|
||||||
counter = kMaxCounter;
|
|
||||||
CHECK_EQ(0, pc >> (SANITIZER_WORDSIZE - kBundleCounterBits));
|
|
||||||
return pc | (counter << (SANITIZER_WORDSIZE - kBundleCounterBits));
|
|
||||||
}
|
|
||||||
|
|
||||||
uptr UnbundlePc(uptr bundle) {
|
|
||||||
if (SANITIZER_WORDSIZE != 64 || !common_flags()->coverage_order_pcs)
|
|
||||||
return bundle;
|
|
||||||
return (bundle << kBundleCounterBits) >> kBundleCounterBits;
|
|
||||||
}
|
|
||||||
|
|
||||||
uptr UnbundleCounter(uptr bundle) {
|
|
||||||
if (SANITIZER_WORDSIZE != 64 || !common_flags()->coverage_order_pcs)
|
|
||||||
return 0;
|
|
||||||
return bundle >> (SANITIZER_WORDSIZE - kBundleCounterBits);
|
|
||||||
}
|
|
||||||
|
|
||||||
// If guard is negative, atomically set it to -guard and store the PC in
|
|
||||||
// pc_array.
|
|
||||||
void CoverageData::Add(uptr pc, u32 *guard) {
|
|
||||||
atomic_uint32_t *atomic_guard = reinterpret_cast<atomic_uint32_t*>(guard);
|
|
||||||
s32 guard_value = atomic_load(atomic_guard, memory_order_relaxed);
|
|
||||||
if (guard_value >= 0) return;
|
|
||||||
|
|
||||||
atomic_store(atomic_guard, -guard_value, memory_order_relaxed);
|
|
||||||
if (!pc_array) return;
|
|
||||||
|
|
||||||
uptr idx = -guard_value - 1;
|
|
||||||
if (idx >= atomic_load(&pc_array_index, memory_order_acquire))
|
|
||||||
return; // May happen after fork when pc_array_index becomes 0.
|
|
||||||
CHECK_LT(idx, atomic_load(&pc_array_size, memory_order_acquire));
|
|
||||||
uptr counter = atomic_fetch_add(&coverage_counter, 1, memory_order_relaxed);
|
|
||||||
pc_array[idx] = BundlePcAndCounter(pc, counter);
|
|
||||||
}
|
|
||||||
|
|
||||||
uptr *CoverageData::data() {
|
|
||||||
return pc_array;
|
|
||||||
}
|
|
||||||
|
|
||||||
uptr CoverageData::size() const {
|
|
||||||
return atomic_load(&pc_array_index, memory_order_relaxed);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Block layout for packed file format: header, followed by module name (no
|
|
||||||
// trailing zero), followed by data blob.
|
|
||||||
struct CovHeader {
|
|
||||||
int pid;
|
|
||||||
unsigned int module_name_length;
|
|
||||||
unsigned int data_length;
|
|
||||||
};
|
|
||||||
|
|
||||||
static void CovWritePacked(int pid, const char *module, const void *blob,
|
|
||||||
unsigned int blob_size) {
|
|
||||||
if (cov_fd == kInvalidFd) return;
|
|
||||||
unsigned module_name_length = internal_strlen(module);
|
|
||||||
CovHeader header = {pid, module_name_length, blob_size};
|
|
||||||
|
|
||||||
if (cov_max_block_size == 0) {
|
|
||||||
// Writing to a file. Just go ahead.
|
|
||||||
WriteToFile(cov_fd, &header, sizeof(header));
|
|
||||||
WriteToFile(cov_fd, module, module_name_length);
|
|
||||||
WriteToFile(cov_fd, blob, blob_size);
|
|
||||||
} else {
|
|
||||||
// Writing to a socket. We want to split the data into appropriately sized
|
|
||||||
// blocks.
|
|
||||||
InternalScopedBuffer<char> block(cov_max_block_size);
|
|
||||||
CHECK_EQ((uptr)block.data(), (uptr)(CovHeader *)block.data());
|
|
||||||
uptr header_size_with_module = sizeof(header) + module_name_length;
|
|
||||||
CHECK_LT(header_size_with_module, cov_max_block_size);
|
|
||||||
unsigned int max_payload_size =
|
|
||||||
cov_max_block_size - header_size_with_module;
|
|
||||||
char *block_pos = block.data();
|
|
||||||
internal_memcpy(block_pos, &header, sizeof(header));
|
|
||||||
block_pos += sizeof(header);
|
|
||||||
internal_memcpy(block_pos, module, module_name_length);
|
|
||||||
block_pos += module_name_length;
|
|
||||||
char *block_data_begin = block_pos;
|
|
||||||
const char *blob_pos = (const char *)blob;
|
|
||||||
while (blob_size > 0) {
|
|
||||||
unsigned int payload_size = Min(blob_size, max_payload_size);
|
|
||||||
blob_size -= payload_size;
|
|
||||||
internal_memcpy(block_data_begin, blob_pos, payload_size);
|
|
||||||
blob_pos += payload_size;
|
|
||||||
((CovHeader *)block.data())->data_length = payload_size;
|
|
||||||
WriteToFile(cov_fd, block.data(), header_size_with_module + payload_size);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// If packed = false: <name>.<pid>.<sancov> (name = module name).
|
|
||||||
// If packed = true and name == 0: <pid>.<sancov>.<packed>.
|
|
||||||
// If packed = true and name != 0: <name>.<sancov>.<packed> (name is
|
|
||||||
// user-supplied).
|
|
||||||
static fd_t CovOpenFile(InternalScopedString *path, bool packed,
|
|
||||||
const char *name, const char *extension = "sancov") {
|
|
||||||
path->clear();
|
|
||||||
if (!packed) {
|
|
||||||
CHECK(name);
|
|
||||||
path->append("%s/%s.%zd.%s", coverage_dir, name, internal_getpid(),
|
|
||||||
extension);
|
|
||||||
} else {
|
|
||||||
if (!name)
|
|
||||||
path->append("%s/%zd.%s.packed", coverage_dir, internal_getpid(),
|
|
||||||
extension);
|
|
||||||
else
|
|
||||||
path->append("%s/%s.%s.packed", coverage_dir, name, extension);
|
|
||||||
}
|
|
||||||
error_t err;
|
|
||||||
fd_t fd = OpenFile(path->data(), WrOnly, &err);
|
|
||||||
if (fd == kInvalidFd)
|
|
||||||
Report("SanitizerCoverage: failed to open %s for writing (reason: %d)\n",
|
|
||||||
path->data(), err);
|
|
||||||
return fd;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::GetRangeOffsets(const NamedPcRange& r, Symbolizer* sym,
|
|
||||||
InternalMmapVector<uptr>* offsets) const {
|
|
||||||
offsets->clear();
|
|
||||||
for (uptr i = 0; i < kNumWordsForMagic; i++)
|
|
||||||
offsets->push_back(0);
|
|
||||||
CHECK(r.copied_module_name);
|
|
||||||
CHECK_LE(r.beg, r.end);
|
|
||||||
CHECK_LE(r.end, size());
|
|
||||||
for (uptr i = r.beg; i < r.end; i++) {
|
|
||||||
uptr pc = UnbundlePc(pc_array[i]);
|
|
||||||
uptr counter = UnbundleCounter(pc_array[i]);
|
|
||||||
if (!pc) continue; // Not visited.
|
|
||||||
uptr offset = 0;
|
|
||||||
sym->GetModuleNameAndOffsetForPC(pc, nullptr, &offset);
|
|
||||||
offsets->push_back(BundlePcAndCounter(offset, counter));
|
|
||||||
}
|
|
||||||
|
|
||||||
CHECK_GE(offsets->size(), kNumWordsForMagic);
|
|
||||||
SortArray(offsets->data(), offsets->size());
|
|
||||||
for (uptr i = 0; i < offsets->size(); i++)
|
|
||||||
(*offsets)[i] = UnbundlePc((*offsets)[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void GenerateHtmlReport(const InternalMmapVector<char *> &cov_files) {
|
|
||||||
if (!common_flags()->html_cov_report) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
char *sancov_path = FindPathToBinary(common_flags()->sancov_path);
|
|
||||||
if (sancov_path == nullptr) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
InternalMmapVector<char *> sancov_argv(cov_files.size() * 2 + 3);
|
|
||||||
sancov_argv.push_back(sancov_path);
|
|
||||||
sancov_argv.push_back(internal_strdup("-html-report"));
|
|
||||||
auto argv_deleter = at_scope_exit([&] {
|
|
||||||
for (uptr i = 0; i < sancov_argv.size(); ++i) {
|
|
||||||
InternalFree(sancov_argv[i]);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
for (const auto &cov_file : cov_files) {
|
|
||||||
sancov_argv.push_back(internal_strdup(cov_file));
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
ListOfModules modules;
|
|
||||||
modules.init();
|
|
||||||
for (const LoadedModule &module : modules) {
|
|
||||||
sancov_argv.push_back(internal_strdup(module.full_name()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
InternalScopedString report_path(kMaxPathLength);
|
|
||||||
fd_t report_fd =
|
|
||||||
CovOpenFile(&report_path, false /* packed */, GetProcessName(), "html");
|
|
||||||
int pid = StartSubprocess(sancov_argv[0], sancov_argv.data(),
|
|
||||||
kInvalidFd /* stdin */, report_fd /* std_out */);
|
|
||||||
if (pid > 0) {
|
|
||||||
int result = WaitForProcess(pid);
|
|
||||||
if (result == 0)
|
|
||||||
Printf("coverage report generated to %s\n", report_path.data());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::DumpOffsets() {
|
|
||||||
auto sym = Symbolizer::GetOrInit();
|
|
||||||
if (!common_flags()->coverage_pcs) return;
|
|
||||||
Printf("**\n***\n***\n");
|
|
||||||
Printf("**WARNING: this implementation of SanitizerCoverage is deprecated\n");
|
|
||||||
Printf("**WARNING: and will be removed in future versions\n");
|
|
||||||
Printf("**WARNING: See https://clang.llvm.org/docs/SanitizerCoverage.html\n");
|
|
||||||
Printf("**\n***\n***\n");
|
|
||||||
|
|
||||||
CHECK_NE(sym, nullptr);
|
|
||||||
InternalMmapVector<uptr> offsets(0);
|
|
||||||
InternalScopedString path(kMaxPathLength);
|
|
||||||
|
|
||||||
InternalMmapVector<char *> cov_files(module_name_vec.size());
|
|
||||||
auto cov_files_deleter = at_scope_exit([&] {
|
|
||||||
for (uptr i = 0; i < cov_files.size(); ++i) {
|
|
||||||
InternalFree(cov_files[i]);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
for (uptr m = 0; m < module_name_vec.size(); m++) {
|
|
||||||
auto r = module_name_vec[m];
|
|
||||||
GetRangeOffsets(r, sym, &offsets);
|
|
||||||
|
|
||||||
uptr num_offsets = offsets.size() - kNumWordsForMagic;
|
|
||||||
u64 *magic_p = reinterpret_cast<u64*>(offsets.data());
|
|
||||||
CHECK_EQ(*magic_p, 0ULL);
|
|
||||||
// FIXME: we may want to write 32-bit offsets even in 64-mode
|
|
||||||
// if all the offsets are small enough.
|
|
||||||
*magic_p = kMagic;
|
|
||||||
|
|
||||||
const char *module_name = StripModuleName(r.copied_module_name);
|
|
||||||
if (cov_sandboxed) {
|
|
||||||
if (cov_fd != kInvalidFd) {
|
|
||||||
CovWritePacked(internal_getpid(), module_name, offsets.data(),
|
|
||||||
offsets.size() * sizeof(offsets[0]));
|
|
||||||
VReport(1, " CovDump: %zd PCs written to packed file\n", num_offsets);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// One file per module per process.
|
|
||||||
fd_t fd = CovOpenFile(&path, false /* packed */, module_name);
|
|
||||||
if (fd == kInvalidFd) continue;
|
|
||||||
WriteToFile(fd, offsets.data(), offsets.size() * sizeof(offsets[0]));
|
|
||||||
CloseFile(fd);
|
|
||||||
cov_files.push_back(internal_strdup(path.data()));
|
|
||||||
VReport(1, " CovDump: %s: %zd PCs written\n", path.data(), num_offsets);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (cov_fd != kInvalidFd)
|
|
||||||
CloseFile(cov_fd);
|
|
||||||
|
|
||||||
GenerateHtmlReport(cov_files);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CoverageData::DumpAll() {
|
|
||||||
if (!coverage_enabled || common_flags()->coverage_direct) return;
|
|
||||||
if (atomic_fetch_add(&dump_once_guard, 1, memory_order_relaxed))
|
|
||||||
return;
|
|
||||||
DumpOffsets();
|
|
||||||
}
|
|
||||||
|
|
||||||
void CovPrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
|
|
||||||
if (!args) return;
|
|
||||||
if (!coverage_enabled) return;
|
|
||||||
cov_sandboxed = args->coverage_sandboxed;
|
|
||||||
if (!cov_sandboxed) return;
|
|
||||||
cov_max_block_size = args->coverage_max_block_size;
|
|
||||||
if (args->coverage_fd >= 0) {
|
|
||||||
cov_fd = (fd_t)args->coverage_fd;
|
|
||||||
} else {
|
|
||||||
InternalScopedString path(kMaxPathLength);
|
|
||||||
// Pre-open the file now. The sandbox won't allow us to do it later.
|
|
||||||
cov_fd = CovOpenFile(&path, true /* packed */, nullptr);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fd_t MaybeOpenCovFile(const char *name) {
|
|
||||||
CHECK(name);
|
|
||||||
if (!coverage_enabled) return kInvalidFd;
|
|
||||||
InternalScopedString path(kMaxPathLength);
|
|
||||||
return CovOpenFile(&path, true /* packed */, name);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CovBeforeFork() {
|
|
||||||
coverage_data.BeforeFork();
|
|
||||||
}
|
|
||||||
|
|
||||||
void CovAfterFork(int child_pid) {
|
|
||||||
coverage_data.AfterFork(child_pid);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void MaybeDumpCoverage() {
|
|
||||||
if (common_flags()->coverage)
|
|
||||||
__sanitizer_cov_dump();
|
|
||||||
}
|
|
||||||
|
|
||||||
void InitializeCoverage(bool enabled, const char *dir) {
|
|
||||||
if (coverage_enabled)
|
|
||||||
return; // May happen if two sanitizer enable coverage in the same process.
|
|
||||||
coverage_enabled = enabled;
|
|
||||||
coverage_dir = dir;
|
|
||||||
coverage_data.Init();
|
|
||||||
if (enabled) coverage_data.Enable();
|
|
||||||
if (!common_flags()->coverage_direct) Atexit(__sanitizer_cov_dump);
|
|
||||||
AddDieCallback(MaybeDumpCoverage);
|
|
||||||
}
|
|
||||||
|
|
||||||
void ReInitializeCoverage(bool enabled, const char *dir) {
|
|
||||||
coverage_enabled = enabled;
|
|
||||||
coverage_dir = dir;
|
|
||||||
coverage_data.ReInit();
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace __sanitizer
|
|
||||||
|
|
||||||
extern "C" {
|
|
||||||
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_cov_dump() {
|
|
||||||
__sanitizer_dump_trace_pc_guard_coverage();
|
|
||||||
}
|
|
||||||
// Default empty implementations (weak). Users should redefine them.
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp1, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp2, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp4, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp8, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_switch, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_div4, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_div8, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_gep, void) {}
|
|
||||||
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_indir, void) {}
|
|
||||||
} // extern "C"
|
|
|
@ -146,6 +146,17 @@ static TracePcGuardController pc_guard_controller;
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace __sancov
|
} // namespace __sancov
|
||||||
|
|
||||||
|
namespace __sanitizer {
|
||||||
|
void InitializeCoverage(bool enabled, const char *dir) {
|
||||||
|
static bool coverage_enabled = false;
|
||||||
|
if (coverage_enabled)
|
||||||
|
return; // May happen if two sanitizer enable coverage in the same process.
|
||||||
|
coverage_enabled = enabled;
|
||||||
|
Atexit(__sanitizer_cov_dump);
|
||||||
|
AddDieCallback(__sanitizer_cov_dump);
|
||||||
|
}
|
||||||
|
} // namespace __sanitizer
|
||||||
|
|
||||||
extern "C" {
|
extern "C" {
|
||||||
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_dump_coverage( // NOLINT
|
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_dump_coverage( // NOLINT
|
||||||
const uptr* pcs, uptr len) {
|
const uptr* pcs, uptr len) {
|
||||||
|
@ -166,4 +177,18 @@ SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_guard_init,
|
||||||
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_dump_trace_pc_guard_coverage() {
|
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_dump_trace_pc_guard_coverage() {
|
||||||
__sancov::pc_guard_controller.Dump();
|
__sancov::pc_guard_controller.Dump();
|
||||||
}
|
}
|
||||||
|
SANITIZER_INTERFACE_ATTRIBUTE void __sanitizer_cov_dump() {
|
||||||
|
__sanitizer_dump_trace_pc_guard_coverage();
|
||||||
|
}
|
||||||
|
// Default empty implementations (weak). Users should redefine them.
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp1, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp2, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp4, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_cmp8, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_switch, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_div4, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_div8, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_gep, void) {}
|
||||||
|
SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_indir, void) {}
|
||||||
} // extern "C"
|
} // extern "C"
|
||||||
|
|
|
@ -264,7 +264,6 @@ void PrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
|
||||||
// Same for /proc/self/exe in the symbolizer.
|
// Same for /proc/self/exe in the symbolizer.
|
||||||
#if !SANITIZER_GO
|
#if !SANITIZER_GO
|
||||||
Symbolizer::GetOrInit()->PrepareForSandboxing();
|
Symbolizer::GetOrInit()->PrepareForSandboxing();
|
||||||
CovPrepareForSandboxing(args);
|
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -400,9 +400,6 @@ void ReExec() {
|
||||||
}
|
}
|
||||||
|
|
||||||
void PrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
|
void PrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
|
||||||
#if !SANITIZER_GO
|
|
||||||
CovPrepareForSandboxing(args);
|
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool StackSizeIsUnlimited() {
|
bool StackSizeIsUnlimited() {
|
||||||
|
|
Loading…
Reference in New Issue