Home | History | Annotate | Line # | Download | only in librumpuser
sp_common.c revision 1.41
      1  1.41  christos /*      $NetBSD: sp_common.c,v 1.41 2020/05/06 12:44:36 christos Exp $	*/
      2   1.1     pooka 
      3   1.1     pooka /*
      4  1.18     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  * Common client/server sysproxy routines.  #included.
     30   1.1     pooka  */
     31   1.1     pooka 
     32  1.32     pooka #include "rumpuser_port.h"
     33   1.1     pooka 
     34   1.1     pooka #include <sys/types.h>
     35   1.1     pooka #include <sys/mman.h>
     36   1.4     pooka #include <sys/queue.h>
     37   1.1     pooka #include <sys/socket.h>
     38   1.2     pooka #include <sys/un.h>
     39  1.38     pooka #include <sys/uio.h>
     40   1.1     pooka 
     41   1.1     pooka #include <arpa/inet.h>
     42   1.1     pooka #include <netinet/in.h>
     43   1.1     pooka #include <netinet/tcp.h>
     44   1.1     pooka 
     45   1.1     pooka #include <assert.h>
     46   1.1     pooka #include <errno.h>
     47   1.1     pooka #include <fcntl.h>
     48  1.13     pooka #include <inttypes.h>
     49  1.32     pooka #include <limits.h>
     50   1.1     pooka #include <poll.h>
     51   1.4     pooka #include <pthread.h>
     52   1.1     pooka #include <stdarg.h>
     53  1.11     pooka #include <stddef.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.32     pooka /*
     60  1.32     pooka  * XXX: NetBSD's __unused collides with Linux headers, so we cannot
     61  1.32     pooka  * define it before we've included everything.
     62  1.32     pooka  */
     63  1.39  dholland #if !defined(__unused) && (defined(__clang__) || defined(__GNUC__))
     64  1.32     pooka #define __unused __attribute__((__unused__))
     65  1.32     pooka #endif
     66  1.39  dholland #if !defined(__printflike) && (defined(__clang__) || defined(__GNUC__))
     67  1.39  dholland #define __printflike(a,b) __attribute__((__format__(__printf__, a, b))))
     68  1.39  dholland #endif
     69  1.32     pooka 
     70   1.1     pooka //#define DEBUG
     71   1.1     pooka #ifdef DEBUG
     72   1.1     pooka #define DPRINTF(x) mydprintf x
     73  1.39  dholland static __printflike(1, 2) void
     74   1.1     pooka mydprintf(const char *fmt, ...)
     75   1.1     pooka {
     76   1.1     pooka 	va_list ap;
     77   1.1     pooka 
     78   1.1     pooka 	va_start(ap, fmt);
     79   1.1     pooka 	vfprintf(stderr, fmt, ap);
     80   1.1     pooka 	va_end(ap);
     81   1.1     pooka }
     82   1.1     pooka #else
     83   1.1     pooka #define DPRINTF(x)
     84   1.1     pooka #endif
     85   1.1     pooka 
     86  1.20     pooka #ifndef HOSTOPS
     87  1.20     pooka #define host_poll poll
     88  1.20     pooka #define host_read read
     89  1.31     pooka #define host_sendmsg sendmsg
     90  1.20     pooka #define host_setsockopt setsockopt
     91  1.20     pooka #endif
     92  1.20     pooka 
     93  1.34     pooka #define IOVPUT(_io_, _b_) _io_.iov_base = 			\
     94  1.34     pooka     (void *)&_b_; _io_.iov_len = sizeof(_b_);
     95  1.34     pooka #define IOVPUT_WITHSIZE(_io_, _b_, _l_) _io_.iov_base =		\
     96  1.34     pooka     (void *)(_b_); _io_.iov_len = _l_;
     97  1.31     pooka #define SENDIOV(_spc_, _iov_) dosend(_spc_, _iov_, __arraycount(_iov_))
     98  1.31     pooka 
     99   1.1     pooka /*
    100   1.1     pooka  * Bah, I hate writing on-off-wire conversions in C
    101   1.1     pooka  */
    102   1.1     pooka 
    103  1.13     pooka enum { RUMPSP_REQ, RUMPSP_RESP, RUMPSP_ERROR };
    104  1.17     pooka enum {	RUMPSP_HANDSHAKE,
    105  1.17     pooka 	RUMPSP_SYSCALL,
    106  1.10     pooka 	RUMPSP_COPYIN, RUMPSP_COPYINSTR,
    107  1.10     pooka 	RUMPSP_COPYOUT, RUMPSP_COPYOUTSTR,
    108  1.18     pooka 	RUMPSP_ANONMMAP,
    109  1.24     pooka 	RUMPSP_PREFORK,
    110  1.24     pooka 	RUMPSP_RAISE };
    111   1.1     pooka 
    112  1.28     pooka enum { HANDSHAKE_GUEST, HANDSHAKE_AUTH, HANDSHAKE_FORK, HANDSHAKE_EXEC };
    113  1.18     pooka 
    114  1.33     pooka /*
    115  1.33     pooka  * error types used for RUMPSP_ERROR
    116  1.33     pooka  */
    117  1.33     pooka enum rumpsp_err { RUMPSP_ERR_NONE = 0, RUMPSP_ERR_TRYAGAIN, RUMPSP_ERR_AUTH,
    118  1.33     pooka 	RUMPSP_ERR_INVALID_PREFORK, RUMPSP_ERR_RFORK_FAILED,
    119  1.33     pooka 	RUMPSP_ERR_INEXEC, RUMPSP_ERR_NOMEM, RUMPSP_ERR_MALFORMED_REQUEST };
    120  1.33     pooka 
    121  1.33     pooka /*
    122  1.33     pooka  * The mapping of the above types to errno.  They are almost never exposed
    123  1.33     pooka  * to the client after handshake (except for a server resource shortage
    124  1.33     pooka  * and the client trying to be funny).  This is a function instead of
    125  1.33     pooka  * an array to catch missing values.  Theoretically, the compiled code
    126  1.33     pooka  * should be the same.
    127  1.33     pooka  */
    128  1.33     pooka static int
    129  1.33     pooka errmap(enum rumpsp_err error)
    130  1.33     pooka {
    131  1.33     pooka 
    132  1.33     pooka 	switch (error) {
    133  1.33     pooka 	/* XXX: no EAUTH on Linux */
    134  1.33     pooka 	case RUMPSP_ERR_NONE:			return 0;
    135  1.33     pooka 	case RUMPSP_ERR_AUTH:			return EPERM;
    136  1.33     pooka 	case RUMPSP_ERR_TRYAGAIN:		return EAGAIN;
    137  1.33     pooka 	case RUMPSP_ERR_INVALID_PREFORK:	return ESRCH;
    138  1.33     pooka 	case RUMPSP_ERR_RFORK_FAILED:		return EIO; /* got a light? */
    139  1.33     pooka 	case RUMPSP_ERR_INEXEC:			return EBUSY;
    140  1.33     pooka 	case RUMPSP_ERR_NOMEM:			return ENOMEM;
    141  1.33     pooka 	case RUMPSP_ERR_MALFORMED_REQUEST:	return EINVAL;
    142  1.33     pooka 	}
    143  1.33     pooka 
    144  1.33     pooka 	return -1;
    145  1.33     pooka }
    146  1.33     pooka 
    147  1.18     pooka #define AUTHLEN 4 /* 128bit fork auth */
    148  1.17     pooka 
    149   1.1     pooka struct rsp_hdr {
    150   1.1     pooka 	uint64_t rsp_len;
    151   1.1     pooka 	uint64_t rsp_reqno;
    152   1.4     pooka 	uint16_t rsp_class;
    153   1.4     pooka 	uint16_t rsp_type;
    154   1.1     pooka 	/*
    155   1.1     pooka 	 * We want this structure 64bit-aligned for typecast fun,
    156   1.1     pooka 	 * so might as well use the following for something.
    157   1.1     pooka 	 */
    158  1.13     pooka 	union {
    159  1.13     pooka 		uint32_t sysnum;
    160  1.13     pooka 		uint32_t error;
    161  1.17     pooka 		uint32_t handshake;
    162  1.24     pooka 		uint32_t signo;
    163  1.13     pooka 	} u;
    164   1.1     pooka };
    165   1.1     pooka #define HDRSZ sizeof(struct rsp_hdr)
    166  1.13     pooka #define rsp_sysnum u.sysnum
    167  1.13     pooka #define rsp_error u.error
    168  1.17     pooka #define rsp_handshake u.handshake
    169  1.24     pooka #define rsp_signo u.signo
    170   1.1     pooka 
    171  1.16     pooka #define MAXBANNER 96
    172  1.16     pooka 
    173   1.1     pooka /*
    174   1.1     pooka  * Data follows the header.  We have two types of structured data.
    175   1.1     pooka  */
    176   1.1     pooka 
    177   1.1     pooka /* copyin/copyout */
    178   1.1     pooka struct rsp_copydata {
    179   1.1     pooka 	size_t rcp_len;
    180   1.1     pooka 	void *rcp_addr;
    181   1.1     pooka 	uint8_t rcp_data[0];
    182   1.1     pooka };
    183   1.1     pooka 
    184   1.1     pooka /* syscall response */
    185   1.1     pooka struct rsp_sysresp {
    186   1.1     pooka 	int rsys_error;
    187   1.1     pooka 	register_t rsys_retval[2];
    188   1.1     pooka };
    189   1.1     pooka 
    190  1.18     pooka struct handshake_fork {
    191  1.18     pooka 	uint32_t rf_auth[4];
    192  1.18     pooka 	int rf_cancel;
    193  1.18     pooka };
    194  1.18     pooka 
    195   1.4     pooka struct respwait {
    196   1.4     pooka 	uint64_t rw_reqno;
    197   1.4     pooka 	void *rw_data;
    198   1.4     pooka 	size_t rw_dlen;
    199  1.21     pooka 	int rw_done;
    200  1.13     pooka 	int rw_error;
    201   1.4     pooka 
    202   1.4     pooka 	pthread_cond_t rw_cv;
    203   1.4     pooka 
    204   1.4     pooka 	TAILQ_ENTRY(respwait) rw_entries;
    205   1.4     pooka };
    206   1.1     pooka 
    207  1.18     pooka struct prefork;
    208   1.1     pooka struct spclient {
    209   1.1     pooka 	int spc_fd;
    210   1.7     pooka 	int spc_refcnt;
    211  1.17     pooka 	int spc_state;
    212   1.6     pooka 
    213  1.11     pooka 	pthread_mutex_t spc_mtx;
    214  1.11     pooka 	pthread_cond_t spc_cv;
    215  1.11     pooka 
    216   1.6     pooka 	struct lwp *spc_mainlwp;
    217   1.6     pooka 	pid_t spc_pid;
    218   1.1     pooka 
    219  1.11     pooka 	TAILQ_HEAD(, respwait) spc_respwait;
    220  1.11     pooka 
    221  1.11     pooka 	/* rest of the fields are zeroed upon disconnect */
    222  1.12     pooka #define SPC_ZEROFF offsetof(struct spclient, spc_pfd)
    223   1.7     pooka 	struct pollfd *spc_pfd;
    224   1.7     pooka 
    225   1.1     pooka 	struct rsp_hdr spc_hdr;
    226   1.1     pooka 	uint8_t *spc_buf;
    227   1.1     pooka 	size_t spc_off;
    228   1.1     pooka 
    229   1.4     pooka 	uint64_t spc_nextreq;
    230  1.25     pooka 	uint64_t spc_syscallreq;
    231  1.26     pooka 	uint64_t spc_generation;
    232   1.4     pooka 	int spc_ostatus, spc_istatus;
    233  1.26     pooka 	int spc_reconnecting;
    234  1.30     pooka 	int spc_inexec;
    235  1.18     pooka 
    236  1.18     pooka 	LIST_HEAD(, prefork) spc_pflist;
    237   1.1     pooka };
    238   1.4     pooka #define SPCSTATUS_FREE 0
    239   1.4     pooka #define SPCSTATUS_BUSY 1
    240   1.4     pooka #define SPCSTATUS_WANTED 2
    241   1.1     pooka 
    242  1.17     pooka #define SPCSTATE_NEW     0
    243  1.17     pooka #define SPCSTATE_RUNNING 1
    244  1.17     pooka #define SPCSTATE_DYING   2
    245  1.17     pooka 
    246   1.1     pooka typedef int (*addrparse_fn)(const char *, struct sockaddr **, int);
    247   1.1     pooka typedef int (*connecthook_fn)(int);
    248  1.15     pooka typedef void (*cleanup_fn)(struct sockaddr *);
    249   1.1     pooka 
    250   1.4     pooka static int readframe(struct spclient *);
    251   1.4     pooka static void handlereq(struct spclient *);
    252   1.4     pooka 
    253  1.13     pooka static __inline void
    254  1.13     pooka spcresetbuf(struct spclient *spc)
    255  1.13     pooka {
    256  1.13     pooka 
    257  1.13     pooka 	spc->spc_buf = NULL;
    258  1.13     pooka 	spc->spc_off = 0;
    259  1.13     pooka }
    260  1.13     pooka 
    261  1.13     pooka static __inline void
    262  1.13     pooka spcfreebuf(struct spclient *spc)
    263  1.13     pooka {
    264  1.13     pooka 
    265  1.13     pooka 	free(spc->spc_buf);
    266  1.13     pooka 	spcresetbuf(spc);
    267  1.13     pooka }
    268  1.13     pooka 
    269   1.4     pooka static void
    270  1.12     pooka sendlockl(struct spclient *spc)
    271   1.4     pooka {
    272   1.4     pooka 
    273   1.4     pooka 	while (spc->spc_ostatus != SPCSTATUS_FREE) {
    274   1.4     pooka 		spc->spc_ostatus = SPCSTATUS_WANTED;
    275   1.4     pooka 		pthread_cond_wait(&spc->spc_cv, &spc->spc_mtx);
    276   1.4     pooka 	}
    277   1.4     pooka 	spc->spc_ostatus = SPCSTATUS_BUSY;
    278  1.12     pooka }
    279  1.12     pooka 
    280  1.26     pooka static void __unused
    281  1.12     pooka sendlock(struct spclient *spc)
    282  1.12     pooka {
    283  1.12     pooka 
    284  1.12     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    285  1.12     pooka 	sendlockl(spc);
    286   1.4     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    287   1.4     pooka }
    288   1.4     pooka 
    289   1.4     pooka static void
    290  1.12     pooka sendunlockl(struct spclient *spc)
    291   1.4     pooka {
    292   1.4     pooka 
    293   1.4     pooka 	if (spc->spc_ostatus == SPCSTATUS_WANTED)
    294   1.4     pooka 		pthread_cond_broadcast(&spc->spc_cv);
    295   1.4     pooka 	spc->spc_ostatus = SPCSTATUS_FREE;
    296  1.12     pooka }
    297  1.12     pooka 
    298  1.12     pooka static void
    299  1.12     pooka sendunlock(struct spclient *spc)
    300  1.12     pooka {
    301  1.12     pooka 
    302  1.12     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    303  1.12     pooka 	sendunlockl(spc);
    304   1.4     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    305   1.4     pooka }
    306   1.1     pooka 
    307   1.1     pooka static int
    308  1.31     pooka dosend(struct spclient *spc, struct iovec *iov, size_t iovlen)
    309   1.1     pooka {
    310  1.31     pooka 	struct msghdr msg;
    311   1.1     pooka 	struct pollfd pfd;
    312  1.31     pooka 	ssize_t n = 0;
    313   1.1     pooka 	int fd = spc->spc_fd;
    314   1.1     pooka 
    315   1.1     pooka 	pfd.fd = fd;
    316   1.1     pooka 	pfd.events = POLLOUT;
    317   1.1     pooka 
    318  1.31     pooka 	memset(&msg, 0, sizeof(msg));
    319  1.31     pooka 
    320  1.31     pooka 	for (;;) {
    321  1.31     pooka 		/* not first round?  poll */
    322   1.1     pooka 		if (n) {
    323  1.20     pooka 			if (host_poll(&pfd, 1, INFTIM) == -1) {
    324   1.1     pooka 				if (errno == EINTR)
    325   1.1     pooka 					continue;
    326   1.1     pooka 				return errno;
    327   1.1     pooka 			}
    328   1.1     pooka 		}
    329   1.1     pooka 
    330  1.31     pooka 		msg.msg_iov = iov;
    331  1.31     pooka 		msg.msg_iovlen = iovlen;
    332  1.31     pooka 		n = host_sendmsg(fd, &msg, MSG_NOSIGNAL);
    333  1.10     pooka 		if (n == -1)  {
    334  1.26     pooka 			if (errno == EPIPE)
    335  1.26     pooka 				return ENOTCONN;
    336  1.10     pooka 			if (errno != EAGAIN)
    337  1.18     pooka 				return errno;
    338  1.10     pooka 			continue;
    339   1.1     pooka 		}
    340  1.26     pooka 		if (n == 0) {
    341  1.26     pooka 			return ENOTCONN;
    342  1.26     pooka 		}
    343  1.31     pooka 
    344  1.31     pooka 		/* ok, need to adjust iovec for potential next round */
    345  1.40     kamil 		while (iovlen && n >= (ssize_t)iov[0].iov_len) {
    346  1.31     pooka 			n -= iov[0].iov_len;
    347  1.31     pooka 			iov++;
    348  1.31     pooka 			iovlen--;
    349  1.31     pooka 		}
    350  1.31     pooka 
    351  1.31     pooka 		if (iovlen == 0) {
    352  1.31     pooka 			_DIAGASSERT(n == 0);
    353  1.31     pooka 			break;
    354  1.31     pooka 		} else {
    355  1.34     pooka 			iov[0].iov_base =
    356  1.34     pooka 			    (void *)((uint8_t *)iov[0].iov_base + n);
    357  1.31     pooka 			iov[0].iov_len -= n;
    358  1.31     pooka 		}
    359   1.1     pooka 	}
    360   1.1     pooka 
    361   1.1     pooka 	return 0;
    362   1.1     pooka }
    363   1.1     pooka 
    364   1.4     pooka static void
    365  1.26     pooka doputwait(struct spclient *spc, struct respwait *rw, struct rsp_hdr *rhdr)
    366   1.4     pooka {
    367   1.4     pooka 
    368   1.4     pooka 	rw->rw_data = NULL;
    369  1.22     pooka 	rw->rw_dlen = rw->rw_done = rw->rw_error = 0;
    370   1.4     pooka 	pthread_cond_init(&rw->rw_cv, NULL);
    371   1.4     pooka 
    372   1.4     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    373   1.4     pooka 	rw->rw_reqno = rhdr->rsp_reqno = spc->spc_nextreq++;
    374   1.4     pooka 	TAILQ_INSERT_TAIL(&spc->spc_respwait, rw, rw_entries);
    375  1.26     pooka }
    376  1.26     pooka 
    377  1.26     pooka static void __unused
    378  1.26     pooka putwait_locked(struct spclient *spc, struct respwait *rw, struct rsp_hdr *rhdr)
    379  1.26     pooka {
    380  1.26     pooka 
    381  1.26     pooka 	doputwait(spc, rw, rhdr);
    382  1.26     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    383  1.26     pooka }
    384  1.26     pooka 
    385  1.26     pooka static void
    386  1.26     pooka putwait(struct spclient *spc, struct respwait *rw, struct rsp_hdr *rhdr)
    387  1.26     pooka {
    388  1.12     pooka 
    389  1.26     pooka 	doputwait(spc, rw, rhdr);
    390  1.12     pooka 	sendlockl(spc);
    391  1.23     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    392   1.8     pooka }
    393   1.8     pooka 
    394   1.8     pooka static void
    395  1.26     pooka dounputwait(struct spclient *spc, struct respwait *rw)
    396  1.26     pooka {
    397  1.26     pooka 
    398  1.26     pooka 	TAILQ_REMOVE(&spc->spc_respwait, rw, rw_entries);
    399  1.26     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    400  1.26     pooka 	pthread_cond_destroy(&rw->rw_cv);
    401  1.26     pooka 
    402  1.26     pooka }
    403  1.26     pooka 
    404  1.26     pooka static void __unused
    405  1.26     pooka unputwait_locked(struct spclient *spc, struct respwait *rw)
    406  1.26     pooka {
    407  1.26     pooka 
    408  1.26     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    409  1.26     pooka 	dounputwait(spc, rw);
    410  1.26     pooka }
    411  1.26     pooka 
    412  1.26     pooka static void
    413   1.8     pooka unputwait(struct spclient *spc, struct respwait *rw)
    414   1.8     pooka {
    415   1.8     pooka 
    416  1.23     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    417  1.12     pooka 	sendunlockl(spc);
    418  1.12     pooka 
    419  1.26     pooka 	dounputwait(spc, rw);
    420   1.4     pooka }
    421   1.4     pooka 
    422   1.4     pooka static void
    423   1.4     pooka kickwaiter(struct spclient *spc)
    424   1.4     pooka {
    425   1.4     pooka 	struct respwait *rw;
    426  1.22     pooka 	int error = 0;
    427   1.4     pooka 
    428   1.4     pooka 	pthread_mutex_lock(&spc->spc_mtx);
    429   1.4     pooka 	TAILQ_FOREACH(rw, &spc->spc_respwait, rw_entries) {
    430   1.4     pooka 		if (rw->rw_reqno == spc->spc_hdr.rsp_reqno)
    431   1.4     pooka 			break;
    432   1.4     pooka 	}
    433   1.4     pooka 	if (rw == NULL) {
    434  1.13     pooka 		DPRINTF(("no waiter found, invalid reqno %" PRIu64 "?\n",
    435  1.13     pooka 		    spc->spc_hdr.rsp_reqno));
    436  1.23     pooka 		pthread_mutex_unlock(&spc->spc_mtx);
    437  1.18     pooka 		spcfreebuf(spc);
    438   1.4     pooka 		return;
    439   1.4     pooka 	}
    440  1.10     pooka 	DPRINTF(("rump_sp: client %p woke up waiter at %p\n", spc, rw));
    441   1.4     pooka 	rw->rw_data = spc->spc_buf;
    442  1.21     pooka 	rw->rw_done = 1;
    443  1.11     pooka 	rw->rw_dlen = (size_t)(spc->spc_off - HDRSZ);
    444  1.13     pooka 	if (spc->spc_hdr.rsp_class == RUMPSP_ERROR) {
    445  1.33     pooka 		error = rw->rw_error = errmap(spc->spc_hdr.rsp_error);
    446  1.13     pooka 	}
    447   1.4     pooka 	pthread_cond_signal(&rw->rw_cv);
    448   1.4     pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    449   1.4     pooka 
    450  1.14     pooka 	if (error)
    451  1.13     pooka 		spcfreebuf(spc);
    452  1.13     pooka 	else
    453  1.13     pooka 		spcresetbuf(spc);
    454   1.4     pooka }
    455   1.4     pooka 
    456   1.4     pooka static void
    457   1.4     pooka kickall(struct spclient *spc)
    458   1.4     pooka {
    459   1.4     pooka 	struct respwait *rw;
    460   1.4     pooka 
    461   1.4     pooka 	/* DIAGASSERT(mutex_owned(spc_lock)) */
    462   1.4     pooka 	TAILQ_FOREACH(rw, &spc->spc_respwait, rw_entries)
    463  1.12     pooka 		pthread_cond_broadcast(&rw->rw_cv);
    464   1.4     pooka }
    465   1.4     pooka 
    466   1.4     pooka static int
    467   1.1     pooka readframe(struct spclient *spc)
    468   1.1     pooka {
    469   1.1     pooka 	int fd = spc->spc_fd;
    470   1.1     pooka 	size_t left;
    471   1.1     pooka 	size_t framelen;
    472   1.1     pooka 	ssize_t n;
    473   1.1     pooka 
    474   1.1     pooka 	/* still reading header? */
    475   1.1     pooka 	if (spc->spc_off < HDRSZ) {
    476   1.1     pooka 		DPRINTF(("rump_sp: readframe getting header at offset %zu\n",
    477   1.1     pooka 		    spc->spc_off));
    478   1.1     pooka 
    479   1.1     pooka 		left = HDRSZ - spc->spc_off;
    480   1.1     pooka 		/*LINTED: cast ok */
    481  1.20     pooka 		n = host_read(fd, (uint8_t*)&spc->spc_hdr + spc->spc_off, left);
    482   1.1     pooka 		if (n == 0) {
    483   1.1     pooka 			return -1;
    484   1.1     pooka 		}
    485   1.1     pooka 		if (n == -1) {
    486   1.1     pooka 			if (errno == EAGAIN)
    487   1.1     pooka 				return 0;
    488   1.1     pooka 			return -1;
    489   1.1     pooka 		}
    490   1.1     pooka 
    491   1.1     pooka 		spc->spc_off += n;
    492  1.29     pooka 		if (spc->spc_off < HDRSZ) {
    493  1.29     pooka 			return 0;
    494  1.29     pooka 		}
    495   1.1     pooka 
    496   1.1     pooka 		/*LINTED*/
    497   1.1     pooka 		framelen = spc->spc_hdr.rsp_len;
    498   1.1     pooka 
    499   1.1     pooka 		if (framelen < HDRSZ) {
    500   1.1     pooka 			return -1;
    501   1.1     pooka 		} else if (framelen == HDRSZ) {
    502   1.1     pooka 			return 1;
    503   1.1     pooka 		}
    504   1.1     pooka 
    505  1.41  christos 		/* Add an extra byte so that we are always NUL-terminated */
    506  1.41  christos 		spc->spc_buf = malloc(framelen - HDRSZ + 1);
    507   1.1     pooka 		if (spc->spc_buf == NULL) {
    508   1.1     pooka 			return -1;
    509   1.1     pooka 		}
    510  1.41  christos 		memset(spc->spc_buf, 0, framelen - HDRSZ + 1);
    511   1.1     pooka 
    512   1.1     pooka 		/* "fallthrough" */
    513   1.1     pooka 	} else {
    514   1.1     pooka 		/*LINTED*/
    515   1.1     pooka 		framelen = spc->spc_hdr.rsp_len;
    516   1.1     pooka 	}
    517   1.1     pooka 
    518   1.1     pooka 	left = framelen - spc->spc_off;
    519   1.1     pooka 
    520   1.1     pooka 	DPRINTF(("rump_sp: readframe getting body at offset %zu, left %zu\n",
    521   1.1     pooka 	    spc->spc_off, left));
    522   1.1     pooka 
    523   1.1     pooka 	if (left == 0)
    524   1.1     pooka 		return 1;
    525  1.20     pooka 	n = host_read(fd, spc->spc_buf + (spc->spc_off - HDRSZ), left);
    526   1.1     pooka 	if (n == 0) {
    527   1.1     pooka 		return -1;
    528   1.1     pooka 	}
    529   1.1     pooka 	if (n == -1) {
    530   1.1     pooka 		if (errno == EAGAIN)
    531   1.1     pooka 			return 0;
    532   1.1     pooka 		return -1;
    533   1.1     pooka 	}
    534   1.1     pooka 	spc->spc_off += n;
    535   1.1     pooka 	left -= n;
    536   1.1     pooka 
    537   1.1     pooka 	/* got everything? */
    538   1.1     pooka 	if (left == 0)
    539   1.1     pooka 		return 1;
    540   1.1     pooka 	else
    541   1.1     pooka 		return 0;
    542   1.1     pooka }
    543   1.1     pooka 
    544   1.1     pooka static int
    545   1.1     pooka tcp_parse(const char *addr, struct sockaddr **sa, int allow_wildcard)
    546   1.1     pooka {
    547   1.1     pooka 	struct sockaddr_in sin;
    548   1.1     pooka 	char buf[64];
    549   1.1     pooka 	const char *p;
    550   1.1     pooka 	size_t l;
    551   1.1     pooka 	int port;
    552   1.1     pooka 
    553   1.1     pooka 	memset(&sin, 0, sizeof(sin));
    554  1.35     pooka 	SIN_SETLEN(sin, sizeof(sin));
    555   1.1     pooka 	sin.sin_family = AF_INET;
    556   1.1     pooka 
    557   1.1     pooka 	p = strchr(addr, ':');
    558   1.1     pooka 	if (!p) {
    559   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: missing port specifier\n");
    560   1.1     pooka 		return EINVAL;
    561   1.1     pooka 	}
    562   1.1     pooka 
    563   1.1     pooka 	l = p - addr;
    564   1.1     pooka 	if (l > sizeof(buf)-1) {
    565   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: address too long\n");
    566   1.1     pooka 		return EINVAL;
    567   1.1     pooka 	}
    568   1.1     pooka 	strncpy(buf, addr, l);
    569   1.1     pooka 	buf[l] = '\0';
    570   1.1     pooka 
    571   1.1     pooka 	/* special INADDR_ANY treatment */
    572   1.1     pooka 	if (strcmp(buf, "*") == 0 || strcmp(buf, "0") == 0) {
    573   1.1     pooka 		sin.sin_addr.s_addr = INADDR_ANY;
    574   1.1     pooka 	} else {
    575   1.1     pooka 		switch (inet_pton(AF_INET, buf, &sin.sin_addr)) {
    576   1.1     pooka 		case 1:
    577   1.1     pooka 			break;
    578   1.1     pooka 		case 0:
    579   1.1     pooka 			fprintf(stderr, "rump_sp_tcp: cannot parse %s\n", buf);
    580   1.1     pooka 			return EINVAL;
    581   1.1     pooka 		case -1:
    582   1.1     pooka 			fprintf(stderr, "rump_sp_tcp: inet_pton failed\n");
    583   1.1     pooka 			return errno;
    584   1.1     pooka 		default:
    585   1.1     pooka 			assert(/*CONSTCOND*/0);
    586   1.1     pooka 			return EINVAL;
    587   1.1     pooka 		}
    588   1.1     pooka 	}
    589   1.1     pooka 
    590   1.1     pooka 	if (!allow_wildcard && sin.sin_addr.s_addr == INADDR_ANY) {
    591   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: client needs !INADDR_ANY\n");
    592   1.1     pooka 		return EINVAL;
    593   1.1     pooka 	}
    594   1.1     pooka 
    595   1.1     pooka 	/* advance to port number & parse */
    596   1.1     pooka 	p++;
    597   1.1     pooka 	l = strspn(p, "0123456789");
    598   1.1     pooka 	if (l == 0) {
    599   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: port now found: %s\n", p);
    600   1.1     pooka 		return EINVAL;
    601   1.1     pooka 	}
    602   1.1     pooka 	strncpy(buf, p, l);
    603   1.1     pooka 	buf[l] = '\0';
    604   1.1     pooka 
    605   1.1     pooka 	if (*(p+l) != '/' && *(p+l) != '\0') {
    606   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: junk at end of port: %s\n", addr);
    607   1.1     pooka 		return EINVAL;
    608   1.1     pooka 	}
    609   1.1     pooka 
    610   1.1     pooka 	port = atoi(buf);
    611   1.1     pooka 	if (port < 0 || port >= (1<<(8*sizeof(in_port_t)))) {
    612   1.1     pooka 		fprintf(stderr, "rump_sp_tcp: port %d out of range\n", port);
    613   1.1     pooka 		return ERANGE;
    614   1.1     pooka 	}
    615   1.1     pooka 	sin.sin_port = htons(port);
    616   1.1     pooka 
    617   1.1     pooka 	*sa = malloc(sizeof(sin));
    618   1.1     pooka 	if (*sa == NULL)
    619   1.1     pooka 		return errno;
    620   1.1     pooka 	memcpy(*sa, &sin, sizeof(sin));
    621   1.1     pooka 	return 0;
    622   1.1     pooka }
    623   1.1     pooka 
    624   1.1     pooka static int
    625   1.1     pooka tcp_connecthook(int s)
    626   1.1     pooka {
    627   1.1     pooka 	int x;
    628   1.1     pooka 
    629   1.1     pooka 	x = 1;
    630  1.20     pooka 	host_setsockopt(s, IPPROTO_TCP, TCP_NODELAY, &x, sizeof(x));
    631   1.1     pooka 
    632   1.1     pooka 	return 0;
    633   1.1     pooka }
    634   1.1     pooka 
    635  1.27     pooka static char parsedurl[256];
    636  1.27     pooka 
    637   1.5     pooka /*ARGSUSED*/
    638   1.2     pooka static int
    639   1.2     pooka unix_parse(const char *addr, struct sockaddr **sa, int allow_wildcard)
    640   1.2     pooka {
    641  1.34     pooka 	struct sockaddr_un s_un;
    642   1.5     pooka 	size_t slen;
    643  1.27     pooka 	int savepath = 0;
    644   1.2     pooka 
    645  1.34     pooka 	if (strlen(addr) >= sizeof(s_un.sun_path))
    646   1.2     pooka 		return ENAMETOOLONG;
    647   1.2     pooka 
    648   1.2     pooka 	/*
    649   1.2     pooka 	 * The pathname can be all kinds of spaghetti elementals,
    650  1.15     pooka 	 * so meek and obidient we accept everything.  However, use
    651  1.15     pooka 	 * full path for easy cleanup in case someone gives a relative
    652  1.15     pooka 	 * one and the server does a chdir() between now than the
    653  1.15     pooka 	 * cleanup.
    654   1.2     pooka 	 */
    655  1.34     pooka 	memset(&s_un, 0, sizeof(s_un));
    656  1.34     pooka 	s_un.sun_family = AF_LOCAL;
    657  1.15     pooka 	if (*addr != '/') {
    658  1.15     pooka 		char mywd[PATH_MAX];
    659  1.15     pooka 
    660  1.15     pooka 		if (getcwd(mywd, sizeof(mywd)) == NULL) {
    661  1.15     pooka 			fprintf(stderr, "warning: cannot determine cwd, "
    662  1.15     pooka 			    "omitting socket cleanup\n");
    663  1.15     pooka 		} else {
    664  1.34     pooka 			if (strlen(addr)+strlen(mywd)+1
    665  1.34     pooka 			    >= sizeof(s_un.sun_path))
    666  1.15     pooka 				return ENAMETOOLONG;
    667  1.34     pooka 			strcpy(s_un.sun_path, mywd);
    668  1.34     pooka 			strcat(s_un.sun_path, "/");
    669  1.27     pooka 			savepath = 1;
    670  1.15     pooka 		}
    671  1.15     pooka 	}
    672  1.34     pooka 	strcat(s_un.sun_path, addr);
    673  1.36     pooka #if defined(__linux__) || defined(__sun__) || defined(__CYGWIN__)
    674  1.34     pooka 	slen = sizeof(s_un);
    675  1.32     pooka #else
    676  1.34     pooka 	s_un.sun_len = SUN_LEN(&s_un);
    677  1.34     pooka 	slen = s_un.sun_len+1; /* get the 0 too */
    678  1.32     pooka #endif
    679   1.2     pooka 
    680  1.27     pooka 	if (savepath && *parsedurl == '\0') {
    681  1.27     pooka 		snprintf(parsedurl, sizeof(parsedurl),
    682  1.34     pooka 		    "unix://%s", s_un.sun_path);
    683  1.27     pooka 	}
    684  1.27     pooka 
    685   1.5     pooka 	*sa = malloc(slen);
    686   1.2     pooka 	if (*sa == NULL)
    687   1.2     pooka 		return errno;
    688  1.34     pooka 	memcpy(*sa, &s_un, slen);
    689   1.2     pooka 
    690   1.2     pooka 	return 0;
    691   1.2     pooka }
    692   1.2     pooka 
    693  1.15     pooka static void
    694  1.15     pooka unix_cleanup(struct sockaddr *sa)
    695  1.15     pooka {
    696  1.34     pooka 	struct sockaddr_un *s_sun = (void *)sa;
    697  1.15     pooka 
    698  1.15     pooka 	/*
    699  1.15     pooka 	 * cleanup only absolute paths.  see unix_parse() above
    700  1.15     pooka 	 */
    701  1.34     pooka 	if (*s_sun->sun_path == '/') {
    702  1.34     pooka 		unlink(s_sun->sun_path);
    703  1.15     pooka 	}
    704  1.15     pooka }
    705  1.15     pooka 
    706   1.1     pooka /*ARGSUSED*/
    707   1.1     pooka static int
    708   1.1     pooka notsupp(void)
    709   1.1     pooka {
    710   1.1     pooka 
    711   1.1     pooka 	fprintf(stderr, "rump_sp: support not yet implemented\n");
    712   1.1     pooka 	return EOPNOTSUPP;
    713   1.1     pooka }
    714   1.1     pooka 
    715   1.1     pooka static int
    716   1.1     pooka success(void)
    717   1.1     pooka {
    718   1.1     pooka 
    719   1.1     pooka 	return 0;
    720   1.1     pooka }
    721   1.1     pooka 
    722  1.37     pooka static struct {
    723   1.1     pooka 	const char *id;
    724   1.1     pooka 	int domain;
    725  1.32     pooka 	socklen_t slen;
    726   1.1     pooka 	addrparse_fn ap;
    727   1.1     pooka 	connecthook_fn connhook;
    728  1.15     pooka 	cleanup_fn cleanup;
    729   1.1     pooka } parsetab[] = {
    730  1.32     pooka 	{ "tcp", PF_INET, sizeof(struct sockaddr_in),
    731  1.32     pooka 	    tcp_parse, tcp_connecthook, (cleanup_fn)success },
    732  1.32     pooka 	{ "unix", PF_LOCAL, sizeof(struct sockaddr_un),
    733  1.32     pooka 	    unix_parse, (connecthook_fn)success, unix_cleanup },
    734  1.32     pooka 	{ "tcp6", PF_INET6, sizeof(struct sockaddr_in6),
    735  1.32     pooka 	    (addrparse_fn)notsupp, (connecthook_fn)success,
    736  1.32     pooka 	    (cleanup_fn)success },
    737   1.1     pooka };
    738   1.1     pooka #define NPARSE (sizeof(parsetab)/sizeof(parsetab[0]))
    739   1.1     pooka 
    740   1.1     pooka static int
    741   1.1     pooka parseurl(const char *url, struct sockaddr **sap, unsigned *idxp,
    742   1.1     pooka 	int allow_wildcard)
    743   1.1     pooka {
    744   1.1     pooka 	char id[16];
    745   1.1     pooka 	const char *p, *p2;
    746   1.1     pooka 	size_t l;
    747   1.1     pooka 	unsigned i;
    748   1.1     pooka 	int error;
    749   1.1     pooka 
    750   1.1     pooka 	/*
    751   1.1     pooka 	 * Parse the url
    752   1.1     pooka 	 */
    753   1.1     pooka 
    754   1.1     pooka 	p = url;
    755   1.1     pooka 	p2 = strstr(p, "://");
    756   1.1     pooka 	if (!p2) {
    757   1.1     pooka 		fprintf(stderr, "rump_sp: invalid locator ``%s''\n", p);
    758   1.1     pooka 		return EINVAL;
    759   1.1     pooka 	}
    760   1.1     pooka 	l = p2-p;
    761   1.1     pooka 	if (l > sizeof(id)-1) {
    762   1.1     pooka 		fprintf(stderr, "rump_sp: identifier too long in ``%s''\n", p);
    763   1.1     pooka 		return EINVAL;
    764   1.1     pooka 	}
    765   1.1     pooka 
    766   1.1     pooka 	strncpy(id, p, l);
    767   1.1     pooka 	id[l] = '\0';
    768   1.1     pooka 	p2 += 3; /* beginning of address */
    769   1.1     pooka 
    770   1.1     pooka 	for (i = 0; i < NPARSE; i++) {
    771   1.1     pooka 		if (strcmp(id, parsetab[i].id) == 0) {
    772   1.1     pooka 			error = parsetab[i].ap(p2, sap, allow_wildcard);
    773   1.1     pooka 			if (error)
    774   1.1     pooka 				return error;
    775   1.1     pooka 			break;
    776   1.1     pooka 		}
    777   1.1     pooka 	}
    778   1.1     pooka 	if (i == NPARSE) {
    779   1.1     pooka 		fprintf(stderr, "rump_sp: invalid identifier ``%s''\n", p);
    780   1.1     pooka 		return EINVAL;
    781   1.1     pooka 	}
    782   1.1     pooka 
    783   1.1     pooka 	*idxp = i;
    784   1.1     pooka 	return 0;
    785   1.1     pooka }
    786