/* $NetBSD: nbperf.c,v 1.9 2024/09/22 20:34:26 christos Exp $ */ /*- * Copyright (c) 2009 The NetBSD Foundation, Inc. * All rights reserved. * * This code is derived from software contributed to The NetBSD Foundation * by Joerg Sonnenberger. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #if HAVE_NBTOOL_CONFIG_H #include "nbtool_config.h" #endif #include __RCSID("$NetBSD: nbperf.c,v 1.9 2024/09/22 20:34:26 christos Exp $"); #include #include #include #include #include #include #include #include #include #include "nbperf.h" static int predictable; static __dead void usage(void) { fprintf(stderr, "%s [-ps] [-c utilisation] [-i iterations] [-n name] " "[-o output] input\n", getprogname()); exit(1); } #if HAVE_NBTOOL_CONFIG_H #define arc4random() rand() #endif static void mi_vector_hash_seed_hash(struct nbperf *nbperf) { static uint32_t predictable_counter; if (predictable) nbperf->seed[0] = predictable_counter++; else nbperf->seed[0] = arc4random(); } static void mi_vector_hash_compute(struct nbperf *nbperf, const void *key, size_t keylen, uint32_t *hashes) { mi_vector_hash(key, keylen, nbperf->seed[0], hashes); } static void mi_vector_hash_print_hash(struct nbperf *nbperf, const char *indent, const char *key, const char *keylen, const char *hash) { fprintf(nbperf->output, "%smi_vector_hash(%s, %s, 0x%08" PRIx32 "U, %s);\n", indent, key, keylen, nbperf->seed[0], hash); } static void set_hash(struct nbperf *nbperf, const char *arg) { if (strcmp(arg, "mi_vector_hash") == 0) { nbperf->hash_size = 3; nbperf->seed_hash = mi_vector_hash_seed_hash; nbperf->compute_hash = mi_vector_hash_compute; nbperf->print_hash = mi_vector_hash_print_hash; return; } if (nbperf->hash_size > NBPERF_MAX_HASH_SIZE) errx(1, "Hash function creates too many output values"); errx(1, "Unknown hash function: %s", arg); } int main(int argc, char **argv) { struct nbperf nbperf = { .c = 0, .hash_name = "hash", .map_output = NULL, .output = NULL, .static_hash = 0, .check_duplicates = 0, .has_duplicates = 0, .allow_hash_fudging = 0, }; FILE *input; size_t curlen = 0, curalloc = 0; char *line, *eos; ssize_t line_len; size_t line_allocated; const void **keys = NULL; size_t *keylens = NULL; uint32_t max_iterations = ~0U; long long tmp; int looped, ch; int (*build_hash)(struct nbperf *) = chm_compute; set_hash(&nbperf, "mi_vector_hash"); while ((ch = getopt(argc, argv, "a:c:fh:i:m:n:o:ps")) != -1) { switch (ch) { case 'a': /* Accept bdz as alias for netbsd-6 compat. */ if (strcmp(optarg, "chm") == 0) build_hash = chm_compute; else if (strcmp(optarg, "chm3") == 0) build_hash = chm3_compute; else if (strcmp(optarg, "bpz") == 0 || strcmp(optarg, "bdz") == 0) build_hash = bpz_compute; else errx(1, "Unsupported algorithm: %s", optarg); break; case 'c': errno = 0; nbperf.c = strtod(optarg, &eos); if (errno || eos[0] || !nbperf.c) errx(2, "Invalid argument for -c"); break; case 'f': nbperf.allow_hash_fudging = 1; break; case 'h': set_hash(&nbperf, optarg); break; case 'i': errno = 0; tmp = strtoll(optarg, &eos, 0); if (errno || eos == optarg || eos[0] || tmp < 0 || tmp > 0xffffffffU) errx(2, "Iteration count must be " "a 32bit integer"); max_iterations = (uint32_t)tmp; break; case 'm': if (nbperf.map_output) fclose(nbperf.map_output); nbperf.map_output = fopen(optarg, "w"); if (nbperf.map_output == NULL) err(2, "cannot open map file"); break; case 'n': nbperf.hash_name = optarg; break; case 'o': if (nbperf.output) fclose(nbperf.output); nbperf.output = fopen(optarg, "w"); if (nbperf.output == NULL) err(2, "cannot open output file"); break; case 'p': predictable = 1; break; case 's': nbperf.static_hash = 1; break; default: usage(); } } argc -= optind; argv += optind; if (argc > 1) usage(); if (argc == 1) { input = fopen(argv[0], "r"); if (input == NULL) err(1, "can't open input file"); } else input = stdin; if (nbperf.output == NULL) nbperf.output = stdout; line = NULL; line_allocated = 0; while ((line_len = getline(&line, &line_allocated, input)) != -1) { if (line_len && line[line_len - 1] == '\n') --line_len; if (curlen == curalloc) { if (curalloc < 256) curalloc = 256; else curalloc += curalloc; keys = realloc(keys, curalloc * sizeof(*keys)); if (keys == NULL) err(1, "realloc failed"); keylens = realloc(keylens, curalloc * sizeof(*keylens)); if (keylens == NULL) err(1, "realloc failed"); } if ((keys[curlen] = strndup(line, line_len)) == NULL) err(1, "malloc failed"); keylens[curlen] = line_len; ++curlen; } free(line); if (input != stdin) fclose(input); nbperf.n = curlen; nbperf.keys = keys; nbperf.keylens = keylens; looped = 0; int rv; for (;;) { rv = (*build_hash)(&nbperf); if (!rv) break; if (nbperf.has_duplicates) { fputc('\n', stderr); errx(1, "Duplicate keys detected"); } fputc('.', stderr); if (!looped) nbperf.check_duplicates = 1; looped = 1; if (max_iterations == ~0U) continue; if (--max_iterations == 0) { fputc('\n', stderr); errx(1, "Iteration count reached"); } } if (looped) fputc('\n', stderr); return 0; }