| // Copyright 2018 The ChromiumOS Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef RUNTIME_PROBE_DAEMON_H_ |
| #define RUNTIME_PROBE_DAEMON_H_ |
| |
| #include <memory> |
| |
| #include <brillo/daemons/dbus_daemon.h> |
| #include <brillo/dbus/async_event_sequencer.h> |
| #include <brillo/dbus/dbus_method_response.h> |
| #include <brillo/dbus/dbus_object.h> |
| |
| // Include the protobuf before generated D-Bus adaptors to ensure the protobuf |
| // messages are defined before adaptors. |
| // TODO(crbug.com/1255584): Includes headers in alphabetical order. |
| #include "runtime_probe/proto_bindings/runtime_probe.pb.h" |
| #include "runtime_probe/dbus_adaptors/org.chromium.RuntimeProbe.h" // NOLINT(build/include_alpha) |
| #include "runtime_probe/probe_config_loader.h" |
| |
| namespace runtime_probe { |
| |
| // Implementation of the runtime_probe D-Bus methods. |
| // Daemon class for the runtime_probe D-Bus service daemon. |
| class Daemon : public brillo::DBusServiceDaemon, |
| public org::chromium::RuntimeProbeAdaptor, |
| public org::chromium::RuntimeProbeInterface { |
| public: |
| explicit Daemon(ProbeConfigLoader* config_loader); |
| Daemon(const Daemon&) = delete; |
| Daemon& operator=(const Daemon&) = delete; |
| |
| ~Daemon() override = default; |
| |
| template <typename... Types> |
| using DBusCallback = typename std::unique_ptr< |
| brillo::dbus_utils::DBusMethodResponse<Types...>>; |
| |
| // org::chromium::RuntimeProbeInterface overrides. |
| void ProbeCategories(DBusCallback<ProbeResult> cb, |
| const ProbeRequest& request) override; |
| void GetKnownComponents(DBusCallback<GetKnownComponentsResult> cb, |
| const GetKnownComponentsRequest& request) override; |
| |
| private: |
| // brillo::DBusServiceDaemon overrides. |
| int OnInit() override; |
| |
| void RegisterDBusObjectsAsync( |
| brillo::dbus_utils::AsyncEventSequencer* sequencer) override; |
| |
| std::unique_ptr<brillo::dbus_utils::DBusObject> dbus_object_; |
| |
| // Dependent classes. |
| ProbeConfigLoader* const config_loader_; |
| }; |
| |
| } // namespace runtime_probe |
| |
| #endif // RUNTIME_PROBE_DAEMON_H_ |