Home | History | Annotate | Line # | Download | only in librumpclient
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