xref: /linux/drivers/char/hw_random/n2-drv.c (revision 975ef7ff81bb000af6e6c8e63e81f89f3468dcf7)
1 /* n2-drv.c: Niagara-2 RNG driver.
2  *
3  * Copyright (C) 2008, 2011 David S. Miller <davem@davemloft.net>
4  */
5 
6 #include <linux/kernel.h>
7 #include <linux/module.h>
8 #include <linux/types.h>
9 #include <linux/delay.h>
10 #include <linux/slab.h>
11 #include <linux/workqueue.h>
12 #include <linux/preempt.h>
13 #include <linux/hw_random.h>
14 
15 #include <linux/of.h>
16 #include <linux/of_device.h>
17 
18 #include <asm/hypervisor.h>
19 
20 #include "n2rng.h"
21 
22 #define DRV_MODULE_NAME		"n2rng"
23 #define PFX DRV_MODULE_NAME	": "
24 #define DRV_MODULE_VERSION	"0.3"
25 #define DRV_MODULE_RELDATE	"Jan 7, 2017"
26 
27 static char version[] =
28 	DRV_MODULE_NAME " v" DRV_MODULE_VERSION " (" DRV_MODULE_RELDATE ")\n";
29 
30 MODULE_AUTHOR("David S. Miller (davem@davemloft.net)");
31 MODULE_DESCRIPTION("Niagara2 RNG driver");
32 MODULE_LICENSE("GPL");
33 MODULE_VERSION(DRV_MODULE_VERSION);
34 
35 /* The Niagara2 RNG provides a 64-bit read-only random number
36  * register, plus a control register.  Access to the RNG is
37  * virtualized through the hypervisor so that both guests and control
38  * nodes can access the device.
39  *
40  * The entropy source consists of raw entropy sources, each
41  * constructed from a voltage controlled oscillator whose phase is
42  * jittered by thermal noise sources.
43  *
44  * The oscillator in each of the three raw entropy sources run at
45  * different frequencies.  Normally, all three generator outputs are
46  * gathered, xored together, and fed into a CRC circuit, the output of
47  * which is the 64-bit read-only register.
48  *
49  * Some time is necessary for all the necessary entropy to build up
50  * such that a full 64-bits of entropy are available in the register.
51  * In normal operating mode (RNG_CTL_LFSR is set), the chip implements
52  * an interlock which blocks register reads until sufficient entropy
53  * is available.
54  *
55  * A control register is provided for adjusting various aspects of RNG
56  * operation, and to enable diagnostic modes.  Each of the three raw
57  * entropy sources has an enable bit (RNG_CTL_ES{1,2,3}).  Also
58  * provided are fields for controlling the minimum time in cycles
59  * between read accesses to the register (RNG_CTL_WAIT, this controls
60  * the interlock described in the previous paragraph).
61  *
62  * The standard setting is to have the mode bit (RNG_CTL_LFSR) set,
63  * all three entropy sources enabled, and the interlock time set
64  * appropriately.
65  *
66  * The CRC polynomial used by the chip is:
67  *
68  * P(X) = x64 + x61 + x57 + x56 + x52 + x51 + x50 + x48 + x47 + x46 +
69  *        x43 + x42 + x41 + x39 + x38 + x37 + x35 + x32 + x28 + x25 +
70  *        x22 + x21 + x17 + x15 + x13 + x12 + x11 + x7 + x5 + x + 1
71  *
72  * The RNG_CTL_VCO value of each noise cell must be programmed
73  * separately.  This is why 4 control register values must be provided
74  * to the hypervisor.  During a write, the hypervisor writes them all,
75  * one at a time, to the actual RNG_CTL register.  The first three
76  * values are used to setup the desired RNG_CTL_VCO for each entropy
77  * source, for example:
78  *
79  *	control 0: (1 << RNG_CTL_VCO_SHIFT) | RNG_CTL_ES1
80  *	control 1: (2 << RNG_CTL_VCO_SHIFT) | RNG_CTL_ES2
81  *	control 2: (3 << RNG_CTL_VCO_SHIFT) | RNG_CTL_ES3
82  *
83  * And then the fourth value sets the final chip state and enables
84  * desired.
85  */
86 
87 static int n2rng_hv_err_trans(unsigned long hv_err)
88 {
89 	switch (hv_err) {
90 	case HV_EOK:
91 		return 0;
92 	case HV_EWOULDBLOCK:
93 		return -EAGAIN;
94 	case HV_ENOACCESS:
95 		return -EPERM;
96 	case HV_EIO:
97 		return -EIO;
98 	case HV_EBUSY:
99 		return -EBUSY;
100 	case HV_EBADALIGN:
101 	case HV_ENORADDR:
102 		return -EFAULT;
103 	default:
104 		return -EINVAL;
105 	}
106 }
107 
108 static unsigned long n2rng_generic_read_control_v2(unsigned long ra,
109 						   unsigned long unit)
110 {
111 	unsigned long hv_err, state, ticks, watchdog_delta, watchdog_status;
112 	int block = 0, busy = 0;
113 
114 	while (1) {
115 		hv_err = sun4v_rng_ctl_read_v2(ra, unit, &state,
116 					       &ticks,
117 					       &watchdog_delta,
118 					       &watchdog_status);
119 		if (hv_err == HV_EOK)
120 			break;
121 
122 		if (hv_err == HV_EBUSY) {
123 			if (++busy >= N2RNG_BUSY_LIMIT)
124 				break;
125 
126 			udelay(1);
127 		} else if (hv_err == HV_EWOULDBLOCK) {
128 			if (++block >= N2RNG_BLOCK_LIMIT)
129 				break;
130 
131 			__delay(ticks);
132 		} else
133 			break;
134 	}
135 
136 	return hv_err;
137 }
138 
139 /* In multi-socket situations, the hypervisor might need to
140  * queue up the RNG control register write if it's for a unit
141  * that is on a cpu socket other than the one we are executing on.
142  *
143  * We poll here waiting for a successful read of that control
144  * register to make sure the write has been actually performed.
145  */
146 static unsigned long n2rng_control_settle_v2(struct n2rng *np, int unit)
147 {
148 	unsigned long ra = __pa(&np->scratch_control[0]);
149 
150 	return n2rng_generic_read_control_v2(ra, unit);
151 }
152 
153 static unsigned long n2rng_write_ctl_one(struct n2rng *np, int unit,
154 					 unsigned long state,
155 					 unsigned long control_ra,
156 					 unsigned long watchdog_timeout,
157 					 unsigned long *ticks)
158 {
159 	unsigned long hv_err;
160 
161 	if (np->hvapi_major == 1) {
162 		hv_err = sun4v_rng_ctl_write_v1(control_ra, state,
163 						watchdog_timeout, ticks);
164 	} else {
165 		hv_err = sun4v_rng_ctl_write_v2(control_ra, state,
166 						watchdog_timeout, unit);
167 		if (hv_err == HV_EOK)
168 			hv_err = n2rng_control_settle_v2(np, unit);
169 		*ticks = N2RNG_ACCUM_CYCLES_DEFAULT;
170 	}
171 
172 	return hv_err;
173 }
174 
175 static int n2rng_generic_read_data(unsigned long data_ra)
176 {
177 	unsigned long ticks, hv_err;
178 	int block = 0, hcheck = 0;
179 
180 	while (1) {
181 		hv_err = sun4v_rng_data_read(data_ra, &ticks);
182 		if (hv_err == HV_EOK)
183 			return 0;
184 
185 		if (hv_err == HV_EWOULDBLOCK) {
186 			if (++block >= N2RNG_BLOCK_LIMIT)
187 				return -EWOULDBLOCK;
188 			__delay(ticks);
189 		} else if (hv_err == HV_ENOACCESS) {
190 			return -EPERM;
191 		} else if (hv_err == HV_EIO) {
192 			if (++hcheck >= N2RNG_HCHECK_LIMIT)
193 				return -EIO;
194 			udelay(10000);
195 		} else
196 			return -ENODEV;
197 	}
198 }
199 
200 static unsigned long n2rng_read_diag_data_one(struct n2rng *np,
201 					      unsigned long unit,
202 					      unsigned long data_ra,
203 					      unsigned long data_len,
204 					      unsigned long *ticks)
205 {
206 	unsigned long hv_err;
207 
208 	if (np->hvapi_major == 1) {
209 		hv_err = sun4v_rng_data_read_diag_v1(data_ra, data_len, ticks);
210 	} else {
211 		hv_err = sun4v_rng_data_read_diag_v2(data_ra, data_len,
212 						     unit, ticks);
213 		if (!*ticks)
214 			*ticks = N2RNG_ACCUM_CYCLES_DEFAULT;
215 	}
216 	return hv_err;
217 }
218 
219 static int n2rng_generic_read_diag_data(struct n2rng *np,
220 					unsigned long unit,
221 					unsigned long data_ra,
222 					unsigned long data_len)
223 {
224 	unsigned long ticks, hv_err;
225 	int block = 0;
226 
227 	while (1) {
228 		hv_err = n2rng_read_diag_data_one(np, unit,
229 						  data_ra, data_len,
230 						  &ticks);
231 		if (hv_err == HV_EOK)
232 			return 0;
233 
234 		if (hv_err == HV_EWOULDBLOCK) {
235 			if (++block >= N2RNG_BLOCK_LIMIT)
236 				return -EWOULDBLOCK;
237 			__delay(ticks);
238 		} else if (hv_err == HV_ENOACCESS) {
239 			return -EPERM;
240 		} else if (hv_err == HV_EIO) {
241 			return -EIO;
242 		} else
243 			return -ENODEV;
244 	}
245 }
246 
247 
248 static int n2rng_generic_write_control(struct n2rng *np,
249 				       unsigned long control_ra,
250 				       unsigned long unit,
251 				       unsigned long state)
252 {
253 	unsigned long hv_err, ticks;
254 	int block = 0, busy = 0;
255 
256 	while (1) {
257 		hv_err = n2rng_write_ctl_one(np, unit, state, control_ra,
258 					     np->wd_timeo, &ticks);
259 		if (hv_err == HV_EOK)
260 			return 0;
261 
262 		if (hv_err == HV_EWOULDBLOCK) {
263 			if (++block >= N2RNG_BLOCK_LIMIT)
264 				return -EWOULDBLOCK;
265 			__delay(ticks);
266 		} else if (hv_err == HV_EBUSY) {
267 			if (++busy >= N2RNG_BUSY_LIMIT)
268 				return -EBUSY;
269 			udelay(1);
270 		} else
271 			return -ENODEV;
272 	}
273 }
274 
275 /* Just try to see if we can successfully access the control register
276  * of the RNG on the domain on which we are currently executing.
277  */
278 static int n2rng_try_read_ctl(struct n2rng *np)
279 {
280 	unsigned long hv_err;
281 	unsigned long x;
282 
283 	if (np->hvapi_major == 1) {
284 		hv_err = sun4v_rng_get_diag_ctl();
285 	} else {
286 		/* We purposefully give invalid arguments, HV_NOACCESS
287 		 * is higher priority than the errors we'd get from
288 		 * these other cases, and that's the error we are
289 		 * truly interested in.
290 		 */
291 		hv_err = sun4v_rng_ctl_read_v2(0UL, ~0UL, &x, &x, &x, &x);
292 		switch (hv_err) {
293 		case HV_EWOULDBLOCK:
294 		case HV_ENOACCESS:
295 			break;
296 		default:
297 			hv_err = HV_EOK;
298 			break;
299 		}
300 	}
301 
302 	return n2rng_hv_err_trans(hv_err);
303 }
304 
305 static u64 n2rng_control_default(struct n2rng *np, int ctl)
306 {
307 	u64 val = 0;
308 
309 	if (np->data->chip_version == 1) {
310 		val = ((2 << RNG_v1_CTL_ASEL_SHIFT) |
311 			(N2RNG_ACCUM_CYCLES_DEFAULT << RNG_v1_CTL_WAIT_SHIFT) |
312 			 RNG_CTL_LFSR);
313 
314 		switch (ctl) {
315 		case 0:
316 			val |= (1 << RNG_v1_CTL_VCO_SHIFT) | RNG_CTL_ES1;
317 			break;
318 		case 1:
319 			val |= (2 << RNG_v1_CTL_VCO_SHIFT) | RNG_CTL_ES2;
320 			break;
321 		case 2:
322 			val |= (3 << RNG_v1_CTL_VCO_SHIFT) | RNG_CTL_ES3;
323 			break;
324 		case 3:
325 			val |= RNG_CTL_ES1 | RNG_CTL_ES2 | RNG_CTL_ES3;
326 			break;
327 		default:
328 			break;
329 		}
330 
331 	} else {
332 		val = ((2 << RNG_v2_CTL_ASEL_SHIFT) |
333 			(N2RNG_ACCUM_CYCLES_DEFAULT << RNG_v2_CTL_WAIT_SHIFT) |
334 			 RNG_CTL_LFSR);
335 
336 		switch (ctl) {
337 		case 0:
338 			val |= (1 << RNG_v2_CTL_VCO_SHIFT) | RNG_CTL_ES1;
339 			break;
340 		case 1:
341 			val |= (2 << RNG_v2_CTL_VCO_SHIFT) | RNG_CTL_ES2;
342 			break;
343 		case 2:
344 			val |= (3 << RNG_v2_CTL_VCO_SHIFT) | RNG_CTL_ES3;
345 			break;
346 		case 3:
347 			val |= RNG_CTL_ES1 | RNG_CTL_ES2 | RNG_CTL_ES3;
348 			break;
349 		default:
350 			break;
351 		}
352 	}
353 
354 	return val;
355 }
356 
357 static void n2rng_control_swstate_init(struct n2rng *np)
358 {
359 	int i;
360 
361 	np->flags |= N2RNG_FLAG_CONTROL;
362 
363 	np->health_check_sec = N2RNG_HEALTH_CHECK_SEC_DEFAULT;
364 	np->accum_cycles = N2RNG_ACCUM_CYCLES_DEFAULT;
365 	np->wd_timeo = N2RNG_WD_TIMEO_DEFAULT;
366 
367 	for (i = 0; i < np->num_units; i++) {
368 		struct n2rng_unit *up = &np->units[i];
369 
370 		up->control[0] = n2rng_control_default(np, 0);
371 		up->control[1] = n2rng_control_default(np, 1);
372 		up->control[2] = n2rng_control_default(np, 2);
373 		up->control[3] = n2rng_control_default(np, 3);
374 	}
375 
376 	np->hv_state = HV_RNG_STATE_UNCONFIGURED;
377 }
378 
379 static int n2rng_grab_diag_control(struct n2rng *np)
380 {
381 	int i, busy_count, err = -ENODEV;
382 
383 	busy_count = 0;
384 	for (i = 0; i < 100; i++) {
385 		err = n2rng_try_read_ctl(np);
386 		if (err != -EAGAIN)
387 			break;
388 
389 		if (++busy_count > 100) {
390 			dev_err(&np->op->dev,
391 				"Grab diag control timeout.\n");
392 			return -ENODEV;
393 		}
394 
395 		udelay(1);
396 	}
397 
398 	return err;
399 }
400 
401 static int n2rng_init_control(struct n2rng *np)
402 {
403 	int err = n2rng_grab_diag_control(np);
404 
405 	/* Not in the control domain, that's OK we are only a consumer
406 	 * of the RNG data, we don't setup and program it.
407 	 */
408 	if (err == -EPERM)
409 		return 0;
410 	if (err)
411 		return err;
412 
413 	n2rng_control_swstate_init(np);
414 
415 	return 0;
416 }
417 
418 static int n2rng_data_read(struct hwrng *rng, u32 *data)
419 {
420 	struct n2rng *np = (struct n2rng *) rng->priv;
421 	unsigned long ra = __pa(&np->test_data);
422 	int len;
423 
424 	if (!(np->flags & N2RNG_FLAG_READY)) {
425 		len = 0;
426 	} else if (np->flags & N2RNG_FLAG_BUFFER_VALID) {
427 		np->flags &= ~N2RNG_FLAG_BUFFER_VALID;
428 		*data = np->buffer;
429 		len = 4;
430 	} else {
431 		int err = n2rng_generic_read_data(ra);
432 		if (!err) {
433 			np->flags |= N2RNG_FLAG_BUFFER_VALID;
434 			np->buffer = np->test_data >> 32;
435 			*data = np->test_data & 0xffffffff;
436 			len = 4;
437 		} else {
438 			dev_err(&np->op->dev, "RNG error, retesting\n");
439 			np->flags &= ~N2RNG_FLAG_READY;
440 			if (!(np->flags & N2RNG_FLAG_SHUTDOWN))
441 				schedule_delayed_work(&np->work, 0);
442 			len = 0;
443 		}
444 	}
445 
446 	return len;
447 }
448 
449 /* On a guest node, just make sure we can read random data properly.
450  * If a control node reboots or reloads it's n2rng driver, this won't
451  * work during that time.  So we have to keep probing until the device
452  * becomes usable.
453  */
454 static int n2rng_guest_check(struct n2rng *np)
455 {
456 	unsigned long ra = __pa(&np->test_data);
457 
458 	return n2rng_generic_read_data(ra);
459 }
460 
461 static int n2rng_entropy_diag_read(struct n2rng *np, unsigned long unit,
462 				   u64 *pre_control, u64 pre_state,
463 				   u64 *buffer, unsigned long buf_len,
464 				   u64 *post_control, u64 post_state)
465 {
466 	unsigned long post_ctl_ra = __pa(post_control);
467 	unsigned long pre_ctl_ra = __pa(pre_control);
468 	unsigned long buffer_ra = __pa(buffer);
469 	int err;
470 
471 	err = n2rng_generic_write_control(np, pre_ctl_ra, unit, pre_state);
472 	if (err)
473 		return err;
474 
475 	err = n2rng_generic_read_diag_data(np, unit,
476 					   buffer_ra, buf_len);
477 
478 	(void) n2rng_generic_write_control(np, post_ctl_ra, unit,
479 					   post_state);
480 
481 	return err;
482 }
483 
484 static u64 advance_polynomial(u64 poly, u64 val, int count)
485 {
486 	int i;
487 
488 	for (i = 0; i < count; i++) {
489 		int highbit_set = ((s64)val < 0);
490 
491 		val <<= 1;
492 		if (highbit_set)
493 			val ^= poly;
494 	}
495 
496 	return val;
497 }
498 
499 static int n2rng_test_buffer_find(struct n2rng *np, u64 val)
500 {
501 	int i, count = 0;
502 
503 	/* Purposefully skip over the first word.  */
504 	for (i = 1; i < SELFTEST_BUFFER_WORDS; i++) {
505 		if (np->test_buffer[i] == val)
506 			count++;
507 	}
508 	return count;
509 }
510 
511 static void n2rng_dump_test_buffer(struct n2rng *np)
512 {
513 	int i;
514 
515 	for (i = 0; i < SELFTEST_BUFFER_WORDS; i++)
516 		dev_err(&np->op->dev, "Test buffer slot %d [0x%016llx]\n",
517 			i, np->test_buffer[i]);
518 }
519 
520 static int n2rng_check_selftest_buffer(struct n2rng *np, unsigned long unit)
521 {
522 	u64 val;
523 	int err, matches, limit;
524 
525 	switch (np->data->id) {
526 	case N2_n2_rng:
527 	case N2_vf_rng:
528 	case N2_kt_rng:
529 	case N2_m4_rng:  /* yes, m4 uses the old value */
530 		val = RNG_v1_SELFTEST_VAL;
531 		break;
532 	default:
533 		val = RNG_v2_SELFTEST_VAL;
534 		break;
535 	}
536 
537 	matches = 0;
538 	for (limit = 0; limit < SELFTEST_LOOPS_MAX; limit++) {
539 		matches += n2rng_test_buffer_find(np, val);
540 		if (matches >= SELFTEST_MATCH_GOAL)
541 			break;
542 		val = advance_polynomial(SELFTEST_POLY, val, 1);
543 	}
544 
545 	err = 0;
546 	if (limit >= SELFTEST_LOOPS_MAX) {
547 		err = -ENODEV;
548 		dev_err(&np->op->dev, "Selftest failed on unit %lu\n", unit);
549 		n2rng_dump_test_buffer(np);
550 	} else
551 		dev_info(&np->op->dev, "Selftest passed on unit %lu\n", unit);
552 
553 	return err;
554 }
555 
556 static int n2rng_control_selftest(struct n2rng *np, unsigned long unit)
557 {
558 	int err;
559 	u64 base, base3;
560 
561 	switch (np->data->id) {
562 	case N2_n2_rng:
563 	case N2_vf_rng:
564 	case N2_kt_rng:
565 		base = RNG_v1_CTL_ASEL_NOOUT << RNG_v1_CTL_ASEL_SHIFT;
566 		base3 = base | RNG_CTL_LFSR |
567 			((RNG_v1_SELFTEST_TICKS - 2) << RNG_v1_CTL_WAIT_SHIFT);
568 		break;
569 	case N2_m4_rng:
570 		base = RNG_v2_CTL_ASEL_NOOUT << RNG_v2_CTL_ASEL_SHIFT;
571 		base3 = base | RNG_CTL_LFSR |
572 			((RNG_v1_SELFTEST_TICKS - 2) << RNG_v2_CTL_WAIT_SHIFT);
573 		break;
574 	default:
575 		base = RNG_v2_CTL_ASEL_NOOUT << RNG_v2_CTL_ASEL_SHIFT;
576 		base3 = base | RNG_CTL_LFSR |
577 			(RNG_v2_SELFTEST_TICKS << RNG_v2_CTL_WAIT_SHIFT);
578 		break;
579 	}
580 
581 	np->test_control[0] = base;
582 	np->test_control[1] = base;
583 	np->test_control[2] = base;
584 	np->test_control[3] = base3;
585 
586 	err = n2rng_entropy_diag_read(np, unit, np->test_control,
587 				      HV_RNG_STATE_HEALTHCHECK,
588 				      np->test_buffer,
589 				      sizeof(np->test_buffer),
590 				      &np->units[unit].control[0],
591 				      np->hv_state);
592 	if (err)
593 		return err;
594 
595 	return n2rng_check_selftest_buffer(np, unit);
596 }
597 
598 static int n2rng_control_check(struct n2rng *np)
599 {
600 	int i;
601 
602 	for (i = 0; i < np->num_units; i++) {
603 		int err = n2rng_control_selftest(np, i);
604 		if (err)
605 			return err;
606 	}
607 	return 0;
608 }
609 
610 /* The sanity checks passed, install the final configuration into the
611  * chip, it's ready to use.
612  */
613 static int n2rng_control_configure_units(struct n2rng *np)
614 {
615 	int unit, err;
616 
617 	err = 0;
618 	for (unit = 0; unit < np->num_units; unit++) {
619 		struct n2rng_unit *up = &np->units[unit];
620 		unsigned long ctl_ra = __pa(&up->control[0]);
621 		int esrc;
622 		u64 base, shift;
623 
624 		if (np->data->chip_version == 1) {
625 			base = ((np->accum_cycles << RNG_v1_CTL_WAIT_SHIFT) |
626 			      (RNG_v1_CTL_ASEL_NOOUT << RNG_v1_CTL_ASEL_SHIFT) |
627 			      RNG_CTL_LFSR);
628 			shift = RNG_v1_CTL_VCO_SHIFT;
629 		} else {
630 			base = ((np->accum_cycles << RNG_v2_CTL_WAIT_SHIFT) |
631 			      (RNG_v2_CTL_ASEL_NOOUT << RNG_v2_CTL_ASEL_SHIFT) |
632 			      RNG_CTL_LFSR);
633 			shift = RNG_v2_CTL_VCO_SHIFT;
634 		}
635 
636 		/* XXX This isn't the best.  We should fetch a bunch
637 		 * XXX of words using each entropy source combined XXX
638 		 * with each VCO setting, and see which combinations
639 		 * XXX give the best random data.
640 		 */
641 		for (esrc = 0; esrc < 3; esrc++)
642 			up->control[esrc] = base |
643 				(esrc << shift) |
644 				(RNG_CTL_ES1 << esrc);
645 
646 		up->control[3] = base |
647 			(RNG_CTL_ES1 | RNG_CTL_ES2 | RNG_CTL_ES3);
648 
649 		err = n2rng_generic_write_control(np, ctl_ra, unit,
650 						  HV_RNG_STATE_CONFIGURED);
651 		if (err)
652 			break;
653 	}
654 
655 	return err;
656 }
657 
658 static void n2rng_work(struct work_struct *work)
659 {
660 	struct n2rng *np = container_of(work, struct n2rng, work.work);
661 	int err = 0;
662 	static int retries = 4;
663 
664 	if (!(np->flags & N2RNG_FLAG_CONTROL)) {
665 		err = n2rng_guest_check(np);
666 	} else {
667 		preempt_disable();
668 		err = n2rng_control_check(np);
669 		preempt_enable();
670 
671 		if (!err)
672 			err = n2rng_control_configure_units(np);
673 	}
674 
675 	if (!err) {
676 		np->flags |= N2RNG_FLAG_READY;
677 		dev_info(&np->op->dev, "RNG ready\n");
678 	}
679 
680 	if (--retries == 0)
681 		dev_err(&np->op->dev, "Self-test retries failed, RNG not ready\n");
682 	else if (err && !(np->flags & N2RNG_FLAG_SHUTDOWN))
683 		schedule_delayed_work(&np->work, HZ * 2);
684 }
685 
686 static void n2rng_driver_version(void)
687 {
688 	static int n2rng_version_printed;
689 
690 	if (n2rng_version_printed++ == 0)
691 		pr_info("%s", version);
692 }
693 
694 static const struct of_device_id n2rng_match[];
695 static int n2rng_probe(struct platform_device *op)
696 {
697 	const struct of_device_id *match;
698 	int err = -ENOMEM;
699 	struct n2rng *np;
700 
701 	match = of_match_device(n2rng_match, &op->dev);
702 	if (!match)
703 		return -EINVAL;
704 
705 	n2rng_driver_version();
706 	np = devm_kzalloc(&op->dev, sizeof(*np), GFP_KERNEL);
707 	if (!np)
708 		goto out;
709 	np->op = op;
710 	np->data = (struct n2rng_template *)match->data;
711 
712 	INIT_DELAYED_WORK(&np->work, n2rng_work);
713 
714 	if (np->data->multi_capable)
715 		np->flags |= N2RNG_FLAG_MULTI;
716 
717 	err = -ENODEV;
718 	np->hvapi_major = 2;
719 	if (sun4v_hvapi_register(HV_GRP_RNG,
720 				 np->hvapi_major,
721 				 &np->hvapi_minor)) {
722 		np->hvapi_major = 1;
723 		if (sun4v_hvapi_register(HV_GRP_RNG,
724 					 np->hvapi_major,
725 					 &np->hvapi_minor)) {
726 			dev_err(&op->dev, "Cannot register suitable "
727 				"HVAPI version.\n");
728 			goto out;
729 		}
730 	}
731 
732 	if (np->flags & N2RNG_FLAG_MULTI) {
733 		if (np->hvapi_major < 2) {
734 			dev_err(&op->dev, "multi-unit-capable RNG requires "
735 				"HVAPI major version 2 or later, got %lu\n",
736 				np->hvapi_major);
737 			goto out_hvapi_unregister;
738 		}
739 		np->num_units = of_getintprop_default(op->dev.of_node,
740 						      "rng-#units", 0);
741 		if (!np->num_units) {
742 			dev_err(&op->dev, "VF RNG lacks rng-#units property\n");
743 			goto out_hvapi_unregister;
744 		}
745 	} else {
746 		np->num_units = 1;
747 	}
748 
749 	dev_info(&op->dev, "Registered RNG HVAPI major %lu minor %lu\n",
750 		 np->hvapi_major, np->hvapi_minor);
751 	np->units = devm_kcalloc(&op->dev, np->num_units, sizeof(*np->units),
752 				 GFP_KERNEL);
753 	err = -ENOMEM;
754 	if (!np->units)
755 		goto out_hvapi_unregister;
756 
757 	err = n2rng_init_control(np);
758 	if (err)
759 		goto out_hvapi_unregister;
760 
761 	dev_info(&op->dev, "Found %s RNG, units: %d\n",
762 		 ((np->flags & N2RNG_FLAG_MULTI) ?
763 		  "multi-unit-capable" : "single-unit"),
764 		 np->num_units);
765 
766 	np->hwrng.name = DRV_MODULE_NAME;
767 	np->hwrng.data_read = n2rng_data_read;
768 	np->hwrng.priv = (unsigned long) np;
769 
770 	err = hwrng_register(&np->hwrng);
771 	if (err)
772 		goto out_hvapi_unregister;
773 
774 	platform_set_drvdata(op, np);
775 
776 	schedule_delayed_work(&np->work, 0);
777 
778 	return 0;
779 
780 out_hvapi_unregister:
781 	sun4v_hvapi_unregister(HV_GRP_RNG);
782 
783 out:
784 	return err;
785 }
786 
787 static int n2rng_remove(struct platform_device *op)
788 {
789 	struct n2rng *np = platform_get_drvdata(op);
790 
791 	np->flags |= N2RNG_FLAG_SHUTDOWN;
792 
793 	cancel_delayed_work_sync(&np->work);
794 
795 	hwrng_unregister(&np->hwrng);
796 
797 	sun4v_hvapi_unregister(HV_GRP_RNG);
798 
799 	return 0;
800 }
801 
802 static struct n2rng_template n2_template = {
803 	.id = N2_n2_rng,
804 	.multi_capable = 0,
805 	.chip_version = 1,
806 };
807 
808 static struct n2rng_template vf_template = {
809 	.id = N2_vf_rng,
810 	.multi_capable = 1,
811 	.chip_version = 1,
812 };
813 
814 static struct n2rng_template kt_template = {
815 	.id = N2_kt_rng,
816 	.multi_capable = 1,
817 	.chip_version = 1,
818 };
819 
820 static struct n2rng_template m4_template = {
821 	.id = N2_m4_rng,
822 	.multi_capable = 1,
823 	.chip_version = 2,
824 };
825 
826 static struct n2rng_template m7_template = {
827 	.id = N2_m7_rng,
828 	.multi_capable = 1,
829 	.chip_version = 2,
830 };
831 
832 static const struct of_device_id n2rng_match[] = {
833 	{
834 		.name		= "random-number-generator",
835 		.compatible	= "SUNW,n2-rng",
836 		.data		= &n2_template,
837 	},
838 	{
839 		.name		= "random-number-generator",
840 		.compatible	= "SUNW,vf-rng",
841 		.data		= &vf_template,
842 	},
843 	{
844 		.name		= "random-number-generator",
845 		.compatible	= "SUNW,kt-rng",
846 		.data		= &kt_template,
847 	},
848 	{
849 		.name		= "random-number-generator",
850 		.compatible	= "ORCL,m4-rng",
851 		.data		= &m4_template,
852 	},
853 	{
854 		.name		= "random-number-generator",
855 		.compatible	= "ORCL,m7-rng",
856 		.data		= &m7_template,
857 	},
858 	{},
859 };
860 MODULE_DEVICE_TABLE(of, n2rng_match);
861 
862 static struct platform_driver n2rng_driver = {
863 	.driver = {
864 		.name = "n2rng",
865 		.of_match_table = n2rng_match,
866 	},
867 	.probe		= n2rng_probe,
868 	.remove		= n2rng_remove,
869 };
870 
871 module_platform_driver(n2rng_driver);
872