1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright (c) 2019, Linaro Limited
4 */
5
6 #include <common.h>
7 #include <dm.h>
8 #include <log.h>
9 #include <rng.h>
10 #include <dm/test.h>
11 #include <test/ut.h>
12
13 /* Basic test of the rng uclass */
dm_test_rng_read(struct unit_test_state * uts)14 static int dm_test_rng_read(struct unit_test_state *uts)
15 {
16 unsigned long rand1 = 0, rand2 = 0;
17 struct udevice *dev;
18
19 ut_assertok(uclass_get_device(UCLASS_RNG, 0, &dev));
20 ut_assertnonnull(dev);
21 ut_assertok(dm_rng_read(dev, &rand1, sizeof(rand1)));
22 ut_assertok(dm_rng_read(dev, &rand2, sizeof(rand2)));
23 ut_assert(rand1 != rand2);
24
25 return 0;
26 }
27 DM_TEST(dm_test_rng_read, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
28
29 /* Test the rng command */
dm_test_rng_cmd(struct unit_test_state * uts)30 static int dm_test_rng_cmd(struct unit_test_state *uts)
31 {
32 struct udevice *dev;
33
34 ut_assertok(uclass_get_device(UCLASS_RNG, 0, &dev));
35 ut_assertnonnull(dev);
36
37 ut_assertok(console_record_reset_enable());
38
39 run_command("rng", 0);
40 ut_assert_nextlinen("00000000:");
41 ut_assert_nextlinen("00000010:");
42 ut_assert_nextlinen("00000020:");
43 ut_assert_nextlinen("00000030:");
44 ut_assert_console_end();
45
46 run_command("rng 0 10", 0);
47 ut_assert_nextlinen("00000000:");
48 ut_assert_console_end();
49
50 run_command("rng 20", 0);
51 ut_assert_nextlinen("No RNG device");
52 ut_assert_console_end();
53
54 return 0;
55 }
56 DM_TEST(dm_test_rng_cmd, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT | UT_TESTF_CONSOLE_REC);
57