51579656c5219668ca5891138153e3542c46e1f8
[WebKit-https.git] / Source / WebKit2 / WebProcess / InjectedBundle / InjectedBundle.cpp
1 /*
2  * Copyright (C) 2010 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 "InjectedBundle.h"
28
29 #include "Arguments.h"
30 #include "ImmutableArray.h"
31 #include "InjectedBundleMessageKinds.h"
32 #include "InjectedBundleScriptWorld.h"
33 #include "InjectedBundleUserMessageCoders.h"
34 #include "WKAPICast.h"
35 #include "WKBundleAPICast.h"
36 #include "WebContextMessageKinds.h"
37 #include "WebCoreArgumentCoders.h"
38 #include "WebDatabaseManager.h"
39 #include "WebFrame.h"
40 #include "WebPage.h"
41 #include "WebPreferencesStore.h"
42 #include "WebProcess.h"
43 #include <JavaScriptCore/APICast.h>
44 #include <JavaScriptCore/JSLock.h>
45 #include <WebCore/Frame.h>
46 #include <WebCore/FrameView.h>
47 #include <WebCore/GCController.h>
48 #include <WebCore/JSDOMWindow.h>
49 #include <WebCore/Page.h>
50 #include <WebCore/PageGroup.h>
51 #include <WebCore/PrintContext.h>
52 #include <WebCore/Settings.h>
53 #include <wtf/OwnArrayPtr.h>
54 #include <wtf/PassOwnArrayPtr.h>
55
56 using namespace WebCore;
57 using namespace JSC;
58
59 namespace WebKit {
60
61 InjectedBundle::InjectedBundle(const String& path)
62     : m_path(path)
63     , m_platformBundle(0)
64 {
65     initializeClient(0);
66 }
67
68 InjectedBundle::~InjectedBundle()
69 {
70 }
71
72 void InjectedBundle::initializeClient(WKBundleClient* client)
73 {
74     m_client.initialize(client);
75 }
76
77 void InjectedBundle::postMessage(const String& messageName, APIObject* messageBody)
78 {
79     WebProcess::shared().connection()->deprecatedSend(WebContextLegacyMessage::PostMessage, 0, CoreIPC::In(messageName, InjectedBundleUserMessageEncoder(messageBody)));
80 }
81
82 void InjectedBundle::postSynchronousMessage(const String& messageName, APIObject* messageBody, RefPtr<APIObject>& returnData)
83 {
84     RefPtr<APIObject> returnDataTmp;
85     InjectedBundleUserMessageDecoder messageDecoder(returnDataTmp);
86     
87     bool succeeded = WebProcess::shared().connection()->deprecatedSendSync(WebContextLegacyMessage::PostSynchronousMessage, 0, CoreIPC::In(messageName, InjectedBundleUserMessageEncoder(messageBody)), CoreIPC::Out(messageDecoder));
88
89     if (!succeeded)
90         return;
91
92     returnData = returnDataTmp;
93 }
94
95 void InjectedBundle::setShouldTrackVisitedLinks(bool shouldTrackVisitedLinks)
96 {
97     PageGroup::setShouldTrackVisitedLinks(shouldTrackVisitedLinks);
98 }
99
100 void InjectedBundle::removeAllVisitedLinks()
101 {
102     PageGroup::removeAllVisitedLinks();
103 }
104
105 void InjectedBundle::overrideXSSAuditorEnabledForTestRunner(WebPageGroupProxy* pageGroup, bool enabled)
106 {
107     // Override the preference for all future pages.
108     WebPreferencesStore::overrideXSSAuditorEnabledForTestRunner(enabled);
109
110     // Change the setting for existing ones.
111     const HashSet<Page*>& pages = PageGroup::pageGroup(pageGroup->identifier())->pages();
112     for (HashSet<Page*>::iterator iter = pages.begin(); iter != pages.end(); ++iter)
113         (*iter)->settings()->setXSSAuditorEnabled(enabled);
114 }
115
116 void InjectedBundle::overrideAllowUniversalAccessFromFileURLsForTestRunner(WebPageGroupProxy* pageGroup, bool enabled)
117 {
118     // Override the preference for all future pages.
119     WebPreferencesStore::overrideAllowUniversalAccessFromFileURLsForTestRunner(enabled);
120
121     // Change the setting for existing ones.
122     const HashSet<Page*>& pages = PageGroup::pageGroup(pageGroup->identifier())->pages();
123     for (HashSet<Page*>::iterator iter = pages.begin(); iter != pages.end(); ++iter)
124         (*iter)->settings()->setAllowUniversalAccessFromFileURLs(enabled);
125 }
126
127 void InjectedBundle::setAllowFileAccessFromFileURLs(WebPageGroupProxy* pageGroup, bool enabled)
128 {
129     // Override the preference for all future pages.
130     WebPreferencesStore::overrideAllowFileAccessFromFileURLsForTestRunner(enabled);
131
132     // Change the setting for existing ones.
133     const HashSet<Page*>& pages = PageGroup::pageGroup(pageGroup->identifier())->pages();
134     for (HashSet<Page*>::iterator iter = pages.begin(); iter != pages.end(); ++iter)
135         (*iter)->settings()->setAllowFileAccessFromFileURLs(enabled);
136 }
137
138 void InjectedBundle::clearAllDatabases()
139 {
140     WebDatabaseManager::shared().deleteAllDatabases();
141 }
142
143 void InjectedBundle::setDatabaseQuota(uint64_t quota)
144 {
145     WebDatabaseManager::shared().setQuotaForOrigin("file:///", quota);
146 }
147
148 int InjectedBundle::numberOfPages(WebFrame* frame, double pageWidthInPixels, double pageHeightInPixels)
149 {
150     Frame* coreFrame = frame ? frame->coreFrame() : 0;
151     if (!coreFrame)
152         return -1;
153     if (!pageWidthInPixels)
154         pageWidthInPixels = coreFrame->view()->width();
155     if (!pageHeightInPixels)
156         pageHeightInPixels = coreFrame->view()->height();
157
158     return PrintContext::numberOfPages(coreFrame, FloatSize(pageWidthInPixels, pageHeightInPixels));
159 }
160
161 int InjectedBundle::pageNumberForElementById(WebFrame* frame, const String& id, double pageWidthInPixels, double pageHeightInPixels)
162 {
163     Frame* coreFrame = frame ? frame->coreFrame() : 0;
164     if (!coreFrame)
165         return -1;
166
167     Element* element = coreFrame->document()->getElementById(AtomicString(id));
168     if (!element)
169         return -1;
170
171     if (!pageWidthInPixels)
172         pageWidthInPixels = coreFrame->view()->width();
173     if (!pageHeightInPixels)
174         pageHeightInPixels = coreFrame->view()->height();
175
176     return PrintContext::pageNumberForElement(element, FloatSize(pageWidthInPixels, pageHeightInPixels));
177 }
178
179 static PassOwnPtr<Vector<String> > toStringVector(ImmutableArray* patterns)
180 {
181     if (!patterns)
182         return 0;
183
184     size_t size =  patterns->size();
185     if (!size)
186         return 0;
187
188     Vector<String>* patternsVector = new Vector<String>;
189     patternsVector->reserveInitialCapacity(size);
190     for (size_t i = 0; i < size; ++i) {
191         WebString* entry = patterns->at<WebString>(i);
192         if (entry)
193             patternsVector->uncheckedAppend(entry->string());
194     }
195     return patternsVector;
196 }
197
198 void InjectedBundle::addUserScript(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& source, const String& url, ImmutableArray* whitelist, ImmutableArray* blacklist, WebCore::UserScriptInjectionTime injectionTime, WebCore::UserContentInjectedFrames injectedFrames)
199 {
200     // url is not from KURL::string(), i.e. it has not already been parsed by KURL, so we have to use the relative URL constructor for KURL instead of the ParsedURLStringTag version.
201     PageGroup::pageGroup(pageGroup->identifier())->addUserScriptToWorld(scriptWorld->coreWorld(), source, KURL(KURL(), url), toStringVector(whitelist), toStringVector(blacklist), injectionTime, injectedFrames);
202 }
203
204 void InjectedBundle::addUserStyleSheet(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& source, const String& url, ImmutableArray* whitelist, ImmutableArray* blacklist, WebCore::UserContentInjectedFrames injectedFrames)
205 {
206     // url is not from KURL::string(), i.e. it has not already been parsed by KURL, so we have to use the relative URL constructor for KURL instead of the ParsedURLStringTag version.
207     PageGroup::pageGroup(pageGroup->identifier())->addUserStyleSheetToWorld(scriptWorld->coreWorld(), source, KURL(KURL(), url), toStringVector(whitelist), toStringVector(blacklist), injectedFrames);
208 }
209
210 void InjectedBundle::removeUserScript(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& url)
211 {
212     // url is not from KURL::string(), i.e. it has not already been parsed by KURL, so we have to use the relative URL constructor for KURL instead of the ParsedURLStringTag version.
213     PageGroup::pageGroup(pageGroup->identifier())->removeUserScriptFromWorld(scriptWorld->coreWorld(), KURL(KURL(), url));
214 }
215
216 void InjectedBundle::removeUserStyleSheet(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& url)
217 {
218     // url is not from KURL::string(), i.e. it has not already been parsed by KURL, so we have to use the relative URL constructor for KURL instead of the ParsedURLStringTag version.
219     PageGroup::pageGroup(pageGroup->identifier())->removeUserStyleSheetFromWorld(scriptWorld->coreWorld(), KURL(KURL(), url));
220 }
221
222 void InjectedBundle::removeUserScripts(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld)
223 {
224     PageGroup::pageGroup(pageGroup->identifier())->removeUserScriptsFromWorld(scriptWorld->coreWorld());
225 }
226
227 void InjectedBundle::removeUserStyleSheets(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld)
228 {
229     PageGroup::pageGroup(pageGroup->identifier())->removeUserStyleSheetsFromWorld(scriptWorld->coreWorld());
230 }
231
232 void InjectedBundle::removeAllUserContent(WebPageGroupProxy* pageGroup)
233 {
234     PageGroup::pageGroup(pageGroup->identifier())->removeAllUserContent();
235 }
236
237 void InjectedBundle::garbageCollectJavaScriptObjects()
238 {
239     gcController().garbageCollectNow();
240 }
241
242 void InjectedBundle::garbageCollectJavaScriptObjectsOnAlternateThreadForDebugging(bool waitUntilDone)
243 {
244     gcController().garbageCollectOnAlternateThreadForDebugging(waitUntilDone);
245 }
246
247 size_t InjectedBundle::javaScriptObjectsCount()
248 {
249     JSLock lock(SilenceAssertionsOnly);
250     return JSDOMWindow::commonJSGlobalData()->heap.objectCount();
251 }
252
253 void InjectedBundle::reportException(JSContextRef context, JSValueRef exception)
254 {
255     if (!context || !exception)
256         return;
257
258     JSLock lock(JSC::SilenceAssertionsOnly);
259     JSC::ExecState* execState = toJS(context);
260
261     // Make sure the context has a DOMWindow global object, otherwise this context didn't originate from a Page.
262     if (!toJSDOMWindow(execState->lexicalGlobalObject()))
263         return;
264
265     WebCore::reportException(execState, toJS(execState, exception));
266 }
267
268 void InjectedBundle::didCreatePage(WebPage* page)
269 {
270     m_client.didCreatePage(this, page);
271 }
272
273 void InjectedBundle::willDestroyPage(WebPage* page)
274 {
275     m_client.willDestroyPage(this, page);
276 }
277
278 void InjectedBundle::didInitializePageGroup(WebPageGroupProxy* pageGroup)
279 {
280     m_client.didInitializePageGroup(this, pageGroup);
281 }
282
283 void InjectedBundle::didReceiveMessage(const String& messageName, APIObject* messageBody)
284 {
285     m_client.didReceiveMessage(this, messageName, messageBody);
286 }
287
288 void InjectedBundle::didReceiveMessage(CoreIPC::Connection* connection, CoreIPC::MessageID messageID, CoreIPC::ArgumentDecoder* arguments)
289 {
290     switch (messageID.get<InjectedBundleMessage::Kind>()) {
291         case InjectedBundleMessage::PostMessage: {
292             String messageName;            
293             RefPtr<APIObject> messageBody;
294             InjectedBundleUserMessageDecoder messageDecoder(messageBody);
295             if (!arguments->decode(CoreIPC::Out(messageName, messageDecoder)))
296                 return;
297
298             didReceiveMessage(messageName, messageBody.get());
299             return;
300         }
301     }
302
303     ASSERT_NOT_REACHED();
304 }
305
306 } // namespace WebKit