| // Copyright 2014 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <string> |
| #include <sysexits.h> |
| |
| #include <base/command_line.h> |
| #include <brillo/daemons/dbus_daemon.h> |
| #include <brillo/dbus/async_event_sequencer.h> |
| #include <brillo/syslog_logging.h> |
| |
| #include "peerd/dbus_constants.h" |
| #include "peerd/manager.h" |
| |
| using brillo::DBusServiceDaemon; |
| using brillo::dbus_utils::AsyncEventSequencer; |
| using peerd::Manager; |
| using peerd::dbus_constants::kRootServicePath; |
| using peerd::dbus_constants::kServiceName; |
| |
| namespace { |
| |
| static const char kHelpFlag[] = "help"; |
| static const char kLogToStdErrFlag[] = "log_to_stderr"; |
| static const char kInitialMDnsPrefix[] = "mdns_prefix"; |
| static const char kHelpMessage[] = "\n" |
| "This is the peer discovery service daemon.\n" |
| "Usage: peerd [--v=<logging level>]\n" |
| " [--vmodule=<see base/logging.h>]\n" |
| " [--log_to_stderr]\n" |
| " [--mdns_prefix=<first mdns record prefix to try>]\n"; |
| |
| class Daemon : public DBusServiceDaemon { |
| public: |
| explicit Daemon(const std::string& initial_mdns_prefix) |
| : DBusServiceDaemon(kServiceName, kRootServicePath), |
| initial_mdns_prefix_(initial_mdns_prefix) { |
| } |
| |
| protected: |
| void RegisterDBusObjectsAsync(AsyncEventSequencer* sequencer) override { |
| manager_.reset(new Manager(object_manager_.get(), initial_mdns_prefix_)); |
| manager_->RegisterAsync( |
| sequencer->GetHandler("Manager.RegisterAsync() failed.", true)); |
| LOG(INFO) << "peerd starting"; |
| } |
| |
| void OnShutdown(int* return_code) override { |
| DBusServiceDaemon::OnShutdown(return_code); |
| manager_.reset(); |
| } |
| |
| private: |
| const std::string initial_mdns_prefix_; |
| std::unique_ptr<Manager> manager_; |
| |
| DISALLOW_COPY_AND_ASSIGN(Daemon); |
| }; |
| |
| } // namespace |
| |
| int main(int argc, char* argv[]) { |
| base::CommandLine::Init(argc, argv); |
| base::CommandLine* cl = base::CommandLine::ForCurrentProcess(); |
| if (cl->HasSwitch(kHelpFlag)) { |
| LOG(INFO) << kHelpMessage; |
| return EX_USAGE; |
| } |
| brillo::InitFlags flags = brillo::kLogToSyslog; |
| if (cl->HasSwitch(kLogToStdErrFlag)) { |
| flags = brillo::kLogToStderr; |
| } |
| std::string initial_mdns_prefix(cl->GetSwitchValueASCII(kInitialMDnsPrefix)); |
| brillo::InitLog(flags | brillo::kLogHeader); |
| Daemon daemon(initial_mdns_prefix); |
| return daemon.Run(); |
| } |