1/*- 2 * Copyright (c) 2011 Google, Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 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. 13 * 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 24 * SUCH DAMAGE. 25 */ 26 27#include <sys/cdefs.h> 28__FBSDID("$FreeBSD$"); 29 30/* 31 * Userboot disk image handling. 32 */ 33 34#include <sys/disk.h> 35#include <stand.h> 36#include <stdarg.h> 37#include <bootstrap.h> 38 39#include "disk.h" 40#include "libuserboot.h" 41 42struct userdisk_info { 43 uint64_t mediasize; 44 uint16_t sectorsize; 45}; 46 47int userboot_disk_maxunit = 0; 48 49static int userdisk_maxunit = 0; 50static struct userdisk_info *ud_info; 51 52static int userdisk_init(void); 53static void userdisk_cleanup(void); 54static int userdisk_strategy(void *devdata, int flag, daddr_t dblk, 55 size_t size, char *buf, size_t *rsize); 56static int userdisk_open(struct open_file *f, ...); 57static int userdisk_close(struct open_file *f); 58static int userdisk_ioctl(struct open_file *f, u_long cmd, void *data); 59static void userdisk_print(int verbose); 60 61struct devsw userboot_disk = { 62 "disk", 63 DEVT_DISK, 64 userdisk_init, 65 userdisk_strategy, 66 userdisk_open, 67 userdisk_close, 68 userdisk_ioctl, 69 userdisk_print, 70 userdisk_cleanup 71}; 72 73/* 74 * Initialize userdisk_info structure for each disk. 75 */ 76static int 77userdisk_init(void) 78{ 79 off_t mediasize; 80 u_int sectorsize; 81 int i; 82 83 userdisk_maxunit = userboot_disk_maxunit; 84 if (userdisk_maxunit > 0) { 85 ud_info = malloc(sizeof(*ud_info) * userdisk_maxunit); 86 if (ud_info == NULL) 87 return (ENOMEM); 88 for (i = 0; i < userdisk_maxunit; i++) { 89 if (CALLBACK(diskioctl, i, DIOCGSECTORSIZE, 90 §orsize) != 0 || CALLBACK(diskioctl, i, 91 DIOCGMEDIASIZE, &mediasize) != 0) 92 return (ENXIO); 93 ud_info[i].mediasize = mediasize; 94 ud_info[i].sectorsize = sectorsize; 95 } 96 } 97 98 return(0); 99} 100 101static void 102userdisk_cleanup(void) 103{ 104 105 if (userdisk_maxunit > 0) 106 free(ud_info); 107 disk_cleanup(&userboot_disk); 108} 109 110/* 111 * Print information about disks 112 */ 113static void 114userdisk_print(int verbose) 115{ 116 struct disk_devdesc dev; 117 char line[80]; 118 int i; 119 120 for (i = 0; i < userdisk_maxunit; i++) { 121 sprintf(line, " disk%d: Guest drive image\n", i); 122 pager_output(line); 123 dev.d_dev = &userboot_disk; 124 dev.d_unit = i; 125 dev.d_slice = -1; 126 dev.d_partition = -1; 127 if (disk_open(&dev, ud_info[i].mediasize, 128 ud_info[i].sectorsize, 0) == 0) { 129 sprintf(line, " disk%d", i); 130 disk_print(&dev, line, verbose); 131 disk_close(&dev); 132 } 133 } 134} 135 136/* 137 * Attempt to open the disk described by (dev) for use by (f). 138 */ 139static int 140userdisk_open(struct open_file *f, ...) 141{ 142 va_list ap; 143 struct disk_devdesc *dev; 144 145 va_start(ap, f); 146 dev = va_arg(ap, struct disk_devdesc *); 147 va_end(ap); 148 149 if (dev->d_unit < 0 || dev->d_unit >= userdisk_maxunit) 150 return (EIO); 151 152 return (disk_open(dev, ud_info[dev->d_unit].mediasize, 153 ud_info[dev->d_unit].sectorsize, 0)); 154} 155 156static int 157userdisk_close(struct open_file *f) 158{ 159 struct disk_devdesc *dev; 160 161 dev = (struct disk_devdesc *)f->f_devdata; 162 return (disk_close(dev)); 163} 164 165static int 166userdisk_strategy(void *devdata, int rw, daddr_t dblk, size_t size, 167 char *buf, size_t *rsize) 168{ 169 struct disk_devdesc *dev = devdata; 170 uint64_t off; 171 size_t resid; 172 int rc; 173 174 if (rw == F_WRITE) 175 return (EROFS); 176 if (rw != F_READ) 177 return (EINVAL); 178 if (rsize) 179 *rsize = 0; 180 off = (dblk + dev->d_offset) * ud_info[dev->d_unit].sectorsize; 181 rc = CALLBACK(diskread, dev->d_unit, off, buf, size, &resid); 182 if (rc) 183 return (rc); 184 if (rsize) 185 *rsize = size - resid; 186 return (0); 187} 188 189static int 190userdisk_ioctl(struct open_file *f, u_long cmd, void *data) 191{ 192 struct disk_devdesc *dev; 193 194 dev = (struct disk_devdesc *)f->f_devdata; 195 return (CALLBACK(diskioctl, dev->d_unit, cmd, data)); 196} 197