1/*
2 * Copyright (C) 2018 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#if ENABLE_PHYSICAL_PAGE_MAP
29
30#include "VMAllocate.h"
31#include <unordered_set>
32
33namespace bmalloc {
34
35// This class is useful for debugging bmalloc's footprint.
36class PhysicalPageMap {
37public:
38
39 void commit(void* ptr, size_t size)
40 {
41 forEachPhysicalPage(ptr, size, [&] (void* ptr) {
42 m_physicalPages.insert(ptr);
43 });
44 }
45
46 void decommit(void* ptr, size_t size)
47 {
48 forEachPhysicalPage(ptr, size, [&] (void* ptr) {
49 m_physicalPages.erase(ptr);
50 });
51 }
52
53 size_t footprint()
54 {
55 return static_cast<size_t>(m_physicalPages.size()) * vmPageSizePhysical();
56 }
57
58private:
59 template <typename F>
60 void forEachPhysicalPage(void* ptr, size_t size, F f)
61 {
62 char* begin = roundUpToMultipleOf(vmPageSizePhysical(), static_cast<char*>(ptr));
63 char* end = roundDownToMultipleOf(vmPageSizePhysical(), static_cast<char*>(ptr) + size);
64 while (begin < end) {
65 f(begin);
66 begin += vmPageSizePhysical();
67 }
68 }
69
70 std::unordered_set<void*> m_physicalPages;
71};
72
73} // namespace bmalloc
74
75#endif // ENABLE_PHYSICAL_PAGE_MAP
76