[linux-2.6-block.git] / drivers / gpio / gpio-mlxbf2.c

// SPDX-License-Identifier: GPL-2.0

/*
 * Copyright (C) 2020-2021 NVIDIA CORPORATION & AFFILIATES
 */

#include <linux/bitfield.h>
#include <linux/bitops.h>
#include <linux/device.h>
#include <linux/gpio/driver.h>
#include <linux/interrupt.h>
#include <linux/io.h>
#include <linux/ioport.h>
#include <linux/kernel.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/pm.h>
#include <linux/resource.h>
#include <linux/seq_file.h>
#include <linux/spinlock.h>
#include <linux/types.h>

/*
 * There are 3 YU GPIO blocks:
 * gpio[0]: HOST_GPIO0->HOST_GPIO31
 * gpio[1]: HOST_GPIO32->HOST_GPIO63
 * gpio[2]: HOST_GPIO64->HOST_GPIO69
 */
#define MLXBF2_GPIO_MAX_PINS_PER_BLOCK 32

/*
 * arm_gpio_lock register:
 * bit[31]	lock status: active if set
 * bit[15:0]	set lock
 * The lock is enabled only if 0xd42f is written to this field
 */
#define YU_ARM_GPIO_LOCK_ADDR		0x2801088
#define YU_ARM_GPIO_LOCK_SIZE		0x8
#define YU_LOCK_ACTIVE_BIT(val)		(val >> 31)
#define YU_ARM_GPIO_LOCK_ACQUIRE	0xd42f
#define YU_ARM_GPIO_LOCK_RELEASE	0x0

/*
 * gpio[x] block registers and their offset
 */
#define YU_GPIO_DATAIN			0x04
#define YU_GPIO_MODE1			0x08
#define YU_GPIO_MODE0			0x0c
#define YU_GPIO_DATASET			0x14
#define YU_GPIO_DATACLEAR		0x18
#define YU_GPIO_CAUSE_RISE_EN		0x44
#define YU_GPIO_CAUSE_FALL_EN		0x48
#define YU_GPIO_MODE1_CLEAR		0x50
#define YU_GPIO_MODE0_SET		0x54
#define YU_GPIO_MODE0_CLEAR		0x58
#define YU_GPIO_CAUSE_OR_CAUSE_EVTEN0	0x80
#define YU_GPIO_CAUSE_OR_EVTEN0		0x94
#define YU_GPIO_CAUSE_OR_CLRCAUSE	0x98

struct mlxbf2_gpio_context_save_regs {
	u32 gpio_mode0;
	u32 gpio_mode1;
};

/* BlueField-2 gpio block context structure. */
struct mlxbf2_gpio_context {
	struct gpio_chip gc;

	/* YU GPIO blocks address */
	void __iomem *gpio_io;
	struct device *dev;

	struct mlxbf2_gpio_context_save_regs *csave_regs;
};

/* BlueField-2 gpio shared structure. */
struct mlxbf2_gpio_param {
	void __iomem *io;
	struct resource *res;
	struct mutex *lock;
};

static struct resource yu_arm_gpio_lock_res =
	DEFINE_RES_MEM_NAMED(YU_ARM_GPIO_LOCK_ADDR, YU_ARM_GPIO_LOCK_SIZE, "YU_ARM_GPIO_LOCK");

static DEFINE_MUTEX(yu_arm_gpio_lock_mutex);

static struct mlxbf2_gpio_param yu_arm_gpio_lock_param = {
	.res = &yu_arm_gpio_lock_res,
	.lock = &yu_arm_gpio_lock_mutex,
};

/* Request memory region and map yu_arm_gpio_lock resource */
static int mlxbf2_gpio_get_lock_res(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct resource *res;
	resource_size_t size;
	int ret = 0;

	mutex_lock(yu_arm_gpio_lock_param.lock);

	/* Check if the memory map already exists */
	if (yu_arm_gpio_lock_param.io)
		goto exit;

	res = yu_arm_gpio_lock_param.res;
	size = resource_size(res);

	if (!devm_request_mem_region(dev, res->start, size, res->name)) {
		ret = -EFAULT;
		goto exit;
	}

	yu_arm_gpio_lock_param.io = devm_ioremap(dev, res->start, size);
	if (!yu_arm_gpio_lock_param.io)
		ret = -ENOMEM;

exit:
	mutex_unlock(yu_arm_gpio_lock_param.lock);

	return ret;
}

/*
 * Acquire the YU arm_gpio_lock to be able to change the direction
 * mode. If the lock_active bit is already set, return an error.
 */
