Vitaly Buka | cad20f0 | 2015-10-16 17:27:15 -0700 | [diff] [blame] | 1 | // Copyright 2015 The Android Open Source Project |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 14 | |
| 15 | #include <string> |
| 16 | |
Alex Vakulenko | bc64849 | 2015-09-08 15:39:18 -0700 | [diff] [blame] | 17 | #include <signal.h> |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 18 | #include <sysexits.h> |
Alex Vakulenko | bc64849 | 2015-09-08 15:39:18 -0700 | [diff] [blame] | 19 | |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 20 | #include <base/files/file_path.h> |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 21 | #include <binderwrapper/binder_wrapper.h> |
| 22 | #include <brillo/binder_watcher.h> |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 23 | #include <brillo/daemons/dbus_daemon.h> |
| 24 | #include <brillo/dbus/async_event_sequencer.h> |
| 25 | #include <brillo/dbus/exported_object_manager.h> |
| 26 | #include <brillo/flag_helper.h> |
| 27 | #include <brillo/strings/string_utils.h> |
| 28 | #include <brillo/syslog_logging.h> |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 29 | |
Vitaly Buka | becd461 | 2015-08-16 23:31:55 -0700 | [diff] [blame] | 30 | #include "buffet/buffet_config.h" |
Alex Vakulenko | 9485896 | 2014-12-01 17:53:27 -0800 | [diff] [blame] | 31 | #include "buffet/dbus_constants.h" |
Christopher Wiley | 4b5f04c | 2014-03-27 14:45:37 -0700 | [diff] [blame] | 32 | #include "buffet/manager.h" |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 33 | #include "common/binder_constants.h" |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 34 | |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 35 | using brillo::dbus_utils::AsyncEventSequencer; |
| 36 | using brillo::DBusServiceDaemon; |
Robert Ginda | cf92c66 | 2015-08-20 09:30:11 -0700 | [diff] [blame] | 37 | using buffet::dbus_constants::kServiceName; |
| 38 | using buffet::dbus_constants::kRootServicePath; |
Christopher Wiley | 54028f9 | 2014-04-01 17:33:29 -0700 | [diff] [blame] | 39 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 40 | namespace buffet { |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 41 | |
Alex Vakulenko | 1e3a66b | 2015-05-22 15:48:53 -0700 | [diff] [blame] | 42 | class Daemon final : public DBusServiceDaemon { |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 43 | public: |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 44 | explicit Daemon(const Manager::Options& options) |
| 45 | : DBusServiceDaemon(kServiceName, kRootServicePath), options_{options} {} |
Chris Sosa | ea6456d | 2014-04-09 15:42:01 -0700 | [diff] [blame] | 46 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 47 | protected: |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 48 | int OnInit() override { |
| 49 | android::BinderWrapper::Create(); |
| 50 | if (!binder_watcher_.Init()) |
| 51 | return EX_OSERR; |
| 52 | |
| 53 | return brillo::DBusServiceDaemon::OnInit(); |
| 54 | } |
| 55 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 56 | void RegisterDBusObjectsAsync(AsyncEventSequencer* sequencer) override { |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 57 | manager_ = new Manager{options_, bus_}; |
| 58 | android::BinderWrapper::Get()->RegisterService( |
| 59 | weaved::binder::kWeaveServiceName, |
| 60 | android::IInterface::asBinder(manager_)); |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 61 | manager_->Start(sequencer); |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 62 | } |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 63 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 64 | void OnShutdown(int* return_code) override { manager_->Stop(); } |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 65 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 66 | private: |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 67 | Manager::Options options_; |
Alex Vakulenko | ae29f7d | 2015-12-21 16:30:37 -0800 | [diff] [blame^] | 68 | brillo::BinderWatcher binder_watcher_; |
| 69 | android::sp<buffet::Manager> manager_; |
Vitaly Buka | 58a288b | 2015-07-31 00:33:31 -0700 | [diff] [blame] | 70 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 71 | DISALLOW_COPY_AND_ASSIGN(Daemon); |
| 72 | }; |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 73 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 74 | } // namespace buffet |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 75 | |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 76 | namespace { |
| 77 | |
Alex Vakulenko | 8e34636 | 2015-08-21 10:01:57 -0700 | [diff] [blame] | 78 | const char kDefaultConfigFilePath[] = "/etc/weaved/weaved.conf"; |
Alex Vakulenko | d42e09d | 2015-08-21 18:23:32 -0700 | [diff] [blame] | 79 | const char kDefaultStateFilePath[] = "/data/misc/weaved/device_reg_info"; |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 80 | |
| 81 | } // namespace |
| 82 | |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 83 | int main(int argc, char* argv[]) { |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 84 | DEFINE_bool(log_to_stderr, false, "log trace messages to stderr as well"); |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 85 | DEFINE_string(config_path, kDefaultConfigFilePath, |
| 86 | "Path to file containing config information."); |
| 87 | DEFINE_string(state_path, kDefaultStateFilePath, |
| 88 | "Path to file containing state information."); |
Christopher Wiley | 1d35ecc | 2015-04-07 11:11:18 -0700 | [diff] [blame] | 89 | DEFINE_bool(enable_xmpp, true, |
| 90 | "Connect to GCD via a persistent XMPP connection."); |
Alex Vakulenko | 49c36ee | 2015-08-19 11:06:04 -0700 | [diff] [blame] | 91 | DEFINE_bool(disable_privet, false, "disable Privet protocol"); |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 92 | DEFINE_bool(enable_ping, false, "enable test HTTP handler at /privet/ping"); |
| 93 | DEFINE_string(device_whitelist, "", |
| 94 | "Comma separated list of network interfaces to monitor for " |
| 95 | "connectivity (an empty list enables all interfaces)."); |
Vitaly Buka | 8b511e9 | 2015-07-18 20:43:58 -0700 | [diff] [blame] | 96 | |
| 97 | DEFINE_bool(disable_security, false, |
| 98 | "disable Privet security for tests. For test only."); |
| 99 | DEFINE_string(test_privet_ssid, "", |
| 100 | "Fixed SSID for WiFi bootstrapping. For test only."); |
| 101 | DEFINE_string(test_definitions_path, "", |
| 102 | "Path to directory containing additional command " |
| 103 | "and state definitions. For test only."); |
| 104 | |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 105 | brillo::FlagHelper::Init(argc, argv, "Privet protocol handler daemon"); |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 106 | if (FLAGS_config_path.empty()) |
| 107 | FLAGS_config_path = kDefaultConfigFilePath; |
| 108 | if (FLAGS_state_path.empty()) |
| 109 | FLAGS_state_path = kDefaultStateFilePath; |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 110 | int flags = brillo::kLogToSyslog | brillo::kLogHeader; |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 111 | if (FLAGS_log_to_stderr) |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 112 | flags |= brillo::kLogToStderr; |
| 113 | brillo::InitLog(flags); |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 114 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 115 | auto device_whitelist = |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 116 | brillo::string_utils::Split(FLAGS_device_whitelist, ",", true, true); |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 117 | |
Alex Vakulenko | bc64849 | 2015-09-08 15:39:18 -0700 | [diff] [blame] | 118 | // We are handling write errors on closed sockets correctly and not relying on |
| 119 | // (nor handling) SIGPIPE signal, which just kills the process. |
| 120 | // Mark it to be ignored. |
| 121 | signal(SIGPIPE, SIG_IGN); |
| 122 | |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 123 | buffet::Manager::Options options; |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 124 | options.xmpp_enabled = FLAGS_enable_xmpp; |
Vitaly Buka | 0c6dcd2 | 2015-07-10 00:12:25 -0700 | [diff] [blame] | 125 | options.disable_privet = FLAGS_disable_privet; |
Vitaly Buka | 0c6dcd2 | 2015-07-10 00:12:25 -0700 | [diff] [blame] | 126 | options.enable_ping = FLAGS_enable_ping; |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 127 | options.device_whitelist = {device_whitelist.begin(), device_whitelist.end()}; |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 128 | |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 129 | options.config_options.defaults = base::FilePath{FLAGS_config_path}; |
| 130 | options.config_options.settings = base::FilePath{FLAGS_state_path}; |
| 131 | options.config_options.definitions = base::FilePath{"/etc/weaved"}; |
| 132 | options.config_options.test_definitions = |
| 133 | base::FilePath{FLAGS_test_definitions_path}; |
| 134 | options.config_options.disable_security = FLAGS_disable_security; |
| 135 | options.config_options.test_privet_ssid = FLAGS_test_privet_ssid; |
| 136 | |
| 137 | buffet::Daemon daemon{options}; |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 138 | return daemon.Run(); |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 139 | } |