pciguard/daemon.h

// Copyright 2020 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 PCIGUARD_DAEMON_H_
#define PCIGUARD_DAEMON_H_

#include "pciguard/session_monitor.h"
#include "pciguard/udev_monitor.h"

#include <brillo/daemons/dbus_daemon.h>
#include <memory>
#include <string>

namespace pciguard {

using brillo::dbus_utils::DBusSignal;

class Daemon : public brillo::DBusServiceDaemon {
 public:
  Daemon();
  Daemon(const Daemon&) = delete;
  Daemon& operator=(const Daemon&) = delete;
  ~Daemon() = default;

 protected:
  int OnInit() override;
  void RegisterDBusObjectsAsync(
      brillo::dbus_utils::AsyncEventSequencer* sequencer) override;
  void HandleUserPermissionChanged(bool ext_pci_allowed);

 private:
  std::unique_ptr<SysfsUtils> utils_;
  std::unique_ptr<EventHandler> event_handler_;
  std::unique_ptr<SessionMonitor> session_monitor_;
  std::unique_ptr<UdevMonitor> udev_monitor_;
  std::unique_ptr<brillo::dbus_utils::DBusObject> dbus_object_;
  std::weak_ptr<DBusSignal<std::string>> dev_blocked_signal_;
  void HandlePCIDeviceBlocked(const std::string& drvr);
};

}  // namespace pciguard

#endif  // PCIGUARD_DAEMON_H__