1e858faa9SBrian Feldman /*-
2e858faa9SBrian Feldman * Copyright (c) 2004 Brian Fundakowski Feldman
3e858faa9SBrian Feldman * All rights reserved.
4e858faa9SBrian Feldman *
5e858faa9SBrian Feldman * Redistribution and use in source and binary forms, with or without
6e858faa9SBrian Feldman * modification, are permitted provided that the following conditions
7e858faa9SBrian Feldman * are met:
8e858faa9SBrian Feldman * 1. Redistributions of source code must retain the above copyright
9e858faa9SBrian Feldman * notice, this list of conditions and the following disclaimer.
10e858faa9SBrian Feldman * 2. Redistributions in binary form must reproduce the above copyright
11e858faa9SBrian Feldman * notice, this list of conditions and the following disclaimer in the
12e858faa9SBrian Feldman * documentation and/or other materials provided with the distribution.
13e858faa9SBrian Feldman *
14e858faa9SBrian Feldman * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15e858faa9SBrian Feldman * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16e858faa9SBrian Feldman * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17e858faa9SBrian Feldman * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18e858faa9SBrian Feldman * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19e858faa9SBrian Feldman * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20e858faa9SBrian Feldman * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21e858faa9SBrian Feldman * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22e858faa9SBrian Feldman * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23e858faa9SBrian Feldman * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24e858faa9SBrian Feldman * SUCH DAMAGE.
25e858faa9SBrian Feldman */
26e858faa9SBrian Feldman
27e858faa9SBrian Feldman #include <sys/types.h>
28e858faa9SBrian Feldman #include <sys/socket.h>
29bdb764aaSBrian Feldman #include <sys/time.h>
30e858faa9SBrian Feldman
31e858faa9SBrian Feldman #include <netinet/in.h>
32e858faa9SBrian Feldman
33e858faa9SBrian Feldman #include <err.h>
34e858faa9SBrian Feldman #include <netdb.h>
35e858faa9SBrian Feldman #include <pthread.h>
36e858faa9SBrian Feldman #include <resolv.h>
37e858faa9SBrian Feldman #include <stdio.h>
38e858faa9SBrian Feldman #include <stdint.h>
39e858faa9SBrian Feldman #include <stdlib.h>
40e858faa9SBrian Feldman #include <string.h>
41e858faa9SBrian Feldman #include <unistd.h>
42e858faa9SBrian Feldman
43e858faa9SBrian Feldman /* Per-thread struct containing all important data. */
44e858faa9SBrian Feldman struct worker {
45e858faa9SBrian Feldman pthread_t w_thread; /* self */
46e858faa9SBrian Feldman uintmax_t w_lookup_success, w_lookup_failure; /* getaddrinfo stats */
47bdb764aaSBrian Feldman struct timespec w_max_lookup_time;
48e858faa9SBrian Feldman };
49e858faa9SBrian Feldman
50e858faa9SBrian Feldman static volatile int workers_stop = 0;
51e858faa9SBrian Feldman static double max_random_sleep = 1.0;
52e858faa9SBrian Feldman static char **randwords;
53e858faa9SBrian Feldman static size_t nrandwords;
54b99c0fd2SBrian Feldman static const struct addrinfo *hints, hintipv4only = { .ai_family = AF_INET };
55e858faa9SBrian Feldman
56e858faa9SBrian Feldman /*
57e858faa9SBrian Feldman * We don't have good random(3)-type functions that are thread-safe,
58e858faa9SBrian Feldman * unfortunately.
59e858faa9SBrian Feldman */
60e858faa9SBrian Feldman static u_int32_t
my_arc4random_r(void)61e858faa9SBrian Feldman my_arc4random_r(void)
62e858faa9SBrian Feldman {
63e858faa9SBrian Feldman static pthread_mutex_t mymutex = PTHREAD_MUTEX_INITIALIZER;
64e858faa9SBrian Feldman u_int32_t ret;
65e858faa9SBrian Feldman
66e858faa9SBrian Feldman (void)pthread_mutex_lock(&mymutex);
67e858faa9SBrian Feldman ret = arc4random();
68e858faa9SBrian Feldman (void)pthread_mutex_unlock(&mymutex);
69e858faa9SBrian Feldman return (ret);
70e858faa9SBrian Feldman }
71e858faa9SBrian Feldman
72e858faa9SBrian Feldman static void
randomsleep(double max_sleep_sec)73e858faa9SBrian Feldman randomsleep(double max_sleep_sec)
74e858faa9SBrian Feldman {
75e858faa9SBrian Feldman struct timespec slptime = { 0, 0 };
76e858faa9SBrian Feldman double rndsleep;
77e858faa9SBrian Feldman
78e858faa9SBrian Feldman rndsleep = (double)my_arc4random_r() / 4294967296.0 * max_sleep_sec;
79e858faa9SBrian Feldman while (rndsleep >= 1.0) {
80e858faa9SBrian Feldman slptime.tv_sec++;
81e858faa9SBrian Feldman rndsleep -= 1.0;
82e858faa9SBrian Feldman }
83e858faa9SBrian Feldman slptime.tv_nsec = rndsleep * 1e9;
84e858faa9SBrian Feldman (void)nanosleep(&slptime, NULL);
85e858faa9SBrian Feldman }
86e858faa9SBrian Feldman
87e858faa9SBrian Feldman /*
88e858faa9SBrian Feldman * Start looking up arbitrary hostnames and record the successes/failures.
89e858faa9SBrian Feldman * Between lookups, sleep a random amount of time to make sure threads
90e858faa9SBrian Feldman * stay well out of synchronization.
91b99c0fd2SBrian Feldman *
92b99c0fd2SBrian Feldman * Host name: part probability
93b99c0fd2SBrian Feldman * ---- -----------
94b99c0fd2SBrian Feldman * www. 1/2
95b99c0fd2SBrian Feldman * random word always, equal
96b99c0fd2SBrian Feldman * random word 1/3, equal
97b99c0fd2SBrian Feldman * .(net|com|org) equal
98e858faa9SBrian Feldman */
99e858faa9SBrian Feldman static void *
work(void * arg)100e858faa9SBrian Feldman work(void *arg)
101e858faa9SBrian Feldman {
102e858faa9SBrian Feldman struct worker *w = arg;
103e858faa9SBrian Feldman
104b99c0fd2SBrian Feldman /* Turn off domain name list searching as much as possible. */
105e858faa9SBrian Feldman if (_res.options & RES_INIT || res_init() == 0)
106e858faa9SBrian Feldman _res.options &= ~RES_DNSRCH;
107e858faa9SBrian Feldman do {
108e858faa9SBrian Feldman const char *suffixes[] = { "net", "com", "org" };
109e858faa9SBrian Feldman const size_t nsuffixes = sizeof(suffixes) / sizeof(suffixes[0]);
110bdb764aaSBrian Feldman struct timespec ts_begintime, ts_total;
111e858faa9SBrian Feldman struct addrinfo *res;
112e858faa9SBrian Feldman char *hostname;
113e858faa9SBrian Feldman int error;
114e858faa9SBrian Feldman
115e858faa9SBrian Feldman randomsleep(max_random_sleep);
116e858faa9SBrian Feldman if (asprintf(&hostname, "%s%s%s.%s",
117e858faa9SBrian Feldman (my_arc4random_r() % 2) == 0 ? "www." : "",
118e858faa9SBrian Feldman randwords[my_arc4random_r() % nrandwords],
119e858faa9SBrian Feldman (my_arc4random_r() % 3) == 0 ?
120e858faa9SBrian Feldman randwords[my_arc4random_r() % nrandwords] : "",
121e858faa9SBrian Feldman suffixes[my_arc4random_r() % nsuffixes]) == -1)
122e858faa9SBrian Feldman continue;
123bdb764aaSBrian Feldman (void)clock_gettime(CLOCK_REALTIME, &ts_begintime);
124b99c0fd2SBrian Feldman error = getaddrinfo(hostname, NULL, hints, &res);
125bdb764aaSBrian Feldman (void)clock_gettime(CLOCK_REALTIME, &ts_total);
126bdb764aaSBrian Feldman ts_total.tv_sec -= ts_begintime.tv_sec;
127bdb764aaSBrian Feldman ts_total.tv_nsec -= ts_begintime.tv_nsec;
128bdb764aaSBrian Feldman if (ts_total.tv_nsec < 0) {
129bdb764aaSBrian Feldman ts_total.tv_sec--;
130bdb764aaSBrian Feldman ts_total.tv_nsec += 1000000000;
131bdb764aaSBrian Feldman }
132bdb764aaSBrian Feldman if (ts_total.tv_sec > w->w_max_lookup_time.tv_sec ||
133bdb764aaSBrian Feldman (ts_total.tv_sec == w->w_max_lookup_time.tv_sec &&
134bdb764aaSBrian Feldman ts_total.tv_nsec > w->w_max_lookup_time.tv_sec))
135bdb764aaSBrian Feldman w->w_max_lookup_time = ts_total;
136e858faa9SBrian Feldman free(hostname);
137e858faa9SBrian Feldman if (error == 0) {
138e858faa9SBrian Feldman w->w_lookup_success++;
139e858faa9SBrian Feldman freeaddrinfo(res);
140e858faa9SBrian Feldman } else {
141e858faa9SBrian Feldman w->w_lookup_failure++;
142e858faa9SBrian Feldman }
143e858faa9SBrian Feldman } while (!workers_stop);
144e858faa9SBrian Feldman
145e858faa9SBrian Feldman pthread_exit(NULL);
146e858faa9SBrian Feldman }
147e858faa9SBrian Feldman
148e858faa9SBrian Feldman int
dowordfile(const char * fname)149e858faa9SBrian Feldman dowordfile(const char *fname)
150e858faa9SBrian Feldman {
151e858faa9SBrian Feldman FILE *fp;
152e858faa9SBrian Feldman char newword[64];
153e858faa9SBrian Feldman size_t n;
154e858faa9SBrian Feldman
155e858faa9SBrian Feldman fp = fopen(fname, "r");
156e858faa9SBrian Feldman if (fp == NULL)
157e858faa9SBrian Feldman return (-1);
158e858faa9SBrian Feldman nrandwords = 0;
159e858faa9SBrian Feldman while (fgets(newword, sizeof(newword), fp) != NULL)
160e858faa9SBrian Feldman nrandwords++;
161e858faa9SBrian Feldman if (ferror(fp) || fseek(fp, 0, SEEK_SET) != 0)
162e858faa9SBrian Feldman goto fail;
163e858faa9SBrian Feldman randwords = calloc(nrandwords, sizeof(char *));
164e858faa9SBrian Feldman if (randwords == NULL)
165e858faa9SBrian Feldman goto fail;
166e858faa9SBrian Feldman n = nrandwords;
167e858faa9SBrian Feldman nrandwords = 0;
168e858faa9SBrian Feldman while (fgets(newword, sizeof(newword), fp) != NULL) {
169e858faa9SBrian Feldman newword[strcspn(newword, "\r\n")] = '\0';
170e858faa9SBrian Feldman randwords[nrandwords] = strdup(newword);
171e858faa9SBrian Feldman if (randwords[nrandwords] == NULL)
172e858faa9SBrian Feldman err(1, "reading words file");
173e858faa9SBrian Feldman if (++nrandwords == n)
174e858faa9SBrian Feldman break;
175e858faa9SBrian Feldman }
176e858faa9SBrian Feldman nrandwords = n;
177e858faa9SBrian Feldman fclose(fp);
178e858faa9SBrian Feldman return (0);
179e858faa9SBrian Feldman fail:
180e858faa9SBrian Feldman fclose(fp);
181e858faa9SBrian Feldman return (-1);
182e858faa9SBrian Feldman }
183e858faa9SBrian Feldman
184e858faa9SBrian Feldman int
main(int argc,char ** argv)185e858faa9SBrian Feldman main(int argc, char **argv) {
186e858faa9SBrian Feldman unsigned long nworkers = 1;
187e858faa9SBrian Feldman struct worker *workers;
188e858faa9SBrian Feldman size_t i;
189e858faa9SBrian Feldman char waiting[3], *send, *wordfile = "/usr/share/dict/words";
190e858faa9SBrian Feldman int ch;
191e858faa9SBrian Feldman
192e858faa9SBrian Feldman if (getprogname() == NULL)
193e858faa9SBrian Feldman setprogname(argv[0]);
194e858faa9SBrian Feldman printf("%s: threaded stress-tester for getaddrinfo(3)\n",
195e858faa9SBrian Feldman getprogname());
196e858faa9SBrian Feldman printf("(c) 2004 Brian Feldman <green@FreeBSD.org>\n");
197b99c0fd2SBrian Feldman while ((ch = getopt(argc, argv, "4s:t:w:")) != -1) {
198e858faa9SBrian Feldman switch (ch) {
199b99c0fd2SBrian Feldman case '4':
200b99c0fd2SBrian Feldman hints = &hintipv4only;
201b99c0fd2SBrian Feldman break;
202e858faa9SBrian Feldman case 's':
203e858faa9SBrian Feldman max_random_sleep = strtod(optarg, &send);
204e858faa9SBrian Feldman if (*send != '\0')
205e858faa9SBrian Feldman goto usage;
206e858faa9SBrian Feldman break;
207e858faa9SBrian Feldman case 't':
208e858faa9SBrian Feldman nworkers = strtoul(optarg, &send, 0);
209e858faa9SBrian Feldman if (*send != '\0')
210e858faa9SBrian Feldman goto usage;
211e858faa9SBrian Feldman break;
212e858faa9SBrian Feldman case 'w':
213e858faa9SBrian Feldman wordfile = optarg;
214e858faa9SBrian Feldman break;
215e858faa9SBrian Feldman default:
216e858faa9SBrian Feldman usage:
217b99c0fd2SBrian Feldman fprintf(stderr, "usage: %s [-4] [-s sleep] "
218b99c0fd2SBrian Feldman "[-t threads] [-w wordfile]\n", getprogname());
219e858faa9SBrian Feldman exit(2);
220e858faa9SBrian Feldman }
221e858faa9SBrian Feldman }
222e858faa9SBrian Feldman argc -= optind;
223e858faa9SBrian Feldman argv += optind;
224e858faa9SBrian Feldman
225e858faa9SBrian Feldman if (nworkers < 1 || nworkers != (size_t)nworkers)
226e858faa9SBrian Feldman goto usage;
227e858faa9SBrian Feldman if (dowordfile(wordfile) == -1)
228e858faa9SBrian Feldman err(1, "reading word file %s", wordfile);
229e858faa9SBrian Feldman if (nrandwords < 1)
230e858faa9SBrian Feldman errx(1, "word file %s did not have >0 words", wordfile);
231f19fbaf8SEnji Cooper printf("Read %zu random words from %s.\n", nrandwords, wordfile);
232e858faa9SBrian Feldman workers = calloc(nworkers, sizeof(*workers));
233e858faa9SBrian Feldman if (workers == NULL)
234e858faa9SBrian Feldman err(1, "allocating workers");
235e858faa9SBrian Feldman printf("Intra-query delay time is from 0 to %g seconds (random).\n",
236e858faa9SBrian Feldman max_random_sleep);
237e858faa9SBrian Feldman
238e858faa9SBrian Feldman printf("Starting %lu worker%.*s: ", nworkers, nworkers > 1, "s");
239e858faa9SBrian Feldman fflush(stdout);
240e858faa9SBrian Feldman for (i = 0; i < nworkers; i++) {
241e858faa9SBrian Feldman if (pthread_create(&workers[i].w_thread, NULL, work,
2422949bd10SRuslan Ermilov &workers[i]) != 0)
243f19fbaf8SEnji Cooper err(1, "creating worker %zu", i);
244f19fbaf8SEnji Cooper printf("%zu%s", i, i == nworkers - 1 ? ".\n" : ", ");
245e858faa9SBrian Feldman fflush(stdout);
246e858faa9SBrian Feldman }
247e858faa9SBrian Feldman
248e858faa9SBrian Feldman printf("<Press enter key to end test.>\n");
249e858faa9SBrian Feldman (void)fgets(waiting, sizeof(waiting), stdin);
250e858faa9SBrian Feldman workers_stop = 1;
251e858faa9SBrian Feldman
252e858faa9SBrian Feldman printf("Stopping %lu worker%.*s: ", nworkers, nworkers > 1, "s");
253e858faa9SBrian Feldman fflush(stdout);
254e858faa9SBrian Feldman for (i = 0; i < nworkers; i++) {
255e858faa9SBrian Feldman pthread_join(workers[i].w_thread, NULL);
256f19fbaf8SEnji Cooper printf("%zu%s", i, i == nworkers - 1 ? ".\n" : ", ");
257e858faa9SBrian Feldman fflush(stdout);
258e858faa9SBrian Feldman }
259e858faa9SBrian Feldman
260bdb764aaSBrian Feldman printf("%-10s%-20s%-20s%-29s\n", "Worker", "Successful GAI",
261bdb764aaSBrian Feldman "Failed GAI", "Max resolution time (M:SS*)");
262bdb764aaSBrian Feldman printf("%-10s%-20s%-20s%-29s\n", "------", "--------------",
263bdb764aaSBrian Feldman "----------", "---------------------------");
264e858faa9SBrian Feldman for (i = 0; i < nworkers; i++) {
265f19fbaf8SEnji Cooper printf("%-10zu%-20ju%-20ju%ld:%s%.2f\n", i,
266bdb764aaSBrian Feldman workers[i].w_lookup_success, workers[i].w_lookup_failure,
267bdb764aaSBrian Feldman workers[i].w_max_lookup_time.tv_sec / 60,
268bdb764aaSBrian Feldman workers[i].w_max_lookup_time.tv_sec % 60 < 10 ? "0" : "",
269bdb764aaSBrian Feldman (double)(workers[i].w_max_lookup_time.tv_sec % 60) +
270bdb764aaSBrian Feldman (double)workers[i].w_max_lookup_time.tv_nsec / 1e9);
271e858faa9SBrian Feldman }
272e858faa9SBrian Feldman
273e858faa9SBrian Feldman exit(0);
274e858faa9SBrian Feldman }
275