rumpclient.c revision 1.21 1 1.21 pooka /* $NetBSD: rumpclient.c,v 1.21 2011/01/28 19:21:28 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.13 pooka
72 1.17 pooka int (*host_kqueue)(void);
73 1.17 pooka int (*host_kevent)(int, const struct kevent *, size_t,
74 1.17 pooka struct kevent *, size_t, const struct timespec *);
75 1.17 pooka
76 1.1 pooka #include "sp_common.c"
77 1.1 pooka
78 1.11 pooka static struct spclient clispc = {
79 1.11 pooka .spc_fd = -1,
80 1.11 pooka };
81 1.1 pooka
82 1.18 pooka static int kq = -1;
83 1.15 pooka static sigset_t fullset;
84 1.12 pooka
85 1.20 pooka static int doconnect(bool);
86 1.18 pooka static int handshake_req(struct spclient *, uint32_t *, int, bool);
87 1.18 pooka
88 1.20 pooka time_t retrytimo = RUMPCLIENT_RETRYCONN_ONCE;
89 1.18 pooka
90 1.18 pooka static int
91 1.18 pooka send_with_recon(struct spclient *spc, const void *data, size_t dlen)
92 1.18 pooka {
93 1.20 pooka struct timeval starttime, curtime;
94 1.20 pooka time_t prevreconmsg;
95 1.20 pooka unsigned reconretries;
96 1.18 pooka int rv;
97 1.18 pooka
98 1.20 pooka for (prevreconmsg = 0, reconretries = 0;;) {
99 1.18 pooka rv = dosend(spc, data, dlen);
100 1.18 pooka if (__predict_false(rv == ENOTCONN || rv == EBADF)) {
101 1.20 pooka /* no persistent connections */
102 1.20 pooka if (retrytimo == 0)
103 1.20 pooka break;
104 1.20 pooka
105 1.20 pooka if (!prevreconmsg) {
106 1.20 pooka prevreconmsg = time(NULL);
107 1.20 pooka gettimeofday(&starttime, NULL);
108 1.20 pooka }
109 1.20 pooka if (reconretries == 1) {
110 1.20 pooka if (retrytimo == RUMPCLIENT_RETRYCONN_ONCE) {
111 1.20 pooka rv = ENOTCONN;
112 1.20 pooka break;
113 1.20 pooka }
114 1.20 pooka fprintf(stderr, "rump_sp: connection to "
115 1.20 pooka "kernel lost, trying to reconnect ...\n");
116 1.20 pooka } else if (time(NULL) - prevreconmsg > 120) {
117 1.20 pooka fprintf(stderr, "rump_sp: still trying to "
118 1.20 pooka "reconnect ...\n");
119 1.20 pooka prevreconmsg = time(NULL);
120 1.20 pooka }
121 1.20 pooka
122 1.20 pooka /* check that we aren't over the limit */
123 1.20 pooka if (retrytimo > 0) {
124 1.20 pooka struct timeval tmp;
125 1.20 pooka
126 1.20 pooka gettimeofday(&curtime, NULL);
127 1.20 pooka timersub(&curtime, &starttime, &tmp);
128 1.20 pooka if (tmp.tv_sec >= retrytimo) {
129 1.20 pooka fprintf(stderr, "rump_sp: reconnect "
130 1.20 pooka "failed, %lld second timeout\n",
131 1.20 pooka (long long)retrytimo);
132 1.20 pooka return ENOTCONN;
133 1.20 pooka }
134 1.20 pooka }
135 1.20 pooka
136 1.20 pooka /* adhoc backoff timer */
137 1.20 pooka if (reconretries < 10) {
138 1.20 pooka usleep(100000 * reconretries);
139 1.20 pooka } else {
140 1.20 pooka sleep(MIN(10, reconretries-9));
141 1.20 pooka }
142 1.20 pooka reconretries++;
143 1.20 pooka
144 1.20 pooka if ((rv = doconnect(false)) != 0)
145 1.18 pooka continue;
146 1.18 pooka if ((rv = handshake_req(&clispc, NULL, 0, true)) != 0)
147 1.18 pooka continue;
148 1.20 pooka
149 1.20 pooka /*
150 1.20 pooka * ok, reconnect succesful. we need to return to
151 1.20 pooka * the upper layer to get the entire PDU resent.
152 1.20 pooka */
153 1.20 pooka if (reconretries != 1)
154 1.20 pooka fprintf(stderr, "rump_sp: reconnected!\n");
155 1.20 pooka rv = EAGAIN;
156 1.20 pooka break;
157 1.20 pooka } else {
158 1.20 pooka _DIAGASSERT(errno != EAGAIN);
159 1.18 pooka break;
160 1.18 pooka }
161 1.20 pooka }
162 1.18 pooka
163 1.18 pooka return rv;
164 1.18 pooka }
165 1.18 pooka
166 1.12 pooka static int
167 1.18 pooka cliwaitresp(struct spclient *spc, struct respwait *rw, sigset_t *mask,
168 1.18 pooka bool keeplock)
169 1.12 pooka {
170 1.18 pooka uint64_t mygen;
171 1.18 pooka bool imalive = true;
172 1.12 pooka
173 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
174 1.18 pooka if (!keeplock)
175 1.18 pooka sendunlockl(spc);
176 1.18 pooka mygen = spc->spc_generation;
177 1.12 pooka
178 1.12 pooka rw->rw_error = 0;
179 1.18 pooka while (!rw->rw_done && rw->rw_error == 0) {
180 1.18 pooka if (__predict_false(spc->spc_generation != mygen || !imalive))
181 1.18 pooka break;
182 1.18 pooka
183 1.12 pooka /* are we free to receive? */
184 1.12 pooka if (spc->spc_istatus == SPCSTATUS_FREE) {
185 1.15 pooka struct kevent kev[8];
186 1.15 pooka int gotresp, dosig, rv, i;
187 1.15 pooka
188 1.12 pooka spc->spc_istatus = SPCSTATUS_BUSY;
189 1.12 pooka pthread_mutex_unlock(&spc->spc_mtx);
190 1.12 pooka
191 1.15 pooka dosig = 0;
192 1.15 pooka for (gotresp = 0; !gotresp; ) {
193 1.15 pooka switch (readframe(spc)) {
194 1.15 pooka case 0:
195 1.17 pooka rv = host_kevent(kq, NULL, 0,
196 1.15 pooka kev, __arraycount(kev), NULL);
197 1.18 pooka
198 1.18 pooka /*
199 1.18 pooka * XXX: don't know how this can
200 1.18 pooka * happen (timeout cannot expire
201 1.18 pooka * since there isn't one), but
202 1.18 pooka * it does happen
203 1.18 pooka */
204 1.18 pooka if (__predict_false(rv == 0))
205 1.18 pooka continue;
206 1.18 pooka
207 1.15 pooka for (i = 0; i < rv; i++) {
208 1.15 pooka if (kev[i].filter
209 1.15 pooka == EVFILT_SIGNAL)
210 1.15 pooka dosig++;
211 1.15 pooka }
212 1.15 pooka if (dosig)
213 1.15 pooka goto cleanup;
214 1.15 pooka
215 1.15 pooka continue;
216 1.15 pooka case -1:
217 1.18 pooka imalive = false;
218 1.15 pooka goto cleanup;
219 1.15 pooka default:
220 1.15 pooka break;
221 1.15 pooka }
222 1.12 pooka
223 1.15 pooka switch (spc->spc_hdr.rsp_class) {
224 1.12 pooka case RUMPSP_RESP:
225 1.12 pooka case RUMPSP_ERROR:
226 1.12 pooka kickwaiter(spc);
227 1.15 pooka gotresp = spc->spc_hdr.rsp_reqno ==
228 1.15 pooka rw->rw_reqno;
229 1.12 pooka break;
230 1.12 pooka case RUMPSP_REQ:
231 1.12 pooka handlereq(spc);
232 1.12 pooka break;
233 1.12 pooka default:
234 1.12 pooka /* panic */
235 1.12 pooka break;
236 1.15 pooka }
237 1.12 pooka }
238 1.12 pooka
239 1.15 pooka cleanup:
240 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
241 1.15 pooka if (spc->spc_istatus == SPCSTATUS_WANTED)
242 1.15 pooka kickall(spc);
243 1.15 pooka spc->spc_istatus = SPCSTATUS_FREE;
244 1.15 pooka
245 1.15 pooka /* take one for the team */
246 1.15 pooka if (dosig) {
247 1.15 pooka pthread_mutex_unlock(&spc->spc_mtx);
248 1.15 pooka pthread_sigmask(SIG_SETMASK, mask, NULL);
249 1.15 pooka pthread_sigmask(SIG_SETMASK, &fullset, NULL);
250 1.15 pooka pthread_mutex_lock(&spc->spc_mtx);
251 1.15 pooka }
252 1.12 pooka } else {
253 1.12 pooka spc->spc_istatus = SPCSTATUS_WANTED;
254 1.12 pooka pthread_cond_wait(&rw->rw_cv, &spc->spc_mtx);
255 1.12 pooka }
256 1.12 pooka }
257 1.12 pooka TAILQ_REMOVE(&spc->spc_respwait, rw, rw_entries);
258 1.12 pooka pthread_mutex_unlock(&spc->spc_mtx);
259 1.12 pooka pthread_cond_destroy(&rw->rw_cv);
260 1.12 pooka
261 1.18 pooka if (spc->spc_generation != mygen || !imalive) {
262 1.12 pooka return ENOTCONN;
263 1.18 pooka }
264 1.12 pooka return rw->rw_error;
265 1.12 pooka }
266 1.12 pooka
267 1.1 pooka static int
268 1.3 pooka syscall_req(struct spclient *spc, int sysnum,
269 1.3 pooka const void *data, size_t dlen, void **resp)
270 1.1 pooka {
271 1.1 pooka struct rsp_hdr rhdr;
272 1.3 pooka struct respwait rw;
273 1.12 pooka sigset_t omask;
274 1.3 pooka int rv;
275 1.1 pooka
276 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
277 1.3 pooka rhdr.rsp_class = RUMPSP_REQ;
278 1.3 pooka rhdr.rsp_type = RUMPSP_SYSCALL;
279 1.1 pooka rhdr.rsp_sysnum = sysnum;
280 1.1 pooka
281 1.12 pooka pthread_sigmask(SIG_SETMASK, &fullset, &omask);
282 1.6 pooka do {
283 1.6 pooka putwait(spc, &rw, &rhdr);
284 1.18 pooka if ((rv = send_with_recon(spc, &rhdr, sizeof(rhdr))) != 0) {
285 1.18 pooka unputwait(spc, &rw);
286 1.18 pooka continue;
287 1.18 pooka }
288 1.18 pooka if ((rv = send_with_recon(spc, data, dlen)) != 0) {
289 1.6 pooka unputwait(spc, &rw);
290 1.18 pooka continue;
291 1.6 pooka }
292 1.6 pooka
293 1.18 pooka rv = cliwaitresp(spc, &rw, &omask, false);
294 1.20 pooka if (rv == ENOTCONN)
295 1.20 pooka rv = EAGAIN;
296 1.20 pooka } while (rv == EAGAIN);
297 1.12 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
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.18 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
348 1.18 pooka return rv;
349 1.18 pooka }
350 1.18 pooka } else {
351 1.18 pooka rv = cliwaitresp(spc, &rw, &omask, haslock);
352 1.10 pooka }
353 1.12 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
354 1.10 pooka if (rv)
355 1.10 pooka return rv;
356 1.10 pooka
357 1.10 pooka rv = *(int *)rw.rw_data;
358 1.10 pooka free(rw.rw_data);
359 1.10 pooka
360 1.10 pooka return rv;
361 1.10 pooka }
362 1.10 pooka
363 1.10 pooka static int
364 1.11 pooka prefork_req(struct spclient *spc, void **resp)
365 1.11 pooka {
366 1.11 pooka struct rsp_hdr rhdr;
367 1.11 pooka struct respwait rw;
368 1.12 pooka sigset_t omask;
369 1.11 pooka int rv;
370 1.11 pooka
371 1.11 pooka rhdr.rsp_len = sizeof(rhdr);
372 1.11 pooka rhdr.rsp_class = RUMPSP_REQ;
373 1.11 pooka rhdr.rsp_type = RUMPSP_PREFORK;
374 1.11 pooka rhdr.rsp_error = 0;
375 1.11 pooka
376 1.12 pooka pthread_sigmask(SIG_SETMASK, &fullset, &omask);
377 1.18 pooka do {
378 1.18 pooka putwait(spc, &rw, &rhdr);
379 1.18 pooka rv = send_with_recon(spc, &rhdr, sizeof(rhdr));
380 1.18 pooka if (rv != 0) {
381 1.18 pooka unputwait(spc, &rw);
382 1.18 pooka continue;
383 1.18 pooka }
384 1.11 pooka
385 1.18 pooka rv = cliwaitresp(spc, &rw, &omask, false);
386 1.20 pooka if (rv == ENOTCONN)
387 1.20 pooka rv = EAGAIN;
388 1.20 pooka } while (rv == EAGAIN);
389 1.12 pooka pthread_sigmask(SIG_SETMASK, &omask, NULL);
390 1.18 pooka
391 1.11 pooka *resp = rw.rw_data;
392 1.11 pooka return rv;
393 1.11 pooka }
394 1.11 pooka
395 1.18 pooka /*
396 1.18 pooka * prevent response code from deadlocking with reconnect code
397 1.18 pooka */
398 1.11 pooka static int
399 1.18 pooka resp_sendlock(struct spclient *spc)
400 1.18 pooka {
401 1.18 pooka int rv = 0;
402 1.18 pooka
403 1.18 pooka pthread_mutex_lock(&spc->spc_mtx);
404 1.18 pooka while (spc->spc_ostatus != SPCSTATUS_FREE) {
405 1.18 pooka if (__predict_false(spc->spc_reconnecting)) {
406 1.18 pooka rv = EBUSY;
407 1.18 pooka goto out;
408 1.18 pooka }
409 1.18 pooka spc->spc_ostatus = SPCSTATUS_WANTED;
410 1.18 pooka pthread_cond_wait(&spc->spc_cv, &spc->spc_mtx);
411 1.18 pooka }
412 1.18 pooka spc->spc_ostatus = SPCSTATUS_BUSY;
413 1.18 pooka
414 1.18 pooka out:
415 1.18 pooka pthread_mutex_unlock(&spc->spc_mtx);
416 1.18 pooka return rv;
417 1.18 pooka }
418 1.18 pooka
419 1.18 pooka static void
420 1.5 pooka send_copyin_resp(struct spclient *spc, uint64_t reqno, void *data, size_t dlen,
421 1.5 pooka int wantstr)
422 1.1 pooka {
423 1.1 pooka struct rsp_hdr rhdr;
424 1.1 pooka
425 1.5 pooka if (wantstr)
426 1.5 pooka dlen = MIN(dlen, strlen(data)+1);
427 1.5 pooka
428 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
429 1.1 pooka rhdr.rsp_reqno = reqno;
430 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
431 1.3 pooka rhdr.rsp_type = RUMPSP_COPYIN;
432 1.1 pooka rhdr.rsp_sysnum = 0;
433 1.1 pooka
434 1.18 pooka if (resp_sendlock(spc) != 0)
435 1.18 pooka return;
436 1.18 pooka (void)dosend(spc, &rhdr, sizeof(rhdr));
437 1.18 pooka (void)dosend(spc, data, dlen);
438 1.3 pooka sendunlock(spc);
439 1.1 pooka }
440 1.1 pooka
441 1.18 pooka static void
442 1.1 pooka send_anonmmap_resp(struct spclient *spc, uint64_t reqno, void *addr)
443 1.1 pooka {
444 1.1 pooka struct rsp_hdr rhdr;
445 1.1 pooka
446 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + sizeof(addr);
447 1.1 pooka rhdr.rsp_reqno = reqno;
448 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
449 1.3 pooka rhdr.rsp_type = RUMPSP_ANONMMAP;
450 1.1 pooka rhdr.rsp_sysnum = 0;
451 1.1 pooka
452 1.18 pooka if (resp_sendlock(spc) != 0)
453 1.18 pooka return;
454 1.18 pooka (void)dosend(spc, &rhdr, sizeof(rhdr));
455 1.18 pooka (void)dosend(spc, &addr, sizeof(addr));
456 1.3 pooka sendunlock(spc);
457 1.1 pooka }
458 1.1 pooka
459 1.1 pooka int
460 1.1 pooka rumpclient_syscall(int sysnum, const void *data, size_t dlen,
461 1.1 pooka register_t *retval)
462 1.1 pooka {
463 1.1 pooka struct rsp_sysresp *resp;
464 1.3 pooka void *rdata;
465 1.3 pooka int rv;
466 1.3 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.3 pooka rv = syscall_req(&clispc, sysnum, data, dlen, &rdata);
471 1.3 pooka if (rv)
472 1.3 pooka return rv;
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.3 pooka return rv;
483 1.3 pooka }
484 1.1 pooka
485 1.3 pooka static void
486 1.3 pooka handlereq(struct spclient *spc)
487 1.3 pooka {
488 1.3 pooka struct rsp_copydata *copydata;
489 1.16 pooka struct rsp_hdr *rhdr = &spc->spc_hdr;
490 1.3 pooka void *mapaddr;
491 1.3 pooka size_t maplen;
492 1.5 pooka int reqtype = spc->spc_hdr.rsp_type;
493 1.1 pooka
494 1.5 pooka switch (reqtype) {
495 1.3 pooka case RUMPSP_COPYIN:
496 1.5 pooka case RUMPSP_COPYINSTR:
497 1.3 pooka /*LINTED*/
498 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
499 1.3 pooka DPRINTF(("rump_sp handlereq: copyin request: %p/%zu\n",
500 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
501 1.3 pooka send_copyin_resp(spc, spc->spc_hdr.rsp_reqno,
502 1.5 pooka copydata->rcp_addr, copydata->rcp_len,
503 1.5 pooka reqtype == RUMPSP_COPYINSTR);
504 1.3 pooka break;
505 1.3 pooka case RUMPSP_COPYOUT:
506 1.5 pooka case RUMPSP_COPYOUTSTR:
507 1.3 pooka /*LINTED*/
508 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
509 1.3 pooka DPRINTF(("rump_sp handlereq: copyout request: %p/%zu\n",
510 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
511 1.3 pooka /*LINTED*/
512 1.3 pooka memcpy(copydata->rcp_addr, copydata->rcp_data,
513 1.3 pooka copydata->rcp_len);
514 1.3 pooka break;
515 1.3 pooka case RUMPSP_ANONMMAP:
516 1.3 pooka /*LINTED*/
517 1.3 pooka maplen = *(size_t *)spc->spc_buf;
518 1.3 pooka mapaddr = mmap(NULL, maplen, PROT_READ|PROT_WRITE,
519 1.3 pooka MAP_ANON, -1, 0);
520 1.3 pooka if (mapaddr == MAP_FAILED)
521 1.3 pooka mapaddr = NULL;
522 1.3 pooka DPRINTF(("rump_sp handlereq: anonmmap: %p\n", mapaddr));
523 1.3 pooka send_anonmmap_resp(spc, spc->spc_hdr.rsp_reqno, mapaddr);
524 1.3 pooka break;
525 1.16 pooka case RUMPSP_RAISE:
526 1.16 pooka DPRINTF(("rump_sp handlereq: raise sig %d\n", rhdr->rsp_signo));
527 1.18 pooka raise((int)rhdr->rsp_signo);
528 1.16 pooka /*
529 1.16 pooka * We most likely have signals blocked, but the signal
530 1.16 pooka * will be handled soon enough when we return.
531 1.16 pooka */
532 1.16 pooka break;
533 1.3 pooka default:
534 1.12 pooka printf("PANIC: INVALID TYPE %d\n", reqtype);
535 1.3 pooka abort();
536 1.3 pooka break;
537 1.1 pooka }
538 1.1 pooka
539 1.6 pooka spcfreebuf(spc);
540 1.1 pooka }
541 1.1 pooka
542 1.11 pooka static unsigned ptab_idx;
543 1.11 pooka static struct sockaddr *serv_sa;
544 1.11 pooka
545 1.11 pooka static int
546 1.20 pooka doconnect(bool noisy)
547 1.1 pooka {
548 1.18 pooka struct respwait rw;
549 1.18 pooka struct rsp_hdr rhdr;
550 1.15 pooka struct kevent kev[NSIG+1];
551 1.9 pooka char banner[MAXBANNER];
552 1.18 pooka struct pollfd pfd;
553 1.15 pooka int s, error, flags, i;
554 1.9 pooka ssize_t n;
555 1.1 pooka
556 1.18 pooka if (kq != -1)
557 1.18 pooka host_close(kq);
558 1.18 pooka kq = -1;
559 1.19 pooka s = -1;
560 1.18 pooka
561 1.18 pooka if (clispc.spc_fd != -1)
562 1.18 pooka host_close(clispc.spc_fd);
563 1.18 pooka clispc.spc_fd = -1;
564 1.18 pooka
565 1.18 pooka /*
566 1.18 pooka * for reconnect, gate everyone out of the receiver code
567 1.18 pooka */
568 1.18 pooka putwait_locked(&clispc, &rw, &rhdr);
569 1.18 pooka
570 1.18 pooka pthread_mutex_lock(&clispc.spc_mtx);
571 1.18 pooka clispc.spc_reconnecting = 1;
572 1.18 pooka pthread_cond_broadcast(&clispc.spc_cv);
573 1.18 pooka clispc.spc_generation++;
574 1.18 pooka while (clispc.spc_istatus != SPCSTATUS_FREE) {
575 1.18 pooka clispc.spc_istatus = SPCSTATUS_WANTED;
576 1.18 pooka pthread_cond_wait(&rw.rw_cv, &clispc.spc_mtx);
577 1.18 pooka }
578 1.18 pooka kickall(&clispc);
579 1.18 pooka
580 1.18 pooka /*
581 1.18 pooka * we can release it already since we hold the
582 1.18 pooka * send lock during reconnect
583 1.18 pooka * XXX: assert it
584 1.18 pooka */
585 1.18 pooka clispc.spc_istatus = SPCSTATUS_FREE;
586 1.18 pooka pthread_mutex_unlock(&clispc.spc_mtx);
587 1.18 pooka unputwait_locked(&clispc, &rw);
588 1.18 pooka
589 1.18 pooka free(clispc.spc_buf);
590 1.18 pooka clispc.spc_off = 0;
591 1.18 pooka
592 1.13 pooka s = host_socket(parsetab[ptab_idx].domain, SOCK_STREAM, 0);
593 1.11 pooka if (s == -1)
594 1.2 pooka return -1;
595 1.1 pooka
596 1.18 pooka pfd.fd = s;
597 1.18 pooka pfd.events = POLLIN;
598 1.18 pooka while (host_connect(s, serv_sa, (socklen_t)serv_sa->sa_len) == -1) {
599 1.18 pooka if (errno == EINTR)
600 1.18 pooka continue;
601 1.20 pooka error = errno;
602 1.20 pooka if (noisy)
603 1.18 pooka fprintf(stderr, "rump_sp: client connect failed: %s\n",
604 1.18 pooka strerror(errno));
605 1.20 pooka errno = error;
606 1.20 pooka return -1;
607 1.2 pooka }
608 1.1 pooka
609 1.11 pooka if ((error = parsetab[ptab_idx].connhook(s)) != 0) {
610 1.2 pooka error = errno;
611 1.20 pooka if (noisy)
612 1.20 pooka fprintf(stderr, "rump_sp: connect hook failed\n");
613 1.2 pooka errno = error;
614 1.2 pooka return -1;
615 1.1 pooka }
616 1.4 pooka
617 1.13 pooka if ((n = host_read(s, banner, sizeof(banner)-1)) < 0) {
618 1.2 pooka error = errno;
619 1.20 pooka if (noisy)
620 1.20 pooka fprintf(stderr, "rump_sp: failed to read banner\n");
621 1.2 pooka errno = error;
622 1.2 pooka return -1;
623 1.1 pooka }
624 1.9 pooka
625 1.9 pooka if (banner[n-1] != '\n') {
626 1.20 pooka if (noisy)
627 1.20 pooka fprintf(stderr, "rump_sp: invalid banner\n");
628 1.9 pooka errno = EINVAL;
629 1.9 pooka return -1;
630 1.9 pooka }
631 1.9 pooka banner[n] = '\0';
632 1.18 pooka /* parse the banner some day */
633 1.9 pooka
634 1.15 pooka flags = host_fcntl(s, F_GETFL, 0);
635 1.15 pooka if (host_fcntl(s, F_SETFL, flags | O_NONBLOCK) == -1) {
636 1.20 pooka if (noisy)
637 1.20 pooka fprintf(stderr, "rump_sp: socket fd NONBLOCK: %s\n",
638 1.20 pooka strerror(errno));
639 1.15 pooka errno = EINVAL;
640 1.15 pooka return -1;
641 1.15 pooka }
642 1.18 pooka clispc.spc_fd = s;
643 1.18 pooka clispc.spc_state = SPCSTATE_RUNNING;
644 1.18 pooka clispc.spc_reconnecting = 0;
645 1.15 pooka
646 1.15 pooka /* setup kqueue, we want all signals and the fd */
647 1.17 pooka if ((kq = host_kqueue()) == -1) {
648 1.15 pooka error = errno;
649 1.20 pooka if (noisy)
650 1.20 pooka fprintf(stderr, "rump_sp: cannot setup kqueue");
651 1.15 pooka errno = error;
652 1.15 pooka return -1;
653 1.15 pooka }
654 1.15 pooka
655 1.15 pooka for (i = 0; i < NSIG; i++) {
656 1.15 pooka EV_SET(&kev[i], i+1, EVFILT_SIGNAL, EV_ADD|EV_ENABLE, 0, 0, 0);
657 1.15 pooka }
658 1.18 pooka EV_SET(&kev[NSIG], clispc.spc_fd,
659 1.18 pooka EVFILT_READ, EV_ADD|EV_ENABLE, 0, 0, 0);
660 1.17 pooka if (host_kevent(kq, kev, NSIG+1, NULL, 0, NULL) == -1) {
661 1.15 pooka error = errno;
662 1.20 pooka if (noisy)
663 1.20 pooka fprintf(stderr, "rump_sp: kevent() failed");
664 1.15 pooka errno = error;
665 1.15 pooka return -1;
666 1.15 pooka }
667 1.15 pooka
668 1.18 pooka return 0;
669 1.18 pooka }
670 1.18 pooka
671 1.18 pooka static int
672 1.18 pooka doinit(void)
673 1.18 pooka {
674 1.18 pooka
675 1.11 pooka TAILQ_INIT(&clispc.spc_respwait);
676 1.11 pooka pthread_mutex_init(&clispc.spc_mtx, NULL);
677 1.11 pooka pthread_cond_init(&clispc.spc_cv, NULL);
678 1.11 pooka
679 1.11 pooka return 0;
680 1.11 pooka }
681 1.11 pooka
682 1.13 pooka void *(*rumpclient_dlsym)(void *, const char *);
683 1.13 pooka
684 1.11 pooka int
685 1.11 pooka rumpclient_init()
686 1.11 pooka {
687 1.11 pooka char *p;
688 1.11 pooka int error;
689 1.11 pooka
690 1.13 pooka /* dlsym overrided by rumphijack? */
691 1.13 pooka if (!rumpclient_dlsym)
692 1.13 pooka rumpclient_dlsym = dlsym;
693 1.13 pooka
694 1.13 pooka /*
695 1.13 pooka * sag mir, wo die symbol sind. zogen fort, der krieg beginnt.
696 1.13 pooka * wann wird man je verstehen? wann wird man je verstehen?
697 1.13 pooka */
698 1.13 pooka #define FINDSYM2(_name_,_syscall_) \
699 1.13 pooka if ((host_##_name_ = rumpclient_dlsym(RTLD_NEXT, \
700 1.13 pooka #_syscall_)) == NULL) \
701 1.13 pooka /* host_##_name_ = _syscall_ */;
702 1.13 pooka #define FINDSYM(_name_) FINDSYM2(_name_,_name_)
703 1.13 pooka FINDSYM2(socket,__socket30);
704 1.13 pooka FINDSYM(close);
705 1.13 pooka FINDSYM(connect);
706 1.15 pooka FINDSYM(fcntl);
707 1.13 pooka FINDSYM(poll);
708 1.13 pooka FINDSYM(read);
709 1.13 pooka FINDSYM(sendto);
710 1.13 pooka FINDSYM(setsockopt);
711 1.17 pooka FINDSYM(kqueue);
712 1.17 pooka FINDSYM(kevent);
713 1.13 pooka #undef FINDSYM
714 1.13 pooka #undef FINDSY2
715 1.13 pooka
716 1.11 pooka if ((p = getenv("RUMP_SERVER")) == NULL) {
717 1.11 pooka errno = ENOENT;
718 1.11 pooka return -1;
719 1.11 pooka }
720 1.11 pooka
721 1.11 pooka if ((error = parseurl(p, &serv_sa, &ptab_idx, 0)) != 0) {
722 1.11 pooka errno = error;
723 1.11 pooka return -1;
724 1.11 pooka }
725 1.11 pooka
726 1.18 pooka if (doinit() == -1)
727 1.18 pooka return -1;
728 1.20 pooka if (doconnect(true) == -1)
729 1.11 pooka return -1;
730 1.11 pooka
731 1.18 pooka error = handshake_req(&clispc, NULL, 0, false);
732 1.11 pooka if (error) {
733 1.11 pooka pthread_mutex_destroy(&clispc.spc_mtx);
734 1.11 pooka pthread_cond_destroy(&clispc.spc_cv);
735 1.18 pooka if (clispc.spc_fd != -1)
736 1.18 pooka host_close(clispc.spc_fd);
737 1.10 pooka errno = error;
738 1.10 pooka return -1;
739 1.10 pooka }
740 1.10 pooka
741 1.12 pooka sigfillset(&fullset);
742 1.11 pooka return 0;
743 1.11 pooka }
744 1.11 pooka
745 1.11 pooka struct rumpclient_fork {
746 1.11 pooka uint32_t fork_auth[AUTHLEN];
747 1.11 pooka };
748 1.11 pooka
749 1.11 pooka struct rumpclient_fork *
750 1.11 pooka rumpclient_prefork(void)
751 1.11 pooka {
752 1.11 pooka struct rumpclient_fork *rpf;
753 1.11 pooka void *resp;
754 1.11 pooka int rv;
755 1.11 pooka
756 1.11 pooka rpf = malloc(sizeof(*rpf));
757 1.11 pooka if (rpf == NULL)
758 1.11 pooka return NULL;
759 1.11 pooka
760 1.11 pooka if ((rv = prefork_req(&clispc, &resp)) != 0) {
761 1.11 pooka free(rpf);
762 1.11 pooka errno = rv;
763 1.11 pooka return NULL;
764 1.11 pooka }
765 1.11 pooka
766 1.11 pooka memcpy(rpf->fork_auth, resp, sizeof(rpf->fork_auth));
767 1.11 pooka free(resp);
768 1.11 pooka
769 1.11 pooka return rpf;
770 1.11 pooka }
771 1.11 pooka
772 1.11 pooka int
773 1.11 pooka rumpclient_fork_init(struct rumpclient_fork *rpf)
774 1.11 pooka {
775 1.11 pooka int error;
776 1.11 pooka
777 1.11 pooka memset(&clispc, 0, sizeof(clispc));
778 1.11 pooka clispc.spc_fd = -1;
779 1.18 pooka kq = -1;
780 1.11 pooka
781 1.18 pooka if (doinit() == -1)
782 1.18 pooka return -1;
783 1.20 pooka if (doconnect(false) == -1)
784 1.11 pooka return -1;
785 1.10 pooka
786 1.18 pooka error = handshake_req(&clispc, rpf->fork_auth, 0, false);
787 1.10 pooka if (error) {
788 1.10 pooka pthread_mutex_destroy(&clispc.spc_mtx);
789 1.10 pooka pthread_cond_destroy(&clispc.spc_cv);
790 1.11 pooka errno = error;
791 1.11 pooka return -1;
792 1.10 pooka }
793 1.11 pooka
794 1.11 pooka return 0;
795 1.1 pooka }
796 1.20 pooka
797 1.20 pooka void
798 1.20 pooka rumpclient_setconnretry(time_t timeout)
799 1.20 pooka {
800 1.20 pooka
801 1.20 pooka if (timeout < RUMPCLIENT_RETRYCONN_ONCE)
802 1.20 pooka return; /* gigo */
803 1.20 pooka
804 1.20 pooka retrytimo = timeout;
805 1.20 pooka }
806