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> |
| 18 | |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 19 | #include <base/files/file_path.h> |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 20 | #include <brillo/daemons/dbus_daemon.h> |
| 21 | #include <brillo/dbus/async_event_sequencer.h> |
| 22 | #include <brillo/dbus/exported_object_manager.h> |
| 23 | #include <brillo/flag_helper.h> |
| 24 | #include <brillo/strings/string_utils.h> |
| 25 | #include <brillo/syslog_logging.h> |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 26 | |
Vitaly Buka | becd461 | 2015-08-16 23:31:55 -0700 | [diff] [blame] | 27 | #include "buffet/buffet_config.h" |
Alex Vakulenko | 9485896 | 2014-12-01 17:53:27 -0800 | [diff] [blame] | 28 | #include "buffet/dbus_constants.h" |
Christopher Wiley | 4b5f04c | 2014-03-27 14:45:37 -0700 | [diff] [blame] | 29 | #include "buffet/manager.h" |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 30 | |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 31 | using brillo::dbus_utils::AsyncEventSequencer; |
| 32 | using brillo::DBusServiceDaemon; |
Robert Ginda | cf92c66 | 2015-08-20 09:30:11 -0700 | [diff] [blame] | 33 | using buffet::dbus_constants::kServiceName; |
| 34 | using buffet::dbus_constants::kRootServicePath; |
Christopher Wiley | 54028f9 | 2014-04-01 17:33:29 -0700 | [diff] [blame] | 35 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 36 | namespace buffet { |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 37 | |
Alex Vakulenko | 1e3a66b | 2015-05-22 15:48:53 -0700 | [diff] [blame] | 38 | class Daemon final : public DBusServiceDaemon { |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 39 | public: |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 40 | explicit Daemon(const Manager::Options& options) |
| 41 | : DBusServiceDaemon(kServiceName, kRootServicePath), options_{options} {} |
Chris Sosa | ea6456d | 2014-04-09 15:42:01 -0700 | [diff] [blame] | 42 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 43 | protected: |
| 44 | void RegisterDBusObjectsAsync(AsyncEventSequencer* sequencer) override { |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 45 | manager_.reset(new Manager(options_, object_manager_->AsWeakPtr())); |
| 46 | manager_->Start(sequencer); |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 47 | } |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 48 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 49 | void OnShutdown(int* return_code) override { manager_->Stop(); } |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 50 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 51 | private: |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 52 | Manager::Options options_; |
Vitaly Buka | 58a288b | 2015-07-31 00:33:31 -0700 | [diff] [blame] | 53 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 54 | std::unique_ptr<buffet::Manager> manager_; |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 55 | DISALLOW_COPY_AND_ASSIGN(Daemon); |
| 56 | }; |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 57 | |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 58 | } // namespace buffet |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 59 | |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 60 | namespace { |
| 61 | |
Alex Vakulenko | 8e34636 | 2015-08-21 10:01:57 -0700 | [diff] [blame] | 62 | const char kDefaultConfigFilePath[] = "/etc/weaved/weaved.conf"; |
Alex Vakulenko | d42e09d | 2015-08-21 18:23:32 -0700 | [diff] [blame] | 63 | const char kDefaultStateFilePath[] = "/data/misc/weaved/device_reg_info"; |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 64 | |
| 65 | } // namespace |
| 66 | |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 67 | int main(int argc, char* argv[]) { |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 68 | 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] | 69 | DEFINE_string(config_path, kDefaultConfigFilePath, |
| 70 | "Path to file containing config information."); |
| 71 | DEFINE_string(state_path, kDefaultStateFilePath, |
| 72 | "Path to file containing state information."); |
Christopher Wiley | 1d35ecc | 2015-04-07 11:11:18 -0700 | [diff] [blame] | 73 | DEFINE_bool(enable_xmpp, true, |
| 74 | "Connect to GCD via a persistent XMPP connection."); |
Alex Vakulenko | 49c36ee | 2015-08-19 11:06:04 -0700 | [diff] [blame] | 75 | DEFINE_bool(disable_privet, false, "disable Privet protocol"); |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 76 | DEFINE_bool(enable_ping, false, "enable test HTTP handler at /privet/ping"); |
| 77 | DEFINE_string(device_whitelist, "", |
| 78 | "Comma separated list of network interfaces to monitor for " |
| 79 | "connectivity (an empty list enables all interfaces)."); |
Vitaly Buka | 8b511e9 | 2015-07-18 20:43:58 -0700 | [diff] [blame] | 80 | |
| 81 | DEFINE_bool(disable_security, false, |
| 82 | "disable Privet security for tests. For test only."); |
| 83 | DEFINE_string(test_privet_ssid, "", |
| 84 | "Fixed SSID for WiFi bootstrapping. For test only."); |
| 85 | DEFINE_string(test_definitions_path, "", |
| 86 | "Path to directory containing additional command " |
| 87 | "and state definitions. For test only."); |
| 88 | |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 89 | brillo::FlagHelper::Init(argc, argv, "Privet protocol handler daemon"); |
Christopher Wiley | 357deca | 2015-02-07 18:29:32 -0800 | [diff] [blame] | 90 | if (FLAGS_config_path.empty()) |
| 91 | FLAGS_config_path = kDefaultConfigFilePath; |
| 92 | if (FLAGS_state_path.empty()) |
| 93 | FLAGS_state_path = kDefaultStateFilePath; |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 94 | int flags = brillo::kLogToSyslog | brillo::kLogHeader; |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 95 | if (FLAGS_log_to_stderr) |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 96 | flags |= brillo::kLogToStderr; |
| 97 | brillo::InitLog(flags); |
Vitaly Buka | b12e995 | 2015-03-13 23:54:21 -0700 | [diff] [blame] | 98 | |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 99 | auto device_whitelist = |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 100 | brillo::string_utils::Split(FLAGS_device_whitelist, ",", true, true); |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 101 | |
Alex Vakulenko | bc64849 | 2015-09-08 15:39:18 -0700 | [diff] [blame] | 102 | // We are handling write errors on closed sockets correctly and not relying on |
| 103 | // (nor handling) SIGPIPE signal, which just kills the process. |
| 104 | // Mark it to be ignored. |
| 105 | signal(SIGPIPE, SIG_IGN); |
| 106 | |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 107 | buffet::Manager::Options options; |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 108 | options.xmpp_enabled = FLAGS_enable_xmpp; |
Vitaly Buka | 0c6dcd2 | 2015-07-10 00:12:25 -0700 | [diff] [blame] | 109 | options.disable_privet = FLAGS_disable_privet; |
Vitaly Buka | 0c6dcd2 | 2015-07-10 00:12:25 -0700 | [diff] [blame] | 110 | options.enable_ping = FLAGS_enable_ping; |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 111 | options.device_whitelist = {device_whitelist.begin(), device_whitelist.end()}; |
Vitaly Buka | 84fd6dd | 2015-06-09 17:22:18 -0700 | [diff] [blame] | 112 | |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 113 | options.config_options.defaults = base::FilePath{FLAGS_config_path}; |
| 114 | options.config_options.settings = base::FilePath{FLAGS_state_path}; |
| 115 | options.config_options.definitions = base::FilePath{"/etc/weaved"}; |
| 116 | options.config_options.test_definitions = |
| 117 | base::FilePath{FLAGS_test_definitions_path}; |
| 118 | options.config_options.disable_security = FLAGS_disable_security; |
| 119 | options.config_options.test_privet_ssid = FLAGS_test_privet_ssid; |
| 120 | |
| 121 | buffet::Daemon daemon{options}; |
Alex Vakulenko | 79e6a28 | 2014-09-08 17:07:19 -0700 | [diff] [blame] | 122 | return daemon.Run(); |
Chris Sosa | 5bac949 | 2014-03-24 11:18:54 -0700 | [diff] [blame] | 123 | } |