xref: /linux/lib/tests/shdi3_kunit.c (revision 6bba2ae43e8f379d3bed4c3eb258ea6d81baae80)
1 // SPDX-License-Identifier: GPL-2.0-or-later OR Apache-2.0
2 /*
3  * Test cases for __ashldi3(), __ashrdi3(), and __lshrdi3().
4  */
5 
6 #include <linux/array_size.h>
7 #include <linux/module.h>
8 #include <linux/libgcc.h>
9 #include <kunit/test.h>
10 
11 struct shdi3_test_entry {
12 	long long input;
13 	int shift;
14 	long long result;
15 };
16 
17 static const struct shdi3_test_entry ashldi3_testdata[] = {
18 	/* https://github.com/llvm/llvm-project/compiler-rt/test/builtins/Unit/ashldi3_test.c */
19 	{ 0x0123456789ABCDEFLL, 0, 0x123456789ABCDEFLL },
20 	{ 0x0123456789ABCDEFLL, 1, 0x2468ACF13579BDELL },
21 	{ 0x0123456789ABCDEFLL, 2, 0x48D159E26AF37BCLL },
22 	{ 0x0123456789ABCDEFLL, 3, 0x91A2B3C4D5E6F78LL },
23 	{ 0x0123456789ABCDEFLL, 4, 0x123456789ABCDEF0LL },
24 	{ 0x0123456789ABCDEFLL, 28, 0x789ABCDEF0000000LL },
25 	{ 0x0123456789ABCDEFLL, 29, 0xF13579BDE0000000LL },
26 	{ 0x0123456789ABCDEFLL, 30, 0xE26AF37BC0000000LL },
27 	{ 0x0123456789ABCDEFLL, 31, 0xC4D5E6F780000000LL },
28 	{ 0x0123456789ABCDEFLL, 32, 0x89ABCDEF00000000LL },
29 	{ 0x0123456789ABCDEFLL, 33, 0x13579BDE00000000LL },
30 	{ 0x0123456789ABCDEFLL, 34, 0x26AF37BC00000000LL },
31 	{ 0x0123456789ABCDEFLL, 35, 0x4D5E6F7800000000LL },
32 	{ 0x0123456789ABCDEFLL, 36, 0x9ABCDEF000000000LL },
33 	{ 0x0123456789ABCDEFLL, 60, 0xF000000000000000LL },
34 	{ 0x0123456789ABCDEFLL, 61, 0xE000000000000000LL },
35 	{ 0x0123456789ABCDEFLL, 62, 0xC000000000000000LL },
36 	{ 0x0123456789ABCDEFLL, 63, 0x8000000000000000LL },
37 };
38 
39 static void shdi3_test_ashldi3(struct kunit *test)
40 {
41 	const struct shdi3_test_entry *e;
42 	long long ret;
43 
44 	for (e = ashldi3_testdata;
45 	     e < ashldi3_testdata + ARRAY_SIZE(ashldi3_testdata); e++) {
46 		ret = __ashldi3(e->input, e->shift);
47 		KUNIT_EXPECT_EQ_MSG(test, ret, e->result,
48 				    "    when evaluating __ashldi3(%lld, %d)",
49 				    e->input, e->shift);
50 	}
51 }
52 
53 static const struct shdi3_test_entry ashrdi3_testdata[] = {
54 	/* https://github.com/llvm/llvm-project/compiler-rt/test/builtins/Unit/ashrdi3_test.c */
55 	{ 0x0123456789ABCDEFLL, 0, 0x123456789ABCDEFLL },
56 	{ 0x0123456789ABCDEFLL, 1, 0x91A2B3C4D5E6F7LL },
57 	{ 0x0123456789ABCDEFLL, 2, 0x48D159E26AF37BLL },
58 	{ 0x0123456789ABCDEFLL, 3, 0x2468ACF13579BDLL },
59 	{ 0x0123456789ABCDEFLL, 4, 0x123456789ABCDELL },
60 	{ 0x0123456789ABCDEFLL, 28, 0x12345678LL },
61 	{ 0x0123456789ABCDEFLL, 29, 0x91A2B3CLL },
62 	{ 0x0123456789ABCDEFLL, 30, 0x48D159ELL },
63 	{ 0x0123456789ABCDEFLL, 31, 0x2468ACFLL },
64 	{ 0x0123456789ABCDEFLL, 32, 0x1234567LL },
65 	{ 0x0123456789ABCDEFLL, 33, 0x91A2B3LL },
66 	{ 0x0123456789ABCDEFLL, 34, 0x48D159LL },
67 	{ 0x0123456789ABCDEFLL, 35, 0x2468ACLL },
68 	{ 0x0123456789ABCDEFLL, 36, 0x123456LL },
69 	{ 0x0123456789ABCDEFLL, 60, 0 },
70 	{ 0x0123456789ABCDEFLL, 61, 0 },
71 	{ 0x0123456789ABCDEFLL, 62, 0 },
72 	{ 0x0123456789ABCDEFLL, 63, 0 },
73 	{ 0xFEDCBA9876543210LL, 0, 0xFEDCBA9876543210LL },
74 	{ 0xFEDCBA9876543210LL, 1, 0xFF6E5D4C3B2A1908LL },
75 	{ 0xFEDCBA9876543210LL, 2, 0xFFB72EA61D950C84LL },
76 	{ 0xFEDCBA9876543210LL, 3, 0xFFDB97530ECA8642LL },
77 	{ 0xFEDCBA9876543210LL, 4, 0xFFEDCBA987654321LL },
78 	{ 0xFEDCBA9876543210LL, 28, 0xFFFFFFFFEDCBA987LL },
79 	{ 0xFEDCBA9876543210LL, 29, 0xFFFFFFFFF6E5D4C3LL },
80 	{ 0xFEDCBA9876543210LL, 30, 0xFFFFFFFFFB72EA61LL },
81 	{ 0xFEDCBA9876543210LL, 31, 0xFFFFFFFFFDB97530LL },
82 	{ 0xFEDCBA9876543210LL, 32, 0xFFFFFFFFFEDCBA98LL },
83 	{ 0xFEDCBA9876543210LL, 33, 0xFFFFFFFFFF6E5D4CLL },
84 	{ 0xFEDCBA9876543210LL, 34, 0xFFFFFFFFFFB72EA6LL },
85 	{ 0xFEDCBA9876543210LL, 35, 0xFFFFFFFFFFDB9753LL },
86 	{ 0xFEDCBA9876543210LL, 36, 0xFFFFFFFFFFEDCBA9LL },
87 	{ 0xAEDCBA9876543210LL, 60, 0xFFFFFFFFFFFFFFFALL },
88 	{ 0xAEDCBA9876543210LL, 61, 0xFFFFFFFFFFFFFFFDLL },
89 	{ 0xAEDCBA9876543210LL, 62, 0xFFFFFFFFFFFFFFFELL },
90 	{ 0xAEDCBA9876543210LL, 63, 0xFFFFFFFFFFFFFFFFLL },
91 };
92 
93 static void shdi3_test_ashrdi3(struct kunit *test)
94 {
95 	const struct shdi3_test_entry *e;
96 	long long ret;
97 
98 	for (e = ashrdi3_testdata;
99 	     e < ashrdi3_testdata + ARRAY_SIZE(ashrdi3_testdata); e++) {
100 		ret = __ashrdi3(e->input, e->shift);
101 		KUNIT_EXPECT_EQ_MSG(test, ret, e->result,
102 				    "    when evaluating __ashrdi3(%lld, %d)",
103 				    e->input, e->shift);
104 	}
105 }
106 
107 static const struct shdi3_test_entry lshrdi3_testdata[] = {
108 	/* https://github.com/llvm/llvm-project/compiler-rt/test/builtins/Unit/lshrdi3_test.c */
109 	{ 0x0123456789ABCDEFLL, 0, 0x123456789ABCDEFLL },
110 	{ 0x0123456789ABCDEFLL, 1, 0x91A2B3C4D5E6F7LL },
111 	{ 0x0123456789ABCDEFLL, 2, 0x48D159E26AF37BLL },
112 	{ 0x0123456789ABCDEFLL, 3, 0x2468ACF13579BDLL },
113 	{ 0x0123456789ABCDEFLL, 4, 0x123456789ABCDELL },
114 	{ 0x0123456789ABCDEFLL, 28, 0x12345678LL },
115 	{ 0x0123456789ABCDEFLL, 29, 0x91A2B3CLL },
116 	{ 0x0123456789ABCDEFLL, 30, 0x48D159ELL },
117 	{ 0x0123456789ABCDEFLL, 31, 0x2468ACFLL },
118 	{ 0x0123456789ABCDEFLL, 32, 0x1234567LL },
119 	{ 0x0123456789ABCDEFLL, 33, 0x91A2B3LL },
120 	{ 0x0123456789ABCDEFLL, 34, 0x48D159LL },
121 	{ 0x0123456789ABCDEFLL, 35, 0x2468ACLL },
122 	{ 0x0123456789ABCDEFLL, 36, 0x123456LL },
123 	{ 0x0123456789ABCDEFLL, 60, 0 },
124 	{ 0x0123456789ABCDEFLL, 61, 0 },
125 	{ 0x0123456789ABCDEFLL, 62, 0 },
126 	{ 0x0123456789ABCDEFLL, 63, 0 },
127 	{ 0xFEDCBA9876543210LL, 0, 0xFEDCBA9876543210LL },
128 	{ 0xFEDCBA9876543210LL, 1, 0x7F6E5D4C3B2A1908LL },
129 	{ 0xFEDCBA9876543210LL, 2, 0x3FB72EA61D950C84LL },
130 	{ 0xFEDCBA9876543210LL, 3, 0x1FDB97530ECA8642LL },
131 	{ 0xFEDCBA9876543210LL, 4, 0xFEDCBA987654321LL },
132 	{ 0xFEDCBA9876543210LL, 28, 0xFEDCBA987LL },
133 	{ 0xFEDCBA9876543210LL, 29, 0x7F6E5D4C3LL },
134 	{ 0xFEDCBA9876543210LL, 30, 0x3FB72EA61LL },
135 	{ 0xFEDCBA9876543210LL, 31, 0x1FDB97530LL },
136 	{ 0xFEDCBA9876543210LL, 32, 0xFEDCBA98LL },
137 	{ 0xFEDCBA9876543210LL, 33, 0x7F6E5D4CLL },
138 	{ 0xFEDCBA9876543210LL, 34, 0x3FB72EA6LL },
139 	{ 0xFEDCBA9876543210LL, 35, 0x1FDB9753LL },
140 	{ 0xFEDCBA9876543210LL, 36, 0xFEDCBA9LL },
141 	{ 0xAEDCBA9876543210LL, 60, 0xALL },
142 	{ 0xAEDCBA9876543210LL, 61, 0x5LL },
143 	{ 0xAEDCBA9876543210LL, 62, 0x2LL },
144 	{ 0xAEDCBA9876543210LL, 63, 0x1LL },
145 };
146 
147 static void shdi3_test_lshrdi3(struct kunit *test)
148 {
149 	const struct shdi3_test_entry *e;
150 	long long ret;
151 
152 	for (e = lshrdi3_testdata;
153 	     e < lshrdi3_testdata + ARRAY_SIZE(lshrdi3_testdata); e++) {
154 		ret = __lshrdi3(e->input, e->shift);
155 		KUNIT_EXPECT_EQ_MSG(test, ret, e->result,
156 				    "    when evaluating __lshrdi3(%lld, %d)",
157 				    e->input, e->shift);
158 	}
159 }
160 
161 static struct kunit_case shdi3_test_cases[] = {
162 	KUNIT_CASE(shdi3_test_ashldi3),
163 	KUNIT_CASE(shdi3_test_ashrdi3),
164 	KUNIT_CASE(shdi3_test_lshrdi3),
165 	{}
166 };
167 
168 static struct kunit_suite shdi3_test_suite = {
169 	.name = "shdi3",
170 	.test_cases = shdi3_test_cases,
171 };
172 kunit_test_suite(shdi3_test_suite);
173 
174 MODULE_DESCRIPTION("Test cases for __ashldi3(), __ashrdi3(), and __lshrdi3()");
175 MODULE_LICENSE("GPL");
176