1// SPDX-License-Identifier: GPL-2.0-or-later OR MIT
2/*
3 * Copyright (c) 2023 David Yang
4 */
5
6#include <linux/err.h>
7#include <linux/hw_random.h>
8#include <linux/io.h>
9#include <linux/iopoll.h>
10#include <linux/kernel.h>
11#include <linux/mod_devicetable.h>
12#include <linux/module.h>
13#include <linux/platform_device.h>
14
15#define RNG_CTRL		0x0
16#define  RNG_SOURCE			GENMASK(1, 0)
17#define  DROP_ENABLE			BIT(5)
18#define  POST_PROCESS_ENABLE		BIT(7)
19#define  POST_PROCESS_DEPTH		GENMASK(15, 8)
20#define RNG_NUMBER		0x4
21#define RNG_STAT		0x8
22#define  DATA_COUNT			GENMASK(2, 0)	/* max 4 */
23
24struct histb_rng_priv {
25	struct hwrng rng;
26	void __iomem *base;
27};
28
29/*
30 * Observed:
31 * depth = 1 -> ~1ms
32 * depth = 255 -> ~16ms
33 */
34static int histb_rng_wait(void __iomem *base)
35{
36	u32 val;
37
38	return readl_relaxed_poll_timeout(base + RNG_STAT, val,
39					  val & DATA_COUNT, 1000, 30 * 1000);
40}
41
42static void histb_rng_init(void __iomem *base, unsigned int depth)
43{
44	u32 val;
45
46	val = readl_relaxed(base + RNG_CTRL);
47
48	val &= ~RNG_SOURCE;
49	val |= 2;
50
51	val &= ~POST_PROCESS_DEPTH;
52	val |= min(depth, 0xffu) << 8;
53
54	val |= POST_PROCESS_ENABLE;
55	val |= DROP_ENABLE;
56
57	writel_relaxed(val, base + RNG_CTRL);
58}
59
60static int histb_rng_read(struct hwrng *rng, void *data, size_t max, bool wait)
61{
62	struct histb_rng_priv *priv = container_of(rng, typeof(*priv), rng);
63	void __iomem *base = priv->base;
64
65	for (int i = 0; i < max; i += sizeof(u32)) {
66		if (!(readl_relaxed(base + RNG_STAT) & DATA_COUNT)) {
67			if (!wait)
68				return i;
69			if (histb_rng_wait(base)) {
70				pr_err("failed to generate random number, generated %d\n",
71				       i);
72				return i ? i : -ETIMEDOUT;
73			}
74		}
75		*(u32 *) (data + i) = readl_relaxed(base + RNG_NUMBER);
76	}
77
78	return max;
79}
80
81static unsigned int histb_rng_get_depth(void __iomem *base)
82{
83	return (readl_relaxed(base + RNG_CTRL) & POST_PROCESS_DEPTH) >> 8;
84}
85
86static ssize_t
87depth_show(struct device *dev, struct device_attribute *attr, char *buf)
88{
89	struct histb_rng_priv *priv = dev_get_drvdata(dev);
90	void __iomem *base = priv->base;
91
92	return sprintf(buf, "%d\n", histb_rng_get_depth(base));
93}
94
95static ssize_t
96depth_store(struct device *dev, struct device_attribute *attr,
97	    const char *buf, size_t count)
98{
99	struct histb_rng_priv *priv = dev_get_drvdata(dev);
100	void __iomem *base = priv->base;
101	unsigned int depth;
102
103	if (kstrtouint(buf, 0, &depth))
104		return -ERANGE;
105
106	histb_rng_init(base, depth);
107	return count;
108}
109
110static DEVICE_ATTR_RW(depth);
111
112static struct attribute *histb_rng_attrs[] = {
113	&dev_attr_depth.attr,
114	NULL,
115};
116
117ATTRIBUTE_GROUPS(histb_rng);
118
119static int histb_rng_probe(struct platform_device *pdev)
120{
121	struct device *dev = &pdev->dev;
122	struct histb_rng_priv *priv;
123	void __iomem *base;
124	int ret;
125
126	priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
127	if (!priv)
128		return -ENOMEM;
129
130	base = devm_platform_ioremap_resource(pdev, 0);
131	if (IS_ERR(base))
132		return PTR_ERR(base);
133
134	histb_rng_init(base, 144);
135	if (histb_rng_wait(base)) {
136		dev_err(dev, "cannot bring up device\n");
137		return -ENODEV;
138	}
139
140	priv->base = base;
141	priv->rng.name = pdev->name;
142	priv->rng.read = histb_rng_read;
143	ret = devm_hwrng_register(dev, &priv->rng);
144	if (ret) {
145		dev_err(dev, "failed to register hwrng: %d\n", ret);
146		return ret;
147	}
148
149	platform_set_drvdata(pdev, priv);
150	dev_set_drvdata(dev, priv);
151	return 0;
152}
153
154static const struct of_device_id histb_rng_of_match[] = {
155	{ .compatible = "hisilicon,histb-rng", },
156	{ }
157};
158MODULE_DEVICE_TABLE(of, histb_rng_of_match);
159
160static struct platform_driver histb_rng_driver = {
161	.probe = histb_rng_probe,
162	.driver = {
163		.name = "histb-rng",
164		.of_match_table = histb_rng_of_match,
165		.dev_groups = histb_rng_groups,
166	},
167};
168
169module_platform_driver(histb_rng_driver);
170
171MODULE_DESCRIPTION("Hisilicon STB random number generator driver");
172MODULE_LICENSE("Dual MIT/GPL");
173MODULE_AUTHOR("David Yang <mmyangfl@gmail.com>");
174