// SPDX-License-Identifier: GPL-2.0-only
/*
 * Intel IFC VF NIC driver for virtio dataplane offloading
 *
 * Copyright (C) 2020 Intel Corporation.
 *
 * Author: Zhu Lingshan <lingshan.zhu@intel.com>
 *
 */

#include <linux/interrupt.h>
#include <linux/module.h>
#include <linux/pci.h>
#include <linux/sysfs.h>
#include "ifcvf_base.h"

#define DRIVER_AUTHOR   "Intel Corporation"
#define IFCVF_DRIVER_NAME       "ifcvf"

static irqreturn_t ifcvf_config_changed(int irq, void *arg)
{
        struct ifcvf_hw *vf = arg;

        if (vf->config_cb.callback)
                return vf->config_cb.callback(vf->config_cb.private);

        return IRQ_HANDLED;
}

static irqreturn_t ifcvf_vq_intr_handler(int irq, void *arg)
{
        struct vring_info *vring = arg;

        if (vring->cb.callback)
                return vring->cb.callback(vring->cb.private);

        return IRQ_HANDLED;
}

static irqreturn_t ifcvf_vqs_reused_intr_handler(int irq, void *arg)
{
        struct ifcvf_hw *vf = arg;
        struct vring_info *vring;
        int i;

        for (i = 0; i < vf->nr_vring; i++) {
                vring = &vf->vring[i];
                if (vring->cb.callback)
                        vring->cb.callback(vring->cb.private);
        }

        return IRQ_HANDLED;
}

static irqreturn_t ifcvf_dev_intr_handler(int irq, void *arg)
{
        struct ifcvf_hw *vf = arg;
        u8 isr;

        isr = vp_ioread8(vf->isr);
        if (isr & VIRTIO_PCI_ISR_CONFIG)
                ifcvf_config_changed(irq, arg);

        return ifcvf_vqs_reused_intr_handler(irq, arg);
}

static void ifcvf_free_irq_vectors(void *data)
{
        pci_free_irq_vectors(data);
}

static void ifcvf_free_per_vq_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int i;

        for (i = 0; i < vf->nr_vring; i++) {
                if (vf->vring[i].irq != -EINVAL) {
                        devm_free_irq(&pdev->dev, vf->vring[i].irq, &vf->vring[i]);
                        vf->vring[i].irq = -EINVAL;
                }
        }
}

static void ifcvf_free_vqs_reused_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;

        if (vf->vqs_reused_irq != -EINVAL) {
                devm_free_irq(&pdev->dev, vf->vqs_reused_irq, vf);
                vf->vqs_reused_irq = -EINVAL;
        }

}

static void ifcvf_free_vq_irq(struct ifcvf_hw *vf)
{
        if (vf->msix_vector_status == MSIX_VECTOR_PER_VQ_AND_CONFIG)
                ifcvf_free_per_vq_irq(vf);
        else
                ifcvf_free_vqs_reused_irq(vf);
}

static void ifcvf_free_config_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;

        if (vf->config_irq == -EINVAL)
                return;

        /* If the irq is shared by all vqs and the config interrupt,
         * it is already freed in ifcvf_free_vq_irq, so here only
         * need to free config irq when msix_vector_status != MSIX_VECTOR_DEV_SHARED
         */
        if (vf->msix_vector_status != MSIX_VECTOR_DEV_SHARED) {
                devm_free_irq(&pdev->dev, vf->config_irq, vf);
                vf->config_irq = -EINVAL;
        }
}

static void ifcvf_free_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;

        ifcvf_free_vq_irq(vf);
        ifcvf_free_config_irq(vf);
        ifcvf_free_irq_vectors(pdev);
        vf->num_msix_vectors = 0;
}

/* ifcvf MSIX vectors allocator, this helper tries to allocate
 * vectors for all virtqueues and the config interrupt.
 * It returns the number of allocated vectors, negative
 * return value when fails.
 */
