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 |