irix-657m-src/irix/cmd/stress/Select/us.c

/*
 * Copyright 1990-1996 Silicon Graphics, Inc.
 * All Rights Reserved.
 *
 * This is UNPUBLISHED PROPRIETARY SOURCE CODE of Silicon Graphics, Inc.;
 * the contents of this file may not be disclosed to third parties, copied or
 * duplicated in any form, in whole or in part, without the prior written
 * permission of Silicon Graphics, Inc.
 *
 * RESTRICTED RIGHTS LEGEND:
 * Use, duplication or disclosure by the Government is subject to restrictions
 * as set forth in subdivision (c)(1)(ii) of the Rights in Technical Data
 * and Computer Software clause at DFARS 252.227-7013, and/or in similar or
 * successor clauses in the FAR, DOD or NASA FAR Supplement. Unpublished -
 * rights reserved under the Copyright Laws of the United States.
 */
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netdb.h>
#include <stdio.h>
#include <signal.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/sysmp.h>
#include <unistd.h>

char *port = "/tmp/9999";
int num_servers = 1;
int backlog = 32;
unsigned aint = 10;
int lock = 0;
int acc_cnt = 0;

/* ARGSUSED */
void
ahand(int sig)
{
	exit(0);
}

void
error(char *s)
{
	perror(s);
	exit(1);
}

void
usage(void)
{
	(void) fprintf(stderr, 
		"usage: us [-i secs] [-L] [-s int] [-l backlog] [-n num] [-p port]\n");
	exit(1);
}

#if 0
/* not used for stress */
/* ARGSUSED */
void
ahand(int sig)
{
	static int acnt = 0;

	acnt++;
	(void) printf("# of connections accepted == %d\n",acc_cnt);
	(void) printf("rate = %f conn/s\n", 
		(float)acc_cnt/((float)acnt * (float)aint));
	(void) alarm(aint);
}
#endif

char buf[BUFSIZ];

main(int argc, char **argv)
{
	int sunlen;
	struct sockaddr_un sun;
	int s, r;
	int i;
	int secs = 300;
	int c;
	fd_set	readfds, tempfds;
	int on = 1;
	int ncpus;
	int cpu = 0;
	pid_t pid;

	ncpus = (int)sysmp(MP_NAPROCS);
	(void) printf("ncpus = %d\n",ncpus);

	s = socket(AF_UNIX, SOCK_STREAM, 0);

	if (s < 0) {
		error("socket");
	}

	r = setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on));
	if (r < 0) {
		error("SO_REUSEADDR");
	}

	while ((c = getopt(argc, argv, "i:Ls:l:p:n:")) != EOF) {
		switch (c) {
		case 'i':
			secs = atoi(optarg);
			break;
		case 'L':
			lock = 1;
			break;
		case 'n':
			num_servers = atoi(optarg);
			break;
		case 'p':
			port = optarg;
			break;
		case 'l':
			backlog = atoi(optarg);
			break;
		case 's':
			aint = atoi(optarg);
			break;
		default:
			usage();
		}
	}

	(void) unlink(port);	
	memset((char *)&sun, 0, sizeof(sun));
	sun.sun_family = AF_UNIX;
	strcpy(sun.sun_path, port);

	r = bind(s, (struct sockaddr *)&sun, (int)(sizeof(sun.sun_family) +
		strlen(port) + 1));
	if (r < 0) {
		error("bind");
	}

	(void) listen(s, backlog);

	for (i = 0; i < num_servers; i++) {
		if ((pid = fork()) == 0) {
			if (lock) {
				sysmp(MP_MUSTRUN, cpu);
				cpu++;
				if (cpu == ncpus) {
					cpu = 0;
				}					
			}
			goto child;
		}
		if (pid == -1) {
			error("fork");
			(void) kill(0, SIGTERM);
			exit(0);
		}
	}
	(void) sigset(SIGALRM, ahand);
	(void) alarm(secs);
	(void) pause();
	(void) kill(0, SIGTERM);
	exit(0);
	
child:
	(void) sigset(SIGALRM, ahand);
#if 0
	(void) alarm(aint);
#else
	(void) alarm(secs);
#endif
	FD_ZERO(&readfds);
	FD_SET(s, &readfds);
	while (1) {
		int n;
		int fd;
		int n2;

		sunlen = sizeof(sun);
		tempfds = readfds;
		n = select(getdtablehi() + 1, &tempfds, (fd_set *)0, 
			(fd_set *)0, (struct timeval *)0);
		if (r < 0) {
			if (errno == EINTR) {
				continue;
			}
			error("select");
		}
		if (FD_ISSET(s, &tempfds)) {
			r = accept(s, (struct sockaddr *)&sun, &sunlen);
			if (r < 0) {
				if (errno == EINTR) {
					continue;
				} else {
					error("accept");
				}
			}
			acc_cnt++;
			FD_SET(r, &readfds);
			n--;
		}
		n2 = getdtablehi();
		for (fd = 0; (fd < n2) && n; fd++) {
			if ((fd != s) && FD_ISSET(fd, &tempfds)) {
				r = read(fd, buf, sizeof(buf));
				if (r <= 0) {
					FD_CLR(fd, &readfds);
					close(fd);
				}
				n--;
			}
		}
	}
	/* NOTREACHED */
	exit(0);
}