// SPDX-License-Identifier: GPL-2.0+
// Copyright 2017 IBM Corp.
#include <linux/interrupt.h>
#include <asm/pnv-ocxl.h>
#include <asm/xive.h>
#include "ocxl_internal.h"
#include "trace.h"

struct afu_irq {
        int id;
        int hw_irq;
        unsigned int virq;
        char *name;
        irqreturn_t (*handler)(void *private);
        void (*free_private)(void *private);
        void *private;
};

int ocxl_irq_offset_to_id(struct ocxl_context *ctx, u64 offset)
{
        return (offset - ctx->afu->irq_base_offset) >> PAGE_SHIFT;
}

u64 ocxl_irq_id_to_offset(struct ocxl_context *ctx, int irq_id)
{
        return ctx->afu->irq_base_offset + (irq_id << PAGE_SHIFT);
}

int ocxl_irq_set_handler(struct ocxl_context *ctx, int irq_id,
                irqreturn_t (*handler)(void *private),
                void (*free_private)(void *private),
                void *private)
{
        struct afu_irq *irq;
        int rc;

        mutex_lock(&ctx->irq_lock);
        irq = idr_find(&ctx->irq_idr, irq_id);
        if (!irq) {
                rc = -EINVAL;
                goto unlock;
        }

        irq->handler = handler;
        irq->private = private;
        irq->free_private = free_private;

        rc = 0;
        // Fall through to unlock

unlock:
        mutex_unlock(&ctx->irq_lock);
        return rc;
}
EXPORT_SYMBOL_GPL(ocxl_irq_set_handler);

static irqreturn_t afu_irq_handler(int virq, void *data)
{
        struct afu_irq *irq = (struct afu_irq *) data;

        trace_ocxl_afu_irq_receive(virq);

        if (irq->handler)
                return irq->handler(irq->private);

        return IRQ_HANDLED; // Just drop it on the ground
}

static int setup_afu_irq(struct ocxl_context *ctx, struct afu_irq *irq)
{
        int rc;

        irq->virq = irq_create_mapping(NULL, irq->hw_irq);
        if (!irq->virq) {
                pr_err("irq_create_mapping failed\n");
                return -ENOMEM;
        }
        pr_debug("hw_irq %d mapped to virq %u\n", irq->hw_irq, irq->virq);

        irq->name = kasprintf(GFP_KERNEL, "ocxl-afu-%u", irq->virq);
        if (!irq->name) {
                irq_dispose_mapping(irq->virq);
                return -ENOMEM;
        }

        rc = request_irq(irq->virq, afu_irq_handler, 0, irq->name, irq);
        if (rc) {
                kfree(irq->name);
                irq->name = NULL;
                irq_dispose_mapping(irq->virq);
                pr_err("request_irq failed: %d\n", rc);
                return rc;
        }
        return 0;
}

static void release_afu_irq(struct afu_irq *irq)
{
        free_irq(irq->virq, irq);
        irq_dispose_mapping(irq->virq);
        kfree(irq->name);
}

int ocxl_afu_irq_alloc(struct ocxl_context *ctx, int *irq_id)
{
        struct afu_irq *irq;
        int rc;

        irq = kzalloc(sizeof(struct afu_irq), GFP_KERNEL);
        if (!irq)
                return -ENOMEM;

        /*
         * We limit the number of afu irqs per context and per link to
         * avoid a single process or user depleting the pool of IPIs
         */

        mutex_lock(&ctx->irq_lock);

        irq->id = idr_alloc(&ctx->irq_idr, irq, 0, MAX_IRQ_PER_CONTEXT,
                        GFP_KERNEL);
        if (irq->id < 0) {
                rc = -ENOSPC;
                goto err_unlock;
        }

        rc = ocxl_link_irq_alloc(ctx->afu->fn->link, &irq->hw_irq);
        if (rc)
                goto err_idr;

        rc = setup_afu_irq(ctx, irq);
        if (rc)
                goto err_alloc;

        trace_ocxl_afu_irq_alloc(ctx->pasid, irq->id, irq->virq, irq->hw_irq);
        mutex_unlock(&ctx->irq_lock);

        *irq_id = irq->id;

        return 0;

err_alloc:
        ocxl_link_free_irq(ctx->afu->fn->link, irq->hw_irq);
err_idr:
        idr_remove(&ctx->irq_idr, irq->id);
err_unlock:
        mutex_unlock(&ctx->irq_lock);
        kfree(irq);
        return rc;
}
EXPORT_SYMBOL_GPL(ocxl_afu_irq_alloc);

static void afu_irq_free(struct afu_irq *irq, struct ocxl_context *ctx)
{
        trace_ocxl_afu_irq_free(ctx->pasid, irq->id);
        if (ctx->mapping)
                unmap_mapping_range(ctx->mapping,
                                ocxl_irq_id_to_offset(ctx, irq->id),
                                1 << PAGE_SHIFT, 1);
        release_afu_irq(irq);
        if (irq->free_private)
                irq->free_private(irq->private);
        ocxl_link_free_irq(ctx->afu->fn->link, irq->hw_irq);
        kfree(irq);
}

int ocxl_afu_irq_free(struct ocxl_context *ctx, int irq_id)
{
        struct afu_irq *irq;

        mutex_lock(&ctx->irq_lock);

        irq = idr_find(&ctx->irq_idr, irq_id);
        if (!irq) {
                mutex_unlock(&ctx->irq_lock);
                return -EINVAL;
        }
        idr_remove(&ctx->irq_idr, irq->id);
        afu_irq_free(irq, ctx);
        mutex_unlock(&ctx->irq_lock);
        return 0;
}
EXPORT_SYMBOL_GPL(ocxl_afu_irq_free);

void ocxl_afu_irq_free_all(struct ocxl_context *ctx)
{
        struct afu_irq *irq;
        int id;

        mutex_lock(&ctx->irq_lock);
        idr_for_each_entry(&ctx->irq_idr, irq, id)
                afu_irq_free(irq, ctx);
        mutex_unlock(&ctx->irq_lock);
}

u64 ocxl_afu_irq_get_addr(struct ocxl_context *ctx, int irq_id)
{
        struct xive_irq_data *xd;
        struct afu_irq *irq;
        u64 addr = 0;

        mutex_lock(&ctx->irq_lock);
        irq = idr_find(&ctx->irq_idr, irq_id);
        if (irq) {
                xd = irq_get_handler_data(irq->virq);
                addr = xd ? xd->trig_page : 0;
        }
        mutex_unlock(&ctx->irq_lock);
        return addr;
}
EXPORT_SYMBOL_GPL(ocxl_afu_irq_get_addr);