static int ifcvf_alloc_vectors(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int max_intr, ret;

        /* all queues and config interrupt  */
        max_intr = vf->nr_vring + 1;
        ret = pci_alloc_irq_vectors(pdev, 1, max_intr, PCI_IRQ_MSIX | PCI_IRQ_AFFINITY);

        if (ret < 0) {
                IFCVF_ERR(pdev, "Failed to alloc IRQ vectors\n");
                return ret;
        }

        if (ret < max_intr)
                IFCVF_INFO(pdev,
                           "Requested %u vectors, however only %u allocated, lower performance\n",
                           max_intr, ret);

        return ret;
}

static int ifcvf_request_per_vq_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int i, vector, ret, irq;

        vf->vqs_reused_irq = -EINVAL;
        for (i = 0; i < vf->nr_vring; i++) {
                snprintf(vf->vring[i].msix_name, 256, "ifcvf[%s]-%d\n", pci_name(pdev), i);
                vector = i;
                irq = pci_irq_vector(pdev, vector);
                ret = devm_request_irq(&pdev->dev, irq,
                                       ifcvf_vq_intr_handler, 0,
                                       vf->vring[i].msix_name,
                                       &vf->vring[i]);
                if (ret) {
                        IFCVF_ERR(pdev, "Failed to request irq for vq %d\n", i);
                        goto err;
                }

                vf->vring[i].irq = irq;
                ret = ifcvf_set_vq_vector(vf, i, vector);
                if (ret == VIRTIO_MSI_NO_VECTOR) {
                        IFCVF_ERR(pdev, "No msix vector for vq %u\n", i);
                        goto err;
                }
        }

        return 0;
err:
        ifcvf_free_irq(vf);

        return -EFAULT;
}

static int ifcvf_request_vqs_reused_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int i, vector, ret, irq;

        vector = 0;
        snprintf(vf->vring[0].msix_name, 256, "ifcvf[%s]-vqs-reused-irq\n", pci_name(pdev));
        irq = pci_irq_vector(pdev, vector);
        ret = devm_request_irq(&pdev->dev, irq,
                               ifcvf_vqs_reused_intr_handler, 0,
                               vf->vring[0].msix_name, vf);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to request reused irq for the device\n");
                goto err;
        }

        vf->vqs_reused_irq = irq;
        for (i = 0; i < vf->nr_vring; i++) {
                vf->vring[i].irq = -EINVAL;
                ret = ifcvf_set_vq_vector(vf, i, vector);
                if (ret == VIRTIO_MSI_NO_VECTOR) {
                        IFCVF_ERR(pdev, "No msix vector for vq %u\n", i);
                        goto err;
                }
        }

        return 0;
err:
        ifcvf_free_irq(vf);

        return -EFAULT;
}

static int ifcvf_request_dev_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int i, vector, ret, irq;

        vector = 0;
        snprintf(vf->vring[0].msix_name, 256, "ifcvf[%s]-dev-irq\n", pci_name(pdev));
        irq = pci_irq_vector(pdev, vector);
        ret = devm_request_irq(&pdev->dev, irq,
                               ifcvf_dev_intr_handler, 0,
                               vf->vring[0].msix_name, vf);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to request irq for the device\n");
                goto err;
        }

        vf->vqs_reused_irq = irq;
        for (i = 0; i < vf->nr_vring; i++) {
                vf->vring[i].irq = -EINVAL;
                ret = ifcvf_set_vq_vector(vf, i, vector);
                if (ret == VIRTIO_MSI_NO_VECTOR) {
                        IFCVF_ERR(pdev, "No msix vector for vq %u\n", i);
                        goto err;
                }
        }

        vf->config_irq = irq;
        ret = ifcvf_set_config_vector(vf, vector);
        if (ret == VIRTIO_MSI_NO_VECTOR) {
                IFCVF_ERR(pdev, "No msix vector for device config\n");
                goto err;
        }

        return 0;
err:
        ifcvf_free_irq(vf);

        return -EFAULT;

}

static int ifcvf_request_vq_irq(struct ifcvf_hw *vf)
{
        int ret;

        if (vf->msix_vector_status == MSIX_VECTOR_PER_VQ_AND_CONFIG)
                ret = ifcvf_request_per_vq_irq(vf);
        else
                ret = ifcvf_request_vqs_reused_irq(vf);

        return ret;
}

