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