xref: /freebsd/tests/sys/kern/libkern_crc32.c (revision b3e7694832e81d7a904a10f525f8797b753bf0d3)
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 
27 #include <sys/param.h>
28 #include <sys/gsb_crc32.h>
29 
30 #include <stdint.h>
31 
32 #include <atf-c.h>
33 
34 #if defined(__amd64__) || defined(__i386__)
35 #include <machine/cpufunc.h>
36 #include <machine/specialreg.h>
37 
38 static bool
have_sse42(void)39 have_sse42(void)
40 {
41 	u_int cpu_registers[4];
42 
43 	do_cpuid(1, cpu_registers);
44 
45 	return ((cpu_registers[2] & CPUID2_SSE42) != 0);
46 }
47 #endif
48 
49 static void
check_crc32c(uint32_t expected,uint32_t crc32c,const void * buffer,size_t length)50 check_crc32c(uint32_t expected, uint32_t crc32c, const void *buffer,
51     size_t length)
52 {
53 	uint32_t act;
54 
55 #if defined(__amd64__) || defined(__i386__)
56 	if (have_sse42()) {
57 		act = sse42_crc32c(crc32c, buffer, length);
58 		ATF_CHECK_EQ_MSG(expected, act,
59 		    "sse42_crc32c expected 0x%08x, got 0x%08x", expected, act);
60 	}
61 #elif defined(__aarch64__)
62 	act = armv8_crc32c(crc32c, buffer, length);
63 	ATF_CHECK_EQ_MSG(expected, act,
64 	    "armv8_crc32c expected 0x%08x, got 0x%08x", expected, act);
65 #endif
66 	act = singletable_crc32c(crc32c, buffer, length);
67 	ATF_CHECK_EQ_MSG(expected, act,
68 	    "singletable_crc32c expected 0x%08x, got 0x%08x", expected, act);
69 	act = multitable_crc32c(crc32c, buffer, length);
70 	ATF_CHECK_EQ_MSG(expected, act,
71 	    "multitable_crc32c expected 0x%08x, got 0x%08x", expected, act);
72 }
73 
74 ATF_TC_WITHOUT_HEAD(crc32c_basic_correctness);
ATF_TC_BODY(crc32c_basic_correctness,tc)75 ATF_TC_BODY(crc32c_basic_correctness, tc)
76 {
77 	const uint64_t inputs[] = {
78 		0xf408c634b3a9142,
79 		0x80539e8c7c352e2b,
80 		0x62e9121db6e4d649,
81 		0x899345850ed0a286,
82 		0x2302df11b4a43b15,
83 		0xe943de7b3d35d70,
84 		0xdf1ff2bf41abf56b,
85 		0x9bc138abae315de2,
86 		0x31cc82e56234f0ff,
87 		0xce63c0cd6988e847,
88 		0x3e42f6b78ee352fa,
89 		0xfa4085436078cfa6,
90 		0x53349558bf670a4b,
91 		0x2714e10e7d722c61,
92 		0xc0d3261addfc6908,
93 		0xd1567c3181d3a1bf,
94 	};
95 	const uint32_t results[] = {
96 		0x2ce33ede,
97 		0xc49cc573,
98 		0xb8683c96,
99 		0x6918660d,
100 		0xa904e522,
101 		0x52dbc42c,
102 		0x98863c22,
103 		0x894d5d2c,
104 		0xb003745d,
105 		0xfc496dbd,
106 		0x97d2fbb5,
107 		0x3c062ef1,
108 		0xcc2eff18,
109 		0x6a9b09f6,
110 		0x420242c1,
111 		0xfd562dc3,
112 	};
113 	size_t i;
114 
115 	ATF_REQUIRE(nitems(inputs) == nitems(results));
116 
117 	for (i = 0; i < nitems(inputs); i++) {
118 		check_crc32c(results[i], ~0u, &inputs[i], sizeof(inputs[0]));
119 	}
120 }
121 
122 ATF_TC_WITHOUT_HEAD(crc32c_alignment);
ATF_TC_BODY(crc32c_alignment,tc)123 ATF_TC_BODY(crc32c_alignment, tc)
124 {
125 	const uint64_t input = 0xf408c634b3a9142;
126 	const uint32_t result = 0x2ce33ede;
127 	unsigned char buf[15];
128 	size_t i;
129 
130 	for (i = 1; i < 8; i++) {
131 		memcpy(&buf[i], &input, sizeof(input));
132 		check_crc32c(result, ~0u, &buf[i], sizeof(input));
133 	}
134 }
135 
136 ATF_TC_WITHOUT_HEAD(crc32c_trailing_bytes);
ATF_TC_BODY(crc32c_trailing_bytes,tc)137 ATF_TC_BODY(crc32c_trailing_bytes, tc)
138 {
139 	const unsigned char input[] = {
140 		0x87, 0x54, 0x74, 0xd2, 0xb, 0x9b, 0xdd, 0xf6, 0x68, 0x37,
141 		0xd4, 0x4, 0x5e, 0xa9, 0xb3
142 	};
143 	const uint32_t result = 0xec638d62;
144 
145 	check_crc32c(result, ~0u, input, sizeof(input));
146 }
147 
ATF_TP_ADD_TCS(tp)148 ATF_TP_ADD_TCS(tp)
149 {
150 
151 	ATF_TP_ADD_TC(tp, crc32c_basic_correctness);
152 	ATF_TP_ADD_TC(tp, crc32c_alignment);
153 	ATF_TP_ADD_TC(tp, crc32c_trailing_bytes);
154 	return (atf_no_error());
155 }
156