static int ifcvf_request_config_irq(struct ifcvf_hw *vf)
{
        struct pci_dev *pdev = vf->pdev;
        int config_vector, ret;

        if (vf->msix_vector_status == MSIX_VECTOR_PER_VQ_AND_CONFIG)
                config_vector = vf->nr_vring;
        else if (vf->msix_vector_status ==  MSIX_VECTOR_SHARED_VQ_AND_CONFIG)
                /* vector 0 for vqs and 1 for config interrupt */
                config_vector = 1;
        else if (vf->msix_vector_status == MSIX_VECTOR_DEV_SHARED)
                /* re-use the vqs vector */
                return 0;
        else
                return -EINVAL;

        snprintf(vf->config_msix_name, 256, "ifcvf[%s]-config\n",
                 pci_name(pdev));
        vf->config_irq = pci_irq_vector(pdev, config_vector);
        ret = devm_request_irq(&pdev->dev, vf->config_irq,
                               ifcvf_config_changed, 0,
                               vf->config_msix_name, vf);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to request config irq\n");
                goto err;
        }

        ret = ifcvf_set_config_vector(vf, config_vector);
        if (ret == VIRTIO_MSI_NO_VECTOR) {
                IFCVF_ERR(pdev, "No msix vector for device config\n");
                goto err;
        }

        return 0;
err:
        ifcvf_free_irq(vf);

        return -EFAULT;
}

static int ifcvf_request_irq(struct ifcvf_hw *vf)
{
        int nvectors, ret, max_intr;

        nvectors = ifcvf_alloc_vectors(vf);
        if (nvectors <= 0)
                return -EFAULT;

        vf->msix_vector_status = MSIX_VECTOR_PER_VQ_AND_CONFIG;
        max_intr = vf->nr_vring + 1;
        if (nvectors < max_intr)
                vf->msix_vector_status = MSIX_VECTOR_SHARED_VQ_AND_CONFIG;

        if (nvectors == 1) {
                vf->msix_vector_status = MSIX_VECTOR_DEV_SHARED;
                ret = ifcvf_request_dev_irq(vf);

                return ret;
        }

        ret = ifcvf_request_vq_irq(vf);
        if (ret)
                return ret;

        ret = ifcvf_request_config_irq(vf);

        if (ret)
                return ret;

        vf->num_msix_vectors = nvectors;

        return 0;
}

static struct ifcvf_adapter *vdpa_to_adapter(struct vdpa_device *vdpa_dev)
{
        return container_of(vdpa_dev, struct ifcvf_adapter, vdpa);
}

static struct ifcvf_hw *vdpa_to_vf(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_adapter *adapter = vdpa_to_adapter(vdpa_dev);

        return adapter->vf;
}

static u64 ifcvf_vdpa_get_device_features(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_adapter *adapter = vdpa_to_adapter(vdpa_dev);
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);
        struct pci_dev *pdev = adapter->pdev;
        u32 type = vf->dev_type;
        u64 features;

        if (type == VIRTIO_ID_NET || type == VIRTIO_ID_BLOCK)
                features = ifcvf_get_dev_features(vf);
        else {
                features = 0;
                IFCVF_ERR(pdev, "VIRTIO ID %u not supported\n", vf->dev_type);
        }

        return features;
}

static int ifcvf_vdpa_set_driver_features(struct vdpa_device *vdpa_dev, u64 features)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);
        int ret;

        ret = ifcvf_verify_min_features(vf, features);
        if (ret)
                return ret;

        ifcvf_set_driver_features(vf, features);

        return 0;
}

static u64 ifcvf_vdpa_get_driver_features(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);
        u64 features;

        features = ifcvf_get_driver_features(vf);

        return features;
}

static u8 ifcvf_vdpa_get_status(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_get_status(vf);
}

