xref: /freebsd/tests/sys/kern/libkern_crc32.c (revision 9768746b)
1 /*
2  * Copyright (c) 2017 Conrad Meyer <cem@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #include <sys/param.h>
30 #include <sys/gsb_crc32.h>
31 
32 #include <stdint.h>
33 
34 #include <atf-c.h>
35 
36 #if defined(__amd64__) || defined(__i386__)
37 #include <machine/cpufunc.h>
38 #include <machine/specialreg.h>
39 
40 static bool
41 have_sse42(void)
42 {
43 	u_int cpu_registers[4];
44 
45 	do_cpuid(1, cpu_registers);
46 
47 	return ((cpu_registers[2] & CPUID2_SSE42) != 0);
48 }
49 #endif
50 
51 static void
52 check_crc32c(uint32_t expected, uint32_t crc32c, const void *buffer,
53     size_t length)
54 {
55 	uint32_t act;
56 
57 #if defined(__amd64__) || defined(__i386__)
58 	if (have_sse42()) {
59 		act = sse42_crc32c(crc32c, buffer, length);
60 		ATF_CHECK_EQ_MSG(expected, act,
61 		    "sse42_crc32c expected 0x%08x, got 0x%08x", expected, act);
62 	}
63 #elif defined(__aarch64__)
64 	act = armv8_crc32c(crc32c, buffer, length);
65 	ATF_CHECK_EQ_MSG(expected, act,
66 	    "armv8_crc32c expected 0x%08x, got 0x%08x", expected, act);
67 #endif
68 	act = singletable_crc32c(crc32c, buffer, length);
69 	ATF_CHECK_EQ_MSG(expected, act,
70 	    "singletable_crc32c expected 0x%08x, got 0x%08x", expected, act);
71 	act = multitable_crc32c(crc32c, buffer, length);
72 	ATF_CHECK_EQ_MSG(expected, act,
73 	    "multitable_crc32c expected 0x%08x, got 0x%08x", expected, act);
74 }
75 
76 ATF_TC_WITHOUT_HEAD(crc32c_basic_correctness);
77 ATF_TC_BODY(crc32c_basic_correctness, tc)
78 {
79 	const uint64_t inputs[] = {
80 		0xf408c634b3a9142,
81 		0x80539e8c7c352e2b,
82 		0x62e9121db6e4d649,
83 		0x899345850ed0a286,
84 		0x2302df11b4a43b15,
85 		0xe943de7b3d35d70,
86 		0xdf1ff2bf41abf56b,
87 		0x9bc138abae315de2,
88 		0x31cc82e56234f0ff,
89 		0xce63c0cd6988e847,
90 		0x3e42f6b78ee352fa,
91 		0xfa4085436078cfa6,
92 		0x53349558bf670a4b,
93 		0x2714e10e7d722c61,
94 		0xc0d3261addfc6908,
95 		0xd1567c3181d3a1bf,
96 	};
97 	const uint32_t results[] = {
98 		0x2ce33ede,
99 		0xc49cc573,
100 		0xb8683c96,
101 		0x6918660d,
102 		0xa904e522,
103 		0x52dbc42c,
104 		0x98863c22,
105 		0x894d5d2c,
106 		0xb003745d,
107 		0xfc496dbd,
108 		0x97d2fbb5,
109 		0x3c062ef1,
110 		0xcc2eff18,
111 		0x6a9b09f6,
112 		0x420242c1,
113 		0xfd562dc3,
114 	};
115 	size_t i;
116 
117 	ATF_REQUIRE(nitems(inputs) == nitems(results));
118 
119 	for (i = 0; i < nitems(inputs); i++) {
120 		check_crc32c(results[i], ~0u, &inputs[i], sizeof(inputs[0]));
121 	}
122 }
123 
124 ATF_TC_WITHOUT_HEAD(crc32c_alignment);
125 ATF_TC_BODY(crc32c_alignment, tc)
126 {
127 	const uint64_t input = 0xf408c634b3a9142;
128 	const uint32_t result = 0x2ce33ede;
129 	unsigned char buf[15];
130 	size_t i;
131 
132 	for (i = 1; i < 8; i++) {
133 		memcpy(&buf[i], &input, sizeof(input));
134 		check_crc32c(result, ~0u, &buf[i], sizeof(input));
135 	}
136 }
137 
138 ATF_TC_WITHOUT_HEAD(crc32c_trailing_bytes);
139 ATF_TC_BODY(crc32c_trailing_bytes, tc)
140 {
141 	const unsigned char input[] = {
142 		0x87, 0x54, 0x74, 0xd2, 0xb, 0x9b, 0xdd, 0xf6, 0x68, 0x37,
143 		0xd4, 0x4, 0x5e, 0xa9, 0xb3
144 	};
145 	const uint32_t result = 0xec638d62;
146 
147 	check_crc32c(result, ~0u, input, sizeof(input));
148 }
149 
150 ATF_TP_ADD_TCS(tp)
151 {
152 
153 	ATF_TP_ADD_TC(tp, crc32c_basic_correctness);
154 	ATF_TP_ADD_TC(tp, crc32c_alignment);
155 	ATF_TP_ADD_TC(tp, crc32c_trailing_bytes);
156 	return (atf_no_error());
157 }
158