OLD | NEW |
1 // Copyright 2016 the V8 project authors. All rights reserved. | 1 // Copyright 2016 the V8 project 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 "src/base/accounting-allocator.h" | 5 #include "src/base/accounting-allocator.h" |
6 | 6 |
7 #include <cstdlib> | 7 #include <cstdlib> |
8 | 8 |
9 #if V8_LIBC_BIONIC | 9 #if V8_LIBC_BIONIC |
10 #include <malloc.h> // NOLINT | 10 #include <malloc.h> // NOLINT |
11 #endif | 11 #endif |
12 | 12 |
13 namespace v8 { | 13 namespace v8 { |
14 namespace base { | 14 namespace base { |
15 | 15 |
16 void* AccountingAllocator::Allocate(size_t bytes) { | 16 void* AccountingAllocator::Allocate(size_t bytes) { |
17 void* memory = malloc(bytes); | 17 void* memory = malloc(bytes); |
18 if (memory) { | 18 if (memory) { |
19 AtomicWord current = | 19 ChangeCurrentMemoryUsage(bytes); |
20 NoBarrier_AtomicIncrement(¤t_memory_usage_, bytes); | |
21 AtomicWord max = NoBarrier_Load(&max_memory_usage_); | |
22 while (current > max) { | |
23 max = NoBarrier_CompareAndSwap(&max_memory_usage_, max, current); | |
24 } | |
25 } | 20 } |
26 return memory; | 21 return memory; |
27 } | 22 } |
28 | 23 |
29 void AccountingAllocator::Free(void* memory, size_t bytes) { | 24 void AccountingAllocator::Free(void* memory, size_t bytes) { |
30 free(memory); | 25 free(memory); |
31 NoBarrier_AtomicIncrement(¤t_memory_usage_, | 26 ChangeCurrentMemoryUsage(-static_cast<int64_t>(bytes)); |
32 -static_cast<AtomicWord>(bytes)); | 27 } |
| 28 |
| 29 void AccountingAllocator::ChangeCurrentMemoryUsage(const int64_t bytes) { |
| 30 AtomicWord current = NoBarrier_AtomicIncrement(¤t_memory_usage_, bytes); |
| 31 AtomicWord max = NoBarrier_Load(&max_memory_usage_); |
| 32 if (bytes > 0) { |
| 33 while (current > max) { |
| 34 max = NoBarrier_CompareAndSwap(&max_memory_usage_, max, current); |
| 35 } |
| 36 } |
33 } | 37 } |
34 | 38 |
35 size_t AccountingAllocator::GetCurrentMemoryUsage() const { | 39 size_t AccountingAllocator::GetCurrentMemoryUsage() const { |
36 return NoBarrier_Load(¤t_memory_usage_); | 40 return NoBarrier_Load(¤t_memory_usage_); |
37 } | 41 } |
38 | 42 |
39 size_t AccountingAllocator::GetMaxMemoryUsage() const { | 43 size_t AccountingAllocator::GetMaxMemoryUsage() const { |
40 return NoBarrier_Load(&max_memory_usage_); | 44 return NoBarrier_Load(&max_memory_usage_); |
41 } | 45 } |
42 | 46 |
43 } // namespace base | 47 } // namespace base |
44 } // namespace v8 | 48 } // namespace v8 |
OLD | NEW |