OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_extension.h" | 5 #include "base/allocator/allocator_extension.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 | 8 |
9 #if defined(USE_TCMALLOC) | 9 #if defined(USE_TCMALLOC) |
10 #include "third_party/tcmalloc/chromium/src/gperftools/heap-profiler.h" | 10 #include "third_party/tcmalloc/chromium/src/gperftools/heap-profiler.h" |
11 #include "third_party/tcmalloc/chromium/src/gperftools/malloc_extension.h" | 11 #include "third_party/tcmalloc/chromium/src/gperftools/malloc_extension.h" |
| 12 #include "third_party/tcmalloc/chromium/src/gperftools/malloc_hook.h" |
12 #endif | 13 #endif |
13 | 14 |
14 namespace base { | 15 namespace base { |
15 namespace allocator { | 16 namespace allocator { |
16 | 17 |
17 void ReleaseFreeMemory() { | 18 void ReleaseFreeMemory() { |
18 #if defined(USE_TCMALLOC) | 19 #if defined(USE_TCMALLOC) |
19 ::MallocExtension::instance()->ReleaseFreeMemory(); | 20 ::MallocExtension::instance()->ReleaseFreeMemory(); |
20 #endif | 21 #endif |
21 } | 22 } |
22 | 23 |
23 bool GetNumericProperty(const char* name, size_t* value) { | 24 bool GetNumericProperty(const char* name, size_t* value) { |
24 #if defined(USE_TCMALLOC) | 25 #if defined(USE_TCMALLOC) |
25 return ::MallocExtension::instance()->GetNumericProperty(name, value); | 26 return ::MallocExtension::instance()->GetNumericProperty(name, value); |
26 #endif | 27 #endif |
27 return false; | 28 return false; |
28 } | 29 } |
29 | 30 |
30 bool IsHeapProfilerRunning() { | 31 bool IsHeapProfilerRunning() { |
31 #if defined(USE_TCMALLOC) | 32 #if defined(USE_TCMALLOC) |
32 return ::IsHeapProfilerRunning(); | 33 return ::IsHeapProfilerRunning(); |
33 #endif | 34 #endif |
34 return false; | 35 return false; |
35 } | 36 } |
36 | 37 |
| 38 void SetHooks(AllocHookFunc alloc_hook, FreeHookFunc free_hook) { |
| 39 // TODO(sque): Use allocator shim layer instead. |
| 40 #if defined(USE_TCMALLOC) |
| 41 // Make sure no hooks get overwritten. |
| 42 auto prev_alloc_hook = MallocHook::SetNewHook(alloc_hook); |
| 43 if (alloc_hook) |
| 44 DCHECK(!prev_alloc_hook); |
| 45 |
| 46 auto prev_free_hook = MallocHook::SetDeleteHook(free_hook); |
| 47 if (free_hook) |
| 48 DCHECK(!prev_free_hook); |
| 49 #endif |
| 50 } |
| 51 |
| 52 int GetCallStack(void** stack, int max_stack_size) { |
| 53 #if defined(USE_TCMALLOC) |
| 54 return MallocHook::GetCallerStackTrace(stack, max_stack_size, 0); |
| 55 #endif |
| 56 return 0; |
| 57 } |
| 58 |
37 } // namespace allocator | 59 } // namespace allocator |
38 } // namespace base | 60 } // namespace base |
OLD | NEW |