//===-- stats_client.cpp --------------------------------------------------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// // // Sanitizer statistics gathering. Manages statistics for a module (executable // or DSO) and registers statistics with the process. // // This is linked into each individual modle and cannot directly use functions // declared in sanitizer_common. // //===----------------------------------------------------------------------===// #ifdef _WIN32 #define WIN32_LEAN_AND_MEAN #include <windows.h> #else #include <dlfcn.h> #endif #include <stdint.h> #include <stdio.h> #include "sanitizer_common/sanitizer_internal_defs.h" #include "stats/stats.h" using namespace __sanitizer; namespace { void *LookupSymbolFromMain(const char *name) { #ifdef _WIN32 return reinterpret_cast<void *>(GetProcAddress(GetModuleHandle(0), name)); #else return dlsym(RTLD_DEFAULT, name); #endif } StatModule *list; struct RegisterSanStats { unsigned module_id; RegisterSanStats() { typedef unsigned (*reg_func_t)(StatModule **); reg_func_t reg_func = reinterpret_cast<reg_func_t>( LookupSymbolFromMain("__sanitizer_stats_register")); if (reg_func) module_id = reg_func(&list); } ~RegisterSanStats() { typedef void (*unreg_func_t)(unsigned); unreg_func_t unreg_func = reinterpret_cast<unreg_func_t>( LookupSymbolFromMain("__sanitizer_stats_unregister")); if (unreg_func) unreg_func(module_id); } } reg; } extern "C" void __sanitizer_stat_init(StatModule *mod) { mod->next = list; list = mod; } extern "C" void __sanitizer_stat_report(StatInfo *s) { s->addr = GET_CALLER_PC(); #if defined(_WIN64) && !defined(__clang__) uptr old_data = InterlockedIncrement64(reinterpret_cast<LONG64 *>(&s->data)); #elif defined(_WIN32) && !defined(__clang__) uptr old_data = InterlockedIncrement(&s->data); #else uptr old_data = __sync_fetch_and_add(&s->data, 1); #endif // Overflow check. if (CountFromData(old_data + 1) == 0) Trap(); } |