Home | History | Annotate | Line # | Download | only in rumpvfs
rumpblk.c revision 1.37.4.1
      1  1.37.4.1   rmind /*	$NetBSD: rumpblk.c,v 1.37.4.1 2010/05/30 05:18:07 rmind 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.37.4.1   rmind __KERNEL_RCSID(0, "$NetBSD: rumpblk.c,v 1.37.4.1 2010/05/30 05:18:07 rmind 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.36   pooka #define STARTWIN(off)		((off) & ~((off_t)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.31   pooka 	struct disklabel rblk_label;
    117       1.1   pooka } minors[RUMPBLK_SIZE];
    118       1.1   pooka 
    119      1.20   pooka static struct evcnt ev_io_total;
    120      1.20   pooka static struct evcnt ev_io_async;
    121      1.20   pooka 
    122      1.20   pooka static struct evcnt ev_memblk_hits;
    123      1.20   pooka static struct evcnt ev_memblk_busy;
    124      1.20   pooka 
    125      1.20   pooka static struct evcnt ev_bwrite_total;
    126      1.20   pooka static struct evcnt ev_bwrite_async;
    127      1.20   pooka static struct evcnt ev_bread_total;
    128      1.18   pooka 
    129       1.1   pooka dev_type_open(rumpblk_open);
    130       1.1   pooka dev_type_close(rumpblk_close);
    131       1.1   pooka dev_type_read(rumpblk_read);
    132       1.1   pooka dev_type_write(rumpblk_write);
    133       1.1   pooka dev_type_ioctl(rumpblk_ioctl);
    134       1.1   pooka dev_type_strategy(rumpblk_strategy);
    135       1.3   pooka dev_type_strategy(rumpblk_strategy_fail);
    136       1.1   pooka dev_type_dump(rumpblk_dump);
    137       1.1   pooka dev_type_size(rumpblk_size);
    138       1.1   pooka 
    139       1.1   pooka static const struct bdevsw rumpblk_bdevsw = {
    140       1.1   pooka 	rumpblk_open, rumpblk_close, rumpblk_strategy, rumpblk_ioctl,
    141       1.1   pooka 	nodump, nosize, D_DISK
    142       1.1   pooka };
    143       1.1   pooka 
    144       1.3   pooka static const struct bdevsw rumpblk_bdevsw_fail = {
    145       1.3   pooka 	rumpblk_open, rumpblk_close, rumpblk_strategy_fail, rumpblk_ioctl,
    146       1.3   pooka 	nodump, nosize, D_DISK
    147       1.3   pooka };
    148       1.3   pooka 
    149       1.1   pooka static const struct cdevsw rumpblk_cdevsw = {
    150       1.1   pooka 	rumpblk_open, rumpblk_close, rumpblk_read, rumpblk_write,
    151       1.1   pooka 	rumpblk_ioctl, nostop, notty, nopoll, nommap, nokqfilter, D_DISK
    152       1.1   pooka };
    153       1.1   pooka 
    154       1.3   pooka /* fail every n out of BLKFAIL_MAX */
    155       1.3   pooka #define BLKFAIL_MAX 10000
    156       1.3   pooka static int blkfail;
    157       1.3   pooka static unsigned randstate;
    158      1.24   pooka static kmutex_t rumpblk_lock;
    159      1.37   pooka static int sectshift = DEV_BSHIFT;
    160       1.1   pooka 
    161      1.31   pooka static void
    162      1.31   pooka makedefaultlabel(struct disklabel *lp, off_t size, int part)
    163      1.31   pooka {
    164      1.31   pooka 	int i;
    165      1.31   pooka 
    166      1.31   pooka 	memset(lp, 0, sizeof(*lp));
    167      1.31   pooka 
    168      1.31   pooka 	lp->d_secperunit = size;
    169      1.37   pooka 	lp->d_secsize = 1 << sectshift;
    170      1.37   pooka 	lp->d_nsectors = size >> sectshift;
    171      1.31   pooka 	lp->d_ntracks = 1;
    172      1.31   pooka 	lp->d_ncylinders = 1;
    173      1.31   pooka 	lp->d_secpercyl = lp->d_nsectors;
    174      1.31   pooka 
    175      1.31   pooka 	/* oh dear oh dear */
    176      1.31   pooka 	strncpy(lp->d_typename, "rumpd", sizeof(lp->d_typename));
    177      1.31   pooka 	strncpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
    178      1.31   pooka 
    179      1.31   pooka 	lp->d_type = DTYPE_RUMPD;
    180      1.31   pooka 	lp->d_rpm = 11;
    181      1.31   pooka 	lp->d_interleave = 1;
    182      1.31   pooka 	lp->d_flags = 0;
    183      1.31   pooka 
    184      1.31   pooka 	/* XXX: RAW_PART handling? */
    185      1.31   pooka 	for (i = 0; i < part; i++) {
    186      1.31   pooka 		lp->d_partitions[i].p_fstype = FS_UNUSED;
    187      1.31   pooka 	}
    188      1.37   pooka 	lp->d_partitions[part].p_size = size >> sectshift;
    189      1.31   pooka 	lp->d_npartitions = part+1;
    190      1.31   pooka 	/* XXX: file system type? */
    191      1.31   pooka 
    192      1.31   pooka 	lp->d_magic = DISKMAGIC;
    193      1.31   pooka 	lp->d_magic2 = DISKMAGIC;
    194      1.31   pooka 	lp->d_checksum = 0; /* XXX */
    195      1.31   pooka }
    196      1.31   pooka 
    197      1.16   pooka static struct blkwin *
    198      1.16   pooka getwindow(struct rblkdev *rblk, off_t off, int *wsize, int *error)
    199      1.16   pooka {
    200      1.16   pooka 	struct blkwin *win;
    201      1.16   pooka 
    202      1.16   pooka 	mutex_enter(&rblk->rblk_memmtx);
    203      1.16   pooka  retry:
    204      1.16   pooka 	/* search for window */
    205      1.16   pooka 	TAILQ_FOREACH(win, &rblk->rblk_lruq, win_lru) {
    206      1.16   pooka 		if (INWIN(win, off) && WINVALID(win))
    207      1.16   pooka 			break;
    208      1.16   pooka 	}
    209      1.16   pooka 
    210      1.16   pooka 	/* found?  return */
    211      1.16   pooka 	if (win) {
    212      1.20   pooka 		ev_memblk_hits.ev_count++;
    213      1.16   pooka 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    214      1.16   pooka 		goto good;
    215      1.16   pooka 	}
    216      1.16   pooka 
    217      1.16   pooka 	/*
    218      1.16   pooka 	 * Else, create new window.  If the least recently used is not
    219      1.16   pooka 	 * currently in use, reuse that.  Otherwise we need to wait.
    220      1.16   pooka 	 */
    221      1.16   pooka 	win = TAILQ_LAST(&rblk->rblk_lruq, winlru);
    222      1.16   pooka 	if (win->win_refcnt == 0) {
    223      1.16   pooka 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    224      1.16   pooka 		mutex_exit(&rblk->rblk_memmtx);
    225      1.16   pooka 
    226      1.16   pooka 		if (WINVALID(win)) {
    227      1.16   pooka 			DPRINTF(("win %p, unmap mem %p, off 0x%" PRIx64 "\n",
    228      1.16   pooka 			    win, win->win_mem, win->win_off));
    229      1.16   pooka 			rumpuser_unmap(win->win_mem, WINSIZE(rblk, win));
    230      1.16   pooka 			WINVALIDATE(win);
    231      1.16   pooka 		}
    232      1.16   pooka 
    233      1.16   pooka 		win->win_off = STARTWIN(off);
    234      1.16   pooka 		win->win_mem = rumpuser_filemmap(rblk->rblk_fd, win->win_off,
    235      1.16   pooka 		    WINSIZE(rblk, win), rblk->rblk_mmflags, error);
    236      1.16   pooka 		DPRINTF(("win %p, off 0x%" PRIx64 ", mem %p\n",
    237      1.16   pooka 		    win, win->win_off, win->win_mem));
    238      1.16   pooka 
    239      1.16   pooka 		mutex_enter(&rblk->rblk_memmtx);
    240      1.16   pooka 		if (win->win_mem == NULL) {
    241      1.16   pooka 			WINVALIDATE(win);
    242      1.16   pooka 			TAILQ_INSERT_TAIL(&rblk->rblk_lruq, win, win_lru);
    243      1.16   pooka 			mutex_exit(&rblk->rblk_memmtx);
    244      1.16   pooka 			return NULL;
    245      1.16   pooka 		}
    246      1.16   pooka 	} else {
    247      1.16   pooka 		DPRINTF(("memwin wait\n"));
    248      1.20   pooka 		ev_memblk_busy.ev_count++;
    249      1.18   pooka 
    250      1.16   pooka 		rblk->rblk_waiting = true;
    251      1.16   pooka 		cv_wait(&rblk->rblk_memcv, &rblk->rblk_memmtx);
    252      1.16   pooka 		goto retry;
    253      1.16   pooka 	}
    254      1.16   pooka 
    255      1.16   pooka  good:
    256      1.16   pooka 	KASSERT(win);
    257      1.16   pooka 	win->win_refcnt++;
    258      1.16   pooka 	TAILQ_INSERT_HEAD(&rblk->rblk_lruq, win, win_lru);
    259      1.16   pooka 	mutex_exit(&rblk->rblk_memmtx);
    260      1.19   pooka 	*wsize = MIN(*wsize, memwinsize - (off-win->win_off));
    261      1.16   pooka 	KASSERT(*wsize);
    262      1.16   pooka 
    263      1.16   pooka 	return win;
    264      1.16   pooka }
    265      1.16   pooka 
    266      1.16   pooka static void
    267      1.16   pooka putwindow(struct rblkdev *rblk, struct blkwin *win)
    268      1.16   pooka {
    269      1.16   pooka 
    270      1.16   pooka 	mutex_enter(&rblk->rblk_memmtx);
    271      1.16   pooka 	if (--win->win_refcnt == 0 && rblk->rblk_waiting) {
    272      1.16   pooka 		rblk->rblk_waiting = false;
    273      1.16   pooka 		cv_signal(&rblk->rblk_memcv);
    274      1.16   pooka 	}
    275      1.16   pooka 	KASSERT(win->win_refcnt >= 0);
    276      1.16   pooka 	mutex_exit(&rblk->rblk_memmtx);
    277      1.16   pooka }
    278      1.16   pooka 
    279      1.16   pooka static void
    280      1.16   pooka wincleanup(struct rblkdev *rblk)
    281      1.16   pooka {
    282      1.16   pooka 	struct blkwin *win;
    283      1.16   pooka 
    284      1.16   pooka 	while ((win = TAILQ_FIRST(&rblk->rblk_lruq)) != NULL) {
    285      1.16   pooka 		TAILQ_REMOVE(&rblk->rblk_lruq, win, win_lru);
    286      1.16   pooka 		if (WINVALID(win)) {
    287      1.16   pooka 			DPRINTF(("cleanup win %p addr %p\n",
    288      1.16   pooka 			    win, win->win_mem));
    289      1.16   pooka 			rumpuser_unmap(win->win_mem, WINSIZE(rblk, win));
    290      1.16   pooka 		}
    291      1.16   pooka 		kmem_free(win, sizeof(*win));
    292      1.16   pooka 	}
    293      1.16   pooka 	rblk->rblk_mmflags = 0;
    294      1.16   pooka }
    295      1.16   pooka 
    296       1.1   pooka int
    297       1.8  cegger rumpblk_init(void)
    298       1.1   pooka {
    299       1.3   pooka 	char buf[64];
    300  1.37.4.1   rmind 	devmajor_t rumpblkmaj = RUMPBLK_DEVMAJOR;
    301      1.19   pooka 	unsigned tmp;
    302      1.16   pooka 	int error, i;
    303       1.3   pooka 
    304      1.24   pooka 	mutex_init(&rumpblk_lock, MUTEX_DEFAULT, IPL_NONE);
    305      1.24   pooka 
    306       1.3   pooka 	if (rumpuser_getenv("RUMP_BLKFAIL", buf, sizeof(buf), &error) == 0) {
    307       1.3   pooka 		blkfail = strtoul(buf, NULL, 10);
    308       1.3   pooka 		/* fail everything */
    309       1.3   pooka 		if (blkfail > BLKFAIL_MAX)
    310       1.3   pooka 			blkfail = BLKFAIL_MAX;
    311       1.3   pooka 		if (rumpuser_getenv("RUMP_BLKFAIL_SEED", buf, sizeof(buf),
    312       1.3   pooka 		    &error) == 0) {
    313       1.3   pooka 			randstate = strtoul(buf, NULL, 10);
    314       1.3   pooka 		} else {
    315      1.23   pooka 			randstate = arc4random();
    316       1.3   pooka 		}
    317      1.22   pooka 		printf("rumpblk: FAULT INJECTION ACTIVE! fail %d/%d. "
    318      1.22   pooka 		    "seed %u\n", blkfail, BLKFAIL_MAX, randstate);
    319       1.3   pooka 	} else {
    320       1.3   pooka 		blkfail = 0;
    321       1.3   pooka 	}
    322       1.1   pooka 
    323      1.19   pooka 	if (rumpuser_getenv("RUMP_BLKWINSIZE", buf, sizeof(buf), &error) == 0) {
    324      1.19   pooka 		printf("rumpblk: ");
    325      1.19   pooka 		tmp = strtoul(buf, NULL, 10);
    326      1.19   pooka 		if (tmp && !(tmp & (tmp-1)))
    327      1.19   pooka 			memwinsize = tmp;
    328      1.19   pooka 		else
    329      1.19   pooka 			printf("invalid RUMP_BLKWINSIZE %d, ", tmp);
    330      1.19   pooka 		printf("using %d for memwinsize\n", memwinsize);
    331      1.19   pooka 	}
    332      1.19   pooka 	if (rumpuser_getenv("RUMP_BLKWINCOUNT", buf, sizeof(buf), &error) == 0){
    333      1.19   pooka 		printf("rumpblk: ");
    334      1.19   pooka 		tmp = strtoul(buf, NULL, 10);
    335      1.19   pooka 		if (tmp)
    336      1.19   pooka 			memwincnt = tmp;
    337      1.19   pooka 		else
    338      1.19   pooka 			printf("invalid RUMP_BLKWINCOUNT %d, ", tmp);
    339      1.19   pooka 		printf("using %d for memwincount\n", memwincnt);
    340      1.19   pooka 	}
    341      1.37   pooka 	if (rumpuser_getenv("RUMP_BLKSECTSHIFT", buf, sizeof(buf), &error)==0){
    342      1.37   pooka 		printf("rumpblk: ");
    343      1.37   pooka 		tmp = strtoul(buf, NULL, 10);
    344      1.37   pooka 		if (tmp >= DEV_BSHIFT)
    345      1.37   pooka 			sectshift = tmp;
    346      1.37   pooka 		else
    347      1.37   pooka 			printf("RUMP_BLKSECTSHIFT must be least %d (now %d), ",
    348      1.37   pooka 			   DEV_BSHIFT, tmp);
    349      1.37   pooka 		printf("using %d for sector shift (size %d)\n",
    350      1.37   pooka 		    sectshift, 1<<sectshift);
    351      1.37   pooka 	}
    352      1.19   pooka 
    353      1.16   pooka 	memset(minors, 0, sizeof(minors));
    354      1.16   pooka 	for (i = 0; i < RUMPBLK_SIZE; i++) {
    355      1.16   pooka 		mutex_init(&minors[i].rblk_memmtx, MUTEX_DEFAULT, IPL_NONE);
    356      1.16   pooka 		cv_init(&minors[i].rblk_memcv, "rblkmcv");
    357      1.16   pooka 	}
    358      1.16   pooka 
    359      1.20   pooka 	evcnt_attach_dynamic(&ev_io_total, EVCNT_TYPE_MISC, NULL,
    360  1.37.4.1   rmind 	    "rumpblk", "I/O reqs");
    361      1.20   pooka 	evcnt_attach_dynamic(&ev_io_async, EVCNT_TYPE_MISC, NULL,
    362  1.37.4.1   rmind 	    "rumpblk", "async I/O");
    363      1.20   pooka 
    364      1.20   pooka 	evcnt_attach_dynamic(&ev_bread_total, EVCNT_TYPE_MISC, NULL,
    365  1.37.4.1   rmind 	    "rumpblk", "bytes read");
    366      1.20   pooka 	evcnt_attach_dynamic(&ev_bwrite_total, EVCNT_TYPE_MISC, NULL,
    367  1.37.4.1   rmind 	    "rumpblk", "bytes written");
    368      1.20   pooka 	evcnt_attach_dynamic(&ev_bwrite_async, EVCNT_TYPE_MISC, NULL,
    369  1.37.4.1   rmind 	    "rumpblk", "bytes written async");
    370      1.20   pooka 
    371      1.20   pooka 	evcnt_attach_dynamic(&ev_memblk_hits, EVCNT_TYPE_MISC, NULL,
    372  1.37.4.1   rmind 	    "rumpblk", "window hits");
    373      1.20   pooka 	evcnt_attach_dynamic(&ev_memblk_busy, EVCNT_TYPE_MISC, NULL,
    374  1.37.4.1   rmind 	    "rumpblk", "all windows busy");
    375      1.18   pooka 
    376       1.3   pooka 	if (blkfail) {
    377  1.37.4.1   rmind 		return devsw_attach("rumpblk",
    378  1.37.4.1   rmind 		    &rumpblk_bdevsw_fail, &rumpblkmaj,
    379  1.37.4.1   rmind 		    &rumpblk_cdevsw, &rumpblkmaj);
    380       1.3   pooka 	} else {
    381  1.37.4.1   rmind 		return devsw_attach("rumpblk",
    382  1.37.4.1   rmind 		    &rumpblk_bdevsw, &rumpblkmaj,
    383  1.37.4.1   rmind 		    &rumpblk_cdevsw, &rumpblkmaj);
    384       1.3   pooka 	}
    385       1.1   pooka }
    386       1.1   pooka 
    387      1.24   pooka /* XXX: no deregister */
    388       1.1   pooka int
    389      1.27   pooka rumpblk_register(const char *path, devminor_t *dmin,
    390      1.27   pooka 	uint64_t offset, uint64_t size)
    391       1.1   pooka {
    392      1.27   pooka 	struct rblkdev *rblk;
    393      1.24   pooka 	uint64_t flen;
    394       1.1   pooka 	size_t len;
    395      1.30   pooka 	int ftype, error, i;
    396      1.24   pooka 
    397      1.27   pooka 	/* devices might not report correct size unless they're open */
    398      1.30   pooka 	if (rumpuser_getfileinfo(path, &flen, &ftype, &error) == -1)
    399      1.27   pooka 		return error;
    400      1.27   pooka 
    401      1.24   pooka 	/* verify host file is of supported type */
    402      1.24   pooka 	if (!(ftype == RUMPUSER_FT_REG
    403      1.24   pooka 	   || ftype == RUMPUSER_FT_BLK
    404      1.24   pooka 	   || ftype == RUMPUSER_FT_CHR))
    405      1.24   pooka 		return EINVAL;
    406       1.1   pooka 
    407      1.24   pooka 	mutex_enter(&rumpblk_lock);
    408      1.24   pooka 	for (i = 0; i < RUMPBLK_SIZE; i++) {
    409      1.24   pooka 		if (minors[i].rblk_path&&strcmp(minors[i].rblk_path, path)==0) {
    410      1.24   pooka 			mutex_exit(&rumpblk_lock);
    411      1.24   pooka 			*dmin = i;
    412      1.24   pooka 			return 0;
    413      1.24   pooka 		}
    414      1.24   pooka 	}
    415       1.1   pooka 
    416       1.1   pooka 	for (i = 0; i < RUMPBLK_SIZE; i++)
    417       1.1   pooka 		if (minors[i].rblk_path == NULL)
    418       1.1   pooka 			break;
    419      1.24   pooka 	if (i == RUMPBLK_SIZE) {
    420      1.24   pooka 		mutex_exit(&rumpblk_lock);
    421      1.24   pooka 		return EBUSY;
    422      1.24   pooka 	}
    423       1.1   pooka 
    424      1.27   pooka 	rblk = &minors[i];
    425       1.1   pooka 	len = strlen(path);
    426      1.27   pooka 	rblk->rblk_path = malloc(len + 1, M_TEMP, M_WAITOK);
    427      1.27   pooka 	strcpy(rblk->rblk_path, path);
    428      1.27   pooka 	rblk->rblk_fd = -1;
    429      1.27   pooka 	rblk->rblk_hostoffset = offset;
    430      1.33   pooka 	if (size != RUMPBLK_SIZENOTSET) {
    431      1.27   pooka 		KASSERT(size + offset <= flen);
    432      1.27   pooka 		rblk->rblk_size = size;
    433      1.27   pooka 	} else {
    434      1.27   pooka 		KASSERT(offset < flen);
    435      1.27   pooka 		rblk->rblk_size = flen - offset;
    436      1.27   pooka 	}
    437      1.27   pooka 	rblk->rblk_ftype = ftype;
    438      1.31   pooka 	makedefaultlabel(&rblk->rblk_label, rblk->rblk_size, i);
    439      1.24   pooka 	mutex_exit(&rumpblk_lock);
    440      1.24   pooka 
    441      1.24   pooka 	*dmin = i;
    442      1.24   pooka 	return 0;
    443       1.1   pooka }
    444       1.1   pooka 
    445       1.1   pooka int
    446       1.1   pooka rumpblk_open(dev_t dev, int flag, int fmt, struct lwp *l)
    447       1.1   pooka {
    448       1.1   pooka 	struct rblkdev *rblk = &minors[minor(dev)];
    449       1.1   pooka 	int error, fd;
    450       1.1   pooka 
    451      1.27   pooka 	if (rblk->rblk_path == NULL)
    452      1.27   pooka 		return ENXIO;
    453      1.27   pooka 
    454      1.25   pooka 	if (rblk->rblk_fd != -1)
    455      1.25   pooka 		return 0; /* XXX: refcount, open mode */
    456       1.1   pooka 	fd = rumpuser_open(rblk->rblk_path, OFLAGS(flag), &error);
    457       1.1   pooka 	if (error)
    458       1.1   pooka 		return error;
    459       1.1   pooka 
    460      1.20   pooka #ifdef HAS_ODIRECT
    461      1.20   pooka 	rblk->rblk_dfd = rumpuser_open(rblk->rblk_path,
    462      1.20   pooka 	    OFLAGS(flag) | O_DIRECT, &error);
    463      1.20   pooka 	if (error)
    464      1.20   pooka 		return error;
    465      1.20   pooka #endif
    466      1.20   pooka 
    467      1.27   pooka 	if (rblk->rblk_ftype == RUMPUSER_FT_REG) {
    468      1.30   pooka 		uint64_t fsize = rblk->rblk_size, off = rblk->rblk_hostoffset;
    469      1.16   pooka 		struct blkwin *win;
    470      1.16   pooka 		int i, winsize;
    471      1.16   pooka 
    472       1.1   pooka 		/*
    473      1.16   pooka 		 * Use mmap to access a regular file.  Allocate and
    474      1.16   pooka 		 * cache initial windows here.  Failure to allocate one
    475      1.16   pooka 		 * means fallback to read/write i/o.
    476       1.1   pooka 		 */
    477       1.9   pooka 
    478      1.16   pooka 		rblk->rblk_mmflags = 0;
    479      1.16   pooka 		if (flag & FREAD)
    480      1.16   pooka 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_READ;
    481      1.16   pooka 		if (flag & FWRITE) {
    482      1.16   pooka 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_WRITE;
    483      1.16   pooka 			rblk->rblk_mmflags |= RUMPUSER_FILEMMAP_SHARED;
    484      1.16   pooka 		}
    485      1.16   pooka 
    486      1.16   pooka 		TAILQ_INIT(&rblk->rblk_lruq);
    487      1.16   pooka 		rblk->rblk_fd = fd;
    488      1.16   pooka 
    489      1.28   pooka 		for (i = 0; i < memwincnt && off + i*memwinsize < fsize; i++) {
    490      1.16   pooka 			win = kmem_zalloc(sizeof(*win), KM_SLEEP);
    491      1.16   pooka 			WINVALIDATE(win);
    492      1.16   pooka 			TAILQ_INSERT_TAIL(&rblk->rblk_lruq, win, win_lru);
    493      1.16   pooka 
    494      1.16   pooka 			/*
    495      1.16   pooka 			 * Allocate first windows.  Here we just generally
    496      1.16   pooka 			 * make sure a) we can mmap at all b) we have the
    497      1.16   pooka 			 * necessary VA available
    498      1.16   pooka 			 */
    499      1.29   pooka 			winsize = memwinsize;
    500      1.28   pooka 			win = getwindow(rblk, off + i*memwinsize, &winsize,
    501      1.28   pooka 			    &error);
    502      1.16   pooka 			if (win) {
    503      1.16   pooka 				putwindow(rblk, win);
    504      1.16   pooka 			} else {
    505      1.16   pooka 				wincleanup(rblk);
    506      1.16   pooka 				break;
    507      1.12   pooka 			}
    508       1.9   pooka 		}
    509      1.35   pooka 	} else {
    510      1.35   pooka 		rblk->rblk_fd = fd;
    511       1.1   pooka 	}
    512       1.1   pooka 
    513      1.16   pooka 	KASSERT(rblk->rblk_fd != -1);
    514       1.1   pooka 	return 0;
    515       1.1   pooka }
    516       1.1   pooka 
    517       1.1   pooka int
    518       1.1   pooka rumpblk_close(dev_t dev, int flag, int fmt, struct lwp *l)
    519       1.1   pooka {
    520       1.1   pooka 	struct rblkdev *rblk = &minors[minor(dev)];
    521       1.1   pooka 	int dummy;
    522       1.1   pooka 
    523      1.16   pooka 	if (rblk->rblk_mmflags)
    524      1.16   pooka 		wincleanup(rblk);
    525      1.16   pooka 	rumpuser_fsync(rblk->rblk_fd, &dummy);
    526       1.1   pooka 	rumpuser_close(rblk->rblk_fd, &dummy);
    527       1.1   pooka 	rblk->rblk_fd = -1;
    528       1.1   pooka 
    529       1.1   pooka 	return 0;
    530       1.1   pooka }
    531       1.1   pooka 
    532       1.1   pooka int
    533       1.1   pooka rumpblk_ioctl(dev_t dev, u_long xfer, void *addr, int flag, struct lwp *l)
    534       1.1   pooka {
    535      1.31   pooka 	devminor_t dmin = minor(dev);
    536      1.31   pooka 	struct rblkdev *rblk = &minors[dmin];
    537      1.31   pooka 	struct partinfo *pi;
    538      1.31   pooka 	int error = 0;
    539      1.31   pooka 
    540      1.31   pooka 	/* well, me should support a few more, but we don't for now */
    541      1.31   pooka 	switch (xfer) {
    542      1.31   pooka 	case DIOCGDINFO:
    543      1.31   pooka 		*(struct disklabel *)addr = rblk->rblk_label;
    544      1.31   pooka 		break;
    545      1.31   pooka 
    546      1.31   pooka 	case DIOCGPART:
    547      1.31   pooka 		pi = addr;
    548      1.31   pooka 		pi->part = &rblk->rblk_label.d_partitions[DISKPART(dmin)];
    549      1.31   pooka 		pi->disklab = &rblk->rblk_label;
    550      1.31   pooka 		break;
    551      1.32   pooka 
    552      1.32   pooka 	/* it's synced enough along the write path */
    553      1.32   pooka 	case DIOCCACHESYNC:
    554      1.32   pooka 		break;
    555      1.32   pooka 
    556      1.31   pooka 	default:
    557      1.31   pooka 		error = ENOTTY;
    558      1.31   pooka 		break;
    559       1.1   pooka 	}
    560       1.1   pooka 
    561      1.31   pooka 	return error;
    562       1.1   pooka }
    563       1.1   pooka 
    564      1.25   pooka static int
    565      1.25   pooka do_physio(dev_t dev, struct uio *uio, int which)
    566      1.25   pooka {
    567      1.25   pooka 	void (*strat)(struct buf *);
    568      1.25   pooka 
    569      1.25   pooka 	if (blkfail)
    570      1.25   pooka 		strat = rumpblk_strategy_fail;
    571      1.25   pooka 	else
    572      1.25   pooka 		strat = rumpblk_strategy;
    573      1.25   pooka 
    574      1.25   pooka 	return physio(strat, NULL, dev, which, minphys, uio);
    575      1.25   pooka }
    576      1.25   pooka 
    577       1.1   pooka int
    578       1.1   pooka rumpblk_read(dev_t dev, struct uio *uio, int flags)
    579       1.1   pooka {
    580       1.1   pooka 
    581      1.25   pooka 	return do_physio(dev, uio, B_READ);
    582       1.1   pooka }
    583       1.1   pooka 
    584       1.1   pooka int
    585       1.1   pooka rumpblk_write(dev_t dev, struct uio *uio, int flags)
    586       1.1   pooka {
    587       1.1   pooka 
    588      1.25   pooka 	return do_physio(dev, uio, B_WRITE);
    589       1.1   pooka }
    590       1.1   pooka 
    591       1.3   pooka static void
    592       1.3   pooka dostrategy(struct buf *bp)
    593       1.1   pooka {
    594       1.1   pooka 	struct rblkdev *rblk = &minors[minor(bp->b_dev)];
    595       1.1   pooka 	off_t off;
    596      1.21   pooka 	int async = bp->b_flags & B_ASYNC;
    597      1.21   pooka 	int error;
    598       1.1   pooka 
    599      1.20   pooka 	/* collect statistics */
    600      1.20   pooka 	ev_io_total.ev_count++;
    601      1.20   pooka 	if (async)
    602      1.20   pooka 		ev_io_async.ev_count++;
    603      1.20   pooka 	if (BUF_ISWRITE(bp)) {
    604      1.20   pooka 		ev_bwrite_total.ev_count += bp->b_bcount;
    605      1.20   pooka 		if (async)
    606      1.20   pooka 			ev_bwrite_async.ev_count += bp->b_bcount;
    607      1.20   pooka 	} else {
    608      1.20   pooka 		ev_bread_total.ev_count++;
    609      1.20   pooka 	}
    610      1.20   pooka 
    611      1.37   pooka 	off = bp->b_blkno << sectshift;
    612      1.11   pooka 	/*
    613      1.11   pooka 	 * Do bounds checking if we're working on a file.  Otherwise
    614      1.11   pooka 	 * invalid file systems might attempt to read beyond EOF.  This
    615      1.11   pooka 	 * is bad(tm) especially on mmapped images.  This is essentially
    616      1.11   pooka 	 * the kernel bounds_check() routines.
    617      1.11   pooka 	 */
    618      1.27   pooka 	if (off + bp->b_bcount > rblk->rblk_size) {
    619      1.11   pooka 		int64_t sz = rblk->rblk_size - off;
    620      1.11   pooka 
    621      1.11   pooka 		/* EOF */
    622      1.11   pooka 		if (sz == 0) {
    623      1.11   pooka 			rump_biodone(bp, 0, 0);
    624      1.11   pooka 			return;
    625      1.11   pooka 		}
    626      1.11   pooka 		/* beyond EOF ==> error */
    627      1.11   pooka 		if (sz < 0) {
    628      1.11   pooka 			rump_biodone(bp, 0, EINVAL);
    629      1.11   pooka 			return;
    630      1.11   pooka 		}
    631      1.11   pooka 
    632      1.11   pooka 		/* truncate to device size */
    633      1.11   pooka 		bp->b_bcount = sz;
    634      1.11   pooka 	}
    635      1.11   pooka 
    636      1.34   pooka 	off += rblk->rblk_hostoffset;
    637       1.1   pooka 	DPRINTF(("rumpblk_strategy: 0x%x bytes %s off 0x%" PRIx64
    638      1.20   pooka 	    " (0x%" PRIx64 " - 0x%" PRIx64 "), %ssync\n",
    639      1.16   pooka 	    bp->b_bcount, BUF_ISREAD(bp) ? "READ" : "WRITE",
    640      1.20   pooka 	    off, off, (off + bp->b_bcount), async ? "a" : ""));
    641       1.1   pooka 
    642      1.16   pooka 	/* mmap?  handle here and return */
    643      1.16   pooka 	if (rblk->rblk_mmflags) {
    644      1.16   pooka 		struct blkwin *win;
    645      1.16   pooka 		int winsize, iodone;
    646      1.16   pooka 		uint8_t *ioaddr, *bufaddr;
    647      1.16   pooka 
    648      1.16   pooka 		for (iodone = 0; iodone < bp->b_bcount;
    649      1.16   pooka 		    iodone += winsize, off += winsize) {
    650      1.16   pooka 			winsize = bp->b_bcount - iodone;
    651      1.16   pooka 			win = getwindow(rblk, off, &winsize, &error);
    652      1.16   pooka 			if (win == NULL) {
    653      1.16   pooka 				rump_biodone(bp, iodone, error);
    654      1.16   pooka 				return;
    655      1.16   pooka 			}
    656      1.16   pooka 
    657      1.16   pooka 			ioaddr = (uint8_t *)win->win_mem + (off-STARTWIN(off));
    658      1.16   pooka 			bufaddr = (uint8_t *)bp->b_data + iodone;
    659      1.11   pooka 
    660      1.16   pooka 			DPRINTF(("strat: %p off 0x%" PRIx64
    661      1.16   pooka 			    ", ioaddr %p (%p)/buf %p\n", win,
    662      1.16   pooka 			    win->win_off, ioaddr, win->win_mem, bufaddr));
    663      1.16   pooka 			if (BUF_ISREAD(bp)) {
    664      1.16   pooka 				memcpy(bufaddr, ioaddr, winsize);
    665      1.16   pooka 			} else {
    666      1.16   pooka 				memcpy(ioaddr, bufaddr, winsize);
    667      1.16   pooka 			}
    668      1.16   pooka 
    669      1.16   pooka 			/* synchronous write, sync bits back to disk */
    670      1.16   pooka 			if (BUF_ISWRITE(bp) && !async) {
    671      1.16   pooka 				rumpuser_memsync(ioaddr, winsize, &error);
    672      1.16   pooka 			}
    673      1.16   pooka 			putwindow(rblk, win);
    674       1.9   pooka 		}
    675       1.9   pooka 
    676       1.9   pooka 		rump_biodone(bp, bp->b_bcount, 0);
    677       1.9   pooka 		return;
    678       1.9   pooka 	}
    679       1.9   pooka 
    680       1.1   pooka 	/*
    681       1.1   pooka 	 * Do I/O.  We have different paths for async and sync I/O.
    682       1.1   pooka 	 * Async I/O is done by passing a request to rumpuser where
    683       1.1   pooka 	 * it is executed.  The rumpuser routine then calls
    684       1.1   pooka 	 * biodone() to signal any waiters in the kernel.  I/O's are
    685       1.1   pooka 	 * executed in series.  Technically executing them in parallel
    686       1.1   pooka 	 * would produce better results, but then we'd need either
    687       1.1   pooka 	 * more threads or posix aio.  Maybe worth investigating
    688       1.1   pooka 	 * this later.
    689      1.14   pooka 	 *
    690      1.14   pooka 	 * Using bufq here might be a good idea.
    691       1.1   pooka 	 */
    692      1.20   pooka 
    693      1.13   pooka 	if (rump_threads) {
    694       1.1   pooka 		struct rumpuser_aio *rua;
    695      1.20   pooka 		int op, fd;
    696      1.20   pooka 
    697      1.20   pooka 		fd = rblk->rblk_fd;
    698      1.20   pooka 		if (BUF_ISREAD(bp)) {
    699      1.20   pooka 			op = RUA_OP_READ;
    700      1.20   pooka 		} else {
    701      1.20   pooka 			op = RUA_OP_WRITE;
    702      1.20   pooka 			if (!async) {
    703      1.20   pooka 				/* O_DIRECT not fully automatic yet */
    704      1.20   pooka #ifdef HAS_ODIRECT
    705      1.37   pooka 				if ((off & ((1<<sectshift)-1)) == 0
    706      1.37   pooka 				    && ((intptr_t)bp->b_data
    707      1.37   pooka 				      & ((1<<sectshift)-1)) == 0
    708      1.37   pooka 				    && (bp->b_bcount & ((1<<sectshift)-1)) == 0)
    709      1.20   pooka 					fd = rblk->rblk_dfd;
    710      1.20   pooka 				else
    711      1.20   pooka #endif
    712      1.20   pooka 					op |= RUA_OP_SYNC;
    713      1.20   pooka 			}
    714      1.20   pooka 		}
    715       1.1   pooka 
    716       1.1   pooka 		rumpuser_mutex_enter(&rumpuser_aio_mtx);
    717      1.20   pooka 		while ((rumpuser_aio_head+1) % N_AIOS == rumpuser_aio_tail) {
    718      1.13   pooka 			rumpuser_cv_wait(&rumpuser_aio_cv, &rumpuser_aio_mtx);
    719      1.20   pooka 		}
    720       1.1   pooka 
    721       1.2   pooka 		rua = &rumpuser_aios[rumpuser_aio_head];
    722       1.2   pooka 		KASSERT(rua->rua_bp == NULL);
    723      1.20   pooka 		rua->rua_fd = fd;
    724       1.2   pooka 		rua->rua_data = bp->b_data;
    725       1.2   pooka 		rua->rua_dlen = bp->b_bcount;
    726       1.2   pooka 		rua->rua_off = off;
    727       1.2   pooka 		rua->rua_bp = bp;
    728      1.20   pooka 		rua->rua_op = op;
    729       1.2   pooka 
    730       1.1   pooka 		/* insert into queue & signal */
    731       1.6   pooka 		rumpuser_aio_head = (rumpuser_aio_head+1) % N_AIOS;
    732       1.1   pooka 		rumpuser_cv_signal(&rumpuser_aio_cv);
    733       1.1   pooka 		rumpuser_mutex_exit(&rumpuser_aio_mtx);
    734       1.1   pooka 	} else {
    735       1.1   pooka 		if (BUF_ISREAD(bp)) {
    736       1.1   pooka 			rumpuser_read_bio(rblk->rblk_fd, bp->b_data,
    737       1.1   pooka 			    bp->b_bcount, off, rump_biodone, bp);
    738       1.1   pooka 		} else {
    739       1.1   pooka 			rumpuser_write_bio(rblk->rblk_fd, bp->b_data,
    740       1.1   pooka 			    bp->b_bcount, off, rump_biodone, bp);
    741       1.1   pooka 		}
    742      1.20   pooka 		if (BUF_ISWRITE(bp) && !async)
    743      1.20   pooka 			rumpuser_fsync(rblk->rblk_fd, &error);
    744       1.1   pooka 	}
    745       1.1   pooka }
    746       1.3   pooka 
    747       1.3   pooka void
    748       1.3   pooka rumpblk_strategy(struct buf *bp)
    749       1.3   pooka {
    750       1.3   pooka 
    751       1.3   pooka 	dostrategy(bp);
    752       1.3   pooka }
    753       1.3   pooka 
    754       1.3   pooka /*
    755       1.4   pooka  * Simple random number generator.  This is private so that we can
    756       1.4   pooka  * very repeatedly control which blocks will fail.
    757       1.4   pooka  *
    758       1.3   pooka  * <mlelstv> pooka, rand()
    759       1.3   pooka  * <mlelstv> [paste]
    760       1.3   pooka  */
    761       1.3   pooka static unsigned
    762       1.3   pooka gimmerand(void)
    763       1.3   pooka {
    764       1.3   pooka 
    765       1.3   pooka 	return (randstate = randstate * 1103515245 + 12345) % (0x80000000L);
    766       1.3   pooka }
    767       1.3   pooka 
    768       1.3   pooka /*
    769       1.3   pooka  * Block device with very simple fault injection.  Fails every
    770       1.3   pooka  * n out of BLKFAIL_MAX I/O with EIO.  n is determined by the env
    771       1.3   pooka  * variable RUMP_BLKFAIL.
    772       1.3   pooka  */
    773       1.3   pooka void
    774       1.3   pooka rumpblk_strategy_fail(struct buf *bp)
    775       1.3   pooka {
    776       1.3   pooka 
    777       1.3   pooka 	if (gimmerand() % BLKFAIL_MAX >= blkfail) {
    778       1.3   pooka 		dostrategy(bp);
    779       1.3   pooka 	} else {
    780       1.3   pooka 		printf("block fault injection: failing I/O on block %lld\n",
    781       1.3   pooka 		    (long long)bp->b_blkno);
    782       1.3   pooka 		bp->b_error = EIO;
    783       1.3   pooka 		biodone(bp);
    784       1.3   pooka 	}
    785       1.3   pooka }
    786