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