bmalloc should support strictly type-segregated isolated heaps
[WebKit-https.git] / Source / bmalloc / bmalloc / IsoHeapImplInlines.h
1 /*
2  * Copyright (C) 2017 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 #include "IsoHeapImpl.h"
29 #include "IsoTLSDeallocatorEntry.h"
30
31 namespace bmalloc {
32
33 template<typename Config>
34 IsoHeapImpl<Config>::IsoHeapImpl()
35     : lock(PerProcess<IsoTLSDeallocatorEntry<Config>>::get()->lock)
36     , m_inlineDirectory(*this)
37     , m_allocator(*this)
38 {
39 }
40
41 template<typename Config>
42 EligibilityResult<Config> IsoHeapImpl<Config>::takeFirstEligible()
43 {
44     if (m_isInlineDirectoryEligible) {
45         EligibilityResult<Config> result = m_inlineDirectory.takeFirstEligible();
46         if (result.kind == EligibilityKind::Full)
47             m_isInlineDirectoryEligible = false;
48         else
49             return result;
50     }
51     
52     if (!m_firstEligibleDirectory) {
53         // If nothing is eligible, it can only be because we have no directories. It wouldn't be the end
54         // of the world if we broke this invariant. It would only mean that didBecomeEligible() would need
55         // a null check.
56         RELEASE_BASSERT(!m_headDirectory);
57         RELEASE_BASSERT(!m_tailDirectory);
58     }
59     
60     for (; m_firstEligibleDirectory; m_firstEligibleDirectory = m_firstEligibleDirectory->next) {
61         EligibilityResult<Config> result = m_firstEligibleDirectory->payload.takeFirstEligible();
62         if (result.kind != EligibilityKind::Full)
63             return result;
64     }
65     
66     auto* newDirectory = new IsoDirectoryPage<Config>(*this, m_numDirectoryPages++);
67     if (m_headDirectory) {
68         m_tailDirectory->next = newDirectory;
69         m_tailDirectory = newDirectory;
70     } else {
71         RELEASE_BASSERT(!m_tailDirectory);
72         m_headDirectory = newDirectory;
73         m_tailDirectory = newDirectory;
74     }
75     m_firstEligibleDirectory = newDirectory;
76     EligibilityResult<Config> result = newDirectory->payload.takeFirstEligible();
77     RELEASE_BASSERT(result.kind != EligibilityKind::Full);
78     return result;
79 }
80
81 template<typename Config>
82 void IsoHeapImpl<Config>::didBecomeEligible(IsoDirectory<Config, numPagesInInlineDirectory>* directory)
83 {
84     RELEASE_BASSERT(directory == &m_inlineDirectory);
85     m_isInlineDirectoryEligible = true;
86 }
87
88 template<typename Config>
89 void IsoHeapImpl<Config>::didBecomeEligible(IsoDirectory<Config, IsoDirectoryPage<Config>::numPages>* directory)
90 {
91     RELEASE_BASSERT(m_firstEligibleDirectory);
92     auto* directoryPage = IsoDirectoryPage<Config>::pageFor(directory);
93     if (directoryPage->index() < m_firstEligibleDirectory->index())
94         m_firstEligibleDirectory = directoryPage;
95 }
96
97 template<typename Config>
98 void IsoHeapImpl<Config>::scavenge(Vector<DeferredDecommit>& decommits)
99 {
100     forEachDirectory(
101         [&] (auto& directory) {
102             directory.scavenge(decommits);
103         });
104 }
105
106 template<typename Config>
107 unsigned IsoHeapImpl<Config>::allocatorOffset()
108 {
109     return m_allocator.offset();
110 }
111
112 template<typename Config>
113 unsigned IsoHeapImpl<Config>::deallocatorOffset()
114 {
115     return PerProcess<IsoTLSDeallocatorEntry<Config>>::get()->offset();
116 }
117
118 template<typename Config>
119 unsigned IsoHeapImpl<Config>::numLiveObjects()
120 {
121     unsigned result = 0;
122     forEachLiveObject(
123         [&] (void*) {
124             result++;
125         });
126     return result;
127 }
128
129 template<typename Config>
130 unsigned IsoHeapImpl<Config>::numCommittedPages()
131 {
132     unsigned result = 0;
133     forEachCommittedPage(
134         [&] (IsoPage<Config>&) {
135             result++;
136         });
137     return result;
138 }
139
140 template<typename Config>
141 template<typename Func>
142 void IsoHeapImpl<Config>::forEachDirectory(const Func& func)
143 {
144     func(m_inlineDirectory);
145     for (IsoDirectoryPage<Config>* page = m_headDirectory; page; page = page->next)
146         func(page->payload);
147 }
148
149 template<typename Config>
150 template<typename Func>
151 void IsoHeapImpl<Config>::forEachCommittedPage(const Func& func)
152 {
153     forEachDirectory(
154         [&] (auto& directory) {
155             directory.forEachCommittedPage(func);
156         });
157 }
158
159 template<typename Config>
160 template<typename Func>
161 void IsoHeapImpl<Config>::forEachLiveObject(const Func& func)
162 {
163     forEachCommittedPage(
164         [&] (IsoPage<Config>& page) {
165             page.forEachLiveObject(func);
166         });
167 }
168
169 } // namespace bmalloc
170