// Copyright 2018 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 LIB_ASYNC_CPP_TRAP_H_
#define LIB_ASYNC_CPP_TRAP_H_

#include <lib/async/trap.h>
#include <lib/fit/function.h>
#include <lib/zx/guest.h>

#include <utility>

namespace async {

// Holds context for a bell trap and its handler.
//
// After successfully posting setting the trap, the client is responsible for retaining
// the structure in memory (and unmodified) until the guest has been destroyed or the
// dispatcher shuts down.  There is no way to cancel a trap which has been set.
//
// Concrete implementations: |async::GuestBellTrap|, |async::GuestBellTrapMethod|.
// Please do not create subclasses of GuestBellTrapBase outside of this library.
class GuestBellTrapBase {
 protected:
  explicit GuestBellTrapBase(async_guest_bell_trap_handler_t* handler);
  ~GuestBellTrapBase();

  GuestBellTrapBase(const GuestBellTrapBase&) = delete;
  GuestBellTrapBase(GuestBellTrapBase&&) = delete;
  GuestBellTrapBase& operator=(const GuestBellTrapBase&) = delete;
  GuestBellTrapBase& operator=(GuestBellTrapBase&&) = delete;

 public:
  // Sets a bell trap in the guest to be handled asynchronously via a handler.
  //
  // |guest| is the handle of the guest the trap will be set on.
  // |addr| is the base address for the trap in the guest's physical address space.
  // |length| is the size of the trap in the guest's physical address space.
  //
  // Returns |ZX_OK| if the trap was successfully set.
  // Returns |ZX_ERR_ACCESS_DENIED| if the guest does not have |ZX_RIGHT_WRITE|.
  // Returns |ZX_ERR_ALREADY_EXISTS| if a bell trap with the same |addr| exists.
  // Returns |ZX_ERR_INVALID_ARGS| if |addr| or |length| are invalid.
  // Returns |ZX_ERR_OUT_OF_RANGE| if |addr| or |length| are out of range of the
  // address space.
  // Returns |ZX_ERR_WRONG_TYPE| if |guest| is not a handle to a guest.
  // Returns |ZX_ERR_BAD_STATE| if the dispatcher is shutting down.
  // Returns |ZX_ERR_NOT_SUPPORTED| if not supported by the dispatcher.
  //
  // This operation is thread-safe.
  zx_status_t SetTrap(async_dispatcher_t* dispatcher, const zx::guest& guest, zx_vaddr_t addr,
                      size_t length);

 protected:
  template <typename T>
  static T* Dispatch(async_guest_bell_trap_t* trap) {
    static_assert(offsetof(GuestBellTrapBase, trap_) == 0, "");
    auto self = reinterpret_cast<GuestBellTrapBase*>(trap);
    return static_cast<T*>(self);
  }

 private:
  async_guest_bell_trap_t trap_;
};

// A bell trap whose handler is bound to a |async::Task::Handler| function.
//
// Prefer using |async::GuestBellTrapMethod| instead for binding to a fixed class member
// function since it is more efficient to dispatch.
class GuestBellTrap final : public GuestBellTrapBase {
 public:
  // Handles an asynchronous trap access.
  //
  // The |status| is |ZX_OK| if the bell was received and |bell| contains the
  // information from the packet, otherwise |bell| is null.
  using Handler = fit::function<void(async_dispatcher_t* dispatcher, async::GuestBellTrap* trap,
                                     zx_status_t status, const zx_packet_guest_bell_t* bell)>;

  explicit GuestBellTrap(Handler handler = nullptr);
  ~GuestBellTrap();

  void set_handler(Handler handler) { handler_ = std::move(handler); }
  bool has_handler() const { return !!handler_; }

 private:
  static void CallHandler(async_dispatcher_t* dispatcher, async_guest_bell_trap_t* trap,
                          zx_status_t status, const zx_packet_guest_bell_t* bell);

  Handler handler_;
};

// A bell trap whose handler is bound to a fixed class member function.
//
// Usage:
//
// class Foo {
//     void Handle(async_dispatcher_t* dispatcher, async::GuestBellTrapBase* trap, zx_status_t
//     status,
//                 const zx_packet_guest_bell_t* bell) { ... }
//     async::GuestBellTrapMethod<Foo, &Foo::Handle> trap_{this};
// };
template <class Class,
          void (Class::*method)(async_dispatcher_t* dispatcher, async::GuestBellTrapBase* trap,
                                zx_status_t status, const zx_packet_guest_bell_t* bell)>
class GuestBellTrapMethod final : public GuestBellTrapBase {
 public:
  explicit GuestBellTrapMethod(Class* instance)
      : GuestBellTrapBase(&GuestBellTrapMethod::CallHandler), instance_(instance) {}

 private:
  static void CallHandler(async_dispatcher_t* dispatcher, async_guest_bell_trap_t* trap,
                          zx_status_t status, const zx_packet_guest_bell_t* bell) {
    auto self = Dispatch<GuestBellTrapMethod>(trap);
    (self->instance_->*method)(dispatcher, self, status, bell);
  }

  Class* const instance_;
};

}  // namespace async

#endif  // LIB_ASYNC_CPP_TRAP_H_