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 <errno.h>
  19 #include <common.h>
  20 #include <mapmem.h>
  21 #include <part.h>
  22 #include <ext4fs.h>
  23 #include <fat.h>
  24 #include <fs.h>
  25 #include <sandboxfs.h>
  26 #include <ubifs_uboot.h>
  27 #include <asm/io.h>
  28 #include <div64.h>
  29 #include <linux/math64.h>
  30
  31 DECLARE_GLOBAL_DATA_PTR;
  32
  33 static block_dev_desc_t *fs_dev_desc;
  34 static disk_partition_t fs_partition;
  35 static int fs_type = FS_TYPE_ANY;
  36
  37 static inline int fs_probe_unsupported(block_dev_desc_t *fs_dev_desc,
  38                                       disk_partition_t *fs_partition)
  39 {
  40         printf("** Unrecognized filesystem type **\n");
  41         return -1;
  42 }
  43
  44 static inline int fs_ls_unsupported(const char *dirname)
  45 {
  46         return -1;
  47 }
  48
  49 static inline int fs_exists_unsupported(const char *filename)
  50 {
  51         return 0;
  52 }
  53
  54 static inline int fs_size_unsupported(const char *filename, loff_t *size)
  55 {
  56         return -1;
  57 }
  58
  59 static inline int fs_read_unsupported(const char *filename, void *buf,
  60                                       loff_t offset, loff_t len,
  61                                       loff_t *actread)
  62 {
  63         return -1;
  64 }
  65
  66 static inline int fs_write_unsupported(const char *filename, void *buf,
  67                                       loff_t offset, loff_t len,
  68                                       loff_t *actwrite)
  69 {
  70         return -1;
  71 }
  72
  73 static inline void fs_close_unsupported(void)
  74 {
  75 }
  76
  77 static inline int fs_uuid_unsupported(char *uuid_str)
  78 {
  79         return -1;
  80 }
  81
  82 struct fstype_info {
  83         int fstype;
  84         char *name;
  85         /*
  86          * Is it legal to pass NULL as .probe()'s  fs_dev_desc parameter? This
  87          * should be false in most cases. For "virtual" filesystems which
  88          * aren't based on a U-Boot block device (e.g. sandbox), this can be
  89          * set to true. This should also be true for the dumm entry at the end
  90          * of fstypes[], since that is essentially a "virtual" (non-existent)
  91          * filesystem.
  92          */
  93         bool null_dev_desc_ok;
  94         int (*probe)(block_dev_desc_t *fs_dev_desc,
  95                      disk_partition_t *fs_partition);
  96         int (*ls)(const char *dirname);
  97         int (*exists)(const char *filename);
  98         int (*size)(const char *filename, loff_t *size);
  99         int (*read)(const char *filename, void *buf, loff_t offset,
 100                     loff_t len, loff_t *actread);
 101         int (*write)(const char *filename, void *buf, loff_t offset,
 102                      loff_t len, loff_t *actwrite);
 103         void (*close)(void);
 104         int (*uuid)(char *uuid_str);
 105 };
 106
 107 static struct fstype_info fstypes[] = {
 108 #ifdef CONFIG_FS_FAT
 109         {
 110                 .fstype = FS_TYPE_FAT,
 111                 .name = "fat",
 112                 .null_dev_desc_ok = false,
 113                 .probe = fat_set_blk_dev,
 114                 .close = fat_close,
 115                 .ls = file_fat_ls,
 116                 .exists = fat_exists,
 117                 .size = fat_size,
 118                 .read = fat_read_file,
 119 #ifdef CONFIG_FAT_WRITE
 120                 .write = file_fat_write,
 121 #else
 122                 .write = fs_write_unsupported,
 123 #endif
 124                 .uuid = fs_uuid_unsupported,
 125         },
 126 #endif
 127 #ifdef CONFIG_FS_EXT4
 128         {
 129                 .fstype = FS_TYPE_EXT,
 130                 .name = "ext4",
 131                 .null_dev_desc_ok = false,
 132                 .probe = ext4fs_probe,
 133                 .close = ext4fs_close,
 134                 .ls = ext4fs_ls,
 135                 .exists = ext4fs_exists,
 136                 .size = ext4fs_size,
 137                 .read = ext4_read_file,
 138 #ifdef CONFIG_CMD_EXT4_WRITE
 139                 .write = ext4_write_file,
 140 #else
 141                 .write = fs_write_unsupported,
 142 #endif
 143                 .uuid = ext4fs_uuid,
 144         },
 145 #endif
 146 #ifdef CONFIG_SANDBOX
 147         {
 148                 .fstype = FS_TYPE_SANDBOX,
 149                 .name = "sandbox",
 150                 .null_dev_desc_ok = true,
 151                 .probe = sandbox_fs_set_blk_dev,
 152                 .close = sandbox_fs_close,
 153                 .ls = sandbox_fs_ls,
 154                 .exists = sandbox_fs_exists,
 155                 .size = sandbox_fs_size,
 156                 .read = fs_read_sandbox,
 157                 .write = fs_write_sandbox,
 158                 .uuid = fs_uuid_unsupported,
 159         },
 160 #endif
 161 #ifdef CONFIG_CMD_UBIFS
 162         {
 163                 .fstype = FS_TYPE_UBIFS,
 164                 .name = "ubifs",
 165                 .null_dev_desc_ok = true,
 166                 .probe = ubifs_set_blk_dev,
 167                 .close = ubifs_close,
 168                 .ls = ubifs_ls,
 169                 .exists = ubifs_exists,
 170                 .size = ubifs_size,
 171                 .read = ubifs_read,
 172                 .write = fs_write_unsupported,
 173                 .uuid = fs_uuid_unsupported,
 174         },
 175 #endif
 176         {
 177                 .fstype = FS_TYPE_ANY,
 178                 .name = "unsupported",
 179                 .null_dev_desc_ok = true,
 180                 .probe = fs_probe_unsupported,
 181                 .close = fs_close_unsupported,
 182                 .ls = fs_ls_unsupported,
 183                 .exists = fs_exists_unsupported,
 184                 .size = fs_size_unsupported,
 185                 .read = fs_read_unsupported,
 186                 .write = fs_write_unsupported,
 187                 .uuid = fs_uuid_unsupported,
 188         },
 189 };
 190
 191 static struct fstype_info *fs_get_info(int fstype)
 192 {
 193         struct fstype_info *info;
 194         int i;
 195
 196         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
 197                 if (fstype == info->fstype)
 198                         return info;
 199         }
 200
 201         /* Return the 'unsupported' sentinel */
 202         return info;
 203 }
 204
 205 int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
 206 {
 207         struct fstype_info *info;
 208         int part, i;
 209 #ifdef CONFIG_NEEDS_MANUAL_RELOC
 210         static int relocated;
 211
 212         if (!relocated) {
 213                 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
 214                                 i++, info++) {
 215                         info->name += gd->reloc_off;
 216                         info->probe += gd->reloc_off;
 217                         info->close += gd->reloc_off;
 218                         info->ls += gd->reloc_off;
 219                         info->read += gd->reloc_off;
 220                         info->write += gd->reloc_off;
 221                 }
 222                 relocated = 1;
 223         }
 224 #endif
 225
 226         part = get_device_and_partition(ifname, dev_part_str, &fs_dev_desc,
 227                                         &fs_partition, 1);
 228         if (part < 0)
 229                 return -1;
 230
 231         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 232                 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
 233                                 fstype != info->fstype)
 234                         continue;
 235
 236                 if (!fs_dev_desc && !info->null_dev_desc_ok)
 237                         continue;
 238
 239                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 240                         fs_type = info->fstype;
 241                         return 0;
 242                 }
 243         }
 244
 245         return -1;
 246 }
 247
 248 static void fs_close(void)
 249 {
 250         struct fstype_info *info = fs_get_info(fs_type);
 251
 252         info->close();
 253
 254         fs_type = FS_TYPE_ANY;
 255 }
 256
 257 int fs_uuid(char *uuid_str)
 258 {
 259         struct fstype_info *info = fs_get_info(fs_type);
 260
 261         return info->uuid(uuid_str);
 262 }
 263
 264 int fs_ls(const char *dirname)
 265 {
 266         int ret;
 267
 268         struct fstype_info *info = fs_get_info(fs_type);
 269
 270         ret = info->ls(dirname);
 271
 272         fs_type = FS_TYPE_ANY;
 273         fs_close();
 274
 275         return ret;
 276 }
 277
 278 int fs_exists(const char *filename)
 279 {
 280         int ret;
 281
 282         struct fstype_info *info = fs_get_info(fs_type);
 283
 284         ret = info->exists(filename);
 285
 286         fs_close();
 287
 288         return ret;
 289 }
 290
 291 int fs_size(const char *filename, loff_t *size)
 292 {
 293         int ret;
 294
 295         struct fstype_info *info = fs_get_info(fs_type);
 296
 297         ret = info->size(filename, size);
 298
 299         fs_close();
 300
 301         return ret;
 302 }
 303
 304 int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
 305             loff_t *actread)
 306 {
 307         struct fstype_info *info = fs_get_info(fs_type);
 308         void *buf;
 309         int ret;
 310
 311         /*
 312          * We don't actually know how many bytes are being read, since len==0
 313          * means read the whole file.
 314          */
 315         buf = map_sysmem(addr, len);
 316         ret = info->read(filename, buf, offset, len, actread);
 317         unmap_sysmem(buf);
 318
 319         /* If we requested a specific number of bytes, check we got it */
 320         if (ret == 0 && len && *actread != len)
 321                 printf("** %s shorter than offset + len **\n", filename);
 322         fs_close();
 323
 324         return ret;
 325 }
 326
 327 int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
 328              loff_t *actwrite)
 329 {
 330         struct fstype_info *info = fs_get_info(fs_type);
 331         void *buf;
 332         int ret;
 333
 334         buf = map_sysmem(addr, len);
 335         ret = info->write(filename, buf, offset, len, actwrite);
 336         unmap_sysmem(buf);
 337
 338         if (ret < 0 && len != *actwrite) {
 339                 printf("** Unable to write file %s **\n", filename);
 340                 ret = -1;
 341         }
 342         fs_close();
 343
 344         return ret;
 345 }
 346
 347 int do_size(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 348                 int fstype)
 349 {
 350         loff_t size;
 351
 352         if (argc != 4)
 353                 return CMD_RET_USAGE;
 354
 355         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 356                 return 1;
 357
 358         if (fs_size(argv[3], &size) < 0)
 359                 return CMD_RET_FAILURE;
 360
 361         setenv_hex("filesize", size);
 362
 363         return 0;
 364 }
 365
 366 int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 367                 int fstype)
 368 {
 369         unsigned long addr;
 370         const char *addr_str;
 371         const char *filename;
 372         loff_t bytes;
 373         loff_t pos;
 374         loff_t len_read;
 375         int ret;
 376         unsigned long time;
 377         char *ep;
 378
 379         if (argc < 2)
 380                 return CMD_RET_USAGE;
 381         if (argc > 7)
 382                 return CMD_RET_USAGE;
 383
 384         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
 385                 return 1;
 386
 387         if (argc >= 4) {
 388                 addr = simple_strtoul(argv[3], &ep, 16);
 389                 if (ep == argv[3] || *ep != '\0')
 390                         return CMD_RET_USAGE;
 391         } else {
 392                 addr_str = getenv("loadaddr");
 393                 if (addr_str != NULL)
 394                         addr = simple_strtoul(addr_str, NULL, 16);
 395                 else
 396                         addr = CONFIG_SYS_LOAD_ADDR;
 397         }
 398         if (argc >= 5) {
 399                 filename = argv[4];
 400         } else {
 401                 filename = getenv("bootfile");
 402                 if (!filename) {
 403                         puts("** No boot file defined **\n");
 404                         return 1;
 405                 }
 406         }
 407         if (argc >= 6)
 408                 bytes = simple_strtoul(argv[5], NULL, 16);
 409         else
 410                 bytes = 0;
 411         if (argc >= 7)
 412                 pos = simple_strtoul(argv[6], NULL, 16);
 413         else
 414                 pos = 0;
 415
 416         time = get_timer(0);
 417         ret = fs_read(filename, addr, pos, bytes, &len_read);
 418         time = get_timer(time);
 419         if (ret < 0)
 420                 return 1;
 421
 422         printf("%llu bytes read in %lu ms", len_read, time);
 423         if (time > 0) {
 424                 puts(" (");
 425                 print_size(div_u64(len_read, time) * 1000, "/s");
 426                 puts(")");
 427         }
 428         puts("\n");
 429
 430         setenv_hex("filesize", len_read);
 431
 432         return 0;
 433 }
 434
 435 int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 436         int fstype)
 437 {
 438         if (argc < 2)
 439                 return CMD_RET_USAGE;
 440         if (argc > 4)
 441                 return CMD_RET_USAGE;
 442
 443         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
 444                 return 1;
 445
 446         if (fs_ls(argc >= 4 ? argv[3] : "/"))
 447                 return 1;
 448
 449         return 0;
 450 }
 451
 452 int file_exists(const char *dev_type, const char *dev_part, const char *file,
 453                 int fstype)
 454 {
 455         if (fs_set_blk_dev(dev_type, dev_part, fstype))
 456                 return 0;
 457
 458         return fs_exists(file);
 459 }
 460
 461 int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 462                 int fstype)
 463 {
 464         unsigned long addr;
 465         const char *filename;
 466         loff_t bytes;
 467         loff_t pos;
 468         loff_t len;
 469         int ret;
 470         unsigned long time;
 471
 472         if (argc < 6 || argc > 7)
 473                 return CMD_RET_USAGE;
 474
 475         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 476                 return 1;
 477
 478         addr = simple_strtoul(argv[3], NULL, 16);
 479         filename = argv[4];
 480         bytes = simple_strtoul(argv[5], NULL, 16);
 481         if (argc >= 7)
 482                 pos = simple_strtoul(argv[6], NULL, 16);
 483         else
 484                 pos = 0;
 485
 486         time = get_timer(0);
 487         ret = fs_write(filename, addr, pos, bytes, &len);
 488         time = get_timer(time);
 489         if (ret < 0)
 490                 return 1;
 491
 492         printf("%llu bytes written in %lu ms", len, time);
 493         if (time > 0) {
 494                 puts(" (");
 495                 print_size(div_u64(len, time) * 1000, "/s");
 496                 puts(")");
 497         }
 498         puts("\n");
 499
 500         return 0;
 501 }
 502
 503 int do_fs_uuid(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
 504                 int fstype)
 505 {
 506         int ret;
 507         char uuid[37];
 508         memset(uuid, 0, sizeof(uuid));
 509
 510         if (argc < 3 || argc > 4)
 511                 return CMD_RET_USAGE;
 512
 513         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 514                 return 1;
 515
 516         ret = fs_uuid(uuid);
 517         if (ret)
 518                 return CMD_RET_FAILURE;
 519
 520         if (argc == 4)
 521                 setenv(argv[3], uuid);
 522         else
 523                 printf("%s\n", uuid);
 524
 525         return CMD_RET_SUCCESS;
 526 }
 527
 528 int do_fs_type(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 529 {
 530         struct fstype_info *info;
 531
 532         if (argc < 3 || argc > 4)
 533                 return CMD_RET_USAGE;
 534
 535         if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
 536                 return 1;
 537
 538         info = fs_get_info(fs_type);
 539
 540         if (argc == 4)
 541                 setenv(argv[3], info->name);
 542         else
 543                 printf("%s\n", info->name);
 544
 545         return CMD_RET_SUCCESS;
 546 }
 547