projects/linux·blob master · refs

182 lines · 5.5 KiB

1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Test cases for random32 functions.
4 */
5
6#include <linux/prandom.h>
7#include <kunit/test.h>
8
9/* prandom_warmup() is static in lib/random32.c; exposed for testing only. */
10void prandom_warmup(struct rnd_state *state);
11
12static const struct prandom_test1 {
13 u32 seed;
14 u32 result;
15} test1[] = {
16 { 1U, 3484351685U },
17 { 2U, 2623130059U },
18 { 3U, 3125133893U },
19 { 4U, 984847254U },
20};
21
22static const struct prandom_test2 {
23 u32 seed;
24 u32 iteration;
25 u32 result;
26} test2[] = {
27 /* Test cases against taus113 from GSL library. */
28 { 931557656U, 959U, 2975593782U },
29 { 1339693295U, 876U, 3887776532U },
30 { 1545556285U, 961U, 1615538833U },
31 { 601730776U, 723U, 1776162651U },
32 { 1027516047U, 687U, 511983079U },
33 { 416526298U, 700U, 916156552U },
34 { 1395522032U, 652U, 2222063676U },
35 { 366221443U, 617U, 2992857763U },
36 { 1539836965U, 714U, 3783265725U },
37 { 556206671U, 994U, 799626459U },
38 { 684907218U, 799U, 367789491U },
39 { 2121230701U, 931U, 2115467001U },
40 { 1668516451U, 644U, 3620590685U },
41 { 768046066U, 883U, 2034077390U },
42 { 1989159136U, 833U, 1195767305U },
43 { 536585145U, 996U, 3577259204U },
44 { 1008129373U, 642U, 1478080776U },
45 { 1740775604U, 939U, 1264980372U },
46 { 1967883163U, 508U, 10734624U },
47 { 1923019697U, 730U, 3821419629U },
48 { 442079932U, 560U, 3440032343U },
49 { 1961302714U, 845U, 841962572U },
50 { 2030205964U, 962U, 1325144227U },
51 { 1160407529U, 507U, 240940858U },
52 { 635482502U, 779U, 4200489746U },
53 { 1252788931U, 699U, 867195434U },
54 { 1961817131U, 719U, 668237657U },
55 { 1071468216U, 983U, 917876630U },
56 { 1281848367U, 932U, 1003100039U },
57 { 582537119U, 780U, 1127273778U },
58 { 1973672777U, 853U, 1071368872U },
59 { 1896756996U, 762U, 1127851055U },
60 { 847917054U, 500U, 1717499075U },
61 { 1240520510U, 951U, 2849576657U },
62 { 1685071682U, 567U, 1961810396U },
63 { 1516232129U, 557U, 3173877U },
64 { 1208118903U, 612U, 1613145022U },
65 { 1817269927U, 693U, 4279122573U },
66 { 1510091701U, 717U, 638191229U },
67 { 365916850U, 807U, 600424314U },
68 { 399324359U, 702U, 1803598116U },
69 { 1318480274U, 779U, 2074237022U },
70 { 697758115U, 840U, 1483639402U },
71 { 1696507773U, 840U, 577415447U },
72 { 2081979121U, 981U, 3041486449U },
73 { 955646687U, 742U, 3846494357U },
74 { 1250683506U, 749U, 836419859U },
75 { 595003102U, 534U, 366794109U },
76 { 47485338U, 558U, 3521120834U },
77 { 619433479U, 610U, 3991783875U },
78 { 704096520U, 518U, 4139493852U },
79 { 1712224984U, 606U, 2393312003U },
80 { 1318233152U, 922U, 3880361134U },
81 { 855572992U, 761U, 1472974787U },
82 { 64721421U, 703U, 683860550U },
83 { 678931758U, 840U, 380616043U },
84 { 692711973U, 778U, 1382361947U },
85 { 677703619U, 530U, 2826914161U },
86 { 92393223U, 586U, 1522128471U },
87 { 1222592920U, 743U, 3466726667U },
88 { 358288986U, 695U, 1091956998U },
89 { 1935056945U, 958U, 514864477U },
90 { 735675993U, 990U, 1294239989U },
91 { 1560089402U, 897U, 2238551287U },
92 { 70616361U, 829U, 22483098U },
93 { 368234700U, 731U, 2913875084U },
94 { 20221190U, 879U, 1564152970U },
95 { 539444654U, 682U, 1835141259U },
96 { 1314987297U, 840U, 1801114136U },
97 { 2019295544U, 645U, 3286438930U },
98 { 469023838U, 716U, 1637918202U },
99 { 1843754496U, 653U, 2562092152U },
100 { 400672036U, 809U, 4264212785U },
101 { 404722249U, 965U, 2704116999U },
102 { 600702209U, 758U, 584979986U },
103 { 519953954U, 667U, 2574436237U },
104 { 1658071126U, 694U, 2214569490U },
105 { 420480037U, 749U, 3430010866U },
106 { 690103647U, 969U, 3700758083U },
107 { 1029424799U, 937U, 3787746841U },
108 { 2012608669U, 506U, 3362628973U },
109 { 1535432887U, 998U, 42610943U },
110 { 1330635533U, 857U, 3040806504U },
111 { 1223800550U, 539U, 3954229517U },
112 { 1322411537U, 680U, 3223250324U },
113 { 1877847898U, 945U, 2915147143U },
114 { 1646356099U, 874U, 965988280U },
115 { 805687536U, 744U, 4032277920U },
116 { 1948093210U, 633U, 1346597684U },
117 { 392609744U, 783U, 1636083295U },
118 { 690241304U, 770U, 1201031298U },
119 { 1360302965U, 696U, 1665394461U },
120 { 1220090946U, 780U, 1316922812U },
121 { 447092251U, 500U, 3438743375U },
122 { 1613868791U, 592U, 828546883U },
123 { 523430951U, 548U, 2552392304U },
124 { 726692899U, 810U, 1656872867U },
125 { 1364340021U, 836U, 3710513486U },
126 { 1986257729U, 931U, 935013962U },
127 { 407983964U, 921U, 728767059U },
128};
129
130static void prandom_state_test_seed(struct rnd_state *state, u32 seed)
131{
132#define LCG(x) ((x) * 69069U) /* super-duper LCG */
133 state->s1 = __seed(LCG(seed), 2U);
134 state->s2 = __seed(LCG(state->s1), 8U);
135 state->s3 = __seed(LCG(state->s2), 16U);
136 state->s4 = __seed(LCG(state->s3), 128U);
137}
138
139static void test_prandom_seed_boundary(struct kunit *test)
140{
141 int i;
142 struct rnd_state state;
143
144 for (i = 0; i < ARRAY_SIZE(test1); i++) {
145 prandom_state_test_seed(&state, test1[i].seed);
146 prandom_warmup(&state);
147 KUNIT_EXPECT_EQ(test, test1[i].result, prandom_u32_state(&state));
148 }
149}
150
151static void test_prandom_taus113(struct kunit *test)
152{
153 int i, j;
154 struct rnd_state state;
155
156 for (i = 0; i < ARRAY_SIZE(test2); i++) {
157 prandom_state_test_seed(&state, test2[i].seed);
158 prandom_warmup(&state);
159
160 for (j = 0; j < test2[i].iteration - 1; j++)
161 prandom_u32_state(&state);
162
163 KUNIT_EXPECT_EQ(test, test2[i].result, prandom_u32_state(&state));
164 }
165}
166
167static struct kunit_case prandom_test_cases[] = {
168 KUNIT_CASE(test_prandom_seed_boundary),
169 KUNIT_CASE(test_prandom_taus113),
170 {}
171};
172
173static struct kunit_suite prandom_test_suite = {
174 .name = "prandom",
175 .test_cases = prandom_test_cases,
176};
177
178kunit_test_suite(prandom_test_suite);
179
180MODULE_DESCRIPTION("KUnit test for prandom");
181MODULE_LICENSE("GPL");
182MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");