| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "content/worker/websharedworker_stub.h" | |
| 6 | |
| 7 #include "base/command_line.h" | |
| 8 #include "base/compiler_specific.h" | |
| 9 #include "content/child/child_process.h" | |
| 10 #include "content/child/child_thread.h" | |
| 11 #include "content/child/fileapi/file_system_dispatcher.h" | |
| 12 #include "content/child/shared_worker_devtools_agent.h" | |
| 13 #include "content/child/webmessageportchannel_impl.h" | |
| 14 #include "content/common/worker_messages.h" | |
| 15 #include "content/public/common/content_switches.h" | |
| 16 #include "content/worker/worker_thread.h" | |
| 17 #include "third_party/WebKit/public/platform/WebString.h" | |
| 18 #include "third_party/WebKit/public/platform/WebURL.h" | |
| 19 #include "third_party/WebKit/public/web/WebSharedWorker.h" | |
| 20 | |
| 21 namespace content { | |
| 22 | |
| 23 WebSharedWorkerStub::WebSharedWorkerStub( | |
| 24 const GURL& url, | |
| 25 const base::string16& name, | |
| 26 const base::string16& content_security_policy, | |
| 27 blink::WebContentSecurityPolicyType security_policy_type, | |
| 28 bool pause_on_start, | |
| 29 int route_id) | |
| 30 : route_id_(route_id), | |
| 31 client_(route_id, this), | |
| 32 running_(false), | |
| 33 url_(url) { | |
| 34 | |
| 35 WorkerThread* worker_thread = WorkerThread::current(); | |
| 36 DCHECK(worker_thread); | |
| 37 worker_thread->AddWorkerStub(this); | |
| 38 // Start processing incoming IPCs for this worker. | |
| 39 worker_thread->GetRouter()->AddRoute(route_id_, this); | |
| 40 | |
| 41 // TODO(atwilson): Add support for NaCl when they support MessagePorts. | |
| 42 impl_ = blink::WebSharedWorker::create(client()); | |
| 43 if (pause_on_start) { | |
| 44 // Pause worker context when it starts and wait until either DevTools client | |
| 45 // is attached or explicit resume notification is received. | |
| 46 impl_->pauseWorkerContextOnStart(); | |
| 47 } | |
| 48 | |
| 49 worker_devtools_agent_.reset(new SharedWorkerDevToolsAgent(route_id, impl_)); | |
| 50 client()->set_devtools_agent(worker_devtools_agent_.get()); | |
| 51 impl_->startWorkerContext(url_, name, | |
| 52 content_security_policy, security_policy_type); | |
| 53 } | |
| 54 | |
| 55 WebSharedWorkerStub::~WebSharedWorkerStub() { | |
| 56 impl_->clientDestroyed(); | |
| 57 WorkerThread* worker_thread = WorkerThread::current(); | |
| 58 DCHECK(worker_thread); | |
| 59 worker_thread->RemoveWorkerStub(this); | |
| 60 worker_thread->GetRouter()->RemoveRoute(route_id_); | |
| 61 } | |
| 62 | |
| 63 void WebSharedWorkerStub::Shutdown() { | |
| 64 // The worker has exited - free ourselves and the client. | |
| 65 delete this; | |
| 66 } | |
| 67 | |
| 68 void WebSharedWorkerStub::EnsureWorkerContextTerminates() { | |
| 69 client_.EnsureWorkerContextTerminates(); | |
| 70 } | |
| 71 | |
| 72 bool WebSharedWorkerStub::OnMessageReceived(const IPC::Message& message) { | |
| 73 if (worker_devtools_agent_->OnMessageReceived(message)) | |
| 74 return true; | |
| 75 | |
| 76 bool handled = true; | |
| 77 IPC_BEGIN_MESSAGE_MAP(WebSharedWorkerStub, message) | |
| 78 IPC_MESSAGE_HANDLER(WorkerMsg_TerminateWorkerContext, | |
| 79 OnTerminateWorkerContext) | |
| 80 IPC_MESSAGE_HANDLER(WorkerMsg_Connect, OnConnect) | |
| 81 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 82 IPC_END_MESSAGE_MAP() | |
| 83 return handled; | |
| 84 } | |
| 85 | |
| 86 void WebSharedWorkerStub::OnChannelError() { | |
| 87 OnTerminateWorkerContext(); | |
| 88 } | |
| 89 | |
| 90 const GURL& WebSharedWorkerStub::url() { | |
| 91 return url_; | |
| 92 } | |
| 93 | |
| 94 void WebSharedWorkerStub::OnConnect(int sent_message_port_id, int routing_id) { | |
| 95 WebMessagePortChannelImpl* channel = | |
| 96 new WebMessagePortChannelImpl(routing_id, | |
| 97 sent_message_port_id, | |
| 98 base::MessageLoopProxy::current().get()); | |
| 99 if (running_) { | |
| 100 impl_->connect(channel); | |
| 101 WorkerThread::current()->Send( | |
| 102 new WorkerHostMsg_WorkerConnected(channel->message_port_id(), | |
| 103 route_id_)); | |
| 104 } else { | |
| 105 // If two documents try to load a SharedWorker at the same time, the | |
| 106 // WorkerMsg_Connect for one of the documents can come in before the | |
| 107 // worker is started. Just queue up the connect and deliver it once the | |
| 108 // worker starts. | |
| 109 pending_channels_.push_back(channel); | |
| 110 } | |
| 111 } | |
| 112 | |
| 113 void WebSharedWorkerStub::OnTerminateWorkerContext() { | |
| 114 running_ = false; | |
| 115 // Call the client to make sure context exits. | |
| 116 EnsureWorkerContextTerminates(); | |
| 117 // This may call "delete this" via WorkerScriptLoadFailed and Shutdown. | |
| 118 impl_->terminateWorkerContext(); | |
| 119 } | |
| 120 | |
| 121 void WebSharedWorkerStub::WorkerScriptLoaded() { | |
| 122 running_ = true; | |
| 123 // Process any pending connections. | |
| 124 for (PendingChannelList::const_iterator iter = pending_channels_.begin(); | |
| 125 iter != pending_channels_.end(); | |
| 126 ++iter) { | |
| 127 impl_->connect(*iter); | |
| 128 WorkerThread::current()->Send( | |
| 129 new WorkerHostMsg_WorkerConnected((*iter)->message_port_id(), | |
| 130 route_id_)); | |
| 131 } | |
| 132 pending_channels_.clear(); | |
| 133 } | |
| 134 | |
| 135 void WebSharedWorkerStub::WorkerScriptLoadFailed() { | |
| 136 for (PendingChannelList::const_iterator iter = pending_channels_.begin(); | |
| 137 iter != pending_channels_.end(); | |
| 138 ++iter) { | |
| 139 blink::WebMessagePortChannel* channel = *iter; | |
| 140 channel->destroy(); | |
| 141 } | |
| 142 pending_channels_.clear(); | |
| 143 Shutdown(); | |
| 144 } | |
| 145 | |
| 146 } // namespace content | |
| OLD | NEW |