OSchip
/
llvm-project
mirror of https://github.com/THU-DSP-LAB/llvm-project.git
13
0
Fork 6
Code Issues Packages Projects Releases Wiki Activity

[ASan] Add __asan_set_on_error_callback() interface function that allows user to set a callback to be called right when ASan detects an error 

llvm-svn: 161754
This commit is contained in:
Alexey Samsonov 2012-08-13 11:23:40 +00:00
parent bf4871d363
commit f87ff74075
3 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
compiler-rt/lib/asan/asan_interface.h
View File

@ -118,6 +118,13 @@ extern "C" {
void __asan_set_error_report_callback(void (*callback)(const char*)) void __asan_set_error_report_callback(void (*callback)(const char*))
SANITIZER_INTERFACE_ATTRIBUTE; SANITIZER_INTERFACE_ATTRIBUTE;
// Sets the callback to be called right when ASan detects an error.
// This can be used to notice cases when ASan detects an error, but the
// program crashes before ASan report is printed.
// Passing 0 unsets the callback.
void __asan_set_on_error_callback(void (*callback)(void))
SANITIZER_INTERFACE_ATTRIBUTE;
// Returns the estimated number of bytes that will be reserved by allocator // Returns the estimated number of bytes that will be reserved by allocator
// for request of "size" bytes. If ASan allocator can't allocate that much // for request of "size" bytes. If ASan allocator can't allocate that much
// memory, returns the maximal possible allocation size, otherwise returns // memory, returns the maximal possible allocation size, otherwise returns

11
compiler-rt/lib/asan/asan_report.cc
View File

@ -21,7 +21,7 @@
namespace __asan { namespace __asan {
// ---------------------- Error report callback ------------------- {{{1 // -------------------- User-specified callbacks ----------------- {{{1
static void (*error_report_callback)(const char*); static void (*error_report_callback)(const char*);
static char *error_message_buffer = 0; static char *error_message_buffer = 0;
static uptr error_message_buffer_pos = 0; static uptr error_message_buffer_pos = 0;
@ -40,6 +40,8 @@ void AppendToErrorMessageBuffer(const char *buffer) {
} }
} }
static void (*on_error_callback)(void);
// ---------------------- Helper functions ----------------------- {{{1 // ---------------------- Helper functions ----------------------- {{{1
static void PrintBytes(const char *before, uptr *a) { static void PrintBytes(const char *before, uptr *a) {
@ -219,6 +221,9 @@ class ScopedInErrorReport {
SleepForSeconds(Max(5, flags()->sleep_before_dying + 1)); SleepForSeconds(Max(5, flags()->sleep_before_dying + 1));
Die(); Die();
} }
if (on_error_callback) {
on_error_callback();
}
AsanPrintf("====================================================" AsanPrintf("===================================================="
"=============\n"); "=============\n");
AsanThread *curr_thread = asanThreadRegistry().GetCurrent(); AsanThread *curr_thread = asanThreadRegistry().GetCurrent();
@ -409,3 +414,7 @@ void NOINLINE __asan_set_error_report_callback(void (*callback)(const char*)) {
error_message_buffer_pos = 0; error_message_buffer_pos = 0;
} }
} }
void NOINLINE __asan_set_on_error_callback(void (*callback)(void)) {
on_error_callback = callback;
}

6
compiler-rt/lib/asan/tests/asan_noinst_test.cc
View File

@ -530,6 +530,12 @@ TEST(AddressSanitizerInterface, DeathCallbackTest) {
__asan_set_death_callback(NULL); __asan_set_death_callback(NULL);
} }
TEST(AddressSanitizerInterface, OnErrorCallbackTest) {
__asan_set_on_error_callback(MyDeathCallback);
EXPECT_DEATH(DoDoubleFree(), "MyDeathCallback.*double-free");
__asan_set_on_error_callback(NULL);
}
static const char* kUseAfterPoisonErrorMessage = "use-after-poison"; static const char* kUseAfterPoisonErrorMessage = "use-after-poison";
#define GOOD_ACCESS(ptr, offset) \ #define GOOD_ACCESS(ptr, offset) \