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