Home | History | Annotate | Line # | Download | only in mount_9p
      1  1.9     ozaki /*      $NetBSD: ninebuf.c,v 1.9 2019/10/28 02:59:25 ozaki-r Exp $	*/
      2  1.1     pooka 
      3  1.1     pooka /*
      4  1.1     pooka  * Copyright (c) 2006, 2007  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 #include <sys/cdefs.h>
     29  1.1     pooka #ifndef lint
     30  1.9     ozaki __RCSID("$NetBSD: ninebuf.c,v 1.9 2019/10/28 02:59:25 ozaki-r Exp $");
     31  1.1     pooka #endif /* !lint */
     32  1.1     pooka 
     33  1.1     pooka #include <sys/types.h>
     34  1.1     pooka #include <sys/time.h>
     35  1.1     pooka #include <sys/vnode.h>
     36  1.8  christos #include <sys/socket.h>
     37  1.1     pooka 
     38  1.1     pooka #include <err.h>
     39  1.1     pooka #include <errno.h>
     40  1.1     pooka #include <stdlib.h>
     41  1.1     pooka #include <util.h>
     42  1.1     pooka #include <unistd.h>
     43  1.1     pooka 
     44  1.1     pooka #include "ninepuffs.h"
     45  1.1     pooka 
     46  1.2     pooka #define CHECK(v) if (!(v)) abort()
     47  1.2     pooka 
     48  1.2     pooka uint8_t
     49  1.2     pooka p9pbuf_get_type(struct puffs_framebuf *pb)
     50  1.2     pooka {
     51  1.2     pooka 	uint8_t val;
     52  1.2     pooka 
     53  1.2     pooka 	puffs_framebuf_getdata_atoff(pb, 4, &val, 1);
     54  1.2     pooka 	return val;
     55  1.2     pooka }
     56  1.1     pooka 
     57  1.2     pooka uint16_t
     58  1.2     pooka p9pbuf_get_tag(struct puffs_framebuf *pb)
     59  1.1     pooka {
     60  1.2     pooka 	uint16_t val;
     61  1.1     pooka 
     62  1.2     pooka 	puffs_framebuf_getdata_atoff(pb, 5, &val, 2);
     63  1.2     pooka 	return le16toh(val);
     64  1.2     pooka }
     65  1.1     pooka 
     66  1.2     pooka static uint32_t
     67  1.2     pooka p9pbuf_get_len(struct puffs_framebuf *pb)
     68  1.2     pooka {
     69  1.2     pooka 	uint32_t val;
     70  1.1     pooka 
     71  1.2     pooka 	puffs_framebuf_getdata_atoff(pb, 0, &val, 4);
     72  1.2     pooka 	return le32toh(val);
     73  1.2     pooka }
     74  1.1     pooka 
     75  1.2     pooka #define CUROFF(pb) (puffs_framebuf_telloff(pb))
     76  1.2     pooka int
     77  1.2     pooka p9pbuf_read(struct puffs_usermount *pu, struct puffs_framebuf *pb,
     78  1.2     pooka 	int fd, int *done)
     79  1.2     pooka {
     80  1.2     pooka 	void *win;
     81  1.2     pooka 	ssize_t n;
     82  1.2     pooka 	size_t howmuch, winlen;
     83  1.2     pooka 	int lenstate;
     84  1.1     pooka 
     85  1.2     pooka  the_next_level:
     86  1.2     pooka 	if ((lenstate = (CUROFF(pb) < 4)))
     87  1.2     pooka 		howmuch = 4 - CUROFF(pb);
     88  1.2     pooka 	else
     89  1.2     pooka 		howmuch = p9pbuf_get_len(pb) - CUROFF(pb);
     90  1.1     pooka 
     91  1.2     pooka 	if (puffs_framebuf_reserve_space(pb, howmuch) == -1)
     92  1.2     pooka 		return errno;
     93  1.1     pooka 
     94  1.2     pooka 	while (howmuch) {
     95  1.2     pooka 		winlen = howmuch;
     96  1.2     pooka 		if (puffs_framebuf_getwindow(pb, CUROFF(pb), &win, &winlen)==-1)
     97  1.2     pooka 			return errno;
     98  1.2     pooka 		n = read(fd, win, winlen);
     99  1.2     pooka 		switch (n) {
    100  1.2     pooka 		case 0:
    101  1.2     pooka 			return ECONNRESET;
    102  1.2     pooka 		case -1:
    103  1.2     pooka 			if (errno == EAGAIN)
    104  1.2     pooka 				return 0;
    105  1.2     pooka 			return errno;
    106  1.2     pooka 		default:
    107  1.2     pooka 			howmuch -= n;
    108  1.2     pooka 			puffs_framebuf_seekset(pb, CUROFF(pb) + n);
    109  1.2     pooka 			break;
    110  1.2     pooka 		}
    111  1.1     pooka 	}
    112  1.1     pooka 
    113  1.2     pooka 	if (!lenstate) {
    114  1.2     pooka 		puffs_framebuf_seekset(pb, 7);
    115  1.2     pooka 		*done = 1;
    116  1.2     pooka 		return 0;
    117  1.2     pooka 	} else
    118  1.2     pooka 		goto the_next_level;
    119  1.1     pooka }
    120  1.1     pooka 
    121  1.1     pooka int
    122  1.2     pooka p9pbuf_write(struct puffs_usermount *pu, struct puffs_framebuf *pb,
    123  1.2     pooka 	int fd, int *done)
    124  1.1     pooka {
    125  1.2     pooka 	void *win;
    126  1.1     pooka 	ssize_t n;
    127  1.2     pooka 	size_t winlen, howmuch;
    128  1.1     pooka 
    129  1.2     pooka 	if (CUROFF(pb) == 0) {
    130  1.1     pooka 		uint32_t len;
    131  1.1     pooka 
    132  1.2     pooka 		len = htole32(puffs_framebuf_tellsize(pb));
    133  1.2     pooka 		puffs_framebuf_putdata_atoff(pb, 0, &len, 4);
    134  1.1     pooka 	}
    135  1.1     pooka 
    136  1.2     pooka 	howmuch = puffs_framebuf_tellsize(pb) - CUROFF(pb);
    137  1.2     pooka 	while (howmuch) {
    138  1.2     pooka 		winlen = howmuch;
    139  1.2     pooka 		if (puffs_framebuf_getwindow(pb, CUROFF(pb), &win, &winlen)==-1)
    140  1.2     pooka 			return errno;
    141  1.9     ozaki 		n = write(fd, win, winlen);
    142  1.2     pooka 		switch (n) {
    143  1.2     pooka 		case 0:
    144  1.2     pooka 			return ECONNRESET;
    145  1.2     pooka 		case -1:
    146  1.2     pooka 			if (errno == EAGAIN)
    147  1.2     pooka 				return 0;
    148  1.2     pooka 			return errno;
    149  1.2     pooka 		default:
    150  1.2     pooka 			howmuch -= n;
    151  1.2     pooka 			puffs_framebuf_seekset(pb, CUROFF(pb) + n);
    152  1.2     pooka 			break;
    153  1.2     pooka 		}
    154  1.1     pooka 	}
    155  1.1     pooka 
    156  1.2     pooka 	*done = 1;
    157  1.2     pooka 	return 0;
    158  1.2     pooka }
    159  1.2     pooka #undef CUROFF
    160  1.1     pooka 
    161  1.2     pooka int
    162  1.2     pooka p9pbuf_cmp(struct puffs_usermount *pu,
    163  1.7     pooka 	struct puffs_framebuf *c1, struct puffs_framebuf *c2, int *notresp)
    164  1.2     pooka {
    165  1.1     pooka 
    166  1.6     pooka 	return p9pbuf_get_tag(c1) != p9pbuf_get_tag(c2);
    167  1.1     pooka }
    168  1.1     pooka 
    169  1.2     pooka struct puffs_framebuf *
    170  1.2     pooka p9pbuf_makeout()
    171  1.1     pooka {
    172  1.2     pooka 	struct puffs_framebuf *pb;
    173  1.1     pooka 
    174  1.2     pooka 	pb = puffs_framebuf_make();
    175  1.2     pooka 	puffs_framebuf_seekset(pb, 4);
    176  1.1     pooka 	return pb;
    177  1.1     pooka }
    178  1.1     pooka 
    179  1.1     pooka void
    180  1.2     pooka p9pbuf_recycleout(struct puffs_framebuf *pb)
    181  1.1     pooka {
    182  1.1     pooka 
    183  1.2     pooka 	puffs_framebuf_recycle(pb);
    184  1.2     pooka 	puffs_framebuf_seekset(pb, 4);
    185  1.1     pooka }
    186  1.1     pooka 
    187  1.1     pooka void
    188  1.2     pooka p9pbuf_put_1(struct puffs_framebuf *pb, uint8_t val)
    189  1.1     pooka {
    190  1.2     pooka 	int rv;
    191  1.1     pooka 
    192  1.2     pooka 	rv = puffs_framebuf_putdata(pb, &val, 1);
    193  1.2     pooka 	CHECK(rv == 0);
    194  1.1     pooka }
    195  1.1     pooka 
    196  1.2     pooka void
    197  1.2     pooka p9pbuf_put_2(struct puffs_framebuf *pb, uint16_t val)
    198  1.1     pooka {
    199  1.2     pooka 	int rv;
    200  1.1     pooka 
    201  1.1     pooka 	HTOLE16(val);
    202  1.2     pooka 	rv = puffs_framebuf_putdata(pb, &val, 2);
    203  1.2     pooka 	CHECK(rv == 0);
    204  1.1     pooka }
    205  1.1     pooka 
    206  1.2     pooka void
    207  1.2     pooka p9pbuf_put_4(struct puffs_framebuf *pb, uint32_t val)
    208  1.1     pooka {
    209  1.2     pooka 	int rv;
    210  1.1     pooka 
    211  1.1     pooka 	HTOLE32(val);
    212  1.2     pooka 	rv = puffs_framebuf_putdata(pb, &val, 4);
    213  1.2     pooka 	CHECK(rv == 0);
    214  1.1     pooka }
    215  1.1     pooka 
    216  1.2     pooka void
    217  1.2     pooka p9pbuf_put_8(struct puffs_framebuf *pb, uint64_t val)
    218  1.1     pooka {
    219  1.2     pooka 	int rv;
    220  1.1     pooka 
    221  1.1     pooka 	HTOLE64(val);
    222  1.2     pooka 	rv = puffs_framebuf_putdata(pb, &val, 8);
    223  1.2     pooka 	CHECK(rv == 0);
    224  1.1     pooka }
    225  1.1     pooka 
    226  1.2     pooka void
    227  1.2     pooka p9pbuf_put_data(struct puffs_framebuf *pb, const void *data, uint16_t dlen)
    228  1.1     pooka {
    229  1.2     pooka 	int rv;
    230  1.1     pooka 
    231  1.1     pooka 	p9pbuf_put_2(pb, dlen);
    232  1.2     pooka 	rv = puffs_framebuf_putdata(pb, data, dlen);
    233  1.2     pooka 	CHECK(rv == 0);
    234  1.1     pooka }
    235  1.1     pooka 
    236  1.2     pooka void
    237  1.2     pooka p9pbuf_put_str(struct puffs_framebuf *pb, const char *str)
    238  1.1     pooka {
    239  1.1     pooka 
    240  1.2     pooka 	p9pbuf_put_data(pb, str, strlen(str));
    241  1.1     pooka }
    242  1.1     pooka 
    243  1.2     pooka void
    244  1.2     pooka p9pbuf_write_data(struct puffs_framebuf *pb, uint8_t *data, uint32_t dlen)
    245  1.1     pooka {
    246  1.2     pooka 	int rv;
    247  1.1     pooka 
    248  1.2     pooka 	rv = puffs_framebuf_putdata(pb, data, dlen);
    249  1.2     pooka 	CHECK(rv == 0);
    250  1.2     pooka }
    251  1.1     pooka 
    252  1.2     pooka #define ERETURN(rv) return ((rv) == -1 ? errno : 0)
    253  1.1     pooka 
    254  1.1     pooka int
    255  1.2     pooka p9pbuf_get_1(struct puffs_framebuf *pb, uint8_t *val)
    256  1.1     pooka {
    257  1.1     pooka 
    258  1.2     pooka 	ERETURN(puffs_framebuf_getdata(pb, val, 1));
    259  1.1     pooka }
    260  1.1     pooka 
    261  1.1     pooka int
    262  1.2     pooka p9pbuf_get_2(struct puffs_framebuf *pb, uint16_t *val)
    263  1.1     pooka {
    264  1.2     pooka 	int rv;
    265  1.2     pooka 
    266  1.2     pooka 	rv = puffs_framebuf_getdata(pb, val, 2);
    267  1.2     pooka 	LE16TOH(*val);
    268  1.1     pooka 
    269  1.2     pooka 	ERETURN(rv);
    270  1.1     pooka }
    271  1.1     pooka 
    272  1.1     pooka int
    273  1.2     pooka p9pbuf_get_4(struct puffs_framebuf *pb, uint32_t *val)
    274  1.1     pooka {
    275  1.2     pooka 	int rv;
    276  1.2     pooka 
    277  1.2     pooka 	rv = puffs_framebuf_getdata(pb, val, 4);
    278  1.2     pooka 	LE32TOH(*val);
    279  1.1     pooka 
    280  1.2     pooka 	ERETURN(rv);
    281  1.1     pooka }
    282  1.1     pooka 
    283  1.1     pooka int
    284  1.2     pooka p9pbuf_get_8(struct puffs_framebuf *pb, uint64_t *val)
    285  1.1     pooka {
    286  1.2     pooka 	int rv;
    287  1.2     pooka 
    288  1.2     pooka 	rv = puffs_framebuf_getdata(pb, val, 8);
    289  1.2     pooka 	LE64TOH(*val);
    290  1.1     pooka 
    291  1.2     pooka 	ERETURN(rv);
    292  1.1     pooka }
    293  1.1     pooka 
    294  1.1     pooka int
    295  1.2     pooka p9pbuf_get_data(struct puffs_framebuf *pb, uint8_t **dp, uint16_t *dlenp)
    296  1.1     pooka {
    297  1.2     pooka         uint8_t *data;
    298  1.1     pooka 	uint16_t len;
    299  1.2     pooka 	int rv;
    300  1.1     pooka 
    301  1.2     pooka 	rv = p9pbuf_get_2(pb, &len);
    302  1.2     pooka 	if (rv)
    303  1.2     pooka 		return errno;
    304  1.2     pooka 
    305  1.3     pooka         if (puffs_framebuf_remaining(pb) < len)
    306  1.2     pooka                 return EPROTO;
    307  1.2     pooka 
    308  1.1     pooka 	if (dp) {
    309  1.2     pooka 		data = emalloc(len+1);
    310  1.2     pooka 		rv = puffs_framebuf_getdata(pb, data, len);
    311  1.5     pooka 		if (rv) {
    312  1.5     pooka 			free(data);
    313  1.2     pooka 			return errno;
    314  1.5     pooka 		}
    315  1.2     pooka 		data[len] = '\0';
    316  1.2     pooka 		*dp = data;
    317  1.2     pooka 	} else
    318  1.2     pooka 		puffs_framebuf_seekset(pb, puffs_framebuf_telloff(pb)+len);
    319  1.1     pooka 
    320  1.1     pooka 	if (dlenp)
    321  1.1     pooka 		*dlenp = len;
    322  1.1     pooka 
    323  1.2     pooka 	return 0;
    324  1.1     pooka }
    325  1.1     pooka 
    326  1.1     pooka int
    327  1.2     pooka p9pbuf_read_data(struct puffs_framebuf *pb, uint8_t *buf, uint32_t dlen)
    328  1.1     pooka {
    329  1.1     pooka 
    330  1.2     pooka 	ERETURN(puffs_framebuf_getdata(pb, buf, dlen));
    331  1.1     pooka }
    332  1.1     pooka 
    333  1.1     pooka int
    334  1.2     pooka p9pbuf_get_str(struct puffs_framebuf *pb, char **dp, uint16_t *dlenp)
    335  1.1     pooka {
    336  1.1     pooka 
    337  1.1     pooka 	return p9pbuf_get_data(pb, (uint8_t **)dp, dlenp);
    338  1.1     pooka }
    339