// SPDX-License-Identifier: (GPL-2.0 OR BSD-3-Clause)
/*
 * Copyright(c) 2018 - 2020 Intel Corporation.
 */

#include "hfi.h"
#include "affinity.h"
#include "sdma.h"
#include "netdev.h"

/**
 * msix_initialize() - Calculate, request and configure MSIx IRQs
 * @dd: valid hfi1 devdata
 *
 */
int msix_initialize(struct hfi1_devdata *dd)
{
        u32 total;
        int ret;
        struct hfi1_msix_entry *entries;

        /*
         * MSIx interrupt count:
         *      one for the general, "slow path" interrupt
         *      one per used SDMA engine
         *      one per kernel receive context
         *      one for each VNIC context
         *      ...any new IRQs should be added here.
         */
        total = 1 + dd->num_sdma + dd->n_krcv_queues + dd->num_netdev_contexts;

        if (total >= CCE_NUM_MSIX_VECTORS)
                return -EINVAL;

        ret = pci_alloc_irq_vectors(dd->pcidev, total, total, PCI_IRQ_MSIX);
        if (ret < 0) {
                dd_dev_err(dd, "pci_alloc_irq_vectors() failed: %d\n", ret);
                return ret;
        }

        entries = kzalloc_objs(*dd->msix_info.msix_entries, total);
        if (!entries) {
                pci_free_irq_vectors(dd->pcidev);
                return -ENOMEM;
        }

        dd->msix_info.msix_entries = entries;
        spin_lock_init(&dd->msix_info.msix_lock);
        bitmap_zero(dd->msix_info.in_use_msix, total);
        dd->msix_info.max_requested = total;
        dd_dev_info(dd, "%u MSI-X interrupts allocated\n", total);

        return 0;
}

/**
 * msix_request_irq() - Allocate a free MSIx IRQ
 * @dd: valid devdata
 * @arg: context information for the IRQ
 * @handler: IRQ handler
 * @thread: IRQ thread handler (could be NULL)
 * @type: affinty IRQ type
 * @name: IRQ name
 *
 * Allocated an MSIx vector if available, and then create the appropriate
 * meta data needed to keep track of the pci IRQ request.
 *
 * Return:
 *   < 0   Error
 *   >= 0  MSIx vector
 *
 */
static int msix_request_irq(struct hfi1_devdata *dd, void *arg,
                            irq_handler_t handler, irq_handler_t thread,
                            enum irq_type type, const char *name)
{
        unsigned long nr;
        int irq;
        int ret;
        struct hfi1_msix_entry *me;

        /* Allocate an MSIx vector */
        spin_lock(&dd->msix_info.msix_lock);
        nr = find_first_zero_bit(dd->msix_info.in_use_msix,
                                 dd->msix_info.max_requested);
        if (nr < dd->msix_info.max_requested)
                __set_bit(nr, dd->msix_info.in_use_msix);
        spin_unlock(&dd->msix_info.msix_lock);

        if (nr == dd->msix_info.max_requested)
                return -ENOSPC;

        if (type < IRQ_SDMA || type >= IRQ_OTHER)
                return -EINVAL;

        irq = pci_irq_vector(dd->pcidev, nr);
        ret = pci_request_irq(dd->pcidev, nr, handler, thread, arg, name);
        if (ret) {
                dd_dev_err(dd,
                           "%s: request for IRQ %d failed, MSIx %lx, err %d\n",
                           name, irq, nr, ret);
                spin_lock(&dd->msix_info.msix_lock);
                __clear_bit(nr, dd->msix_info.in_use_msix);
                spin_unlock(&dd->msix_info.msix_lock);
                return ret;
        }

        /*
         * assign arg after pci_request_irq call, so it will be
         * cleaned up
         */
        me = &dd->msix_info.msix_entries[nr];
        me->irq = irq;
        me->arg = arg;
        me->type = type;

        /* This is a request, so a failure is not fatal */
        ret = hfi1_get_irq_affinity(dd, me);
        if (ret)
                dd_dev_err(dd, "%s: unable to pin IRQ %d\n", name, ret);

        return nr;
}

static int msix_request_rcd_irq_common(struct hfi1_ctxtdata *rcd,
                                       irq_handler_t handler,
                                       irq_handler_t thread,
                                       const char *name)
{
        int nr = msix_request_irq(rcd->dd, rcd, handler, thread,
                                  rcd->is_vnic ? IRQ_NETDEVCTXT : IRQ_RCVCTXT,
                                  name);
        if (nr < 0)
                return nr;

        /*
         * Set the interrupt register and mask for this
         * context's interrupt.
         */
        rcd->ireg = (IS_RCVAVAIL_START + rcd->ctxt) / 64;
        rcd->imask = ((u64)1) << ((IS_RCVAVAIL_START + rcd->ctxt) % 64);
        rcd->msix_intr = nr;
        remap_intr(rcd->dd, IS_RCVAVAIL_START + rcd->ctxt, nr);

        return 0;
}

/**
 * msix_request_rcd_irq() - Helper function for RCVAVAIL IRQs
 * @rcd: valid rcd context
 *
 */
int msix_request_rcd_irq(struct hfi1_ctxtdata *rcd)
{
        char name[MAX_NAME_SIZE];

        snprintf(name, sizeof(name), DRIVER_NAME "_%d kctxt%d",
                 rcd->dd->unit, rcd->ctxt);

        return msix_request_rcd_irq_common(rcd, receive_context_interrupt,
                                           receive_context_thread, name);
}

/**
 * msix_netdev_request_rcd_irq  - Helper function for RCVAVAIL IRQs
 * for netdev context
 * @rcd: valid netdev contexti
 */
