1//===-- memprof_stack.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// This file is a part of MemProfiler, a memory profiler.
10//
11// Code for MemProf stack trace.
12//===----------------------------------------------------------------------===//
13#include "memprof_stack.h"
14#include "memprof_internal.h"
15#include "sanitizer_common/sanitizer_atomic.h"
16
17namespace __memprof {
18
19static atomic_uint32_t malloc_context_size;
20
21void SetMallocContextSize(u32 size) {
22  atomic_store(&malloc_context_size, size, memory_order_release);
23}
24
25u32 GetMallocContextSize() {
26  return atomic_load(&malloc_context_size, memory_order_acquire);
27}
28
29} // namespace __memprof
30
31void __sanitizer::BufferedStackTrace::UnwindImpl(uptr pc, uptr bp,
32                                                 void *context,
33                                                 bool request_fast,
34                                                 u32 max_depth) {
35  using namespace __memprof;
36  size = 0;
37  if (UNLIKELY(!memprof_inited))
38    return;
39  request_fast = StackTrace::WillUseFastUnwind(request_fast);
40  MemprofThread *t = GetCurrentThread();
41  if (request_fast) {
42    if (t) {
43      Unwind(max_depth, pc, bp, nullptr, t->stack_top(), t->stack_bottom(),
44             true);
45    }
46    return;
47  }
48  Unwind(max_depth, pc, bp, context, 0, 0, false);
49}
50
51// ------------------ Interface -------------- {{{1
52
53extern "C" {
54SANITIZER_INTERFACE_ATTRIBUTE
55void __sanitizer_print_stack_trace() {
56  using namespace __memprof;
57  PRINT_CURRENT_STACK();
58}
59} // extern "C"
60