embedded_worker_dispatcher.cc revision 5c02ac1a9c1b504631c0a3d2b6e737b5d738bae1 
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
3// found in the LICENSE file.
4
5#include "content/renderer/service_worker/embedded_worker_dispatcher.h"
6
7#include "base/memory/scoped_ptr.h"
8#include "base/strings/string16.h"
9#include "base/strings/utf_string_conversions.h"
10#include "content/child/child_process.h"
11#include "content/child/scoped_child_process_reference.h"
12#include "content/child/thread_safe_sender.h"
13#include "content/child/worker_task_runner.h"
14#include "content/common/devtools_messages.h"
15#include "content/common/service_worker/embedded_worker_messages.h"
16#include "content/public/common/content_client.h"
17#include "content/renderer/render_thread_impl.h"
18#include "content/renderer/service_worker/embedded_worker_context_client.h"
19#include "content/renderer/service_worker/embedded_worker_devtools_agent.h"
20#include "third_party/WebKit/public/platform/WebString.h"
21#include "third_party/WebKit/public/platform/WebURL.h"
22#include "third_party/WebKit/public/web/WebEmbeddedWorker.h"
23#include "third_party/WebKit/public/web/WebEmbeddedWorkerStartData.h"
24
25namespace content {
26
27// A thin wrapper of WebEmbeddedWorker which also adds and releases process
28// references automatically.
29class EmbeddedWorkerDispatcher::WorkerWrapper {
30 public:
31  WorkerWrapper(blink::WebEmbeddedWorker* worker, int devtools_agent_route_id)
32      : worker_(worker),
33        dev_tools_agent_(
34            new EmbeddedWorkerDevToolsAgent(worker, devtools_agent_route_id)) {}
35  ~WorkerWrapper() {}
36
37  blink::WebEmbeddedWorker* worker() { return worker_.get(); }
38
39 private:
40  ScopedChildProcessReference process_ref_;
41  scoped_ptr<blink::WebEmbeddedWorker> worker_;
42  scoped_ptr<EmbeddedWorkerDevToolsAgent> dev_tools_agent_;
43};
44
45EmbeddedWorkerDispatcher::EmbeddedWorkerDispatcher() : weak_factory_(this) {}
46
47EmbeddedWorkerDispatcher::~EmbeddedWorkerDispatcher() {}
48
49bool EmbeddedWorkerDispatcher::OnMessageReceived(
50    const IPC::Message& message) {
51  bool handled = true;
52  IPC_BEGIN_MESSAGE_MAP(EmbeddedWorkerDispatcher, message)
53    IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StartWorker, OnStartWorker)
54    IPC_MESSAGE_HANDLER(EmbeddedWorkerMsg_StopWorker, OnStopWorker)
55    IPC_MESSAGE_UNHANDLED(handled = false)
56  IPC_END_MESSAGE_MAP()
57  return handled;
58}
59
60void EmbeddedWorkerDispatcher::WorkerContextDestroyed(
61    int embedded_worker_id) {
62  RenderThreadImpl::current()->thread_safe_sender()->Send(
63      new EmbeddedWorkerHostMsg_WorkerStopped(embedded_worker_id));
64  workers_.Remove(embedded_worker_id);
65}
66
67void EmbeddedWorkerDispatcher::OnStartWorker(
68    const EmbeddedWorkerMsg_StartWorker_Params& params) {
69  DCHECK(!workers_.Lookup(params.embedded_worker_id));
70  RenderThread::Get()->EnsureWebKitInitialized();
71  scoped_ptr<WorkerWrapper> wrapper(
72      new WorkerWrapper(blink::WebEmbeddedWorker::create(
73                            new EmbeddedWorkerContextClient(
74                                params.embedded_worker_id,
75                                params.service_worker_version_id,
76                                params.scope,
77                                params.script_url,
78                                params.worker_devtools_agent_route_id),
79                            NULL),
80                        params.worker_devtools_agent_route_id));
81
82  blink::WebEmbeddedWorkerStartData start_data;
83  start_data.scriptURL = params.script_url;
84  start_data.userAgent = base::UTF8ToUTF16(GetContentClient()->GetUserAgent());
85  start_data.startMode = blink::WebEmbeddedWorkerStartModeDontPauseOnStart;
86
87  wrapper->worker()->startWorkerContext(start_data);
88  workers_.AddWithID(wrapper.release(), params.embedded_worker_id);
89}
90
91void EmbeddedWorkerDispatcher::OnStopWorker(int embedded_worker_id) {
92  WorkerWrapper* wrapper = workers_.Lookup(embedded_worker_id);
93  if (!wrapper) {
94    LOG(WARNING) << "Got OnStopWorker for nonexistent worker";
95    return;
96  }
97
98  // This should eventually call WorkerContextDestroyed. (We may need to post
99  // a delayed task to forcibly abort the worker context if we find it
100  // necessary)
101  wrapper->worker()->terminateWorkerContext();
102}
103
104}  // namespace content
105