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
23 static unsigned int syscalls, bindcalls;
24 static pid_t mypid;
25 static volatile sig_atomic_t doquit;
26
27 static void
signaali(int sig)28 signaali(int sig)
29 {
30
31 doquit = 1;
32 }
33
34 static const int hostnamemib[] = { CTL_KERN, KERN_HOSTNAME };
35 static char hostnamebuf[128];
36 #define HOSTNAMEBASE "rumpclient"
37
38 static int iskiller;
39
40 static void *
client(void * arg)41 client(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
127 int
main(int argc,char * argv[])128 main(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