162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci *  74Hx164 - Generic serial-in/parallel-out 8-bits shift register GPIO driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci *  Copyright (C) 2010 Gabor Juhos <juhosg@openwrt.org>
662306a36Sopenharmony_ci *  Copyright (C) 2010 Miguel Gaio <miguel.gaio@efixo.com>
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/bitops.h>
1062306a36Sopenharmony_ci#include <linux/gpio/consumer.h>
1162306a36Sopenharmony_ci#include <linux/gpio/driver.h>
1262306a36Sopenharmony_ci#include <linux/module.h>
1362306a36Sopenharmony_ci#include <linux/mutex.h>
1462306a36Sopenharmony_ci#include <linux/property.h>
1562306a36Sopenharmony_ci#include <linux/slab.h>
1662306a36Sopenharmony_ci#include <linux/spi/spi.h>
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci#define GEN_74X164_NUMBER_GPIOS	8
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistruct gen_74x164_chip {
2162306a36Sopenharmony_ci	struct gpio_chip	gpio_chip;
2262306a36Sopenharmony_ci	struct mutex		lock;
2362306a36Sopenharmony_ci	struct gpio_desc	*gpiod_oe;
2462306a36Sopenharmony_ci	u32			registers;
2562306a36Sopenharmony_ci	/*
2662306a36Sopenharmony_ci	 * Since the registers are chained, every byte sent will make
2762306a36Sopenharmony_ci	 * the previous byte shift to the next register in the
2862306a36Sopenharmony_ci	 * chain. Thus, the first byte sent will end up in the last
2962306a36Sopenharmony_ci	 * register at the end of the transfer. So, to have a logical
3062306a36Sopenharmony_ci	 * numbering, store the bytes in reverse order.
3162306a36Sopenharmony_ci	 */
3262306a36Sopenharmony_ci	u8			buffer[];
3362306a36Sopenharmony_ci};
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_cistatic int __gen_74x164_write_config(struct gen_74x164_chip *chip)
3662306a36Sopenharmony_ci{
3762306a36Sopenharmony_ci	return spi_write(to_spi_device(chip->gpio_chip.parent), chip->buffer,
3862306a36Sopenharmony_ci			 chip->registers);
3962306a36Sopenharmony_ci}
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_cistatic int gen_74x164_get_value(struct gpio_chip *gc, unsigned offset)
4262306a36Sopenharmony_ci{
4362306a36Sopenharmony_ci	struct gen_74x164_chip *chip = gpiochip_get_data(gc);
4462306a36Sopenharmony_ci	u8 bank = chip->registers - 1 - offset / 8;
4562306a36Sopenharmony_ci	u8 pin = offset % 8;
4662306a36Sopenharmony_ci	int ret;
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	mutex_lock(&chip->lock);
4962306a36Sopenharmony_ci	ret = (chip->buffer[bank] >> pin) & 0x1;
5062306a36Sopenharmony_ci	mutex_unlock(&chip->lock);
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	return ret;
5362306a36Sopenharmony_ci}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_cistatic void gen_74x164_set_value(struct gpio_chip *gc,
5662306a36Sopenharmony_ci		unsigned offset, int val)
5762306a36Sopenharmony_ci{
5862306a36Sopenharmony_ci	struct gen_74x164_chip *chip = gpiochip_get_data(gc);
5962306a36Sopenharmony_ci	u8 bank = chip->registers - 1 - offset / 8;
6062306a36Sopenharmony_ci	u8 pin = offset % 8;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	mutex_lock(&chip->lock);
6362306a36Sopenharmony_ci	if (val)
6462306a36Sopenharmony_ci		chip->buffer[bank] |= (1 << pin);
6562306a36Sopenharmony_ci	else
6662306a36Sopenharmony_ci		chip->buffer[bank] &= ~(1 << pin);
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci	__gen_74x164_write_config(chip);
6962306a36Sopenharmony_ci	mutex_unlock(&chip->lock);
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_cistatic void gen_74x164_set_multiple(struct gpio_chip *gc, unsigned long *mask,
7362306a36Sopenharmony_ci				    unsigned long *bits)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	struct gen_74x164_chip *chip = gpiochip_get_data(gc);
7662306a36Sopenharmony_ci	unsigned long offset;
7762306a36Sopenharmony_ci	unsigned long bankmask;
7862306a36Sopenharmony_ci	size_t bank;
7962306a36Sopenharmony_ci	unsigned long bitmask;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	mutex_lock(&chip->lock);
8262306a36Sopenharmony_ci	for_each_set_clump8(offset, bankmask, mask, chip->registers * 8) {
8362306a36Sopenharmony_ci		bank = chip->registers - 1 - offset / 8;
8462306a36Sopenharmony_ci		bitmask = bitmap_get_value8(bits, offset) & bankmask;
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci		chip->buffer[bank] &= ~bankmask;
8762306a36Sopenharmony_ci		chip->buffer[bank] |= bitmask;
8862306a36Sopenharmony_ci	}
8962306a36Sopenharmony_ci	__gen_74x164_write_config(chip);
9062306a36Sopenharmony_ci	mutex_unlock(&chip->lock);
9162306a36Sopenharmony_ci}
9262306a36Sopenharmony_ci
9362306a36Sopenharmony_cistatic int gen_74x164_direction_output(struct gpio_chip *gc,
9462306a36Sopenharmony_ci		unsigned offset, int val)
9562306a36Sopenharmony_ci{
9662306a36Sopenharmony_ci	gen_74x164_set_value(gc, offset, val);
9762306a36Sopenharmony_ci	return 0;
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic int gen_74x164_probe(struct spi_device *spi)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	struct gen_74x164_chip *chip;
10362306a36Sopenharmony_ci	u32 nregs;
10462306a36Sopenharmony_ci	int ret;
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	/*
10762306a36Sopenharmony_ci	 * bits_per_word cannot be configured in platform data
10862306a36Sopenharmony_ci	 */
10962306a36Sopenharmony_ci	spi->bits_per_word = 8;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	ret = spi_setup(spi);
11262306a36Sopenharmony_ci	if (ret < 0)
11362306a36Sopenharmony_ci		return ret;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	ret = device_property_read_u32(&spi->dev, "registers-number", &nregs);
11662306a36Sopenharmony_ci	if (ret) {
11762306a36Sopenharmony_ci		dev_err(&spi->dev, "Missing 'registers-number' property.\n");
11862306a36Sopenharmony_ci		return -EINVAL;
11962306a36Sopenharmony_ci	}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	chip = devm_kzalloc(&spi->dev, sizeof(*chip) + nregs, GFP_KERNEL);
12262306a36Sopenharmony_ci	if (!chip)
12362306a36Sopenharmony_ci		return -ENOMEM;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	chip->gpiod_oe = devm_gpiod_get_optional(&spi->dev, "enable",
12662306a36Sopenharmony_ci						 GPIOD_OUT_LOW);
12762306a36Sopenharmony_ci	if (IS_ERR(chip->gpiod_oe))
12862306a36Sopenharmony_ci		return PTR_ERR(chip->gpiod_oe);
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	spi_set_drvdata(spi, chip);
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_ci	chip->gpio_chip.label = spi->modalias;
13362306a36Sopenharmony_ci	chip->gpio_chip.direction_output = gen_74x164_direction_output;
13462306a36Sopenharmony_ci	chip->gpio_chip.get = gen_74x164_get_value;
13562306a36Sopenharmony_ci	chip->gpio_chip.set = gen_74x164_set_value;
13662306a36Sopenharmony_ci	chip->gpio_chip.set_multiple = gen_74x164_set_multiple;
13762306a36Sopenharmony_ci	chip->gpio_chip.base = -1;
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	chip->registers = nregs;
14062306a36Sopenharmony_ci	chip->gpio_chip.ngpio = GEN_74X164_NUMBER_GPIOS * chip->registers;
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci	chip->gpio_chip.can_sleep = true;
14362306a36Sopenharmony_ci	chip->gpio_chip.parent = &spi->dev;
14462306a36Sopenharmony_ci	chip->gpio_chip.owner = THIS_MODULE;
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	mutex_init(&chip->lock);
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	ret = __gen_74x164_write_config(chip);
14962306a36Sopenharmony_ci	if (ret) {
15062306a36Sopenharmony_ci		dev_err(&spi->dev, "Failed writing: %d\n", ret);
15162306a36Sopenharmony_ci		goto exit_destroy;
15262306a36Sopenharmony_ci	}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	gpiod_set_value_cansleep(chip->gpiod_oe, 1);
15562306a36Sopenharmony_ci
15662306a36Sopenharmony_ci	ret = gpiochip_add_data(&chip->gpio_chip, chip);
15762306a36Sopenharmony_ci	if (!ret)
15862306a36Sopenharmony_ci		return 0;
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ciexit_destroy:
16162306a36Sopenharmony_ci	mutex_destroy(&chip->lock);
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	return ret;
16462306a36Sopenharmony_ci}
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_cistatic void gen_74x164_remove(struct spi_device *spi)
16762306a36Sopenharmony_ci{
16862306a36Sopenharmony_ci	struct gen_74x164_chip *chip = spi_get_drvdata(spi);
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	gpiod_set_value_cansleep(chip->gpiod_oe, 0);
17162306a36Sopenharmony_ci	gpiochip_remove(&chip->gpio_chip);
17262306a36Sopenharmony_ci	mutex_destroy(&chip->lock);
17362306a36Sopenharmony_ci}
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_cistatic const struct spi_device_id gen_74x164_spi_ids[] = {
17662306a36Sopenharmony_ci	{ .name = "74hc595" },
17762306a36Sopenharmony_ci	{ .name = "74lvc594" },
17862306a36Sopenharmony_ci	{},
17962306a36Sopenharmony_ci};
18062306a36Sopenharmony_ciMODULE_DEVICE_TABLE(spi, gen_74x164_spi_ids);
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_cistatic const struct of_device_id gen_74x164_dt_ids[] = {
18362306a36Sopenharmony_ci	{ .compatible = "fairchild,74hc595" },
18462306a36Sopenharmony_ci	{ .compatible = "nxp,74lvc594" },
18562306a36Sopenharmony_ci	{},
18662306a36Sopenharmony_ci};
18762306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, gen_74x164_dt_ids);
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_cistatic struct spi_driver gen_74x164_driver = {
19062306a36Sopenharmony_ci	.driver = {
19162306a36Sopenharmony_ci		.name		= "74x164",
19262306a36Sopenharmony_ci		.of_match_table	= gen_74x164_dt_ids,
19362306a36Sopenharmony_ci	},
19462306a36Sopenharmony_ci	.probe		= gen_74x164_probe,
19562306a36Sopenharmony_ci	.remove		= gen_74x164_remove,
19662306a36Sopenharmony_ci	.id_table	= gen_74x164_spi_ids,
19762306a36Sopenharmony_ci};
19862306a36Sopenharmony_cimodule_spi_driver(gen_74x164_driver);
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ciMODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");
20162306a36Sopenharmony_ciMODULE_AUTHOR("Miguel Gaio <miguel.gaio@efixo.com>");
20262306a36Sopenharmony_ciMODULE_DESCRIPTION("GPIO expander driver for 74X164 8-bits shift register");
20362306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
204