fs/fs.c

   1 /*
   2  * Copyright (c) 2012, NVIDIA CORPORATION.  All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify it
   5  * under the terms and conditions of the GNU General Public License,
   6  * version 2, as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope it will be useful, but WITHOUT
   9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  11  * more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15  */
  16
  17 #include <config.h>
  18 #include <common.h>
  19 #include <part.h>
  20 #include <ext4fs.h>
  21 #include <fat.h>
  22 #include <fs.h>
  23 #include <sandboxfs.h>
  24 #include <asm/io.h>
  25
  26 DECLARE_GLOBAL_DATA_PTR;
  27
  28 static block_dev_desc_t *fs_dev_desc;
  29 static disk_partition_t fs_partition;
  30 static int fs_type = FS_TYPE_ANY;
  31
  32 static inline int fs_probe_unsupported(block_dev_desc_t *fs_dev_desc,
  33                                       disk_partition_t *fs_partition)
  34 {
  35         printf("** Unrecognized filesystem type **\n");
  36         return -1;
  37 }
  38
  39 static inline int fs_ls_unsupported(const char *dirname)
  40 {
  41         return -1;
  42 }
  43
  44 static inline int fs_read_unsupported(const char *filename, void *buf,
  45                                       int offset, int len)
  46 {
  47         return -1;
  48 }
  49
  50 static inline int fs_write_unsupported(const char *filename, void *buf,
  51                                       int offset, int len)
  52 {
  53         return -1;
  54 }
  55
  56 static inline void fs_close_unsupported(void)
  57 {
  58 }
  59
  60 struct fstype_info {
  61         int fstype;
  62         int (*probe)(block_dev_desc_t *fs_dev_desc,
  63                      disk_partition_t *fs_partition);
  64         int (*ls)(const char *dirname);
  65         int (*read)(const char *filename, void *buf, int offset, int len);
  66         int (*write)(const char *filename, void *buf, int offset, int len);
  67         void (*close)(void);
  68 };
  69
  70 static struct fstype_info fstypes[] = {
  71 #ifdef CONFIG_FS_FAT
  72         {
  73                 .fstype = FS_TYPE_FAT,
  74                 .probe = fat_set_blk_dev,
  75                 .close = fat_close,
  76                 .ls = file_fat_ls,
  77                 .read = fat_read_file,
  78         },
  79 #endif
  80 #ifdef CONFIG_FS_EXT4
  81         {
  82                 .fstype = FS_TYPE_EXT,
  83                 .probe = ext4fs_probe,
  84                 .close = ext4fs_close,
  85                 .ls = ext4fs_ls,
  86                 .read = ext4_read_file,
  87         },
  88 #endif
  89 #ifdef CONFIG_SANDBOX
  90         {
  91                 .fstype = FS_TYPE_SANDBOX,
  92                 .probe = sandbox_fs_set_blk_dev,
  93                 .close = sandbox_fs_close,
  94                 .ls = sandbox_fs_ls,
  95                 .read = fs_read_sandbox,
  96                 .write = fs_write_sandbox,
  97         },
  98 #endif
  99         {
 100                 .fstype = FS_TYPE_ANY,
 101                 .probe = fs_probe_unsupported,
 102                 .close = fs_close_unsupported,
 103                 .ls = fs_ls_unsupported,
 104                 .read = fs_read_unsupported,
 105                 .write = fs_write_unsupported,
 106         },
 107 };
 108
 109 static struct fstype_info *fs_get_info(int fstype)
 110 {
 111         struct fstype_info *info;
 112         int i;
 113
 114         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
 115                 if (fstype == info->fstype)
 116                         return info;
 117         }
 118
 119         /* Return the 'unsupported' sentinel */
 120         return info;
 121 }
 122
 123 int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
 124 {
 125         struct fstype_info *info;
 126         int part, i;
 127 #ifdef CONFIG_NEEDS_MANUAL_RELOC
 128         static int relocated;
 129
 130         if (!relocated) {
 131                 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
 132                                 i++, info++) {
 133                         info->probe += gd->reloc_off;
 134                         info->close += gd->reloc_off;
 135                         info->ls += gd->reloc_off;
 136                         info->read += gd->reloc_off;
 137                         info->write += gd->reloc_off;
 138                 }
 139                 relocated = 1;
 140         }
 141 #endif
 142
 143         part = get_device_and_partition(ifname, dev_part_str, &fs_dev_desc,
 144                                         &fs_partition, 1);
 145         if (part < 0)
 146                 return -1;
 147
 148         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 149                 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
 150                                 fstype != info->fstype)
 151                         continue;
 152
 153                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 154                         fs_type = info->fstype;
 155                         return 0;
 156                 }
 157         }
 158
 159         return -1;
 160 }
 161
 162 static void fs_close(void)
 163 {
 164         struct fstype_info *info = fs_get_info(fs_type);
 165
 166         info->close();
 167
 168         fs_type = FS_TYPE_ANY;
 169 }
 170
 171 int fs_ls(const char *dirname)
 172 {
 173         int ret;
 174
 175         struct fstype_info *info = fs_get_info(fs_type);
 176
 177         ret = info->ls(dirname);
 178
 179         fs_type = FS_TYPE_ANY;
 180         fs_close();
 181
 182         return ret;
 183 }
 184
 185 int fs_read(const char *filename, ulong addr, int offset, int len)
 186 {
 187         struct fstype_info *info = fs_get_info(fs_type);
 188         void *buf;
 189         int ret;
 190
 191         /*
 192          * We don't actually know how many bytes are being read, since len==0
 193          * means read the whole file.
 194          */
 195         buf = map_sysmem(addr, len);
 196         ret = info->read(filename, buf, offset, len);
 197         unmap_sysmem(buf);
 198
 199         /* If we requested a specific number of bytes, check we got it */
 200         if (ret >= 0 && len && ret != len) {
 201                 printf("** Unable to read file %s **\n", filename);
 202                 ret = -1;
 203         }
 204         fs_close();
 205
 206         return ret;
 207 }
 208
 209 int fs_write(const char *filename, ulong addr, int offset, int len)
 210 {
 211         struct fstype_info *info = fs_get_info(fs_type);
 212         void *buf;
 213         int ret;
 214
 215         /*
 216          * We don't actually know how many bytes are being read, since len==0
 217          * means read the whole file.
 218          */
 219         buf = map_sysmem(addr, len);
 220         ret = info->write(filename, buf, offset, len);
 221         unmap_sysmem(buf);
 222
 223         /* If we requested a specific number of bytes, check we got it */
 224         if (ret >= 0 && len && ret != len) {
 225                 printf("** Unable to write file %s **\n", filename);
 226                 ret = -1;
 227         }
 228         fs_close();
 229
 230         return ret;
 231 }
 232
 233 int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 234                 int fstype)
 235 {
 236         unsigned long addr;
 237         const char *addr_str;
 238         const char *filename;
 239         unsigned long bytes;
 240         unsigned long pos;
 241         int len_read;
 242         unsigned long time;
 243
 244         if (argc < 2)
 245                 return CMD_RET_USAGE;
 246         if (argc > 7)
 247                 return CMD_RET_USAGE;
 248
 249         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
 250                 return 1;
 251
 252         if (argc >= 4) {
 253                 addr = simple_strtoul(argv[3], NULL, 16);
 254         } else {
 255                 addr_str = getenv("loadaddr");
 256                 if (addr_str != NULL)
 257                         addr = simple_strtoul(addr_str, NULL, 16);
 258                 else
 259                         addr = CONFIG_SYS_LOAD_ADDR;
 260         }
 261         if (argc >= 5) {
 262                 filename = argv[4];
 263         } else {
 264                 filename = getenv("bootfile");
 265                 if (!filename) {
 266                         puts("** No boot file defined **\n");
 267                         return 1;
 268                 }
 269         }
 270         if (argc >= 6)
 271                 bytes = simple_strtoul(argv[5], NULL, 16);
 272         else
 273                 bytes = 0;
 274         if (argc >= 7)
 275                 pos = simple_strtoul(argv[6], NULL, 16);
 276         else
 277                 pos = 0;
 278
 279         time = get_timer(0);
 280         len_read = fs_read(filename, addr, pos, bytes);
 281         time = get_timer(time);
 282         if (len_read <= 0)
 283                 return 1;
 284
 285         printf("%d bytes read in %lu ms", len_read, time);
 286         if (time > 0) {
 287                 puts(" (");
 288                 print_size(len_read / time * 1000, "/s");
 289                 puts(")");
 290         }
 291         puts("\n");
 292
 293         setenv_hex("filesize", len_read);
 294
 295         return 0;
 296 }
 297
 298 int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 299         int fstype)
 300 {
 301         if (argc < 2)
 302                 return CMD_RET_USAGE;
 303         if (argc > 4)
 304                 return CMD_RET_USAGE;
 305
 306         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
 307                 return 1;
 308
 309         if (fs_ls(argc >= 4 ? argv[3] : "/"))
 310                 return 1;
 311
 312         return 0;
 313 }
 314
 315 int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 316                 int fstype)
 317 {
 318         unsigned long addr;
 319         const char *filename;
 320         unsigned long bytes;
 321         unsigned long pos;
 322         int len;
 323         unsigned long time;
 324
 325         if (argc < 6 || argc > 7)
 326                 return CMD_RET_USAGE;
 327
 328         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 329                 return 1;
 330
 331         filename = argv[3];
 332         addr = simple_strtoul(argv[4], NULL, 16);
 333         bytes = simple_strtoul(argv[5], NULL, 16);
 334         if (argc >= 7)
 335                 pos = simple_strtoul(argv[6], NULL, 16);
 336         else
 337                 pos = 0;
 338
 339         time = get_timer(0);
 340         len = fs_write(filename, addr, pos, bytes);
 341         time = get_timer(time);
 342         if (len <= 0)
 343                 return 1;
 344
 345         printf("%d bytes written in %lu ms", len, time);
 346         if (time > 0) {
 347                 puts(" (");
 348                 print_size(len / time * 1000, "/s");
 349                 puts(")");
 350         }
 351         puts("\n");
 352
 353         return 0;
 354 }