OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
khmel1
2015/11/11 11:58:37
intial file is at 1412863004
| |
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 "components/arc/arc_bridge_service.h" | |
6 | |
7 #include "base/files/file_path.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" | |
13 #include "chromeos/arc/bridge/common/arc_host_messages.h" | |
14 #include "chromeos/arc/bridge/common/arc_instance_messages.h" | |
15 #include "chromeos/dbus/dbus_method_call_status.h" | |
16 #include "chromeos/dbus/dbus_thread_manager.h" | |
17 #include "chromeos/dbus/session_manager_client.h" | |
18 #include "components/arc/arc_pref_names.h" | |
19 #include "ipc/ipc_channel.h" | |
20 | |
21 namespace arc { | |
22 | |
23 namespace { | |
24 | |
25 const base::FilePath::CharType kArcBridgeSocketPath[] = | |
26 FILE_PATH_LITERAL("/home/chronos/ArcBridge/bridge.sock"); | |
27 | |
28 // Weak pointer. This class is owned by ChromeBrowserMainPartsChromeos. | |
29 ArcBridgeService* g_arc_bridge_service = NULL; | |
30 | |
31 } // namespace | |
32 | |
33 ArcBridgeService::ArcBridgeService( | |
34 const scoped_refptr<base::SingleThreadTaskRunner>& ipc_task_runner, | |
35 const scoped_refptr<base::SequencedTaskRunner>& file_task_runner) | |
36 : origin_task_runner_(base::ThreadTaskRunnerHandle::Get()), | |
37 ipc_task_runner_(ipc_task_runner), | |
38 file_task_runner_(file_task_runner), | |
39 available_(false), | |
40 enabled_(false), | |
41 state_(State::STOPPED), | |
42 weak_factory_(this) { | |
43 DCHECK(!g_arc_bridge_service); | |
44 g_arc_bridge_service = this; | |
45 } | |
46 | |
47 ArcBridgeService::~ArcBridgeService() { | |
48 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
49 Shutdown(); | |
50 DCHECK(g_arc_bridge_service); | |
51 g_arc_bridge_service = NULL; | |
52 } | |
53 | |
54 // static | |
55 ArcBridgeService* ArcBridgeService::Get() { | |
56 DCHECK(g_arc_bridge_service); | |
57 DCHECK(g_arc_bridge_service->origin_task_runner_->RunsTasksOnCurrentThread()); | |
58 return g_arc_bridge_service; | |
59 } | |
60 | |
61 void ArcBridgeService::DetectAvailability() { | |
62 chromeos::SessionManagerClient* session_manager_client = | |
63 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
64 session_manager_client->CheckArcAvailability(base::Bind( | |
65 &ArcBridgeService::OnArcAvailable, weak_factory_.GetWeakPtr())); | |
66 } | |
67 | |
68 void ArcBridgeService::HandleStartup() { | |
69 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
70 session_started_ = true; | |
71 PrerequisitesChanged(); | |
72 } | |
73 | |
74 void ArcBridgeService::Shutdown() { | |
75 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
76 session_started_ = false; | |
77 PrerequisitesChanged(); | |
78 } | |
79 | |
80 void ArcBridgeService::PrerequisitesChanged() { | |
81 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
82 if (state_ == State::STOPPED) { | |
83 if (!enabled_ || !available_ || !session_started_) | |
84 return; | |
85 SocketConnect(base::FilePath(kArcBridgeSocketPath)); | |
86 } else { | |
87 if (enabled_ && available_ && session_started_) | |
88 return; | |
89 StopInstance(); | |
90 } | |
91 } | |
92 | |
93 void ArcBridgeService::StopInstance() { | |
94 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
95 if (state_ == State::STOPPED || state_ == State::STOPPING) { | |
96 VLOG(1) << "StopInstance() called when ARC is not running"; | |
97 return; | |
98 } | |
99 if (state_ == State::CONNECTED || state_ == State::CONNECTING) { | |
100 // This was stopped before the D-Bus command to start the instance. Just | |
101 // close the socket (if it was opened). | |
102 if (state_ == State::CONNECTED) { | |
103 ipc_channel_->Close(); | |
104 ipc_channel_.reset(); | |
105 } | |
106 SetState(State::STOPPED); | |
107 return; | |
108 } | |
109 | |
110 SetState(State::STOPPING); | |
111 chromeos::SessionManagerClient* session_manager_client = | |
112 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
113 session_manager_client->StopArcInstance(base::Bind( | |
114 &ArcBridgeService::OnInstanceStopped, weak_factory_.GetWeakPtr())); | |
115 } | |
116 | |
117 void ArcBridgeService::AddObserver(Observer* observer) { | |
118 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
119 observer_list_.AddObserver(observer); | |
120 } | |
121 | |
122 void ArcBridgeService::RemoveObserver(Observer* observer) { | |
123 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
124 observer_list_.RemoveObserver(observer); | |
125 } | |
126 | |
127 void ArcBridgeService::SetEnabled(bool enabled) { | |
128 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
129 | |
130 if (enabled_ == enabled) | |
131 return; | |
132 enabled_ = enabled; | |
133 FOR_EACH_OBSERVER(Observer, observer_list_, OnEnabledChanged(enabled_)); | |
134 PrerequisitesChanged(); | |
135 } | |
136 | |
137 bool ArcBridgeService::RegisterInputDevice(const std::string& name, | |
138 const std::string& device_type, | |
139 base::ScopedFD fd) { | |
140 DCHECK(ipc_task_runner_->RunsTasksOnCurrentThread()); | |
141 if (state_ != State::READY) { | |
142 LOG(ERROR) << "Called RegisterInputDevice when the service is not ready"; | |
143 return false; | |
144 } | |
145 return ipc_channel_->Send(new ArcInstanceMsg_RegisterInputDevice( | |
146 name, device_type, base::FileDescriptor(fd.Pass()))); | |
147 } | |
148 | |
149 bool ArcBridgeService::RefreshApps() { | |
khmel1
2015/11/11 11:58:37
my change starts
| |
150 DCHECK(ipc_task_runner_->RunsTasksOnCurrentThread()); | |
151 if (state_ != State::READY) { | |
152 LOG(ERROR) << "Called RefreshApps when the service is not ready"; | |
153 return false; | |
154 } | |
155 return ipc_channel_->Send(new ArcInstanceMsg_RefreshApps()); | |
156 } | |
157 | |
158 bool ArcBridgeService::LaunchApp(const std::string& package, | |
159 const std::string& activity) { | |
160 DCHECK(ipc_task_runner_->RunsTasksOnCurrentThread()); | |
161 if (state_ != State::READY) { | |
162 LOG(ERROR) << "Called RefreshApps when the service is not ready"; | |
163 return false; | |
164 } | |
165 | |
166 return ipc_channel_->Send(new ArcInstanceMsg_LaunchApp(package, activity)); | |
167 } | |
168 | |
169 bool ArcBridgeService::RequestIcon(const std::string& package, | |
170 const std::string& activity, | |
171 int density) { | |
172 DCHECK(ipc_task_runner_->RunsTasksOnCurrentThread()); | |
173 if (state_ != State::READY) { | |
174 LOG(ERROR) << "Called RefreshApps when the service is not ready"; | |
175 return false; | |
176 } | |
177 | |
178 return ipc_channel_->Send(new ArcInstanceMsg_RequestIcon(package, | |
179 activity, | |
180 density)); | |
181 } | |
khmel1
2015/11/11 11:58:36
my change ends
| |
182 | |
183 void ArcBridgeService::SocketConnect(const base::FilePath& socket_path) { | |
184 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
185 if (state_ != State::STOPPED) { | |
186 VLOG(1) << "SocketConnect() called when instance is not stopped"; | |
187 return; | |
188 } | |
189 SetState(State::CONNECTING); | |
190 base::PostTaskAndReplyWithResult( | |
191 file_task_runner_.get(), FROM_HERE, | |
192 base::Bind(&base::CreateDirectory, socket_path.DirName()), | |
193 base::Bind(&ArcBridgeService::SocketConnectAfterEnsureParentDirectory, | |
194 weak_factory_.GetWeakPtr(), socket_path)); | |
195 } | |
196 | |
197 void ArcBridgeService::SocketConnectAfterEnsureParentDirectory( | |
198 const base::FilePath& socket_path, | |
199 bool directory_present) { | |
200 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
201 if (state_ != State::CONNECTING) { | |
202 VLOG(1) << "StopInstance() called while connecting"; | |
203 return; | |
204 } | |
205 if (!directory_present) { | |
206 LOG(ERROR) << "Error creating directory for " << socket_path.value(); | |
207 StopInstance(); | |
208 return; | |
209 } | |
210 | |
211 if (!Connect(IPC::ChannelHandle(socket_path.value()), | |
212 IPC::Channel::MODE_OPEN_NAMED_SERVER)) { | |
213 LOG(ERROR) << "Error connecting to " << socket_path.value(); | |
214 StopInstance(); | |
215 return; | |
216 } | |
217 | |
218 base::PostTaskAndReplyWithResult( | |
219 file_task_runner_.get(), FROM_HERE, | |
220 // TODO(lhchavez): Tighten the security around the socket by tying it to | |
221 // the user the instance will run as. | |
222 base::Bind(&base::SetPosixFilePermissions, socket_path, 0777), | |
223 base::Bind(&ArcBridgeService::SocketConnectAfterSetSocketPermissions, | |
224 weak_factory_.GetWeakPtr(), socket_path)); | |
225 } | |
226 | |
227 void ArcBridgeService::SocketConnectAfterSetSocketPermissions( | |
228 const base::FilePath& socket_path, | |
229 bool socket_permissions_success) { | |
230 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
231 if (state_ != State::CONNECTED) { | |
232 VLOG(1) << "StopInstance() called while connecting"; | |
233 return; | |
234 } | |
235 | |
236 if (!socket_permissions_success) { | |
237 LOG(ERROR) << "Error setting socket permissions for " | |
238 << socket_path.value(); | |
239 StopInstance(); | |
240 return; | |
241 } | |
242 | |
243 SetState(State::STARTING); | |
244 chromeos::SessionManagerClient* session_manager_client = | |
245 chromeos::DBusThreadManager::Get()->GetSessionManagerClient(); | |
246 session_manager_client->StartArcInstance( | |
247 socket_path.value(), base::Bind(&ArcBridgeService::OnInstanceStarted, | |
248 weak_factory_.GetWeakPtr())); | |
249 } | |
250 | |
251 bool ArcBridgeService::Connect(const IPC::ChannelHandle& handle, | |
252 IPC::Channel::Mode mode) { | |
253 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
254 if (state_ != State::CONNECTING) { | |
255 VLOG(1) << "StopInstance() called while connecting"; | |
256 return false; | |
257 } | |
258 | |
259 ipc_channel_ = | |
260 IPC::ChannelProxy::Create(handle, mode, this, ipc_task_runner_.get()); | |
261 if (!ipc_channel_) | |
262 return false; | |
263 SetState(State::CONNECTED); | |
264 return true; | |
265 } | |
266 | |
267 void ArcBridgeService::OnInstanceReady() { | |
268 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
269 if (state_ != State::STARTING) { | |
270 VLOG(1) << "StopInstance() called while connecting"; | |
271 return; | |
272 } | |
273 SetState(State::READY); | |
274 } | |
275 | |
276 void ArcBridgeService::OnAppsRefreshed( | |
khmel1
2015/11/11 11:58:37
my change starts
| |
277 const std::vector<std::string>& names, | |
278 const std::vector<std::string>& packages) { | |
279 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
280 FOR_EACH_OBSERVER(Observer, observer_list_, OnAppsRefreshed(names, packages)); | |
281 } | |
282 | |
283 void ArcBridgeService::OnAppIcon(const std::string& package, | |
284 const std::string& activity, | |
285 int density, | |
286 const std::vector<uint8_t>& icon_png_data) { | |
287 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
288 FOR_EACH_OBSERVER(Observer, observer_list_, OnAppIcon(package, | |
289 activity, | |
290 density, | |
291 icon_png_data)); | |
292 } | |
khmel1
2015/11/11 11:58:37
my change ends
| |
293 | |
294 void ArcBridgeService::SetState(State state) { | |
295 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
296 DCHECK(state_ != state); | |
297 state_ = state; | |
298 FOR_EACH_OBSERVER(Observer, observer_list_, OnStateChanged(state_)); | |
299 } | |
300 | |
301 bool ArcBridgeService::OnMessageReceived(const IPC::Message& message) { | |
302 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
303 bool handled = true; | |
304 | |
305 IPC_BEGIN_MESSAGE_MAP(ArcBridgeService, message) | |
306 IPC_MESSAGE_HANDLER(ArcInstanceHostMsg_InstanceReady, OnInstanceReady) | |
307 IPC_MESSAGE_HANDLER(ArcInstanceHostMsg_AppsRefreshed, OnAppsRefreshed) | |
khmel1
2015/11/11 11:58:37
my change is 2 lines
| |
308 IPC_MESSAGE_HANDLER(ArcInstanceHostMsg_AppIcon, OnAppIcon) | |
309 | |
310 IPC_MESSAGE_UNHANDLED(handled = false) | |
311 IPC_END_MESSAGE_MAP() | |
312 | |
313 if (!handled) | |
314 LOG(ERROR) << "Invalid message with type = " << message.type(); | |
315 return handled; | |
316 } | |
317 | |
318 void ArcBridgeService::OnArcAvailable(bool available) { | |
319 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
320 if (available_ == available) | |
321 return; | |
322 available_ = available; | |
323 FOR_EACH_OBSERVER(Observer, observer_list_, OnAvailableChanged(available_)); | |
324 PrerequisitesChanged(); | |
325 } | |
326 | |
327 void ArcBridgeService::OnInstanceStarted(bool success) { | |
328 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
329 if (state_ != State::STARTING) { | |
330 VLOG(1) << "StopInstance() called while connecting"; | |
331 return; | |
332 } | |
333 if (!success) { | |
334 LOG(ERROR) << "ARC instance unable to start. Shutting down the bridge"; | |
335 StopInstance(); | |
336 return; | |
337 } | |
338 } | |
339 | |
340 void ArcBridgeService::OnInstanceStopped(bool success) { | |
341 DCHECK(origin_task_runner_->RunsTasksOnCurrentThread()); | |
342 // STOPPING is the only valid state for this function. | |
343 DCHECK(state_ == State::STOPPING); | |
344 ipc_channel_->Close(); | |
345 ipc_channel_.reset(); | |
346 SetState(State::STOPPED); | |
347 } | |
348 | |
349 } // namespace arc | |
OLD | NEW |