// SPDX-License-Identifier: GPL-2.0
/*
 * MSI framework for platform devices
 *
 * Copyright (C) 2015 ARM Limited, All Rights Reserved.
 * Author: Marc Zyngier <marc.zyngier@arm.com>
 * Copyright (C) 2022 Linutronix GmbH
 */

#include <linux/device.h>
#include <linux/irqdomain.h>
#include <linux/msi.h>

/*
 * This indirection can go when platform_device_msi_init_and_alloc_irqs()
 * is switched to a proper irq_chip::irq_write_msi_msg() callback. Keep it
 * simple for now.
 */
static void platform_msi_write_msi_msg(struct irq_data *d, struct msi_msg *msg)
{
        irq_write_msi_msg_t cb = d->chip_data;

        cb(irq_data_get_msi_desc(d), msg);
}

static void platform_msi_set_desc(msi_alloc_info_t *arg, struct msi_desc *desc)
{
        arg->desc = desc;
        arg->hwirq = desc->msi_index;
}

static const struct msi_domain_template platform_msi_template = {
        .chip = {
                .name                   = "pMSI",
                .irq_mask               = irq_chip_mask_parent,
                .irq_unmask             = irq_chip_unmask_parent,
                .irq_write_msi_msg      = platform_msi_write_msi_msg,
                /* The rest is filled in by the platform MSI parent */
        },

        .ops = {
                .set_desc               = platform_msi_set_desc,
        },

        .info = {
                .bus_token              = DOMAIN_BUS_DEVICE_MSI,
        },
};

/**
 * platform_device_msi_init_and_alloc_irqs - Initialize platform device MSI
 *                                           and allocate interrupts for @dev
 * @dev:                The device for which to allocate interrupts
 * @nvec:               The number of interrupts to allocate
 * @write_msi_msg:      Callback to write an interrupt message for @dev
 *
 * Returns:
 * Zero for success, or an error code in case of failure
 *
 * This creates a MSI domain on @dev which has @dev->msi.domain as
 * parent. The parent domain sets up the new domain. The domain has
 * a fixed size of @nvec. The domain is managed by devres and will
 * be removed when the device is removed.
 */
int platform_device_msi_init_and_alloc_irqs(struct device *dev, unsigned int nvec,
                                            irq_write_msi_msg_t write_msi_msg)
{
        struct irq_domain *domain = dev->msi.domain;

        if (!domain || !write_msi_msg)
                return -EINVAL;

        /*
         * @write_msi_msg is stored in the resulting msi_domain_info::data.
         * The underlying domain creation mechanism will assign that
         * callback to the resulting irq chip.
         */
        if (!msi_create_device_irq_domain(dev, MSI_DEFAULT_DOMAIN,
                                          &platform_msi_template,
                                          nvec, NULL, write_msi_msg))
                return -ENODEV;

        return msi_domain_alloc_irqs_range(dev, MSI_DEFAULT_DOMAIN, 0, nvec - 1);
}
EXPORT_SYMBOL_GPL(platform_device_msi_init_and_alloc_irqs);

/**
 * platform_device_msi_free_irqs_all - Free all interrupts for @dev
 * @dev:        The device for which to free interrupts
 */
void platform_device_msi_free_irqs_all(struct device *dev)
{
        msi_domain_free_irqs_all(dev, MSI_DEFAULT_DOMAIN);
        msi_remove_device_irq_domain(dev, MSI_DEFAULT_DOMAIN);
}
EXPORT_SYMBOL_GPL(platform_device_msi_free_irqs_all);