1/*	$NetBSD: h_stresscli.c,v 1.9 2011/06/26 13:17:36 christos Exp $	*/
2
3#include <sys/types.h>
4#include <sys/atomic.h>
5#include <sys/sysctl.h>
6#include <sys/wait.h>
7#include <sys/socket.h>
8
9#include <netinet/in.h>
10
11#include <err.h>
12#include <fcntl.h>
13#include <pthread.h>
14#include <stdio.h>
15#include <stdlib.h>
16#include <string.h>
17#include <unistd.h>
18#include <signal.h>
19
20#include <rump/rump_syscalls.h>
21#include <rump/rumpclient.h>
22
23static unsigned int syscalls, bindcalls;
24static pid_t mypid;
25static volatile sig_atomic_t doquit;
26
27static void
28signaali(int sig)
29{
30
31	doquit = 1;
32}
33
34static const int hostnamemib[] = { CTL_KERN, KERN_HOSTNAME };
35static char hostnamebuf[128];
36#define HOSTNAMEBASE "rumpclient"
37
38static int iskiller;
39
40static void *
41client(void *arg)
42{
43	char buf[256];
44	struct sockaddr_in sin;
45	size_t blen;
46	int port = (int)(uintptr_t)arg;
47	int s, fd, x;
48
49	memset(&sin, 0, sizeof(sin));
50	sin.sin_family = AF_INET;
51	sin.sin_len = sizeof(sin);
52	sin.sin_port = htons(port);
53
54	while (!doquit) {
55		pid_t pidi;
56		blen = sizeof(buf);
57		s = rump_sys_socket(PF_INET, SOCK_STREAM, 0);
58		if (s == -1)
59			err(1, "socket");
60		atomic_inc_uint(&syscalls);
61
62		fd = rump_sys_open("/dev/null", O_RDWR);
63		atomic_inc_uint(&syscalls);
64
65		if (doquit)
66			goto out;
67
68		x = 1;
69		if (rump_sys_setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
70		    &x, sizeof(x)) == -1)
71			err(1, "reuseaddr");
72
73		/*
74		 * we don't really know when the kernel handles our disconnect,
75		 * so be soft about about the failure in case of a killer client
76		 */
77		if (rump_sys_bind(s, (struct sockaddr*)&sin, sizeof(sin))==-1) {
78			if (!iskiller)
79				err(1, "bind to port %d failed",
80				    ntohs(sin.sin_port));
81		} else {
82			atomic_inc_uint(&bindcalls);
83		}
84		atomic_inc_uint(&syscalls);
85
86		if (doquit)
87			goto out;
88
89		if (rump_sys___sysctl(hostnamemib, __arraycount(hostnamemib),
90		    buf, &blen, NULL, 0) == -1)
91			err(1, "sysctl");
92		if (strncmp(buf, hostnamebuf, sizeof(HOSTNAMEBASE)-1) != 0)
93			errx(1, "hostname (%s/%s) mismatch", buf, hostnamebuf);
94		atomic_inc_uint(&syscalls);
95
96		if (doquit)
97			goto out;
98
99		pidi = rump_sys_getpid();
100		if (pidi == -1)
101			err(1, "getpid");
102		if (pidi != mypid)
103			errx(1, "mypid mismatch");
104		atomic_inc_uint(&syscalls);
105
106		if (doquit)
107			goto out;
108
109		if (rump_sys_write(fd, buf, 16) != 16)
110			err(1, "write /dev/null");
111		atomic_inc_uint(&syscalls);
112
113 out:
114		rump_sys_close(fd);
115		atomic_inc_uint(&syscalls);
116		rump_sys_close(s);
117		atomic_inc_uint(&syscalls);
118	}
119
120	return NULL;
121}
122
123/* Stress with max 32 clients, 8 threads each (256 concurrent threads) */
124#define NCLI 32
125#define NTHR 8
126
127int
128main(int argc, char *argv[])
129{
130	pthread_t pt[NTHR-1];
131	pid_t clis[NCLI];
132	pid_t apid;
133	int ncli = 0;
134	int i = 0, j;
135	int status, thesig;
136	int rounds, myport;
137
138	if (argc != 2 && argc != 3)
139		errx(1, "need roundcount");
140
141	if (argc == 3) {
142		if (strcmp(argv[2], "kill") != 0)
143			errx(1, "optional 3rd param must be kill");
144		thesig = SIGKILL;
145		iskiller = 1;
146	} else {
147		thesig = SIGUSR1;
148	}
149
150	signal(SIGUSR1, signaali);
151
152	memset(clis, 0, sizeof(clis));
153	for (rounds = 1; rounds < atoi(argv[1])*10; rounds++) {
154		while (ncli < NCLI) {
155			switch ((apid = fork())) {
156			case -1:
157				err(1, "fork failed");
158			case 0:
159				if (rumpclient_init() == -1)
160					err(1, "rumpclient init");
161
162				mypid = rump_sys_getpid();
163				sprintf(hostnamebuf, HOSTNAMEBASE "%d", mypid);
164				if (rump_sys___sysctl(hostnamemib,
165				    __arraycount(hostnamemib), NULL, NULL,
166				    hostnamebuf, strlen(hostnamebuf)+1) == -1)
167					err(1, "sethostname");
168
169				for (j = 0; j < NTHR-1; j++) {
170					myport = i*NCLI + j+2;
171					if (pthread_create(&pt[j], NULL,
172					    client,
173					    (void*)(uintptr_t)myport) !=0 )
174						err(1, "pthread create");
175				}
176				myport = i*NCLI+1;
177				client((void *)(uintptr_t)myport);
178				for (j = 0; j < NTHR-1; j++)
179					pthread_join(pt[j], NULL);
180				membar_consumer();
181				fprintf(stderr, "done %d\n", syscalls);
182				exit(0);
183				/* NOTREACHED */
184			default:
185				ncli++;
186				clis[i] = apid;
187				break;
188			}
189
190			i = (i + 1) % NCLI;
191		}
192
193		usleep(100000);
194		kill(clis[i], thesig);
195
196		apid = wait(&status);
197		if (apid != clis[i])
198			errx(1, "wanted pid %d, got %d\n", clis[i], apid);
199		clis[i] = 0;
200		ncli--;
201		if (thesig == SIGUSR1) {
202			if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
203				fprintf(stderr, "child died with 0x%x\n",
204				    status);
205				exit(1);
206			}
207		} else {
208			if (!WIFSIGNALED(status) || WTERMSIG(status) != thesig){
209				fprintf(stderr, "child died with 0x%x\n",
210				    status);
211				exit(1);
212			}
213		}
214	}
215
216	for (i = 0; i < NCLI; i++)
217		if (clis[i])
218			kill(clis[i], SIGKILL);
219}
220