#include "device_proxy.h"
#include "common.h"
#include <ddk/binding.h>
#include <ddk/debug.h>

#define DEVICE_PROXY_UNIMPLEMENTED                       \
    zxlogf(INFO, "[DeviceProxy] called %s\n", __func__); \
    return ZX_ERR_NOT_SUPPORTED


// This file contains the PciProtocol implementation that is proxied over
// a channel to the specific pci::Device objects in the PCI Bus Driver.
namespace pci {

zx_status_t DeviceProxy::Create(zx_device_t* parent, zx_handle_t rpcch, const char* name) {
    DeviceProxy* dp = new DeviceProxy(parent, rpcch);
    return dp->DdkAdd(name);
}

zx_status_t DeviceProxy::RpcRequest(PciRpcOp op, zx_handle_t* handle, PciRpcMsg* req,
                                    PciRpcMsg* resp) {
    if (rpcch_ == ZX_HANDLE_INVALID) {
        return ZX_ERR_NOT_SUPPORTED;
    }

    uint32_t handle_cnt = 0;
    if (handle) {
        // Since only the caller knows if they expected a valid handle back, make
        // sure the handle is invalid if we didn't get one.
        *handle = ZX_HANDLE_INVALID;
        handle_cnt = 1;
    }

    req->op = op;
    zx_channel_call_args_t cc_args = {};
    cc_args.wr_bytes = req;
    cc_args.wr_num_bytes = sizeof(*req);
    cc_args.rd_bytes = resp;
    cc_args.rd_num_bytes = sizeof(*resp);
    cc_args.rd_handles = handle;
    cc_args.rd_num_handles = handle_cnt;
    cc_args.wr_handles = nullptr;
    cc_args.wr_num_handles = 0;

    uint32_t actual_bytes;
    uint32_t actual_handles;
    zx_status_t st = rpcch_.call(0, zx::time(ZX_TIME_INFINITE), &cc_args, &actual_bytes,
                                 &actual_handles);
    if (st != ZX_OK) {
        return st;
    }

    if (actual_bytes != sizeof(*resp)) {
        return ZX_ERR_INTERNAL;
    }

    return resp->ret;
}

zx_status_t DeviceProxy::DdkGetProtocol(uint32_t proto_id, void* out) {
    if (proto_id == ZX_PROTOCOL_PCI) {
        auto proto = static_cast<pci_protocol_t*>(out);
        proto->ctx = this;
        proto->ops = &pci_protocol_ops_;
        return ZX_OK;
    }

    return ZX_ERR_NOT_SUPPORTED;
}

// TODO(ZX-3927): Convert this to using a better wire format when we no longer
// have to support the kernel driver.
zx_status_t DeviceProxy::PciGetBar(uint32_t bar_id, zx_pci_bar_t* out_bar) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciEnableBusMaster(bool enable) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciResetDevice() {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciMapInterrupt(zx_status_t which_irq, zx::interrupt* out_handle) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciQueryIrqMode(zx_pci_irq_mode_t mode, uint32_t* out_max_irqs) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciSetIrqMode(zx_pci_irq_mode_t mode, uint32_t requested_irq_count) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciGetDeviceInfo(zx_pcie_device_info_t* out_info) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

template <typename T>
zx_status_t DeviceProxy::PciConfigRead(uint16_t offset, T* out_value) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciConfigRead8(uint16_t offset, uint8_t* out_value) {
    return PciConfigRead(offset, out_value);
}

zx_status_t DeviceProxy::PciConfigRead16(uint16_t offset, uint16_t* out_value) {
    return PciConfigRead(offset, out_value);
}

zx_status_t DeviceProxy::PciConfigRead32(uint16_t offset, uint32_t* out_value) {
    return PciConfigRead(offset, out_value);
}

template <typename T>
zx_status_t DeviceProxy::PciConfigWrite(uint16_t offset, T value) {
    return ZX_ERR_NOT_SUPPORTED;
}

zx_status_t DeviceProxy::PciConfigWrite8(uint16_t offset, uint8_t value) {
    return PciConfigWrite(offset, value);
}

zx_status_t DeviceProxy::PciConfigWrite16(uint16_t offset, uint16_t value) {
    return PciConfigWrite(offset, value);
}

zx_status_t DeviceProxy::PciConfigWrite32(uint16_t offset, uint32_t value) {
    return PciConfigWrite(offset, value);
}

zx_status_t DeviceProxy::PciGetFirstCapability(uint8_t cap_id, uint8_t* out_offset) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciGetNextCapability(uint8_t cap_id, uint8_t offset,
                                              uint8_t* out_offset) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

// TODO(ZX-3146): These methods need to be deleted, or refactored.
zx_status_t DeviceProxy::PciGetAuxdata(const char* args,
                                       void* out_data_buffer,
                                       size_t data_size,
                                       size_t* out_data_actual) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

zx_status_t DeviceProxy::PciGetBti(uint32_t index, zx::bti* out_bti) {
    DEVICE_PROXY_UNIMPLEMENTED;
}

} // namespace pci

static zx_status_t pci_device_proxy_create(void* ctx, zx_device_t* parent, const char* name,
                                           const char* args, zx_handle_t rpcch) {
    return pci::DeviceProxy::Create(parent, rpcch, name);
}

static constexpr zx_driver_ops_t pci_device_proxy_driver_ops = []() {
    zx_driver_ops_t ops = {};
    ops.version = DRIVER_OPS_VERSION;
    ops.create = pci_device_proxy_create;
    return ops;
}();

ZIRCON_DRIVER_BEGIN(pci_device_proxy, pci_device_proxy_driver_ops, "zircon", "0.1", 1)
    BI_ABORT_IF_AUTOBIND,
ZIRCON_DRIVER_END(pci_device_proxy)