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