crosdns/crosdns_daemon.h

// Copyright 2018 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.

#ifndef CROSDNS_CROSDNS_DAEMON_H_
#define CROSDNS_CROSDNS_DAEMON_H_

#include <memory>
#include <string>

#include <base/macros.h>
#include <brillo/daemons/dbus_daemon.h>
#include <brillo/errors/error.h>

#include "crosdns/dbus_adaptors/org.chromium.CrosDns.h"
#include "crosdns/hosts_modifier.h"

namespace crosdns {

// D-Bus service daemon implementation, this is used to receive requests to add
// and remove entries from the /etc/hosts files.
class CrosDnsDaemon : public brillo::DBusServiceDaemon,
                      public org::chromium::CrosDnsInterface {
 public:
  CrosDnsDaemon();
  CrosDnsDaemon(const CrosDnsDaemon&) = delete;
  CrosDnsDaemon& operator=(const CrosDnsDaemon&) = delete;

  ~CrosDnsDaemon() override;

  // Implementations of the public methods interface.
  // org::chromium::CrosDnsInterface:
  bool SetHostnameIpMapping(brillo::ErrorPtr* err,
                            const std::string& hostname,
                            const std::string& ipv4,
                            const std::string& ipv6) override;
  bool RemoveHostnameIpMapping(brillo::ErrorPtr* err,
                               const std::string& hostname) override;

 protected:
  // brillo::DBusServiceDaemon:
  void RegisterDBusObjectsAsync(
      brillo::dbus_utils::AsyncEventSequencer* sequencer) override;
  int OnInit() override;

 private:
  std::unique_ptr<brillo::dbus_utils::DBusObject> dbus_object_;

  org::chromium::CrosDnsAdaptor dbus_adaptor_;

  HostsModifier hosts_modifier_;
};

}  // namespace crosdns

#endif  // CROSDNS_CROSDNS_DAEMON_H_