static void ifcvf_vdpa_set_status(struct vdpa_device *vdpa_dev, u8 status)
{
        struct ifcvf_hw *vf;
        u8 status_old;
        int ret;

        vf  = vdpa_to_vf(vdpa_dev);
        status_old = ifcvf_get_status(vf);

        if (status_old == status)
                return;

        if ((status & VIRTIO_CONFIG_S_DRIVER_OK) &&
            !(status_old & VIRTIO_CONFIG_S_DRIVER_OK)) {
                ret = ifcvf_request_irq(vf);
                if (ret) {
                        IFCVF_ERR(vf->pdev, "failed to request irq with error %d\n", ret);
                        return;
                }
        }

        ifcvf_set_status(vf, status);
}

static int ifcvf_vdpa_reset(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);
        u8 status = ifcvf_get_status(vf);

        ifcvf_stop(vf);

        if (status & VIRTIO_CONFIG_S_DRIVER_OK)
                ifcvf_free_irq(vf);

        ifcvf_reset(vf);

        return 0;
}

static u16 ifcvf_vdpa_get_vq_num_max(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_get_max_vq_size(vf);
}

static u16 ifcvf_vdpa_get_vq_num_min(struct vdpa_device *vdpa_dev)
{
        return IFCVF_MIN_VQ_SIZE;
}

static int ifcvf_vdpa_get_vq_state(struct vdpa_device *vdpa_dev, u16 qid,
                                   struct vdpa_vq_state *state)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        state->split.avail_index = ifcvf_get_vq_state(vf, qid);
        return 0;
}

static int ifcvf_vdpa_set_vq_state(struct vdpa_device *vdpa_dev, u16 qid,
                                   const struct vdpa_vq_state *state)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_set_vq_state(vf, qid, state->split.avail_index);
}

static void ifcvf_vdpa_set_vq_cb(struct vdpa_device *vdpa_dev, u16 qid,
                                 struct vdpa_callback *cb)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        vf->vring[qid].cb = *cb;
}

static void ifcvf_vdpa_set_vq_ready(struct vdpa_device *vdpa_dev,
                                    u16 qid, bool ready)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        ifcvf_set_vq_ready(vf, qid, ready);
}

static bool ifcvf_vdpa_get_vq_ready(struct vdpa_device *vdpa_dev, u16 qid)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_get_vq_ready(vf, qid);
}

static void ifcvf_vdpa_set_vq_num(struct vdpa_device *vdpa_dev, u16 qid,
                                  u32 num)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        ifcvf_set_vq_num(vf, qid, num);
}

static int ifcvf_vdpa_set_vq_address(struct vdpa_device *vdpa_dev, u16 qid,
                                     u64 desc_area, u64 driver_area,
                                     u64 device_area)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_set_vq_address(vf, qid, desc_area, driver_area, device_area);
}

static void ifcvf_vdpa_kick_vq(struct vdpa_device *vdpa_dev, u16 qid)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        ifcvf_notify_queue(vf, qid);
}

static u32 ifcvf_vdpa_get_generation(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return vp_ioread8(&vf->common_cfg->config_generation);
}

static u32 ifcvf_vdpa_get_device_id(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return vf->dev_type;
}

static u32 ifcvf_vdpa_get_vendor_id(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_adapter *adapter = vdpa_to_adapter(vdpa_dev);
        struct pci_dev *pdev = adapter->pdev;

        return pdev->subsystem_vendor;
}

static u32 ifcvf_vdpa_get_vq_align(struct vdpa_device *vdpa_dev)
{
        return IFCVF_QUEUE_ALIGNMENT;
}

static size_t ifcvf_vdpa_get_config_size(struct vdpa_device *vdpa_dev)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return  vf->config_size;
}

static u32 ifcvf_vdpa_get_vq_group(struct vdpa_device *vdpa, u16 idx)
{
        return 0;
}

static void ifcvf_vdpa_get_config(struct vdpa_device *vdpa_dev,
                                  unsigned int offset,
                                  void *buf, unsigned int len)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        ifcvf_read_dev_config(vf, offset, buf, len);
}

static void ifcvf_vdpa_set_config(struct vdpa_device *vdpa_dev,
                                  unsigned int offset, const void *buf,
                                  unsigned int len)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        ifcvf_write_dev_config(vf, offset, buf, len);
}

static void ifcvf_vdpa_set_config_cb(struct vdpa_device *vdpa_dev,
                                     struct vdpa_callback *cb)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        vf->config_cb.callback = cb->callback;
        vf->config_cb.private = cb->private;
}

