OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/allocator/allocator_shim.h" | 5 #include "base/allocator/allocator_shim.h" |
6 | 6 |
7 #include <malloc.h> | 7 #include <malloc.h> |
8 | 8 |
9 // This translation unit defines a default dispatch for the allocator shim which | 9 // This translation unit defines a default dispatch for the allocator shim which |
10 // routes allocations to libc functions. | 10 // routes allocations to libc functions. |
11 // The code here is strongly inspired from tcmalloc's libc_override_glibc.h. | 11 // The code here is strongly inspired from tcmalloc's libc_override_glibc.h. |
12 | 12 |
13 extern "C" { | 13 extern "C" { |
14 void* __libc_malloc(size_t size); | 14 void* __libc_malloc(size_t size); |
15 void* __libc_calloc(size_t n, size_t size); | 15 void* __libc_calloc(size_t n, size_t size); |
16 void* __libc_realloc(void* address, size_t size); | 16 void* __libc_realloc(void* address, size_t size); |
17 void* __libc_memalign(size_t alignment, size_t size); | 17 void* __libc_memalign(size_t alignment, size_t size); |
18 void __libc_free(void* ptr); | 18 void __libc_free(void* ptr); |
19 } // extern "C" | 19 } // extern "C" |
20 | 20 |
21 namespace { | 21 namespace { |
22 | 22 |
23 using base::allocator::AllocatorDispatch; | 23 using base::allocator::AllocatorDispatch; |
24 | 24 |
25 void* GlibcMalloc(const AllocatorDispatch*, size_t size) { | 25 void* GlibcMalloc(const AllocatorDispatch*, size_t size, void* context) { |
26 return __libc_malloc(size); | 26 return __libc_malloc(size); |
27 } | 27 } |
28 | 28 |
29 void* GlibcCalloc(const AllocatorDispatch*, size_t n, size_t size) { | 29 void* GlibcCalloc(const AllocatorDispatch*, |
| 30 size_t n, |
| 31 size_t size, |
| 32 void* context) { |
30 return __libc_calloc(n, size); | 33 return __libc_calloc(n, size); |
31 } | 34 } |
32 | 35 |
33 void* GlibcRealloc(const AllocatorDispatch*, void* address, size_t size) { | 36 void* GlibcRealloc(const AllocatorDispatch*, |
| 37 void* address, |
| 38 size_t size, |
| 39 void* context) { |
34 return __libc_realloc(address, size); | 40 return __libc_realloc(address, size); |
35 } | 41 } |
36 | 42 |
37 void* GlibcMemalign(const AllocatorDispatch*, size_t alignment, size_t size) { | 43 void* GlibcMemalign(const AllocatorDispatch*, |
| 44 size_t alignment, |
| 45 size_t size, |
| 46 void* context) { |
38 return __libc_memalign(alignment, size); | 47 return __libc_memalign(alignment, size); |
39 } | 48 } |
40 | 49 |
41 void GlibcFree(const AllocatorDispatch*, void* address) { | 50 void GlibcFree(const AllocatorDispatch*, void* address, void* context) { |
42 __libc_free(address); | 51 __libc_free(address); |
43 } | 52 } |
44 | 53 |
45 size_t GlibcGetSizeEstimate(const AllocatorDispatch*, void* address) { | 54 size_t GlibcGetSizeEstimate(const AllocatorDispatch*, |
| 55 void* address, |
| 56 void* context) { |
46 // TODO(siggi, primiano): malloc_usable_size may need redirection in the | 57 // TODO(siggi, primiano): malloc_usable_size may need redirection in the |
47 // presence of interposing shims that divert allocations. | 58 // presence of interposing shims that divert allocations. |
48 return malloc_usable_size(address); | 59 return malloc_usable_size(address); |
49 } | 60 } |
50 | 61 |
51 } // namespace | 62 } // namespace |
52 | 63 |
53 const AllocatorDispatch AllocatorDispatch::default_dispatch = { | 64 const AllocatorDispatch AllocatorDispatch::default_dispatch = { |
54 &GlibcMalloc, /* alloc_function */ | 65 &GlibcMalloc, /* alloc_function */ |
55 &GlibcCalloc, /* alloc_zero_initialized_function */ | 66 &GlibcCalloc, /* alloc_zero_initialized_function */ |
56 &GlibcMemalign, /* alloc_aligned_function */ | 67 &GlibcMemalign, /* alloc_aligned_function */ |
57 &GlibcRealloc, /* realloc_function */ | 68 &GlibcRealloc, /* realloc_function */ |
58 &GlibcFree, /* free_function */ | 69 &GlibcFree, /* free_function */ |
59 &GlibcGetSizeEstimate, /* get_size_estimate_function */ | 70 &GlibcGetSizeEstimate, /* get_size_estimate_function */ |
60 nullptr, /* batch_malloc_function */ | 71 nullptr, /* batch_malloc_function */ |
61 nullptr, /* batch_free_function */ | 72 nullptr, /* batch_free_function */ |
62 nullptr, /* free_definite_size_function */ | 73 nullptr, /* free_definite_size_function */ |
63 nullptr, /* next */ | 74 nullptr, /* next */ |
64 }; | 75 }; |
OLD | NEW |