llvm-project

1
//===-- stats_client.cpp --------------------------------------------------===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8
//
9
// Sanitizer statistics gathering. Manages statistics for a module (executable
10
// or DSO) and registers statistics with the process.
11
//
12
// This is linked into each individual modle and cannot directly use functions
13
// declared in sanitizer_common.
14
//
15
//===----------------------------------------------------------------------===//
16

17
#ifdef _WIN32
18
#define WIN32_LEAN_AND_MEAN
19
#include <windows.h>
20
#else
21
#include <dlfcn.h>
22
#endif
23
#include <stdint.h>
24
#include <stdio.h>
25

26
#include "sanitizer_common/sanitizer_internal_defs.h"
27
#include "stats/stats.h"
28

29
using namespace __sanitizer;
30

31
namespace {
32

33
void *LookupSymbolFromMain(const char *name) {
34
#ifdef _WIN32
35
  return reinterpret_cast<void *>(GetProcAddress(GetModuleHandle(0), name));
36
#else
37
  return dlsym(RTLD_DEFAULT, name);
38
#endif
39
}
40

41
StatModule *list;
42

43
struct RegisterSanStats {
44
  unsigned module_id;
45

46
  RegisterSanStats() {
47
    typedef unsigned (*reg_func_t)(StatModule **);
48
    reg_func_t reg_func = reinterpret_cast<reg_func_t>(
49
        LookupSymbolFromMain("__sanitizer_stats_register"));
50
    if (reg_func)
51
      module_id = reg_func(&list);
52
  }
53

54
  ~RegisterSanStats() {
55
    typedef void (*unreg_func_t)(unsigned);
56
    unreg_func_t unreg_func = reinterpret_cast<unreg_func_t>(
57
        LookupSymbolFromMain("__sanitizer_stats_unregister"));
58
    if (unreg_func)
59
      unreg_func(module_id);
60
  }
61
} reg;
62

63
}
64

65
extern "C" void __sanitizer_stat_init(StatModule *mod) {
66
  mod->next = list;
67
  list = mod;
68
}
69

70
extern "C" void __sanitizer_stat_report(StatInfo *s) {
71
  s->addr = GET_CALLER_PC();
72
#if defined(_WIN64) && !defined(__clang__)
73
  uptr old_data = InterlockedIncrement64(reinterpret_cast<LONG64 *>(&s->data));
74
#elif defined(_WIN32) && !defined(__clang__)
75
  uptr old_data = InterlockedIncrement(&s->data);
76
#else
77
  uptr old_data = __sync_fetch_and_add(&s->data, 1);
78
#endif
79

80
  // Overflow check.
81
  if (CountFromData(old_data + 1) == 0)
82
    Trap();
83
}
84