Commit | Line | Data |
---|---|---|
b64eb60f MW |
1 | /* -*-c-*- |
2 | * | |
3 | * Test driver for universal hashing | |
4 | * | |
5 | * (c) 2009 Straylight/Edgeware | |
6 | */ | |
7 | ||
8 | /*----- Licensing notice --------------------------------------------------* | |
9 | * | |
10 | * This file is part of the mLib utilities library. | |
11 | * | |
12 | * mLib is free software; you can redistribute it and/or modify | |
13 | * it under the terms of the GNU Library General Public License as | |
14 | * published by the Free Software Foundation; either version 2 of the | |
15 | * License, or (at your option) any later version. | |
16 | * | |
17 | * mLib is distributed in the hope that it will be useful, | |
18 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
20 | * GNU Library General Public License for more details. | |
21 | * | |
22 | * You should have received a copy of the GNU Library General Public | |
23 | * License along with mLib; if not, write to the Free | |
24 | * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, | |
25 | * MA 02111-1307, USA. | |
26 | */ | |
27 | ||
28 | /*----- Header files ------------------------------------------------------*/ | |
29 | ||
30 | #include "crc32.h" | |
31 | #include "unihash.h" | |
32 | #include "tvec.h" | |
33 | ||
34 | /*----- Main code ---------------------------------------------------------*/ | |
35 | ||
36 | enum { | |
37 | RH, NROUT, | |
38 | RK = NROUT, RM, NREG | |
39 | }; | |
40 | ||
41 | struct step { size_t s; }; | |
42 | ||
43 | static void test_crc32(const struct tvec_reg *in, struct tvec_reg *out, | |
44 | void *ctx) | |
45 | { | |
46 | const struct step *step = ctx; | |
47 | const unsigned char *p = in[RM].v.bytes.p; size_t sz = in[RM].v.bytes.sz; | |
48 | uint32 h; | |
49 | ||
50 | if (!step) | |
51 | out[RH].v.u = crc32(0, p, sz); | |
52 | else { | |
53 | for (h = 0; sz > step->s; p += step->s, sz -= step->s) | |
54 | h = crc32(h, p, step->s); | |
55 | out[RH].v.u = crc32(h, p, sz); | |
56 | } | |
57 | } | |
58 | ||
59 | static void bench_crc32(const struct tvec_reg *in, struct tvec_reg *out, | |
60 | void *ctx) | |
61 | { crc32(0, in[RM].v.bytes.p, in[RM].v.bytes.sz); } | |
62 | ||
63 | static void test_unihash(const struct tvec_reg *in, struct tvec_reg *out, | |
64 | void *ctx) | |
65 | { | |
66 | const struct step *step = ctx; | |
67 | unihash_info ui; | |
68 | const unsigned char *p = in[RM].v.bytes.p; size_t sz = in[RM].v.bytes.sz; | |
69 | uint32 h; | |
70 | ||
71 | unihash_setkey(&ui, in[RK].v.u); | |
72 | if (!step) | |
73 | out[RH].v.u = unihash(&ui, p, sz); | |
74 | else { | |
75 | for (h = UNIHASH_INIT(&ui); sz > step->s; p += step->s, sz -= step->s) | |
76 | h = unihash_hash(&ui, h, p, step->s); | |
77 | out[RH].v.u = unihash_hash(&ui, h, p, sz); | |
78 | } | |
79 | } | |
80 | ||
81 | static void bench_unihash(const struct tvec_reg *in, struct tvec_reg *out, | |
82 | void *ctx) | |
83 | { unihash_hash(ctx, 0, in[RM].v.bytes.p, in[RM].v.bytes.sz); } | |
e63124bc MW |
84 | static int setup_unihash(struct tvec_state *tv, |
85 | const struct tvec_env *env, void *pctx, void *ctx) | |
b64eb60f | 86 | { unihash_setkey(ctx, 0); return (0); } |
e63124bc MW |
87 | static const struct tvec_env unihash_benchenv = |
88 | { sizeof(unihash_info), setup_unihash, 0, 0 }; | |
b64eb60f | 89 | |
e63124bc | 90 | static void run_step(struct tvec_state *tv, tvec_testfn *fn, void *ctx) |
b64eb60f MW |
91 | { |
92 | static const size_t steps[] = { 1, 5, 6, 7, 8, 23 }; | |
93 | struct step step; | |
94 | size_t i; | |
95 | ||
e63124bc | 96 | fn(tv->in, tv->out, 0); |
b64eb60f MW |
97 | tvec_check(tv, "whole buffer"); |
98 | ||
99 | for (i = 0; i < N(steps); i++) { | |
100 | step.s = steps[i]; | |
e63124bc | 101 | fn(tv->in, tv->out, &step); |
b64eb60f MW |
102 | tvec_check(tv, "step = %lu", (unsigned long)steps[i]); |
103 | } | |
104 | } | |
105 | ||
e63124bc MW |
106 | static const struct tvec_env step_testenv = { 0, 0, 0, 0, run_step, 0, 0 }; |
107 | ||
b64eb60f MW |
108 | static const struct tvec_regdef unihash_regs[] = { |
109 | { "k", RK, &tvty_uint, 0, { &tvrange_u32 } }, | |
110 | { "m", RM, &tvty_bytes, 0 }, | |
111 | { "h", RH, &tvty_uint, 0, { &tvrange_u32 } }, | |
db2bf411 | 112 | TVEC_ENDREGS |
b64eb60f MW |
113 | }; |
114 | ||
115 | static const struct tvec_regdef crc32_regs[] = { | |
116 | { "m", RM, &tvty_bytes, 0 }, | |
117 | { "h", RH, &tvty_uint, 0, { &tvrange_u32 } }, | |
db2bf411 | 118 | TVEC_ENDREGS |
b64eb60f MW |
119 | }; |
120 | ||
121 | static const struct tvec_regdef bench_regs[] = { | |
122 | { "msz", RM, &tvty_buffer, TVRF_ID }, | |
db2bf411 | 123 | TVEC_ENDREGS |
b64eb60f MW |
124 | }; |
125 | ||
e63124bc MW |
126 | static const struct tvec_bench crc32_bench = |
127 | { TVEC_BENCHINIT, 1, -1, RM, 0 }; | |
128 | static const struct tvec_bench unihash_bench = | |
129 | { TVEC_BENCHINIT, 1, -1, RM, &unihash_benchenv }; | |
b64eb60f MW |
130 | |
131 | static const struct tvec_test tests[] = { | |
e63124bc MW |
132 | { "crc32", crc32_regs, &step_testenv, test_crc32 }, |
133 | { "unihash", unihash_regs, &step_testenv, test_unihash }, | |
134 | { "crc32-bench", bench_regs, &crc32_bench._env, bench_crc32 }, | |
135 | { "unihash-bench", bench_regs, &unihash_bench._env, bench_unihash }, | |
db2bf411 | 136 | TVEC_ENDTESTS |
b64eb60f MW |
137 | }; |
138 | ||
c5e0e403 | 139 | static const struct tvec_config testconfig = |
b64eb60f MW |
140 | { tests, NROUT, NREG, sizeof(struct tvec_reg) }; |
141 | ||
142 | int main(int argc, char *argv[]) | |
c5e0e403 | 143 | { return (tvec_main(argc, argv, &testconfig, 0)); } |
b64eb60f MW |
144 | |
145 | /*----- That's all, folks -------------------------------------------------*/ |