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#ifndef UPDATE_ENGINE_CLIENT_LIBRARY_CLIENT_DBUS_H_
18#define UPDATE_ENGINE_CLIENT_LIBRARY_CLIENT_DBUS_H_
19
20#include <cstdint>
21#include <memory>
22#include <string>
23#include <vector>
24
25#include <base/macros.h>
26
27#include "update_engine/client_library/include/update_engine/client.h"
28#include "update_engine/dbus-proxies.h"
29
30namespace update_engine {
31namespace internal {
32
33class DBusUpdateEngineClient : public UpdateEngineClient {
34 public:
35  DBusUpdateEngineClient() = default;
36  bool Init();
37
38  virtual ~DBusUpdateEngineClient() = default;
39
40  bool AttemptUpdate(const std::string& app_version,
41                     const std::string& omaha_url,
42                     bool at_user_request) override;
43
44  bool GetStatus(int64_t* out_last_checked_time,
45                 double* out_progress,
46                 UpdateStatus* out_update_status,
47                 std::string* out_new_version,
48                 int64_t* out_new_size) const override;
49
50  bool SetCohortHint(const std::string& cohort_hint) override;
51  bool GetCohortHint(std::string* cohort_hint) const override;
52
53  bool SetUpdateOverCellularPermission(bool allowed) override;
54  bool GetUpdateOverCellularPermission(bool* allowed) const override;
55
56  bool SetP2PUpdatePermission(bool enabled) override;
57  bool GetP2PUpdatePermission(bool* enabled) const override;
58
59  bool Rollback(bool powerwash) override;
60
61  bool GetRollbackPartition(std::string* rollback_partition) const override;
62
63  void RebootIfNeeded() override;
64
65  bool GetPrevVersion(std::string* prev_version) const override;
66
67  bool ResetStatus() override;
68
69  bool SetTargetChannel(const std::string& target_channel,
70                        bool allow_powerwash) override;
71
72  bool GetTargetChannel(std::string* out_channel) const override;
73
74  bool GetChannel(std::string* out_channel) const override;
75
76  bool RegisterStatusUpdateHandler(StatusUpdateHandler* handler) override;
77  bool UnregisterStatusUpdateHandler(StatusUpdateHandler* handler) override;
78
79  bool GetLastAttemptError(int32_t* last_attempt_error) const override;
80
81  bool GetEolStatus(int32_t* eol_status) const override;
82
83 private:
84  void DBusStatusHandlersRegistered(const std::string& interface,
85                                    const std::string& signal_name,
86                                    bool success) const;
87
88  // Send an initial event to new StatusUpdateHandlers. If the handler argument
89  // is not nullptr, only that handler receives the event. Otherwise all
90  // registered handlers receive the event.
91  void StatusUpdateHandlersRegistered(StatusUpdateHandler* handler) const;
92
93  void RunStatusUpdateHandlers(int64_t last_checked_time,
94                               double progress,
95                               const std::string& current_operation,
96                               const std::string& new_version,
97                               int64_t new_size);
98
99  std::unique_ptr<org::chromium::UpdateEngineInterfaceProxy> proxy_;
100  std::vector<update_engine::StatusUpdateHandler*> handlers_;
101  bool dbus_handler_registered_{false};
102
103  DISALLOW_COPY_AND_ASSIGN(DBusUpdateEngineClient);
104};  // class DBusUpdateEngineClient
105
106}  // namespace internal
107}  // namespace update_engine
108
109#endif  // UPDATE_ENGINE_CLIENT_LIBRARY_CLIENT_DBUS_H_
110