1//===-- sanitizer_stackdepot.h ----------------------------------*- C++ -*-===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file is shared between AddressSanitizer and ThreadSanitizer 11// run-time libraries. 12//===----------------------------------------------------------------------===// 13 14#ifndef SANITIZER_STACKDEPOT_H 15#define SANITIZER_STACKDEPOT_H 16 17#include "sanitizer_common.h" 18#include "sanitizer_internal_defs.h" 19#include "sanitizer_stacktrace.h" 20 21namespace __sanitizer { 22 23// StackDepot efficiently stores huge amounts of stack traces. 24struct StackDepotNode; 25struct StackDepotHandle { 26 StackDepotNode *node_; 27 StackDepotHandle() : node_(nullptr) {} 28 explicit StackDepotHandle(StackDepotNode *node) : node_(node) {} 29 bool valid() { return node_; } 30 u32 id(); 31 int use_count(); 32 void inc_use_count_unsafe(); 33}; 34 35const int kStackDepotMaxUseCount = 1U << 20; 36 37StackDepotStats *StackDepotGetStats(); 38u32 StackDepotPut(StackTrace stack); 39StackDepotHandle StackDepotPut_WithHandle(StackTrace stack); 40// Retrieves a stored stack trace by the id. 41StackTrace StackDepotGet(u32 id); 42 43void StackDepotLockAll(); 44void StackDepotUnlockAll(); 45 46// Instantiating this class creates a snapshot of StackDepot which can be 47// efficiently queried with StackDepotGet(). You can use it concurrently with 48// StackDepot, but the snapshot is only guaranteed to contain those stack traces 49// which were stored before it was instantiated. 50class StackDepotReverseMap { 51 public: 52 StackDepotReverseMap(); 53 StackTrace Get(u32 id); 54 55 private: 56 struct IdDescPair { 57 u32 id; 58 StackDepotNode *desc; 59 60 static bool IdComparator(const IdDescPair &a, const IdDescPair &b); 61 }; 62 63 InternalMmapVector<IdDescPair> map_; 64 65 // Disallow evil constructors. 66 StackDepotReverseMap(const StackDepotReverseMap&); 67 void operator=(const StackDepotReverseMap&); 68}; 69 70} // namespace __sanitizer 71 72#endif // SANITIZER_STACKDEPOT_H 73