rumpclient.c revision 1.27 1 1.27 pooka /* $NetBSD: rumpclient.c,v 1.27 2011/02/09 14:29:58 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.11 pooka * Copyright (c) 2010, 2011 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Redistribution and use in source and binary forms, with or without
7 1.1 pooka * modification, are permitted provided that the following conditions
8 1.1 pooka * are met:
9 1.1 pooka * 1. Redistributions of source code must retain the above copyright
10 1.1 pooka * notice, this list of conditions and the following disclaimer.
11 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 pooka * notice, this list of conditions and the following disclaimer in the
13 1.1 pooka * documentation and/or other materials provided with the distribution.
14 1.1 pooka *
15 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 pooka * SUCH DAMAGE.
26 1.1 pooka */
27 1.1 pooka
28 1.1 pooka /*
29 1.1 pooka * Client side routines for rump syscall proxy.
30 1.1 pooka */
31 1.1 pooka
32 1.1 pooka #include <sys/cdefs.h>
33 1.1 pooka __RCSID("$NetBSD");
34 1.1 pooka
35 1.5 pooka #include <sys/param.h>
36 1.15 pooka #include <sys/event.h>
37 1.1 pooka #include <sys/mman.h>
38 1.1 pooka #include <sys/socket.h>
39 1.1 pooka
40 1.1 pooka #include <arpa/inet.h>
41 1.1 pooka #include <netinet/in.h>
42 1.1 pooka #include <netinet/tcp.h>
43 1.1 pooka
44 1.1 pooka #include <assert.h>
45 1.13 pooka #include <dlfcn.h>
46 1.1 pooka #include <errno.h>
47 1.1 pooka #include <fcntl.h>
48 1.13 pooka #include <link.h>
49 1.1 pooka #include <poll.h>
50 1.1 pooka #include <pthread.h>
51 1.11 pooka #include <signal.h>
52 1.1 pooka #include <stdarg.h>
53 1.18 pooka #include <stdbool.h>
54 1.1 pooka #include <stdio.h>
55 1.1 pooka #include <stdlib.h>
56 1.1 pooka #include <string.h>
57 1.1 pooka #include <unistd.h>
58 1.1 pooka
59 1.1 pooka #include <rump/rumpclient.h>
60 1.1 pooka
61 1.13 pooka #define HOSTOPS
62 1.13 pooka int (*host_socket)(int, int, int);
63 1.13 pooka int (*host_close)(int);
64 1.13 pooka int (*host_connect)(int, const struct sockaddr *, socklen_t);
65 1.15 pooka int (*host_fcntl)(int, int, ...);
66 1.13 pooka int (*host_poll)(struct pollfd *, nfds_t, int);
67 1.13 pooka ssize_t (*host_read)(int, void *, size_t);
68 1.13 pooka ssize_t (*host_sendto)(int, const void *, size_t, int,
69 1.13 pooka const struct sockaddr *, socklen_t);
70 1.13 pooka int (*host_setsockopt)(int, int, int, const void *, socklen_t);
71 1.27 pooka int (*host_dup)(int);
72 1.13 pooka
73 1.17 pooka int (*host_kqueue)(void);
74 1.17 pooka int (*host_kevent)(int, const struct kevent *, size_t,
75 1.17 pooka struct kevent *, size_t, const struct timespec *);
76 1.17 pooka
77 1.1 pooka #include "sp_common.c"
78 1.1 pooka
79 1.11 pooka static struct spclient clispc = {
80 1.11 pooka .spc_fd = -1,
81 1.11 pooka };
82 1.1 pooka
83 1.18 pooka static int kq = -1;
84 1.15 pooka static sigset_t fullset;
85 1.12 pooka
86 1.20 pooka static int doconnect(bool);
87 1.18 pooka static int handshake_req(struct spclient *, uint32_t *, int, bool);
88 1.18 pooka
89 1.20 pooka time_t retrytimo = RUMPCLIENT_RETRYCONN_ONCE;
90 1.18 pooka
91 1.18 pooka static int
92 1.18 pooka send_with_recon(struct spclient *spc, const void *data, size_t dlen)
93 1.18 pooka {
94 1.20 pooka struct timeval starttime, curtime;
95 1.20 pooka time_t prevreconmsg;
96 1.20 pooka unsigned reconretries;
97 1.18 pooka int rv;
98 1.18 pooka
99 1.20 pooka for (prevreconmsg = 0, reconretries = 0;;) {
100 1.18 pooka rv = dosend(spc, data, dlen);
101 1.18 pooka if (__predict_false(rv == ENOTCONN || rv == EBADF)) {
102 1.20 pooka /* no persistent connections */
103 1.20 pooka if (retrytimo == 0)
104 1.20 pooka break;
105 1.24 pooka if (retrytimo == RUMPCLIENT_RETRYCONN_DIE)
106 1.24 pooka exit(1);
107 1.20 pooka
108 1.20 pooka if (!prevreconmsg) {
109 1.20 pooka prevreconmsg = time(NULL);
110 1.20 pooka gettimeofday(&starttime, NULL);
111 1.20 pooka }
112 1.20 pooka if (reconretries == 1) {
113 1.20 pooka if (retrytimo == RUMPCLIENT_RETRYCONN_ONCE) {
114 1.20 pooka rv = ENOTCONN;
115 1.20 pooka break;
116 1.20 pooka }
117 1.20 pooka fprintf(stderr, "rump_sp: connection to "
118 1.20 pooka "kernel lost, trying to reconnect ...\n");
119 1.20 pooka } else if (time(NULL) - prevreconmsg > 120) {
120 1.20 pooka fprintf(stderr, "rump_sp: still trying to "
121 1.20 pooka "reconnect ...\n");
122 1.20 pooka prevreconmsg = time(NULL);
123 1.20 pooka }
124 1.20 pooka
125 1.20 pooka /* check that we aren't over the limit */
126 1.20 pooka if (retrytimo > 0) {
127 1.20 pooka struct timeval tmp;
128 1.20 pooka
129 1.20 pooka gettimeofday(&curtime, NULL);
130 1.20 pooka timersub(&curtime, &starttime, &tmp);
131 1.20 pooka if (tmp.tv_sec >= retrytimo) {
132 1.20 pooka fprintf(stderr, "rump_sp: reconnect "
133 1.20 pooka "failed, %lld second timeout\n",
134 1.20 pooka (long long)retrytimo);
135 1.20 pooka return ENOTCONN;
136 1.20 pooka }
137 1.20 pooka }
138 1.20 pooka
139 1.20 pooka /* adhoc backoff timer */
140 1.20 pooka if (reconretries < 10) {
141 1.20 pooka usleep(100000 * reconretries);
142 1.20 pooka } else {
143 1.20 pooka sleep(MIN(10, reconretries-9));
144 1.20 pooka }
145 1.20 pooka reconretries++;
146 1.20 pooka
147 1.20 pooka if ((rv = doconnect(false)) != 0)
148 1.18 pooka continue;
149 1.18 pooka if ((rv = handshake_req(&clispc, NULL, 0, true)) != 0)
150 1.18 pooka continue;
151 1.20 pooka
152 1.20 pooka /*
153 1.20 pooka * ok, reconnect succesful. we need to return to
154 1.20 pooka * the upper layer to get the entire PDU resent.
155 1.20 pooka */
156 1.20 pooka if (reconretries != 1)
157 1.20 pooka fprintf(stderr, "rump_sp: reconnected!\n");
158 1.20 pooka rv = EAGAIN;
159 1.20 pooka break;
160 1.20 pooka } else {
161 1.20 pooka _DIAGASSERT(errno != EAGAIN);
162 1.18 pooka break;
163 1.18 pooka }
164 1.20 pooka }
165 1.18 pooka
166 1.18 pooka return rv;
167 1.18 pooka }
168 1.18 pooka
169 1.12 pooka static int
170 1.18 pooka cliwaitresp(struct spclient *spc, struct respwait *rw, sigset_t *mask,
171 1.18 pooka bool keeplock)
172 1.12 pooka {
173 1.18 pooka uint64_t mygen;
174 1.18 pooka bool imalive = true;
175 1.12 pooka
176 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
177 1.18 pooka if (!keeplock)
178 1.18 pooka sendunlockl(spc);
179 1.18 pooka mygen = spc->spc_generation;
180 1.12 pooka
181 1.12 pooka rw->rw_error = 0;
182 1.18 pooka while (!rw->rw_done && rw->rw_error == 0) {
183 1.18 pooka if (__predict_false(spc->spc_generation != mygen || !imalive))
184 1.18 pooka break;
185 1.18 pooka
186 1.12 pooka /* are we free to receive? */
187 1.12 pooka if (spc->spc_istatus == SPCSTATUS_FREE) {
188 1.15 pooka struct kevent kev[8];
189 1.15 pooka int gotresp, dosig, rv, i;
190 1.15 pooka
191 1.12 pooka spc->spc_istatus = SPCSTATUS_BUSY;
192 1.12 pooka pthread_mutex_unlock(&spc->spc_mtx);
193 1.12 pooka
194 1.15 pooka dosig = 0;
195 1.15 pooka for (gotresp = 0; !gotresp; ) {
196 1.15 pooka switch (readframe(spc)) {
197 1.15 pooka case 0:
198 1.17 pooka rv = host_kevent(kq, NULL, 0,
199 1.15 pooka kev, __arraycount(kev), NULL);
200 1.18 pooka
201 1.18 pooka /*
202 1.18 pooka * XXX: don't know how this can
203 1.18 pooka * happen (timeout cannot expire
204 1.18 pooka * since there isn't one), but
205 1.18 pooka * it does happen
206 1.18 pooka */
207 1.18 pooka if (__predict_false(rv == 0))
208 1.18 pooka continue;
209 1.18 pooka
210 1.15 pooka for (i = 0; i < rv; i++) {
211 1.15 pooka if (kev[i].filter
212 1.15 pooka == EVFILT_SIGNAL)
213 1.15 pooka dosig++;
214 1.15 pooka }
215 1.15 pooka if (dosig)
216 1.15 pooka goto cleanup;
217 1.15 pooka
218 1.15 pooka continue;
219 1.15 pooka case -1:
220 1.18 pooka imalive = false;
221 1.15 pooka goto cleanup;
222 1.15 pooka default:
223 1.15 pooka break;
224 1.15 pooka }
225 1.12 pooka
226 1.15 pooka switch (spc->spc_hdr.rsp_class) {
227 1.12 pooka case RUMPSP_RESP:
228 1.12 pooka case RUMPSP_ERROR:
229 1.12 pooka kickwaiter(spc);
230 1.15 pooka gotresp = spc->spc_hdr.rsp_reqno ==
231 1.15 pooka rw->rw_reqno;
232 1.12 pooka break;
233 1.12 pooka case RUMPSP_REQ:
234 1.12 pooka handlereq(spc);
235 1.12 pooka break;
236 1.12 pooka default:
237 1.12 pooka /* panic */
238 1.12 pooka break;
239 1.15 pooka }
240 1.12 pooka }
241 1.12 pooka
242 1.15 pooka cleanup:
243 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
244 1.15 pooka if (spc->spc_istatus == SPCSTATUS_WANTED)
245 1.15 pooka kickall(spc);
246 1.15 pooka spc->spc_istatus = SPCSTATUS_FREE;
247 1.15 pooka
248 1.15 pooka /* take one for the team */
249 1.15 pooka if (dosig) {
250 1.15 pooka pthread_mutex_unlock(&spc->spc_mtx);
251 1.15 pooka pthread_sigmask(SIG_SETMASK, mask, NULL);
252 1.15 pooka pthread_sigmask(SIG_SETMASK, &fullset, NULL);
253 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
254 1.15 pooka }
255 1.12 pooka } else {
256 1.12 pooka spc->spc_istatus = SPCSTATUS_WANTED;
257 1.12 pooka pthread_cond_wait(&rw->rw_cv, &spc->spc_mtx);
258 1.12 pooka }
259 1.12 pooka }
260 1.12 pooka TAILQ_REMOVE(&spc->spc_respwait, rw, rw_entries);
261 1.12 pooka pthread_mutex_unlock(&spc->spc_mtx);
262 1.12 pooka pthread_cond_destroy(&rw->rw_cv);
263 1.12 pooka
264 1.18 pooka if (spc->spc_generation != mygen || !imalive) {
265 1.12 pooka return ENOTCONN;
266 1.18 pooka }
267 1.12 pooka return rw->rw_error;
268 1.12 pooka }
269 1.12 pooka
270 1.1 pooka static int
271 1.26 pooka syscall_req(struct spclient *spc, sigset_t *omask, int sysnum,
272 1.3 pooka const void *data, size_t dlen, void **resp)
273 1.1 pooka {
274 1.1 pooka struct rsp_hdr rhdr;
275 1.3 pooka struct respwait rw;
276 1.3 pooka int rv;
277 1.1 pooka
278 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
279 1.3 pooka rhdr.rsp_class = RUMPSP_REQ;
280 1.3 pooka rhdr.rsp_type = RUMPSP_SYSCALL;
281 1.1 pooka rhdr.rsp_sysnum = sysnum;
282 1.1 pooka
283 1.6 pooka do {
284 1.6 pooka putwait(spc, &rw, &rhdr);
285 1.18 pooka if ((rv = send_with_recon(spc, &rhdr, sizeof(rhdr))) != 0) {
286 1.18 pooka unputwait(spc, &rw);
287 1.18 pooka continue;
288 1.18 pooka }
289 1.18 pooka if ((rv = send_with_recon(spc, data, dlen)) != 0) {
290 1.6 pooka unputwait(spc, &rw);
291 1.18 pooka continue;
292 1.6 pooka }
293 1.6 pooka
294 1.26 pooka rv = cliwaitresp(spc, &rw, omask, false);
295 1.20 pooka if (rv == ENOTCONN)
296 1.20 pooka rv = EAGAIN;
297 1.20 pooka } while (rv == EAGAIN);
298 1.3 pooka
299 1.3 pooka *resp = rw.rw_data;
300 1.3 pooka return rv;
301 1.1 pooka }
302 1.1 pooka
303 1.1 pooka static int
304 1.18 pooka handshake_req(struct spclient *spc, uint32_t *auth, int cancel, bool haslock)
305 1.10 pooka {
306 1.11 pooka struct handshake_fork rf;
307 1.10 pooka struct rsp_hdr rhdr;
308 1.10 pooka struct respwait rw;
309 1.12 pooka sigset_t omask;
310 1.21 pooka size_t bonus;
311 1.10 pooka int rv;
312 1.10 pooka
313 1.21 pooka if (auth) {
314 1.21 pooka bonus = sizeof(rf);
315 1.21 pooka } else {
316 1.21 pooka bonus = strlen(getprogname())+1;
317 1.21 pooka }
318 1.21 pooka
319 1.10 pooka /* performs server handshake */
320 1.21 pooka rhdr.rsp_len = sizeof(rhdr) + bonus;
321 1.10 pooka rhdr.rsp_class = RUMPSP_REQ;
322 1.10 pooka rhdr.rsp_type = RUMPSP_HANDSHAKE;
323 1.11 pooka if (auth)
324 1.11 pooka rhdr.rsp_handshake = HANDSHAKE_FORK;
325 1.11 pooka else
326 1.11 pooka rhdr.rsp_handshake = HANDSHAKE_GUEST;
327 1.10 pooka
328 1.12 pooka pthread_sigmask(SIG_SETMASK, &fullset, &omask);
329 1.18 pooka if (haslock)
330 1.18 pooka putwait_locked(spc, &rw, &rhdr);
331 1.18 pooka else
332 1.18 pooka putwait(spc, &rw, &rhdr);
333 1.10 pooka rv = dosend(spc, &rhdr, sizeof(rhdr));
334 1.11 pooka if (auth) {
335 1.11 pooka memcpy(rf.rf_auth, auth, AUTHLEN*sizeof(*auth));
336 1.11 pooka rf.rf_cancel = cancel;
337 1.18 pooka rv = send_with_recon(spc, &rf, sizeof(rf));
338 1.21 pooka } else {
339 1.21 pooka rv = dosend(spc, getprogname(), strlen(getprogname())+1);
340 1.11 pooka }
341 1.18 pooka if (rv || cancel) {
342 1.18 pooka if (haslock)
343 1.18 pooka unputwait_locked(spc, &rw);
344 1.18 pooka else
345 1.18 pooka unputwait(spc, &rw);
346 1.18 pooka if (cancel) {
347 1.26 pooka goto out;
348 1.18 pooka }
349 1.18 pooka } else {
350 1.18 pooka rv = cliwaitresp(spc, &rw, &omask, haslock);
351 1.10 pooka }
352 1.10 pooka if (rv)
353 1.26 pooka goto out;
354 1.10 pooka
355 1.10 pooka rv = *(int *)rw.rw_data;
356 1.10 pooka free(rw.rw_data);
357 1.10 pooka
358 1.26 pooka out:
359 1.26 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
360 1.10 pooka return rv;
361 1.10 pooka }
362 1.10 pooka
363 1.10 pooka static int
364 1.26 pooka prefork_req(struct spclient *spc, sigset_t *omask, void **resp)
365 1.11 pooka {
366 1.11 pooka struct rsp_hdr rhdr;
367 1.11 pooka struct respwait rw;
368 1.11 pooka int rv;
369 1.11 pooka
370 1.11 pooka rhdr.rsp_len = sizeof(rhdr);
371 1.11 pooka rhdr.rsp_class = RUMPSP_REQ;
372 1.11 pooka rhdr.rsp_type = RUMPSP_PREFORK;
373 1.11 pooka rhdr.rsp_error = 0;
374 1.11 pooka
375 1.18 pooka do {
376 1.18 pooka putwait(spc, &rw, &rhdr);
377 1.18 pooka rv = send_with_recon(spc, &rhdr, sizeof(rhdr));
378 1.18 pooka if (rv != 0) {
379 1.18 pooka unputwait(spc, &rw);
380 1.18 pooka continue;
381 1.18 pooka }
382 1.11 pooka
383 1.26 pooka rv = cliwaitresp(spc, &rw, omask, false);
384 1.20 pooka if (rv == ENOTCONN)
385 1.20 pooka rv = EAGAIN;
386 1.20 pooka } while (rv == EAGAIN);
387 1.18 pooka
388 1.11 pooka *resp = rw.rw_data;
389 1.11 pooka return rv;
390 1.11 pooka }
391 1.11 pooka
392 1.18 pooka /*
393 1.18 pooka * prevent response code from deadlocking with reconnect code
394 1.18 pooka */
395 1.11 pooka static int
396 1.18 pooka resp_sendlock(struct spclient *spc)
397 1.18 pooka {
398 1.18 pooka int rv = 0;
399 1.18 pooka
400 1.18 pooka pthread_mutex_lock(&spc->spc_mtx);
401 1.18 pooka while (spc->spc_ostatus != SPCSTATUS_FREE) {
402 1.18 pooka if (__predict_false(spc->spc_reconnecting)) {
403 1.18 pooka rv = EBUSY;
404 1.18 pooka goto out;
405 1.18 pooka }
406 1.18 pooka spc->spc_ostatus = SPCSTATUS_WANTED;
407 1.18 pooka pthread_cond_wait(&spc->spc_cv, &spc->spc_mtx);
408 1.18 pooka }
409 1.18 pooka spc->spc_ostatus = SPCSTATUS_BUSY;
410 1.18 pooka
411 1.18 pooka out:
412 1.18 pooka pthread_mutex_unlock(&spc->spc_mtx);
413 1.18 pooka return rv;
414 1.18 pooka }
415 1.18 pooka
416 1.18 pooka static void
417 1.5 pooka send_copyin_resp(struct spclient *spc, uint64_t reqno, void *data, size_t dlen,
418 1.5 pooka int wantstr)
419 1.1 pooka {
420 1.1 pooka struct rsp_hdr rhdr;
421 1.1 pooka
422 1.5 pooka if (wantstr)
423 1.5 pooka dlen = MIN(dlen, strlen(data)+1);
424 1.5 pooka
425 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
426 1.1 pooka rhdr.rsp_reqno = reqno;
427 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
428 1.3 pooka rhdr.rsp_type = RUMPSP_COPYIN;
429 1.1 pooka rhdr.rsp_sysnum = 0;
430 1.1 pooka
431 1.18 pooka if (resp_sendlock(spc) != 0)
432 1.18 pooka return;
433 1.18 pooka (void)dosend(spc, &rhdr, sizeof(rhdr));
434 1.18 pooka (void)dosend(spc, data, dlen);
435 1.3 pooka sendunlock(spc);
436 1.1 pooka }
437 1.1 pooka
438 1.18 pooka static void
439 1.1 pooka send_anonmmap_resp(struct spclient *spc, uint64_t reqno, void *addr)
440 1.1 pooka {
441 1.1 pooka struct rsp_hdr rhdr;
442 1.1 pooka
443 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + sizeof(addr);
444 1.1 pooka rhdr.rsp_reqno = reqno;
445 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
446 1.3 pooka rhdr.rsp_type = RUMPSP_ANONMMAP;
447 1.1 pooka rhdr.rsp_sysnum = 0;
448 1.1 pooka
449 1.18 pooka if (resp_sendlock(spc) != 0)
450 1.18 pooka return;
451 1.18 pooka (void)dosend(spc, &rhdr, sizeof(rhdr));
452 1.18 pooka (void)dosend(spc, &addr, sizeof(addr));
453 1.3 pooka sendunlock(spc);
454 1.1 pooka }
455 1.1 pooka
456 1.1 pooka int
457 1.1 pooka rumpclient_syscall(int sysnum, const void *data, size_t dlen,
458 1.1 pooka register_t *retval)
459 1.1 pooka {
460 1.1 pooka struct rsp_sysresp *resp;
461 1.26 pooka sigset_t omask;
462 1.3 pooka void *rdata;
463 1.3 pooka int rv;
464 1.3 pooka
465 1.26 pooka pthread_sigmask(SIG_SETMASK, &fullset, &omask);
466 1.26 pooka
467 1.3 pooka DPRINTF(("rumpsp syscall_req: syscall %d with %p/%zu\n",
468 1.3 pooka sysnum, data, dlen));
469 1.3 pooka
470 1.26 pooka rv = syscall_req(&clispc, &omask, sysnum, data, dlen, &rdata);
471 1.3 pooka if (rv)
472 1.26 pooka goto out;
473 1.3 pooka
474 1.3 pooka resp = rdata;
475 1.3 pooka DPRINTF(("rumpsp syscall_resp: syscall %d error %d, rv: %d/%d\n",
476 1.3 pooka sysnum, rv, resp->rsys_retval[0], resp->rsys_retval[1]));
477 1.1 pooka
478 1.3 pooka memcpy(retval, &resp->rsys_retval, sizeof(resp->rsys_retval));
479 1.3 pooka rv = resp->rsys_error;
480 1.3 pooka free(rdata);
481 1.1 pooka
482 1.26 pooka out:
483 1.26 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
484 1.3 pooka return rv;
485 1.3 pooka }
486 1.1 pooka
487 1.3 pooka static void
488 1.3 pooka handlereq(struct spclient *spc)
489 1.3 pooka {
490 1.3 pooka struct rsp_copydata *copydata;
491 1.16 pooka struct rsp_hdr *rhdr = &spc->spc_hdr;
492 1.3 pooka void *mapaddr;
493 1.3 pooka size_t maplen;
494 1.5 pooka int reqtype = spc->spc_hdr.rsp_type;
495 1.1 pooka
496 1.5 pooka switch (reqtype) {
497 1.3 pooka case RUMPSP_COPYIN:
498 1.5 pooka case RUMPSP_COPYINSTR:
499 1.3 pooka /*LINTED*/
500 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
501 1.3 pooka DPRINTF(("rump_sp handlereq: copyin request: %p/%zu\n",
502 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
503 1.3 pooka send_copyin_resp(spc, spc->spc_hdr.rsp_reqno,
504 1.5 pooka copydata->rcp_addr, copydata->rcp_len,
505 1.5 pooka reqtype == RUMPSP_COPYINSTR);
506 1.3 pooka break;
507 1.3 pooka case RUMPSP_COPYOUT:
508 1.5 pooka case RUMPSP_COPYOUTSTR:
509 1.3 pooka /*LINTED*/
510 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
511 1.3 pooka DPRINTF(("rump_sp handlereq: copyout request: %p/%zu\n",
512 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
513 1.3 pooka /*LINTED*/
514 1.3 pooka memcpy(copydata->rcp_addr, copydata->rcp_data,
515 1.3 pooka copydata->rcp_len);
516 1.3 pooka break;
517 1.3 pooka case RUMPSP_ANONMMAP:
518 1.3 pooka /*LINTED*/
519 1.3 pooka maplen = *(size_t *)spc->spc_buf;
520 1.3 pooka mapaddr = mmap(NULL, maplen, PROT_READ|PROT_WRITE,
521 1.3 pooka MAP_ANON, -1, 0);
522 1.3 pooka if (mapaddr == MAP_FAILED)
523 1.3 pooka mapaddr = NULL;
524 1.3 pooka DPRINTF(("rump_sp handlereq: anonmmap: %p\n", mapaddr));
525 1.3 pooka send_anonmmap_resp(spc, spc->spc_hdr.rsp_reqno, mapaddr);
526 1.3 pooka break;
527 1.16 pooka case RUMPSP_RAISE:
528 1.16 pooka DPRINTF(("rump_sp handlereq: raise sig %d\n", rhdr->rsp_signo));
529 1.18 pooka raise((int)rhdr->rsp_signo);
530 1.16 pooka /*
531 1.16 pooka * We most likely have signals blocked, but the signal
532 1.16 pooka * will be handled soon enough when we return.
533 1.16 pooka */
534 1.16 pooka break;
535 1.3 pooka default:
536 1.12 pooka printf("PANIC: INVALID TYPE %d\n", reqtype);
537 1.3 pooka abort();
538 1.3 pooka break;
539 1.1 pooka }
540 1.1 pooka
541 1.6 pooka spcfreebuf(spc);
542 1.1 pooka }
543 1.1 pooka
544 1.11 pooka static unsigned ptab_idx;
545 1.11 pooka static struct sockaddr *serv_sa;
546 1.11 pooka
547 1.27 pooka /* dup until we get a "good" fd which does not collide with stdio */
548 1.27 pooka static int
549 1.27 pooka dupgood(int myfd)
550 1.27 pooka {
551 1.27 pooka int ofds[3];
552 1.27 pooka int i;
553 1.27 pooka
554 1.27 pooka for (i = 0; myfd <= 2 && myfd != -1; i++) {
555 1.27 pooka assert(i < __arraycount(ofds));
556 1.27 pooka ofds[i] = myfd;
557 1.27 pooka myfd = host_dup(myfd);
558 1.27 pooka }
559 1.27 pooka
560 1.27 pooka for (i--; i >= 0; i--) {
561 1.27 pooka host_close(ofds[i]);
562 1.27 pooka }
563 1.27 pooka
564 1.27 pooka return myfd;
565 1.27 pooka }
566 1.27 pooka
567 1.11 pooka static int
568 1.20 pooka doconnect(bool noisy)
569 1.1 pooka {
570 1.18 pooka struct respwait rw;
571 1.18 pooka struct rsp_hdr rhdr;
572 1.15 pooka struct kevent kev[NSIG+1];
573 1.9 pooka char banner[MAXBANNER];
574 1.18 pooka struct pollfd pfd;
575 1.15 pooka int s, error, flags, i;
576 1.9 pooka ssize_t n;
577 1.1 pooka
578 1.18 pooka if (kq != -1)
579 1.18 pooka host_close(kq);
580 1.18 pooka kq = -1;
581 1.19 pooka s = -1;
582 1.18 pooka
583 1.18 pooka if (clispc.spc_fd != -1)
584 1.18 pooka host_close(clispc.spc_fd);
585 1.18 pooka clispc.spc_fd = -1;
586 1.18 pooka
587 1.18 pooka /*
588 1.18 pooka * for reconnect, gate everyone out of the receiver code
589 1.18 pooka */
590 1.18 pooka putwait_locked(&clispc, &rw, &rhdr);
591 1.18 pooka
592 1.18 pooka pthread_mutex_lock(&clispc.spc_mtx);
593 1.18 pooka clispc.spc_reconnecting = 1;
594 1.18 pooka pthread_cond_broadcast(&clispc.spc_cv);
595 1.18 pooka clispc.spc_generation++;
596 1.18 pooka while (clispc.spc_istatus != SPCSTATUS_FREE) {
597 1.18 pooka clispc.spc_istatus = SPCSTATUS_WANTED;
598 1.18 pooka pthread_cond_wait(&rw.rw_cv, &clispc.spc_mtx);
599 1.18 pooka }
600 1.18 pooka kickall(&clispc);
601 1.18 pooka
602 1.18 pooka /*
603 1.18 pooka * we can release it already since we hold the
604 1.18 pooka * send lock during reconnect
605 1.18 pooka * XXX: assert it
606 1.18 pooka */
607 1.18 pooka clispc.spc_istatus = SPCSTATUS_FREE;
608 1.18 pooka pthread_mutex_unlock(&clispc.spc_mtx);
609 1.18 pooka unputwait_locked(&clispc, &rw);
610 1.18 pooka
611 1.18 pooka free(clispc.spc_buf);
612 1.18 pooka clispc.spc_off = 0;
613 1.18 pooka
614 1.27 pooka s = dupgood(host_socket(parsetab[ptab_idx].domain, SOCK_STREAM, 0));
615 1.11 pooka if (s == -1)
616 1.2 pooka return -1;
617 1.1 pooka
618 1.18 pooka pfd.fd = s;
619 1.18 pooka pfd.events = POLLIN;
620 1.18 pooka while (host_connect(s, serv_sa, (socklen_t)serv_sa->sa_len) == -1) {
621 1.18 pooka if (errno == EINTR)
622 1.18 pooka continue;
623 1.20 pooka error = errno;
624 1.20 pooka if (noisy)
625 1.18 pooka fprintf(stderr, "rump_sp: client connect failed: %s\n",
626 1.18 pooka strerror(errno));
627 1.20 pooka errno = error;
628 1.20 pooka return -1;
629 1.2 pooka }
630 1.1 pooka
631 1.11 pooka if ((error = parsetab[ptab_idx].connhook(s)) != 0) {
632 1.2 pooka error = errno;
633 1.20 pooka if (noisy)
634 1.20 pooka fprintf(stderr, "rump_sp: connect hook failed\n");
635 1.2 pooka errno = error;
636 1.2 pooka return -1;
637 1.1 pooka }
638 1.4 pooka
639 1.13 pooka if ((n = host_read(s, banner, sizeof(banner)-1)) < 0) {
640 1.2 pooka error = errno;
641 1.20 pooka if (noisy)
642 1.20 pooka fprintf(stderr, "rump_sp: failed to read banner\n");
643 1.2 pooka errno = error;
644 1.2 pooka return -1;
645 1.1 pooka }
646 1.9 pooka
647 1.9 pooka if (banner[n-1] != '\n') {
648 1.20 pooka if (noisy)
649 1.20 pooka fprintf(stderr, "rump_sp: invalid banner\n");
650 1.9 pooka errno = EINVAL;
651 1.9 pooka return -1;
652 1.9 pooka }
653 1.9 pooka banner[n] = '\0';
654 1.18 pooka /* parse the banner some day */
655 1.9 pooka
656 1.15 pooka flags = host_fcntl(s, F_GETFL, 0);
657 1.15 pooka if (host_fcntl(s, F_SETFL, flags | O_NONBLOCK) == -1) {
658 1.20 pooka if (noisy)
659 1.20 pooka fprintf(stderr, "rump_sp: socket fd NONBLOCK: %s\n",
660 1.20 pooka strerror(errno));
661 1.15 pooka errno = EINVAL;
662 1.15 pooka return -1;
663 1.15 pooka }
664 1.18 pooka clispc.spc_fd = s;
665 1.18 pooka clispc.spc_state = SPCSTATE_RUNNING;
666 1.18 pooka clispc.spc_reconnecting = 0;
667 1.15 pooka
668 1.15 pooka /* setup kqueue, we want all signals and the fd */
669 1.27 pooka if ((kq = dupgood(host_kqueue())) == -1) {
670 1.15 pooka error = errno;
671 1.20 pooka if (noisy)
672 1.20 pooka fprintf(stderr, "rump_sp: cannot setup kqueue");
673 1.15 pooka errno = error;
674 1.15 pooka return -1;
675 1.15 pooka }
676 1.15 pooka
677 1.15 pooka for (i = 0; i < NSIG; i++) {
678 1.15 pooka EV_SET(&kev[i], i+1, EVFILT_SIGNAL, EV_ADD|EV_ENABLE, 0, 0, 0);
679 1.15 pooka }
680 1.18 pooka EV_SET(&kev[NSIG], clispc.spc_fd,
681 1.18 pooka EVFILT_READ, EV_ADD|EV_ENABLE, 0, 0, 0);
682 1.17 pooka if (host_kevent(kq, kev, NSIG+1, NULL, 0, NULL) == -1) {
683 1.15 pooka error = errno;
684 1.20 pooka if (noisy)
685 1.20 pooka fprintf(stderr, "rump_sp: kevent() failed");
686 1.15 pooka errno = error;
687 1.15 pooka return -1;
688 1.15 pooka }
689 1.15 pooka
690 1.18 pooka return 0;
691 1.18 pooka }
692 1.18 pooka
693 1.18 pooka static int
694 1.18 pooka doinit(void)
695 1.18 pooka {
696 1.18 pooka
697 1.11 pooka TAILQ_INIT(&clispc.spc_respwait);
698 1.11 pooka pthread_mutex_init(&clispc.spc_mtx, NULL);
699 1.11 pooka pthread_cond_init(&clispc.spc_cv, NULL);
700 1.11 pooka
701 1.11 pooka return 0;
702 1.11 pooka }
703 1.11 pooka
704 1.13 pooka void *(*rumpclient_dlsym)(void *, const char *);
705 1.13 pooka
706 1.11 pooka int
707 1.11 pooka rumpclient_init()
708 1.11 pooka {
709 1.11 pooka char *p;
710 1.11 pooka int error;
711 1.11 pooka
712 1.25 pooka sigfillset(&fullset);
713 1.25 pooka
714 1.13 pooka /* dlsym overrided by rumphijack? */
715 1.13 pooka if (!rumpclient_dlsym)
716 1.13 pooka rumpclient_dlsym = dlsym;
717 1.13 pooka
718 1.13 pooka /*
719 1.13 pooka * sag mir, wo die symbol sind. zogen fort, der krieg beginnt.
720 1.13 pooka * wann wird man je verstehen? wann wird man je verstehen?
721 1.13 pooka */
722 1.13 pooka #define FINDSYM2(_name_,_syscall_) \
723 1.13 pooka if ((host_##_name_ = rumpclient_dlsym(RTLD_NEXT, \
724 1.13 pooka #_syscall_)) == NULL) \
725 1.13 pooka /* host_##_name_ = _syscall_ */;
726 1.13 pooka #define FINDSYM(_name_) FINDSYM2(_name_,_name_)
727 1.13 pooka FINDSYM2(socket,__socket30);
728 1.13 pooka FINDSYM(close);
729 1.13 pooka FINDSYM(connect);
730 1.15 pooka FINDSYM(fcntl);
731 1.13 pooka FINDSYM(poll);
732 1.13 pooka FINDSYM(read);
733 1.13 pooka FINDSYM(sendto);
734 1.13 pooka FINDSYM(setsockopt);
735 1.27 pooka FINDSYM(dup);
736 1.17 pooka FINDSYM(kqueue);
737 1.22 pooka #if !__NetBSD_Prereq__(5,99,7)
738 1.17 pooka FINDSYM(kevent);
739 1.22 pooka #else
740 1.22 pooka FINDSYM2(kevent,_sys___kevent50);
741 1.22 pooka #endif
742 1.13 pooka #undef FINDSYM
743 1.13 pooka #undef FINDSY2
744 1.13 pooka
745 1.11 pooka if ((p = getenv("RUMP_SERVER")) == NULL) {
746 1.11 pooka errno = ENOENT;
747 1.11 pooka return -1;
748 1.11 pooka }
749 1.11 pooka
750 1.11 pooka if ((error = parseurl(p, &serv_sa, &ptab_idx, 0)) != 0) {
751 1.11 pooka errno = error;
752 1.11 pooka return -1;
753 1.11 pooka }
754 1.11 pooka
755 1.18 pooka if (doinit() == -1)
756 1.18 pooka return -1;
757 1.20 pooka if (doconnect(true) == -1)
758 1.11 pooka return -1;
759 1.11 pooka
760 1.18 pooka error = handshake_req(&clispc, NULL, 0, false);
761 1.11 pooka if (error) {
762 1.11 pooka pthread_mutex_destroy(&clispc.spc_mtx);
763 1.11 pooka pthread_cond_destroy(&clispc.spc_cv);
764 1.18 pooka if (clispc.spc_fd != -1)
765 1.18 pooka host_close(clispc.spc_fd);
766 1.10 pooka errno = error;
767 1.10 pooka return -1;
768 1.10 pooka }
769 1.10 pooka
770 1.11 pooka return 0;
771 1.11 pooka }
772 1.11 pooka
773 1.11 pooka struct rumpclient_fork {
774 1.11 pooka uint32_t fork_auth[AUTHLEN];
775 1.11 pooka };
776 1.11 pooka
777 1.11 pooka struct rumpclient_fork *
778 1.11 pooka rumpclient_prefork(void)
779 1.11 pooka {
780 1.11 pooka struct rumpclient_fork *rpf;
781 1.26 pooka sigset_t omask;
782 1.11 pooka void *resp;
783 1.11 pooka int rv;
784 1.11 pooka
785 1.26 pooka pthread_sigmask(SIG_SETMASK, &fullset, &omask);
786 1.11 pooka rpf = malloc(sizeof(*rpf));
787 1.11 pooka if (rpf == NULL)
788 1.11 pooka return NULL;
789 1.11 pooka
790 1.26 pooka if ((rv = prefork_req(&clispc, &omask, &resp)) != 0) {
791 1.11 pooka free(rpf);
792 1.11 pooka errno = rv;
793 1.26 pooka rpf = NULL;
794 1.26 pooka goto out;
795 1.11 pooka }
796 1.11 pooka
797 1.11 pooka memcpy(rpf->fork_auth, resp, sizeof(rpf->fork_auth));
798 1.11 pooka free(resp);
799 1.11 pooka
800 1.26 pooka out:
801 1.26 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
802 1.11 pooka return rpf;
803 1.11 pooka }
804 1.11 pooka
805 1.11 pooka int
806 1.11 pooka rumpclient_fork_init(struct rumpclient_fork *rpf)
807 1.11 pooka {
808 1.11 pooka int error;
809 1.23 pooka int osock;
810 1.11 pooka
811 1.23 pooka osock = clispc.spc_fd;
812 1.11 pooka memset(&clispc, 0, sizeof(clispc));
813 1.23 pooka clispc.spc_fd = osock;
814 1.23 pooka
815 1.23 pooka kq = -1; /* kqueue descriptor is not copied over fork() */
816 1.11 pooka
817 1.18 pooka if (doinit() == -1)
818 1.18 pooka return -1;
819 1.20 pooka if (doconnect(false) == -1)
820 1.11 pooka return -1;
821 1.10 pooka
822 1.18 pooka error = handshake_req(&clispc, rpf->fork_auth, 0, false);
823 1.10 pooka if (error) {
824 1.10 pooka pthread_mutex_destroy(&clispc.spc_mtx);
825 1.10 pooka pthread_cond_destroy(&clispc.spc_cv);
826 1.11 pooka errno = error;
827 1.11 pooka return -1;
828 1.10 pooka }
829 1.11 pooka
830 1.11 pooka return 0;
831 1.1 pooka }
832 1.20 pooka
833 1.20 pooka void
834 1.20 pooka rumpclient_setconnretry(time_t timeout)
835 1.20 pooka {
836 1.20 pooka
837 1.24 pooka if (timeout < RUMPCLIENT_RETRYCONN_DIE)
838 1.20 pooka return; /* gigo */
839 1.20 pooka
840 1.20 pooka retrytimo = timeout;
841 1.20 pooka }
842