// 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 GARNET_DRIVERS_WLAN_TESTING_PHY_DEVICE_H
#define GARNET_DRIVERS_WLAN_TESTING_PHY_DEVICE_H

#include <ddk/device.h>
#include <wlan/async/dispatcher.h>
#include <zircon/types.h>

#include <fuchsia/wlan/device/cpp/fidl.h>

#include <memory>
#include <mutex>
#include <unordered_map>

namespace wlan {
namespace testing {

class IfaceDevice;

class PhyDevice : public ::fuchsia::wlan::device::Phy {
   public:
    PhyDevice(zx_device_t* device);
    virtual ~PhyDevice() = default;

    zx_status_t Bind();

    void Unbind();
    void Release();
    zx_status_t Ioctl(uint32_t op, const void* in_buf, size_t in_len, void* out_buf, size_t out_len,
                      size_t* out_actual);

    virtual void Query(QueryCallback callback) override;
    virtual void CreateIface(::fuchsia::wlan::device::CreateIfaceRequest req,
                             CreateIfaceCallback callback) override;
    virtual void DestroyIface(::fuchsia::wlan::device::DestroyIfaceRequest req,
                              DestroyIfaceCallback callback) override;

   private:
    zx_status_t Connect(const void* buf, size_t len);

    zx_device_t* zxdev_;
    zx_device_t* parent_;

    std::mutex lock_;
    std::unique_ptr<wlan::async::Dispatcher<::fuchsia::wlan::device::Phy>> dispatcher_;
    std::unordered_map<uint16_t, IfaceDevice*> ifaces_;
    // Next available Iface id. Must be checked against the map to prevent overwriting an existing
    // IfaceDevice pointer in the map.
    uint16_t next_id_ = 0;
};

}  // namespace testing
}  // namespace wlan

#endif  // GARNET_DRIVERS_WLAN_TESTING_PHY_DEVICE_H