1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * TRNG driver for the StarFive JH7110 SoC
4  *
5  * Copyright (C) 2022 StarFive Technology Co.
6  */
7 
8 #include <linux/clk.h>
9 #include <linux/completion.h>
10 #include <linux/delay.h>
11 #include <linux/err.h>
12 #include <linux/hw_random.h>
13 #include <linux/interrupt.h>
14 #include <linux/io.h>
15 #include <linux/iopoll.h>
16 #include <linux/kernel.h>
17 #include <linux/module.h>
18 #include <linux/of.h>
19 #include <linux/platform_device.h>
20 #include <linux/pm_runtime.h>
21 #include <linux/random.h>
22 #include <linux/reset.h>
23 
24 /* trng register offset */
25 #define STARFIVE_CTRL			0x00
26 #define STARFIVE_STAT			0x04
27 #define STARFIVE_MODE			0x08
28 #define STARFIVE_SMODE			0x0C
29 #define STARFIVE_IE			0x10
30 #define STARFIVE_ISTAT			0x14
31 #define STARFIVE_RAND0			0x20
32 #define STARFIVE_RAND1			0x24
33 #define STARFIVE_RAND2			0x28
34 #define STARFIVE_RAND3			0x2C
35 #define STARFIVE_RAND4			0x30
36 #define STARFIVE_RAND5			0x34
37 #define STARFIVE_RAND6			0x38
38 #define STARFIVE_RAND7			0x3C
39 #define STARFIVE_AUTO_RQSTS		0x60
40 #define STARFIVE_AUTO_AGE		0x64
41 
42 /* CTRL CMD */
43 #define STARFIVE_CTRL_EXEC_NOP		0x0
44 #define STARFIVE_CTRL_GENE_RANDNUM	0x1
45 #define STARFIVE_CTRL_EXEC_RANDRESEED	0x2
46 
47 /* STAT */
48 #define STARFIVE_STAT_NONCE_MODE	BIT(2)
49 #define STARFIVE_STAT_R256		BIT(3)
50 #define STARFIVE_STAT_MISSION_MODE	BIT(8)
51 #define STARFIVE_STAT_SEEDED		BIT(9)
52 #define STARFIVE_STAT_LAST_RESEED(x)	((x) << 16)
53 #define STARFIVE_STAT_SRVC_RQST		BIT(27)
54 #define STARFIVE_STAT_RAND_GENERATING	BIT(30)
55 #define STARFIVE_STAT_RAND_SEEDING	BIT(31)
56 
57 /* MODE */
58 #define STARFIVE_MODE_R256		BIT(3)
59 
60 /* SMODE */
61 #define STARFIVE_SMODE_NONCE_MODE	BIT(2)
62 #define STARFIVE_SMODE_MISSION_MODE	BIT(8)
63 #define STARFIVE_SMODE_MAX_REJECTS(x)	((x) << 16)
64 
65 /* IE */
66 #define STARFIVE_IE_RAND_RDY_EN		BIT(0)
67 #define STARFIVE_IE_SEED_DONE_EN	BIT(1)
68 #define STARFIVE_IE_LFSR_LOCKUP_EN	BIT(4)
69 #define STARFIVE_IE_GLBL_EN		BIT(31)
70 
71 #define STARFIVE_IE_ALL			(STARFIVE_IE_GLBL_EN | \
72 					 STARFIVE_IE_RAND_RDY_EN | \
73 					 STARFIVE_IE_SEED_DONE_EN | \
74 					 STARFIVE_IE_LFSR_LOCKUP_EN)
75 
76 /* ISTAT */
77 #define STARFIVE_ISTAT_RAND_RDY		BIT(0)
78 #define STARFIVE_ISTAT_SEED_DONE	BIT(1)
79 #define STARFIVE_ISTAT_LFSR_LOCKUP	BIT(4)
80 
81 #define STARFIVE_RAND_LEN		sizeof(u32)
82 
83 #define to_trng(p)			container_of(p, struct starfive_trng, rng)
84 
85 enum reseed {
86 	RANDOM_RESEED,
87 	NONCE_RESEED,
88 };
89 
90 enum mode {
91 	PRNG_128BIT,
92 	PRNG_256BIT,
93 };
94 
95 struct starfive_trng {
96 	struct device		*dev;
97 	void __iomem		*base;
98 	struct clk		*hclk;
99 	struct clk		*ahb;
100 	struct reset_control	*rst;
101 	struct hwrng		rng;
102 	struct completion	random_done;
103 	struct completion	reseed_done;
104 	u32			mode;
105 	u32			mission;
106 	u32			reseed;
107 	/* protects against concurrent write to ctrl register */
108 	spinlock_t		write_lock;
109 };
110 
111 static u16 autoreq;
112 module_param(autoreq, ushort, 0);
113 MODULE_PARM_DESC(autoreq, "Auto-reseeding after random number requests by host reaches specified counter:\n"
114 				" 0 - disable counter\n"
115 				" other - reload value for internal counter");
116 
117 static u16 autoage;
118 module_param(autoage, ushort, 0);
119 MODULE_PARM_DESC(autoage, "Auto-reseeding after specified timer countdowns to 0:\n"
120 				" 0 - disable timer\n"
121 				" other - reload value for internal timer");
122 
123 static inline int starfive_trng_wait_idle(struct starfive_trng *trng)
124 {
125 	u32 stat;
126 
127 	return readl_relaxed_poll_timeout(trng->base + STARFIVE_STAT, stat,
128 					  !(stat & (STARFIVE_STAT_RAND_GENERATING |
129 						    STARFIVE_STAT_RAND_SEEDING)),
130 					  10, 100000);
131 }
132 
133 static inline void starfive_trng_irq_mask_clear(struct starfive_trng *trng)
134 {
135 	/* clear register: ISTAT */
136 	u32 data = readl(trng->base + STARFIVE_ISTAT);
137 
138 	writel(data, trng->base + STARFIVE_ISTAT);
139 }
140 
141 static int starfive_trng_cmd(struct starfive_trng *trng, u32 cmd, bool wait)
142 {
143 	int wait_time = 1000;
144 
145 	/* allow up to 40 us for wait == 0 */
146 	if (!wait)
147 		wait_time = 40;
148 
149 	switch (cmd) {
150 	case STARFIVE_CTRL_GENE_RANDNUM:
151 		reinit_completion(&trng->random_done);
152 		spin_lock_irq(&trng->write_lock);
153 		writel(cmd, trng->base + STARFIVE_CTRL);
154 		spin_unlock_irq(&trng->write_lock);
155 		if (!wait_for_completion_timeout(&trng->random_done, usecs_to_jiffies(wait_time)))
156 			return -ETIMEDOUT;
157 		break;
158 	case STARFIVE_CTRL_EXEC_RANDRESEED:
159 		reinit_completion(&trng->reseed_done);
160 		spin_lock_irq(&trng->write_lock);
161 		writel(cmd, trng->base + STARFIVE_CTRL);
162 		spin_unlock_irq(&trng->write_lock);
163 		if (!wait_for_completion_timeout(&trng->reseed_done, usecs_to_jiffies(wait_time)))
164 			return -ETIMEDOUT;
165 		break;
166 	default:
167 		return -EINVAL;
168 	}
169 
170 	return 0;
171 }
172 
173 static int starfive_trng_init(struct hwrng *rng)
174 {
175 	struct starfive_trng *trng = to_trng(rng);
176 	u32 mode, intr = 0;
177 
178 	/* setup Auto Request/Age register */
179 	writel(autoage, trng->base + STARFIVE_AUTO_AGE);
180 	writel(autoreq, trng->base + STARFIVE_AUTO_RQSTS);
181 
182 	/* clear register: ISTAT */
183 	starfive_trng_irq_mask_clear(trng);
184 
185 	intr |= STARFIVE_IE_ALL;
186 	writel(intr, trng->base + STARFIVE_IE);
187 
188 	mode  = readl(trng->base + STARFIVE_MODE);
189 
190 	switch (trng->mode) {
191 	case PRNG_128BIT:
192 		mode &= ~STARFIVE_MODE_R256;
193 		break;
194 	case PRNG_256BIT:
195 		mode |= STARFIVE_MODE_R256;
196 		break;
197 	default:
198 		mode |= STARFIVE_MODE_R256;
199 		break;
200 	}
201 
202 	writel(mode, trng->base + STARFIVE_MODE);
203 
204 	return starfive_trng_cmd(trng, STARFIVE_CTRL_EXEC_RANDRESEED, 1);
205 }
206 
207 static irqreturn_t starfive_trng_irq(int irq, void *priv)
208 {
209 	u32 status;
210 	struct starfive_trng *trng = (struct starfive_trng *)priv;
211 
212 	status = readl(trng->base + STARFIVE_ISTAT);
213 	if (status & STARFIVE_ISTAT_RAND_RDY) {
214 		writel(STARFIVE_ISTAT_RAND_RDY, trng->base + STARFIVE_ISTAT);
215 		complete(&trng->random_done);
216 	}
217 
218 	if (status & STARFIVE_ISTAT_SEED_DONE) {
219 		writel(STARFIVE_ISTAT_SEED_DONE, trng->base + STARFIVE_ISTAT);
220 		complete(&trng->reseed_done);
221 	}
222 
223 	if (status & STARFIVE_ISTAT_LFSR_LOCKUP) {
224 		writel(STARFIVE_ISTAT_LFSR_LOCKUP, trng->base + STARFIVE_ISTAT);
225 		/* SEU occurred, reseeding required*/
226 		spin_lock(&trng->write_lock);
227 		writel(STARFIVE_CTRL_EXEC_RANDRESEED, trng->base + STARFIVE_CTRL);
228 		spin_unlock(&trng->write_lock);
229 	}
230 
231 	return IRQ_HANDLED;
232 }
233 
234 static void starfive_trng_cleanup(struct hwrng *rng)
235 {
236 	struct starfive_trng *trng = to_trng(rng);
237 
238 	writel(0, trng->base + STARFIVE_CTRL);
239 
240 	reset_control_assert(trng->rst);
241 	clk_disable_unprepare(trng->hclk);
242 	clk_disable_unprepare(trng->ahb);
243 }
244 
245 static int starfive_trng_read(struct hwrng *rng, void *buf, size_t max, bool wait)
246 {
247 	struct starfive_trng *trng = to_trng(rng);
248 	int ret;
249 
250 	pm_runtime_get_sync(trng->dev);
251 
252 	if (trng->mode == PRNG_256BIT)
253 		max = min_t(size_t, max, (STARFIVE_RAND_LEN * 8));
254 	else
255 		max = min_t(size_t, max, (STARFIVE_RAND_LEN * 4));
256 
257 	if (wait) {
258 		ret = starfive_trng_wait_idle(trng);
259 		if (ret)
260 			return -ETIMEDOUT;
261 	}
262 
263 	ret = starfive_trng_cmd(trng, STARFIVE_CTRL_GENE_RANDNUM, wait);
264 	if (ret)
265 		return ret;
266 
267 	memcpy_fromio(buf, trng->base + STARFIVE_RAND0, max);
268 
269 	pm_runtime_put_sync_autosuspend(trng->dev);
270 
271 	return max;
272 }
273 
274 static int starfive_trng_probe(struct platform_device *pdev)
275 {
276 	int ret;
277 	int irq;
278 	struct starfive_trng *trng;
279 
280 	trng = devm_kzalloc(&pdev->dev, sizeof(*trng), GFP_KERNEL);
281 	if (!trng)
282 		return -ENOMEM;
283 
284 	platform_set_drvdata(pdev, trng);
285 	trng->dev = &pdev->dev;
286 
287 	trng->base = devm_platform_ioremap_resource(pdev, 0);
288 	if (IS_ERR(trng->base))
289 		return dev_err_probe(&pdev->dev, PTR_ERR(trng->base),
290 				     "Error remapping memory for platform device.\n");
291 
292 	irq = platform_get_irq(pdev, 0);
293 	if (irq < 0)
294 		return irq;
295 
296 	init_completion(&trng->random_done);
297 	init_completion(&trng->reseed_done);
298 	spin_lock_init(&trng->write_lock);
299 
300 	ret = devm_request_irq(&pdev->dev, irq, starfive_trng_irq, 0, pdev->name,
301 			       (void *)trng);
302 	if (ret)
303 		return dev_err_probe(&pdev->dev, irq,
304 				     "Failed to register interrupt handler\n");
305 
306 	trng->hclk = devm_clk_get(&pdev->dev, "hclk");
307 	if (IS_ERR(trng->hclk))
308 		return dev_err_probe(&pdev->dev, PTR_ERR(trng->hclk),
309 				     "Error getting hardware reference clock\n");
310 
311 	trng->ahb = devm_clk_get(&pdev->dev, "ahb");
312 	if (IS_ERR(trng->ahb))
313 		return dev_err_probe(&pdev->dev, PTR_ERR(trng->ahb),
314 				     "Error getting ahb reference clock\n");
315 
316 	trng->rst = devm_reset_control_get_shared(&pdev->dev, NULL);
317 	if (IS_ERR(trng->rst))
318 		return dev_err_probe(&pdev->dev, PTR_ERR(trng->rst),
319 				     "Error getting hardware reset line\n");
320 
321 	clk_prepare_enable(trng->hclk);
322 	clk_prepare_enable(trng->ahb);
323 	reset_control_deassert(trng->rst);
324 
325 	trng->rng.name = dev_driver_string(&pdev->dev);
326 	trng->rng.init = starfive_trng_init;
327 	trng->rng.cleanup = starfive_trng_cleanup;
328 	trng->rng.read = starfive_trng_read;
329 
330 	trng->mode = PRNG_256BIT;
331 	trng->mission = 1;
332 	trng->reseed = RANDOM_RESEED;
333 
334 	pm_runtime_use_autosuspend(&pdev->dev);
335 	pm_runtime_set_autosuspend_delay(&pdev->dev, 100);
336 	pm_runtime_enable(&pdev->dev);
337 
338 	ret = devm_hwrng_register(&pdev->dev, &trng->rng);
339 	if (ret) {
340 		pm_runtime_disable(&pdev->dev);
341 
342 		reset_control_assert(trng->rst);
343 		clk_disable_unprepare(trng->ahb);
344 		clk_disable_unprepare(trng->hclk);
345 
346 		return dev_err_probe(&pdev->dev, ret, "Failed to register hwrng\n");
347 	}
348 
349 	return 0;
350 }
351 
352 static int __maybe_unused starfive_trng_suspend(struct device *dev)
353 {
354 	struct starfive_trng *trng = dev_get_drvdata(dev);
355 
356 	clk_disable_unprepare(trng->hclk);
357 	clk_disable_unprepare(trng->ahb);
358 
359 	return 0;
360 }
361 
362 static int __maybe_unused starfive_trng_resume(struct device *dev)
363 {
364 	struct starfive_trng *trng = dev_get_drvdata(dev);
365 
366 	clk_prepare_enable(trng->hclk);
367 	clk_prepare_enable(trng->ahb);
368 
369 	return 0;
370 }
371 
372 static DEFINE_SIMPLE_DEV_PM_OPS(starfive_trng_pm_ops, starfive_trng_suspend,
373 				starfive_trng_resume);
374 
375 static const struct of_device_id trng_dt_ids[] __maybe_unused = {
376 	{ .compatible = "starfive,jh7110-trng" },
377 	{ }
378 };
379 MODULE_DEVICE_TABLE(of, trng_dt_ids);
380 
381 static struct platform_driver starfive_trng_driver = {
382 	.probe	= starfive_trng_probe,
383 	.driver	= {
384 		.name		= "jh7110-trng",
385 		.pm		= &starfive_trng_pm_ops,
386 		.of_match_table	= of_match_ptr(trng_dt_ids),
387 	},
388 };
389 
390 module_platform_driver(starfive_trng_driver);
391 
392 MODULE_LICENSE("GPL");
393 MODULE_DESCRIPTION("StarFive True Random Number Generator");
394