forked from OSchip/llvm-project
[HWASan] Use hwasan_memalign for aligned new.
Aligned new does not require size to be a multiple of alignment, so memalign is the correct choice instead of aligned_alloc. Fixes false reports for unaligned sizes. Reviewed By: eugenis Differential Revision: https://reviews.llvm.org/D119161
This commit is contained in:
parent
142cedc283
commit
c990d56d42
|
@ -28,11 +28,11 @@
|
|||
if (!nothrow && UNLIKELY(!res)) \
|
||||
ReportOutOfMemory(size, &stack); \
|
||||
return res
|
||||
# define OPERATOR_NEW_ALIGN_BODY(nothrow) \
|
||||
GET_MALLOC_STACK_TRACE; \
|
||||
void *res = hwasan_aligned_alloc(static_cast<uptr>(align), size, &stack); \
|
||||
if (!nothrow && UNLIKELY(!res)) \
|
||||
ReportOutOfMemory(size, &stack); \
|
||||
# define OPERATOR_NEW_ALIGN_BODY(nothrow) \
|
||||
GET_MALLOC_STACK_TRACE; \
|
||||
void *res = hwasan_memalign(static_cast<uptr>(align), size, &stack); \
|
||||
if (!nothrow && UNLIKELY(!res)) \
|
||||
ReportOutOfMemory(size, &stack); \
|
||||
return res
|
||||
|
||||
# define OPERATOR_DELETE_BODY \
|
||||
|
|
|
@ -1,11 +1,13 @@
|
|||
// Test basic new functionality.
|
||||
// RUN: %clangxx_hwasan %s -o %t
|
||||
// RUN: %clangxx_hwasan -std=c++17 %s -o %t
|
||||
// RUN: %run %t
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <assert.h>
|
||||
#include <sanitizer/hwasan_interface.h>
|
||||
#include <cassert>
|
||||
#include <cstdint>
|
||||
#include <cstdlib>
|
||||
#include <new>
|
||||
#include <sanitizer/allocator_interface.h>
|
||||
#include <sanitizer/hwasan_interface.h>
|
||||
|
||||
int main() {
|
||||
__hwasan_enable_allocator_tagging();
|
||||
|
@ -15,4 +17,14 @@ int main() {
|
|||
assert(a1 != nullptr);
|
||||
assert(__sanitizer_get_allocated_size(a1) == 0);
|
||||
delete[] a1;
|
||||
|
||||
#ifdef __cpp_aligned_new
|
||||
// Aligned new/delete
|
||||
constexpr auto kAlign = std::align_val_t{8};
|
||||
void *a2 = ::operator new(4, kAlign);
|
||||
assert(a2 != nullptr);
|
||||
assert(reinterpret_cast<uintptr_t>(a2) % static_cast<uintptr_t>(kAlign) == 0);
|
||||
assert(__sanitizer_get_allocated_size(a2) >= 4);
|
||||
::operator delete(a2, kAlign);
|
||||
#endif
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue