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