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 "chrome/browser/worker_host/worker_process_host.h" |
| 6 |
| 7 #include "base/command_line.h" |
| 8 #include "base/debug_util.h" |
| 9 #include "base/logging.h" |
| 10 #include "base/path_service.h" |
| 11 #include "base/process_util.h" |
| 12 #include "chrome/browser/browser_process.h" |
| 13 #include "chrome/browser/renderer_host/resource_message_filter.h" |
| 14 #include "chrome/browser/worker_host/worker_service.h" |
| 15 #include "chrome/common/chrome_switches.h" |
| 16 #include "chrome/common/debug_flags.h" |
| 17 #include "chrome/common/process_watcher.h" |
| 18 #include "chrome/common/render_messages.h" |
| 19 #include "chrome/common/worker_messages.h" |
| 20 |
| 21 |
| 22 WorkerProcessHost::WorkerProcessHost(MessageLoop* main_message_loop) |
| 23 : ChildProcessHost(WORKER_PROCESS, main_message_loop) { |
| 24 } |
| 25 |
| 26 WorkerProcessHost::~WorkerProcessHost() { |
| 27 } |
| 28 |
| 29 bool WorkerProcessHost::Init() { |
| 30 // TODO(jabdelmalek): figure out what to set as the title. |
| 31 set_name(L"TBD"); |
| 32 |
| 33 if (!CreateChannel()) |
| 34 return false; |
| 35 |
| 36 std::wstring exe_path; |
| 37 if (!PathService::Get(base::FILE_EXE, &exe_path)) |
| 38 return false; |
| 39 |
| 40 CommandLine cmd_line(exe_path); |
| 41 |
| 42 // TODO(jabdelmalek): factor out common code from renderer/plugin that does |
| 43 // sandboxing and command line copying and reuse here. |
| 44 cmd_line.AppendSwitchWithValue(switches::kProcessType, |
| 45 switches::kWorkerProcess); |
| 46 cmd_line.AppendSwitchWithValue(switches::kProcessChannelID, channel_id()); |
| 47 base::ProcessHandle handle; |
| 48 if (!base::LaunchApp(cmd_line, false, false, &handle)) |
| 49 return false; |
| 50 SetHandle(handle); |
| 51 |
| 52 return true; |
| 53 } |
| 54 |
| 55 void WorkerProcessHost::CreateWorker(const GURL& url, |
| 56 int worker_route_id, |
| 57 int renderer_route_id, |
| 58 ResourceMessageFilter* filter) { |
| 59 WorkerInstance instance; |
| 60 instance.worker_route_id = worker_route_id; |
| 61 instance.renderer_route_id = renderer_route_id; |
| 62 instance.filter = filter; |
| 63 instances_.push_back(instance); |
| 64 Send(new WorkerProcessMsg_CreateWorker(url, worker_route_id)); |
| 65 } |
| 66 |
| 67 bool WorkerProcessHost::FilterMessage(const IPC::Message& message) { |
| 68 for (Instances::iterator i = instances_.begin(); i != instances_.end(); ++i) { |
| 69 if (i->renderer_route_id == message.routing_id()) { |
| 70 IPC::Message* new_message = new IPC::Message(message); |
| 71 new_message->set_routing_id(i->worker_route_id); |
| 72 Send(new_message); |
| 73 return true; |
| 74 } |
| 75 } |
| 76 |
| 77 return false; |
| 78 } |
| 79 |
| 80 void WorkerProcessHost::OnMessageReceived(const IPC::Message& message) { |
| 81 for (Instances::iterator i = instances_.begin(); i != instances_.end(); ++i) { |
| 82 if (i->worker_route_id == message.routing_id()) { |
| 83 IPC::Message* new_message = new IPC::Message(message); |
| 84 new_message->set_routing_id(i->renderer_route_id); |
| 85 i->filter->Send(new_message); |
| 86 break; |
| 87 } |
| 88 } |
| 89 } |
| 90 |
| 91 void WorkerProcessHost::RendererShutdown(ResourceMessageFilter* filter) { |
| 92 for (Instances::iterator i = instances_.begin(); i != instances_.end();) { |
| 93 if (i->filter == filter) { |
| 94 i = instances_.erase(i); |
| 95 } else { |
| 96 ++i; |
| 97 } |
| 98 } |
| 99 } |
OLD | NEW |