162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only OR BSD-3-Clause
262306a36Sopenharmony_ci/* Copyright (C) 2022 NVIDIA CORPORATION & AFFILIATES */
362306a36Sopenharmony_ci
462306a36Sopenharmony_ci#include <linux/bitfield.h>
562306a36Sopenharmony_ci#include <linux/bitops.h>
662306a36Sopenharmony_ci#include <linux/device.h>
762306a36Sopenharmony_ci#include <linux/err.h>
862306a36Sopenharmony_ci#include <linux/gpio/driver.h>
962306a36Sopenharmony_ci#include <linux/interrupt.h>
1062306a36Sopenharmony_ci#include <linux/io.h>
1162306a36Sopenharmony_ci#include <linux/module.h>
1262306a36Sopenharmony_ci#include <linux/platform_device.h>
1362306a36Sopenharmony_ci#include <linux/spinlock.h>
1462306a36Sopenharmony_ci#include <linux/types.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci/*
1762306a36Sopenharmony_ci * There are 2 YU GPIO blocks:
1862306a36Sopenharmony_ci * gpio[0]: HOST_GPIO0->HOST_GPIO31
1962306a36Sopenharmony_ci * gpio[1]: HOST_GPIO32->HOST_GPIO55
2062306a36Sopenharmony_ci */
2162306a36Sopenharmony_ci#define MLXBF3_GPIO_MAX_PINS_PER_BLOCK 32
2262306a36Sopenharmony_ci#define MLXBF3_GPIO_MAX_PINS_BLOCK0    32
2362306a36Sopenharmony_ci#define MLXBF3_GPIO_MAX_PINS_BLOCK1    24
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci/*
2662306a36Sopenharmony_ci * fw_gpio[x] block registers and their offset
2762306a36Sopenharmony_ci */
2862306a36Sopenharmony_ci#define MLXBF_GPIO_FW_OUTPUT_ENABLE_SET	  0x00
2962306a36Sopenharmony_ci#define MLXBF_GPIO_FW_DATA_OUT_SET        0x04
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci#define MLXBF_GPIO_FW_OUTPUT_ENABLE_CLEAR 0x00
3262306a36Sopenharmony_ci#define MLXBF_GPIO_FW_DATA_OUT_CLEAR      0x04
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci#define MLXBF_GPIO_CAUSE_RISE_EN          0x00
3562306a36Sopenharmony_ci#define MLXBF_GPIO_CAUSE_FALL_EN          0x04
3662306a36Sopenharmony_ci#define MLXBF_GPIO_READ_DATA_IN           0x08
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci#define MLXBF_GPIO_CAUSE_OR_CAUSE_EVTEN0  0x00
3962306a36Sopenharmony_ci#define MLXBF_GPIO_CAUSE_OR_EVTEN0        0x14
4062306a36Sopenharmony_ci#define MLXBF_GPIO_CAUSE_OR_CLRCAUSE      0x18
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_cistruct mlxbf3_gpio_context {
4362306a36Sopenharmony_ci	struct gpio_chip gc;
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	/* YU GPIO block address */
4662306a36Sopenharmony_ci	void __iomem *gpio_set_io;
4762306a36Sopenharmony_ci	void __iomem *gpio_clr_io;
4862306a36Sopenharmony_ci	void __iomem *gpio_io;
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	/* YU GPIO cause block address */
5162306a36Sopenharmony_ci	void __iomem *gpio_cause_io;
5262306a36Sopenharmony_ci};
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_cistatic void mlxbf3_gpio_irq_enable(struct irq_data *irqd)
5562306a36Sopenharmony_ci{
5662306a36Sopenharmony_ci	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
5762306a36Sopenharmony_ci	struct mlxbf3_gpio_context *gs = gpiochip_get_data(gc);
5862306a36Sopenharmony_ci	irq_hw_number_t offset = irqd_to_hwirq(irqd);
5962306a36Sopenharmony_ci	unsigned long flags;
6062306a36Sopenharmony_ci	u32 val;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	gpiochip_enable_irq(gc, offset);
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
6562306a36Sopenharmony_ci	writel(BIT(offset), gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_CLRCAUSE);
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	val = readl(gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_EVTEN0);
6862306a36Sopenharmony_ci	val |= BIT(offset);
6962306a36Sopenharmony_ci	writel(val, gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_EVTEN0);
7062306a36Sopenharmony_ci	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
7162306a36Sopenharmony_ci}
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_cistatic void mlxbf3_gpio_irq_disable(struct irq_data *irqd)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
7662306a36Sopenharmony_ci	struct mlxbf3_gpio_context *gs = gpiochip_get_data(gc);
7762306a36Sopenharmony_ci	irq_hw_number_t offset = irqd_to_hwirq(irqd);
7862306a36Sopenharmony_ci	unsigned long flags;
7962306a36Sopenharmony_ci	u32 val;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
8262306a36Sopenharmony_ci	val = readl(gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_EVTEN0);
8362306a36Sopenharmony_ci	val &= ~BIT(offset);
8462306a36Sopenharmony_ci	writel(val, gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_EVTEN0);
8562306a36Sopenharmony_ci	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	gpiochip_disable_irq(gc, offset);
8862306a36Sopenharmony_ci}
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_cistatic irqreturn_t mlxbf3_gpio_irq_handler(int irq, void *ptr)
9162306a36Sopenharmony_ci{
9262306a36Sopenharmony_ci	struct mlxbf3_gpio_context *gs = ptr;
9362306a36Sopenharmony_ci	struct gpio_chip *gc = &gs->gc;
9462306a36Sopenharmony_ci	unsigned long pending;
9562306a36Sopenharmony_ci	u32 level;
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	pending = readl(gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_CAUSE_EVTEN0);
9862306a36Sopenharmony_ci	writel(pending, gs->gpio_cause_io + MLXBF_GPIO_CAUSE_OR_CLRCAUSE);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	for_each_set_bit(level, &pending, gc->ngpio)
10162306a36Sopenharmony_ci		generic_handle_domain_irq(gc->irq.domain, level);
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci	return IRQ_RETVAL(pending);
10462306a36Sopenharmony_ci}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_cistatic int
10762306a36Sopenharmony_cimlxbf3_gpio_irq_set_type(struct irq_data *irqd, unsigned int type)
10862306a36Sopenharmony_ci{
10962306a36Sopenharmony_ci	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
11062306a36Sopenharmony_ci	struct mlxbf3_gpio_context *gs = gpiochip_get_data(gc);
11162306a36Sopenharmony_ci	irq_hw_number_t offset = irqd_to_hwirq(irqd);
11262306a36Sopenharmony_ci	unsigned long flags;
11362306a36Sopenharmony_ci	u32 val;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	switch (type & IRQ_TYPE_SENSE_MASK) {
11862306a36Sopenharmony_ci	case IRQ_TYPE_EDGE_BOTH:
11962306a36Sopenharmony_ci		val = readl(gs->gpio_io + MLXBF_GPIO_CAUSE_FALL_EN);
12062306a36Sopenharmony_ci		val |= BIT(offset);
12162306a36Sopenharmony_ci		writel(val, gs->gpio_io + MLXBF_GPIO_CAUSE_FALL_EN);
12262306a36Sopenharmony_ci		val = readl(gs->gpio_io + MLXBF_GPIO_CAUSE_RISE_EN);
12362306a36Sopenharmony_ci		val |= BIT(offset);
12462306a36Sopenharmony_ci		writel(val, gs->gpio_io + MLXBF_GPIO_CAUSE_RISE_EN);
12562306a36Sopenharmony_ci		break;
12662306a36Sopenharmony_ci	case IRQ_TYPE_EDGE_RISING:
12762306a36Sopenharmony_ci		val = readl(gs->gpio_io + MLXBF_GPIO_CAUSE_RISE_EN);
12862306a36Sopenharmony_ci		val |= BIT(offset);
12962306a36Sopenharmony_ci		writel(val, gs->gpio_io + MLXBF_GPIO_CAUSE_RISE_EN);
13062306a36Sopenharmony_ci		break;
13162306a36Sopenharmony_ci	case IRQ_TYPE_EDGE_FALLING:
13262306a36Sopenharmony_ci		val = readl(gs->gpio_io + MLXBF_GPIO_CAUSE_FALL_EN);
13362306a36Sopenharmony_ci		val |= BIT(offset);
13462306a36Sopenharmony_ci		writel(val, gs->gpio_io + MLXBF_GPIO_CAUSE_FALL_EN);
13562306a36Sopenharmony_ci		break;
13662306a36Sopenharmony_ci	default:
13762306a36Sopenharmony_ci		raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
13862306a36Sopenharmony_ci		return -EINVAL;
13962306a36Sopenharmony_ci	}
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_ci	irq_set_handler_locked(irqd, handle_edge_irq);
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	return 0;
14662306a36Sopenharmony_ci}
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci/* This function needs to be defined for handle_edge_irq() */
14962306a36Sopenharmony_cistatic void mlxbf3_gpio_irq_ack(struct irq_data *data)
15062306a36Sopenharmony_ci{
15162306a36Sopenharmony_ci}
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_cistatic const struct irq_chip gpio_mlxbf3_irqchip = {
15462306a36Sopenharmony_ci	.name = "MLNXBF33",
15562306a36Sopenharmony_ci	.irq_ack = mlxbf3_gpio_irq_ack,
15662306a36Sopenharmony_ci	.irq_set_type = mlxbf3_gpio_irq_set_type,
15762306a36Sopenharmony_ci	.irq_enable = mlxbf3_gpio_irq_enable,
15862306a36Sopenharmony_ci	.irq_disable = mlxbf3_gpio_irq_disable,
15962306a36Sopenharmony_ci	.flags = IRQCHIP_IMMUTABLE,
16062306a36Sopenharmony_ci	GPIOCHIP_IRQ_RESOURCE_HELPERS,
16162306a36Sopenharmony_ci};
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_cistatic int mlxbf3_gpio_add_pin_ranges(struct gpio_chip *chip)
16462306a36Sopenharmony_ci{
16562306a36Sopenharmony_ci	unsigned int id;
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci	switch(chip->ngpio) {
16862306a36Sopenharmony_ci	case MLXBF3_GPIO_MAX_PINS_BLOCK0:
16962306a36Sopenharmony_ci		id = 0;
17062306a36Sopenharmony_ci		break;
17162306a36Sopenharmony_ci	case MLXBF3_GPIO_MAX_PINS_BLOCK1:
17262306a36Sopenharmony_ci		id = 1;
17362306a36Sopenharmony_ci		break;
17462306a36Sopenharmony_ci	default:
17562306a36Sopenharmony_ci		return -EINVAL;
17662306a36Sopenharmony_ci	}
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	return gpiochip_add_pin_range(chip, "MLNXBF34:00",
17962306a36Sopenharmony_ci			chip->base, id * MLXBF3_GPIO_MAX_PINS_PER_BLOCK,
18062306a36Sopenharmony_ci			chip->ngpio);
18162306a36Sopenharmony_ci}
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_cistatic int mlxbf3_gpio_probe(struct platform_device *pdev)
18462306a36Sopenharmony_ci{
18562306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
18662306a36Sopenharmony_ci	struct mlxbf3_gpio_context *gs;
18762306a36Sopenharmony_ci	struct gpio_irq_chip *girq;
18862306a36Sopenharmony_ci	struct gpio_chip *gc;
18962306a36Sopenharmony_ci	int ret, irq;
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	gs = devm_kzalloc(dev, sizeof(*gs), GFP_KERNEL);
19262306a36Sopenharmony_ci	if (!gs)
19362306a36Sopenharmony_ci		return -ENOMEM;
19462306a36Sopenharmony_ci
19562306a36Sopenharmony_ci	gs->gpio_io = devm_platform_ioremap_resource(pdev, 0);
19662306a36Sopenharmony_ci	if (IS_ERR(gs->gpio_io))
19762306a36Sopenharmony_ci		return PTR_ERR(gs->gpio_io);
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_ci	gs->gpio_cause_io = devm_platform_ioremap_resource(pdev, 1);
20062306a36Sopenharmony_ci	if (IS_ERR(gs->gpio_cause_io))
20162306a36Sopenharmony_ci		return PTR_ERR(gs->gpio_cause_io);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	gs->gpio_set_io = devm_platform_ioremap_resource(pdev, 2);
20462306a36Sopenharmony_ci	if (IS_ERR(gs->gpio_set_io))
20562306a36Sopenharmony_ci		return PTR_ERR(gs->gpio_set_io);
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	gs->gpio_clr_io = devm_platform_ioremap_resource(pdev, 3);
20862306a36Sopenharmony_ci	if (IS_ERR(gs->gpio_clr_io))
20962306a36Sopenharmony_ci		return PTR_ERR(gs->gpio_clr_io);
21062306a36Sopenharmony_ci	gc = &gs->gc;
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_ci	ret = bgpio_init(gc, dev, 4,
21362306a36Sopenharmony_ci			gs->gpio_io + MLXBF_GPIO_READ_DATA_IN,
21462306a36Sopenharmony_ci			gs->gpio_set_io + MLXBF_GPIO_FW_DATA_OUT_SET,
21562306a36Sopenharmony_ci			gs->gpio_clr_io + MLXBF_GPIO_FW_DATA_OUT_CLEAR,
21662306a36Sopenharmony_ci			gs->gpio_set_io + MLXBF_GPIO_FW_OUTPUT_ENABLE_SET,
21762306a36Sopenharmony_ci			gs->gpio_clr_io + MLXBF_GPIO_FW_OUTPUT_ENABLE_CLEAR, 0);
21862306a36Sopenharmony_ci	if (ret)
21962306a36Sopenharmony_ci		return dev_err_probe(dev, ret, "%s: bgpio_init() failed", __func__);
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_ci	gc->request = gpiochip_generic_request;
22262306a36Sopenharmony_ci	gc->free = gpiochip_generic_free;
22362306a36Sopenharmony_ci	gc->owner = THIS_MODULE;
22462306a36Sopenharmony_ci	gc->add_pin_ranges = mlxbf3_gpio_add_pin_ranges;
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	irq = platform_get_irq(pdev, 0);
22762306a36Sopenharmony_ci	if (irq >= 0) {
22862306a36Sopenharmony_ci		girq = &gs->gc.irq;
22962306a36Sopenharmony_ci		gpio_irq_chip_set_chip(girq, &gpio_mlxbf3_irqchip);
23062306a36Sopenharmony_ci		girq->default_type = IRQ_TYPE_NONE;
23162306a36Sopenharmony_ci		/* This will let us handle the parent IRQ in the driver */
23262306a36Sopenharmony_ci		girq->num_parents = 0;
23362306a36Sopenharmony_ci		girq->parents = NULL;
23462306a36Sopenharmony_ci		girq->parent_handler = NULL;
23562306a36Sopenharmony_ci		girq->handler = handle_bad_irq;
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_ci		/*
23862306a36Sopenharmony_ci		 * Directly request the irq here instead of passing
23962306a36Sopenharmony_ci		 * a flow-handler because the irq is shared.
24062306a36Sopenharmony_ci		 */
24162306a36Sopenharmony_ci		ret = devm_request_irq(dev, irq, mlxbf3_gpio_irq_handler,
24262306a36Sopenharmony_ci				       IRQF_SHARED, dev_name(dev), gs);
24362306a36Sopenharmony_ci		if (ret)
24462306a36Sopenharmony_ci			return dev_err_probe(dev, ret, "failed to request IRQ");
24562306a36Sopenharmony_ci	}
24662306a36Sopenharmony_ci
24762306a36Sopenharmony_ci	platform_set_drvdata(pdev, gs);
24862306a36Sopenharmony_ci
24962306a36Sopenharmony_ci	ret = devm_gpiochip_add_data(dev, &gs->gc, gs);
25062306a36Sopenharmony_ci	if (ret)
25162306a36Sopenharmony_ci		dev_err_probe(dev, ret, "Failed adding memory mapped gpiochip\n");
25262306a36Sopenharmony_ci
25362306a36Sopenharmony_ci	return 0;
25462306a36Sopenharmony_ci}
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_cistatic const struct acpi_device_id mlxbf3_gpio_acpi_match[] = {
25762306a36Sopenharmony_ci	{ "MLNXBF33", 0 },
25862306a36Sopenharmony_ci	{}
25962306a36Sopenharmony_ci};
26062306a36Sopenharmony_ciMODULE_DEVICE_TABLE(acpi, mlxbf3_gpio_acpi_match);
26162306a36Sopenharmony_ci
26262306a36Sopenharmony_cistatic struct platform_driver mlxbf3_gpio_driver = {
26362306a36Sopenharmony_ci	.driver = {
26462306a36Sopenharmony_ci		.name = "mlxbf3_gpio",
26562306a36Sopenharmony_ci		.acpi_match_table = mlxbf3_gpio_acpi_match,
26662306a36Sopenharmony_ci	},
26762306a36Sopenharmony_ci	.probe    = mlxbf3_gpio_probe,
26862306a36Sopenharmony_ci};
26962306a36Sopenharmony_cimodule_platform_driver(mlxbf3_gpio_driver);
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ciMODULE_SOFTDEP("pre: pinctrl-mlxbf3");
27262306a36Sopenharmony_ciMODULE_DESCRIPTION("NVIDIA BlueField-3 GPIO Driver");
27362306a36Sopenharmony_ciMODULE_AUTHOR("Asmaa Mnebhi <asmaa@nvidia.com>");
27462306a36Sopenharmony_ciMODULE_LICENSE("Dual BSD/GPL");
275