Home | History | Annotate | Line # | Download | only in librumpuser
sp_common.c revision 1.12
      1  1.12  pooka /*      $NetBSD: sp_common.c,v 1.12 2010/11/26 18:51:03 pooka Exp $	*/
      2   1.1  pooka 
      3   1.1  pooka /*
      4   1.1  pooka  * Copyright (c) 2010 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.1  pooka #include <sys/cdefs.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.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.1  pooka #include <errno.h>
     46   1.1  pooka #include <fcntl.h>
     47   1.1  pooka #include <poll.h>
     48   1.4  pooka #include <pthread.h>
     49   1.1  pooka #include <stdarg.h>
     50  1.11  pooka #include <stddef.h>
     51   1.1  pooka #include <stdio.h>
     52   1.1  pooka #include <stdlib.h>
     53   1.1  pooka #include <string.h>
     54   1.1  pooka #include <unistd.h>
     55   1.1  pooka 
     56   1.1  pooka //#define DEBUG
     57   1.1  pooka #ifdef DEBUG
     58   1.1  pooka #define DPRINTF(x) mydprintf x
     59   1.1  pooka static void
     60   1.1  pooka mydprintf(const char *fmt, ...)
     61   1.1  pooka {
     62   1.1  pooka 	va_list ap;
     63   1.1  pooka 
     64   1.1  pooka 	va_start(ap, fmt);
     65   1.1  pooka 	vfprintf(stderr, fmt, ap);
     66   1.1  pooka 	va_end(ap);
     67   1.1  pooka }
     68   1.1  pooka #else
     69   1.1  pooka #define DPRINTF(x)
     70   1.1  pooka #endif
     71   1.1  pooka 
     72   1.1  pooka /*
     73   1.1  pooka  * Bah, I hate writing on-off-wire conversions in C
     74   1.1  pooka  */
     75   1.1  pooka 
     76   1.4  pooka enum { RUMPSP_REQ, RUMPSP_RESP };
     77  1.10  pooka enum {	RUMPSP_SYSCALL,
     78  1.10  pooka 	RUMPSP_COPYIN, RUMPSP_COPYINSTR,
     79  1.10  pooka 	RUMPSP_COPYOUT, RUMPSP_COPYOUTSTR,
     80  1.10  pooka 	RUMPSP_ANONMMAP };
     81   1.1  pooka 
     82   1.1  pooka struct rsp_hdr {
     83   1.1  pooka 	uint64_t rsp_len;
     84   1.1  pooka 	uint64_t rsp_reqno;
     85   1.4  pooka 	uint16_t rsp_class;
     86   1.4  pooka 	uint16_t rsp_type;
     87   1.1  pooka 	/*
     88   1.1  pooka 	 * We want this structure 64bit-aligned for typecast fun,
     89   1.1  pooka 	 * so might as well use the following for something.
     90   1.1  pooka 	 */
     91   1.1  pooka 	uint32_t rsp_sysnum;
     92   1.1  pooka };
     93   1.1  pooka #define HDRSZ sizeof(struct rsp_hdr)
     94   1.1  pooka 
     95   1.1  pooka /*
     96   1.1  pooka  * Data follows the header.  We have two types of structured data.
     97   1.1  pooka  */
     98   1.1  pooka 
     99   1.1  pooka /* copyin/copyout */
    100   1.1  pooka struct rsp_copydata {
    101   1.1  pooka 	size_t rcp_len;
    102   1.1  pooka 	void *rcp_addr;
    103   1.1  pooka 	uint8_t rcp_data[0];
    104   1.1  pooka };
    105   1.1  pooka 
    106   1.1  pooka /* syscall response */
    107   1.1  pooka struct rsp_sysresp {
    108   1.1  pooka 	int rsys_error;
    109   1.1  pooka 	register_t rsys_retval[2];
    110   1.1  pooka };
    111   1.1  pooka 
    112   1.4  pooka struct respwait {
    113   1.4  pooka 	uint64_t rw_reqno;
    114   1.4  pooka 	void *rw_data;
    115   1.4  pooka 	size_t rw_dlen;
    116   1.4  pooka 
    117   1.4  pooka 	pthread_cond_t rw_cv;
    118   1.4  pooka 
    119   1.4  pooka 	TAILQ_ENTRY(respwait) rw_entries;
    120   1.4  pooka };
    121   1.1  pooka 
    122   1.1  pooka struct spclient {
    123   1.1  pooka 	int spc_fd;
    124   1.7  pooka 	int spc_refcnt;
    125   1.7  pooka 	int spc_dying;
    126   1.6  pooka 
    127  1.11  pooka 	pthread_mutex_t spc_mtx;
    128  1.11  pooka 	pthread_cond_t spc_cv;
    129  1.11  pooka 
    130   1.6  pooka 	struct lwp *spc_mainlwp;
    131   1.6  pooka 	pid_t spc_pid;
    132   1.1  pooka 
    133  1.11  pooka 	TAILQ_HEAD(, respwait) spc_respwait;
    134  1.11  pooka 
    135  1.11  pooka 	/* rest of the fields are zeroed upon disconnect */
    136  1.12  pooka #define SPC_ZEROFF offsetof(struct spclient, spc_pfd)
    137   1.7  pooka 	struct pollfd *spc_pfd;
    138   1.7  pooka 
    139   1.1  pooka 	struct rsp_hdr spc_hdr;
    140   1.1  pooka 	uint8_t *spc_buf;
    141   1.1  pooka 	size_t spc_off;
    142   1.1  pooka 
    143   1.4  pooka 	uint64_t spc_nextreq;
    144   1.4  pooka 	int spc_ostatus, spc_istatus;
    145   1.1  pooka };
    146   1.4  pooka #define SPCSTATUS_FREE 0
    147   1.4  pooka #define SPCSTATUS_BUSY 1
    148   1.4  pooka #define SPCSTATUS_WANTED 2
    149   1.1  pooka 
    150   1.1  pooka typedef int (*addrparse_fn)(const char *, struct sockaddr **, int);
    151   1.1  pooka typedef int (*connecthook_fn)(int);
    152   1.1  pooka 
    153   1.4  pooka static int readframe(struct spclient *);
    154   1.4  pooka static void handlereq(struct spclient *);
    155   1.4  pooka 
    156   1.4  pooka static void
    157  1.12  pooka sendlockl(struct spclient *spc)
    158   1.4  pooka {
    159   1.4  pooka 
    160  1.12  pooka 	/* assert(pthread_mutex_owned) */
    161   1.4  pooka 	while (spc->spc_ostatus != SPCSTATUS_FREE) {
    162   1.4  pooka 		spc->spc_ostatus = SPCSTATUS_WANTED;
    163   1.4  pooka 		pthread_cond_wait(&spc->spc_cv, &spc->spc_mtx);
    164   1.4  pooka 	}
    165   1.4  pooka 	spc->spc_ostatus = SPCSTATUS_BUSY;
    166  1.12  pooka }
    167  1.12  pooka 
    168  1.12  pooka static void
    169  1.12  pooka sendlock(struct spclient *spc)
    170  1.12  pooka {
    171  1.12  pooka 
    172  1.12  pooka 	pthread_mutex_lock(&spc->spc_mtx);
    173  1.12  pooka 	sendlockl(spc);
    174   1.4  pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    175   1.4  pooka }
    176   1.4  pooka 
    177   1.4  pooka static void
    178  1.12  pooka sendunlockl(struct spclient *spc)
    179   1.4  pooka {
    180   1.4  pooka 
    181  1.12  pooka 	/* assert(pthread_mutex_owned) */
    182   1.4  pooka 	if (spc->spc_ostatus == SPCSTATUS_WANTED)
    183   1.4  pooka 		pthread_cond_broadcast(&spc->spc_cv);
    184   1.4  pooka 	spc->spc_ostatus = SPCSTATUS_FREE;
    185  1.12  pooka }
    186  1.12  pooka 
    187  1.12  pooka static void
    188  1.12  pooka sendunlock(struct spclient *spc)
    189  1.12  pooka {
    190  1.12  pooka 
    191  1.12  pooka 	pthread_mutex_lock(&spc->spc_mtx);
    192  1.12  pooka 	sendunlockl(spc);
    193   1.4  pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    194   1.4  pooka }
    195   1.1  pooka 
    196   1.1  pooka static int
    197   1.1  pooka dosend(struct spclient *spc, const void *data, size_t dlen)
    198   1.1  pooka {
    199   1.1  pooka 	struct pollfd pfd;
    200   1.1  pooka 	const uint8_t *sdata = data;
    201   1.1  pooka 	ssize_t n;
    202   1.1  pooka 	size_t sent;
    203   1.1  pooka 	int fd = spc->spc_fd;
    204   1.1  pooka 
    205   1.1  pooka 	pfd.fd = fd;
    206   1.1  pooka 	pfd.events = POLLOUT;
    207   1.1  pooka 
    208   1.1  pooka 	for (sent = 0, n = 0; sent < dlen; ) {
    209   1.1  pooka 		if (n) {
    210   1.1  pooka 			if (poll(&pfd, 1, INFTIM) == -1) {
    211   1.1  pooka 				if (errno == EINTR)
    212   1.1  pooka 					continue;
    213   1.1  pooka 				return errno;
    214   1.1  pooka 			}
    215   1.1  pooka 		}
    216   1.1  pooka 
    217   1.3  pooka 		n = send(fd, sdata + sent, dlen - sent, MSG_NOSIGNAL);
    218   1.1  pooka 		if (n == 0) {
    219   1.1  pooka 			return EFAULT;
    220   1.1  pooka 		}
    221  1.10  pooka 		if (n == -1)  {
    222  1.10  pooka 			if (errno != EAGAIN)
    223  1.10  pooka 				return EFAULT;
    224  1.10  pooka 			continue;
    225   1.1  pooka 		}
    226   1.1  pooka 		sent += n;
    227   1.1  pooka 	}
    228   1.1  pooka 
    229   1.1  pooka 	return 0;
    230   1.1  pooka }
    231   1.1  pooka 
    232   1.4  pooka static void
    233   1.4  pooka putwait(struct spclient *spc, struct respwait *rw, struct rsp_hdr *rhdr)
    234   1.4  pooka {
    235   1.4  pooka 
    236   1.4  pooka 	rw->rw_data = NULL;
    237   1.4  pooka 	rw->rw_dlen = 0;
    238   1.4  pooka 	pthread_cond_init(&rw->rw_cv, NULL);
    239   1.4  pooka 
    240   1.4  pooka 	pthread_mutex_lock(&spc->spc_mtx);
    241   1.4  pooka 	rw->rw_reqno = rhdr->rsp_reqno = spc->spc_nextreq++;
    242   1.4  pooka 	TAILQ_INSERT_TAIL(&spc->spc_respwait, rw, rw_entries);
    243  1.12  pooka 
    244  1.12  pooka 	sendlockl(spc);
    245   1.8  pooka }
    246   1.8  pooka 
    247   1.8  pooka static void
    248   1.8  pooka unputwait(struct spclient *spc, struct respwait *rw)
    249   1.8  pooka {
    250   1.8  pooka 
    251  1.12  pooka 	sendunlockl(spc);
    252  1.12  pooka 
    253   1.8  pooka 	TAILQ_REMOVE(&spc->spc_respwait, rw, rw_entries);
    254   1.4  pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    255   1.8  pooka 	pthread_cond_destroy(&rw->rw_cv);
    256   1.4  pooka }
    257   1.4  pooka 
    258   1.4  pooka static void
    259   1.4  pooka kickwaiter(struct spclient *spc)
    260   1.4  pooka {
    261   1.4  pooka 	struct respwait *rw;
    262   1.4  pooka 
    263   1.4  pooka 	pthread_mutex_lock(&spc->spc_mtx);
    264   1.4  pooka 	TAILQ_FOREACH(rw, &spc->spc_respwait, rw_entries) {
    265   1.4  pooka 		if (rw->rw_reqno == spc->spc_hdr.rsp_reqno)
    266   1.4  pooka 			break;
    267   1.4  pooka 	}
    268   1.4  pooka 	if (rw == NULL) {
    269   1.4  pooka 		printf("PANIC: no waiter\n");
    270   1.6  pooka 		abort();
    271   1.4  pooka 		return;
    272   1.4  pooka 	}
    273  1.10  pooka 	DPRINTF(("rump_sp: client %p woke up waiter at %p\n", spc, rw));
    274   1.4  pooka 	rw->rw_data = spc->spc_buf;
    275  1.11  pooka 	rw->rw_dlen = (size_t)(spc->spc_off - HDRSZ);
    276   1.4  pooka 	pthread_cond_signal(&rw->rw_cv);
    277   1.4  pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    278   1.4  pooka 
    279   1.4  pooka 	spc->spc_buf = NULL;
    280   1.4  pooka 	spc->spc_off = 0;
    281   1.4  pooka }
    282   1.4  pooka 
    283   1.4  pooka static void
    284   1.4  pooka kickall(struct spclient *spc)
    285   1.4  pooka {
    286   1.4  pooka 	struct respwait *rw;
    287   1.4  pooka 
    288   1.4  pooka 	/* DIAGASSERT(mutex_owned(spc_lock)) */
    289   1.4  pooka 	TAILQ_FOREACH(rw, &spc->spc_respwait, rw_entries)
    290  1.12  pooka 		pthread_cond_broadcast(&rw->rw_cv);
    291   1.4  pooka }
    292   1.4  pooka 
    293   1.4  pooka static int
    294   1.4  pooka waitresp(struct spclient *spc, struct respwait *rw)
    295   1.4  pooka {
    296   1.4  pooka 	struct pollfd pfd;
    297   1.4  pooka 	int rv = 0;
    298   1.4  pooka 
    299  1.12  pooka 	sendunlockl(spc);
    300  1.12  pooka 
    301   1.7  pooka 	while (rw->rw_data == NULL && spc->spc_dying == 0) {
    302   1.4  pooka 		/* are we free to receive? */
    303   1.4  pooka 		if (spc->spc_istatus == SPCSTATUS_FREE) {
    304   1.4  pooka 			int gotresp;
    305   1.4  pooka 
    306   1.4  pooka 			spc->spc_istatus = SPCSTATUS_BUSY;
    307   1.4  pooka 			pthread_mutex_unlock(&spc->spc_mtx);
    308   1.4  pooka 
    309   1.4  pooka 			pfd.fd = spc->spc_fd;
    310   1.4  pooka 			pfd.events = POLLIN;
    311   1.4  pooka 
    312   1.4  pooka 			for (gotresp = 0; !gotresp; ) {
    313   1.9  pooka 				switch (readframe(spc)) {
    314   1.8  pooka 				case 0:
    315   1.4  pooka 					poll(&pfd, 1, INFTIM);
    316   1.8  pooka 					continue;
    317   1.8  pooka 				case -1:
    318   1.9  pooka 					rv = errno;
    319   1.8  pooka 					spc->spc_dying = 1;
    320  1.11  pooka 					goto cleanup;
    321   1.8  pooka 				default:
    322   1.8  pooka 					break;
    323   1.8  pooka 				}
    324   1.4  pooka 
    325   1.4  pooka 				switch (spc->spc_hdr.rsp_class) {
    326   1.4  pooka 				case RUMPSP_RESP:
    327   1.4  pooka 					kickwaiter(spc);
    328   1.4  pooka 					gotresp = spc->spc_hdr.rsp_reqno ==
    329   1.4  pooka 					    rw->rw_reqno;
    330   1.4  pooka 					break;
    331   1.4  pooka 				case RUMPSP_REQ:
    332   1.4  pooka 					handlereq(spc);
    333   1.4  pooka 					break;
    334   1.4  pooka 				default:
    335   1.4  pooka 					/* panic */
    336   1.4  pooka 					break;
    337   1.4  pooka 				}
    338   1.4  pooka 			}
    339  1.11  pooka  cleanup:
    340   1.4  pooka 			pthread_mutex_lock(&spc->spc_mtx);
    341   1.4  pooka 			if (spc->spc_istatus == SPCSTATUS_WANTED)
    342   1.4  pooka 				kickall(spc);
    343   1.4  pooka 			spc->spc_istatus = SPCSTATUS_FREE;
    344   1.4  pooka 		} else {
    345   1.4  pooka 			spc->spc_istatus = SPCSTATUS_WANTED;
    346   1.4  pooka 			pthread_cond_wait(&rw->rw_cv, &spc->spc_mtx);
    347   1.4  pooka 		}
    348   1.4  pooka 	}
    349   1.4  pooka 
    350   1.4  pooka 	TAILQ_REMOVE(&spc->spc_respwait, rw, rw_entries);
    351   1.4  pooka 	pthread_mutex_unlock(&spc->spc_mtx);
    352   1.4  pooka 
    353   1.4  pooka 	pthread_cond_destroy(&rw->rw_cv);
    354   1.9  pooka 
    355  1.11  pooka 	if (rv == 0 && spc->spc_dying)
    356  1.11  pooka 		rv = ENOTCONN;
    357   1.4  pooka 	return rv;
    358   1.4  pooka }
    359   1.4  pooka 
    360   1.1  pooka static int
    361   1.1  pooka readframe(struct spclient *spc)
    362   1.1  pooka {
    363   1.1  pooka 	int fd = spc->spc_fd;
    364   1.1  pooka 	size_t left;
    365   1.1  pooka 	size_t framelen;
    366   1.1  pooka 	ssize_t n;
    367   1.1  pooka 
    368   1.1  pooka 	/* still reading header? */
    369   1.1  pooka 	if (spc->spc_off < HDRSZ) {
    370   1.1  pooka 		DPRINTF(("rump_sp: readframe getting header at offset %zu\n",
    371   1.1  pooka 		    spc->spc_off));
    372   1.1  pooka 
    373   1.1  pooka 		left = HDRSZ - spc->spc_off;
    374   1.1  pooka 		/*LINTED: cast ok */
    375   1.1  pooka 		n = read(fd, (uint8_t *)&spc->spc_hdr + spc->spc_off, left);
    376   1.1  pooka 		if (n == 0) {
    377   1.1  pooka 			return -1;
    378   1.1  pooka 		}
    379   1.1  pooka 		if (n == -1) {
    380   1.1  pooka 			if (errno == EAGAIN)
    381   1.1  pooka 				return 0;
    382   1.1  pooka 			return -1;
    383   1.1  pooka 		}
    384   1.1  pooka 
    385   1.1  pooka 		spc->spc_off += n;
    386   1.1  pooka 		if (spc->spc_off < HDRSZ)
    387   1.1  pooka 			return -1;
    388   1.1  pooka 
    389   1.1  pooka 		/*LINTED*/
    390   1.1  pooka 		framelen = spc->spc_hdr.rsp_len;
    391   1.1  pooka 
    392   1.1  pooka 		if (framelen < HDRSZ) {
    393   1.1  pooka 			return -1;
    394   1.1  pooka 		} else if (framelen == HDRSZ) {
    395   1.1  pooka 			return 1;
    396   1.1  pooka 		}
    397   1.1  pooka 
    398   1.1  pooka 		spc->spc_buf = malloc(framelen - HDRSZ);
    399   1.1  pooka 		if (spc->spc_buf == NULL) {
    400   1.1  pooka 			return -1;
    401   1.1  pooka 		}
    402   1.1  pooka 		memset(spc->spc_buf, 0, framelen - HDRSZ);
    403   1.1  pooka 
    404   1.1  pooka 		/* "fallthrough" */
    405   1.1  pooka 	} else {
    406   1.1  pooka 		/*LINTED*/
    407   1.1  pooka 		framelen = spc->spc_hdr.rsp_len;
    408   1.1  pooka 	}
    409   1.1  pooka 
    410   1.1  pooka 	left = framelen - spc->spc_off;
    411   1.1  pooka 
    412   1.1  pooka 	DPRINTF(("rump_sp: readframe getting body at offset %zu, left %zu\n",
    413   1.1  pooka 	    spc->spc_off, left));
    414   1.1  pooka 
    415   1.1  pooka 	if (left == 0)
    416   1.1  pooka 		return 1;
    417   1.1  pooka 	n = read(fd, spc->spc_buf + (spc->spc_off - HDRSZ), left);
    418   1.1  pooka 	if (n == 0) {
    419   1.1  pooka 		return -1;
    420   1.1  pooka 	}
    421   1.1  pooka 	if (n == -1) {
    422   1.1  pooka 		if (errno == EAGAIN)
    423   1.1  pooka 			return 0;
    424   1.1  pooka 		return -1;
    425   1.1  pooka 	}
    426   1.1  pooka 	spc->spc_off += n;
    427   1.1  pooka 	left -= n;
    428   1.1  pooka 
    429   1.1  pooka 	/* got everything? */
    430   1.1  pooka 	if (left == 0)
    431   1.1  pooka 		return 1;
    432   1.1  pooka 	else
    433   1.1  pooka 		return 0;
    434   1.1  pooka }
    435   1.1  pooka 
    436   1.1  pooka static int
    437   1.1  pooka tcp_parse(const char *addr, struct sockaddr **sa, int allow_wildcard)
    438   1.1  pooka {
    439   1.1  pooka 	struct sockaddr_in sin;
    440   1.1  pooka 	char buf[64];
    441   1.1  pooka 	const char *p;
    442   1.1  pooka 	size_t l;
    443   1.1  pooka 	int port;
    444   1.1  pooka 
    445   1.1  pooka 	memset(&sin, 0, sizeof(sin));
    446   1.1  pooka 	sin.sin_len = sizeof(sin);
    447   1.1  pooka 	sin.sin_family = AF_INET;
    448   1.1  pooka 
    449   1.1  pooka 	p = strchr(addr, ':');
    450   1.1  pooka 	if (!p) {
    451   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: missing port specifier\n");
    452   1.1  pooka 		return EINVAL;
    453   1.1  pooka 	}
    454   1.1  pooka 
    455   1.1  pooka 	l = p - addr;
    456   1.1  pooka 	if (l > sizeof(buf)-1) {
    457   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: address too long\n");
    458   1.1  pooka 		return EINVAL;
    459   1.1  pooka 	}
    460   1.1  pooka 	strncpy(buf, addr, l);
    461   1.1  pooka 	buf[l] = '\0';
    462   1.1  pooka 
    463   1.1  pooka 	/* special INADDR_ANY treatment */
    464   1.1  pooka 	if (strcmp(buf, "*") == 0 || strcmp(buf, "0") == 0) {
    465   1.1  pooka 		sin.sin_addr.s_addr = INADDR_ANY;
    466   1.1  pooka 	} else {
    467   1.1  pooka 		switch (inet_pton(AF_INET, buf, &sin.sin_addr)) {
    468   1.1  pooka 		case 1:
    469   1.1  pooka 			break;
    470   1.1  pooka 		case 0:
    471   1.1  pooka 			fprintf(stderr, "rump_sp_tcp: cannot parse %s\n", buf);
    472   1.1  pooka 			return EINVAL;
    473   1.1  pooka 		case -1:
    474   1.1  pooka 			fprintf(stderr, "rump_sp_tcp: inet_pton failed\n");
    475   1.1  pooka 			return errno;
    476   1.1  pooka 		default:
    477   1.1  pooka 			assert(/*CONSTCOND*/0);
    478   1.1  pooka 			return EINVAL;
    479   1.1  pooka 		}
    480   1.1  pooka 	}
    481   1.1  pooka 
    482   1.1  pooka 	if (!allow_wildcard && sin.sin_addr.s_addr == INADDR_ANY) {
    483   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: client needs !INADDR_ANY\n");
    484   1.1  pooka 		return EINVAL;
    485   1.1  pooka 	}
    486   1.1  pooka 
    487   1.1  pooka 	/* advance to port number & parse */
    488   1.1  pooka 	p++;
    489   1.1  pooka 	l = strspn(p, "0123456789");
    490   1.1  pooka 	if (l == 0) {
    491   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: port now found: %s\n", p);
    492   1.1  pooka 		return EINVAL;
    493   1.1  pooka 	}
    494   1.1  pooka 	strncpy(buf, p, l);
    495   1.1  pooka 	buf[l] = '\0';
    496   1.1  pooka 
    497   1.1  pooka 	if (*(p+l) != '/' && *(p+l) != '\0') {
    498   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: junk at end of port: %s\n", addr);
    499   1.1  pooka 		return EINVAL;
    500   1.1  pooka 	}
    501   1.1  pooka 
    502   1.1  pooka 	port = atoi(buf);
    503   1.1  pooka 	if (port < 0 || port >= (1<<(8*sizeof(in_port_t)))) {
    504   1.1  pooka 		fprintf(stderr, "rump_sp_tcp: port %d out of range\n", port);
    505   1.1  pooka 		return ERANGE;
    506   1.1  pooka 	}
    507   1.1  pooka 	sin.sin_port = htons(port);
    508   1.1  pooka 
    509   1.1  pooka 	*sa = malloc(sizeof(sin));
    510   1.1  pooka 	if (*sa == NULL)
    511   1.1  pooka 		return errno;
    512   1.1  pooka 	memcpy(*sa, &sin, sizeof(sin));
    513   1.1  pooka 	return 0;
    514   1.1  pooka }
    515   1.1  pooka 
    516   1.1  pooka static int
    517   1.1  pooka tcp_connecthook(int s)
    518   1.1  pooka {
    519   1.1  pooka 	int x;
    520   1.1  pooka 
    521   1.1  pooka 	x = 1;
    522   1.1  pooka 	setsockopt(s, IPPROTO_TCP, TCP_NODELAY, &x, sizeof(x));
    523   1.1  pooka 
    524   1.1  pooka 	return 0;
    525   1.1  pooka }
    526   1.1  pooka 
    527   1.5  pooka /*ARGSUSED*/
    528   1.2  pooka static int
    529   1.2  pooka unix_parse(const char *addr, struct sockaddr **sa, int allow_wildcard)
    530   1.2  pooka {
    531   1.2  pooka 	struct sockaddr_un sun;
    532   1.5  pooka 	size_t slen;
    533   1.2  pooka 
    534   1.2  pooka 	if (strlen(addr) > sizeof(sun.sun_path))
    535   1.2  pooka 		return ENAMETOOLONG;
    536   1.2  pooka 
    537   1.2  pooka 	/*
    538   1.2  pooka 	 * The pathname can be all kinds of spaghetti elementals,
    539   1.2  pooka 	 * so meek and obidient we accept everything.
    540   1.2  pooka 	 */
    541   1.2  pooka 	memset(&sun, 0, sizeof(sun));
    542   1.2  pooka 	sun.sun_family = AF_LOCAL;
    543   1.2  pooka 	strlcpy(sun.sun_path, addr, sizeof(sun.sun_path));
    544   1.5  pooka 	sun.sun_len = slen = SUN_LEN(&sun);
    545   1.2  pooka 
    546   1.5  pooka 	*sa = malloc(slen);
    547   1.2  pooka 	if (*sa == NULL)
    548   1.2  pooka 		return errno;
    549   1.5  pooka 	memcpy(*sa, &sun, slen);
    550   1.2  pooka 
    551   1.2  pooka 	return 0;
    552   1.2  pooka }
    553   1.2  pooka 
    554   1.1  pooka /*ARGSUSED*/
    555   1.1  pooka static int
    556   1.1  pooka notsupp(void)
    557   1.1  pooka {
    558   1.1  pooka 
    559   1.1  pooka 	fprintf(stderr, "rump_sp: support not yet implemented\n");
    560   1.1  pooka 	return EOPNOTSUPP;
    561   1.1  pooka }
    562   1.1  pooka 
    563   1.1  pooka static int
    564   1.1  pooka success(void)
    565   1.1  pooka {
    566   1.1  pooka 
    567   1.1  pooka 	return 0;
    568   1.1  pooka }
    569   1.1  pooka 
    570   1.1  pooka struct {
    571   1.1  pooka 	const char *id;
    572   1.1  pooka 	int domain;
    573   1.1  pooka 	addrparse_fn ap;
    574   1.1  pooka 	connecthook_fn connhook;
    575   1.1  pooka } parsetab[] = {
    576   1.1  pooka 	{ "tcp", PF_INET, tcp_parse, tcp_connecthook },
    577   1.2  pooka 	{ "unix", PF_LOCAL, unix_parse, (connecthook_fn)success },
    578   1.1  pooka 	{ "tcp6", PF_INET6, (addrparse_fn)notsupp, (connecthook_fn)success },
    579   1.1  pooka };
    580   1.1  pooka #define NPARSE (sizeof(parsetab)/sizeof(parsetab[0]))
    581   1.1  pooka 
    582   1.1  pooka static int
    583   1.1  pooka parseurl(const char *url, struct sockaddr **sap, unsigned *idxp,
    584   1.1  pooka 	int allow_wildcard)
    585   1.1  pooka {
    586   1.1  pooka 	char id[16];
    587   1.1  pooka 	const char *p, *p2;
    588   1.1  pooka 	size_t l;
    589   1.1  pooka 	unsigned i;
    590   1.1  pooka 	int error;
    591   1.1  pooka 
    592   1.1  pooka 	/*
    593   1.1  pooka 	 * Parse the url
    594   1.1  pooka 	 */
    595   1.1  pooka 
    596   1.1  pooka 	p = url;
    597   1.1  pooka 	p2 = strstr(p, "://");
    598   1.1  pooka 	if (!p2) {
    599   1.1  pooka 		fprintf(stderr, "rump_sp: invalid locator ``%s''\n", p);
    600   1.1  pooka 		return EINVAL;
    601   1.1  pooka 	}
    602   1.1  pooka 	l = p2-p;
    603   1.1  pooka 	if (l > sizeof(id)-1) {
    604   1.1  pooka 		fprintf(stderr, "rump_sp: identifier too long in ``%s''\n", p);
    605   1.1  pooka 		return EINVAL;
    606   1.1  pooka 	}
    607   1.1  pooka 
    608   1.1  pooka 	strncpy(id, p, l);
    609   1.1  pooka 	id[l] = '\0';
    610   1.1  pooka 	p2 += 3; /* beginning of address */
    611   1.1  pooka 
    612   1.1  pooka 	for (i = 0; i < NPARSE; i++) {
    613   1.1  pooka 		if (strcmp(id, parsetab[i].id) == 0) {
    614   1.1  pooka 			error = parsetab[i].ap(p2, sap, allow_wildcard);
    615   1.1  pooka 			if (error)
    616   1.1  pooka 				return error;
    617   1.1  pooka 			break;
    618   1.1  pooka 		}
    619   1.1  pooka 	}
    620   1.1  pooka 	if (i == NPARSE) {
    621   1.1  pooka 		fprintf(stderr, "rump_sp: invalid identifier ``%s''\n", p);
    622   1.1  pooka 		return EINVAL;
    623   1.1  pooka 	}
    624   1.1  pooka 
    625   1.1  pooka 	*idxp = i;
    626   1.1  pooka 	return 0;
    627   1.1  pooka }
    628