drivers/block/sandbox.c

   1 /*
   2  * Copyright (C) 2013 Henrik Nordstrom <henrik@henriknordstrom.net>
   3  *
   4  * SPDX-License-Identifier:     GPL-2.0+
   5  */
   6
   7 #include <common.h>
   8 #include <blk.h>
   9 #include <dm.h>
  10 #include <fdtdec.h>
  11 #include <part.h>
  12 #include <os.h>
  13 #include <malloc.h>
  14 #include <sandboxblockdev.h>
  15 #include <linux/errno.h>
  16 #include <dm/device-internal.h>
  17
  18 DECLARE_GLOBAL_DATA_PTR;
  19
  20 #ifndef CONFIG_BLK
  21 static struct host_block_dev host_devices[CONFIG_HOST_MAX_DEVICES];
  22
  23 static struct host_block_dev *find_host_device(int dev)
  24 {
  25         if (dev >= 0 && dev < CONFIG_HOST_MAX_DEVICES)
  26                 return &host_devices[dev];
  27
  28         return NULL;
  29 }
  30 #endif
  31
  32 #ifdef CONFIG_BLK
  33 static unsigned long host_block_read(struct udevice *dev,
  34                                      unsigned long start, lbaint_t blkcnt,
  35                                      void *buffer)
  36 {
  37         struct host_block_dev *host_dev = dev_get_priv(dev);
  38         struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
  39
  40 #else
  41 static unsigned long host_block_read(struct blk_desc *block_dev,
  42                                      unsigned long start, lbaint_t blkcnt,
  43                                      void *buffer)
  44 {
  45         int dev = block_dev->devnum;
  46         struct host_block_dev *host_dev = find_host_device(dev);
  47
  48         if (!host_dev)
  49                 return -1;
  50 #endif
  51
  52         if (os_lseek(host_dev->fd, start * block_dev->blksz, OS_SEEK_SET) ==
  53                         -1) {
  54                 printf("ERROR: Invalid block %lx\n", start);
  55                 return -1;
  56         }
  57         ssize_t len = os_read(host_dev->fd, buffer, blkcnt * block_dev->blksz);
  58         if (len >= 0)
  59                 return len / block_dev->blksz;
  60         return -1;
  61 }
  62
  63 #ifdef CONFIG_BLK
  64 static unsigned long host_block_write(struct udevice *dev,
  65                                       unsigned long start, lbaint_t blkcnt,
  66                                       const void *buffer)
  67 {
  68         struct host_block_dev *host_dev = dev_get_priv(dev);
  69         struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
  70 #else
  71 static unsigned long host_block_write(struct blk_desc *block_dev,
  72                                       unsigned long start, lbaint_t blkcnt,
  73                                       const void *buffer)
  74 {
  75         int dev = block_dev->devnum;
  76         struct host_block_dev *host_dev = find_host_device(dev);
  77 #endif
  78
  79         if (os_lseek(host_dev->fd, start * block_dev->blksz, OS_SEEK_SET) ==
  80                         -1) {
  81                 printf("ERROR: Invalid block %lx\n", start);
  82                 return -1;
  83         }
  84         ssize_t len = os_write(host_dev->fd, buffer, blkcnt * block_dev->blksz);
  85         if (len >= 0)
  86                 return len / block_dev->blksz;
  87         return -1;
  88 }
  89
  90 #ifdef CONFIG_BLK
  91 int host_dev_bind(int devnum, char *filename)
  92 {
  93         struct host_block_dev *host_dev;
  94         struct udevice *dev;
  95         char dev_name[20], *str, *fname;
  96         int ret, fd;
  97
  98         /* Remove and unbind the old device, if any */
  99         ret = blk_get_device(IF_TYPE_HOST, devnum, &dev);
 100         if (ret == 0) {
 101                 ret = device_remove(dev, DM_REMOVE_NORMAL);
 102                 if (ret)
 103                         return ret;
 104                 ret = device_unbind(dev);
 105                 if (ret)
 106                         return ret;
 107         } else if (ret != -ENODEV) {
 108                 return ret;
 109         }
 110
 111         if (!filename)
 112                 return 0;
 113
 114         snprintf(dev_name, sizeof(dev_name), "host%d", devnum);
 115         str = strdup(dev_name);
 116         if (!str)
 117                 return -ENOMEM;
 118         fname = strdup(filename);
 119         if (!fname) {
 120                 free(str);
 121                 return -ENOMEM;
 122         }
 123
 124         fd = os_open(filename, OS_O_RDWR);
 125         if (fd == -1) {
 126                 printf("Failed to access host backing file '%s'\n", filename);
 127                 ret = -ENOENT;
 128                 goto err;
 129         }
 130         ret = blk_create_device(gd->dm_root, "sandbox_host_blk", str,
 131                                 IF_TYPE_HOST, devnum, 512,
 132                                 os_lseek(fd, 0, OS_SEEK_END) / 512, &dev);
 133         if (ret)
 134                 goto err_file;
 135         ret = device_probe(dev);
 136         if (ret) {
 137                 device_unbind(dev);
 138                 goto err_file;
 139         }
 140
 141         host_dev = dev_get_priv(dev);
 142         host_dev->fd = fd;
 143         host_dev->filename = fname;
 144
 145         return blk_prepare_device(dev);
 146 err_file:
 147         os_close(fd);
 148 err:
 149         free(fname);
 150         free(str);
 151         return ret;
 152 }
 153 #else
 154 int host_dev_bind(int dev, char *filename)
 155 {
 156         struct host_block_dev *host_dev = find_host_device(dev);
 157
 158         if (!host_dev)
 159                 return -1;
 160         if (host_dev->blk_dev.priv) {
 161                 os_close(host_dev->fd);
 162                 host_dev->blk_dev.priv = NULL;
 163         }
 164         if (host_dev->filename)
 165                 free(host_dev->filename);
 166         if (filename && *filename) {
 167                 host_dev->filename = strdup(filename);
 168         } else {
 169                 host_dev->filename = NULL;
 170                 return 0;
 171         }
 172
 173         host_dev->fd = os_open(host_dev->filename, OS_O_RDWR);
 174         if (host_dev->fd == -1) {
 175                 printf("Failed to access host backing file '%s'\n",
 176                        host_dev->filename);
 177                 return 1;
 178         }
 179
 180         struct blk_desc *blk_dev = &host_dev->blk_dev;
 181         blk_dev->if_type = IF_TYPE_HOST;
 182         blk_dev->priv = host_dev;
 183         blk_dev->blksz = 512;
 184         blk_dev->lba = os_lseek(host_dev->fd, 0, OS_SEEK_END) / blk_dev->blksz;
 185         blk_dev->block_read = host_block_read;
 186         blk_dev->block_write = host_block_write;
 187         blk_dev->devnum = dev;
 188         blk_dev->part_type = PART_TYPE_UNKNOWN;
 189         part_init(blk_dev);
 190
 191         return 0;
 192 }
 193 #endif
 194
 195 int host_get_dev_err(int devnum, struct blk_desc **blk_devp)
 196 {
 197 #ifdef CONFIG_BLK
 198         struct udevice *dev;
 199         int ret;
 200
 201         ret = blk_get_device(IF_TYPE_HOST, devnum, &dev);
 202         if (ret)
 203                 return ret;
 204         *blk_devp = dev_get_uclass_platdata(dev);
 205 #else
 206         struct host_block_dev *host_dev = find_host_device(devnum);
 207
 208         if (!host_dev)
 209                 return -ENODEV;
 210
 211         if (!host_dev->blk_dev.priv)
 212                 return -ENOENT;
 213
 214         *blk_devp = &host_dev->blk_dev;
 215 #endif
 216
 217         return 0;
 218 }
 219
 220 #ifdef CONFIG_BLK
 221 static const struct blk_ops sandbox_host_blk_ops = {
 222         .read   = host_block_read,
 223         .write  = host_block_write,
 224 };
 225
 226 U_BOOT_DRIVER(sandbox_host_blk) = {
 227         .name           = "sandbox_host_blk",
 228         .id             = UCLASS_BLK,
 229         .ops            = &sandbox_host_blk_ops,
 230         .priv_auto_alloc_size   = sizeof(struct host_block_dev),
 231 };
 232 #else
 233 U_BOOT_LEGACY_BLK(sandbox_host) = {
 234         .if_typename    = "host",
 235         .if_type        = IF_TYPE_HOST,
 236         .max_devs       = CONFIG_HOST_MAX_DEVICES,
 237         .get_dev        = host_get_dev_err,
 238 };
 239 #endif