static int mlxbf2_gpio_lock_acquire(struct mlxbf2_gpio_context *gs)
{
	u32 arm_gpio_lock_val;

	mutex_lock(yu_arm_gpio_lock_param.lock);
	raw_spin_lock(&gs->gc.bgpio_lock);

	arm_gpio_lock_val = readl(yu_arm_gpio_lock_param.io);

	/*
	 * When lock active bit[31] is set, ModeX is write enabled
	 */
	if (YU_LOCK_ACTIVE_BIT(arm_gpio_lock_val)) {
		raw_spin_unlock(&gs->gc.bgpio_lock);
		mutex_unlock(yu_arm_gpio_lock_param.lock);
		return -EINVAL;
	}

	writel(YU_ARM_GPIO_LOCK_ACQUIRE, yu_arm_gpio_lock_param.io);

	return 0;
}

/*
 * Release the YU arm_gpio_lock after changing the direction mode.
 */
static void mlxbf2_gpio_lock_release(struct mlxbf2_gpio_context *gs)
	__releases(&gs->gc.bgpio_lock)
	__releases(yu_arm_gpio_lock_param.lock)
{
	writel(YU_ARM_GPIO_LOCK_RELEASE, yu_arm_gpio_lock_param.io);
	raw_spin_unlock(&gs->gc.bgpio_lock);
	mutex_unlock(yu_arm_gpio_lock_param.lock);
}

/*
 * mode0 and mode1 are both locked by the gpio_lock field.
 *
 * Together, mode0 and mode1 define the gpio Mode dependeing also
 * on Reg_DataOut.
 *
 * {mode1,mode0}:{Reg_DataOut=0,Reg_DataOut=1}->{DataOut=0,DataOut=1}
 *
 * {0,0}:Reg_DataOut{0,1}->{Z,Z} Input PAD
 * {0,1}:Reg_DataOut{0,1}->{0,1} Full drive Output PAD
 * {1,0}:Reg_DataOut{0,1}->{0,Z} 0-set PAD to low, 1-float
 * {1,1}:Reg_DataOut{0,1}->{Z,1} 0-float, 1-set PAD to high
 */

/*
 * Set input direction:
 * {mode1,mode0} = {0,0}
 */
static int mlxbf2_gpio_direction_input(struct gpio_chip *chip,
				       unsigned int offset)
{
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(chip);
	int ret;

	/*
	 * Although the arm_gpio_lock was set in the probe function, check again
	 * if it is still enabled to be able to write to the ModeX registers.
	 */
	ret = mlxbf2_gpio_lock_acquire(gs);
	if (ret < 0)
		return ret;

	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE0_CLEAR);
	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE1_CLEAR);

	mlxbf2_gpio_lock_release(gs);

	return ret;
}

/*
 * Set output direction:
 * {mode1,mode0} = {0,1}
 */
static int mlxbf2_gpio_direction_output(struct gpio_chip *chip,
					unsigned int offset,
					int value)
{
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(chip);
	int ret = 0;

	/*
	 * Although the arm_gpio_lock was set in the probe function,
	 * check again it is still enabled to be able to write to the
	 * ModeX registers.
	 */
	ret = mlxbf2_gpio_lock_acquire(gs);
	if (ret < 0)
		return ret;

	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE1_CLEAR);
	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE0_SET);

	mlxbf2_gpio_lock_release(gs);

	return ret;
}

static void mlxbf2_gpio_irq_enable(struct irq_data *irqd)
{
	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	int offset = irqd_to_hwirq(irqd);
	unsigned long flags;
	u32 val;

	gpiochip_enable_irq(gc, irqd_to_hwirq(irqd));
	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);
	val |= BIT(offset);
	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);

	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	val |= BIT(offset);
	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
}

static void mlxbf2_gpio_irq_disable(struct irq_data *irqd)
{
	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	int offset = irqd_to_hwirq(irqd);
	unsigned long flags;
	u32 val;

	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	val &= ~BIT(offset);
	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
	gpiochip_disable_irq(gc, irqd_to_hwirq(irqd));
}

static irqreturn_t mlxbf2_gpio_irq_handler(int irq, void *ptr)
{
	struct mlxbf2_gpio_context *gs = ptr;
	struct gpio_chip *gc = &gs->gc;
	unsigned long pending;
	u32 level;

	pending = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_CAUSE_EVTEN0);
	writel(pending, gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);

	for_each_set_bit(level, &pending, gc->ngpio)
		generic_handle_domain_irq_safe(gc->irq.domain, level);

	return IRQ_RETVAL(pending);
}