static int ifcvf_vdpa_get_vq_irq(struct vdpa_device *vdpa_dev,
                                 u16 qid)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        if (vf->vqs_reused_irq < 0)
                return vf->vring[qid].irq;
        else
                return -EINVAL;
}

static u16 ifcvf_vdpa_get_vq_size(struct vdpa_device *vdpa_dev,
                             u16 qid)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);

        return ifcvf_get_vq_size(vf, qid);
}

static struct vdpa_notification_area ifcvf_get_vq_notification(struct vdpa_device *vdpa_dev,
                                                               u16 idx)
{
        struct ifcvf_hw *vf = vdpa_to_vf(vdpa_dev);
        struct vdpa_notification_area area;

        area.addr = vf->vring[idx].notify_pa;
        if (!vf->notify_off_multiplier)
                area.size = PAGE_SIZE;
        else
                area.size = vf->notify_off_multiplier;

        return area;
}

/*
 * IFCVF currently doesn't have on-chip IOMMU, so not
 * implemented set_map()/dma_map()/dma_unmap()
 */
static const struct vdpa_config_ops ifc_vdpa_ops = {
        .get_device_features = ifcvf_vdpa_get_device_features,
        .set_driver_features = ifcvf_vdpa_set_driver_features,
        .get_driver_features = ifcvf_vdpa_get_driver_features,
        .get_status     = ifcvf_vdpa_get_status,
        .set_status     = ifcvf_vdpa_set_status,
        .reset          = ifcvf_vdpa_reset,
        .get_vq_num_max = ifcvf_vdpa_get_vq_num_max,
        .get_vq_num_min = ifcvf_vdpa_get_vq_num_min,
        .get_vq_state   = ifcvf_vdpa_get_vq_state,
        .set_vq_state   = ifcvf_vdpa_set_vq_state,
        .set_vq_cb      = ifcvf_vdpa_set_vq_cb,
        .set_vq_ready   = ifcvf_vdpa_set_vq_ready,
        .get_vq_ready   = ifcvf_vdpa_get_vq_ready,
        .set_vq_num     = ifcvf_vdpa_set_vq_num,
        .set_vq_address = ifcvf_vdpa_set_vq_address,
        .get_vq_irq     = ifcvf_vdpa_get_vq_irq,
        .get_vq_size    = ifcvf_vdpa_get_vq_size,
        .kick_vq        = ifcvf_vdpa_kick_vq,
        .get_generation = ifcvf_vdpa_get_generation,
        .get_device_id  = ifcvf_vdpa_get_device_id,
        .get_vendor_id  = ifcvf_vdpa_get_vendor_id,
        .get_vq_align   = ifcvf_vdpa_get_vq_align,
        .get_vq_group   = ifcvf_vdpa_get_vq_group,
        .get_config_size        = ifcvf_vdpa_get_config_size,
        .get_config     = ifcvf_vdpa_get_config,
        .set_config     = ifcvf_vdpa_set_config,
        .set_config_cb  = ifcvf_vdpa_set_config_cb,
        .get_vq_notification = ifcvf_get_vq_notification,
};

static struct virtio_device_id id_table_net[] = {
        {VIRTIO_ID_NET, VIRTIO_DEV_ANY_ID},
        {0},
};

static struct virtio_device_id id_table_blk[] = {
        {VIRTIO_ID_BLOCK, VIRTIO_DEV_ANY_ID},
        {0},
};

static u32 get_dev_type(struct pci_dev *pdev)
{
        u32 dev_type;

        /* This drirver drives both modern virtio devices and transitional
         * devices in modern mode.
         * vDPA requires feature bit VIRTIO_F_ACCESS_PLATFORM,
         * so legacy devices and transitional devices in legacy
         * mode will not work for vDPA, this driver will not
         * drive devices with legacy interface.
         */

        if (pdev->device < 0x1040)
                dev_type =  pdev->subsystem_device;
        else
                dev_type =  pdev->device - 0x1040;

        return dev_type;
}

