| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/renderer/service_worker/embedded_worker_dispatcher.h" | 5 #include "content/renderer/service_worker/embedded_worker_dispatcher.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "base/strings/string16.h" | 8 #include "base/strings/string16.h" |
| 9 #include "base/strings/utf_string_conversions.h" | 9 #include "base/strings/utf_string_conversions.h" |
| 10 #include "content/child/child_process.h" | 10 #include "content/child/child_process.h" |
| 11 #include "content/child/scoped_child_process_reference.h" | 11 #include "content/child/scoped_child_process_reference.h" |
| 12 #include "content/child/thread_safe_sender.h" | 12 #include "content/child/thread_safe_sender.h" |
| 13 #include "content/common/service_worker_messages.h" | 13 #include "content/common/service_worker_messages.h" |
| 14 #include "content/renderer/render_thread_impl.h" | 14 #include "content/renderer/render_thread_impl.h" |
| 15 #include "content/renderer/service_worker/service_worker_context_client.h" | 15 #include "content/renderer/service_worker/embedded_worker_context_client.h" |
| 16 #include "third_party/WebKit/public/platform/WebString.h" | 16 #include "third_party/WebKit/public/platform/WebString.h" |
| 17 #include "third_party/WebKit/public/platform/WebURL.h" | 17 #include "third_party/WebKit/public/platform/WebURL.h" |
| 18 #include "third_party/WebKit/public/web/WebEmbeddedWorker.h" | 18 #include "third_party/WebKit/public/web/WebEmbeddedWorker.h" |
| 19 #include "third_party/WebKit/public/web/WebEmbeddedWorkerStartData.h" | 19 #include "third_party/WebKit/public/web/WebEmbeddedWorkerStartData.h" |
| 20 #include "webkit/child/worker_task_runner.h" | 20 #include "webkit/child/worker_task_runner.h" |
| 21 #include "webkit/common/user_agent/user_agent.h" | 21 #include "webkit/common/user_agent/user_agent.h" |
| 22 | 22 |
| 23 namespace content { | 23 namespace content { |
| 24 | 24 |
| 25 // A thin wrapper of WebEmbeddedWorker which also adds and releases process | 25 // A thin wrapper of WebEmbeddedWorker which also adds and releases process |
| (...skipping 11 matching lines...) Expand all Loading... |
| 37 }; | 37 }; |
| 38 | 38 |
| 39 EmbeddedWorkerDispatcher::EmbeddedWorkerDispatcher() : weak_factory_(this) {} | 39 EmbeddedWorkerDispatcher::EmbeddedWorkerDispatcher() : weak_factory_(this) {} |
| 40 | 40 |
| 41 EmbeddedWorkerDispatcher::~EmbeddedWorkerDispatcher() {} | 41 EmbeddedWorkerDispatcher::~EmbeddedWorkerDispatcher() {} |
| 42 | 42 |
| 43 bool EmbeddedWorkerDispatcher::OnMessageReceived( | 43 bool EmbeddedWorkerDispatcher::OnMessageReceived( |
| 44 const IPC::Message& message) { | 44 const IPC::Message& message) { |
| 45 bool handled = true; | 45 bool handled = true; |
| 46 IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerDispatcher, message) | 46 IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerDispatcher, message) |
| 47 IPC_MESSAGE_HANDLER(ServiceWorkerMsg_StartWorker, OnStartWorker) | 47 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StartWorker, OnStartWorker) |
| 48 IPC_MESSAGE_HANDLER(ServiceWorkerMsg_TerminateWorker, OnTerminateWorker) | 48 IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StopWorker, OnStopWorker) |
| 49 IPC_MESSAGE_UNHANDLED(handled = false) | 49 IPC_MESSAGE_UNHANDLED(handled = false) |
| 50 IPC_END_MESSAGE_MAP() | 50 IPC_END_MESSAGE_MAP() |
| 51 return handled; | 51 return handled; |
| 52 } | 52 } |
| 53 | 53 |
| 54 void EmbeddedWorkerDispatcher::WorkerContextDestroyed( | 54 void EmbeddedWorkerDispatcher::WorkerContextDestroyed( |
| 55 int embedded_worker_id) { | 55 int embedded_worker_id) { |
| 56 RenderThreadImpl::current()->thread_safe_sender()->Send( |
| 57 new EmbeddedWorkerHostMsg_WorkerStopped(embedded_worker_id)); |
| 56 workers_.Remove(embedded_worker_id); | 58 workers_.Remove(embedded_worker_id); |
| 57 } | 59 } |
| 58 | 60 |
| 59 void EmbeddedWorkerDispatcher::OnStartWorker( | 61 void EmbeddedWorkerDispatcher::OnStartWorker( |
| 60 int embedded_worker_id, | 62 int embedded_worker_id, |
| 61 int64 service_worker_version_id, | 63 int64 service_worker_version_id, |
| 62 const GURL& script_url) { | 64 const GURL& script_url) { |
| 63 DCHECK(!workers_.Lookup(embedded_worker_id)); | 65 DCHECK(!workers_.Lookup(embedded_worker_id)); |
| 64 scoped_ptr<WorkerWrapper> wrapper(new WorkerWrapper( | 66 scoped_ptr<WorkerWrapper> wrapper(new WorkerWrapper( |
| 65 blink::WebEmbeddedWorker::create( | 67 blink::WebEmbeddedWorker::create( |
| 66 new ServiceWorkerContextClient( | 68 new EmbeddedWorkerContextClient( |
| 67 embedded_worker_id, | 69 embedded_worker_id, |
| 68 service_worker_version_id, | 70 service_worker_version_id, |
| 69 script_url), | 71 script_url), |
| 70 NULL))); | 72 NULL))); |
| 71 | 73 |
| 72 blink::WebEmbeddedWorkerStartData start_data; | 74 blink::WebEmbeddedWorkerStartData start_data; |
| 73 start_data.scriptURL = script_url; | 75 start_data.scriptURL = script_url; |
| 74 start_data.userAgent = | 76 start_data.userAgent = |
| 75 base::UTF8ToUTF16(webkit_glue::GetUserAgent(script_url)); | 77 base::UTF8ToUTF16(webkit_glue::GetUserAgent(script_url)); |
| 76 | 78 |
| 77 wrapper->worker()->startWorkerContext(start_data); | 79 wrapper->worker()->startWorkerContext(start_data); |
| 78 workers_.AddWithID(wrapper.release(), embedded_worker_id); | 80 workers_.AddWithID(wrapper.release(), embedded_worker_id); |
| 79 } | 81 } |
| 80 | 82 |
| 81 void EmbeddedWorkerDispatcher::OnTerminateWorker(int embedded_worker_id) { | 83 void EmbeddedWorkerDispatcher::OnStopWorker(int embedded_worker_id) { |
| 82 WorkerWrapper* wrapper = workers_.Lookup(embedded_worker_id); | 84 WorkerWrapper* wrapper = workers_.Lookup(embedded_worker_id); |
| 83 if (!wrapper) { | 85 if (!wrapper) { |
| 84 LOG(WARNING) << "Got OnTerminateWorker for nonexistent worker"; | 86 LOG(WARNING) << "Got OnStopWorker for nonexistent worker"; |
| 85 return; | 87 return; |
| 86 } | 88 } |
| 87 | 89 |
| 88 // This should eventually call WorkerContextDestroyed. (We may need to post | 90 // This should eventually call WorkerContextDestroyed. (We may need to post |
| 89 // a delayed task to forcibly abort the worker context if we find it | 91 // a delayed task to forcibly abort the worker context if we find it |
| 90 // necessary) | 92 // necessary) |
| 91 wrapper->worker()->terminateWorkerContext(); | 93 wrapper->worker()->terminateWorkerContext(); |
| 92 } | 94 } |
| 93 | 95 |
| 94 } // namespace content | 96 } // namespace content |
| OLD | NEW |