WebCore/workers/Worker.cpp

   1 /*
   2  * Copyright (C) 2008 Apple Inc. All Rights Reserved.
   3  * Copyright (C) 2009 Google Inc. All Rights Reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  */
  27
  28 #include "config.h"
  29
  30 #if ENABLE(WORKERS)
  31
  32 #include "Worker.h"
  33
  34 #include "DOMWindow.h"
  35 #include "DocLoader.h"
  36 #include "Document.h"
  37 #include "EventException.h"
  38 #include "EventListener.h"
  39 #include "EventNames.h"
  40 #include "ExceptionCode.h"
  41 #include "Frame.h"
  42 #include "FrameLoader.h"
  43 #include "MessageEvent.h"
  44 #include "SecurityOrigin.h"
  45 #include "TextEncoding.h"
  46 #include "WorkerContextProxy.h"
  47 #include "WorkerScriptLoader.h"
  48 #include "WorkerThread.h"
  49 #include <wtf/MainThread.h>
  50
  51 namespace WebCore {
  52
  53 Worker::Worker(const String& url, ScriptExecutionContext* context, ExceptionCode& ec)
  54     : AbstractWorker(context)
  55     , m_contextProxy(WorkerContextProxy::create(this))
  56 {
  57     if (url.isEmpty()) {
  58         ec = SYNTAX_ERR;
  59         return;
  60     }
  61
  62     KURL scriptURL = context->completeURL(url);
  63     if (!scriptURL.isValid()) {
  64         ec = SYNTAX_ERR;
  65         return;
  66     }
  67
  68     if (!context->securityOrigin()->canAccess(SecurityOrigin::create(scriptURL).get())) {
  69         ec = SECURITY_ERR;
  70         return;
  71     }
  72
  73     m_scriptLoader = new WorkerScriptLoader();
  74     m_scriptLoader->loadAsynchronously(scriptExecutionContext(), scriptURL, DenyCrossOriginRedirect, this);
  75     setPendingActivity(this);  // The worker context does not exist while loading, so we must ensure that the worker object is not collected, as well as its event listeners.
  76 }
  77
  78 Worker::~Worker()
  79 {
  80     ASSERT(isMainThread());
  81     ASSERT(scriptExecutionContext()); // The context is protected by worker context proxy, so it cannot be destroyed while a Worker exists.
  82     m_contextProxy->workerObjectDestroyed();
  83 }
  84
  85 void Worker::postMessage(const String& message, ExceptionCode& ec)
  86 {
  87     postMessage(message, 0, ec);
  88 }
  89
  90 void Worker::postMessage(const String& message, MessagePort* messagePort, ExceptionCode& ec)
  91 {
  92     // Disentangle the port in preparation for sending it to the remote context.
  93     OwnPtr<MessagePortChannel> channel = messagePort ? messagePort->disentangle(ec) : 0;
  94     if (ec)
  95         return;
  96     m_contextProxy->postMessageToWorkerContext(message, channel.release());
  97 }
  98
  99 void Worker::terminate()
 100 {
 101     m_contextProxy->terminateWorkerContext();
 102 }
 103
 104 bool Worker::canSuspend() const
 105 {
 106     // FIXME: It is not currently possible to suspend a worker, so pages with workers can not go into page cache.
 107     return false;
 108 }
 109
 110 void Worker::stop()
 111 {
 112     terminate();
 113 }
 114
 115 bool Worker::hasPendingActivity() const
 116 {
 117     return m_contextProxy->hasPendingActivity() || ActiveDOMObject::hasPendingActivity();
 118 }
 119
 120 void Worker::notifyFinished()
 121 {
 122     if (m_scriptLoader->failed())
 123         dispatchLoadErrorEvent();
 124     else
 125         m_contextProxy->startWorkerContext(m_scriptLoader->url(), scriptExecutionContext()->userAgent(m_scriptLoader->url()), m_scriptLoader->script());
 126
 127     m_scriptLoader = 0;
 128
 129     unsetPendingActivity(this);
 130 }
 131
 132 void Worker::dispatchMessage(const String& message, PassRefPtr<MessagePort> port)
 133 {
 134     RefPtr<Event> evt = MessageEvent::create(message, "", "", 0, port);
 135
 136     if (m_onMessageListener.get()) {
 137         evt->setTarget(this);
 138         evt->setCurrentTarget(this);
 139         m_onMessageListener->handleEvent(evt.get(), false);
 140     }
 141
 142     ExceptionCode ec = 0;
 143     dispatchEvent(evt.release(), ec);
 144     ASSERT(!ec);
 145 }
 146
 147 } // namespace WebCore
 148
 149 #endif // ENABLE(WORKERS)