libsanitizer/hwasan/hwasan_new_delete.cpp

   1 //===-- hwasan_new_delete.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-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file is a part of HWAddressSanitizer.
  10 //
  11 // Interceptors for operators new and delete.
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "hwasan.h"
  15 #include "interception/interception.h"
  16 #include "sanitizer_common/sanitizer_allocator.h"
  17 #include "sanitizer_common/sanitizer_allocator_report.h"
  18
  19 #include <stddef.h>
  20 #include <stdlib.h>
  21
  22 #if HWASAN_REPLACE_OPERATORS_NEW_AND_DELETE
  23
  24 // TODO(alekseys): throw std::bad_alloc instead of dying on OOM.
  25 #  define OPERATOR_NEW_BODY(nothrow)         \
  26     GET_MALLOC_STACK_TRACE;                  \
  27     void *res = hwasan_malloc(size, &stack); \
  28     if (!nothrow && UNLIKELY(!res))          \
  29       ReportOutOfMemory(size, &stack);       \
  30     return res
  31 #  define OPERATOR_NEW_ALIGN_BODY(nothrow)                               \
  32     GET_MALLOC_STACK_TRACE;                                              \
  33     void *res = hwasan_memalign(static_cast<uptr>(align), size, &stack); \
  34     if (!nothrow && UNLIKELY(!res))                                      \
  35       ReportOutOfMemory(size, &stack);                                   \
  36     return res
  37
  38 #  define OPERATOR_DELETE_BODY \
  39     GET_MALLOC_STACK_TRACE;    \
  40     if (ptr)                   \
  41     hwasan_free(ptr, &stack)
  42
  43 #elif defined(__ANDROID__)
  44
  45 // We don't actually want to intercept operator new and delete on Android, but
  46 // since we previously released a runtime that intercepted these functions,
  47 // removing the interceptors would break ABI. Therefore we simply forward to
  48 // malloc and free.
  49 #  define OPERATOR_NEW_BODY(nothrow) return malloc(size)
  50 #  define OPERATOR_DELETE_BODY free(ptr)
  51
  52 #endif
  53
  54 #ifdef OPERATOR_NEW_BODY
  55
  56 using namespace __hwasan;
  57
  58 // Fake std::nothrow_t to avoid including <new>.
  59 namespace std {
  60 struct nothrow_t {};
  61 }  // namespace std
  62
  63 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new(size_t size) {
  64   OPERATOR_NEW_BODY(false /*nothrow*/);
  65 }
  66 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new[](
  67     size_t size) {
  68   OPERATOR_NEW_BODY(false /*nothrow*/);
  69 }
  70 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new(
  71     size_t size, std::nothrow_t const &) {
  72   OPERATOR_NEW_BODY(true /*nothrow*/);
  73 }
  74 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new[](
  75     size_t size, std::nothrow_t const &) {
  76   OPERATOR_NEW_BODY(true /*nothrow*/);
  77 }
  78
  79 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete(
  80     void *ptr) NOEXCEPT {
  81   OPERATOR_DELETE_BODY;
  82 }
  83 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete[](
  84     void *ptr) NOEXCEPT {
  85   OPERATOR_DELETE_BODY;
  86 }
  87 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete(
  88     void *ptr, std::nothrow_t const &) {
  89   OPERATOR_DELETE_BODY;
  90 }
  91 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete[](
  92     void *ptr, std::nothrow_t const &) {
  93   OPERATOR_DELETE_BODY;
  94 }
  95
  96 #endif  // OPERATOR_NEW_BODY
  97
  98 #ifdef OPERATOR_NEW_ALIGN_BODY
  99
 100 namespace std {
 101 enum class align_val_t : size_t {};
 102 }  // namespace std
 103
 104 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new(
 105     size_t size, std::align_val_t align) {
 106   OPERATOR_NEW_ALIGN_BODY(false /*nothrow*/);
 107 }
 108 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new[](
 109     size_t size, std::align_val_t align) {
 110   OPERATOR_NEW_ALIGN_BODY(false /*nothrow*/);
 111 }
 112 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new(
 113     size_t size, std::align_val_t align, std::nothrow_t const &) {
 114   OPERATOR_NEW_ALIGN_BODY(true /*nothrow*/);
 115 }
 116 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void *operator new[](
 117     size_t size, std::align_val_t align, std::nothrow_t const &) {
 118   OPERATOR_NEW_ALIGN_BODY(true /*nothrow*/);
 119 }
 120
 121 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete(
 122     void *ptr, std::align_val_t align) NOEXCEPT {
 123   OPERATOR_DELETE_BODY;
 124 }
 125 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete[](
 126     void *ptr, std::align_val_t) NOEXCEPT {
 127   OPERATOR_DELETE_BODY;
 128 }
 129 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete(
 130     void *ptr, std::align_val_t, std::nothrow_t const &) NOEXCEPT {
 131   OPERATOR_DELETE_BODY;
 132 }
 133 INTERCEPTOR_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void operator delete[](
 134     void *ptr, std::align_val_t, std::nothrow_t const &) NOEXCEPT {
 135   OPERATOR_DELETE_BODY;
 136 }
 137
 138 #endif  // OPERATOR_NEW_ALIGN_BODY