/*	$OpenBSD: threads.c,v 1.1.1.1 2012/07/13 17:49:53 eric Exp $	*/
/*
 * Copyright (c) 2012 Eric Faurot <eric@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
#include <sys/time.h>
#include <sys/resource.h>

#include <assert.h>
#include <err.h>
#include <getopt.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>

#include <netdb.h>

#define MAX_THREADS	50

int	  ac;
char	**av;
int	  loop;
int	  nthreads;

int long_err;
int gai_errno;
int rrset_errno;

void async_resolver_done(void *);

void stats(void)
{
	struct rusage	ru;

	getrusage(RUSAGE_SELF, &ru);
	printf("%li\n", ru.ru_maxrss);
}

void*
task(void *arg)
{
	int		 id, i, j, c;
	struct addrinfo *ai, *n;

	id = *((int*) arg);

	n = NULL; c =0;

	for(j = 0; j < loop; j++)
		for(i = 0; i < ac; i++) {
			if (getaddrinfo(av[i], NULL, NULL, &ai) == 0) {
/*
				for (c = 0, n = ai; n; c++, n = n->ai_next);
				printf("%i:%s: ok: %i\n", id, av[i], c);
*/
				freeaddrinfo(ai);
			} else {
/*
				printf("%i:%s: fail\n", id, av[i]);
*/
			}
		}
	return (NULL);
}

void
usage(void)
{
	extern const char *__progname;
	fprintf(stderr, "usage: %s [-L loop] [-l loop] [-t threads] <host> ...\n",
		__progname);
}

int
main(int argc, char **argv)
{
	pthread_t	th[MAX_THREADS];
	int		th_args[MAX_THREADS], r, i, ch;
	int		n, LOOP;

	nthreads = 1;
	loop = 1;
	LOOP = 1;

	while ((ch = getopt(argc, argv, "L:l:t:")) != -1) {
		switch (ch) {
		case 'L':
			LOOP = atoi(optarg);
			break;
		case 'l':
			loop = atoi(optarg);
			break;
		case 't':
			nthreads = atoi(optarg);
			if (nthreads > MAX_THREADS)
				nthreads = MAX_THREADS;
			break;
		default:
			usage();
			/* NOTREACHED */
		}
	}

	argc -= optind;
	argv += optind;

	ac = argc;
	av = argv;

	printf("%i %i %i\n", LOOP, nthreads, loop);
	for (n = 0; n < LOOP; n ++) {
		for (i = 0; i < nthreads; i++) {
			th_args[i] = i;
			r = pthread_create(&th[i], NULL, task, (void *) &th_args[i]);
			if (r == -1)
				errx(1, "pthread_create");
		}
		for (i = 0; i < nthreads; i++)
			pthread_join(th[i], NULL);

		if (nthreads == 0)
			task(&n);

		stats();
	}

	return (0);
}