static int
mlxbf2_gpio_irq_set_type(struct irq_data *irqd, unsigned int type)
{
	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	int offset = irqd_to_hwirq(irqd);
	unsigned long flags;
	bool fall = false;
	bool rise = false;
	u32 val;

	switch (type & IRQ_TYPE_SENSE_MASK) {
	case IRQ_TYPE_EDGE_BOTH:
		fall = true;
		rise = true;
		break;
	case IRQ_TYPE_EDGE_RISING:
		rise = true;
		break;
	case IRQ_TYPE_EDGE_FALLING:
		fall = true;
		break;
	default:
		return -EINVAL;
	}

	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
	if (fall) {
		val = readl(gs->gpio_io + YU_GPIO_CAUSE_FALL_EN);
		val |= BIT(offset);
		writel(val, gs->gpio_io + YU_GPIO_CAUSE_FALL_EN);
	}

	if (rise) {
		val = readl(gs->gpio_io + YU_GPIO_CAUSE_RISE_EN);
		val |= BIT(offset);
		writel(val, gs->gpio_io + YU_GPIO_CAUSE_RISE_EN);
	}
	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);

	return 0;
}

static void mlxbf2_gpio_irq_print_chip(struct irq_data *irqd,
				       struct seq_file *p)
{
	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);

	seq_printf(p, dev_name(gs->dev));
}

static const struct irq_chip mlxbf2_gpio_irq_chip = {
	.irq_set_type = mlxbf2_gpio_irq_set_type,
	.irq_enable = mlxbf2_gpio_irq_enable,
	.irq_disable = mlxbf2_gpio_irq_disable,
	.irq_print_chip = mlxbf2_gpio_irq_print_chip,
	.flags = IRQCHIP_IMMUTABLE,
	GPIOCHIP_IRQ_RESOURCE_HELPERS,
};

/* BlueField-2 GPIO driver initialization routine. */
static int
mlxbf2_gpio_probe(struct platform_device *pdev)
{
	struct mlxbf2_gpio_context *gs;
	struct device *dev = &pdev->dev;
	struct gpio_irq_chip *girq;
	struct gpio_chip *gc;
	unsigned int npins;
	const char *name;
	int ret, irq;

	name = dev_name(dev);

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

	gs->dev = dev;

	/* YU GPIO block address */
	gs->gpio_io = devm_platform_ioremap_resource(pdev, 0);
	if (IS_ERR(gs->gpio_io))
		return PTR_ERR(gs->gpio_io);

	ret = mlxbf2_gpio_get_lock_res(pdev);
	if (ret) {
		dev_err(dev, "Failed to get yu_arm_gpio_lock resource\n");
		return ret;
	}

	if (device_property_read_u32(dev, "npins", &npins))
		npins = MLXBF2_GPIO_MAX_PINS_PER_BLOCK;

	gc = &gs->gc;

	ret = bgpio_init(gc, dev, 4,
			gs->gpio_io + YU_GPIO_DATAIN,
			gs->gpio_io + YU_GPIO_DATASET,
			gs->gpio_io + YU_GPIO_DATACLEAR,
			NULL,
			NULL,
			0);

	if (ret) {
		dev_err(dev, "bgpio_init failed\n");
		return ret;
	}

	gc->direction_input = mlxbf2_gpio_direction_input;
	gc->direction_output = mlxbf2_gpio_direction_output;
	gc->ngpio = npins;
	gc->owner = THIS_MODULE;

	irq = platform_get_irq(pdev, 0);
	if (irq >= 0) {
		girq = &gs->gc.irq;
		gpio_irq_chip_set_chip(girq, &mlxbf2_gpio_irq_chip);
		girq->handler = handle_simple_irq;
		girq->default_type = IRQ_TYPE_NONE;
		/* This will let us handle the parent IRQ in the driver */
		girq->num_parents = 0;
		girq->parents = NULL;
		girq->parent_handler = NULL;

		/*
		 * Directly request the irq here instead of passing
		 * a flow-handler because the irq is shared.
		 */
		ret = devm_request_irq(dev, irq, mlxbf2_gpio_irq_handler,
				       IRQF_SHARED, name, gs);
		if (ret) {
			dev_err(dev, "failed to request IRQ");
			return ret;
		}
	}

	platform_set_drvdata(pdev, gs);

	ret = devm_gpiochip_add_data(dev, &gs->gc, gs);
	if (ret) {
		dev_err(dev, "Failed adding memory mapped gpiochip\n");
		return ret;
	}

	return 0;
}

static int __maybe_unused mlxbf2_gpio_suspend(struct device *dev)
{
	struct mlxbf2_gpio_context *gs = dev_get_drvdata(dev);

	gs->csave_regs->gpio_mode0 = readl(gs->gpio_io +
		YU_GPIO_MODE0);
	gs->csave_regs->gpio_mode1 = readl(gs->gpio_io +
		YU_GPIO_MODE1);

	return 0;
}

