Home | History | Annotate | Line # | Download | only in librumpclient
rumpclient.c revision 1.22
      1  1.22  pooka /*      $NetBSD: rumpclient.c,v 1.22 2011/02/05 12:38:19 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.22  pooka #if !__NetBSD_Prereq__(5,99,7)
    713  1.17  pooka 	FINDSYM(kevent);
    714  1.22  pooka #else
    715  1.22  pooka 	FINDSYM2(kevent,_sys___kevent50);
    716  1.22  pooka #endif
    717  1.13  pooka #undef	FINDSYM
    718  1.13  pooka #undef	FINDSY2
    719  1.13  pooka 
    720  1.11  pooka 	if ((p = getenv("RUMP_SERVER")) == NULL) {
    721  1.11  pooka 		errno = ENOENT;
    722  1.11  pooka 		return -1;
    723  1.11  pooka 	}
    724  1.11  pooka 
    725  1.11  pooka 	if ((error = parseurl(p, &serv_sa, &ptab_idx, 0)) != 0) {
    726  1.11  pooka 		errno = error;
    727  1.11  pooka 		return -1;
    728  1.11  pooka 	}
    729  1.11  pooka 
    730  1.18  pooka 	if (doinit() == -1)
    731  1.18  pooka 		return -1;
    732  1.20  pooka 	if (doconnect(true) == -1)
    733  1.11  pooka 		return -1;
    734  1.11  pooka 
    735  1.18  pooka 	error = handshake_req(&clispc, NULL, 0, false);
    736  1.11  pooka 	if (error) {
    737  1.11  pooka 		pthread_mutex_destroy(&clispc.spc_mtx);
    738  1.11  pooka 		pthread_cond_destroy(&clispc.spc_cv);
    739  1.18  pooka 		if (clispc.spc_fd != -1)
    740  1.18  pooka 			host_close(clispc.spc_fd);
    741  1.10  pooka 		errno = error;
    742  1.10  pooka 		return -1;
    743  1.10  pooka 	}
    744  1.10  pooka 
    745  1.12  pooka 	sigfillset(&fullset);
    746  1.11  pooka 	return 0;
    747  1.11  pooka }
    748  1.11  pooka 
    749  1.11  pooka struct rumpclient_fork {
    750  1.11  pooka 	uint32_t fork_auth[AUTHLEN];
    751  1.11  pooka };
    752  1.11  pooka 
    753  1.11  pooka struct rumpclient_fork *
    754  1.11  pooka rumpclient_prefork(void)
    755  1.11  pooka {
    756  1.11  pooka 	struct rumpclient_fork *rpf;
    757  1.11  pooka 	void *resp;
    758  1.11  pooka 	int rv;
    759  1.11  pooka 
    760  1.11  pooka 	rpf = malloc(sizeof(*rpf));
    761  1.11  pooka 	if (rpf == NULL)
    762  1.11  pooka 		return NULL;
    763  1.11  pooka 
    764  1.11  pooka 	if ((rv = prefork_req(&clispc, &resp)) != 0) {
    765  1.11  pooka 		free(rpf);
    766  1.11  pooka 		errno = rv;
    767  1.11  pooka 		return NULL;
    768  1.11  pooka 	}
    769  1.11  pooka 
    770  1.11  pooka 	memcpy(rpf->fork_auth, resp, sizeof(rpf->fork_auth));
    771  1.11  pooka 	free(resp);
    772  1.11  pooka 
    773  1.11  pooka 	return rpf;
    774  1.11  pooka }
    775  1.11  pooka 
    776  1.11  pooka int
    777  1.11  pooka rumpclient_fork_init(struct rumpclient_fork *rpf)
    778  1.11  pooka {
    779  1.11  pooka 	int error;
    780  1.11  pooka 
    781  1.11  pooka 	memset(&clispc, 0, sizeof(clispc));
    782  1.11  pooka 	clispc.spc_fd = -1;
    783  1.18  pooka 	kq = -1;
    784  1.11  pooka 
    785  1.18  pooka 	if (doinit() == -1)
    786  1.18  pooka 		return -1;
    787  1.20  pooka 	if (doconnect(false) == -1)
    788  1.11  pooka 		return -1;
    789  1.10  pooka 
    790  1.18  pooka 	error = handshake_req(&clispc, rpf->fork_auth, 0, false);
    791  1.10  pooka 	if (error) {
    792  1.10  pooka 		pthread_mutex_destroy(&clispc.spc_mtx);
    793  1.10  pooka 		pthread_cond_destroy(&clispc.spc_cv);
    794  1.11  pooka 		errno = error;
    795  1.11  pooka 		return -1;
    796  1.10  pooka 	}
    797  1.11  pooka 
    798  1.11  pooka 	return 0;
    799   1.1  pooka }
    800  1.20  pooka 
    801  1.20  pooka void
    802  1.20  pooka rumpclient_setconnretry(time_t timeout)
    803  1.20  pooka {
    804  1.20  pooka 
    805  1.20  pooka 	if (timeout < RUMPCLIENT_RETRYCONN_ONCE)
    806  1.20  pooka 		return; /* gigo */
    807  1.20  pooka 
    808  1.20  pooka 	retrytimo = timeout;
    809  1.20  pooka }
    810