llvm-project

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
    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
    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
}
125