Chromium Code Reviews| Index: base/debug/thread_heap_usage_tracker.h |
| diff --git a/base/debug/thread_heap_usage_tracker.h b/base/debug/thread_heap_usage_tracker.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ba9425afd7c9986edb6574ded99e280406ca27e0 |
| --- /dev/null |
| +++ b/base/debug/thread_heap_usage_tracker.h |
| @@ -0,0 +1,118 @@ |
| +// Copyright 2016 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#ifndef BASE_DEBUG_THREAD_HEAP_USAGE_TRACKER_H_ |
| +#define BASE_DEBUG_THREAD_HEAP_USAGE_TRACKER_H_ |
| + |
| +#include <stdint.h> |
| + |
| +#include "base/allocator/features.h" |
| +#include "base/base_export.h" |
| +#include "base/threading/thread_checker.h" |
| + |
| +namespace base { |
| +namespace allocator { |
| +struct AllocatorDispatch; |
| +} // namespace allocator |
| + |
| +namespace debug { |
| + |
| +// Used to store the heap allocator usage in a scope. |
| +struct ThreadHeapUsage { |
| + // The cumulative number of allocation operations. |
| + uint64_t alloc_ops; |
| + |
| + // The cumulative number of allocated bytes. Where available, this is |
| + // inclusive heap padding and estimated or actual heap overhead. |
| + uint64_t alloc_bytes; |
| + |
| + // Where available, cumulative number of heap padding heap |
|
chrisha
2016/10/17 19:17:48
heap padding heap?
Sigurður Ásgeirsson
2016/10/17 19:35:23
Done.
|
| + // and overhead bytes. |
| + uint64_t alloc_overhead_bytes; |
| + |
| + // The cumulative number of free operations. |
| + uint64_t free_ops; |
| + |
| + // The cumulative number of bytes freed. |
| + // Only recorded if the underlying heap shim can return the size of an |
| + // allocation. |
| + uint64_t free_bytes; |
| + |
| + // The maximal value of alloc_bytes - free_bytes seen for this thread. |
| + // Only recorded if the underlying heap shim supports returning the size of |
| + // an allocation. |
| + uint64_t max_allocated_bytes; |
| +}; |
| + |
| +// By keeping a tally on heap operations, it's possible to track: |
| +// - the number of alloc/free operations, where a realloc is zero or one |
| +// of each, depending on the input parameters (see man realloc). |
| +// - the number of bytes allocated/freed. |
| +// - the number of estimated bytes of heap overhead used. |
| +// - the high-watermark amount of bytes allocated in the scope. |
| +// This in turn allows measuring the memory usage and memory usage churn over |
| +// a scope. Scopes must be cleanly nested, and each scope must be |
| +// destroyed on the thread where it's created. |
| +// |
| +// Note that this depends on the capabilities of the underlying heap shim. If |
| +// that shim can not yield a size estimate for an allocation, it's not possible |
| +// to keep track of overhead, freed bytes and the allocation high water mark. |
| +class BASE_EXPORT ThreadHeapUsageTracker { |
| + public: |
| + ThreadHeapUsageTracker(); |
| + ~ThreadHeapUsageTracker(); |
| + |
| + // Start tracking heap usage on this thread. |
| + // Note |IsHeapTrackingEnabled()| must be true. |
| + void Start(); |
| + |
| + // Stop tracking heap usage on this thread and store the usage tallied. |
| + // If |usage_is_exclusive| is true, the usage tallied won't be added to the |
| + // outer scope's usage. If |usage_is_exclusive" is false, the usage tallied |
|
chrisha
2016/10/17 19:17:48
|usage_is_exclusive"
Sigurður Ásgeirsson
2016/10/17 19:35:23
Done.
|
| + // in this scope will also tally to any outer scope. |
| + void Stop(bool usage_is_exclusive); |
| + |
| + // After Stop() returns the usage tallied from Start() to Stop(). |
| + const ThreadHeapUsage& usage() const { return usage_; } |
| + |
| + // Returns this thread's allocator usage from the start of the innermost |
| + // enclosing ThreadHeapUsageTracker instance, if any. |
| + static ThreadHeapUsage CurrentUsage(); |
| + |
| + // Enables the heap intercept. May only be called once, and only if the heap |
| + // shim is available, e.g. if BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM) is |
| + // true. |
| + static void EnableHeapTracking(); |
| + |
| + // Returns true iff heap tracking is enabled. |
| + static bool IsHeapTrackingEnabled(); |
| + |
| + protected: |
| + // Exposed for testing only - note that it's safe to re-EnableHeapTracking() |
| + // after calling this function in tests. |
| + static void DisableHeapTrackingForTesting(); |
| + |
| + // Exposed for testing only. |
| + static void EnsureTLSInitializedForTesting(); |
| + |
| + // Exposed to allow testing the shim without inserting it in the allocator |
| + // shim chain. |
| + static base::allocator::AllocatorDispatch* GetDispatchForTesting(); |
| + |
| + private: |
| + static void EnsureTLSInitialized(); |
| + |
| + ThreadChecker thread_checker_; |
| + |
| + // The allocator usage at Start(), or the difference from Start() to Stop(). |
| + ThreadHeapUsage usage_; |
| + |
| + // This thread's allocator usage, non-null from Start() to Stop(). |
| + ThreadHeapUsage* thread_usage_; |
| +}; |
| + |
| +} // namespace debug |
| +} // namespace base |
| + |
| +#endif // BASE_DEBUG_THREAD_HEAP_USAGE_TRACKER_H_ |