975bb49192ff1ba3bb0367cf4eed1e032641a385
[WebKit-https.git] / Source / WebCore / bindings / v8 / DOMWrapperWorld.cpp
1 /*
2  * Copyright (C) 2009 Google 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 are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "DOMWrapperWorld.h"
33
34 #include "DOMDataStore.h"
35 #include "V8Binding.h"
36 #include "V8DOMActivityLogger.h"
37 #include "V8DOMWindow.h"
38 #include "V8DOMWrapper.h"
39 #include <wtf/HashTraits.h>
40 #include <wtf/MainThread.h>
41 #include <wtf/StdLibExtras.h>
42
43 namespace WebCore {
44
45 int DOMWrapperWorld::isolatedWorldCount = 0;
46
47 PassRefPtr<DOMWrapperWorld>  DOMWrapperWorld::createUninitializedWorld()
48 {
49     return adoptRef(new DOMWrapperWorld(uninitializedWorldId, uninitializedExtensionGroup));
50 }
51
52 PassRefPtr<DOMWrapperWorld> DOMWrapperWorld::createMainWorld()
53 {
54     return adoptRef(new DOMWrapperWorld(mainWorldId, mainWorldExtensionGroup));
55 }
56
57 DOMWrapperWorld::DOMWrapperWorld(int worldId, int extensionGroup)
58     : m_worldId(worldId)
59     , m_extensionGroup(extensionGroup)
60 {
61     if (isIsolatedWorld())
62         m_domDataStore = adoptPtr(new DOMDataStore(DOMDataStore::IsolatedWorld));
63 }
64
65 DOMWrapperWorld* mainThreadNormalWorld()
66 {
67     ASSERT(isMainThread());
68     DEFINE_STATIC_LOCAL(RefPtr<DOMWrapperWorld>, cachedNormalWorld, (DOMWrapperWorld::createMainWorld()));
69     return cachedNormalWorld.get();
70 }
71
72 void DOMWrapperWorld::assertContextHasCorrectPrototype(v8::Handle<v8::Context> context)
73 {
74     ASSERT(isMainThread());
75     ASSERT(V8DOMWrapper::isWrapperOfType(toInnerGlobalObject(context), &V8DOMWindow::info));
76 }
77
78 static void isolatedWorldWeakCallback(v8::Isolate* isolate, v8::Persistent<v8::Value> object, void* parameter)
79 {
80     object.Dispose(isolate);
81     object.Clear();
82     static_cast<DOMWrapperWorld*>(parameter)->deref();
83 }
84
85 void DOMWrapperWorld::makeContextWeak(v8::Handle<v8::Context> context)
86 {
87     ASSERT(isIsolatedWorld());
88     ASSERT(isolatedWorld(context) == this);
89     v8::Isolate* isolate = context->GetIsolate();
90     v8::Persistent<v8::Context>::New(isolate, context).MakeWeak(isolate, this, isolatedWorldWeakCallback);
91     // Matching deref is in weak callback.
92     this->ref();
93 }
94
95 void DOMWrapperWorld::setIsolatedWorldField(v8::Handle<v8::Context> context)
96 {
97     context->SetAlignedPointerInEmbedderData(v8ContextIsolatedWorld, isMainWorld() ? 0 : this);
98 }
99
100 typedef HashMap<int, DOMWrapperWorld*> WorldMap;
101 static WorldMap& isolatedWorldMap()
102 {
103     ASSERT(isMainThread());
104     DEFINE_STATIC_LOCAL(WorldMap, map, ());
105     return map;
106 }
107
108 void DOMWrapperWorld::getAllWorlds(Vector<RefPtr<DOMWrapperWorld> >& worlds)
109 {
110     worlds.append(mainThreadNormalWorld());
111     WorldMap& isolatedWorlds = isolatedWorldMap();
112     for (WorldMap::iterator it = isolatedWorlds.begin(); it != isolatedWorlds.end(); ++it)
113         worlds.append(it->value);
114 }
115
116 DOMWrapperWorld::~DOMWrapperWorld()
117 {
118     ASSERT(!isMainWorld());
119
120     if (!isIsolatedWorld())
121         return;
122
123     WorldMap& map = isolatedWorldMap();
124     WorldMap::iterator i = map.find(m_worldId);
125     if (i == map.end()) {
126         ASSERT_NOT_REACHED();
127         return;
128     }
129     ASSERT(i->value == this);
130
131     map.remove(i);
132     isolatedWorldCount--;
133     ASSERT(map.size() == isolatedWorldCount);
134 }
135
136 static int temporaryWorldId = DOMWrapperWorld::uninitializedWorldId-1;
137
138 PassRefPtr<DOMWrapperWorld> DOMWrapperWorld::ensureIsolatedWorld(int worldId, int extensionGroup)
139 {
140     ASSERT(worldId != mainWorldId);
141     ASSERT(worldId >= uninitializedWorldId);
142
143     WorldMap& map = isolatedWorldMap();
144     if (worldId == uninitializedWorldId)
145         worldId = temporaryWorldId--;
146     else {
147         WorldMap::iterator i = map.find(worldId);
148         if (i != map.end()) {
149             ASSERT(i->value->worldId() == worldId);
150             ASSERT(i->value->extensionGroup() == extensionGroup);
151             return i->value;
152         }
153     }
154
155     RefPtr<DOMWrapperWorld> world = adoptRef(new DOMWrapperWorld(worldId, extensionGroup));
156     map.add(worldId, world.get());
157     isolatedWorldCount++;
158     ASSERT(map.size() == isolatedWorldCount);
159
160     return world.release();
161 }
162
163 typedef HashMap<int, RefPtr<SecurityOrigin> > IsolatedWorldSecurityOriginMap;
164 static IsolatedWorldSecurityOriginMap& isolatedWorldSecurityOrigins()
165 {
166     ASSERT(isMainThread());
167     DEFINE_STATIC_LOCAL(IsolatedWorldSecurityOriginMap, map, ());
168     return map;
169 }
170
171 SecurityOrigin* DOMWrapperWorld::isolatedWorldSecurityOrigin()
172 {
173     ASSERT(this->isIsolatedWorld());
174     IsolatedWorldSecurityOriginMap& origins = isolatedWorldSecurityOrigins();
175     IsolatedWorldSecurityOriginMap::iterator it = origins.find(worldId());
176     return it == origins.end() ? 0 : it->value.get();
177 }
178
179 void DOMWrapperWorld::setIsolatedWorldSecurityOrigin(int worldID, PassRefPtr<SecurityOrigin> securityOrigin)
180 {
181     ASSERT(DOMWrapperWorld::isIsolatedWorldId(worldID));
182     if (securityOrigin)
183         isolatedWorldSecurityOrigins().set(worldID, securityOrigin);
184     else
185         isolatedWorldSecurityOrigins().remove(worldID);
186 }
187
188 void DOMWrapperWorld::clearIsolatedWorldSecurityOrigin(int worldID)
189 {
190     ASSERT(DOMWrapperWorld::isIsolatedWorldId(worldID));
191     isolatedWorldSecurityOrigins().remove(worldID);
192 }
193
194 typedef HashMap<int, bool> IsolatedWorldContentSecurityPolicyMap;
195 static IsolatedWorldContentSecurityPolicyMap& isolatedWorldContentSecurityPolicies()
196 {
197     ASSERT(isMainThread());
198     DEFINE_STATIC_LOCAL(IsolatedWorldContentSecurityPolicyMap, map, ());
199     return map;
200 }
201
202 bool DOMWrapperWorld::isolatedWorldHasContentSecurityPolicy()
203 {
204     ASSERT(this->isIsolatedWorld());
205     IsolatedWorldContentSecurityPolicyMap& policies = isolatedWorldContentSecurityPolicies();
206     IsolatedWorldContentSecurityPolicyMap::iterator it = policies.find(worldId());
207     return it == policies.end() ? false : it->value;
208 }
209
210 void DOMWrapperWorld::setIsolatedWorldContentSecurityPolicy(int worldID, const String& policy)
211 {
212     ASSERT(DOMWrapperWorld::isIsolatedWorldId(worldID));
213     if (!policy.isEmpty())
214         isolatedWorldContentSecurityPolicies().set(worldID, true);
215     else
216         isolatedWorldContentSecurityPolicies().remove(worldID);
217 }
218
219 void DOMWrapperWorld::clearIsolatedWorldContentSecurityPolicy(int worldID)
220 {
221     ASSERT(DOMWrapperWorld::isIsolatedWorldId(worldID));
222     isolatedWorldContentSecurityPolicies().remove(worldID);
223 }
224
225 typedef HashMap<int, OwnPtr<V8DOMActivityLogger>, WTF::IntHash<int>, WTF::UnsignedWithZeroKeyHashTraits<int> > DOMActivityLoggerMap; 
226 static DOMActivityLoggerMap& domActivityLoggers()
227 {
228     ASSERT(isMainThread());
229     DEFINE_STATIC_LOCAL(DOMActivityLoggerMap, map, ());
230     return map;
231 }
232
233 void DOMWrapperWorld::setActivityLogger(int worldId, PassOwnPtr<V8DOMActivityLogger> logger)
234 {
235     domActivityLoggers().set(worldId, logger);
236
237
238 V8DOMActivityLogger* DOMWrapperWorld::activityLogger(int worldId)
239 {
240     DOMActivityLoggerMap& loggers = domActivityLoggers();
241     DOMActivityLoggerMap::iterator it = loggers.find(worldId);   
242     return it == loggers.end() ? 0 : it->value.get();
243
244
245 } // namespace WebCore