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