OLD | NEW |
| (Empty) |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 // When possible, we implement allocator functions on top of the basic | |
6 // low-level functions malloc() and free(). This way, including a new | |
7 // allocator is as simple as providing just a small interface. | |
8 // | |
9 // As such, this file should not contain any allocator-specific code. | |
10 | |
11 // Implement a C++ style allocation, which always calls the new_handler | |
12 // on failure. | |
13 inline void* generic_cpp_alloc(size_t size, bool nothrow) { | |
14 void* ptr; | |
15 for (;;) { | |
16 ptr = malloc(size); | |
17 if (ptr) | |
18 return ptr; | |
19 if (!call_new_handler(nothrow, size)) | |
20 break; | |
21 } | |
22 return ptr; | |
23 } | |
24 | |
25 extern "C++" { | |
26 | |
27 void* operator new(size_t size) { | |
28 return generic_cpp_alloc(size, false); | |
29 } | |
30 | |
31 void operator delete(void* p) throw() { | |
32 free(p); | |
33 } | |
34 | |
35 void* operator new[](size_t size) { | |
36 return generic_cpp_alloc(size, false); | |
37 } | |
38 | |
39 void operator delete[](void* p) throw() { | |
40 free(p); | |
41 } | |
42 | |
43 void* operator new(size_t size, const std::nothrow_t& nt) { | |
44 return generic_cpp_alloc(size, true); | |
45 } | |
46 | |
47 void operator delete(void* p, const std::nothrow_t& nt) throw() { | |
48 free(p); | |
49 } | |
50 | |
51 void* operator new[](size_t size, const std::nothrow_t& nt) { | |
52 return generic_cpp_alloc(size, true); | |
53 } | |
54 | |
55 void operator delete[](void* p, const std::nothrow_t& nt) throw() { | |
56 free(p); | |
57 } | |
58 | |
59 // This function behaves similarly to MSVC's _set_new_mode. | |
60 // If flag is 0 (default), calls to malloc will behave normally. | |
61 // If flag is 1, calls to malloc will behave like calls to new, | |
62 // and the std_new_handler will be invoked on failure. | |
63 // Returns the previous mode. | |
64 int _set_new_mode(int flag) throw() { | |
65 int old_mode = new_mode; | |
66 new_mode = flag; | |
67 return old_mode; | |
68 } | |
69 | |
70 } // extern "C++" | |
71 | |
72 extern "C" { | |
73 | |
74 void* calloc(size_t n, size_t elem_size) { | |
75 // Overflow check | |
76 const size_t size = n * elem_size; | |
77 if (elem_size != 0 && size / elem_size != n) return NULL; | |
78 | |
79 void* result = malloc(size); | |
80 if (result != NULL) { | |
81 memset(result, 0, size); | |
82 } | |
83 return result; | |
84 } | |
85 | |
86 #ifdef WIN32 | |
87 | |
88 void* _recalloc(void* p, size_t n, size_t elem_size) { | |
89 if (!p) | |
90 return calloc(n, elem_size); | |
91 | |
92 // This API is a bit odd. | |
93 // Note: recalloc only guarantees zeroed memory when p is NULL. | |
94 // Generally, calls to malloc() have padding. So a request | |
95 // to malloc N bytes actually malloc's N+x bytes. Later, if | |
96 // that buffer is passed to recalloc, we don't know what N | |
97 // was anymore. We only know what N+x is. As such, there is | |
98 // no way to know what to zero out. | |
99 const size_t size = n * elem_size; | |
100 if (elem_size != 0 && size / elem_size != n) return NULL; | |
101 return realloc(p, size); | |
102 } | |
103 | |
104 void* _calloc_impl(size_t n, size_t size) { | |
105 return calloc(n, size); | |
106 } | |
107 | |
108 #ifndef NDEBUG | |
109 #undef malloc | |
110 #undef free | |
111 #undef calloc | |
112 | |
113 static int error_handler(int reportType) { | |
114 switch (reportType) { | |
115 case 0: // _CRT_WARN | |
116 __debugbreak(); | |
117 return 0; | |
118 | |
119 case 1: // _CRT_ERROR | |
120 __debugbreak(); | |
121 return 0; | |
122 | |
123 case 2: // _CRT_ASSERT | |
124 __debugbreak(); | |
125 return 0; | |
126 } | |
127 char* p = NULL; | |
128 *p = '\0'; | |
129 return 0; | |
130 } | |
131 | |
132 int _CrtDbgReport(int reportType, | |
133 const char*, | |
134 int, const char*, | |
135 const char*, | |
136 ...) { | |
137 return error_handler(reportType); | |
138 } | |
139 | |
140 int _CrtDbgReportW(int reportType, | |
141 const wchar_t*, | |
142 int, const wchar_t*, | |
143 const wchar_t*, | |
144 ...) { | |
145 return error_handler(reportType); | |
146 } | |
147 | |
148 int _CrtSetReportMode(int, int) { | |
149 return 0; | |
150 } | |
151 | |
152 void* _malloc_dbg(size_t size, int , const char*, int) { | |
153 return malloc(size); | |
154 } | |
155 | |
156 void* _realloc_dbg(void* ptr, size_t size, int, const char*, int) { | |
157 return realloc(ptr, size); | |
158 } | |
159 | |
160 void _free_dbg(void* ptr, int) { | |
161 free(ptr); | |
162 } | |
163 | |
164 void* _calloc_dbg(size_t n, size_t size, int, const char*, int) { | |
165 return calloc(n, size); | |
166 } | |
167 #endif // NDEBUG | |
168 | |
169 #endif // WIN32 | |
170 | |
171 } // extern C | |
172 | |
OLD | NEW |