2011-04-06 Chang Shu <cshu@webkit.org>
[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 String InjectedBundle::pageSizeAndMarginsInPixels(WebFrame* frame, int pageIndex, int width, int height, int marginTop, int marginRight, int marginBottom, int marginLeft)
180 {
181     Frame* coreFrame = frame ? frame->coreFrame() : 0;
182     if (!coreFrame)
183         return String();
184
185     return PrintContext::pageSizeAndMarginsInPixels(coreFrame, pageIndex, width, height, marginTop, marginRight, marginBottom, marginLeft);
186 }
187
188 static PassOwnPtr<Vector<String> > toStringVector(ImmutableArray* patterns)
189 {
190     if (!patterns)
191         return 0;
192
193     size_t size =  patterns->size();
194     if (!size)
195         return 0;
196
197     Vector<String>* patternsVector = new Vector<String>;
198     patternsVector->reserveInitialCapacity(size);
199     for (size_t i = 0; i < size; ++i) {
200         WebString* entry = patterns->at<WebString>(i);
201         if (entry)
202             patternsVector->uncheckedAppend(entry->string());
203     }
204     return patternsVector;
205 }
206
207 void InjectedBundle::addUserScript(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& source, const String& url, ImmutableArray* whitelist, ImmutableArray* blacklist, WebCore::UserScriptInjectionTime injectionTime, WebCore::UserContentInjectedFrames injectedFrames)
208 {
209     // 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.
210     PageGroup::pageGroup(pageGroup->identifier())->addUserScriptToWorld(scriptWorld->coreWorld(), source, KURL(KURL(), url), toStringVector(whitelist), toStringVector(blacklist), injectionTime, injectedFrames);
211 }
212
213 void InjectedBundle::addUserStyleSheet(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& source, const String& url, ImmutableArray* whitelist, ImmutableArray* blacklist, WebCore::UserContentInjectedFrames injectedFrames)
214 {
215     // 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.
216     PageGroup::pageGroup(pageGroup->identifier())->addUserStyleSheetToWorld(scriptWorld->coreWorld(), source, KURL(KURL(), url), toStringVector(whitelist), toStringVector(blacklist), injectedFrames);
217 }
218
219 void InjectedBundle::removeUserScript(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& url)
220 {
221     // 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.
222     PageGroup::pageGroup(pageGroup->identifier())->removeUserScriptFromWorld(scriptWorld->coreWorld(), KURL(KURL(), url));
223 }
224
225 void InjectedBundle::removeUserStyleSheet(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld, const String& url)
226 {
227     // 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.
228     PageGroup::pageGroup(pageGroup->identifier())->removeUserStyleSheetFromWorld(scriptWorld->coreWorld(), KURL(KURL(), url));
229 }
230
231 void InjectedBundle::removeUserScripts(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld)
232 {
233     PageGroup::pageGroup(pageGroup->identifier())->removeUserScriptsFromWorld(scriptWorld->coreWorld());
234 }
235
236 void InjectedBundle::removeUserStyleSheets(WebPageGroupProxy* pageGroup, InjectedBundleScriptWorld* scriptWorld)
237 {
238     PageGroup::pageGroup(pageGroup->identifier())->removeUserStyleSheetsFromWorld(scriptWorld->coreWorld());
239 }
240
241 void InjectedBundle::removeAllUserContent(WebPageGroupProxy* pageGroup)
242 {
243     PageGroup::pageGroup(pageGroup->identifier())->removeAllUserContent();
244 }
245
246 void InjectedBundle::garbageCollectJavaScriptObjects()
247 {
248     gcController().garbageCollectNow();
249 }
250
251 void InjectedBundle::garbageCollectJavaScriptObjectsOnAlternateThreadForDebugging(bool waitUntilDone)
252 {
253     gcController().garbageCollectOnAlternateThreadForDebugging(waitUntilDone);
254 }
255
256 size_t InjectedBundle::javaScriptObjectsCount()
257 {
258     JSLock lock(SilenceAssertionsOnly);
259     return JSDOMWindow::commonJSGlobalData()->heap.objectCount();
260 }
261
262 void InjectedBundle::reportException(JSContextRef context, JSValueRef exception)
263 {
264     if (!context || !exception)
265         return;
266
267     JSLock lock(JSC::SilenceAssertionsOnly);
268     JSC::ExecState* execState = toJS(context);
269
270     // Make sure the context has a DOMWindow global object, otherwise this context didn't originate from a Page.
271     if (!toJSDOMWindow(execState->lexicalGlobalObject()))
272         return;
273
274     WebCore::reportException(execState, toJS(execState, exception));
275 }
276
277 void InjectedBundle::didCreatePage(WebPage* page)
278 {
279     m_client.didCreatePage(this, page);
280 }
281
282 void InjectedBundle::willDestroyPage(WebPage* page)
283 {
284     m_client.willDestroyPage(this, page);
285 }
286
287 void InjectedBundle::didInitializePageGroup(WebPageGroupProxy* pageGroup)
288 {
289     m_client.didInitializePageGroup(this, pageGroup);
290 }
291
292 void InjectedBundle::didReceiveMessage(const String& messageName, APIObject* messageBody)
293 {
294     m_client.didReceiveMessage(this, messageName, messageBody);
295 }
296
297 void InjectedBundle::didReceiveMessage(CoreIPC::Connection* connection, CoreIPC::MessageID messageID, CoreIPC::ArgumentDecoder* arguments)
298 {
299     switch (messageID.get<InjectedBundleMessage::Kind>()) {
300         case InjectedBundleMessage::PostMessage: {
301             String messageName;            
302             RefPtr<APIObject> messageBody;
303             InjectedBundleUserMessageDecoder messageDecoder(messageBody);
304             if (!arguments->decode(CoreIPC::Out(messageName, messageDecoder)))
305                 return;
306
307             didReceiveMessage(messageName, messageBody.get());
308             return;
309         }
310     }
311
312     ASSERT_NOT_REACHED();
313 }
314
315 } // namespace WebKit