60 lines · cpp
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-exception6//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 __memprof30 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 -------------- {{{152 53extern "C" {54SANITIZER_INTERFACE_ATTRIBUTE55void __sanitizer_print_stack_trace() {56 using namespace __memprof;57 PRINT_CURRENT_STACK();58}59} // extern "C"60