From 9f7e8c143302fe051a8c8fca45565cde1bee88a7 Mon Sep 17 00:00:00 2001 From: Evgeniy Stepanov Date: Mon, 28 Jan 2013 13:52:49 +0000 Subject: [msan] A runtime call to support custom allocators. llvm-svn: 173687 --- compiler-rt/lib/msan/msan_interceptors.cc | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'compiler-rt/lib') diff --git a/compiler-rt/lib/msan/msan_interceptors.cc b/compiler-rt/lib/msan/msan_interceptors.cc index 0672bcf78c9..635bff14dbc 100644 --- a/compiler-rt/lib/msan/msan_interceptors.cc +++ b/compiler-rt/lib/msan/msan_interceptors.cc @@ -20,6 +20,7 @@ #include "msan_platform_limits_posix.h" #include "sanitizer_common/sanitizer_allocator.h" #include "sanitizer_common/sanitizer_common.h" +#include "sanitizer_common/sanitizer_stackdepot.h" #include "sanitizer_common/sanitizer_libc.h" #include @@ -710,6 +711,18 @@ INTERCEPTOR(void *, malloc, SIZE_T size) { return MsanReallocate(&stack, 0, size, sizeof(u64), false); } +void __msan_allocated_memory(void* data, uptr size) { + GET_MALLOC_STACK_TRACE; + if (flags()->poison_in_malloc) + __msan_poison(data, size); + if (__msan_get_track_origins()) { + u32 stack_id = StackDepotPut(stack.trace, stack.size); + CHECK(stack_id); + CHECK_EQ((stack_id >> 31), 0); // Higher bit is occupied by stack origins. + __msan_set_origin(data, size, stack_id); + } +} + INTERCEPTOR(void *, mmap, void *addr, SIZE_T length, int prot, int flags, int fd, OFF_T offset) { ENSURE_MSAN_INITED(); -- cgit v1.2.3