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. */ 10 void prandom_warmup(struct rnd_state *state); 11 12 static 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 22 static 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 130 static 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 139 static 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 151 static 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 167 static struct kunit_case prandom_test_cases[] = { 168 KUNIT_CASE(test_prandom_seed_boundary), 169 KUNIT_CASE(test_prandom_taus113), 170 {} 171 }; 172 173 static struct kunit_suite prandom_test_suite = { 174 .name = "prandom", 175 .test_cases = prandom_test_cases, 176 }; 177 178 kunit_test_suite(prandom_test_suite); 179 180 MODULE_DESCRIPTION("KUnit test for prandom"); 181 MODULE_LICENSE("GPL"); 182 MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING"); 183