1 /*- 2 * Copyright (c) 2008 Semihalf, Rafal Jaworowski 3 * Copyright (c) 2009 Semihalf, Piotr Ziecik 4 * Copyright (c) 2012 Andrey V. Elsukov <ae@FreeBSD.org> 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 * 28 */ 29 30 /* 31 * Block storage I/O routines for U-Boot 32 */ 33 34 #include <sys/cdefs.h> 35 __FBSDID("$FreeBSD$"); 36 37 #include <sys/param.h> 38 #include <sys/disk.h> 39 #include <machine/stdarg.h> 40 #include <stand.h> 41 42 #include "api_public.h" 43 #include "bootstrap.h" 44 #include "disk.h" 45 #include "glue.h" 46 #include "libuboot.h" 47 48 #define stor_printf(fmt, args...) do { \ 49 printf("%s%d: ", dev->d_dev->dv_name, dev->d_unit); \ 50 printf(fmt, ##args); \ 51 } while (0) 52 53 #ifdef DEBUG 54 #define debugf(fmt, args...) do { printf("%s(): ", __func__); \ 55 printf(fmt,##args); } while (0) 56 #else 57 #define debugf(fmt, args...) 58 #endif 59 60 static struct { 61 int opened; /* device is opened */ 62 int handle; /* storage device handle */ 63 int type; /* storage type */ 64 off_t blocks; /* block count */ 65 u_int bsize; /* block size */ 66 } stor_info[UB_MAX_DEV]; 67 68 #define SI(dev) (stor_info[(dev)->d_unit]) 69 70 static int stor_info_no = 0; 71 static int stor_opendev(struct disk_devdesc *); 72 static int stor_readdev(struct disk_devdesc *, daddr_t, size_t, char *); 73 74 /* devsw I/F */ 75 static int stor_init(void); 76 static int stor_strategy(void *, int, daddr_t, size_t, char *, size_t *); 77 static int stor_open(struct open_file *, ...); 78 static int stor_close(struct open_file *); 79 static int stor_ioctl(struct open_file *f, u_long cmd, void *data); 80 static int stor_print(int); 81 static void stor_cleanup(void); 82 83 struct devsw uboot_storage = { 84 "disk", 85 DEVT_DISK, 86 stor_init, 87 stor_strategy, 88 stor_open, 89 stor_close, 90 stor_ioctl, 91 stor_print, 92 stor_cleanup 93 }; 94 95 static int 96 stor_init(void) 97 { 98 struct device_info *di; 99 int i; 100 101 if (devs_no == 0) { 102 printf("No U-Boot devices! Really enumerated?\n"); 103 return (-1); 104 } 105 106 for (i = 0; i < devs_no; i++) { 107 di = ub_dev_get(i); 108 if ((di != NULL) && (di->type & DEV_TYP_STOR)) { 109 if (stor_info_no >= UB_MAX_DEV) { 110 printf("Too many storage devices: %d\n", 111 stor_info_no); 112 return (-1); 113 } 114 stor_info[stor_info_no].handle = i; 115 stor_info[stor_info_no].opened = 0; 116 stor_info[stor_info_no].type = di->type; 117 stor_info[stor_info_no].blocks = 118 di->di_stor.block_count; 119 stor_info[stor_info_no].bsize = 120 di->di_stor.block_size; 121 stor_info_no++; 122 } 123 } 124 125 if (!stor_info_no) { 126 debugf("No storage devices\n"); 127 return (-1); 128 } 129 130 debugf("storage devices found: %d\n", stor_info_no); 131 return (0); 132 } 133 134 static void 135 stor_cleanup(void) 136 { 137 int i; 138 139 for (i = 0; i < stor_info_no; i++) 140 if (stor_info[i].opened > 0) 141 ub_dev_close(stor_info[i].handle); 142 disk_cleanup(&uboot_storage); 143 } 144 145 static int 146 stor_strategy(void *devdata, int rw, daddr_t blk, size_t size, char *buf, 147 size_t *rsize) 148 { 149 struct disk_devdesc *dev = (struct disk_devdesc *)devdata; 150 daddr_t bcount; 151 int err; 152 153 rw &= F_MASK; 154 if (rw != F_READ) { 155 stor_printf("write attempt, operation not supported!\n"); 156 return (EROFS); 157 } 158 159 if (size % SI(dev).bsize) { 160 stor_printf("size=%zu not multiple of device " 161 "block size=%d\n", 162 size, SI(dev).bsize); 163 return (EIO); 164 } 165 bcount = size / SI(dev).bsize; 166 if (rsize) 167 *rsize = 0; 168 169 err = stor_readdev(dev, blk + dev->d_offset, bcount, buf); 170 if (!err && rsize) 171 *rsize = size; 172 173 return (err); 174 } 175 176 static int 177 stor_open(struct open_file *f, ...) 178 { 179 va_list ap; 180 struct disk_devdesc *dev; 181 182 va_start(ap, f); 183 dev = va_arg(ap, struct disk_devdesc *); 184 va_end(ap); 185 186 return (stor_opendev(dev)); 187 } 188 189 static int 190 stor_opendev(struct disk_devdesc *dev) 191 { 192 int err; 193 194 if (dev->d_unit < 0 || dev->d_unit >= stor_info_no) 195 return (EIO); 196 197 if (SI(dev).opened == 0) { 198 err = ub_dev_open(SI(dev).handle); 199 if (err != 0) { 200 stor_printf("device open failed with error=%d, " 201 "handle=%d\n", err, SI(dev).handle); 202 return (ENXIO); 203 } 204 SI(dev).opened++; 205 } 206 return (disk_open(dev, SI(dev).blocks * SI(dev).bsize, SI(dev).bsize)); 207 } 208 209 static int 210 stor_close(struct open_file *f) 211 { 212 struct disk_devdesc *dev; 213 214 dev = (struct disk_devdesc *)(f->f_devdata); 215 return (disk_close(dev)); 216 } 217 218 static int 219 stor_readdev(struct disk_devdesc *dev, daddr_t blk, size_t size, char *buf) 220 { 221 lbasize_t real_size; 222 int err; 223 224 debugf("reading blk=%d size=%d @ 0x%08x\n", (int)blk, size, (uint32_t)buf); 225 226 err = ub_dev_read(SI(dev).handle, buf, size, blk, &real_size); 227 if (err != 0) { 228 stor_printf("read failed, error=%d\n", err); 229 return (EIO); 230 } 231 232 if (real_size != size) { 233 stor_printf("real size != size\n"); 234 err = EIO; 235 } 236 237 return (err); 238 } 239 240 static int 241 stor_print(int verbose) 242 { 243 struct disk_devdesc dev; 244 static char line[80]; 245 int i; 246 247 if (stor_info_no == 0) 248 return (ret); 249 250 printf("%s devices:", uboot_storage.dv_name); 251 if ((ret = pager_output("\n")) != 0) 252 return (ret); 253 254 for (i = 0; i < stor_info_no; i++) { 255 dev.d_dev = &uboot_storage; 256 dev.d_unit = i; 257 dev.d_slice = -1; 258 dev.d_partition = -1; 259 sprintf(line, "\tdisk%d (%s)\n", i, 260 ub_stor_type(SI(&dev).type)); 261 pager_output(line); 262 if (stor_opendev(&dev) == 0) { 263 sprintf(line, "\tdisk%d", i); 264 disk_print(&dev, line, verbose); 265 disk_close(&dev); 266 } 267 } 268 } 269 270 static int 271 stor_ioctl(struct open_file *f, u_long cmd, void *data) 272 { 273 struct disk_devdesc *dev; 274 275 dev = (struct disk_devdesc *)f->f_devdata; 276 switch (cmd) { 277 case DIOCGSECTORSIZE: 278 *(u_int *)data = SI(dev).bsize; 279 break; 280 case DIOCGMEDIASIZE: 281 *(off_t *)data = SI(dev).bsize * SI(dev).blocks; 282 break; 283 default: 284 return (ENOTTY); 285 } 286 return (0); 287 } 288 289 290 /* 291 * Return the device unit number for the given type and type-relative unit 292 * number. 293 */ 294 int 295 uboot_diskgetunit(int type, int type_unit) 296 { 297 int local_type_unit; 298 int i; 299 300 local_type_unit = 0; 301 for (i = 0; i < stor_info_no; i++) { 302 if ((stor_info[i].type & type) == type) { 303 if (local_type_unit == type_unit) { 304 return (i); 305 } 306 local_type_unit++; 307 } 308 } 309 310 return (-1); 311 } 312