static int __maybe_unused mlxbf2_gpio_resume(struct device *dev)
{
	struct mlxbf2_gpio_context *gs = dev_get_drvdata(dev);

	writel(gs->csave_regs->gpio_mode0, gs->gpio_io +
		YU_GPIO_MODE0);
	writel(gs->csave_regs->gpio_mode1, gs->gpio_io +
		YU_GPIO_MODE1);

	return 0;
}
static SIMPLE_DEV_PM_OPS(mlxbf2_pm_ops, mlxbf2_gpio_suspend, mlxbf2_gpio_resume);

static const struct acpi_device_id __maybe_unused mlxbf2_gpio_acpi_match[] = {
	{ "MLNXBF22", 0 },
	{},
};
MODULE_DEVICE_TABLE(acpi, mlxbf2_gpio_acpi_match);

static struct platform_driver mlxbf2_gpio_driver = {
	.driver = {
		.name = "mlxbf2_gpio",
		.acpi_match_table = mlxbf2_gpio_acpi_match,
		.pm = &mlxbf2_pm_ops,
	},
	.probe    = mlxbf2_gpio_probe,
};

module_platform_driver(mlxbf2_gpio_driver);

MODULE_DESCRIPTION("Mellanox BlueField-2 GPIO Driver");
MODULE_AUTHOR("Asmaa Mnebhi <asmaa@nvidia.com>");
MODULE_LICENSE("GPL v2");
Commit	Line	Data
bc0ae0e7 AM	1	// SPDX-License-Identifier: GPL-2.0
bc0ae0e7 AM	2
2b725265 AM	3	/*
	4	* Copyright (C) 2020-2021 NVIDIA CORPORATION & AFFILIATES
	5	*/
	6
bc0ae0e7 AM	7	#include <linux/bitfield.h>
	8	#include <linux/bitops.h>
	9	#include <linux/device.h>
	10	#include <linux/gpio/driver.h>
2b725265	11	#include <linux/interrupt.h>
bc0ae0e7 AM	12	#include <linux/io.h>
	13	#include <linux/ioport.h>
	14	#include <linux/kernel.h>
603607e7	15	#include <linux/mod_devicetable.h>
bc0ae0e7 AM	16	#include <linux/module.h>
	17	#include <linux/platform_device.h>
	18	#include <linux/pm.h>
	19	#include <linux/resource.h>
5bfff76d	20	#include <linux/seq_file.h>
bc0ae0e7 AM	21	#include <linux/spinlock.h>
bc0ae0e7 AM	22	#include <linux/types.h>
bc0ae0e7 AM	23
	24	/*
	25	* There are 3 YU GPIO blocks:
	26	* gpio[0]: HOST_GPIO0->HOST_GPIO31
	27	* gpio[1]: HOST_GPIO32->HOST_GPIO63
	28	* gpio[2]: HOST_GPIO64->HOST_GPIO69
	29	*/
	30	#define MLXBF2_GPIO_MAX_PINS_PER_BLOCK 32
	31
	32	/*
	33	* arm_gpio_lock register:
	34	* bit[31] lock status: active if set
	35	* bit[15:0] set lock
	36	* The lock is enabled only if 0xd42f is written to this field
	37	*/
	38	#define YU_ARM_GPIO_LOCK_ADDR 0x2801088
	39	#define YU_ARM_GPIO_LOCK_SIZE 0x8
	40	#define YU_LOCK_ACTIVE_BIT(val) (val >> 31)
	41	#define YU_ARM_GPIO_LOCK_ACQUIRE 0xd42f
	42	#define YU_ARM_GPIO_LOCK_RELEASE 0x0
	43
	44	/*
	45	* gpio[x] block registers and their offset
	46	*/
	47	#define YU_GPIO_DATAIN 0x04
	48	#define YU_GPIO_MODE1 0x08
	49	#define YU_GPIO_MODE0 0x0c
	50	#define YU_GPIO_DATASET 0x14
	51	#define YU_GPIO_DATACLEAR 0x18
2b725265 AM	52	#define YU_GPIO_CAUSE_RISE_EN 0x44
2b725265 AM	53	#define YU_GPIO_CAUSE_FALL_EN 0x48
bc0ae0e7 AM	54	#define YU_GPIO_MODE1_CLEAR 0x50
	55	#define YU_GPIO_MODE0_SET 0x54
	56	#define YU_GPIO_MODE0_CLEAR 0x58
