Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef BASE_MEMORY_SHARED_MEMORY_TRACKER_H_ | |
| 6 #define BASE_MEMORY_SHARED_MEMORY_TRACKER_H_ | |
| 7 | |
| 8 #include "base/atomicops.h" | |
| 9 #include "base/memory/shared_memory_handle.h" | |
| 10 #include "base/memory/singleton.h" | |
| 11 #include "base/trace_event/memory_dump_provider.h" | |
| 12 | |
| 13 namespace base { | |
| 14 | |
| 15 class SharedMemory; | |
| 16 | |
| 17 namespace trace_event { | |
| 18 | |
| 19 class MemoryAllocatorDumpGuid; | |
| 20 class ProcessMemoryDump; | |
| 21 } | |
| 22 | |
| 23 class BASE_EXPORT SharedMemoryTracker | |
| 24 : public base::trace_event::MemoryDumpProvider { | |
| 25 public: | |
| 26 // Returns a singleton instance. | |
| 27 static SharedMemoryTracker* GetInstance(); | |
| 28 | |
| 29 static bool AddOwnershipEdgeToSharedGlobalDump( | |
| 30 trace_event::ProcessMemoryDump* pmd, | |
| 31 const trace_event::MemoryAllocatorDumpGuid& source, | |
| 32 const base::SharedMemoryHandle& shared_memory_handle, | |
| 33 size_t mapped_size); | |
| 34 | |
| 35 // base::trace_event::MemoryDumpProvider implementation. | |
| 36 bool OnMemoryDump(const base::trace_event::MemoryDumpArgs& args, | |
| 37 base::trace_event::ProcessMemoryDump* pmd) override; | |
| 38 | |
| 39 bool IncrementMemoryUsage(const SharedMemory& shared_memory); | |
| 40 bool DecrementMemoryUsage(const SharedMemory& shared_memory); | |
| 41 | |
| 42 private: | |
| 43 struct TrackerBucket { | |
| 44 TrackerBucket(); | |
| 45 ~TrackerBucket(); | |
| 46 base::Lock lock; | |
| 47 // The first map's key is a device ID and the second map's key is a file | |
| 48 // ID. See SharedMemoryHandleID struct. | |
| 49 std::unordered_map<int64_t, std::unordered_map<int64_t, size_t>> usage; | |
|
Primiano Tucci (use gerrit)
2017/01/20 16:34:52
I don't think you should expose device id and file
hajimehoshi
2017/01/23 11:59:09
Done.
| |
| 50 }; | |
| 51 | |
| 52 friend struct base::DefaultSingletonTraits<SharedMemoryTracker>; | |
| 53 | |
| 54 static const int kTrackerBucketNum = 1024; | |
| 55 | |
| 56 SharedMemoryTracker(); | |
| 57 ~SharedMemoryTracker() override; | |
| 58 | |
| 59 TrackerBucket buckets_[kTrackerBucketNum]; | |
| 60 | |
| 61 DISALLOW_COPY_AND_ASSIGN(SharedMemoryTracker); | |
| 62 }; | |
| 63 | |
| 64 } // namespace base | |
| 65 | |
| 66 #endif // BASE_MEMORY_SHARED_MEMORY_TRACKER_H_ | |
| OLD | NEW |