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 "web/SharedWorkerRepositoryClientImpl.h"
33
34#include "bindings/core/v8/ExceptionMessages.h"
35#include "bindings/core/v8/ExceptionState.h"
36#include "core/dom/ExceptionCode.h"
37#include "core/dom/ExecutionContext.h"
38#include "core/events/Event.h"
39#include "core/frame/csp/ContentSecurityPolicy.h"
40#include "core/inspector/InspectorInstrumentation.h"
41#include "core/workers/SharedWorker.h"
42#include "core/workers/WorkerScriptLoader.h"
43#include "core/workers/WorkerScriptLoaderClient.h"
44#include "platform/network/ResourceResponse.h"
45#include "public/platform/WebMessagePortChannel.h"
46#include "public/platform/WebString.h"
47#include "public/platform/WebURL.h"
48#include "public/web/WebContentSecurityPolicy.h"
49#include "public/web/WebFrameClient.h"
50#include "public/web/WebKit.h"
51#include "public/web/WebSharedWorker.h"
52#include "public/web/WebSharedWorkerRepositoryClient.h"
53#include "web/WebLocalFrameImpl.h"
54
55namespace blink {
56
57// Callback class that keeps the SharedWorker and WebSharedWorker objects alive while connecting.
58class SharedWorkerConnector : private WebSharedWorkerConnector::ConnectListener {
59public:
60    SharedWorkerConnector(PassRefPtrWillBeRawPtr<SharedWorker> worker, const KURL& url, const String& name, PassOwnPtr<WebMessagePortChannel> channel, PassOwnPtr<WebSharedWorkerConnector> webWorkerConnector)
61        : m_worker(worker)
62        , m_url(url)
63        , m_name(name)
64        , m_webWorkerConnector(webWorkerConnector)
65        , m_channel(channel) { }
66
67    virtual ~SharedWorkerConnector();
68    void connect();
69
70private:
71    // WebSharedWorkerConnector::ConnectListener overrides.
72    virtual void connected() OVERRIDE;
73    virtual void scriptLoadFailed() OVERRIDE;
74
75    RefPtrWillBePersistent<SharedWorker> m_worker;
76    KURL m_url;
77    String m_name;
78    OwnPtr<WebSharedWorkerConnector> m_webWorkerConnector;
79    OwnPtr<WebMessagePortChannel> m_channel;
80};
81
82SharedWorkerConnector::~SharedWorkerConnector()
83{
84    m_worker->setIsBeingConnected(false);
85}
86
87void SharedWorkerConnector::connect()
88{
89    m_worker->setIsBeingConnected(true);
90    m_webWorkerConnector->connect(m_channel.leakPtr(), this);
91}
92
93void SharedWorkerConnector::connected()
94{
95    // Free ourselves (this releases the SharedWorker so it can be freed as well if unreferenced).
96    delete this;
97}
98
99void SharedWorkerConnector::scriptLoadFailed()
100{
101    m_worker->dispatchEvent(Event::createCancelable(EventTypeNames::error));
102    // Free ourselves (this releases the SharedWorker so it can be freed as well if unreferenced).
103    delete this;
104}
105
106static WebSharedWorkerRepositoryClient::DocumentID getId(void* document)
107{
108    ASSERT(document);
109    return reinterpret_cast<WebSharedWorkerRepositoryClient::DocumentID>(document);
110}
111
112void SharedWorkerRepositoryClientImpl::connect(PassRefPtrWillBeRawPtr<SharedWorker> worker, PassOwnPtr<WebMessagePortChannel> port, const KURL& url, const String& name, ExceptionState& exceptionState)
113{
114    ASSERT(m_client);
115
116    // No nested workers (for now) - connect() should only be called from document context.
117    ASSERT(worker->executionContext()->isDocument());
118    Document* document = toDocument(worker->executionContext());
119    OwnPtr<WebSharedWorkerConnector> webWorkerConnector = adoptPtr(m_client->createSharedWorkerConnector(url, name, getId(document), worker->executionContext()->contentSecurityPolicy()->deprecatedHeader(), static_cast<WebContentSecurityPolicyType>(worker->executionContext()->contentSecurityPolicy()->deprecatedHeaderType())));
120    if (!webWorkerConnector) {
121        // Existing worker does not match this url, so return an error back to the caller.
122        exceptionState.throwDOMException(URLMismatchError, "The location of the SharedWorker named '" + name + "' does not exactly match the provided URL ('" + url.elidedString() + "').");
123        return;
124    }
125
126    // The connector object manages its own lifecycle (and the lifecycles of the two worker objects).
127    // It will free itself once connecting is completed.
128    SharedWorkerConnector* connector = new SharedWorkerConnector(worker, url, name, port, webWorkerConnector.release());
129    connector->connect();
130}
131
132void SharedWorkerRepositoryClientImpl::documentDetached(Document* document)
133{
134    ASSERT(m_client);
135    m_client->documentDetached(getId(document));
136}
137
138SharedWorkerRepositoryClientImpl::SharedWorkerRepositoryClientImpl(WebSharedWorkerRepositoryClient* client)
139    : m_client(client)
140{
141}
142
143} // namespace blink
144