runtime_probe/system/context_impl.h

// Copyright 2021 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_SYSTEM_CONTEXT_IMPL_H_
#define RUNTIME_PROBE_SYSTEM_CONTEXT_IMPL_H_

#include <memory>

#include <base/check.h>
#include <brillo/dbus/dbus_connection.h>
#include <chromeos-config/libcros_config/cros_config.h>
#include <diagnostics/mojom/public/cros_healthd.mojom.h>
#include <libcrossystem/crossystem.h>
#include <mojo/public/cpp/bindings/remote.h>
#include <shill/dbus-proxies.h>

#include "runtime_probe/system/context.h"
#include "runtime_probe/system/syscaller_impl.h"

namespace runtime_probe {

class ContextImpl : public Context {
 public:
  ~ContextImpl() override;

  brillo::CrosConfigInterface* cros_config() override { return &cros_config_; }

  crossystem::Crossystem* crossystem() override { return &crossystem_; }

  Syscaller* syscaller() override { return &syscaller_; }

  org::chromium::debugdProxyInterface* debugd_proxy() override {
    CHECK(debugd_proxy_);
    return debugd_proxy_.get();
  };

  HelperInvoker* helper_invoker() override {
    CHECK(helper_invoker_);
    return helper_invoker_.get();
  }

  org::chromium::flimflam::ManagerProxyInterface* shill_manager_proxy()
      override {
    CHECK(shill_manager_proxy_);
    return shill_manager_proxy_.get();
  }

  std::unique_ptr<org::chromium::flimflam::DeviceProxyInterface>
  CreateShillDeviceProxy(const dbus::ObjectPath& path) override {
    CHECK(dbus_bus_);
    return std::make_unique<org::chromium::flimflam::DeviceProxy>(dbus_bus_,
                                                                  path);
  }

  cros_healthd_mojom::CrosHealthdProbeService* GetCrosHealthdProbeServiceProxy()
      override;

 protected:
  // This interface should be used through its derived classes.
  ContextImpl();

  // Setups the dbus connection and the dbus services.
  bool SetupDBusServices();

  // The object to hold the dbus connection.
  brillo::DBusConnection connection_;
  // The object to access the ChromeOS model configuration.
  brillo::CrosConfig cros_config_;
  // The object to access crossystem system properties.
  crossystem::Crossystem crossystem_;
  // The object to make syscalls.
  SyscallerImpl syscaller_;
  // The reference of the dbus connection.
  scoped_refptr<dbus::Bus> dbus_bus_;
  // The proxy object for dbugd dbus service.
  std::unique_ptr<org::chromium::debugdProxyInterface> debugd_proxy_;
  // The object for invoking helper.
  std::unique_ptr<HelperInvoker> helper_invoker_;
  // The proxy object for shill manager.
  std::unique_ptr<org::chromium::flimflam::ManagerProxyInterface>
      shill_manager_proxy_;
  // The mojo remote for cros_healthd probe service.
  mojo::Remote<cros_healthd_mojom::CrosHealthdProbeService>
      cros_healthd_service_;
};

}  // namespace runtime_probe

#endif  // RUNTIME_PROBE_SYSTEM_CONTEXT_IMPL_H_