eba831e95a72547c91b625c4c6e09199e6db3a27
[WebKit-https.git] / Source / JavaScriptCore / runtime / WeakMapData.cpp
1 /*
2  * Copyright (C) 2013, 2015 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. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "WeakMapData.h"
28
29 #include "ExceptionHelpers.h"
30 #include "JSCInlines.h"
31
32 #include <wtf/MathExtras.h>
33
34 namespace JSC {
35
36 const ClassInfo WeakMapData::s_info = { "WeakMapData", 0, 0, CREATE_METHOD_TABLE(WeakMapData) };
37
38 WeakMapData::WeakMapData(VM& vm)
39     : Base(vm, vm.weakMapDataStructure.get())
40     , m_deadKeyCleaner(this)
41 {
42 }
43
44 void WeakMapData::finishCreation(VM& vm)
45 {
46     Base::finishCreation(vm);
47 }
48
49 void WeakMapData::destroy(JSCell* cell)
50 {
51     static_cast<WeakMapData*>(cell)->~WeakMapData();
52 }
53
54 size_t WeakMapData::estimatedSize(JSCell* cell)
55 {
56     WeakMapData* thisObj = jsCast<WeakMapData*>(cell);
57     return Base::estimatedSize(cell) + (thisObj->m_map.capacity() * (sizeof(JSObject*) + sizeof(WriteBarrier<Unknown>)));
58 }
59
60 void WeakMapData::visitChildren(JSCell* cell, SlotVisitor& visitor)
61 {
62     Base::visitChildren(cell, visitor);
63     WeakMapData* thisObj = jsCast<WeakMapData*>(cell);
64     visitor.addUnconditionalFinalizer(&thisObj->m_deadKeyCleaner);
65     visitor.addWeakReferenceHarvester(&thisObj->m_deadKeyCleaner);
66
67     // Rough approximation of the external storage needed for the hashtable.
68     // This isn't exact, but it is close enough, and proportional to the actual
69     // external memory usage.
70     visitor.reportExtraMemoryVisited(thisObj->m_map.capacity() * (sizeof(JSObject*) + sizeof(WriteBarrier<Unknown>)));
71 }
72
73 void WeakMapData::set(VM& vm, JSObject* key, JSValue value)
74 {
75     // Here we force the write barrier on the key.
76     auto result = m_map.add(WriteBarrier<JSObject>(vm, this, key).get(), WriteBarrier<Unknown>());
77     result.iterator->value.set(vm, this, value);
78 }
79
80 JSValue WeakMapData::get(JSObject* key)
81 {
82     auto iter = m_map.find(key);
83     if (iter == m_map.end())
84         return jsUndefined();
85     return iter->value.get();
86 }
87
88 bool WeakMapData::remove(JSObject* key)
89 {
90     auto iter = m_map.find(key);
91     if (iter == m_map.end())
92         return false;
93
94     m_map.remove(iter);
95     return true;
96 }
97
98 bool WeakMapData::contains(JSObject* key)
99 {
100     return m_map.contains(key);
101 }
102
103 void WeakMapData::clear()
104 {
105     m_map.clear();
106 }
107
108 void WeakMapData::DeadKeyCleaner::visitWeakReferences(SlotVisitor& visitor)
109 {
110     m_liveKeyCount = 0;
111     for (auto it = m_target->m_map.begin(), end = m_target->m_map.end(); it != end; ++it) {
112         if (!Heap::isMarked(it->key))
113             continue;
114         m_liveKeyCount++;
115         visitor.append(it->value);
116     }
117     RELEASE_ASSERT(m_liveKeyCount <= m_target->m_map.size());
118 }
119
120 void WeakMapData::DeadKeyCleaner::finalizeUnconditionally()
121 {
122     if (m_liveKeyCount > m_target->m_map.size() / 2) {
123         RELEASE_ASSERT(m_liveKeyCount <= m_target->m_map.size());
124         int deadCount = m_target->m_map.size() - m_liveKeyCount;
125         if (!deadCount)
126             return;
127         Vector<JSObject*> deadEntries;
128         deadEntries.reserveCapacity(deadCount);
129         for (auto it = m_target->m_map.begin(), end = m_target->m_map.end(); it != end; ++it) {
130             if (Heap::isMarked(it->key))
131                 continue;
132             deadEntries.uncheckedAppend(it->key);
133         }
134         for (size_t i = 0; i < deadEntries.size(); i++)
135             m_target->m_map.remove(deadEntries[i]);
136     } else {
137         MapType newMap;
138         for (auto it = m_target->m_map.begin(), end = m_target->m_map.end(); it != end; ++it) {
139             if (!Heap::isMarked(it->key))
140                 continue;
141             newMap.add(it->key, it->value);
142         }
143         m_target->m_map.swap(newMap);
144     }
145 }
146
147 }