2b725265 AM	57	#define YU_GPIO_CAUSE_OR_CAUSE_EVTEN0 0x80
	58	#define YU_GPIO_CAUSE_OR_EVTEN0 0x94
	59	#define YU_GPIO_CAUSE_OR_CLRCAUSE 0x98
bc0ae0e7	60
bc0ae0e7 AM	61	struct mlxbf2_gpio_context_save_regs {
	62	u32 gpio_mode0;
	63	u32 gpio_mode1;
	64	};
bc0ae0e7 AM	65
	66	/* BlueField-2 gpio block context structure. */
	67	struct mlxbf2_gpio_context {
	68	struct gpio_chip gc;
	69
	70	/* YU GPIO blocks address */
	71	void __iomem *gpio_io;
5bfff76d	72	struct device *dev;
bc0ae0e7	73
bc0ae0e7	74	struct mlxbf2_gpio_context_save_regs *csave_regs;
bc0ae0e7 AM	75	};
	76
	77	/* BlueField-2 gpio shared structure. */
	78	struct mlxbf2_gpio_param {
	79	void __iomem *io;
	80	struct resource *res;
	81	struct mutex *lock;
	82	};
	83
d0ef631d AS	84	static struct resource yu_arm_gpio_lock_res =
d0ef631d AS	85	DEFINE_RES_MEM_NAMED(YU_ARM_GPIO_LOCK_ADDR, YU_ARM_GPIO_LOCK_SIZE, "YU_ARM_GPIO_LOCK");
bc0ae0e7 AM	86
	87	static DEFINE_MUTEX(yu_arm_gpio_lock_mutex);
	88
	89	static struct mlxbf2_gpio_param yu_arm_gpio_lock_param = {
	90	.res = &yu_arm_gpio_lock_res,
	91	.lock = &yu_arm_gpio_lock_mutex,
	92	};
	93
	94	/* Request memory region and map yu_arm_gpio_lock resource */
	95	static int mlxbf2_gpio_get_lock_res(struct platform_device *pdev)
	96	{
	97	struct device *dev = &pdev->dev;
	98	struct resource *res;
	99	resource_size_t size;
	100	int ret = 0;
	101
	102	mutex_lock(yu_arm_gpio_lock_param.lock);
	103
	104	/* Check if the memory map already exists */
	105	if (yu_arm_gpio_lock_param.io)
	106	goto exit;
	107
	108	res = yu_arm_gpio_lock_param.res;
	109	size = resource_size(res);
	110
	111	if (!devm_request_mem_region(dev, res->start, size, res->name)) {
	112	ret = -EFAULT;
	113	goto exit;
	114	}
	115
	116	yu_arm_gpio_lock_param.io = devm_ioremap(dev, res->start, size);
66d8ad67 WY	117	if (!yu_arm_gpio_lock_param.io)
66d8ad67 WY	118	ret = -ENOMEM;
bc0ae0e7 AM	119
	120	exit:
	121	mutex_unlock(yu_arm_gpio_lock_param.lock);
	122
	123	return ret;
	124	}
	125
	126	/*
	127	* Acquire the YU arm_gpio_lock to be able to change the direction
	128	* mode. If the lock_active bit is already set, return an error.
	129	*/
	130	static int mlxbf2_gpio_lock_acquire(struct mlxbf2_gpio_context *gs)
	131	{
	132	u32 arm_gpio_lock_val;
	133
bc0ae0e7	134	mutex_lock(yu_arm_gpio_lock_param.lock);
3c938cc5	135	raw_spin_lock(&gs->gc.bgpio_lock);
bc0ae0e7 AM	136
	137	arm_gpio_lock_val = readl(yu_arm_gpio_lock_param.io);
	138
	139	/*
	140	* When lock active bit[31] is set, ModeX is write enabled
	141	*/
	142	if (YU_LOCK_ACTIVE_BIT(arm_gpio_lock_val)) {
3c938cc5	143	raw_spin_unlock(&gs->gc.bgpio_lock);
e6862430	144	mutex_unlock(yu_arm_gpio_lock_param.lock);
bc0ae0e7 AM	145	return -EINVAL;
	146	}
	147
	148	writel(YU_ARM_GPIO_LOCK_ACQUIRE, yu_arm_gpio_lock_param.io);
	149
	150	return 0;
	151	}
	152
	153	/*
	154	* Release the YU arm_gpio_lock after changing the direction mode.
	155	*/
	156	static void mlxbf2_gpio_lock_release(struct mlxbf2_gpio_context *gs)
