| Alex Deymo | aea4c1c | 2015-08-19 20:24:43 -0700 | [diff] [blame] | 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 | // | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 16 |  | 
|  | 17 | #ifndef UPDATE_ENGINE_DAEMON_H_ | 
|  | 18 | #define UPDATE_ENGINE_DAEMON_H_ | 
|  | 19 |  | 
|  | 20 | #include <memory> | 
|  | 21 | #include <string> | 
|  | 22 |  | 
| Alex Deymo | dc066f1 | 2017-03-27 13:01:29 -0700 | [diff] [blame] | 23 | #if USE_BINDER | 
| Alex Vakulenko | e119e6a | 2016-01-06 17:13:11 -0800 | [diff] [blame] | 24 | #include <brillo/binder_watcher.h> | 
| Alex Deymo | dc066f1 | 2017-03-27 13:01:29 -0700 | [diff] [blame] | 25 | #endif  // USE_BINDER | 
| Alex Deymo | a91cc48 | 2016-01-20 16:51:56 -0800 | [diff] [blame] | 26 | #include <brillo/daemons/daemon.h> | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 27 |  | 
| Casey Dahlin | a93cd53 | 2016-01-14 16:55:11 -0800 | [diff] [blame] | 28 | #if USE_BINDER | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 29 | #if USE_OMAHA | 
| Alex Deymo | fa78f14 | 2016-01-26 21:36:16 -0800 | [diff] [blame] | 30 | #include "update_engine/binder_service_brillo.h" | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 31 | #else  // !USE_OMAHA | 
| Alex Deymo | e97b39c | 2016-01-20 13:22:17 -0800 | [diff] [blame] | 32 | #include "update_engine/binder_service_android.h" | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 33 | #endif  // USE_OMAHA | 
| Casey Dahlin | a93cd53 | 2016-01-14 16:55:11 -0800 | [diff] [blame] | 34 | #endif  // USE_BINDER | 
| Alex Deymo | 39910dc | 2015-11-09 17:04:30 -0800 | [diff] [blame] | 35 | #include "update_engine/common/subprocess.h" | 
| Alex Deymo | fa78f14 | 2016-01-26 21:36:16 -0800 | [diff] [blame] | 36 | #include "update_engine/daemon_state_interface.h" | 
| Alex Deymo | e97b39c | 2016-01-20 13:22:17 -0800 | [diff] [blame] | 37 | #if USE_DBUS | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 38 | #include "update_engine/dbus_service.h" | 
| Alex Deymo | e97b39c | 2016-01-20 13:22:17 -0800 | [diff] [blame] | 39 | #endif  // USE_DBUS | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 40 |  | 
|  | 41 | namespace chromeos_update_engine { | 
|  | 42 |  | 
| Alex Deymo | a91cc48 | 2016-01-20 16:51:56 -0800 | [diff] [blame] | 43 | class UpdateEngineDaemon : public brillo::Daemon { | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 44 | public: | 
|  | 45 | UpdateEngineDaemon() = default; | 
|  | 46 |  | 
|  | 47 | protected: | 
|  | 48 | int OnInit() override; | 
|  | 49 |  | 
|  | 50 | private: | 
| Alex Deymo | e97b39c | 2016-01-20 13:22:17 -0800 | [diff] [blame] | 51 | #if USE_DBUS | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 52 | // Run from the main loop when the |dbus_adaptor_| object is registered. At | 
|  | 53 | // this point we can request ownership of the DBus service name and continue | 
|  | 54 | // initialization. | 
|  | 55 | void OnDBusRegistered(bool succeeded); | 
|  | 56 |  | 
| Sen Jiang | 299128e | 2016-06-03 17:48:18 -0700 | [diff] [blame] | 57 | // Main D-Bus service adaptor. | 
| Alex Deymo | a91cc48 | 2016-01-20 16:51:56 -0800 | [diff] [blame] | 58 | std::unique_ptr<UpdateEngineAdaptor> dbus_adaptor_; | 
| Alex Deymo | e97b39c | 2016-01-20 13:22:17 -0800 | [diff] [blame] | 59 | #endif  // USE_DBUS | 
| Alex Deymo | a91cc48 | 2016-01-20 16:51:56 -0800 | [diff] [blame] | 60 |  | 
| Alex Vakulenko | 3f39d5c | 2015-10-13 09:27:13 -0700 | [diff] [blame] | 61 | // The Subprocess singleton class requires a brillo::MessageLoop in the | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 62 | // current thread, so we need to initialize it from this class instead of | 
|  | 63 | // the main() function. | 
|  | 64 | Subprocess subprocess_; | 
|  | 65 |  | 
| Alex Deymo | dc066f1 | 2017-03-27 13:01:29 -0700 | [diff] [blame] | 66 | #if USE_BINDER | 
| Alex Vakulenko | e119e6a | 2016-01-06 17:13:11 -0800 | [diff] [blame] | 67 | brillo::BinderWatcher binder_watcher_; | 
| Alex Deymo | dc066f1 | 2017-03-27 13:01:29 -0700 | [diff] [blame] | 68 | #endif  // USE_BINDER | 
| Casey Dahlin | a93cd53 | 2016-01-14 16:55:11 -0800 | [diff] [blame] | 69 |  | 
|  | 70 | #if USE_BINDER | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 71 | #if USE_OMAHA | 
| Alex Deymo | fa78f14 | 2016-01-26 21:36:16 -0800 | [diff] [blame] | 72 | android::sp<BinderUpdateEngineBrilloService> binder_service_; | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 73 | #else  // !USE_OMAHA | 
| Alex Deymo | fa78f14 | 2016-01-26 21:36:16 -0800 | [diff] [blame] | 74 | android::sp<BinderUpdateEngineAndroidService> binder_service_; | 
| Sen Jiang | 192a86a | 2016-05-19 17:21:24 -0700 | [diff] [blame] | 75 | #endif  // USE_OMAHA | 
| Casey Dahlin | a93cd53 | 2016-01-14 16:55:11 -0800 | [diff] [blame] | 76 | #endif  // USE_BINDER | 
| Alex Vakulenko | e119e6a | 2016-01-06 17:13:11 -0800 | [diff] [blame] | 77 |  | 
| Alex Deymo | fa78f14 | 2016-01-26 21:36:16 -0800 | [diff] [blame] | 78 | // The daemon state with all the required daemon classes for the configured | 
|  | 79 | // platform. | 
|  | 80 | std::unique_ptr<DaemonStateInterface> daemon_state_; | 
| Alex Deymo | 2b4268c | 2015-12-04 13:56:25 -0800 | [diff] [blame] | 81 |  | 
| Alex Deymo | b7ca096 | 2014-10-01 17:58:07 -0700 | [diff] [blame] | 82 | DISALLOW_COPY_AND_ASSIGN(UpdateEngineDaemon); | 
|  | 83 | }; | 
|  | 84 |  | 
|  | 85 | }  // namespace chromeos_update_engine | 
|  | 86 |  | 
|  | 87 | #endif  // UPDATE_ENGINE_DAEMON_H_ |