162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Ingenic Random Number Generator driver
462306a36Sopenharmony_ci * Copyright (c) 2017 PrasannaKumar Muralidharan <prasannatsmkumar@gmail.com>
562306a36Sopenharmony_ci * Copyright (c) 2020 周琰杰 (Zhou Yanjie) <zhouyanjie@wanyeetech.com>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/err.h>
962306a36Sopenharmony_ci#include <linux/kernel.h>
1062306a36Sopenharmony_ci#include <linux/hw_random.h>
1162306a36Sopenharmony_ci#include <linux/io.h>
1262306a36Sopenharmony_ci#include <linux/iopoll.h>
1362306a36Sopenharmony_ci#include <linux/module.h>
1462306a36Sopenharmony_ci#include <linux/of_device.h>
1562306a36Sopenharmony_ci#include <linux/platform_device.h>
1662306a36Sopenharmony_ci#include <linux/slab.h>
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci/* RNG register offsets */
1962306a36Sopenharmony_ci#define RNG_REG_ERNG_OFFSET		0x0
2062306a36Sopenharmony_ci#define RNG_REG_RNG_OFFSET		0x4
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci/* bits within the ERND register */
2362306a36Sopenharmony_ci#define ERNG_READY				BIT(31)
2462306a36Sopenharmony_ci#define ERNG_ENABLE				BIT(0)
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cienum ingenic_rng_version {
2762306a36Sopenharmony_ci	ID_JZ4780,
2862306a36Sopenharmony_ci	ID_X1000,
2962306a36Sopenharmony_ci};
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci/* Device associated memory */
3262306a36Sopenharmony_cistruct ingenic_rng {
3362306a36Sopenharmony_ci	enum ingenic_rng_version version;
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	void __iomem *base;
3662306a36Sopenharmony_ci	struct hwrng rng;
3762306a36Sopenharmony_ci};
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_cistatic int ingenic_rng_init(struct hwrng *rng)
4062306a36Sopenharmony_ci{
4162306a36Sopenharmony_ci	struct ingenic_rng *priv = container_of(rng, struct ingenic_rng, rng);
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	writel(ERNG_ENABLE, priv->base + RNG_REG_ERNG_OFFSET);
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	return 0;
4662306a36Sopenharmony_ci}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_cistatic void ingenic_rng_cleanup(struct hwrng *rng)
4962306a36Sopenharmony_ci{
5062306a36Sopenharmony_ci	struct ingenic_rng *priv = container_of(rng, struct ingenic_rng, rng);
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	writel(0, priv->base + RNG_REG_ERNG_OFFSET);
5362306a36Sopenharmony_ci}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_cistatic int ingenic_rng_read(struct hwrng *rng, void *buf, size_t max, bool wait)
5662306a36Sopenharmony_ci{
5762306a36Sopenharmony_ci	struct ingenic_rng *priv = container_of(rng, struct ingenic_rng, rng);
5862306a36Sopenharmony_ci	u32 *data = buf;
5962306a36Sopenharmony_ci	u32 status;
6062306a36Sopenharmony_ci	int ret;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	if (priv->version >= ID_X1000) {
6362306a36Sopenharmony_ci		ret = readl_poll_timeout(priv->base + RNG_REG_ERNG_OFFSET, status,
6462306a36Sopenharmony_ci					 status & ERNG_READY, 10, 1000);
6562306a36Sopenharmony_ci		if (ret == -ETIMEDOUT) {
6662306a36Sopenharmony_ci			pr_err("%s: Wait for RNG data ready timeout\n", __func__);
6762306a36Sopenharmony_ci			return ret;
6862306a36Sopenharmony_ci		}
6962306a36Sopenharmony_ci	} else {
7062306a36Sopenharmony_ci		/*
7162306a36Sopenharmony_ci		 * A delay is required so that the current RNG data is not bit shifted
7262306a36Sopenharmony_ci		 * version of previous RNG data which could happen if random data is
7362306a36Sopenharmony_ci		 * read continuously from this device.
7462306a36Sopenharmony_ci		 */
7562306a36Sopenharmony_ci		udelay(20);
7662306a36Sopenharmony_ci	}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	*data = readl(priv->base + RNG_REG_RNG_OFFSET);
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	return 4;
8162306a36Sopenharmony_ci}
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_cistatic int ingenic_rng_probe(struct platform_device *pdev)
8462306a36Sopenharmony_ci{
8562306a36Sopenharmony_ci	struct ingenic_rng *priv;
8662306a36Sopenharmony_ci	int ret;
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
8962306a36Sopenharmony_ci	if (!priv)
9062306a36Sopenharmony_ci		return -ENOMEM;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	priv->base = devm_platform_ioremap_resource(pdev, 0);
9362306a36Sopenharmony_ci	if (IS_ERR(priv->base)) {
9462306a36Sopenharmony_ci		pr_err("%s: Failed to map RNG registers\n", __func__);
9562306a36Sopenharmony_ci		return PTR_ERR(priv->base);
9662306a36Sopenharmony_ci	}
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	priv->version = (enum ingenic_rng_version)(uintptr_t)of_device_get_match_data(&pdev->dev);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	priv->rng.name = pdev->name;
10162306a36Sopenharmony_ci	priv->rng.init = ingenic_rng_init;
10262306a36Sopenharmony_ci	priv->rng.cleanup = ingenic_rng_cleanup;
10362306a36Sopenharmony_ci	priv->rng.read = ingenic_rng_read;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	ret = hwrng_register(&priv->rng);
10662306a36Sopenharmony_ci	if (ret) {
10762306a36Sopenharmony_ci		dev_err(&pdev->dev, "Failed to register hwrng\n");
10862306a36Sopenharmony_ci		return ret;
10962306a36Sopenharmony_ci	}
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	platform_set_drvdata(pdev, priv);
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	dev_info(&pdev->dev, "Ingenic RNG driver registered\n");
11462306a36Sopenharmony_ci	return 0;
11562306a36Sopenharmony_ci}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_cistatic int ingenic_rng_remove(struct platform_device *pdev)
11862306a36Sopenharmony_ci{
11962306a36Sopenharmony_ci	struct ingenic_rng *priv = platform_get_drvdata(pdev);
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	hwrng_unregister(&priv->rng);
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	writel(0, priv->base + RNG_REG_ERNG_OFFSET);
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	return 0;
12662306a36Sopenharmony_ci}
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_cistatic const struct of_device_id ingenic_rng_of_match[] = {
12962306a36Sopenharmony_ci	{ .compatible = "ingenic,jz4780-rng", .data = (void *) ID_JZ4780 },
13062306a36Sopenharmony_ci	{ .compatible = "ingenic,x1000-rng", .data = (void *) ID_X1000 },
13162306a36Sopenharmony_ci	{ /* sentinel */ }
13262306a36Sopenharmony_ci};
13362306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, ingenic_rng_of_match);
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_cistatic struct platform_driver ingenic_rng_driver = {
13662306a36Sopenharmony_ci	.probe		= ingenic_rng_probe,
13762306a36Sopenharmony_ci	.remove		= ingenic_rng_remove,
13862306a36Sopenharmony_ci	.driver		= {
13962306a36Sopenharmony_ci		.name	= "ingenic-rng",
14062306a36Sopenharmony_ci		.of_match_table = ingenic_rng_of_match,
14162306a36Sopenharmony_ci	},
14262306a36Sopenharmony_ci};
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_cimodule_platform_driver(ingenic_rng_driver);
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ciMODULE_LICENSE("GPL");
14762306a36Sopenharmony_ciMODULE_AUTHOR("PrasannaKumar Muralidharan <prasannatsmkumar@gmail.com>");
14862306a36Sopenharmony_ciMODULE_AUTHOR("周琰杰 (Zhou Yanjie) <zhouyanjie@wanyeetech.com>");
14962306a36Sopenharmony_ciMODULE_DESCRIPTION("Ingenic Random Number Generator driver");
150