// SPDX-License-Identifier: GPL-2.0-only
/*
 * Intel Elkhart Lake Programmable Service Engine (PSE) I/O
 *
 * Copyright (c) 2025 Intel Corporation.
 *
 * Author: Raag Jadav <raag.jadav@intel.com>
 */

#include <linux/auxiliary_bus.h>
#include <linux/device/devres.h>
#include <linux/errno.h>
#include <linux/gfp_types.h>
#include <linux/ioport.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/pci.h>
#include <linux/sizes.h>
#include <linux/types.h>

#include <linux/ehl_pse_io_aux.h>

#define EHL_PSE_IO_DEV_SIZE	SZ_4K

static int ehl_pse_io_dev_create(struct pci_dev *pci, const char *name, int idx)
{
	struct device *dev = &pci->dev;
	struct auxiliary_device *adev;
	struct ehl_pse_io_data *data;
	resource_size_t start, offset;
	u32 id;

	data = devm_kzalloc(dev, sizeof(*data), GFP_KERNEL);
	if (!data)
		return -ENOMEM;

	id = (pci_domain_nr(pci->bus) << 16) | pci_dev_id(pci);
	start = pci_resource_start(pci, 0);
	offset = EHL_PSE_IO_DEV_SIZE * idx;

	data->mem = DEFINE_RES_MEM(start + offset, EHL_PSE_IO_DEV_SIZE);
	data->irq = pci_irq_vector(pci, idx);

	adev = __devm_auxiliary_device_create(dev, EHL_PSE_IO_NAME, name, data, id);

	return adev ? 0 : -ENODEV;
}

static int ehl_pse_io_probe(struct pci_dev *pci, const struct pci_device_id *id)
{
	int ret;

	ret = pcim_enable_device(pci);
	if (ret)
		return ret;

	pci_set_master(pci);

	ret = pci_alloc_irq_vectors(pci, 2, 2, PCI_IRQ_MSI);
	if (ret < 0)
		return ret;

	ret = ehl_pse_io_dev_create(pci, EHL_PSE_GPIO_NAME, 0);
	if (ret)
		return ret;

	return ehl_pse_io_dev_create(pci, EHL_PSE_TIO_NAME, 1);
}

static const struct pci_device_id ehl_pse_io_ids[] = {
	{ PCI_VDEVICE(INTEL, 0x4b88) },
	{ PCI_VDEVICE(INTEL, 0x4b89) },
	{ }
};
MODULE_DEVICE_TABLE(pci, ehl_pse_io_ids);

static struct pci_driver ehl_pse_io_driver = {
	.name		= EHL_PSE_IO_NAME,
	.id_table	= ehl_pse_io_ids,
	.probe		= ehl_pse_io_probe,
};
module_pci_driver(ehl_pse_io_driver);

MODULE_AUTHOR("Raag Jadav <raag.jadav@intel.com>");
MODULE_DESCRIPTION("Intel Elkhart Lake PSE I/O driver");
MODULE_LICENSE("GPL");