162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2020 Western Digital Corporation or its affiliates.
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci#include <linux/of.h>
662306a36Sopenharmony_ci#include <linux/platform_device.h>
762306a36Sopenharmony_ci#include <linux/reset-controller.h>
862306a36Sopenharmony_ci#include <linux/delay.h>
962306a36Sopenharmony_ci#include <linux/mfd/syscon.h>
1062306a36Sopenharmony_ci#include <linux/regmap.h>
1162306a36Sopenharmony_ci#include <soc/canaan/k210-sysctl.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#include <dt-bindings/reset/k210-rst.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#define K210_RST_MASK	0x27FFFFFF
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistruct k210_rst {
1862306a36Sopenharmony_ci	struct regmap *map;
1962306a36Sopenharmony_ci	struct reset_controller_dev rcdev;
2062306a36Sopenharmony_ci};
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_cistatic inline struct k210_rst *
2362306a36Sopenharmony_cito_k210_rst(struct reset_controller_dev *rcdev)
2462306a36Sopenharmony_ci{
2562306a36Sopenharmony_ci	return container_of(rcdev, struct k210_rst, rcdev);
2662306a36Sopenharmony_ci}
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic inline int k210_rst_assert(struct reset_controller_dev *rcdev,
2962306a36Sopenharmony_ci				  unsigned long id)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	struct k210_rst *ksr = to_k210_rst(rcdev);
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	return regmap_update_bits(ksr->map, K210_SYSCTL_PERI_RESET, BIT(id), 1);
3462306a36Sopenharmony_ci}
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_cistatic inline int k210_rst_deassert(struct reset_controller_dev *rcdev,
3762306a36Sopenharmony_ci				    unsigned long id)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	struct k210_rst *ksr = to_k210_rst(rcdev);
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	return regmap_update_bits(ksr->map, K210_SYSCTL_PERI_RESET, BIT(id), 0);
4262306a36Sopenharmony_ci}
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic int k210_rst_reset(struct reset_controller_dev *rcdev,
4562306a36Sopenharmony_ci			  unsigned long id)
4662306a36Sopenharmony_ci{
4762306a36Sopenharmony_ci	int ret;
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci	ret = k210_rst_assert(rcdev, id);
5062306a36Sopenharmony_ci	if (ret == 0) {
5162306a36Sopenharmony_ci		udelay(10);
5262306a36Sopenharmony_ci		ret = k210_rst_deassert(rcdev, id);
5362306a36Sopenharmony_ci	}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	return ret;
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_cistatic int k210_rst_status(struct reset_controller_dev *rcdev,
5962306a36Sopenharmony_ci			   unsigned long id)
6062306a36Sopenharmony_ci{
6162306a36Sopenharmony_ci	struct k210_rst *ksr = to_k210_rst(rcdev);
6262306a36Sopenharmony_ci	u32 reg, bit = BIT(id);
6362306a36Sopenharmony_ci	int ret;
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci	ret = regmap_read(ksr->map, K210_SYSCTL_PERI_RESET, &reg);
6662306a36Sopenharmony_ci	if (ret)
6762306a36Sopenharmony_ci		return ret;
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	return reg & bit;
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_cistatic int k210_rst_xlate(struct reset_controller_dev *rcdev,
7362306a36Sopenharmony_ci			  const struct of_phandle_args *reset_spec)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	unsigned long id = reset_spec->args[0];
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci	if (!(BIT(id) & K210_RST_MASK))
7862306a36Sopenharmony_ci		return -EINVAL;
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	return id;
8162306a36Sopenharmony_ci}
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_cistatic const struct reset_control_ops k210_rst_ops = {
8462306a36Sopenharmony_ci	.assert		= k210_rst_assert,
8562306a36Sopenharmony_ci	.deassert	= k210_rst_deassert,
8662306a36Sopenharmony_ci	.reset		= k210_rst_reset,
8762306a36Sopenharmony_ci	.status		= k210_rst_status,
8862306a36Sopenharmony_ci};
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_cistatic int k210_rst_probe(struct platform_device *pdev)
9162306a36Sopenharmony_ci{
9262306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
9362306a36Sopenharmony_ci	struct device_node *parent_np = of_get_parent(dev->of_node);
9462306a36Sopenharmony_ci	struct k210_rst *ksr;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	dev_info(dev, "K210 reset controller\n");
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	ksr = devm_kzalloc(dev, sizeof(*ksr), GFP_KERNEL);
9962306a36Sopenharmony_ci	if (!ksr)
10062306a36Sopenharmony_ci		return -ENOMEM;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	ksr->map = syscon_node_to_regmap(parent_np);
10362306a36Sopenharmony_ci	of_node_put(parent_np);
10462306a36Sopenharmony_ci	if (IS_ERR(ksr->map))
10562306a36Sopenharmony_ci		return PTR_ERR(ksr->map);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	ksr->rcdev.owner = THIS_MODULE;
10862306a36Sopenharmony_ci	ksr->rcdev.dev = dev;
10962306a36Sopenharmony_ci	ksr->rcdev.of_node = dev->of_node;
11062306a36Sopenharmony_ci	ksr->rcdev.ops = &k210_rst_ops;
11162306a36Sopenharmony_ci	ksr->rcdev.nr_resets = fls(K210_RST_MASK);
11262306a36Sopenharmony_ci	ksr->rcdev.of_reset_n_cells = 1;
11362306a36Sopenharmony_ci	ksr->rcdev.of_xlate = k210_rst_xlate;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	return devm_reset_controller_register(dev, &ksr->rcdev);
11662306a36Sopenharmony_ci}
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_cistatic const struct of_device_id k210_rst_dt_ids[] = {
11962306a36Sopenharmony_ci	{ .compatible = "canaan,k210-rst" },
12062306a36Sopenharmony_ci	{ /* sentinel */ },
12162306a36Sopenharmony_ci};
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_cistatic struct platform_driver k210_rst_driver = {
12462306a36Sopenharmony_ci	.probe	= k210_rst_probe,
12562306a36Sopenharmony_ci	.driver = {
12662306a36Sopenharmony_ci		.name		= "k210-rst",
12762306a36Sopenharmony_ci		.of_match_table	= k210_rst_dt_ids,
12862306a36Sopenharmony_ci	},
12962306a36Sopenharmony_ci};
13062306a36Sopenharmony_cibuiltin_platform_driver(k210_rst_driver);
131