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/worker_thread.h" 6 7#include "base/command_line.h" 8#include "base/lazy_instance.h" 9#include "base/threading/thread_local.h" 10#include "content/child/appcache/appcache_dispatcher.h" 11#include "content/child/appcache/appcache_frontend_impl.h" 12#include "content/child/db_message_filter.h" 13#include "content/child/indexed_db/indexed_db_message_filter.h" 14#include "content/child/runtime_features.h" 15#include "content/child/web_database_observer_impl.h" 16#include "content/common/child_process_messages.h" 17#include "content/common/worker_messages.h" 18#include "content/public/common/content_switches.h" 19#include "content/worker/websharedworker_stub.h" 20#include "content/worker/worker_webkitplatformsupport_impl.h" 21#include "ipc/ipc_sync_channel.h" 22#include "third_party/WebKit/public/platform/WebBlobRegistry.h" 23#include "third_party/WebKit/public/web/WebDatabase.h" 24#include "third_party/WebKit/public/web/WebKit.h" 25#include "third_party/WebKit/public/web/WebRuntimeFeatures.h" 26#include "webkit/glue/webkit_glue.h" 27 28using blink::WebRuntimeFeatures; 29 30namespace content { 31 32static base::LazyInstance<base::ThreadLocalPointer<WorkerThread> > lazy_tls = 33 LAZY_INSTANCE_INITIALIZER; 34 35WorkerThread::WorkerThread() { 36 lazy_tls.Pointer()->Set(this); 37 webkit_platform_support_.reset(new WorkerWebKitPlatformSupportImpl( 38 thread_safe_sender(), 39 sync_message_filter(), 40 quota_message_filter())); 41 42 const CommandLine& command_line = *CommandLine::ForCurrentProcess(); 43 if (command_line.HasSwitch(switches::kJavaScriptFlags)) { 44 webkit_glue::SetJavaScriptFlags( 45 command_line.GetSwitchValueASCII(switches::kJavaScriptFlags)); 46 } 47 SetRuntimeFeaturesDefaultsAndUpdateFromArgs(command_line); 48 49 blink::initialize(webkit_platform_support_.get()); 50 51 appcache_dispatcher_.reset( 52 new AppCacheDispatcher(this, new AppCacheFrontendImpl())); 53 54 db_message_filter_ = new DBMessageFilter(); 55 channel()->AddFilter(db_message_filter_.get()); 56 57 indexed_db_message_filter_ = new IndexedDBMessageFilter( 58 thread_safe_sender()); 59 channel()->AddFilter(indexed_db_message_filter_->GetFilter()); 60 61} 62 63void WorkerThread::OnShutdown() { 64 // The worker process is to be shut down gracefully. Ask the browser 65 // process to shut it down forcefully instead and wait on the message, so that 66 // there are no races between threads when the process is shutting down. 67 Send(new WorkerProcessHostMsg_ForceKillWorker()); 68} 69 70WorkerThread::~WorkerThread() { 71} 72 73void WorkerThread::Shutdown() { 74 ChildThread::Shutdown(); 75 76 if (webkit_platform_support_) { 77 webkit_platform_support_->web_database_observer_impl()-> 78 WaitForAllDatabasesToClose(); 79 } 80 81 // Shutdown in reverse of the initialization order. 82 indexed_db_message_filter_ = NULL; 83 84 channel()->RemoveFilter(db_message_filter_.get()); 85 db_message_filter_ = NULL; 86 87 blink::shutdown(); 88 lazy_tls.Pointer()->Set(NULL); 89} 90 91WorkerThread* WorkerThread::current() { 92 return lazy_tls.Pointer()->Get(); 93} 94 95bool WorkerThread::OnControlMessageReceived(const IPC::Message& msg) { 96 // Appcache messages are handled by a delegate. 97 if (appcache_dispatcher_->OnMessageReceived(msg)) 98 return true; 99 100 bool handled = true; 101 IPC_BEGIN_MESSAGE_MAP(WorkerThread, msg) 102 IPC_MESSAGE_HANDLER(WorkerProcessMsg_CreateWorker, OnCreateWorker) 103 IPC_MESSAGE_UNHANDLED(handled = false) 104 IPC_END_MESSAGE_MAP() 105 return handled; 106} 107 108bool WorkerThread::OnMessageReceived(const IPC::Message& msg) { 109 bool handled = true; 110 IPC_BEGIN_MESSAGE_MAP(WorkerThread, msg) 111 IPC_MESSAGE_HANDLER(ChildProcessMsg_Shutdown, OnShutdown) 112 IPC_MESSAGE_UNHANDLED(handled = ChildThread::OnMessageReceived(msg)) 113 IPC_END_MESSAGE_MAP() 114 return handled; 115} 116 117void WorkerThread::OnCreateWorker( 118 const WorkerProcessMsg_CreateWorker_Params& params) { 119 WorkerAppCacheInitInfo appcache_init_info( 120 params.creator_process_id, 121 params.shared_worker_appcache_id); 122 123 // WebSharedWorkerStub own themselves. 124 new WebSharedWorkerStub(params.name, params.route_id, appcache_init_info); 125} 126 127// The browser process is likely dead. Terminate all workers. 128void WorkerThread::OnChannelError() { 129 set_on_channel_error_called(true); 130 131 for (WorkerStubsList::iterator it = worker_stubs_.begin(); 132 it != worker_stubs_.end(); ++it) { 133 (*it)->OnChannelError(); 134 } 135} 136 137void WorkerThread::RemoveWorkerStub(WebSharedWorkerStub* stub) { 138 worker_stubs_.erase(stub); 139} 140 141void WorkerThread::AddWorkerStub(WebSharedWorkerStub* stub) { 142 worker_stubs_.insert(stub); 143} 144 145} // namespace content 146