Chromium Code Reviews| 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 #include "build/build_config.h" | |
| 8 | 9 |
| 9 namespace base { | 10 namespace base { |
| 10 namespace allocator { | 11 namespace allocator { |
| 11 | 12 |
| 13 #if defined(USE_TCMALLOC) | |
| 14 // Since the base library does not know about allocator, the initialization | |
| 15 // method would call this function which is declared as a weak symbol. This | |
| 16 // should be replaced by a strong symbol linked from the actual allocator | |
| 17 // library. The executable target is expected include an allocator shim layer | |
|
Primiano Tucci (use gerrit)
2015/11/25 12:47:17
remove "shim layer"
| |
| 18 // replaces this weak symbol. | |
| 19 // Such an implementation of this method should return a set of callbacks that | |
| 20 // are used to implement the required allocator extension functions. | |
| 21 // This method is defined in .cc file so that the allocator implementation can | |
| 22 // still include the header file without getting a "weak" declaration of this | |
| 23 // function. | |
| 24 __attribute__((weak, visibility("default"), noinline)) | |
| 25 thunks::AllocatorExtensionFunctions | |
| 26 InitializeAllocatorWeak(); | |
| 27 #endif | |
| 28 | |
| 29 void InitializeAllocator() { | |
| 30 #if defined(USE_TCMALLOC) | |
| 31 // This calls into the weak function declared above. If a specific allocator | |
| 32 // implementation needs initialization then it should define and link a strong | |
| 33 // symbol of "InitializeAllocatorWeak" function, that returns the necessary | |
| 34 // callbacks. | |
| 35 thunks::AllocatorExtensionFunctions functions = InitializeAllocatorWeak(); | |
| 36 thunks::SetAllocatorExtensionFunctions(functions); | |
| 37 #endif | |
| 38 } | |
| 39 | |
| 12 bool GetAllocatorWasteSize(size_t* size) { | 40 bool GetAllocatorWasteSize(size_t* size) { |
| 13 thunks::GetAllocatorWasteSizeFunction get_allocator_waste_size_function = | 41 auto get_allocator_waste_size_function = |
| 14 thunks::GetGetAllocatorWasteSizeFunction(); | 42 thunks::GetAllocatorExtensionFunctions().get_allocator_waste_size; |
| 15 return get_allocator_waste_size_function != NULL && | 43 return get_allocator_waste_size_function != NULL && |
| 16 get_allocator_waste_size_function(size); | 44 get_allocator_waste_size_function(size); |
| 17 } | 45 } |
| 18 | 46 |
| 19 void GetStats(char* buffer, int buffer_length) { | 47 void GetStats(char* buffer, int buffer_length) { |
| 20 DCHECK_GT(buffer_length, 0); | 48 DCHECK_GT(buffer_length, 0); |
| 21 thunks::GetStatsFunction get_stats_function = thunks::GetGetStatsFunction(); | 49 auto get_stats_function = thunks::GetAllocatorExtensionFunctions().get_stats; |
| 22 if (get_stats_function) | 50 if (get_stats_function) |
| 23 get_stats_function(buffer, buffer_length); | 51 get_stats_function(buffer, buffer_length); |
| 24 else | 52 else |
| 25 buffer[0] = '\0'; | 53 buffer[0] = '\0'; |
| 26 } | 54 } |
| 27 | 55 |
| 28 void ReleaseFreeMemory() { | 56 void ReleaseFreeMemory() { |
| 29 thunks::ReleaseFreeMemoryFunction release_free_memory_function = | 57 auto release_free_memory_function = |
| 30 thunks::GetReleaseFreeMemoryFunction(); | 58 thunks::GetAllocatorExtensionFunctions().release_free_memory; |
| 31 if (release_free_memory_function) | 59 if (release_free_memory_function) |
| 32 release_free_memory_function(); | 60 release_free_memory_function(); |
| 33 } | 61 } |
| 34 | 62 |
| 35 void SetGetAllocatorWasteSizeFunction( | 63 size_t GetBytesAllocatedOnCurrentThread() { |
| 36 thunks::GetAllocatorWasteSizeFunction get_allocator_waste_size_function) { | 64 auto get_bytes_allocated_on_current_thread_function = |
| 37 DCHECK_EQ(thunks::GetGetAllocatorWasteSizeFunction(), | 65 thunks::GetAllocatorExtensionFunctions() |
| 38 reinterpret_cast<thunks::GetAllocatorWasteSizeFunction>(NULL)); | 66 .get_bytes_allocated_on_current_thread; |
| 39 thunks::SetGetAllocatorWasteSizeFunction(get_allocator_waste_size_function); | 67 DCHECK(get_bytes_allocated_on_current_thread_function); |
| 68 return get_bytes_allocated_on_current_thread_function(); | |
| 40 } | 69 } |
| 41 | 70 |
| 42 void SetGetStatsFunction(thunks::GetStatsFunction get_stats_function) { | 71 bool GetNumericProperty(const char* name, size_t* value) { |
| 43 DCHECK_EQ(thunks::GetGetStatsFunction(), | 72 auto get_numeric_property_function = |
| 44 reinterpret_cast<thunks::GetStatsFunction>(NULL)); | 73 thunks::GetAllocatorExtensionFunctions().get_numeric_property; |
| 45 thunks::SetGetStatsFunction(get_stats_function); | 74 return get_numeric_property_function != NULL && |
| 75 get_numeric_property_function(name, value); | |
| 46 } | 76 } |
| 47 | 77 |
| 48 void SetReleaseFreeMemoryFunction( | 78 void HeapProfilerStart(StackGeneratorFunction callback) { |
| 49 thunks::ReleaseFreeMemoryFunction release_free_memory_function) { | 79 auto heap_profiler_start_function = |
| 50 DCHECK_EQ(thunks::GetReleaseFreeMemoryFunction(), | 80 thunks::GetAllocatorExtensionFunctions().heap_profiler_start; |
| 51 reinterpret_cast<thunks::ReleaseFreeMemoryFunction>(NULL)); | 81 if (heap_profiler_start_function) |
| 52 thunks::SetReleaseFreeMemoryFunction(release_free_memory_function); | 82 heap_profiler_start_function(callback); |
| 53 } | 83 } |
| 54 | 84 |
| 55 void SetGetNumericPropertyFunction( | 85 void HeapProfilerStop() { |
| 56 thunks::GetNumericPropertyFunction get_numeric_property_function) { | 86 auto heap_profiler_stop_function = |
| 57 DCHECK_EQ(thunks::GetGetNumericPropertyFunction(), | 87 thunks::GetAllocatorExtensionFunctions().heap_profiler_stop; |
| 58 reinterpret_cast<thunks::GetNumericPropertyFunction>(NULL)); | 88 if (heap_profiler_stop_function) |
| 59 thunks::SetGetNumericPropertyFunction(get_numeric_property_function); | 89 heap_profiler_stop_function(); |
| 90 } | |
| 91 | |
| 92 char* GetHeapProfile() { | |
| 93 auto get_heap_profile_function = | |
| 94 thunks::GetAllocatorExtensionFunctions().get_heap_profile; | |
| 95 if (get_heap_profile_function) | |
| 96 return get_heap_profile_function(); | |
| 97 return NULL; | |
| 98 } | |
| 99 | |
| 100 void HeapProfilerDump(const char* reason) { | |
| 101 auto heap_profiler_dump_function = | |
| 102 thunks::GetAllocatorExtensionFunctions().heap_profiler_dump; | |
| 103 if (heap_profiler_dump_function) | |
| 104 heap_profiler_dump_function(reason); | |
| 105 } | |
| 106 | |
| 107 bool IsHeapProfilerRunning() { | |
| 108 auto is_heap_profiler_running_function = | |
| 109 thunks::GetAllocatorExtensionFunctions().is_heap_profiler_running; | |
| 110 return is_heap_profiler_running_function != NULL && | |
| 111 is_heap_profiler_running_function(); | |
| 60 } | 112 } |
| 61 | 113 |
| 62 } // namespace allocator | 114 } // namespace allocator |
| 63 } // namespace base | 115 } // namespace base |
| OLD | NEW |