1//
2// Copyright (C) 2015 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8//      http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
16
17#include "update_engine/daemon.h"
18
19#include <sysexits.h>
20
21#include <base/bind.h>
22#include <base/location.h>
23#include <base/time/time.h>
24#if USE_WEAVE || USE_BINDER
25#include <binderwrapper/binder_wrapper.h>
26#endif  // USE_WEAVE || USE_BINDER
27
28#if defined(__BRILLO__) || defined(__CHROMEOS__)
29#include "update_engine/real_system_state.h"
30#else  // !(defined(__BRILLO__) || defined(__CHROMEOS__))
31#include "update_engine/daemon_state_android.h"
32#endif  // defined(__BRILLO__) || defined(__CHROMEOS__)
33
34#if USE_DBUS
35namespace {
36const int kDBusSystemMaxWaitSeconds = 2 * 60;
37}  // namespace
38#endif  // USE_DBUS
39
40namespace chromeos_update_engine {
41
42int UpdateEngineDaemon::OnInit() {
43  // Register the |subprocess_| singleton with this Daemon as the signal
44  // handler.
45  subprocess_.Init(this);
46
47  int exit_code = Daemon::OnInit();
48  if (exit_code != EX_OK)
49    return exit_code;
50
51#if USE_WEAVE || USE_BINDER
52  android::BinderWrapper::Create();
53  binder_watcher_.Init();
54#endif  // USE_WEAVE || USE_BINDER
55
56#if USE_DBUS
57  // We wait for the D-Bus connection for up two minutes to avoid re-spawning
58  // the daemon too fast causing thrashing if dbus-daemon is not running.
59  scoped_refptr<dbus::Bus> bus = dbus_connection_.ConnectWithTimeout(
60      base::TimeDelta::FromSeconds(kDBusSystemMaxWaitSeconds));
61
62  if (!bus) {
63    // TODO(deymo): Make it possible to run update_engine even if dbus-daemon
64    // is not running or constantly crashing.
65    LOG(ERROR) << "Failed to initialize DBus, aborting.";
66    return 1;
67  }
68
69  CHECK(bus->SetUpAsyncOperations());
70#endif  // USE_DBUS
71
72#if defined(__BRILLO__) || defined(__CHROMEOS__)
73  // Initialize update engine global state but continue if something fails.
74  // TODO(deymo): Move the daemon_state_ initialization to a factory method
75  // avoiding the explicit re-usage of the |bus| instance, shared between
76  // D-Bus service and D-Bus client calls.
77  RealSystemState* real_system_state = new RealSystemState(bus);
78  daemon_state_.reset(real_system_state);
79  LOG_IF(ERROR, !real_system_state->Initialize())
80      << "Failed to initialize system state.";
81#else  // !(defined(__BRILLO__) || defined(__CHROMEOS__))
82  DaemonStateAndroid* daemon_state_android = new DaemonStateAndroid();
83  daemon_state_.reset(daemon_state_android);
84  LOG_IF(ERROR, !daemon_state_android->Initialize())
85      << "Failed to initialize system state.";
86#endif  // defined(__BRILLO__) || defined(__CHROMEOS__)
87
88#if USE_BINDER
89  // Create the Binder Service.
90#if defined(__BRILLO__) || defined(__CHROMEOS__)
91  binder_service_ = new BinderUpdateEngineBrilloService{real_system_state};
92#else  // !(defined(__BRILLO__) || defined(__CHROMEOS__))
93  binder_service_ = new BinderUpdateEngineAndroidService{
94      daemon_state_android->service_delegate()};
95#endif  // defined(__BRILLO__) || defined(__CHROMEOS__)
96  auto binder_wrapper = android::BinderWrapper::Get();
97  if (!binder_wrapper->RegisterService(binder_service_->ServiceName(),
98                                       binder_service_)) {
99    LOG(ERROR) << "Failed to register binder service.";
100  }
101
102  daemon_state_->AddObserver(binder_service_.get());
103#endif  // USE_BINDER
104
105#if USE_DBUS
106  // Create the DBus service.
107  dbus_adaptor_.reset(new UpdateEngineAdaptor(real_system_state, bus));
108  daemon_state_->AddObserver(dbus_adaptor_.get());
109
110  dbus_adaptor_->RegisterAsync(base::Bind(&UpdateEngineDaemon::OnDBusRegistered,
111                                          base::Unretained(this)));
112  LOG(INFO) << "Waiting for DBus object to be registered.";
113#else  // !USE_DBUS
114  daemon_state_->StartUpdater();
115#endif  // USE_DBUS
116  return EX_OK;
117}
118
119#if USE_DBUS
120void UpdateEngineDaemon::OnDBusRegistered(bool succeeded) {
121  if (!succeeded) {
122    LOG(ERROR) << "Registering the UpdateEngineAdaptor";
123    QuitWithExitCode(1);
124    return;
125  }
126
127  // Take ownership of the service now that everything is initialized. We need
128  // to this now and not before to avoid exposing a well known DBus service
129  // path that doesn't have the service it is supposed to implement.
130  if (!dbus_adaptor_->RequestOwnership()) {
131    LOG(ERROR) << "Unable to take ownership of the DBus service, is there "
132               << "other update_engine daemon running?";
133    QuitWithExitCode(1);
134    return;
135  }
136  daemon_state_->StartUpdater();
137}
138#endif  // USE_DBUS
139
140}  // namespace chromeos_update_engine
141