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
38static bool
39have_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
49static void
50check_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
74ATF_TC_WITHOUT_HEAD(crc32c_basic_correctness);
75ATF_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
122ATF_TC_WITHOUT_HEAD(crc32c_alignment);
123ATF_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
136ATF_TC_WITHOUT_HEAD(crc32c_trailing_bytes);
137ATF_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
148ATF_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