a7a9ad23 LJ	157	__releases(&gs->gc.bgpio_lock)
a7a9ad23 LJ	158	__releases(yu_arm_gpio_lock_param.lock)
bc0ae0e7 AM	159	{
bc0ae0e7 AM	160	writel(YU_ARM_GPIO_LOCK_RELEASE, yu_arm_gpio_lock_param.io);
3c938cc5	161	raw_spin_unlock(&gs->gc.bgpio_lock);
e6862430	162	mutex_unlock(yu_arm_gpio_lock_param.lock);
bc0ae0e7 AM	163	}
	164
	165	/*
	166	* mode0 and mode1 are both locked by the gpio_lock field.
	167	*
	168	* Together, mode0 and mode1 define the gpio Mode dependeing also
	169	* on Reg_DataOut.
	170	*
	171	* {mode1,mode0}:{Reg_DataOut=0,Reg_DataOut=1}->{DataOut=0,DataOut=1}
	172	*
	173	* {0,0}:Reg_DataOut{0,1}->{Z,Z} Input PAD
	174	* {0,1}:Reg_DataOut{0,1}->{0,1} Full drive Output PAD
	175	* {1,0}:Reg_DataOut{0,1}->{0,Z} 0-set PAD to low, 1-float
	176	* {1,1}:Reg_DataOut{0,1}->{Z,1} 0-float, 1-set PAD to high
	177	*/
	178
	179	/*
	180	* Set input direction:
	181	* {mode1,mode0} = {0,0}
	182	*/
	183	static int mlxbf2_gpio_direction_input(struct gpio_chip *chip,
	184	unsigned int offset)
	185	{
	186	struct mlxbf2_gpio_context *gs = gpiochip_get_data(chip);
	187	int ret;
	188
	189	/*
	190	* Although the arm_gpio_lock was set in the probe function, check again
	191	* if it is still enabled to be able to write to the ModeX registers.
	192	*/
	193	ret = mlxbf2_gpio_lock_acquire(gs);
	194	if (ret < 0)
	195	return ret;
	196
	197	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE0_CLEAR);
	198	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE1_CLEAR);
	199
	200	mlxbf2_gpio_lock_release(gs);
	201
	202	return ret;
	203	}
	204
	205	/*
	206	* Set output direction:
	207	* {mode1,mode0} = {0,1}
	208	*/
	209	static int mlxbf2_gpio_direction_output(struct gpio_chip *chip,
	210	unsigned int offset,
	211	int value)
	212	{
	213	struct mlxbf2_gpio_context *gs = gpiochip_get_data(chip);
	214	int ret = 0;
	215
	216	/*
	217	* Although the arm_gpio_lock was set in the probe function,
	218	* check again it is still enabled to be able to write to the
	219	* ModeX registers.
	220	*/
	221	ret = mlxbf2_gpio_lock_acquire(gs);
	222	if (ret < 0)
	223	return ret;
	224
	225	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE1_CLEAR);
	226	writel(BIT(offset), gs->gpio_io + YU_GPIO_MODE0_SET);
227
228	mlxbf2_gpio_lock_release(gs);
229
230	return ret;
231	}
232
2b725265 AM	233	static void mlxbf2_gpio_irq_enable(struct irq_data *irqd)
	234	{
	235	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	236	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	237	int offset = irqd_to_hwirq(irqd);
	238	unsigned long flags;
	239	u32 val;
	240
5bfff76d	241	gpiochip_enable_irq(gc, irqd_to_hwirq(irqd));
3c938cc5	242	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
2b725265 AM	243	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);
	244	val \|= BIT(offset);
	245	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);
	246
	247	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	248	val \|= BIT(offset);
	249	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
