2 Copyright (c) 2013 Cisco and/or its affiliates.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include <vppinfra/pfhash.h>
18 #include <vppinfra/format.h>
19 #include <vppinfra/random.h>
21 #if defined(CLIB_HAVE_VEC128) && ! defined (__ALTIVEC__)
26 random_aligned_string (u32 * seed, uword len)
28 u8 *alphabet = (u8 *) "abcdefghijklmnopqrstuvwxyz";
32 vec_resize_aligned (s, len, 16);
33 for (i = 0; i < len; i++)
34 s[i] = alphabet[random_u32 (seed) % 26];
42 test_pfhash_main (unformat_input_t * input)
44 u32 seed = 0xdeaddabe;
50 u32 sizes[3] = { 16, 8, 4 }, this_size, size;
52 pfhash_t _rec, *p = &_rec;
54 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
56 if (unformat (input, "seed %d", &seed))
58 else if (unformat (input, "niter %d", &niter))
60 else if (unformat (input, "nkeys %d", &nkeys))
62 else if (unformat (input, "nbuckets %d", &nbuckets))
64 else if (unformat (input, "verbose %d", &verbose))
66 else if (unformat (input, "verbose"))
69 clib_error ("unknown input `%U'", format_unformat_error, input);
72 vec_validate (keys, nkeys - 1);
74 for (i = 0; i < nkeys; i++)
79 keys[i] = random_aligned_string (&seed, 16);
80 for (j = 0; j < (i - 1); j++)
82 /* Make sure we don't have a dup key in the min key size */
83 for (k = 0; k < 4; k++)
85 if (keys[i][k] != keys[j][k])
95 /* test 8 byte key, 8 byte value case separately */
97 for (size = 8; size < 9; size++)
101 fformat (stdout, "%d-byte key 8 byte value test\n", this_size);
103 pfhash_init (p, "test", 8 /* key size */ , 8 /* value size */ ,
106 for (iter = 0; iter < niter; iter++)
109 for (i = 0; i < nkeys; i++)
111 bucket = (i % nbuckets) + 1;
112 pfhash_set (p, bucket, keys[i],
113 (void *) (u64) 0x100000000ULL + i + 1);
116 for (i = 0; i < nkeys; i++)
118 bucket = (i % nbuckets) + 1;
119 if (pfhash_get (p, bucket, keys[i])
120 != (u64) 0x100000000ULL + i + 1)
122 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
123 (void) pfhash_get (p, bucket, keys[i]);
127 /* test inline functions */
128 for (i = 0; i < nkeys; i++)
131 u64 value = 0xdeadbeef;
132 bucket = (i % nbuckets) + 1;
134 pfhash_prefetch_bucket (p, bucket);
135 bucket_contents = pfhash_read_bucket_prefetch_kv (p, bucket);
137 value = pfhash_search_kv_8v8 (p, bucket_contents,
139 if (value != (u64) 0x100000000ULL + i + 1)
140 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
144 fformat (stdout, "%U\n", format_pfhash, p, verbose > 1);
146 for (i = 0; i < nkeys; i++)
148 bucket = (i % nbuckets) + 1;
149 pfhash_unset (p, bucket, keys[i]);
152 for (i = 0; i < nkeys; i++)
154 bucket = (i % nbuckets) + 1;
155 if (pfhash_get (p, bucket, keys[i]) != (u64) ~ 0)
157 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
158 (void) pfhash_get (p, bucket, keys[i]);
161 /* test inline functions */
162 for (i = 0; i < nkeys; i++)
165 u64 value = 0xdeadbeef;
166 bucket = (i % nbuckets) + 1;
168 pfhash_prefetch_bucket (p, bucket);
169 bucket_contents = pfhash_read_bucket_prefetch_kv (p, bucket);
171 value = pfhash_search_kv_8v8 (p, bucket_contents,
174 if (value != (u64) ~ 0)
175 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
181 /* test other cases */
183 for (size = 0; size < ARRAY_LEN (sizes); size++)
185 this_size = sizes[size];
187 fformat (stdout, "%d-byte key test\n", this_size);
189 pfhash_init (p, "test", this_size, 4 /* value size */ , nbuckets + 1);
191 for (iter = 0; iter < niter; iter++)
194 for (i = 0; i < nkeys; i++)
196 bucket = (i % nbuckets) + 1;
197 pfhash_set (p, bucket, keys[i], (void *) (u64) i + 1);
200 for (i = 0; i < nkeys; i++)
202 bucket = (i % nbuckets) + 1;
203 if (pfhash_get (p, bucket, keys[i]) != i + 1)
205 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
206 (void) pfhash_get (p, bucket, keys[i]);
210 /* test inline functions */
211 for (i = 0; i < nkeys; i++)
214 u32 value = 0xdeadbeef;
215 bucket = (i % nbuckets) + 1;
217 pfhash_prefetch_bucket (p, bucket);
218 bucket_contents = pfhash_read_bucket_prefetch_kv (p, bucket);
223 pfhash_search_kv_16 (p, bucket_contents,
228 pfhash_search_kv_8 (p, bucket_contents, (u64 *) keys[i]);
232 pfhash_search_kv_4 (p, bucket_contents, (u32 *) keys[i]);
236 if (value != (i + 1))
237 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
241 fformat (stdout, "%U\n", format_pfhash, p, verbose > 1);
243 for (i = 0; i < nkeys; i++)
245 bucket = (i % nbuckets) + 1;
246 pfhash_unset (p, bucket, keys[i]);
249 for (i = 0; i < nkeys; i++)
251 bucket = (i % nbuckets) + 1;
252 if (pfhash_get (p, bucket, keys[i]) != (u64) ~ 0)
254 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
255 (void) pfhash_get (p, bucket, keys[i]);
258 /* test inline functions */
259 for (i = 0; i < nkeys; i++)
262 u32 value = 0xdeadbeef;
263 bucket = (i % nbuckets) + 1;
265 pfhash_prefetch_bucket (p, bucket);
266 bucket_contents = pfhash_read_bucket_prefetch_kv (p, bucket);
271 pfhash_search_kv_16 (p, bucket_contents,
276 pfhash_search_kv_8 (p, bucket_contents, (u64 *) keys[i]);
280 pfhash_search_kv_4 (p, bucket_contents, (u32 *) keys[i]);
283 if (value != (u32) ~ 0)
284 clib_warning ("key %d bucket %d lookup FAIL\n", i, bucket);
294 test_pfhash_main (unformat_input_t * input)
296 clib_warning ("MMX unit not available");
303 main (int argc, char *argv[])
308 unformat_init_command_line (&i, argv);
309 ret = test_pfhash_main (&i);
314 #endif /* CLIB_UNIX */
317 * fd.io coding-style-patch-verification: ON
320 * eval: (c-set-style "gnu")