Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "components/arc/arc_bridge_service.h" | 5 #include "components/arc/arc_bridge_service.h" |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/sequenced_task_runner.h" |
| 8 #include "base/files/file_util.h" | |
| 9 #include "base/prefs/pref_registry_simple.h" | |
| 10 #include "base/prefs/pref_service.h" | |
| 11 #include "base/task_runner_util.h" | |
| 12 #include "base/thread_task_runner_handle.h" | 8 #include "base/thread_task_runner_handle.h" |
| 13 #include "chromeos/dbus/dbus_method_call_status.h" | |
| 14 #include "chromeos/dbus/dbus_thread_manager.h" | |
| 15 #include "chromeos/dbus/session_manager_client.h" | |
| 16 #include "components/arc/common/arc_host_messages.h" | |
| 17 #include "components/arc/common/arc_instance_messages.h" | |
| 18 #include "ipc/ipc_channel.h" | |
| 19 | 9 |
| 20 namespace arc { | 10 namespace arc { |
| 21 | 11 |
| 22 namespace { | 12 namespace { |
| 23 | 13 |
| 24 const base::FilePath::CharType kArcBridgeSocketPath[] = | |
| 25 FILE_PATH_LITERAL("/home/chronos/ArcBridge/bridge.sock"); | |
| 26 | |
| 27 // Weak pointer. This class is owned by ChromeBrowserMainPartsChromeos. | 14 // Weak pointer. This class is owned by ChromeBrowserMainPartsChromeos. |
| 28 ArcBridgeService* g_arc_bridge_service = nullptr; | 15 ArcBridgeService* g_arc_bridge_service = nullptr; |
| 29 | 16 |
| 30 } // namespace | 17 } // namespace |
| 31 | 18 |
| 32 ArcBridgeService::ArcBridgeService( | 19 ArcBridgeService::ArcBridgeService() |
| 33 const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner, | |
| 34 const scoped_refptr<base::SequencedTaskRunner>& file_task_runner) | |
| 35 : origin_task_runner_(base::ThreadTaskRunnerHandle::Get()), | 20 : origin_task_runner_(base::ThreadTaskRunnerHandle::Get()), |
| 36 ipc_task_runner_(ipc_task_runner), | |
| 37 file_task_runner_(file_task_runner), | |
| 38 session_started_(false), | |
| 39 available_(false), | 21 available_(false), |
| 40 state_(State::STOPPED), | 22 state_(State::STOPPED) { |
| 41 weak_factory_(this) { | |
| 42 DCHECK(!g_arc_bridge_service); | 23 DCHECK(!g_arc_bridge_service); |
| 43 g_arc_bridge_service = this; | 24 g_arc_bridge_service = this; |
| 44 } | 25 } |
| 45 | 26 |
| 46 ArcBridgeService::~ArcBridgeService() { | 27 ArcBridgeService::~ArcBridgeService() { |
| 47 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | 28 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); |
| 48 DCHECK(state_ == State::STOPPING || state_ == State::STOPPED); | 29 DCHECK(state_ == State::STOPPING || state_ == State::STOPPED); |
| 49 DCHECK(g_arc_bridge_service); | 30 DCHECK(g_arc_bridge_service); |
| 50 g_arc_bridge_service = nullptr; | 31 g_arc_bridge_service = nullptr; |
| 51 } | 32 } |
| 52 | 33 |
| 53 // static | 34 // static |
| 54 ArcBridgeService* ArcBridgeService::Get() { | 35 ArcBridgeService* ArcBridgeService::Get() { |
| 55 DCHECK(g_arc_bridge_service); | 36 DCHECK(g_arc_bridge_service); |
| 56 DCHECK(g_arc_bridge_service->origin_task_runner_->RunsTasksOnCurrentThread()); | 37 DCHECK(g_arc_bridge_service->origin_task_runner_->RunsTasksOnCurrentThread()); |
| 57 return g_arc_bridge_service; | 38 return g_arc_bridge_service; |
| 58 } | 39 } |
| 59 | 40 |
| 60 void ArcBridgeService::DetectAvailability() { | |
| 61 chromeos::SessionManagerClient* session_manager_client = | |
| 62 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
| 63 session_manager_client->CheckArcAvailability(base::Bind( | |
| 64 &ArcBridgeService::OnArcAvailable, weak_factory_.GetWeakPtr())); | |
| 65 } | |
| 66 | |
| 67 void ArcBridgeService::HandleStartup() { | |
| 68 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 69 session_started_ = true; | |
| 70 PrerequisitesChanged(); | |
| 71 } | |
| 72 | |
| 73 void ArcBridgeService::Shutdown() { | |
| 74 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 75 session_started_ = false; | |
| 76 PrerequisitesChanged(); | |
| 77 } | |
| 78 | |
| 79 void ArcBridgeService::AddObserver(Observer* observer) { | 41 void ArcBridgeService::AddObserver(Observer* observer) { |
| 80 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | 42 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); |
| 81 observer_list_.AddObserver(observer); | 43 observer_list_.AddObserver(observer); |
| 82 } | 44 } |
| 83 | 45 |
| 84 void ArcBridgeService::RemoveObserver(Observer* observer) { | 46 void ArcBridgeService::RemoveObserver(Observer* observer) { |
| 85 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | 47 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); |
| 86 observer_list_.RemoveObserver(observer); | 48 observer_list_.RemoveObserver(observer); |
| 87 } | 49 } |
| 88 | 50 |
| 89 void ArcBridgeService::PrerequisitesChanged() { | |
| 90 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 91 if (state_ == State::STOPPED) { | |
| 92 if (!available_ || !session_started_) | |
| 93 return; | |
| 94 SocketConnect(base::FilePath(kArcBridgeSocketPath)); | |
| 95 } else { | |
| 96 if (available_ && session_started_) | |
| 97 return; | |
| 98 StopInstance(); | |
| 99 } | |
| 100 } | |
| 101 | |
| 102 void ArcBridgeService::StopInstance() { | |
| 103 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 104 if (state_ == State::STOPPED || state_ == State::STOPPING) { | |
| 105 VLOG(1) << "StopInstance() called when ARC is not running"; | |
| 106 return; | |
| 107 } | |
| 108 if (state_ == State::CONNECTED || state_ == State::CONNECTING) { | |
| 109 // This was stopped before the D-Bus command to start the instance. Just | |
| 110 // close the socket (if it was opened). | |
| 111 if (state_ == State::CONNECTED) { | |
| 112 ipc_channel_.reset(); | |
| 113 } | |
| 114 SetState(State::STOPPED); | |
| 115 return; | |
| 116 } | |
| 117 | |
| 118 SetState(State::STOPPING); | |
| 119 chromeos::SessionManagerClient* session_manager_client = | |
| 120 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
| 121 session_manager_client->StopArcInstance(base::Bind( | |
| 122 &ArcBridgeService::OnInstanceStopped, weak_factory_.GetWeakPtr())); | |
| 123 } | |
| 124 | |
| 125 bool ArcBridgeService::RegisterInputDevice(const std::string& name, | |
| 126 const std::string& device_type, | |
| 127 base::ScopedFD fd) { | |
| 128 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 129 if (state_ != State::READY) { | |
| 130 LOG(ERROR) << "Called RegisterInputDevice when the service is not ready"; | |
| 131 return false; | |
| 132 } | |
| 133 return ipc_channel_->Send(new ArcInstanceMsg_RegisterInputDevice( | |
| 134 name, device_type, base::FileDescriptor(fd.Pass()))); | |
| 135 } | |
| 136 | |
| 137 void ArcBridgeService::SocketConnect(const base::FilePath& socket_path) { | |
| 138 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 139 if (state_ != State::STOPPED) { | |
| 140 VLOG(1) << "SocketConnect() called when instance is not stopped"; | |
| 141 return; | |
| 142 } | |
| 143 SetState(State::CONNECTING); | |
| 144 base::PostTaskAndReplyWithResult( | |
| 145 file_task_runner_.get(), FROM_HERE, | |
| 146 base::Bind(&base::CreateDirectory, socket_path.DirName()), | |
| 147 base::Bind(&ArcBridgeService::SocketConnectAfterEnsureParentDirectory, | |
| 148 weak_factory_.GetWeakPtr(), socket_path)); | |
| 149 } | |
| 150 | |
| 151 void ArcBridgeService::SocketConnectAfterEnsureParentDirectory( | |
| 152 const base::FilePath& socket_path, | |
| 153 bool directory_present) { | |
| 154 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 155 if (state_ != State::CONNECTING) { | |
| 156 VLOG(1) << "StopInstance() called while connecting"; | |
| 157 return; | |
| 158 } | |
| 159 if (!directory_present) { | |
| 160 LOG(ERROR) << "Error creating directory for " << socket_path.value(); | |
| 161 StopInstance(); | |
| 162 return; | |
| 163 } | |
| 164 | |
| 165 if (!Connect(IPC::ChannelHandle(socket_path.value()), | |
| 166 IPC::Channel::MODE_OPEN_NAMED_SERVER)) { | |
| 167 LOG(ERROR) << "Error connecting to " << socket_path.value(); | |
| 168 StopInstance(); | |
| 169 return; | |
| 170 } | |
| 171 | |
| 172 base::PostTaskAndReplyWithResult( | |
| 173 file_task_runner_.get(), FROM_HERE, | |
| 174 // TODO(lhchavez): Tighten the security around the socket by tying it to | |
| 175 // the user the instance will run as. | |
| 176 base::Bind(&base::SetPosixFilePermissions, socket_path, 0777), | |
| 177 base::Bind(&ArcBridgeService::SocketConnectAfterSetSocketPermissions, | |
| 178 weak_factory_.GetWeakPtr(), socket_path)); | |
| 179 } | |
| 180 | |
| 181 bool ArcBridgeService::Connect(const IPC::ChannelHandle& handle, | |
| 182 IPC::Channel::Mode mode) { | |
| 183 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 184 if (state_ != State::CONNECTING) { | |
| 185 VLOG(1) << "StopInstance() called while connecting"; | |
| 186 return false; | |
| 187 } | |
| 188 | |
| 189 ipc_channel_ = | |
| 190 IPC::ChannelProxy::Create(handle, mode, this, ipc_task_runner_.get()); | |
| 191 if (!ipc_channel_) | |
| 192 return false; | |
| 193 SetState(State::CONNECTED); | |
| 194 return true; | |
| 195 } | |
| 196 | |
| 197 void ArcBridgeService::SocketConnectAfterSetSocketPermissions( | |
| 198 const base::FilePath& socket_path, | |
| 199 bool socket_permissions_success) { | |
| 200 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 201 if (state_ != State::CONNECTED) { | |
| 202 VLOG(1) << "StopInstance() called while connecting"; | |
| 203 return; | |
| 204 } | |
| 205 | |
| 206 if (!socket_permissions_success) { | |
| 207 LOG(ERROR) << "Error setting socket permissions for " | |
| 208 << socket_path.value(); | |
| 209 StopInstance(); | |
| 210 return; | |
| 211 } | |
| 212 | |
| 213 SetState(State::STARTING); | |
| 214 chromeos::SessionManagerClient* session_manager_client = | |
| 215 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
| 216 session_manager_client->StartArcInstance( | |
| 217 socket_path.value(), base::Bind(&ArcBridgeService::OnInstanceStarted, | |
| 218 weak_factory_.GetWeakPtr())); | |
| 219 } | |
| 220 | |
| 221 void ArcBridgeService::OnInstanceStarted(bool success) { | |
| 222 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 223 if (state_ != State::STARTING) { | |
| 224 VLOG(1) << "StopInstance() called while connecting"; | |
| 225 return; | |
| 226 } | |
| 227 if (!success) { | |
| 228 LOG(ERROR) << "ARC instance unable to start. Shutting down the bridge"; | |
| 229 StopInstance(); | |
| 230 return; | |
| 231 } | |
| 232 } | |
| 233 | |
| 234 void ArcBridgeService::OnInstanceBootPhase(InstanceBootPhase phase) { | |
| 235 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 236 if (state_ != State::STARTING) { | |
| 237 VLOG(1) << "StopInstance() called while connecting"; | |
| 238 return; | |
| 239 } | |
| 240 if (phase == InstanceBootPhase::BRIDGE_READY) { | |
| 241 SetState(State::READY); | |
| 242 } | |
| 243 FOR_EACH_OBSERVER(Observer, observer_list_, OnInstanceBootPhase(phase)); | |
| 244 } | |
| 245 | |
| 246 void ArcBridgeService::SetState(State state) { | 51 void ArcBridgeService::SetState(State state) { |
| 247 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | 52 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); |
| 248 // DCHECK on enum classes not supported. | 53 // DCHECK on enum classes not supported. |
| 249 DCHECK(state_ != state); | 54 DCHECK(state_ != state); |
| 250 state_ = state; | 55 state_ = state; |
| 251 FOR_EACH_OBSERVER(Observer, observer_list_, OnStateChanged(state_)); | 56 FOR_EACH_OBSERVER(Observer, observer_list_, OnStateChanged(state_)); |
| 252 } | 57 } |
| 253 | 58 |
| 254 bool ArcBridgeService::OnMessageReceived(const IPC::Message& message) { | 59 void ArcBridgeService::SetAvailable(bool available) { |
| 255 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | 60 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); |
| 256 bool handled = true; | 61 // DCHECK on enum classes not supported. |
|
hidehiko
2015/11/25 07:47:37
The comment looks unrelated to the code?
khmel1
2015/11/25 15:28:31
Done.
| |
| 257 | 62 DCHECK(available_ != available); |
| 258 IPC_BEGIN_MESSAGE_MAP(ArcBridgeService, message) | |
| 259 IPC_MESSAGE_HANDLER(ArcInstanceHostMsg_InstanceBootPhase, | |
| 260 OnInstanceBootPhase) | |
| 261 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 262 IPC_END_MESSAGE_MAP() | |
| 263 | |
| 264 if (!handled) | |
| 265 LOG(ERROR) << "Invalid message with type = " << message.type(); | |
| 266 return handled; | |
| 267 } | |
| 268 | |
| 269 void ArcBridgeService::OnArcAvailable(bool available) { | |
| 270 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 271 if (available_ == available) | |
| 272 return; | |
| 273 available_ = available; | 63 available_ = available; |
| 274 FOR_EACH_OBSERVER(Observer, observer_list_, OnAvailableChanged(available_)); | 64 FOR_EACH_OBSERVER(Observer, observer_list_, OnAvailableChanged(available_)); |
| 275 PrerequisitesChanged(); | |
| 276 } | |
| 277 | |
| 278 void ArcBridgeService::OnInstanceStopped(bool success) { | |
| 279 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
| 280 // STOPPING is the only valid state for this function. | |
| 281 // DCHECK on enum classes not supported. | |
| 282 DCHECK(state_ == State::STOPPING); | |
| 283 ipc_channel_.reset(); | |
| 284 SetState(State::STOPPED); | |
| 285 } | 65 } |
| 286 | 66 |
| 287 } // namespace arc | 67 } // namespace arc |
| OLD | NEW |