static int ifcvf_vdpa_dev_add(struct vdpa_mgmt_dev *mdev, const char *name,
                              const struct vdpa_dev_set_config *config)
{
        struct ifcvf_vdpa_mgmt_dev *ifcvf_mgmt_dev;
        struct ifcvf_adapter *adapter;
        struct vdpa_device *vdpa_dev;
        struct pci_dev *pdev;
        struct ifcvf_hw *vf;
        u64 device_features;
        int ret;

        ifcvf_mgmt_dev = container_of(mdev, struct ifcvf_vdpa_mgmt_dev, mdev);
        vf = &ifcvf_mgmt_dev->vf;
        pdev = vf->pdev;
        adapter = vdpa_alloc_device(struct ifcvf_adapter, vdpa,
                                    &pdev->dev, &ifc_vdpa_ops,
                                    NULL, 1, 1, NULL, false);
        if (IS_ERR(adapter)) {
                IFCVF_ERR(pdev, "Failed to allocate vDPA structure");
                return PTR_ERR(adapter);
        }

        ifcvf_mgmt_dev->adapter = adapter;
        adapter->pdev = pdev;
        adapter->vdpa.vmap.dma_dev = &pdev->dev;
        adapter->vdpa.mdev = mdev;
        adapter->vf = vf;
        vdpa_dev = &adapter->vdpa;

        device_features = vf->hw_features;
        if (config->mask & BIT_ULL(VDPA_ATTR_DEV_FEATURES)) {
                if (config->device_features & ~device_features) {
                        IFCVF_ERR(pdev, "The provisioned features 0x%llx are not supported by this device with features 0x%llx\n",
                                  config->device_features, device_features);
                        return -EINVAL;
                }
                device_features &= config->device_features;
        }
        vf->dev_features = device_features;

        if (name)
                ret = dev_set_name(&vdpa_dev->dev, "%s", name);
        else
                ret = dev_set_name(&vdpa_dev->dev, "vdpa%u", vdpa_dev->index);

        ret = _vdpa_register_device(&adapter->vdpa, vf->nr_vring);
        if (ret) {
                put_device(&adapter->vdpa.dev);
                IFCVF_ERR(pdev, "Failed to register to vDPA bus");
                return ret;
        }

        return 0;
}

static void ifcvf_vdpa_dev_del(struct vdpa_mgmt_dev *mdev, struct vdpa_device *dev)
{
        struct ifcvf_vdpa_mgmt_dev *ifcvf_mgmt_dev;

        ifcvf_mgmt_dev = container_of(mdev, struct ifcvf_vdpa_mgmt_dev, mdev);
        _vdpa_unregister_device(dev);
        ifcvf_mgmt_dev->adapter = NULL;
}

static const struct vdpa_mgmtdev_ops ifcvf_vdpa_mgmt_dev_ops = {
        .dev_add = ifcvf_vdpa_dev_add,
        .dev_del = ifcvf_vdpa_dev_del
};

