Home | History | Annotate | Line # | Download | only in rumpvfs
rumpblk.c revision 1.39
      1 /*	$NetBSD: rumpblk.c,v 1.39 2010/05/01 14:37:53 pooka Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2009 Antti Kantee.  All Rights Reserved.
      5  *
      6  * Development of this software was supported by the
      7  * Finnish Cultural Foundation.
      8  *
      9  * Redistribution and use in source and binary forms, with or without
     10  * modification, are permitted provided that the following conditions
     11  * are met:
     12  * 1. Redistributions of source code must retain the above copyright
     13  *    notice, this list of conditions and the following disclaimer.
     14  * 2. Redistributions in binary form must reproduce the above copyright
     15  *    notice, this list of conditions and the following disclaimer in the
     16  *    documentation and/or other materials provided with the distribution.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
     19  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     21  * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
     24  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     28  * SUCH DAMAGE.
     29  */
     30 
     31 /*
     32  * Block device emulation.  Presents a block device interface and
     33  * uses rumpuser system calls to satisfy I/O requests.
     34  *
     35  * We provide fault injection.  The driver can be made to fail
     36  * I/O occasionally.
     37  *
     38  * The driver also provides an optimization for regular files by
     39  * using memory-mapped I/O.  This avoids kernel access for every
     40  * I/O operation.  It also gives finer-grained control of how to
     41  * flush data.  Additionally, in case the rump kernel dumps core,
     42  * we get way less carnage.
     43  *
     44  * However, it is quite costly in writing large amounts of
     45  * file data, since old contents cannot merely be overwritten, but
     46  * must be paged in first before replacing (i.e. r/m/w).  Ideally,
     47  * we should use directio.  The problem is that directio can fail
     48  * silently causing improper file system semantics (i.e. unflushed
     49  * data).  Therefore, default to mmap for now.  Even so, directio
     50  * _should_ be safe and can be enabled by compiling this module
     51  * with -DHAS_DIRECTIO.
     52  */
     53 
     54 #include <sys/cdefs.h>
     55 __KERNEL_RCSID(0, "$NetBSD: rumpblk.c,v 1.39 2010/05/01 14:37:53 pooka Exp $");
     56 
     57 #include <sys/param.h>
     58 #include <sys/buf.h>
     59 #include <sys/conf.h>
     60 #include <sys/condvar.h>
     61 #include <sys/disklabel.h>
     62 #include <sys/evcnt.h>
     63 #include <sys/fcntl.h>
     64 #include <sys/kmem.h>
     65 #include <sys/malloc.h>
     66 #include <sys/queue.h>
     67 #include <sys/stat.h>
     68 
     69 #include <rump/rumpuser.h>
     70 
     71 #include "rump_private.h"
     72 #include "rump_vfs_private.h"
     73 
     74 #if 0
     75 #define DPRINTF(x) printf x
     76 #else
     77 #define DPRINTF(x)
     78 #endif
     79 
     80 /* Default: 16 x 1MB windows */
     81 unsigned memwinsize = (1<<20);
     82 unsigned memwincnt = 16;
     83 
     84 #define STARTWIN(off)		((off) & ~((off_t)memwinsize-1))
     85 #define INWIN(win,off)		((win)->win_off == STARTWIN(off))
     86 #define WINSIZE(rblk, win)	(MIN((rblk->rblk_size-win->win_off),memwinsize))
     87 #define WINVALID(win)		((win)->win_off != (off_t)-1)
     88 #define WINVALIDATE(win)	((win)->win_off = (off_t)-1)
     89 struct blkwin {
     90 	off_t win_off;
     91 	void *win_mem;
     92 	int win_refcnt;
     93 
     94 	TAILQ_ENTRY(blkwin) win_lru;
     95 };
     96 
     97 #define RUMPBLK_SIZE 16
     98 static struct rblkdev {
     99 	char *rblk_path;
    100 	int rblk_fd;
    101 	int rblk_opencnt;
    102 #ifdef HAS_ODIRECT
    103 	int rblk_dfd;
    104 #endif
    105 	uint64_t rblk_size;
    106 	uint64_t rblk_hostoffset;
    107 	int rblk_ftype;
    108 
    109 	/* for mmap */
    110 	int rblk_mmflags;
    111 	kmutex_t rblk_memmtx;
    112 	kcondvar_t rblk_memcv;
    113 	TAILQ_HEAD(winlru, blkwin) rblk_lruq;
    114 	bool rblk_waiting;
    115 
    116 	struct disklabel rblk_label;
    117 } minors[RUMPBLK_SIZE];
    118 
    119 static struct evcnt ev_io_total;
    120 static struct evcnt ev_io_async;
    121 
    122 static struct evcnt ev_memblk_hits;
    123 static struct evcnt ev_memblk_busy;
    124 
    125 static struct evcnt ev_bwrite_total;
    126 static struct evcnt ev_bwrite_async;
    127 static struct evcnt ev_bread_total;
    128 
    129 dev_type_open(rumpblk_open);
    130 dev_type_close(rumpblk_close);
    131 dev_type_read(rumpblk_read);
    132 dev_type_write(rumpblk_write);
    133 dev_type_ioctl(rumpblk_ioctl);
    134 dev_type_strategy(rumpblk_strategy);
    135 dev_type_strategy(rumpblk_strategy_fail);
    136 dev_type_dump(rumpblk_dump);
    137 dev_type_size(rumpblk_size);
    138 
    139 static const struct bdevsw rumpblk_bdevsw = {
    140 	rumpblk_open, rumpblk_close, rumpblk_strategy, rumpblk_ioctl,
    141 	nodump, nosize, D_DISK
    142 };
    143 
    144 static const struct bdevsw rumpblk_bdevsw_fail = {
    145 	rumpblk_open, rumpblk_close, rumpblk_strategy_fail, rumpblk_ioctl,
    146 	nodump, nosize, D_DISK
    147 };
    148 
    149 static const struct cdevsw rumpblk_cdevsw = {
    150 	rumpblk_open, rumpblk_close, rumpblk_read, rumpblk_write,
    151 	rumpblk_ioctl, nostop, notty, nopoll, nommap, nokqfilter, D_DISK
    152 };
    153 
    154 /* fail every n out of BLKFAIL_MAX */
    155 #define BLKFAIL_MAX 10000
    156 static int blkfail;
    157 static unsigned randstate;
    158 static kmutex_t rumpblk_lock;
    159 static int sectshift = DEV_BSHIFT;
    160 
    161 static void
    162 makedefaultlabel(struct disklabel *lp, off_t size, int part)
    163 {
    164 	int i;
    165 
    166 	memset(lp, 0, sizeof(*lp));
    167 
    168 	lp->d_secperunit = size;
    169 	lp->d_secsize = 1 << sectshift;
    170 	lp->d_nsectors = size >> sectshift;
    171 	lp->d_ntracks = 1;
    172 	lp->d_ncylinders = 1;
    173 	lp->d_secpercyl = lp->d_nsectors;
    174 
    175 	/* oh dear oh dear */
    176 	strncpy(lp->d_typename, "rumpd", sizeof(lp->d_typename));
    177 	strncpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
    178 
    179 	lp->d_type = DTYPE_RUMPD;
    180 	lp->d_rpm = 11;
    181 	lp->d_interleave = 1;
    182 	lp->d_flags = 0;
    183 
    184 	/* XXX: RAW_PART handling? */
    185 	for (i = 0; i < part; i++) {
    186 		lp->d_partitions[i].p_fstype = FS_UNUSED;
    187 	}
    188 	lp->d_partitions[part].p_size = size >> sectshift;
    189 	lp->d_npartitions = part+1;
    190 	/* XXX: file system type? */
    191 
    192 	lp->d_magic = DISKMAGIC;
    193 	lp->d_magic2 = DISKMAGIC;
    194 	lp->d_checksum = 0; /* XXX */
    195 }
    196 
    197 static struct blkwin *
    198 getwindow(struct rblkdev *rblk, off_t off, int *wsize, int *error)
    199 {
    200 	struct blkwin *win;
    201 
    202 	mutex_enter(&rblk->rblk_memmtx);
    203  retry:
    204 	/* search for window */
    205 	TAILQ_FOREACH(win, &rblk->rblk_lruq, win_lru) {
    206 		if (INWIN(win, off) && WINVALID(win))
    207 			break;
    208 	}
    209 
    210 	/* found?  return */
    211 	if (win) {
    212 		ev_memblk_hits.ev_count++;
    213 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    214 		goto good;
    215 	}
    216 
    217 	/*
    218 	 * Else, create new window.  If the least recently used is not
    219 	 * currently in use, reuse that.  Otherwise we need to wait.
    220 	 */
    221 	win = TAILQ_LAST(&rblk->rblk_lruq, winlru);
    222 	if (win->win_refcnt == 0) {
    223 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    224 		mutex_exit(&rblk->rblk_memmtx);
    225 
    226 		if (WINVALID(win)) {
    227 			DPRINTF(("win %p, unmap mem %p, off 0x%" PRIx64 "\n",
    228 			    win, win->win_mem, win->win_off));
    229 			rumpuser_unmap(win->win_mem, WINSIZE(rblk, win));
    230 			WINVALIDATE(win);
    231 		}
    232 
    233 		win->win_off = STARTWIN(off);
    234 		win->win_mem = rumpuser_filemmap(rblk->rblk_fd, win->win_off,
    235 		    WINSIZE(rblk, win), rblk->rblk_mmflags, error);
    236 		DPRINTF(("win %p, off 0x%" PRIx64 ", mem %p\n",
    237 		    win, win->win_off, win->win_mem));
    238 
    239 		mutex_enter(&rblk->rblk_memmtx);
    240 		if (win->win_mem == NULL) {
    241 			WINVALIDATE(win);
    242 			TAILQ_INSERT_TAIL(&rblk->rblk_lruq, win, win_lru);
    243 			mutex_exit(&rblk->rblk_memmtx);
    244 			return NULL;
    245 		}
    246 	} else {
    247 		DPRINTF(("memwin wait\n"));
    248 		ev_memblk_busy.ev_count++;
    249 
    250 		rblk->rblk_waiting = true;
    251 		cv_wait(&rblk->rblk_memcv, &rblk->rblk_memmtx);
    252 		goto retry;
    253 	}
    254 
    255  good:
    256 	KASSERT(win);
    257 	win->win_refcnt++;
    258 	TAILQ_INSERT_HEAD(&rblk->rblk_lruq, win, win_lru);
    259 	mutex_exit(&rblk->rblk_memmtx);
    260 	*wsize = MIN(*wsize, memwinsize - (off-win->win_off));
    261 	KASSERT(*wsize);
    262 
    263 	return win;
    264 }
    265 
    266 static void
    267 putwindow(struct rblkdev *rblk, struct blkwin *win)
    268 {
    269 
    270 	mutex_enter(&rblk->rblk_memmtx);
    271 	if (--win->win_refcnt == 0 && rblk->rblk_waiting) {
    272 		rblk->rblk_waiting = false;
    273 		cv_signal(&rblk->rblk_memcv);
    274 	}
    275 	KASSERT(win->win_refcnt >= 0);
    276 	mutex_exit(&rblk->rblk_memmtx);
    277 }
    278 
    279 static void
    280 wincleanup(struct rblkdev *rblk)
    281 {
    282 	struct blkwin *win;
    283 
    284 	while ((win = TAILQ_FIRST(&rblk->rblk_lruq)) != NULL) {
    285 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    286 		if (WINVALID(win)) {
    287 			DPRINTF(("cleanup win %p addr %p\n",
    288 			    win, win->win_mem));
    289 			rumpuser_unmap(win->win_mem, WINSIZE(rblk, win));
    290 		}
    291 		kmem_free(win, sizeof(*win));
    292 	}
    293 	rblk->rblk_mmflags = 0;
    294 }
    295 
    296 int
    297 rumpblk_init(void)
    298 {
    299 	char buf[64];
    300 	devmajor_t rumpblkmaj = RUMPBLK_DEVMAJOR;
    301 	unsigned tmp;
    302 	int error, i;
    303 
    304 	mutex_init(&rumpblk_lock, MUTEX_DEFAULT, IPL_NONE);
    305 
    306 	if (rumpuser_getenv("RUMP_BLKFAIL", buf, sizeof(buf), &error) == 0) {
    307 		blkfail = strtoul(buf, NULL, 10);
    308 		/* fail everything */
    309 		if (blkfail > BLKFAIL_MAX)
    310 			blkfail = BLKFAIL_MAX;
    311 		if (rumpuser_getenv("RUMP_BLKFAIL_SEED", buf, sizeof(buf),
    312 		    &error) == 0) {
    313 			randstate = strtoul(buf, NULL, 10);
    314 		} else {
    315 			randstate = arc4random();
    316 		}
    317 		printf("rumpblk: FAULT INJECTION ACTIVE! fail %d/%d. "
    318 		    "seed %u\n", blkfail, BLKFAIL_MAX, randstate);
    319 	} else {
    320 		blkfail = 0;
    321 	}
    322 
    323 	if (rumpuser_getenv("RUMP_BLKWINSIZE", buf, sizeof(buf), &error) == 0) {
    324 		printf("rumpblk: ");
    325 		tmp = strtoul(buf, NULL, 10);
    326 		if (tmp && !(tmp & (tmp-1)))
    327 			memwinsize = tmp;
    328 		else
    329 			printf("invalid RUMP_BLKWINSIZE %d, ", tmp);
    330 		printf("using %d for memwinsize\n", memwinsize);
    331 	}
    332 	if (rumpuser_getenv("RUMP_BLKWINCOUNT", buf, sizeof(buf), &error) == 0){
    333 		printf("rumpblk: ");
    334 		tmp = strtoul(buf, NULL, 10);
    335 		if (tmp)
    336 			memwincnt = tmp;
    337 		else
    338 			printf("invalid RUMP_BLKWINCOUNT %d, ", tmp);
    339 		printf("using %d for memwincount\n", memwincnt);
    340 	}
    341 	if (rumpuser_getenv("RUMP_BLKSECTSHIFT", buf, sizeof(buf), &error)==0){
    342 		printf("rumpblk: ");
    343 		tmp = strtoul(buf, NULL, 10);
    344 		if (tmp >= DEV_BSHIFT)
    345 			sectshift = tmp;
    346 		else
    347 			printf("RUMP_BLKSECTSHIFT must be least %d (now %d), ",
    348 			   DEV_BSHIFT, tmp);
    349 		printf("using %d for sector shift (size %d)\n",
    350 		    sectshift, 1<<sectshift);
    351 	}
    352 
    353 	memset(minors, 0, sizeof(minors));
    354 	for (i = 0; i < RUMPBLK_SIZE; i++) {
    355 		mutex_init(&minors[i].rblk_memmtx, MUTEX_DEFAULT, IPL_NONE);
    356 		cv_init(&minors[i].rblk_memcv, "rblkmcv");
    357 	}
    358 
    359 	evcnt_attach_dynamic(&ev_io_total, EVCNT_TYPE_MISC, NULL,
    360 	    "rumpblk", "I/O reqs");
    361 	evcnt_attach_dynamic(&ev_io_async, EVCNT_TYPE_MISC, NULL,
    362 	    "rumpblk", "async I/O");
    363 
    364 	evcnt_attach_dynamic(&ev_bread_total, EVCNT_TYPE_MISC, NULL,
    365 	    "rumpblk", "bytes read");
    366 	evcnt_attach_dynamic(&ev_bwrite_total, EVCNT_TYPE_MISC, NULL,
    367 	    "rumpblk", "bytes written");
    368 	evcnt_attach_dynamic(&ev_bwrite_async, EVCNT_TYPE_MISC, NULL,
    369 	    "rumpblk", "bytes written async");
    370 
    371 	evcnt_attach_dynamic(&ev_memblk_hits, EVCNT_TYPE_MISC, NULL,
    372 	    "rumpblk", "window hits");
    373 	evcnt_attach_dynamic(&ev_memblk_busy, EVCNT_TYPE_MISC, NULL,
    374 	    "rumpblk", "all windows busy");
    375 
    376 	if (blkfail) {
    377 		return devsw_attach("rumpblk",
    378 		    &rumpblk_bdevsw_fail, &rumpblkmaj,
    379 		    &rumpblk_cdevsw, &rumpblkmaj);
    380 	} else {
    381 		return devsw_attach("rumpblk",
    382 		    &rumpblk_bdevsw, &rumpblkmaj,
    383 		    &rumpblk_cdevsw, &rumpblkmaj);
    384 	}
    385 }
    386 
    387 /* XXX: no deregister */
    388 int
    389 rumpblk_register(const char *path, devminor_t *dmin,
    390 	uint64_t offset, uint64_t size)
    391 {
    392 	struct rblkdev *rblk;
    393 	uint64_t flen;
    394 	size_t len;
    395 	int ftype, error, i;
    396 
    397 	/* devices might not report correct size unless they're open */
    398 	if (rumpuser_getfileinfo(path, &flen, &ftype, &error) == -1)
    399 		return error;
    400 
    401 	/* verify host file is of supported type */
    402 	if (!(ftype == RUMPUSER_FT_REG
    403 	   || ftype == RUMPUSER_FT_BLK
    404 	   || ftype == RUMPUSER_FT_CHR))
    405 		return EINVAL;
    406 
    407 	mutex_enter(&rumpblk_lock);
    408 	for (i = 0; i < RUMPBLK_SIZE; i++) {
    409 		if (minors[i].rblk_path&&strcmp(minors[i].rblk_path, path)==0) {
    410 			mutex_exit(&rumpblk_lock);
    411 			*dmin = i;
    412 			return 0;
    413 		}
    414 	}
    415 
    416 	for (i = 0; i < RUMPBLK_SIZE; i++)
    417 		if (minors[i].rblk_path == NULL)
    418 			break;
    419 	if (i == RUMPBLK_SIZE) {
    420 		mutex_exit(&rumpblk_lock);
    421 		return EBUSY;
    422 	}
    423 
    424 	rblk = &minors[i];
    425 	len = strlen(path);
    426 	rblk->rblk_path = malloc(len + 1, M_TEMP, M_WAITOK);
    427 	strcpy(rblk->rblk_path, path);
    428 	rblk->rblk_fd = -1;
    429 	rblk->rblk_hostoffset = offset;
    430 	if (size != RUMPBLK_SIZENOTSET) {
    431 		KASSERT(size + offset <= flen);
    432 		rblk->rblk_size = size;
    433 	} else {
    434 		KASSERT(offset < flen);
    435 		rblk->rblk_size = flen - offset;
    436 	}
    437 	rblk->rblk_ftype = ftype;
    438 	makedefaultlabel(&rblk->rblk_label, rblk->rblk_size, i);
    439 	mutex_exit(&rumpblk_lock);
    440 
    441 	*dmin = i;
    442 	return 0;
    443 }
    444 
    445 int
    446 rumpblk_open(dev_t dev, int flag, int fmt, struct lwp *l)
    447 {
    448 	struct rblkdev *rblk = &minors[minor(dev)];
    449 	int error, fd;
    450 
    451 	if (rblk->rblk_path == NULL)
    452 		return ENXIO;
    453 
    454 	if (rblk->rblk_fd != -1)
    455 		return 0; /* XXX: refcount, open mode */
    456 	fd = rumpuser_open(rblk->rblk_path, OFLAGS(flag), &error);
    457 	if (error)
    458 		return error;
    459 
    460 #ifdef HAS_ODIRECT
    461 	rblk->rblk_dfd = rumpuser_open(rblk->rblk_path,
    462 	    OFLAGS(flag) | O_DIRECT, &error);
    463 	if (error)
    464 		return error;
    465 #endif
    466 
    467 	if (rblk->rblk_ftype == RUMPUSER_FT_REG) {
    468 		uint64_t fsize = rblk->rblk_size, off = rblk->rblk_hostoffset;
    469 		struct blkwin *win;
    470 		int i, winsize;
    471 
    472 		/*
    473 		 * Use mmap to access a regular file.  Allocate and
    474 		 * cache initial windows here.  Failure to allocate one
    475 		 * means fallback to read/write i/o.
    476 		 */
    477 
    478 		rblk->rblk_mmflags = 0;
    479 		if (flag & FREAD)
    480 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_READ;
    481 		if (flag & FWRITE) {
    482 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_WRITE;
    483 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_SHARED;
    484 		}
    485 
    486 		TAILQ_INIT(&rblk->rblk_lruq);
    487 		rblk->rblk_fd = fd;
    488 
    489 		for (i = 0; i < memwincnt && off + i*memwinsize < fsize; i++) {
    490 			win = kmem_zalloc(sizeof(*win), KM_SLEEP);
    491 			WINVALIDATE(win);
    492 			TAILQ_INSERT_TAIL(&rblk->rblk_lruq, win, win_lru);
    493 
    494 			/*
    495 			 * Allocate first windows.  Here we just generally
    496 			 * make sure a) we can mmap at all b) we have the
    497 			 * necessary VA available
    498 			 */
    499 			winsize = memwinsize;
    500 			win = getwindow(rblk, off + i*memwinsize, &winsize,
    501 			    &error);
    502 			if (win) {
    503 				putwindow(rblk, win);
    504 			} else {
    505 				wincleanup(rblk);
    506 				break;
    507 			}
    508 		}
    509 	} else {
    510 		rblk->rblk_fd = fd;
    511 	}
    512 
    513 	KASSERT(rblk->rblk_fd != -1);
    514 	return 0;
    515 }
    516 
    517 int
    518 rumpblk_close(dev_t dev, int flag, int fmt, struct lwp *l)
    519 {
    520 	struct rblkdev *rblk = &minors[minor(dev)];
    521 	int dummy;
    522 
    523 	if (rblk->rblk_mmflags)
    524 		wincleanup(rblk);
    525 	rumpuser_fsync(rblk->rblk_fd, &dummy);
    526 	rumpuser_close(rblk->rblk_fd, &dummy);
    527 	rblk->rblk_fd = -1;
    528 
    529 	return 0;
    530 }
    531 
    532 int
    533 rumpblk_ioctl(dev_t dev, u_long xfer, void *addr, int flag, struct lwp *l)
    534 {
    535 	devminor_t dmin = minor(dev);
    536 	struct rblkdev *rblk = &minors[dmin];
    537 	struct partinfo *pi;
    538 	int error = 0;
    539 
    540 	/* well, me should support a few more, but we don't for now */
    541 	switch (xfer) {
    542 	case DIOCGDINFO:
    543 		*(struct disklabel *)addr = rblk->rblk_label;
    544 		break;
    545 
    546 	case DIOCGPART:
    547 		pi = addr;
    548 		pi->part = &rblk->rblk_label.d_partitions[DISKPART(dmin)];
    549 		pi->disklab = &rblk->rblk_label;
    550 		break;
    551 
    552 	/* it's synced enough along the write path */
    553 	case DIOCCACHESYNC:
    554 		break;
    555 
    556 	default:
    557 		error = ENOTTY;
    558 		break;
    559 	}
    560 
    561 	return error;
    562 }
    563 
    564 static int
    565 do_physio(dev_t dev, struct uio *uio, int which)
    566 {
    567 	void (*strat)(struct buf *);
    568 
    569 	if (blkfail)
    570 		strat = rumpblk_strategy_fail;
    571 	else
    572 		strat = rumpblk_strategy;
    573 
    574 	return physio(strat, NULL, dev, which, minphys, uio);
    575 }
    576 
    577 int
    578 rumpblk_read(dev_t dev, struct uio *uio, int flags)
    579 {
    580 
    581 	return do_physio(dev, uio, B_READ);
    582 }
    583 
    584 int
    585 rumpblk_write(dev_t dev, struct uio *uio, int flags)
    586 {
    587 
    588 	return do_physio(dev, uio, B_WRITE);
    589 }
    590 
    591 static void
    592 dostrategy(struct buf *bp)
    593 {
    594 	struct rblkdev *rblk = &minors[minor(bp->b_dev)];
    595 	off_t off;
    596 	int async = bp->b_flags & B_ASYNC;
    597 	int error;
    598 
    599 	/* collect statistics */
    600 	ev_io_total.ev_count++;
    601 	if (async)
    602 		ev_io_async.ev_count++;
    603 	if (BUF_ISWRITE(bp)) {
    604 		ev_bwrite_total.ev_count += bp->b_bcount;
    605 		if (async)
    606 			ev_bwrite_async.ev_count += bp->b_bcount;
    607 	} else {
    608 		ev_bread_total.ev_count++;
    609 	}
    610 
    611 	off = bp->b_blkno << sectshift;
    612 	/*
    613 	 * Do bounds checking if we're working on a file.  Otherwise
    614 	 * invalid file systems might attempt to read beyond EOF.  This
    615 	 * is bad(tm) especially on mmapped images.  This is essentially
    616 	 * the kernel bounds_check() routines.
    617 	 */
    618 	if (off + bp->b_bcount > rblk->rblk_size) {
    619 		int64_t sz = rblk->rblk_size - off;
    620 
    621 		/* EOF */
    622 		if (sz == 0) {
    623 			rump_biodone(bp, 0, 0);
    624 			return;
    625 		}
    626 		/* beyond EOF ==> error */
    627 		if (sz < 0) {
    628 			rump_biodone(bp, 0, EINVAL);
    629 			return;
    630 		}
    631 
    632 		/* truncate to device size */
    633 		bp->b_bcount = sz;
    634 	}
    635 
    636 	off += rblk->rblk_hostoffset;
    637 	DPRINTF(("rumpblk_strategy: 0x%x bytes %s off 0x%" PRIx64
    638 	    " (0x%" PRIx64 " - 0x%" PRIx64 "), %ssync\n",
    639 	    bp->b_bcount, BUF_ISREAD(bp) ? "READ" : "WRITE",
    640 	    off, off, (off + bp->b_bcount), async ? "a" : ""));
    641 
    642 	/* mmap?  handle here and return */
    643 	if (rblk->rblk_mmflags) {
    644 		struct blkwin *win;
    645 		int winsize, iodone;
    646 		uint8_t *ioaddr, *bufaddr;
    647 
    648 		for (iodone = 0; iodone < bp->b_bcount;
    649 		    iodone += winsize, off += winsize) {
    650 			winsize = bp->b_bcount - iodone;
    651 			win = getwindow(rblk, off, &winsize, &error);
    652 			if (win == NULL) {
    653 				rump_biodone(bp, iodone, error);
    654 				return;
    655 			}
    656 
    657 			ioaddr = (uint8_t *)win->win_mem + (off-STARTWIN(off));
    658 			bufaddr = (uint8_t *)bp->b_data + iodone;
    659 
    660 			DPRINTF(("strat: %p off 0x%" PRIx64
    661 			    ", ioaddr %p (%p)/buf %p\n", win,
    662 			    win->win_off, ioaddr, win->win_mem, bufaddr));
    663 			if (BUF_ISREAD(bp)) {
    664 				memcpy(bufaddr, ioaddr, winsize);
    665 			} else {
    666 				memcpy(ioaddr, bufaddr, winsize);
    667 			}
    668 
    669 			/* synchronous write, sync bits back to disk */
    670 			if (BUF_ISWRITE(bp) && !async) {
    671 				rumpuser_memsync(ioaddr, winsize, &error);
    672 			}
    673 			putwindow(rblk, win);
    674 		}
    675 
    676 		rump_biodone(bp, bp->b_bcount, 0);
    677 		return;
    678 	}
    679 
    680 	/*
    681 	 * Do I/O.  We have different paths for async and sync I/O.
    682 	 * Async I/O is done by passing a request to rumpuser where
    683 	 * it is executed.  The rumpuser routine then calls
    684 	 * biodone() to signal any waiters in the kernel.  I/O's are
    685 	 * executed in series.  Technically executing them in parallel
    686 	 * would produce better results, but then we'd need either
    687 	 * more threads or posix aio.  Maybe worth investigating
    688 	 * this later.
    689 	 *
    690 	 * Using bufq here might be a good idea.
    691 	 */
    692 
    693 	if (rump_threads) {
    694 		struct rumpuser_aio *rua;
    695 		int op, fd;
    696 
    697 		fd = rblk->rblk_fd;
    698 		if (BUF_ISREAD(bp)) {
    699 			op = RUA_OP_READ;
    700 		} else {
    701 			op = RUA_OP_WRITE;
    702 			if (!async) {
    703 				/* O_DIRECT not fully automatic yet */
    704 #ifdef HAS_ODIRECT
    705 				if ((off & ((1<<sectshift)-1)) == 0
    706 				    && ((intptr_t)bp->b_data
    707 				      & ((1<<sectshift)-1)) == 0
    708 				    && (bp->b_bcount & ((1<<sectshift)-1)) == 0)
    709 					fd = rblk->rblk_dfd;
    710 				else
    711 #endif
    712 					op |= RUA_OP_SYNC;
    713 			}
    714 		}
    715 
    716 		rumpuser_mutex_enter(&rumpuser_aio_mtx);
    717 		while ((rumpuser_aio_head+1) % N_AIOS == rumpuser_aio_tail) {
    718 			rumpuser_cv_wait(&rumpuser_aio_cv, &rumpuser_aio_mtx);
    719 		}
    720 
    721 		rua = &rumpuser_aios[rumpuser_aio_head];
    722 		KASSERT(rua->rua_bp == NULL);
    723 		rua->rua_fd = fd;
    724 		rua->rua_data = bp->b_data;
    725 		rua->rua_dlen = bp->b_bcount;
    726 		rua->rua_off = off;
    727 		rua->rua_bp = bp;
    728 		rua->rua_op = op;
    729 
    730 		/* insert into queue & signal */
    731 		rumpuser_aio_head = (rumpuser_aio_head+1) % N_AIOS;
    732 		rumpuser_cv_signal(&rumpuser_aio_cv);
    733 		rumpuser_mutex_exit(&rumpuser_aio_mtx);
    734 	} else {
    735 		if (BUF_ISREAD(bp)) {
    736 			rumpuser_read_bio(rblk->rblk_fd, bp->b_data,
    737 			    bp->b_bcount, off, rump_biodone, bp);
    738 		} else {
    739 			rumpuser_write_bio(rblk->rblk_fd, bp->b_data,
    740 			    bp->b_bcount, off, rump_biodone, bp);
    741 		}
    742 		if (BUF_ISWRITE(bp) && !async)
    743 			rumpuser_fsync(rblk->rblk_fd, &error);
    744 	}
    745 }
    746 
    747 void
    748 rumpblk_strategy(struct buf *bp)
    749 {
    750 
    751 	dostrategy(bp);
    752 }
    753 
    754 /*
    755  * Simple random number generator.  This is private so that we can
    756  * very repeatedly control which blocks will fail.
    757  *
    758  * <mlelstv> pooka, rand()
    759  * <mlelstv> [paste]
    760  */
    761 static unsigned
    762 gimmerand(void)
    763 {
    764 
    765 	return (randstate = randstate * 1103515245 + 12345) % (0x80000000L);
    766 }
    767 
    768 /*
    769  * Block device with very simple fault injection.  Fails every
    770  * n out of BLKFAIL_MAX I/O with EIO.  n is determined by the env
    771  * variable RUMP_BLKFAIL.
    772  */
    773 void
    774 rumpblk_strategy_fail(struct buf *bp)
    775 {
    776 
    777 	if (gimmerand() % BLKFAIL_MAX >= blkfail) {
    778 		dostrategy(bp);
    779 	} else {
    780 		printf("block fault injection: failing I/O on block %lld\n",
    781 		    (long long)bp->b_blkno);
    782 		bp->b_error = EIO;
    783 		biodone(bp);
    784 	}
    785 }
    786