| OLD | NEW |
| (Empty) |
| 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 "chrome/browser/nacl_host/nacl_broker_service_win.h" | |
| 6 | |
| 7 #include "chrome/browser/nacl_host/nacl_process_host.h" | |
| 8 #include "components/nacl/common/nacl_process_type.h" | |
| 9 #include "content/public/browser/browser_child_process_host_iterator.h" | |
| 10 | |
| 11 using content::BrowserChildProcessHostIterator; | |
| 12 | |
| 13 NaClBrokerService* NaClBrokerService::GetInstance() { | |
| 14 return Singleton<NaClBrokerService>::get(); | |
| 15 } | |
| 16 | |
| 17 NaClBrokerService::NaClBrokerService() | |
| 18 : loaders_running_(0) { | |
| 19 } | |
| 20 | |
| 21 bool NaClBrokerService::StartBroker() { | |
| 22 NaClBrokerHost* broker_host = new NaClBrokerHost; | |
| 23 if (!broker_host->Init()) { | |
| 24 delete broker_host; | |
| 25 return false; | |
| 26 } | |
| 27 return true; | |
| 28 } | |
| 29 | |
| 30 bool NaClBrokerService::LaunchLoader( | |
| 31 base::WeakPtr<NaClProcessHost> nacl_process_host, | |
| 32 const std::string& loader_channel_id) { | |
| 33 // Add task to the list | |
| 34 pending_launches_[loader_channel_id] = nacl_process_host; | |
| 35 NaClBrokerHost* broker_host = GetBrokerHost(); | |
| 36 | |
| 37 if (!broker_host) { | |
| 38 if (!StartBroker()) | |
| 39 return false; | |
| 40 broker_host = GetBrokerHost(); | |
| 41 } | |
| 42 broker_host->LaunchLoader(loader_channel_id); | |
| 43 | |
| 44 return true; | |
| 45 } | |
| 46 | |
| 47 void NaClBrokerService::OnLoaderLaunched(const std::string& channel_id, | |
| 48 base::ProcessHandle handle) { | |
| 49 PendingLaunchesMap::iterator it = pending_launches_.find(channel_id); | |
| 50 if (pending_launches_.end() == it) | |
| 51 NOTREACHED(); | |
| 52 | |
| 53 NaClProcessHost* client = it->second.get(); | |
| 54 if (client) | |
| 55 client->OnProcessLaunchedByBroker(handle); | |
| 56 pending_launches_.erase(it); | |
| 57 ++loaders_running_; | |
| 58 } | |
| 59 | |
| 60 void NaClBrokerService::OnLoaderDied() { | |
| 61 DCHECK(loaders_running_ > 0); | |
| 62 --loaders_running_; | |
| 63 // Stop the broker only if there are no loaders running or being launched. | |
| 64 NaClBrokerHost* broker_host = GetBrokerHost(); | |
| 65 if (loaders_running_ + pending_launches_.size() == 0 && broker_host != NULL) { | |
| 66 broker_host->StopBroker(); | |
| 67 } | |
| 68 } | |
| 69 | |
| 70 bool NaClBrokerService::LaunchDebugExceptionHandler( | |
| 71 base::WeakPtr<NaClProcessHost> nacl_process_host, int32 pid, | |
| 72 base::ProcessHandle process_handle, const std::string& startup_info) { | |
| 73 pending_debuggers_[pid] = nacl_process_host; | |
| 74 NaClBrokerHost* broker_host = GetBrokerHost(); | |
| 75 if (!broker_host) | |
| 76 return false; | |
| 77 return broker_host->LaunchDebugExceptionHandler(pid, process_handle, | |
| 78 startup_info); | |
| 79 } | |
| 80 | |
| 81 void NaClBrokerService::OnDebugExceptionHandlerLaunched(int32 pid, | |
| 82 bool success) { | |
| 83 PendingDebugExceptionHandlersMap::iterator it = pending_debuggers_.find(pid); | |
| 84 if (pending_debuggers_.end() == it) | |
| 85 NOTREACHED(); | |
| 86 | |
| 87 NaClProcessHost* client = it->second.get(); | |
| 88 if (client) | |
| 89 client->OnDebugExceptionHandlerLaunchedByBroker(success); | |
| 90 pending_debuggers_.erase(it); | |
| 91 } | |
| 92 | |
| 93 NaClBrokerHost* NaClBrokerService::GetBrokerHost() { | |
| 94 BrowserChildProcessHostIterator iter(PROCESS_TYPE_NACL_BROKER); | |
| 95 while (!iter.Done()) { | |
| 96 NaClBrokerHost* host = static_cast<NaClBrokerHost*>(iter.GetDelegate()); | |
| 97 if (!host->IsTerminating()) | |
| 98 return host; | |
| 99 ++iter; | |
| 100 } | |
| 101 return NULL; | |
| 102 } | |
| OLD | NEW |