2 * Copyright (c) 2011 Google, Inc.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
31 * Userboot disk image handling.
37 #include <bootstrap.h>
40 #include "libuserboot.h"
42 struct userdisk_info {
45 int ud_open; /* reference counter */
46 void *ud_bcache; /* buffer cache data */
49 int userboot_disk_maxunit = 0;
51 static int userdisk_maxunit = 0;
52 static struct userdisk_info *ud_info;
54 static int userdisk_init(void);
55 static void userdisk_cleanup(void);
56 static int userdisk_strategy(void *devdata, int flag, daddr_t dblk,
57 size_t size, char *buf, size_t *rsize);
58 static int userdisk_realstrategy(void *devdata, int flag, daddr_t dblk,
59 size_t size, char *buf, size_t *rsize);
60 static int userdisk_open(struct open_file *f, ...);
61 static int userdisk_close(struct open_file *f);
62 static int userdisk_ioctl(struct open_file *f, u_long cmd, void *data);
63 static int userdisk_print(int verbose);
65 struct devsw userboot_disk = {
78 * Initialize userdisk_info structure for each disk.
87 userdisk_maxunit = userboot_disk_maxunit;
88 if (userdisk_maxunit > 0) {
89 ud_info = malloc(sizeof(*ud_info) * userdisk_maxunit);
92 for (i = 0; i < userdisk_maxunit; i++) {
93 if (CALLBACK(diskioctl, i, DIOCGSECTORSIZE,
94 §orsize) != 0 || CALLBACK(diskioctl, i,
95 DIOCGMEDIASIZE, &mediasize) != 0)
97 ud_info[i].mediasize = mediasize;
98 ud_info[i].sectorsize = sectorsize;
99 ud_info[i].ud_open = 0;
100 ud_info[i].ud_bcache = NULL;
103 bcache_add_dev(userdisk_maxunit);
108 userdisk_cleanup(void)
111 if (userdisk_maxunit > 0)
116 * Print information about disks
119 userdisk_print(int verbose)
121 struct disk_devdesc dev;
125 if (userdisk_maxunit == 0)
128 printf("%s devices:", userboot_disk.dv_name);
129 if ((ret = pager_output("\n")) != 0)
132 for (i = 0; i < userdisk_maxunit; i++) {
133 snprintf(line, sizeof(line),
134 " disk%d: Guest drive image\n", i);
135 ret = pager_output(line);
138 dev.d_dev = &userboot_disk;
141 dev.d_partition = -1;
142 if (disk_open(&dev, ud_info[i].mediasize,
143 ud_info[i].sectorsize) == 0) {
144 snprintf(line, sizeof(line), " disk%d", i);
145 ret = disk_print(&dev, line, verbose);
155 * Attempt to open the disk described by (dev) for use by (f).
158 userdisk_open(struct open_file *f, ...)
161 struct disk_devdesc *dev;
164 dev = va_arg(ap, struct disk_devdesc *);
167 if (dev->d_unit < 0 || dev->d_unit >= userdisk_maxunit)
169 ud_info[dev->d_unit].ud_open++;
170 if (ud_info[dev->d_unit].ud_bcache == NULL)
171 ud_info[dev->d_unit].ud_bcache = bcache_allocate();
172 return (disk_open(dev, ud_info[dev->d_unit].mediasize,
173 ud_info[dev->d_unit].sectorsize));
177 userdisk_close(struct open_file *f)
179 struct disk_devdesc *dev;
181 dev = (struct disk_devdesc *)f->f_devdata;
182 ud_info[dev->d_unit].ud_open--;
183 if (ud_info[dev->d_unit].ud_open == 0) {
184 bcache_free(ud_info[dev->d_unit].ud_bcache);
185 ud_info[dev->d_unit].ud_bcache = NULL;
187 return (disk_close(dev));
191 userdisk_strategy(void *devdata, int rw, daddr_t dblk, size_t size,
192 char *buf, size_t *rsize)
194 struct bcache_devdata bcd;
195 struct disk_devdesc *dev;
197 dev = (struct disk_devdesc *)devdata;
198 bcd.dv_strategy = userdisk_realstrategy;
199 bcd.dv_devdata = devdata;
200 bcd.dv_cache = ud_info[dev->d_unit].ud_bcache;
201 return (bcache_strategy(&bcd, rw, dblk + dev->d_offset,
206 userdisk_realstrategy(void *devdata, int rw, daddr_t dblk, size_t size,
207 char *buf, size_t *rsize)
209 struct disk_devdesc *dev = devdata;
221 off = dblk * ud_info[dev->d_unit].sectorsize;
222 rc = CALLBACK(diskread, dev->d_unit, off, buf, size, &resid);
226 *rsize = size - resid;
231 userdisk_ioctl(struct open_file *f, u_long cmd, void *data)
233 struct disk_devdesc *dev;
236 dev = (struct disk_devdesc *)f->f_devdata;
237 rc = disk_ioctl(dev, cmd, data);
241 return (CALLBACK(diskioctl, dev->d_unit, cmd, data));