1 //===-- sanitizer_allocator_stats.h -----------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // Part of the Sanitizer Allocator.
12 //===----------------------------------------------------------------------===//
13 #ifndef SANITIZER_ALLOCATOR_H
14 #error This file must be included inside sanitizer_allocator.h
17 // Memory allocator statistics
19 AllocatorStatAllocated,
24 typedef uptr AllocatorStatCounters[AllocatorStatCount];
26 // Per-thread stats, live in per-thread cache.
27 class AllocatorStats {
30 internal_memset(this, 0, sizeof(*this));
32 void InitLinkerInitialized() {}
34 void Add(AllocatorStat i, uptr v) {
35 v += atomic_load(&stats_[i], memory_order_relaxed);
36 atomic_store(&stats_[i], v, memory_order_relaxed);
39 void Sub(AllocatorStat i, uptr v) {
40 v = atomic_load(&stats_[i], memory_order_relaxed) - v;
41 atomic_store(&stats_[i], v, memory_order_relaxed);
44 void Set(AllocatorStat i, uptr v) {
45 atomic_store(&stats_[i], v, memory_order_relaxed);
48 uptr Get(AllocatorStat i) const {
49 return atomic_load(&stats_[i], memory_order_relaxed);
53 friend class AllocatorGlobalStats;
54 AllocatorStats *next_;
55 AllocatorStats *prev_;
56 atomic_uintptr_t stats_[AllocatorStatCount];
59 // Global stats, used for aggregation and querying.
60 class AllocatorGlobalStats : public AllocatorStats {
62 void InitLinkerInitialized() {
67 internal_memset(this, 0, sizeof(*this));
68 InitLinkerInitialized();
71 void Register(AllocatorStats *s) {
72 SpinMutexLock l(&mu_);
79 void Unregister(AllocatorStats *s) {
80 SpinMutexLock l(&mu_);
81 s->prev_->next_ = s->next_;
82 s->next_->prev_ = s->prev_;
83 for (int i = 0; i < AllocatorStatCount; i++)
84 Add(AllocatorStat(i), s->Get(AllocatorStat(i)));
87 void Get(AllocatorStatCounters s) const {
88 internal_memset(s, 0, AllocatorStatCount * sizeof(uptr));
89 SpinMutexLock l(&mu_);
90 const AllocatorStats *stats = this;
92 for (int i = 0; i < AllocatorStatCount; i++)
93 s[i] += stats->Get(AllocatorStat(i));
98 // All stats must be non-negative.
99 for (int i = 0; i < AllocatorStatCount; i++)
100 s[i] = ((sptr)s[i]) >= 0 ? s[i] : 0;
104 mutable SpinMutex mu_;