1 /* $NetBSD: h_resolv.c,v 1.2 2010/11/03 16:10:22 christos Exp $ */
4 * Copyright (c) 2004, 2008 The NetBSD Foundation, Inc.
7 * This code is derived from software contributed to The NetBSD Foundation
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
32 #include <sys/cdefs.h>
33 __COPYRIGHT("@(#) Copyright (c) 2008\
34 The NetBSD Foundation, inc. All rights reserved.");
35 __RCSID("$NetBSD: h_resolv.c,v 1.2 2010/11/03 16:10:22 christos Exp $");
44 #include <stringlist.h>
50 static StringList *hosts = NULL;
52 static int *ask = NULL;
53 static int *got = NULL;
55 static void usage(void) __attribute__((__noreturn__));
56 static void load(const char *);
57 static void resolvone(int);
58 static void *resolvloop(void *);
59 static void run(int *);
61 static pthread_mutex_t stats = PTHREAD_MUTEX_INITIALIZER;
67 "Usage: %s [-d] [-h <nhosts>] [-n <nthreads>] <file> ...\n",
73 load(const char *fname)
79 if ((fp = fopen(fname, "r")) == NULL)
80 err(1, "Cannot open `%s'", fname);
81 while ((line = fgetln(fp, &len)) != NULL) {
85 for (ptr = strtok(line, WS); ptr; ptr = strtok(NULL, WS))
86 sl_add(hosts, strdup(ptr));
97 pthread_t self = pthread_self();
98 size_t i = (random() & 0x0fffffff) % hosts->sl_cur;
99 char *host = hosts->sl_str[i];
100 struct addrinfo *res;
103 len = snprintf(buf, sizeof(buf), "%p: %d resolving %s %d\n",
104 self, n, host, (int)i);
105 (void)write(STDOUT_FILENO, buf, len);
107 error = getaddrinfo(host, NULL, NULL, &res);
109 len = snprintf(buf, sizeof(buf), "%p: host %s %s\n",
110 self, host, error ? "not found" : "ok");
111 (void)write(STDOUT_FILENO, buf, len);
113 pthread_mutex_lock(&stats);
115 got[i] += error == 0;
116 pthread_mutex_unlock(&stats);
124 int *nhosts = (int *)p;
136 pthread_t self = pthread_self();
137 if (pthread_create(&self, NULL, resolvloop, nhosts) != 0)
138 err(1, "pthread_create");
142 main(int argc, char *argv[])
144 int nthreads = NTHREADS;
146 int i, c, done, *nleft;
151 while ((c = getopt(argc, argv, "dh:n:")) != -1)
157 nhosts = atoi(optarg);
160 nthreads = atoi(optarg);
166 for (i = optind; i < argc; i++)
169 if (hosts->sl_cur == 0)
172 if ((nleft = malloc(nthreads * sizeof(int))) == NULL)
174 if ((ask = calloc(hosts->sl_cur, sizeof(int))) == NULL)
176 if ((got = calloc(hosts->sl_cur, sizeof(int))) == NULL)
180 for (i = 0; i < nthreads; i++) {
185 for (done = 0; !done;) {
187 for (i = 0; i < nthreads; i++) {
196 for (i = 0; i < (int)hosts->sl_cur; i++) {
197 if (ask[i] != got[i] && got[i] != 0) {
198 warnx("Error: host %s ask %d got %d\n",
199 hosts->sl_str[i], ask[i], got[i]);