162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2020 NovaTech LLC
462306a36Sopenharmony_ci * George McCollister <george.mccollister@gmail.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/bits.h>
862306a36Sopenharmony_ci#include <linux/i2c.h>
962306a36Sopenharmony_ci#include <linux/module.h>
1062306a36Sopenharmony_ci#include "xrs700x.h"
1162306a36Sopenharmony_ci#include "xrs700x_reg.h"
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_cistruct xrs700x_i2c_cmd {
1462306a36Sopenharmony_ci	__be32 reg;
1562306a36Sopenharmony_ci	__be16 val;
1662306a36Sopenharmony_ci} __packed;
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_cistatic int xrs700x_i2c_reg_read(void *context, unsigned int reg,
1962306a36Sopenharmony_ci				unsigned int *val)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	struct device *dev = context;
2262306a36Sopenharmony_ci	struct i2c_client *i2c = to_i2c_client(dev);
2362306a36Sopenharmony_ci	struct xrs700x_i2c_cmd cmd;
2462306a36Sopenharmony_ci	int ret;
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci	cmd.reg = cpu_to_be32(reg | 1);
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci	ret = i2c_master_send(i2c, (char *)&cmd.reg, sizeof(cmd.reg));
2962306a36Sopenharmony_ci	if (ret < 0) {
3062306a36Sopenharmony_ci		dev_err(dev, "xrs i2c_master_send returned %d\n", ret);
3162306a36Sopenharmony_ci		return ret;
3262306a36Sopenharmony_ci	}
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	ret = i2c_master_recv(i2c, (char *)&cmd.val, sizeof(cmd.val));
3562306a36Sopenharmony_ci	if (ret < 0) {
3662306a36Sopenharmony_ci		dev_err(dev, "xrs i2c_master_recv returned %d\n", ret);
3762306a36Sopenharmony_ci		return ret;
3862306a36Sopenharmony_ci	}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	*val = be16_to_cpu(cmd.val);
4162306a36Sopenharmony_ci	return 0;
4262306a36Sopenharmony_ci}
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic int xrs700x_i2c_reg_write(void *context, unsigned int reg,
4562306a36Sopenharmony_ci				 unsigned int val)
4662306a36Sopenharmony_ci{
4762306a36Sopenharmony_ci	struct device *dev = context;
4862306a36Sopenharmony_ci	struct i2c_client *i2c = to_i2c_client(dev);
4962306a36Sopenharmony_ci	struct xrs700x_i2c_cmd cmd;
5062306a36Sopenharmony_ci	int ret;
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	cmd.reg = cpu_to_be32(reg);
5362306a36Sopenharmony_ci	cmd.val = cpu_to_be16(val);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	ret = i2c_master_send(i2c, (char *)&cmd, sizeof(cmd));
5662306a36Sopenharmony_ci	if (ret < 0) {
5762306a36Sopenharmony_ci		dev_err(dev, "xrs i2c_master_send returned %d\n", ret);
5862306a36Sopenharmony_ci		return ret;
5962306a36Sopenharmony_ci	}
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	return 0;
6262306a36Sopenharmony_ci}
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_cistatic const struct regmap_config xrs700x_i2c_regmap_config = {
6562306a36Sopenharmony_ci	.val_bits = 16,
6662306a36Sopenharmony_ci	.reg_stride = 2,
6762306a36Sopenharmony_ci	.reg_bits = 32,
6862306a36Sopenharmony_ci	.pad_bits = 0,
6962306a36Sopenharmony_ci	.write_flag_mask = 0,
7062306a36Sopenharmony_ci	.read_flag_mask = 0,
7162306a36Sopenharmony_ci	.reg_read = xrs700x_i2c_reg_read,
7262306a36Sopenharmony_ci	.reg_write = xrs700x_i2c_reg_write,
7362306a36Sopenharmony_ci	.max_register = 0,
7462306a36Sopenharmony_ci	.cache_type = REGCACHE_NONE,
7562306a36Sopenharmony_ci	.reg_format_endian = REGMAP_ENDIAN_BIG,
7662306a36Sopenharmony_ci	.val_format_endian = REGMAP_ENDIAN_BIG
7762306a36Sopenharmony_ci};
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_cistatic int xrs700x_i2c_probe(struct i2c_client *i2c)
8062306a36Sopenharmony_ci{
8162306a36Sopenharmony_ci	struct xrs700x *priv;
8262306a36Sopenharmony_ci	int ret;
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	priv = xrs700x_switch_alloc(&i2c->dev, i2c);
8562306a36Sopenharmony_ci	if (!priv)
8662306a36Sopenharmony_ci		return -ENOMEM;
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	priv->regmap = devm_regmap_init(&i2c->dev, NULL, &i2c->dev,
8962306a36Sopenharmony_ci					&xrs700x_i2c_regmap_config);
9062306a36Sopenharmony_ci	if (IS_ERR(priv->regmap)) {
9162306a36Sopenharmony_ci		ret = PTR_ERR(priv->regmap);
9262306a36Sopenharmony_ci		dev_err(&i2c->dev, "Failed to initialize regmap: %d\n", ret);
9362306a36Sopenharmony_ci		return ret;
9462306a36Sopenharmony_ci	}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	i2c_set_clientdata(i2c, priv);
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	ret = xrs700x_switch_register(priv);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	/* Main DSA driver may not be started yet. */
10162306a36Sopenharmony_ci	if (ret)
10262306a36Sopenharmony_ci		return ret;
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	return 0;
10562306a36Sopenharmony_ci}
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_cistatic void xrs700x_i2c_remove(struct i2c_client *i2c)
10862306a36Sopenharmony_ci{
10962306a36Sopenharmony_ci	struct xrs700x *priv = i2c_get_clientdata(i2c);
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	if (!priv)
11262306a36Sopenharmony_ci		return;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	xrs700x_switch_remove(priv);
11562306a36Sopenharmony_ci}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_cistatic void xrs700x_i2c_shutdown(struct i2c_client *i2c)
11862306a36Sopenharmony_ci{
11962306a36Sopenharmony_ci	struct xrs700x *priv = i2c_get_clientdata(i2c);
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	if (!priv)
12262306a36Sopenharmony_ci		return;
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	xrs700x_switch_shutdown(priv);
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	i2c_set_clientdata(i2c, NULL);
12762306a36Sopenharmony_ci}
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_cistatic const struct i2c_device_id xrs700x_i2c_id[] = {
13062306a36Sopenharmony_ci	{ "xrs700x-switch", 0 },
13162306a36Sopenharmony_ci	{},
13262306a36Sopenharmony_ci};
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ciMODULE_DEVICE_TABLE(i2c, xrs700x_i2c_id);
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_cistatic const struct of_device_id __maybe_unused xrs700x_i2c_dt_ids[] = {
13762306a36Sopenharmony_ci	{ .compatible = "arrow,xrs7003e", .data = &xrs7003e_info },
13862306a36Sopenharmony_ci	{ .compatible = "arrow,xrs7003f", .data = &xrs7003f_info },
13962306a36Sopenharmony_ci	{ .compatible = "arrow,xrs7004e", .data = &xrs7004e_info },
14062306a36Sopenharmony_ci	{ .compatible = "arrow,xrs7004f", .data = &xrs7004f_info },
14162306a36Sopenharmony_ci	{},
14262306a36Sopenharmony_ci};
14362306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, xrs700x_i2c_dt_ids);
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_cistatic struct i2c_driver xrs700x_i2c_driver = {
14662306a36Sopenharmony_ci	.driver = {
14762306a36Sopenharmony_ci		.name	= "xrs700x-i2c",
14862306a36Sopenharmony_ci		.of_match_table = of_match_ptr(xrs700x_i2c_dt_ids),
14962306a36Sopenharmony_ci	},
15062306a36Sopenharmony_ci	.probe = xrs700x_i2c_probe,
15162306a36Sopenharmony_ci	.remove	= xrs700x_i2c_remove,
15262306a36Sopenharmony_ci	.shutdown = xrs700x_i2c_shutdown,
15362306a36Sopenharmony_ci	.id_table = xrs700x_i2c_id,
15462306a36Sopenharmony_ci};
15562306a36Sopenharmony_ci
15662306a36Sopenharmony_cimodule_i2c_driver(xrs700x_i2c_driver);
15762306a36Sopenharmony_ci
15862306a36Sopenharmony_ciMODULE_AUTHOR("George McCollister <george.mccollister@gmail.com>");
15962306a36Sopenharmony_ciMODULE_DESCRIPTION("Arrow SpeedChips XRS700x DSA I2C driver");
16062306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
161