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