src/devices/bin/driver_host2/driver_host.h

// Copyright 2020 The Fuchsia 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 SRC_DEVICES_BIN_DRIVER_HOST2_DRIVER_HOST_H_
#define SRC_DEVICES_BIN_DRIVER_HOST2_DRIVER_HOST_H_

#include <fuchsia/driver/framework/llcpp/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/inspect/cpp/inspect.h>
#include <lib/zx/status.h>

#include <fbl/intrusive_double_list.h>

#include "src/devices/lib/driver2/record.h"
#include "src/lib/storage/vfs/cpp/pseudo_dir.h"

class Driver : public fidl::WireInterface<fuchsia_driver_framework::Driver>,
               public fbl::DoublyLinkedListable<std::unique_ptr<Driver>> {
 public:
  static zx::status<std::unique_ptr<Driver>> Load(std::string url, std::string binary, zx::vmo vmo);

  Driver(std::string url, std::string binary, void* library, DriverRecordV1* record);
  ~Driver();

  const std::string& url() const { return url_; }
  const std::string& binary() const { return binary_; }
  void set_binding(fidl::ServerBindingRef<fuchsia_driver_framework::Driver> binding);

  // Starts the driver.
  //
  // The handles in |message| will be consumed.
  zx::status<> Start(fidl::OutgoingMessage& message, async_dispatcher_t* dispatcher);

 private:
  std::string url_;
  std::string binary_;
  void* library_;
  DriverRecordV1* record_;
  void* opaque_ = nullptr;
  std::optional<fidl::ServerBindingRef<fuchsia_driver_framework::Driver>> binding_;
};

class DriverHost : public fidl::WireInterface<fuchsia_driver_framework::DriverHost> {
 public:
  // DriverHost does not take ownership of |loop|.
  DriverHost(inspect::Inspector* inspector, async::Loop* loop);

  zx::status<> PublishDriverHost(const fbl::RefPtr<fs::PseudoDir>& svc_dir);
  fit::promise<inspect::Inspector> Inspect();

 private:
  // fidl::WireInterface<fuchsia_driver_framework::DriverHost>
  void Start(fuchsia_driver_framework::wire::DriverStartArgs start_args,
             fidl::ServerEnd<fuchsia_driver_framework::Driver> request,
             StartCompleter::Sync& completer) override;

  async::Loop* const loop_;
  fbl::DoublyLinkedList<std::unique_ptr<Driver>> drivers_;
};

#endif  // SRC_DEVICES_BIN_DRIVER_HOST2_DRIVER_HOST_H_