int msix_netdev_request_rcd_irq(struct hfi1_ctxtdata *rcd)
{
        char name[MAX_NAME_SIZE];

        snprintf(name, sizeof(name), DRIVER_NAME "_%d nd kctxt%d",
                 rcd->dd->unit, rcd->ctxt);
        return msix_request_rcd_irq_common(rcd, receive_context_interrupt_napi,
                                           NULL, name);
}

/**
 * msix_request_sdma_irq  - Helper for getting SDMA IRQ resources
 * @sde: valid sdma engine
 *
 */
int msix_request_sdma_irq(struct sdma_engine *sde)
{
        int nr;
        char name[MAX_NAME_SIZE];

        snprintf(name, sizeof(name), DRIVER_NAME "_%d sdma%d",
                 sde->dd->unit, sde->this_idx);
        nr = msix_request_irq(sde->dd, sde, sdma_interrupt, NULL,
                              IRQ_SDMA, name);
        if (nr < 0)
                return nr;
        sde->msix_intr = nr;
        remap_sdma_interrupts(sde->dd, sde->this_idx, nr);

        return 0;
}

/**
 * msix_request_general_irq - Helper for getting general IRQ
 * resources
 * @dd: valid device data
 */
int msix_request_general_irq(struct hfi1_devdata *dd)
{
        int nr;
        char name[MAX_NAME_SIZE];

        snprintf(name, sizeof(name), DRIVER_NAME "_%d", dd->unit);
        nr = msix_request_irq(dd, dd, general_interrupt, NULL, IRQ_GENERAL,
                              name);
        if (nr < 0)
                return nr;

        /* general interrupt must be MSIx vector 0 */
        if (nr) {
                msix_free_irq(dd, (u8)nr);
                dd_dev_err(dd, "Invalid index %d for GENERAL IRQ\n", nr);
                return -EINVAL;
        }

        return 0;
}

/**
 * enable_sdma_srcs - Helper to enable SDMA IRQ srcs
 * @dd: valid devdata structure
 * @i: index of SDMA engine
 */
static void enable_sdma_srcs(struct hfi1_devdata *dd, int i)
{
        set_intr_bits(dd, IS_SDMA_START + i, IS_SDMA_START + i, true);
        set_intr_bits(dd, IS_SDMA_PROGRESS_START + i,
                      IS_SDMA_PROGRESS_START + i, true);
        set_intr_bits(dd, IS_SDMA_IDLE_START + i, IS_SDMA_IDLE_START + i, true);
        set_intr_bits(dd, IS_SDMAENG_ERR_START + i, IS_SDMAENG_ERR_START + i,
                      true);
}

/**
 * msix_request_irqs() - Allocate all MSIx IRQs
 * @dd: valid devdata structure
 *
 * Helper function to request the used MSIx IRQs.
 *
 */
int msix_request_irqs(struct hfi1_devdata *dd)
{
        int i;
        int ret = msix_request_general_irq(dd);

        if (ret)
                return ret;

        for (i = 0; i < dd->num_sdma; i++) {
                struct sdma_engine *sde = &dd->per_sdma[i];

                ret = msix_request_sdma_irq(sde);
                if (ret)
                        return ret;
                enable_sdma_srcs(sde->dd, i);
        }

        for (i = 0; i < dd->n_krcv_queues; i++) {
                struct hfi1_ctxtdata *rcd = hfi1_rcd_get_by_index_safe(dd, i);

                if (rcd)
                        ret = msix_request_rcd_irq(rcd);
                hfi1_rcd_put(rcd);
                if (ret)
                        return ret;
        }

        return 0;
}

/**
 * msix_free_irq() - Free the specified MSIx resources and IRQ
 * @dd: valid devdata
 * @msix_intr: MSIx vector to free.
 *
 */
void msix_free_irq(struct hfi1_devdata *dd, u8 msix_intr)
{
        struct hfi1_msix_entry *me;

        if (msix_intr >= dd->msix_info.max_requested)
                return;

        me = &dd->msix_info.msix_entries[msix_intr];

        if (!me->arg) /* => no irq, no affinity */
                return;

        hfi1_put_irq_affinity(dd, me);
        pci_free_irq(dd->pcidev, msix_intr, me->arg);

        me->arg = NULL;

        spin_lock(&dd->msix_info.msix_lock);
        __clear_bit(msix_intr, dd->msix_info.in_use_msix);
        spin_unlock(&dd->msix_info.msix_lock);
}

/**
 * msix_clean_up_interrupts  - Free all MSIx IRQ resources
 * @dd: valid device data data structure
 *
 * Free the MSIx and associated PCI resources, if they have been allocated.
 */
void msix_clean_up_interrupts(struct hfi1_devdata *dd)
{
        int i;
        struct hfi1_msix_entry *me = dd->msix_info.msix_entries;

        /* remove irqs - must happen before disabling/turning off */
        for (i = 0; i < dd->msix_info.max_requested; i++, me++)
                msix_free_irq(dd, i);

        /* clean structures */
        kfree(dd->msix_info.msix_entries);
        dd->msix_info.msix_entries = NULL;
        dd->msix_info.max_requested = 0;

        pci_free_irq_vectors(dd->pcidev);
}

/**
 * msix_netdev_synchronize_irq - netdev IRQ synchronize
 * @dd: valid devdata
 */
void msix_netdev_synchronize_irq(struct hfi1_devdata *dd)
{
        int i;
        int ctxt_count = hfi1_netdev_ctxt_count(dd);

        for (i = 0; i < ctxt_count; i++) {
                struct hfi1_ctxtdata *rcd = hfi1_netdev_get_ctxt(dd, i);
                struct hfi1_msix_entry *me;

                me = &dd->msix_info.msix_entries[rcd->msix_intr];

                synchronize_irq(me->irq);
        }
}