compiler-rt/lib/dfsan/dfsan_new_delete.cpp

   1 //===-- dfsan_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 DataflowSanitizer.
  10 //
  11 // Interceptors for operators new and delete.
  12 //===----------------------------------------------------------------------===//
  13
  14 #include <stddef.h>
  15
  16 #include "dfsan.h"
  17 #include "interception/interception.h"
  18 #include "sanitizer_common/sanitizer_allocator.h"
  19 #include "sanitizer_common/sanitizer_allocator_report.h"
  20
  21 using namespace __dfsan;
  22
  23 // Fake std::nothrow_t and std::align_val_t to avoid including <new>.
  24 namespace std {
  25 struct nothrow_t {};
  26 enum class align_val_t : size_t {};
  27 }  // namespace std
  28
  29 // TODO(alekseys): throw std::bad_alloc instead of dying on OOM.
  30 #define OPERATOR_NEW_BODY(nothrow)   \
  31   void *res = dfsan_malloc(size);    \
  32   if (!nothrow && UNLIKELY(!res)) {  \
  33     UNINITIALIZED BufferedStackTrace stack;        \
  34     ReportOutOfMemory(size, &stack); \
  35   }                                  \
  36   return res
  37 #define OPERATOR_NEW_BODY_ALIGN(nothrow)         \
  38   void *res = dfsan_memalign((uptr)align, size); \
  39   if (!nothrow && UNLIKELY(!res)) {              \
  40     UNINITIALIZED BufferedStackTrace stack;                    \
  41     ReportOutOfMemory(size, &stack);             \
  42   }                                              \
  43   return res;
  44
  45 INTERCEPTOR_ATTRIBUTE
  46 void *operator new(size_t size) { OPERATOR_NEW_BODY(false /*nothrow*/); }
  47 INTERCEPTOR_ATTRIBUTE
  48 void *operator new[](size_t size) { OPERATOR_NEW_BODY(false /*nothrow*/); }
  49 INTERCEPTOR_ATTRIBUTE
  50 void *operator new(size_t size, std::nothrow_t const &) {
  51   OPERATOR_NEW_BODY(true /*nothrow*/);
  52 }
  53 INTERCEPTOR_ATTRIBUTE
  54 void *operator new[](size_t size, std::nothrow_t const &) {
  55   OPERATOR_NEW_BODY(true /*nothrow*/);
  56 }
  57 INTERCEPTOR_ATTRIBUTE
  58 void *operator new(size_t size, std::align_val_t align) {
  59   OPERATOR_NEW_BODY_ALIGN(false /*nothrow*/);
  60 }
  61 INTERCEPTOR_ATTRIBUTE
  62 void *operator new[](size_t size, std::align_val_t align) {
  63   OPERATOR_NEW_BODY_ALIGN(false /*nothrow*/);
  64 }
  65 INTERCEPTOR_ATTRIBUTE
  66 void *operator new(size_t size, std::align_val_t align,
  67                    std::nothrow_t const &) {
  68   OPERATOR_NEW_BODY_ALIGN(true /*nothrow*/);
  69 }
  70 INTERCEPTOR_ATTRIBUTE
  71 void *operator new[](size_t size, std::align_val_t align,
  72                      std::nothrow_t const &) {
  73   OPERATOR_NEW_BODY_ALIGN(true /*nothrow*/);
  74 }
  75
  76 #define OPERATOR_DELETE_BODY \
  77   if (ptr)                   \
  78   dfsan_deallocate(ptr)
  79
  80 INTERCEPTOR_ATTRIBUTE
  81 void operator delete(void *ptr)NOEXCEPT { OPERATOR_DELETE_BODY; }
  82 INTERCEPTOR_ATTRIBUTE
  83 void operator delete[](void *ptr) NOEXCEPT { OPERATOR_DELETE_BODY; }
  84 INTERCEPTOR_ATTRIBUTE
  85 void operator delete(void *ptr, std::nothrow_t const &) {
  86   OPERATOR_DELETE_BODY;
  87 }
  88 INTERCEPTOR_ATTRIBUTE
  89 void operator delete[](void *ptr, std::nothrow_t const &) {
  90   OPERATOR_DELETE_BODY;
  91 }
  92 INTERCEPTOR_ATTRIBUTE
  93 void operator delete(void *ptr, size_t size)NOEXCEPT { OPERATOR_DELETE_BODY; }
  94 INTERCEPTOR_ATTRIBUTE
  95 void operator delete[](void *ptr, size_t size) NOEXCEPT {
  96   OPERATOR_DELETE_BODY;
  97 }
  98 INTERCEPTOR_ATTRIBUTE
  99 void operator delete(void *ptr, std::align_val_t align)NOEXCEPT {
 100   OPERATOR_DELETE_BODY;
 101 }
 102 INTERCEPTOR_ATTRIBUTE
 103 void operator delete[](void *ptr, std::align_val_t align) NOEXCEPT {
 104   OPERATOR_DELETE_BODY;
 105 }
 106 INTERCEPTOR_ATTRIBUTE
 107 void operator delete(void *ptr, std::align_val_t align,
 108                      std::nothrow_t const &) {
 109   OPERATOR_DELETE_BODY;
 110 }
 111 INTERCEPTOR_ATTRIBUTE
 112 void operator delete[](void *ptr, std::align_val_t align,
 113                        std::nothrow_t const &) {
 114   OPERATOR_DELETE_BODY;
 115 }
 116 INTERCEPTOR_ATTRIBUTE
 117 void operator delete(void *ptr, size_t size, std::align_val_t align)NOEXCEPT {
 118   OPERATOR_DELETE_BODY;
 119 }
 120 INTERCEPTOR_ATTRIBUTE
 121 void operator delete[](void *ptr, size_t size,
 122                        std::align_val_t align) NOEXCEPT {
 123   OPERATOR_DELETE_BODY;
 124 }