// Copyright 2016 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.

#pragma once

#include <ddktl/device.h>
#include <ddktl/protocol/usb.h>
#include <ddktl/protocol/usb/hci.h>
#include <ddktl/protocol/usb/hub.h>
#include <fbl/array.h>
#include <fbl/mutex.h>
#include <fbl/ref_counted.h>
#include <fbl/ref_ptr.h>
#include <usb/usb-request.h>
#include <lib/sync/completion.h>
#include <usb/request-cpp.h>
#include <zircon/hw/usb.h>

#include <optional>
#include <threads.h>

namespace usb_bus {

class UsbDevice;
using UsbDeviceType = ddk::Device<UsbDevice, ddk::GetProtocolable, ddk::Messageable,
                                  ddk::Unbindable>;

class UsbDevice : public UsbDeviceType,
                  public ddk::UsbProtocol<UsbDevice, ddk::base_protocol>,
                  public fbl::RefCounted<UsbDevice>  {
  public:
    UsbDevice(zx_device_t* parent, const ddk::UsbHciProtocolClient& hci, uint32_t device_id,
              uint32_t hub_id, usb_speed_t speed)
        : UsbDeviceType(parent), device_id_(device_id), hub_id_(hub_id), speed_(speed), hci_(hci) {}

    static zx_status_t Create(zx_device_t* parent, const ddk::UsbHciProtocolClient& hci,
                              uint32_t device_id, uint32_t hub_id, usb_speed_t speed,
                              fbl::RefPtr<UsbDevice>* out_device);

    // Device protocol implementation.
    zx_status_t DdkGetProtocol(uint32_t proto_id, void* out);
    zx_status_t DdkMessage(fidl_msg_t* msg, fidl_txn_t* txn);
    void DdkUnbind();
    void DdkRelease();

    // USB protocol implementation.
    zx_status_t UsbControlOut(uint8_t request_type, uint8_t request, uint16_t value, uint16_t index,
                              int64_t timeout, const void* write_buffer, size_t write_size);
    zx_status_t UsbControlIn(uint8_t request_type, uint8_t request, uint16_t value, uint16_t index,
                             int64_t timeout, void* out_read_buffer, size_t read_size,
                             size_t* out_read_actual);
    void UsbRequestQueue(usb_request_t* usb_request, const usb_request_complete_t* complete_cb);
    usb_speed_t UsbGetSpeed();
    zx_status_t UsbSetInterface(uint8_t interface_number, uint8_t alt_setting);
    uint8_t UsbGetConfiguration();
    zx_status_t UsbSetConfiguration(uint8_t configuration);
    zx_status_t UsbEnableEndpoint(const usb_endpoint_descriptor_t* ep_desc,
                                  const usb_ss_ep_comp_descriptor_t* ss_com_desc, bool enable);
    zx_status_t UsbResetEndpoint(uint8_t ep_address);
    zx_status_t UsbResetDevice();
    size_t UsbGetMaxTransferSize(uint8_t ep_address);
    uint32_t UsbGetDeviceId();
    void UsbGetDeviceDescriptor(usb_device_descriptor_t* out_desc);
    zx_status_t UsbGetConfigurationDescriptorLength(uint8_t configuration, size_t* out_length);
    zx_status_t UsbGetConfigurationDescriptor(uint8_t configuration, void* out_desc_buffer,
                                              size_t desc_size, size_t* out_desc_actual);
    size_t UsbGetDescriptorsLength();
    void UsbGetDescriptors(void* out_descs_buffer, size_t descs_size, size_t* out_descs_actual);
    zx_status_t UsbGetStringDescriptor(uint8_t desc_id, uint16_t lang_id, uint16_t* out_lang_id,
                                       void* out_string_buffer, size_t string_size,
                                       size_t* out_string_actual);
    zx_status_t UsbCancelAll(uint8_t ep_address);
    uint64_t UsbGetCurrentFrame();
    size_t UsbGetRequestSize();

    // FIDL messages.
    zx_status_t MsgGetDeviceSpeed(fidl_txn_t* txn);
    zx_status_t MsgGetDeviceDescriptor(fidl_txn_t* txn);
    zx_status_t MsgGetConfigurationDescriptorSize(uint8_t config, fidl_txn_t* txn);
    zx_status_t MsgGetConfigurationDescriptor(uint8_t config, fidl_txn_t* txn);
    zx_status_t MsgGetStringDescriptor(uint8_t desc_id, uint16_t lang_id, fidl_txn_t* txn);
    zx_status_t MsgSetInterface(uint8_t interface_number, uint8_t alt_setting, fidl_txn_t* txn);
    zx_status_t MsgGetDeviceId(fidl_txn_t* txn);
    zx_status_t MsgGetHubDeviceId(fidl_txn_t* txn);
    zx_status_t MsgGetConfiguration(fidl_txn_t* txn);
    zx_status_t MsgSetConfiguration(uint8_t configuration, fidl_txn_t* txn);

    // Hub support.
    void SetHubInterface(const usb_hub_interface_t* hub_intf);
    zx_status_t HubResetPort(uint32_t port);

    zx_status_t GetDescriptor(uint16_t type, uint16_t index, uint16_t language, void* data,
                              size_t length, size_t* out_actual);
    zx_status_t Reinitialize();

    inline uint32_t GetHubId() const { return hub_id_; }
    inline usb_speed_t GetSpeed() const { return speed_; }

private:
    DISALLOW_COPY_ASSIGN_AND_MOVE(UsbDevice);

    // Internal storage for USB requests.
    struct UsbRequestInternal {
        // Callback to client driver.
        usb_request_complete_t complete_cb;
    };

    using Request = usb::Request<UsbRequestInternal>;
    using UnownedRequest = usb::UnownedRequest<UsbRequestInternal>;
    using RequestPool = usb::RequestPool<UsbRequestInternal>;
    using UnownedRequestQueue = usb::UnownedRequestQueue<UsbRequestInternal>;

    zx_status_t Init();

    int CallbackThread();
    void StartCallbackThread();
    void StopCallbackThread();

    void RequestComplete(usb_request_t* req);
    static void ControlComplete(void* ctx, usb_request_t* req);
     zx_status_t Control(uint8_t request_type, uint8_t request, uint16_t value, uint16_t index,
                         zx_time_t timeout, const void* write_buffer, size_t write_size,
                         void* out_read_buffer, size_t read_size, size_t* out_read_actual);
    const usb_configuration_descriptor_t* GetConfigDesc(uint8_t config);

    // ID assigned by host controller driver.
    const uint32_t device_id_;
    // device_id of the hub we are attached to (or zero for root hub).
    const uint32_t hub_id_;
    const usb_speed_t speed_;

    // Parent's HCI protocol.
    ddk::UsbHciProtocolClient hci_;

    // Hub interface, for devices that are hubs.
    ddk::UsbHubInterfaceClient hub_intf_;

    usb_device_descriptor_t device_desc_;

    // list of all configuration descriptors
    fbl::Array<fbl::Array<uint8_t>> config_descs_;
    uint8_t current_config_index_  __TA_GUARDED(state_lock_);

    std::optional<usb_langid_desc_t> lang_ids_ __TA_GUARDED(state_lock_);

    bool resetting_ __TA_GUARDED(state_lock_) = false;
    // mutex that protects the resetting state member
    fbl::Mutex state_lock_;

    // thread for calling client's usb request complete callback
    thrd_t callback_thread_ = 0;
    bool callback_thread_stop_ __TA_GUARDED(callback_lock_) = false;
    // completion used for signalling callback_thread
    sync_completion_t callback_thread_completion_;
    // list of requests that need to have client's completion callback called
    UnownedRequestQueue completed_reqs_ __TA_GUARDED(callback_lock_);
    // mutex that protects the callback_* members above
    fbl::Mutex callback_lock_;

    // pool of requests that can be reused
    RequestPool free_reqs_;

    size_t parent_req_size_;
};

} // namespace usb_bus