// SPDX-License-Identifier: GPL-2.0
// Copyright (C) 2022 Microchip Technology Inc.

#include <linux/mfd/core.h>
#include <linux/module.h>
#include <linux/pci.h>
#include <linux/spinlock.h>
#include <linux/gpio/driver.h>
#include <linux/interrupt.h>
#include <linux/io.h>
#include <linux/idr.h>
#include "mchp_pci1xxxx_gp.h"

struct aux_bus_device {
        struct auxiliary_device_wrapper *aux_device_wrapper[2];
};

static DEFINE_IDA(gp_client_ida);
static const char aux_dev_otp_e2p_name[15] = "gp_otp_e2p";
static const char aux_dev_gpio_name[15] = "gp_gpio";

static void gp_auxiliary_device_release(struct device *dev)
{
        struct auxiliary_device_wrapper *aux_device_wrapper =
                (struct auxiliary_device_wrapper *)container_of(dev,
                                struct auxiliary_device_wrapper, aux_dev.dev);

        ida_free(&gp_client_ida, aux_device_wrapper->aux_dev.id);
        kfree(aux_device_wrapper);
}

static int gp_aux_bus_probe(struct pci_dev *pdev, const struct pci_device_id *id)
{
        struct aux_bus_device *aux_bus;
        int retval;

        retval = pcim_enable_device(pdev);
        if (retval)
                return retval;

        aux_bus = devm_kzalloc(&pdev->dev, sizeof(*aux_bus), GFP_KERNEL);
        if (!aux_bus)
                return -ENOMEM;

        aux_bus->aux_device_wrapper[0] = kzalloc_obj(*aux_bus->aux_device_wrapper[0]);
        if (!aux_bus->aux_device_wrapper[0])
                return -ENOMEM;

        retval = ida_alloc(&gp_client_ida, GFP_KERNEL);
        if (retval < 0)
                goto err_ida_alloc_0;

        aux_bus->aux_device_wrapper[0]->aux_dev.name = aux_dev_otp_e2p_name;
        aux_bus->aux_device_wrapper[0]->aux_dev.dev.parent = &pdev->dev;
        aux_bus->aux_device_wrapper[0]->aux_dev.dev.release = gp_auxiliary_device_release;
        aux_bus->aux_device_wrapper[0]->aux_dev.id = retval;

        aux_bus->aux_device_wrapper[0]->gp_aux_data.region_start = pci_resource_start(pdev, 0);
        aux_bus->aux_device_wrapper[0]->gp_aux_data.region_length = pci_resource_end(pdev, 0);

        retval = auxiliary_device_init(&aux_bus->aux_device_wrapper[0]->aux_dev);
        if (retval < 0)
                goto err_aux_dev_init_0;

        retval = auxiliary_device_add(&aux_bus->aux_device_wrapper[0]->aux_dev);
        if (retval)
                goto err_aux_dev_add_0;

        aux_bus->aux_device_wrapper[1] = kzalloc_obj(*aux_bus->aux_device_wrapper[1]);
        if (!aux_bus->aux_device_wrapper[1]) {
                retval =  -ENOMEM;
                goto err_aux_dev_add_0;
        }

        retval = ida_alloc(&gp_client_ida, GFP_KERNEL);
        if (retval < 0)
                goto err_ida_alloc_1;

        aux_bus->aux_device_wrapper[1]->aux_dev.name = aux_dev_gpio_name;
        aux_bus->aux_device_wrapper[1]->aux_dev.dev.parent = &pdev->dev;
        aux_bus->aux_device_wrapper[1]->aux_dev.dev.release = gp_auxiliary_device_release;
        aux_bus->aux_device_wrapper[1]->aux_dev.id = retval;

        aux_bus->aux_device_wrapper[1]->gp_aux_data.region_start = pci_resource_start(pdev, 0);
        aux_bus->aux_device_wrapper[1]->gp_aux_data.region_length = pci_resource_end(pdev, 0);

        retval = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_ALL_TYPES);

        if (retval < 0)
                goto err_aux_dev_init_1;

        retval = pci_irq_vector(pdev, 0);
        if (retval < 0)
                goto err_aux_dev_init_1;

        pdev->irq = retval;
        aux_bus->aux_device_wrapper[1]->gp_aux_data.irq_num = pdev->irq;

        retval = auxiliary_device_init(&aux_bus->aux_device_wrapper[1]->aux_dev);
        if (retval < 0)
                goto err_aux_dev_init_1;

        retval = auxiliary_device_add(&aux_bus->aux_device_wrapper[1]->aux_dev);
        if (retval)
                goto err_aux_dev_add_1;

        pci_set_drvdata(pdev, aux_bus);
        pci_set_master(pdev);

        return 0;

err_aux_dev_add_1:
        auxiliary_device_uninit(&aux_bus->aux_device_wrapper[1]->aux_dev);
        goto err_aux_dev_add_0;

err_aux_dev_init_1:
        ida_free(&gp_client_ida, aux_bus->aux_device_wrapper[1]->aux_dev.id);

err_ida_alloc_1:
        kfree(aux_bus->aux_device_wrapper[1]);

err_aux_dev_add_0:
        auxiliary_device_uninit(&aux_bus->aux_device_wrapper[0]->aux_dev);
        goto err_ret;

err_aux_dev_init_0:
        ida_free(&gp_client_ida, aux_bus->aux_device_wrapper[0]->aux_dev.id);

err_ida_alloc_0:
        kfree(aux_bus->aux_device_wrapper[0]);

err_ret:
        return retval;
}

static void gp_aux_bus_remove(struct pci_dev *pdev)
{
        struct aux_bus_device *aux_bus = pci_get_drvdata(pdev);

        auxiliary_device_delete(&aux_bus->aux_device_wrapper[0]->aux_dev);
        auxiliary_device_uninit(&aux_bus->aux_device_wrapper[0]->aux_dev);
        auxiliary_device_delete(&aux_bus->aux_device_wrapper[1]->aux_dev);
        auxiliary_device_uninit(&aux_bus->aux_device_wrapper[1]->aux_dev);
}

static const struct pci_device_id pci1xxxx_tbl[] = {
        { PCI_DEVICE(0x1055, 0xA005) },
        { PCI_DEVICE(0x1055, 0xA015) },
        { PCI_DEVICE(0x1055, 0xA025) },
        { PCI_DEVICE(0x1055, 0xA035) },
        { PCI_DEVICE(0x1055, 0xA045) },
        { PCI_DEVICE(0x1055, 0xA055) },
        {0,}
};
MODULE_DEVICE_TABLE(pci, pci1xxxx_tbl);

static struct pci_driver pci1xxxx_gp_driver = {
        .name = "PCI1xxxxGP",
        .id_table = pci1xxxx_tbl,
        .probe = gp_aux_bus_probe,
        .remove = gp_aux_bus_remove,
};

module_pci_driver(pci1xxxx_gp_driver);

MODULE_DESCRIPTION("Microchip Technology Inc. PCI1xxxx GP expander");
MODULE_AUTHOR("Kumaravel Thiagarajan <kumaravel.thiagarajan@microchip.com>");
MODULE_LICENSE("GPL");