3c938cc5	250	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
2b725265 AM	251	}
	252
	253	static void mlxbf2_gpio_irq_disable(struct irq_data *irqd)
	254	{
	255	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	256	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	257	int offset = irqd_to_hwirq(irqd);
	258	unsigned long flags;
	259	u32 val;
	260
3c938cc5	261	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
2b725265 AM	262	val = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
	263	val &= ~BIT(offset);
	264	writel(val, gs->gpio_io + YU_GPIO_CAUSE_OR_EVTEN0);
3c938cc5	265	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
5bfff76d	266	gpiochip_disable_irq(gc, irqd_to_hwirq(irqd));
2b725265 AM	267	}
	268
	269	static irqreturn_t mlxbf2_gpio_irq_handler(int irq, void *ptr)
	270	{
	271	struct mlxbf2_gpio_context *gs = ptr;
	272	struct gpio_chip *gc = &gs->gc;
	273	unsigned long pending;
	274	u32 level;
	275
	276	pending = readl(gs->gpio_io + YU_GPIO_CAUSE_OR_CAUSE_EVTEN0);
	277	writel(pending, gs->gpio_io + YU_GPIO_CAUSE_OR_CLRCAUSE);
	278
118c3ba2 SAS	279	for_each_set_bit(level, &pending, gc->ngpio)
118c3ba2 SAS	280	generic_handle_domain_irq_safe(gc->irq.domain, level);
2b725265 AM	281
	282	return IRQ_RETVAL(pending);
	283	}
	284
	285	static int
	286	mlxbf2_gpio_irq_set_type(struct irq_data *irqd, unsigned int type)
	287	{
	288	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	289	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	290	int offset = irqd_to_hwirq(irqd);
	291	unsigned long flags;
	292	bool fall = false;
	293	bool rise = false;
	294	u32 val;
	295
	296	switch (type & IRQ_TYPE_SENSE_MASK) {
	297	case IRQ_TYPE_EDGE_BOTH:
	298	fall = true;
	299	rise = true;
	300	break;
	301	case IRQ_TYPE_EDGE_RISING:
	302	rise = true;
	303	break;
	304	case IRQ_TYPE_EDGE_FALLING:
	305	fall = true;
	306	break;
	307	default:
	308	return -EINVAL;
	309	}
	310
3c938cc5	311	raw_spin_lock_irqsave(&gs->gc.bgpio_lock, flags);
2b725265 AM	312	if (fall) {
	313	val = readl(gs->gpio_io + YU_GPIO_CAUSE_FALL_EN);
	314	val \|= BIT(offset);
	315	writel(val, gs->gpio_io + YU_GPIO_CAUSE_FALL_EN);
	316	}
	317
	318	if (rise) {
	319	val = readl(gs->gpio_io + YU_GPIO_CAUSE_RISE_EN);
	320	val \|= BIT(offset);
	321	writel(val, gs->gpio_io + YU_GPIO_CAUSE_RISE_EN);
	322	}
3c938cc5	323	raw_spin_unlock_irqrestore(&gs->gc.bgpio_lock, flags);
2b725265 AM	324
	325	return 0;
	326	}
	327
5bfff76d LW	328	static void mlxbf2_gpio_irq_print_chip(struct irq_data *irqd,
	329	struct seq_file *p)
	330	{
	331	struct gpio_chip *gc = irq_data_get_irq_chip_data(irqd);
	332	struct mlxbf2_gpio_context *gs = gpiochip_get_data(gc);
	333
	334	seq_printf(p, dev_name(gs->dev));
	335	}
	336
	337	static const struct irq_chip mlxbf2_gpio_irq_chip = {
	338	.irq_set_type = mlxbf2_gpio_irq_set_type,
	339	.irq_enable = mlxbf2_gpio_irq_enable,
	340	.irq_disable = mlxbf2_gpio_irq_disable,
	341	.irq_print_chip = mlxbf2_gpio_irq_print_chip,
	342	.flags = IRQCHIP_IMMUTABLE,
	343	GPIOCHIP_IRQ_RESOURCE_HELPERS,
	344	};
	345
