usr/src/boot/sys/boot/uboot/lib/disk.c

   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 }