static int ifcvf_probe(struct pci_dev *pdev, const struct pci_device_id *id)
{
        struct ifcvf_vdpa_mgmt_dev *ifcvf_mgmt_dev;
        struct device *dev = &pdev->dev;
        struct ifcvf_hw *vf;
        u32 dev_type;
        int ret, i;

        ret = pcim_enable_device(pdev);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to enable device\n");
                return ret;
        }
        ret = pcim_iomap_regions(pdev, BIT(0) | BIT(2) | BIT(4),
                                 IFCVF_DRIVER_NAME);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to request MMIO region\n");
                return ret;
        }

        ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(64));
        if (ret) {
                IFCVF_ERR(pdev, "No usable DMA configuration\n");
                return ret;
        }

        ret = devm_add_action_or_reset(dev, ifcvf_free_irq_vectors, pdev);
        if (ret) {
                IFCVF_ERR(pdev,
                          "Failed for adding devres for freeing irq vectors\n");
                return ret;
        }

        pci_set_master(pdev);
        ifcvf_mgmt_dev = kzalloc_obj(struct ifcvf_vdpa_mgmt_dev);
        if (!ifcvf_mgmt_dev) {
                IFCVF_ERR(pdev, "Failed to alloc memory for the vDPA management device\n");
                return -ENOMEM;
        }

        vf = &ifcvf_mgmt_dev->vf;
        vf->dev_type = get_dev_type(pdev);
        vf->base = pcim_iomap_table(pdev);
        vf->pdev = pdev;

        ret = ifcvf_init_hw(vf, pdev);
        if (ret) {
                IFCVF_ERR(pdev, "Failed to init IFCVF hw\n");
                goto err;
        }

        for (i = 0; i < vf->nr_vring; i++)
                vf->vring[i].irq = -EINVAL;

        vf->hw_features = ifcvf_get_hw_features(vf);
        vf->config_size = ifcvf_get_config_size(vf);

        dev_type = get_dev_type(pdev);
        switch (dev_type) {
        case VIRTIO_ID_NET:
                ifcvf_mgmt_dev->mdev.id_table = id_table_net;
                break;
        case VIRTIO_ID_BLOCK:
                ifcvf_mgmt_dev->mdev.id_table = id_table_blk;
                break;
        default:
                IFCVF_ERR(pdev, "VIRTIO ID %u not supported\n", dev_type);
                ret = -EOPNOTSUPP;
                goto err;
        }

        ifcvf_mgmt_dev->mdev.ops = &ifcvf_vdpa_mgmt_dev_ops;
        ifcvf_mgmt_dev->mdev.device = dev;
        ifcvf_mgmt_dev->mdev.max_supported_vqs = vf->nr_vring;
        ifcvf_mgmt_dev->mdev.supported_features = vf->hw_features;
        ifcvf_mgmt_dev->mdev.config_attr_mask = (1 << VDPA_ATTR_DEV_FEATURES);

        ret = vdpa_mgmtdev_register(&ifcvf_mgmt_dev->mdev);
        if (ret) {
                IFCVF_ERR(pdev,
                          "Failed to initialize the management interfaces\n");
                goto err;
        }

        pci_set_drvdata(pdev, ifcvf_mgmt_dev);

        return 0;

err:
        kfree(ifcvf_mgmt_dev->vf.vring);
        kfree(ifcvf_mgmt_dev);
        return ret;
}

static void ifcvf_remove(struct pci_dev *pdev)
{
        struct ifcvf_vdpa_mgmt_dev *ifcvf_mgmt_dev;

        ifcvf_mgmt_dev = pci_get_drvdata(pdev);
        vdpa_mgmtdev_unregister(&ifcvf_mgmt_dev->mdev);
        kfree(ifcvf_mgmt_dev->vf.vring);
        kfree(ifcvf_mgmt_dev);
}

static struct pci_device_id ifcvf_pci_ids[] = {
        /* N3000 network device */
        { PCI_DEVICE_SUB(PCI_VENDOR_ID_REDHAT_QUMRANET,
                         N3000_DEVICE_ID,
                         PCI_VENDOR_ID_INTEL,
                         N3000_SUBSYS_DEVICE_ID) },
        /* C5000X-PL network device
         * F2000X-PL network device
         */
        { PCI_DEVICE_SUB(PCI_VENDOR_ID_REDHAT_QUMRANET,
                         VIRTIO_TRANS_ID_NET,
                         PCI_VENDOR_ID_INTEL,
                         VIRTIO_ID_NET) },
        /* C5000X-PL block device */
        { PCI_DEVICE_SUB(PCI_VENDOR_ID_REDHAT_QUMRANET,
                         VIRTIO_TRANS_ID_BLOCK,
                         PCI_VENDOR_ID_INTEL,
                         VIRTIO_ID_BLOCK) },

        { 0 },
};
MODULE_DEVICE_TABLE(pci, ifcvf_pci_ids);

static struct pci_driver ifcvf_driver = {
        .name     = IFCVF_DRIVER_NAME,
        .id_table = ifcvf_pci_ids,
        .probe    = ifcvf_probe,
        .remove   = ifcvf_remove,
};

module_pci_driver(ifcvf_driver);

MODULE_DESCRIPTION("Intel IFC VF NIC driver for virtio dataplane offloading");
MODULE_LICENSE("GPL v2");