| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2009 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 "config.h" | |
| 6 | |
| 7 #include "DedicatedWorkerContext.h" | |
| 8 #include "DedicatedWorkerThread.h" | |
| 9 #include "GenericWorkerTask.h" | |
| 10 #include "KURL.h" | |
| 11 #include "MessageEvent.h" | |
| 12 #include "MessagePort.h" | |
| 13 #include "MessagePortChannel.h" | |
| 14 #include "ScriptExecutionContext.h" | |
| 15 #include "SecurityOrigin.h" | |
| 16 #include "SerializedScriptValue.h" | |
| 17 #include "SubstituteData.h" | |
| 18 #include <wtf/MainThread.h> | |
| 19 #include <wtf/Threading.h> | |
| 20 | |
| 21 #undef LOG | |
| 22 | |
| 23 #include "webkit/api/public/WebFrameClient.h" | |
| 24 #include "webkit/api/public/WebMessagePortChannel.h" | |
| 25 #include "webkit/api/public/WebScreenInfo.h" | |
| 26 #include "webkit/api/public/WebString.h" | |
| 27 #include "webkit/api/public/WebURL.h" | |
| 28 #include "webkit/api/public/WebView.h" | |
| 29 #include "webkit/api/public/WebWorkerClient.h" | |
| 30 #include "webkit/api/src/PlatformMessagePortChannel.h" | |
| 31 #include "webkit/api/src/WebDataSourceImpl.h" | |
| 32 #include "webkit/glue/empty_webframeclient.h" | |
| 33 #include "webkit/glue/glue_util.h" | |
| 34 #include "webkit/glue/webframe_impl.h" | |
| 35 #include "webkit/glue/webpreferences.h" | |
| 36 #include "webkit/glue/webworker_impl.h" | |
| 37 | |
| 38 using WebKit::WebCursorInfo; | |
| 39 using WebKit::WebDataSource; | |
| 40 using WebKit::WebDataSourceImpl; | |
| 41 using WebKit::WebFrame; | |
| 42 using WebKit::WebFrameClient; | |
| 43 using WebKit::WebMessagePortChannel; | |
| 44 using WebKit::WebMessagePortChannelArray; | |
| 45 using WebKit::WebNavigationPolicy; | |
| 46 using WebKit::WebRect; | |
| 47 using WebKit::WebScreenInfo; | |
| 48 using WebKit::WebString; | |
| 49 using WebKit::WebURL; | |
| 50 using WebKit::WebWorker; | |
| 51 using WebKit::WebWorkerClient; | |
| 52 using WebKit::WebView; | |
| 53 | |
| 54 #if ENABLE(WORKERS) | |
| 55 | |
| 56 // Dummy WebViewDelegate - we only need it in Worker process to load a | |
| 57 // 'shadow page' which will initialize WebCore loader. | |
| 58 class WorkerWebFrameClient : public webkit_glue::EmptyWebFrameClient { | |
| 59 public: | |
| 60 // Tell the loader to load the data into the 'shadow page' synchronously, | |
| 61 // so we can grab the resulting Document right after load. | |
| 62 virtual void didCreateDataSource(WebFrame* frame, WebDataSource* ds) { | |
| 63 static_cast<WebDataSourceImpl*>(ds)->setDeferMainResourceDataLoad(false); | |
| 64 } | |
| 65 | |
| 66 // Lazy allocate and leak this instance. | |
| 67 static WorkerWebFrameClient* GetSharedInstance() { | |
| 68 static WorkerWebFrameClient client; | |
| 69 return &client; | |
| 70 } | |
| 71 | |
| 72 private: | |
| 73 WorkerWebFrameClient() {} | |
| 74 }; | |
| 75 | |
| 76 namespace WebKit { | |
| 77 | |
| 78 WebWorker* WebWorker::create(WebWorkerClient* client) { | |
| 79 return new WebWorkerImpl(client); | |
| 80 } | |
| 81 | |
| 82 } | |
| 83 | |
| 84 // This function is called on the main thread to force to initialize some static | |
| 85 // values used in WebKit before any worker thread is started. This is because in | |
| 86 // our worker processs, we do not run any WebKit code in main thread and thus | |
| 87 // when multiple workers try to start at the same time, we might hit crash due | |
| 88 // to contention for initializing static values. | |
| 89 void InitializeWebKitStaticValues() { | |
| 90 static bool initialized = false; | |
| 91 if (!initialized) { | |
| 92 initialized= true; | |
| 93 // Note that we have to pass a URL with valid protocol in order to follow | |
| 94 // the path to do static value initializations. | |
| 95 WTF::RefPtr<WebCore::SecurityOrigin> origin = | |
| 96 WebCore::SecurityOrigin::create(WebCore::KURL(WebCore::ParsedURLString, | |
| 97 "http://localhost")); | |
| 98 origin.release(); | |
| 99 } | |
| 100 } | |
| 101 | |
| 102 WebWorkerImpl::WebWorkerImpl(WebWorkerClient* client) | |
| 103 : client_(client), | |
| 104 web_view_(NULL), | |
| 105 asked_to_terminate_(false) { | |
| 106 InitializeWebKitStaticValues(); | |
| 107 } | |
| 108 | |
| 109 WebWorkerImpl::~WebWorkerImpl() { | |
| 110 web_view_->close(); | |
| 111 } | |
| 112 | |
| 113 void WebWorkerImpl::PostMessageToWorkerContextTask( | |
| 114 WebCore::ScriptExecutionContext* context, | |
| 115 WebWorkerImpl* this_ptr, | |
| 116 const WebCore::String& message, | |
| 117 WTF::PassOwnPtr<WebCore::MessagePortChannelArray> channels) { | |
| 118 ASSERT(context->isWorkerContext()); | |
| 119 WebCore::DedicatedWorkerContext* worker_context = | |
| 120 static_cast<WebCore::DedicatedWorkerContext*>(context); | |
| 121 | |
| 122 WTF::OwnPtr<WebCore::MessagePortArray> ports = | |
| 123 WebCore::MessagePort::entanglePorts(*context, channels.release()); | |
| 124 WTF::RefPtr<WebCore::SerializedScriptValue> serialized_message = | |
| 125 WebCore::SerializedScriptValue::create(message); | |
| 126 worker_context->dispatchEvent( | |
| 127 WebCore::MessageEvent::create(ports.release(), | |
| 128 serialized_message.release())); | |
| 129 | |
| 130 this_ptr->confirmMessageFromWorkerObject( | |
| 131 worker_context->hasPendingActivity()); | |
| 132 } | |
| 133 | |
| 134 // WebWorker ------------------------------------------------------------------- | |
| 135 | |
| 136 void WebWorkerImpl::startWorkerContext(const WebURL& script_url, | |
| 137 const WebString& user_agent, | |
| 138 const WebString& source_code) { | |
| 139 // Create 'shadow page'. This page is never displayed, it is used to proxy the | |
| 140 // loading requests from the worker context to the rest of WebKit and Chromium | |
| 141 // infrastructure. | |
| 142 ASSERT(!web_view_); | |
| 143 web_view_ = WebView::create(NULL); | |
| 144 WebPreferences().Apply(web_view_); | |
| 145 web_view_->initializeMainFrame(WorkerWebFrameClient::GetSharedInstance()); | |
| 146 | |
| 147 WebFrameImpl* web_frame = static_cast<WebFrameImpl*>(web_view_->mainFrame()); | |
| 148 | |
| 149 // Construct substitute data source for the 'shadow page'. We only need it | |
| 150 // to have same origin as the worker so the loading checks work correctly. | |
| 151 WebCore::CString content(""); | |
| 152 int len = static_cast<int>(content.length()); | |
| 153 RefPtr<WebCore::SharedBuffer> buf( | |
| 154 WebCore::SharedBuffer::create(content.data(), len)); | |
| 155 WebCore::SubstituteData subst_data(buf, | |
| 156 WebCore::String("text/html"), | |
| 157 WebCore::String("UTF-8"), | |
| 158 WebCore::KURL()); | |
| 159 WebCore::ResourceRequest request(webkit_glue::GURLToKURL(script_url), | |
| 160 WebCore::CString()); | |
| 161 web_frame->frame()->loader()->load(request, subst_data, false); | |
| 162 | |
| 163 // This document will be used as 'loading context' for the worker. | |
| 164 loading_document_ = web_frame->frame()->document(); | |
| 165 | |
| 166 worker_thread_ = WebCore::DedicatedWorkerThread::create( | |
| 167 webkit_glue::WebURLToKURL(script_url), | |
| 168 webkit_glue::WebStringToString(user_agent), | |
| 169 webkit_glue::WebStringToString(source_code), | |
| 170 *this, | |
| 171 *this); | |
| 172 | |
| 173 // Worker initialization means a pending activity. | |
| 174 reportPendingActivity(true); | |
| 175 | |
| 176 worker_thread_->start(); | |
| 177 } | |
| 178 | |
| 179 void WebWorkerImpl::terminateWorkerContext() { | |
| 180 if (asked_to_terminate_) | |
| 181 return; | |
| 182 asked_to_terminate_ = true; | |
| 183 | |
| 184 if (worker_thread_) | |
| 185 worker_thread_->stop(); | |
| 186 } | |
| 187 | |
| 188 void WebWorkerImpl::postMessageToWorkerContext( | |
| 189 const WebString& message, | |
| 190 const WebMessagePortChannelArray& webchannels) { | |
| 191 | |
| 192 WTF::OwnPtr<WebCore::MessagePortChannelArray> channels; | |
| 193 if (webchannels.size()) { | |
| 194 channels = new WebCore::MessagePortChannelArray(webchannels.size()); | |
| 195 for (size_t i = 0; i < webchannels.size(); ++i) { | |
| 196 RefPtr<WebCore::PlatformMessagePortChannel> platform_channel = | |
| 197 WebCore::PlatformMessagePortChannel::create(webchannels[i]); | |
| 198 webchannels[i]->setClient(platform_channel.get()); | |
| 199 (*channels)[i] = WebCore::MessagePortChannel::create(platform_channel); | |
| 200 } | |
| 201 } | |
| 202 | |
| 203 worker_thread_->runLoop().postTask(WebCore::createCallbackTask( | |
| 204 &PostMessageToWorkerContextTask, | |
| 205 this, | |
| 206 webkit_glue::WebStringToString(message), | |
| 207 channels.release())); | |
| 208 } | |
| 209 | |
| 210 void WebWorkerImpl::workerObjectDestroyed() { | |
| 211 // Worker object in the renderer was destroyed, perhaps a result of GC. | |
| 212 // For us, it's a signal to start terminating the WorkerContext too. | |
| 213 // TODO(dimich): when 'kill a worker' html5 spec algorithm is implemented, it | |
| 214 // should be used here instead of 'terminate a worker'. | |
| 215 terminateWorkerContext(); | |
| 216 } | |
| 217 | |
| 218 void WebWorkerImpl::clientDestroyed() { | |
| 219 client_ = NULL; | |
| 220 } | |
| 221 | |
| 222 void WebWorkerImpl::DispatchTaskToMainThread( | |
| 223 PassRefPtr<WebCore::ScriptExecutionContext::Task> task) { | |
| 224 return WTF::callOnMainThread(InvokeTaskMethod, task.releaseRef()); | |
| 225 } | |
| 226 | |
| 227 void WebWorkerImpl::InvokeTaskMethod(void* param) { | |
| 228 WebCore::ScriptExecutionContext::Task* task = | |
| 229 static_cast<WebCore::ScriptExecutionContext::Task*>(param); | |
| 230 task->performTask(NULL); | |
| 231 task->deref(); | |
| 232 } | |
| 233 | |
| 234 // WorkerObjectProxy ----------------------------------------------------------- | |
| 235 | |
| 236 void WebWorkerImpl::postMessageToWorkerObject( | |
| 237 WTF::PassRefPtr<WebCore::SerializedScriptValue> message, | |
| 238 WTF::PassOwnPtr<WebCore::MessagePortChannelArray> channels) { | |
| 239 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 240 &PostMessageTask, | |
| 241 this, | |
| 242 message->toString(), | |
| 243 channels)); | |
| 244 } | |
| 245 | |
| 246 void WebWorkerImpl::PostMessageTask( | |
| 247 WebCore::ScriptExecutionContext* context, | |
| 248 WebWorkerImpl* this_ptr, | |
| 249 WebCore::String message, | |
| 250 WTF::PassOwnPtr<WebCore::MessagePortChannelArray> channels) { | |
| 251 if (!this_ptr->client_) | |
| 252 return; | |
| 253 | |
| 254 WebMessagePortChannelArray web_channels( | |
| 255 channels.get() ? channels->size() : 0); | |
| 256 for (size_t i = 0; i < web_channels.size(); ++i) { | |
| 257 web_channels[i] = (*channels)[i]->channel()->webChannelRelease(); | |
| 258 web_channels[i]->setClient(0); | |
| 259 } | |
| 260 | |
| 261 this_ptr->client_->postMessageToWorkerObject( | |
| 262 webkit_glue::StringToWebString(message), web_channels); | |
| 263 } | |
| 264 | |
| 265 void WebWorkerImpl::postExceptionToWorkerObject( | |
| 266 const WebCore::String& error_message, | |
| 267 int line_number, | |
| 268 const WebCore::String& source_url) { | |
| 269 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 270 &PostExceptionTask, | |
| 271 this, | |
| 272 error_message, | |
| 273 line_number, | |
| 274 source_url)); | |
| 275 } | |
| 276 | |
| 277 void WebWorkerImpl::PostExceptionTask( | |
| 278 WebCore::ScriptExecutionContext* context, | |
| 279 WebWorkerImpl* this_ptr, | |
| 280 const WebCore::String& error_message, | |
| 281 int line_number, | |
| 282 const WebCore::String& source_url) { | |
| 283 if (!this_ptr->client_) | |
| 284 return; | |
| 285 | |
| 286 this_ptr->client_->postExceptionToWorkerObject( | |
| 287 webkit_glue::StringToWebString(error_message), | |
| 288 line_number, | |
| 289 webkit_glue::StringToWebString(source_url)); | |
| 290 } | |
| 291 | |
| 292 void WebWorkerImpl::postConsoleMessageToWorkerObject( | |
| 293 WebCore::MessageDestination destination, | |
| 294 WebCore::MessageSource source, | |
| 295 WebCore::MessageType type, | |
| 296 WebCore::MessageLevel level, | |
| 297 const WebCore::String& message, | |
| 298 int line_number, | |
| 299 const WebCore::String& source_url) { | |
| 300 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 301 &PostConsoleMessageTask, | |
| 302 this, | |
| 303 static_cast<int>(destination), | |
| 304 static_cast<int>(source), | |
| 305 static_cast<int>(type), | |
| 306 static_cast<int>(level), | |
| 307 message, | |
| 308 line_number, | |
| 309 source_url)); | |
| 310 } | |
| 311 | |
| 312 void WebWorkerImpl::PostConsoleMessageTask( | |
| 313 WebCore::ScriptExecutionContext* context, | |
| 314 WebWorkerImpl* this_ptr, | |
| 315 int destination, | |
| 316 int source, | |
| 317 int type, | |
| 318 int level, | |
| 319 const WebCore::String& message, | |
| 320 int line_number, | |
| 321 const WebCore::String& source_url) { | |
| 322 if (!this_ptr->client_) | |
| 323 return; | |
| 324 | |
| 325 this_ptr->client_->postConsoleMessageToWorkerObject( | |
| 326 destination, | |
| 327 source, | |
| 328 type, | |
| 329 level, | |
| 330 webkit_glue::StringToWebString(message), | |
| 331 line_number, | |
| 332 webkit_glue::StringToWebString(source_url)); | |
| 333 } | |
| 334 | |
| 335 void WebWorkerImpl::confirmMessageFromWorkerObject(bool has_pending_activity) { | |
| 336 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 337 &ConfirmMessageTask, | |
| 338 this, | |
| 339 has_pending_activity)); | |
| 340 } | |
| 341 | |
| 342 void WebWorkerImpl::ConfirmMessageTask( | |
| 343 WebCore::ScriptExecutionContext* context, | |
| 344 WebWorkerImpl* this_ptr, | |
| 345 bool has_pending_activity) { | |
| 346 if (!this_ptr->client_) | |
| 347 return; | |
| 348 | |
| 349 this_ptr->client_->confirmMessageFromWorkerObject(has_pending_activity); | |
| 350 } | |
| 351 | |
| 352 void WebWorkerImpl::reportPendingActivity(bool has_pending_activity) { | |
| 353 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 354 &ReportPendingActivityTask, | |
| 355 this, | |
| 356 has_pending_activity)); | |
| 357 } | |
| 358 | |
| 359 void WebWorkerImpl::ReportPendingActivityTask( | |
| 360 WebCore::ScriptExecutionContext* context, | |
| 361 WebWorkerImpl* this_ptr, | |
| 362 bool has_pending_activity) { | |
| 363 if (!this_ptr->client_) | |
| 364 return; | |
| 365 | |
| 366 this_ptr->client_->reportPendingActivity(has_pending_activity); | |
| 367 } | |
| 368 | |
| 369 void WebWorkerImpl::workerContextDestroyed() { | |
| 370 DispatchTaskToMainThread(WebCore::createCallbackTask( | |
| 371 &WorkerContextDestroyedTask, | |
| 372 this)); | |
| 373 } | |
| 374 | |
| 375 // WorkerLoaderProxy ----------------------------------------------------------- | |
| 376 | |
| 377 void WebWorkerImpl::postTaskToLoader( | |
| 378 PassRefPtr<WebCore::ScriptExecutionContext::Task> task) { | |
| 379 ASSERT(loading_document_->isDocument()); | |
| 380 loading_document_->postTask(task); | |
| 381 } | |
| 382 | |
| 383 void WebWorkerImpl::postTaskForModeToWorkerContext( | |
| 384 PassRefPtr<WebCore::ScriptExecutionContext::Task> task, | |
| 385 const WebCore::String& mode) { | |
| 386 worker_thread_->runLoop().postTaskForMode(task, mode); | |
| 387 } | |
| 388 | |
| 389 void WebWorkerImpl::WorkerContextDestroyedTask( | |
| 390 WebCore::ScriptExecutionContext* context, | |
| 391 WebWorkerImpl* this_ptr) { | |
| 392 if (this_ptr->client_) | |
| 393 this_ptr->client_->workerContextDestroyed(); | |
| 394 | |
| 395 // The lifetime of this proxy is controlled by the worker context. | |
| 396 delete this_ptr; | |
| 397 } | |
| 398 | |
| 399 #else | |
| 400 | |
| 401 namespace WebKit { | |
| 402 | |
| 403 WebWorker* WebWorker::create(WebWorkerClient* client) { | |
| 404 return NULL; | |
| 405 } | |
| 406 | |
| 407 } | |
| 408 | |
| 409 #endif | |
| OLD | NEW |