bc0ae0e7 AM	346	/* BlueField-2 GPIO driver initialization routine. */
	347	static int
	348	mlxbf2_gpio_probe(struct platform_device *pdev)
	349	{
	350	struct mlxbf2_gpio_context *gs;
	351	struct device *dev = &pdev->dev;
2b725265	352	struct gpio_irq_chip *girq;
bc0ae0e7	353	struct gpio_chip *gc;
bc0ae0e7	354	unsigned int npins;
2b725265 AM	355	const char *name;
	356	int ret, irq;
	357
	358	name = dev_name(dev);
bc0ae0e7 AM	359
	360	gs = devm_kzalloc(dev, sizeof(*gs), GFP_KERNEL);
	361	if (!gs)
	362	return -ENOMEM;
	363
5bfff76d LW	364	gs->dev = dev;
5bfff76d LW	365
bc0ae0e7	366	/* YU GPIO block address */
4e6864f8 AS	367	gs->gpio_io = devm_platform_ioremap_resource(pdev, 0);
	368	if (IS_ERR(gs->gpio_io))
	369	return PTR_ERR(gs->gpio_io);
bc0ae0e7 AM	370
	371	ret = mlxbf2_gpio_get_lock_res(pdev);
	372	if (ret) {
	373	dev_err(dev, "Failed to get yu_arm_gpio_lock resource\n");
	374	return ret;
	375	}
	376
	377	if (device_property_read_u32(dev, "npins", &npins))
	378	npins = MLXBF2_GPIO_MAX_PINS_PER_BLOCK;
	379
	380	gc = &gs->gc;
	381
	382	ret = bgpio_init(gc, dev, 4,
	383	gs->gpio_io + YU_GPIO_DATAIN,
	384	gs->gpio_io + YU_GPIO_DATASET,
	385	gs->gpio_io + YU_GPIO_DATACLEAR,
	386	NULL,
	387	NULL,
	388	0);
	389
c0eee6fb AM	390	if (ret) {
	391	dev_err(dev, "bgpio_init failed\n");
	392	return ret;
	393	}
	394
bc0ae0e7 AM	395	gc->direction_input = mlxbf2_gpio_direction_input;
	396	gc->direction_output = mlxbf2_gpio_direction_output;
	397	gc->ngpio = npins;
	398	gc->owner = THIS_MODULE;
	399
2b725265 AM	400	irq = platform_get_irq(pdev, 0);
2b725265 AM	401	if (irq >= 0) {
2b725265	402	girq = &gs->gc.irq;
5bfff76d	403	gpio_irq_chip_set_chip(girq, &mlxbf2_gpio_irq_chip);
2b725265 AM	404	girq->handler = handle_simple_irq;
	405	girq->default_type = IRQ_TYPE_NONE;
	406	/* This will let us handle the parent IRQ in the driver */
	407	girq->num_parents = 0;
	408	girq->parents = NULL;
	409	girq->parent_handler = NULL;
	410
	411	/*
	412	* Directly request the irq here instead of passing
	413	* a flow-handler because the irq is shared.
	414	*/
	415	ret = devm_request_irq(dev, irq, mlxbf2_gpio_irq_handler,
	416	IRQF_SHARED, name, gs);
	417	if (ret) {
	418	dev_err(dev, "failed to request IRQ");
	419	return ret;
	420	}
	421	}
	422
bc0ae0e7 AM	423	platform_set_drvdata(pdev, gs);
	424
	425	ret = devm_gpiochip_add_data(dev, &gs->gc, gs);
	426	if (ret) {
	427	dev_err(dev, "Failed adding memory mapped gpiochip\n");
	428	return ret;
	429	}
	430
	431	return 0;
	432	}
	433
dabe57c3	434	static int __maybe_unused mlxbf2_gpio_suspend(struct device *dev)
bc0ae0e7	435	{
dabe57c3	436	struct mlxbf2_gpio_context *gs = dev_get_drvdata(dev);
bc0ae0e7 AM	437
	438	gs->csave_regs->gpio_mode0 = readl(gs->gpio_io +
	439	YU_GPIO_MODE0);
	440	gs->csave_regs->gpio_mode1 = readl(gs->gpio_io +
	441	YU_GPIO_MODE1);
	442
	443	return 0;
	444	}
	445
dabe57c3	446	static int __maybe_unused mlxbf2_gpio_resume(struct device *dev)
bc0ae0e7	447	{
dabe57c3	448	struct mlxbf2_gpio_context *gs = dev_get_drvdata(dev);
bc0ae0e7 AM	449
	450	writel(gs->csave_regs->gpio_mode0, gs->gpio_io +
	451	YU_GPIO_MODE0);
	452	writel(gs->csave_regs->gpio_mode1, gs->gpio_io +
	453	YU_GPIO_MODE1);
	454
	455	return 0;
	456	}
dabe57c3	457	static SIMPLE_DEV_PM_OPS(mlxbf2_pm_ops, mlxbf2_gpio_suspend, mlxbf2_gpio_resume);
bc0ae0e7	458
2f9bce5f	459	static const struct acpi_device_id __maybe_unused mlxbf2_gpio_acpi_match[] = {
bc0ae0e7 AM	460	{ "MLNXBF22", 0 },
	461	{},
	462	};
	463	MODULE_DEVICE_TABLE(acpi, mlxbf2_gpio_acpi_match);
	464
	465	static struct platform_driver mlxbf2_gpio_driver = {
	466	.driver = {
	467	.name = "mlxbf2_gpio",
603607e7	468	.acpi_match_table = mlxbf2_gpio_acpi_match,
dabe57c3	469	.pm = &mlxbf2_pm_ops,
bc0ae0e7 AM	470	},
bc0ae0e7 AM	471	.probe = mlxbf2_gpio_probe,
bc0ae0e7 AM	472	};
	473
	474	module_platform_driver(mlxbf2_gpio_driver);
	475
	476	MODULE_DESCRIPTION("Mellanox BlueField-2 GPIO Driver");
2b725265	477	MODULE_AUTHOR("Asmaa Mnebhi <asmaa@nvidia.com>");
bc0ae0e7	478	MODULE_LICENSE("GPL v2");