1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh * Copyright (c) 2011 Google, Inc.
3ca987d46SWarner Losh * All rights reserved.
4ca987d46SWarner Losh *
5ca987d46SWarner Losh * Redistribution and use in source and binary forms, with or without
6ca987d46SWarner Losh * modification, are permitted provided that the following conditions
7ca987d46SWarner Losh * are met:
8ca987d46SWarner Losh * 1. Redistributions of source code must retain the above copyright
9ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer.
10ca987d46SWarner Losh * 2. Redistributions in binary form must reproduce the above copyright
11ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer in the
12ca987d46SWarner Losh * documentation and/or other materials provided with the distribution.
13ca987d46SWarner Losh *
14ca987d46SWarner Losh * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15ca987d46SWarner Losh * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16ca987d46SWarner Losh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17ca987d46SWarner Losh * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18ca987d46SWarner Losh * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19ca987d46SWarner Losh * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20ca987d46SWarner Losh * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21ca987d46SWarner Losh * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22ca987d46SWarner Losh * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23ca987d46SWarner Losh * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24ca987d46SWarner Losh * SUCH DAMAGE.
25ca987d46SWarner Losh */
26ca987d46SWarner Losh
27ca987d46SWarner Losh /*
28ca987d46SWarner Losh * Userboot disk image handling.
29ca987d46SWarner Losh */
30ca987d46SWarner Losh
31ca987d46SWarner Losh #include <sys/disk.h>
32ca987d46SWarner Losh #include <stand.h>
33ca987d46SWarner Losh #include <stdarg.h>
34ca987d46SWarner Losh #include <bootstrap.h>
35ca987d46SWarner Losh
36ca987d46SWarner Losh #include "disk.h"
37ca987d46SWarner Losh #include "libuserboot.h"
38ca987d46SWarner Losh
39ca987d46SWarner Losh struct userdisk_info {
40ca987d46SWarner Losh uint64_t mediasize;
41ca987d46SWarner Losh uint16_t sectorsize;
42ca987d46SWarner Losh int ud_open; /* reference counter */
43ca987d46SWarner Losh void *ud_bcache; /* buffer cache data */
44ca987d46SWarner Losh };
45ca987d46SWarner Losh
46ca987d46SWarner Losh int userboot_disk_maxunit = 0;
47ca987d46SWarner Losh
48ca987d46SWarner Losh static int userdisk_maxunit = 0;
49ca987d46SWarner Losh static struct userdisk_info *ud_info;
50ca987d46SWarner Losh
51ca987d46SWarner Losh static int userdisk_init(void);
52ca987d46SWarner Losh static void userdisk_cleanup(void);
53ca987d46SWarner Losh static int userdisk_strategy(void *devdata, int flag, daddr_t dblk,
54ca987d46SWarner Losh size_t size, char *buf, size_t *rsize);
55ca987d46SWarner Losh static int userdisk_realstrategy(void *devdata, int flag, daddr_t dblk,
56ca987d46SWarner Losh size_t size, char *buf, size_t *rsize);
57ca987d46SWarner Losh static int userdisk_open(struct open_file *f, ...);
58ca987d46SWarner Losh static int userdisk_close(struct open_file *f);
59ca987d46SWarner Losh static int userdisk_ioctl(struct open_file *f, u_long cmd, void *data);
60ca987d46SWarner Losh static int userdisk_print(int verbose);
61ca987d46SWarner Losh
62ca987d46SWarner Losh struct devsw userboot_disk = {
634c460aafSWarner Losh .dv_name = "disk",
644c460aafSWarner Losh .dv_type = DEVT_DISK,
654c460aafSWarner Losh .dv_init = userdisk_init,
664c460aafSWarner Losh .dv_strategy = userdisk_strategy,
674c460aafSWarner Losh .dv_open = userdisk_open,
684c460aafSWarner Losh .dv_close = userdisk_close,
694c460aafSWarner Losh .dv_ioctl = userdisk_ioctl,
704c460aafSWarner Losh .dv_print = userdisk_print,
714c460aafSWarner Losh .dv_cleanup = userdisk_cleanup,
72ad759c73SWarner Losh .dv_fmtdev = disk_fmtdev,
738337ab69SWarner Losh .dv_parsedev = disk_parsedev,
74ca987d46SWarner Losh };
75ca987d46SWarner Losh
76ca987d46SWarner Losh /*
77ca987d46SWarner Losh * Initialize userdisk_info structure for each disk.
78ca987d46SWarner Losh */
79ca987d46SWarner Losh static int
userdisk_init(void)80ca987d46SWarner Losh userdisk_init(void)
81ca987d46SWarner Losh {
82ca987d46SWarner Losh off_t mediasize;
83ca987d46SWarner Losh u_int sectorsize;
84ca987d46SWarner Losh int i;
85ca987d46SWarner Losh
86ca987d46SWarner Losh userdisk_maxunit = userboot_disk_maxunit;
87ca987d46SWarner Losh if (userdisk_maxunit > 0) {
88ca987d46SWarner Losh ud_info = malloc(sizeof(*ud_info) * userdisk_maxunit);
89ca987d46SWarner Losh if (ud_info == NULL)
90ca987d46SWarner Losh return (ENOMEM);
91ca987d46SWarner Losh for (i = 0; i < userdisk_maxunit; i++) {
92ca987d46SWarner Losh if (CALLBACK(diskioctl, i, DIOCGSECTORSIZE,
93ca987d46SWarner Losh §orsize) != 0 || CALLBACK(diskioctl, i,
94ca987d46SWarner Losh DIOCGMEDIASIZE, &mediasize) != 0)
95ca987d46SWarner Losh return (ENXIO);
96ca987d46SWarner Losh ud_info[i].mediasize = mediasize;
97ca987d46SWarner Losh ud_info[i].sectorsize = sectorsize;
98ca987d46SWarner Losh ud_info[i].ud_open = 0;
99ca987d46SWarner Losh ud_info[i].ud_bcache = NULL;
100ca987d46SWarner Losh }
101ca987d46SWarner Losh }
102ca987d46SWarner Losh bcache_add_dev(userdisk_maxunit);
103ca987d46SWarner Losh return(0);
104ca987d46SWarner Losh }
105ca987d46SWarner Losh
106ca987d46SWarner Losh static void
userdisk_cleanup(void)107ca987d46SWarner Losh userdisk_cleanup(void)
108ca987d46SWarner Losh {
109ca987d46SWarner Losh
110ca987d46SWarner Losh if (userdisk_maxunit > 0)
111ca987d46SWarner Losh free(ud_info);
112ca987d46SWarner Losh }
113ca987d46SWarner Losh
114ca987d46SWarner Losh /*
115ca987d46SWarner Losh * Print information about disks
116ca987d46SWarner Losh */
117ca987d46SWarner Losh static int
userdisk_print(int verbose)118ca987d46SWarner Losh userdisk_print(int verbose)
119ca987d46SWarner Losh {
120ca987d46SWarner Losh struct disk_devdesc dev;
121ca987d46SWarner Losh char line[80];
122ca987d46SWarner Losh int i, ret = 0;
123ca987d46SWarner Losh
124ca987d46SWarner Losh if (userdisk_maxunit == 0)
125ca987d46SWarner Losh return (0);
126ca987d46SWarner Losh
127ca987d46SWarner Losh printf("%s devices:", userboot_disk.dv_name);
128ca987d46SWarner Losh if ((ret = pager_output("\n")) != 0)
129ca987d46SWarner Losh return (ret);
130ca987d46SWarner Losh
131ca987d46SWarner Losh for (i = 0; i < userdisk_maxunit; i++) {
132ca987d46SWarner Losh snprintf(line, sizeof(line),
133ca987d46SWarner Losh " disk%d: Guest drive image\n", i);
134ca987d46SWarner Losh ret = pager_output(line);
135ca987d46SWarner Losh if (ret != 0)
136ca987d46SWarner Losh break;
137de04d704SWarner Losh dev.dd.d_dev = &userboot_disk;
138de04d704SWarner Losh dev.dd.d_unit = i;
13914243f8dSIan Lepore dev.d_slice = D_SLICENONE;
14014243f8dSIan Lepore dev.d_partition = D_PARTNONE;
141ca987d46SWarner Losh if (disk_open(&dev, ud_info[i].mediasize,
142ca987d46SWarner Losh ud_info[i].sectorsize) == 0) {
143ca987d46SWarner Losh snprintf(line, sizeof(line), " disk%d", i);
144ca987d46SWarner Losh ret = disk_print(&dev, line, verbose);
145ca987d46SWarner Losh disk_close(&dev);
146ca987d46SWarner Losh if (ret != 0)
147ca987d46SWarner Losh break;
148ca987d46SWarner Losh }
149ca987d46SWarner Losh }
150ca987d46SWarner Losh return (ret);
151ca987d46SWarner Losh }
152ca987d46SWarner Losh
153ca987d46SWarner Losh /*
154ca987d46SWarner Losh * Attempt to open the disk described by (dev) for use by (f).
155ca987d46SWarner Losh */
156ca987d46SWarner Losh static int
userdisk_open(struct open_file * f,...)157ca987d46SWarner Losh userdisk_open(struct open_file *f, ...)
158ca987d46SWarner Losh {
159ca987d46SWarner Losh va_list ap;
160ca987d46SWarner Losh struct disk_devdesc *dev;
161ca987d46SWarner Losh
162ca987d46SWarner Losh va_start(ap, f);
163ca987d46SWarner Losh dev = va_arg(ap, struct disk_devdesc *);
164ca987d46SWarner Losh va_end(ap);
165ca987d46SWarner Losh
166de04d704SWarner Losh if (dev->dd.d_unit < 0 || dev->dd.d_unit >= userdisk_maxunit)
167ca987d46SWarner Losh return (EIO);
168de04d704SWarner Losh ud_info[dev->dd.d_unit].ud_open++;
169de04d704SWarner Losh if (ud_info[dev->dd.d_unit].ud_bcache == NULL)
170de04d704SWarner Losh ud_info[dev->dd.d_unit].ud_bcache = bcache_allocate();
171de04d704SWarner Losh return (disk_open(dev, ud_info[dev->dd.d_unit].mediasize,
172de04d704SWarner Losh ud_info[dev->dd.d_unit].sectorsize));
173ca987d46SWarner Losh }
174ca987d46SWarner Losh
175ca987d46SWarner Losh static int
userdisk_close(struct open_file * f)176ca987d46SWarner Losh userdisk_close(struct open_file *f)
177ca987d46SWarner Losh {
178ca987d46SWarner Losh struct disk_devdesc *dev;
179ca987d46SWarner Losh
180ca987d46SWarner Losh dev = (struct disk_devdesc *)f->f_devdata;
181de04d704SWarner Losh ud_info[dev->dd.d_unit].ud_open--;
182de04d704SWarner Losh if (ud_info[dev->dd.d_unit].ud_open == 0) {
183de04d704SWarner Losh bcache_free(ud_info[dev->dd.d_unit].ud_bcache);
184de04d704SWarner Losh ud_info[dev->dd.d_unit].ud_bcache = NULL;
185ca987d46SWarner Losh }
186ca987d46SWarner Losh return (disk_close(dev));
187ca987d46SWarner Losh }
188ca987d46SWarner Losh
189ca987d46SWarner Losh static int
userdisk_strategy(void * devdata,int rw,daddr_t dblk,size_t size,char * buf,size_t * rsize)190ca987d46SWarner Losh userdisk_strategy(void *devdata, int rw, daddr_t dblk, size_t size,
191ca987d46SWarner Losh char *buf, size_t *rsize)
192ca987d46SWarner Losh {
193ca987d46SWarner Losh struct bcache_devdata bcd;
194ca987d46SWarner Losh struct disk_devdesc *dev;
195ca987d46SWarner Losh
196ca987d46SWarner Losh dev = (struct disk_devdesc *)devdata;
197ca987d46SWarner Losh bcd.dv_strategy = userdisk_realstrategy;
198ca987d46SWarner Losh bcd.dv_devdata = devdata;
199de04d704SWarner Losh bcd.dv_cache = ud_info[dev->dd.d_unit].ud_bcache;
200ca987d46SWarner Losh return (bcache_strategy(&bcd, rw, dblk + dev->d_offset,
201ca987d46SWarner Losh size, buf, rsize));
202ca987d46SWarner Losh }
203ca987d46SWarner Losh
204ca987d46SWarner Losh static int
userdisk_realstrategy(void * devdata,int rw,daddr_t dblk,size_t size,char * buf,size_t * rsize)205ca987d46SWarner Losh userdisk_realstrategy(void *devdata, int rw, daddr_t dblk, size_t size,
206ca987d46SWarner Losh char *buf, size_t *rsize)
207ca987d46SWarner Losh {
208ca987d46SWarner Losh struct disk_devdesc *dev = devdata;
209ca987d46SWarner Losh uint64_t off;
210ca987d46SWarner Losh size_t resid;
211ca987d46SWarner Losh int rc;
212ca987d46SWarner Losh
213ca987d46SWarner Losh if (rsize)
214ca987d46SWarner Losh *rsize = 0;
215de04d704SWarner Losh off = dblk * ud_info[dev->dd.d_unit].sectorsize;
216e307eb94SToomas Soome switch (rw & F_MASK) {
217e307eb94SToomas Soome case F_READ:
218de04d704SWarner Losh rc = CALLBACK(diskread, dev->dd.d_unit, off, buf, size, &resid);
219e307eb94SToomas Soome break;
220e307eb94SToomas Soome case F_WRITE:
221e307eb94SToomas Soome rc = CALLBACK(diskwrite, dev->dd.d_unit, off, buf, size,
222e307eb94SToomas Soome &resid);
223e307eb94SToomas Soome break;
224e307eb94SToomas Soome default:
225e307eb94SToomas Soome rc = EINVAL;
226e307eb94SToomas Soome break;
227e307eb94SToomas Soome }
228ca987d46SWarner Losh if (rc)
229ca987d46SWarner Losh return (rc);
230ca987d46SWarner Losh if (rsize)
231ca987d46SWarner Losh *rsize = size - resid;
232ca987d46SWarner Losh return (0);
233ca987d46SWarner Losh }
234ca987d46SWarner Losh
235ca987d46SWarner Losh static int
userdisk_ioctl(struct open_file * f,u_long cmd,void * data)236ca987d46SWarner Losh userdisk_ioctl(struct open_file *f, u_long cmd, void *data)
237ca987d46SWarner Losh {
238ca987d46SWarner Losh struct disk_devdesc *dev;
239ca987d46SWarner Losh int rc;
240ca987d46SWarner Losh
241ca987d46SWarner Losh dev = (struct disk_devdesc *)f->f_devdata;
242ca987d46SWarner Losh rc = disk_ioctl(dev, cmd, data);
243ca987d46SWarner Losh if (rc != ENOTTY)
244ca987d46SWarner Losh return (rc);
245ca987d46SWarner Losh
246de04d704SWarner Losh return (CALLBACK(diskioctl, dev->dd.d_unit, cmd, data));
247ca987d46SWarner Losh }
248