2721e27c3a
We have some significant amount of duplication around CheckFailed functionality. Each sanitizer copy-pasted a chunk of code. Some got random improvements like dealing with recursive failures better. These improvements could benefit all sanitizers, but they don't. Deduplicate CheckFailed logic across sanitizers and let each sanitizer only print the current stack trace. I've tried to dedup stack printing as well, but this got me into cmake hell. So let's keep this part duplicated in each sanitizer for now. Reviewed By: vitalybuka Differential Revision: https://reviews.llvm.org/D102221
72 lines
2.7 KiB
C++
72 lines
2.7 KiB
C++
//===-- asan_stack.h --------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of AddressSanitizer, an address sanity checker.
|
|
//
|
|
// ASan-private header for asan_stack.cpp.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef ASAN_STACK_H
|
|
#define ASAN_STACK_H
|
|
|
|
#include "asan_flags.h"
|
|
#include "asan_thread.h"
|
|
#include "sanitizer_common/sanitizer_flags.h"
|
|
#include "sanitizer_common/sanitizer_stacktrace.h"
|
|
|
|
namespace __asan {
|
|
|
|
static const u32 kDefaultMallocContextSize = 30;
|
|
|
|
void SetMallocContextSize(u32 size);
|
|
u32 GetMallocContextSize();
|
|
|
|
} // namespace __asan
|
|
|
|
// NOTE: A Rule of thumb is to retrieve stack trace in the interceptors
|
|
// as early as possible (in functions exposed to the user), as we generally
|
|
// don't want stack trace to contain functions from ASan internals.
|
|
|
|
#define GET_STACK_TRACE(max_size, fast) \
|
|
BufferedStackTrace stack; \
|
|
if (max_size <= 2) { \
|
|
stack.size = max_size; \
|
|
if (max_size > 0) { \
|
|
stack.top_frame_bp = GET_CURRENT_FRAME(); \
|
|
stack.trace_buffer[0] = StackTrace::GetCurrentPc(); \
|
|
if (max_size > 1) stack.trace_buffer[1] = GET_CALLER_PC(); \
|
|
} \
|
|
} else { \
|
|
stack.Unwind(StackTrace::GetCurrentPc(), \
|
|
GET_CURRENT_FRAME(), nullptr, fast, max_size); \
|
|
}
|
|
|
|
#define GET_STACK_TRACE_FATAL(pc, bp) \
|
|
BufferedStackTrace stack; \
|
|
stack.Unwind(pc, bp, nullptr, \
|
|
common_flags()->fast_unwind_on_fatal)
|
|
|
|
#define GET_STACK_TRACE_FATAL_HERE \
|
|
GET_STACK_TRACE(kStackTraceMax, common_flags()->fast_unwind_on_fatal)
|
|
|
|
#define GET_STACK_TRACE_THREAD \
|
|
GET_STACK_TRACE(kStackTraceMax, true)
|
|
|
|
#define GET_STACK_TRACE_MALLOC \
|
|
GET_STACK_TRACE(GetMallocContextSize(), common_flags()->fast_unwind_on_malloc)
|
|
|
|
#define GET_STACK_TRACE_FREE GET_STACK_TRACE_MALLOC
|
|
|
|
#define PRINT_CURRENT_STACK() \
|
|
{ \
|
|
GET_STACK_TRACE_FATAL_HERE; \
|
|
stack.Print(); \
|
|
}
|
|
|
|
#endif // ASAN_STACK_H
|