fs/fs.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Copyright (c) 2012, NVIDIA CORPORATION.  All rights reserved.
   4  */
   5
   6 #define LOG_CATEGORY LOGC_CORE
   7
   8 #include <command.h>
   9 #include <config.h>
  10 #include <display_options.h>
  11 #include <errno.h>
  12 #include <env.h>
  13 #include <lmb.h>
  14 #include <log.h>
  15 #include <malloc.h>
  16 #include <mapmem.h>
  17 #include <part.h>
  18 #include <ext4fs.h>
  19 #include <fat.h>
  20 #include <fs.h>
  21 #include <sandboxfs.h>
  22 #include <semihostingfs.h>
  23 #include <time.h>
  24 #include <ubifs_uboot.h>
  25 #include <btrfs.h>
  26 #include <asm/global_data.h>
  27 #include <asm/io.h>
  28 #include <div64.h>
  29 #include <linux/math64.h>
  30 #include <linux/sizes.h>
  31 #include <efi_loader.h>
  32 #include <squashfs.h>
  33 #include <erofs.h>
  34
  35 DECLARE_GLOBAL_DATA_PTR;
  36
  37 static struct blk_desc *fs_dev_desc;
  38 static int fs_dev_part;
  39 static struct disk_partition fs_partition;
  40 static int fs_type = FS_TYPE_ANY;
  41
  42 void fs_set_type(int type)
  43 {
  44         fs_type = type;
  45 }
  46
  47 static inline int fs_probe_unsupported(struct blk_desc *fs_dev_desc,
  48                                       struct disk_partition *fs_partition)
  49 {
  50         log_debug("Unrecognized filesystem type\n");
  51         return -1;
  52 }
  53
  54 static inline int fs_ls_unsupported(const char *dirname)
  55 {
  56         return -1;
  57 }
  58
  59 /* generic implementation of ls in terms of opendir/readdir/closedir */
  60 __maybe_unused
  61 static int fs_ls_generic(const char *dirname)
  62 {
  63         struct fs_dir_stream *dirs;
  64         struct fs_dirent *dent;
  65         int nfiles = 0, ndirs = 0;
  66
  67         dirs = fs_opendir(dirname);
  68         if (!dirs)
  69                 return -errno;
  70
  71         while ((dent = fs_readdir(dirs))) {
  72                 if (dent->type == FS_DT_DIR) {
  73                         printf("            %s/\n", dent->name);
  74                         ndirs++;
  75                 } else if (dent->type == FS_DT_LNK) {
  76                         printf("    <SYM>   %s\n", dent->name);
  77                         nfiles++;
  78                 } else {
  79                         printf(" %8lld   %s\n", dent->size, dent->name);
  80                         nfiles++;
  81                 }
  82         }
  83
  84         fs_closedir(dirs);
  85
  86         printf("\n%d file(s), %d dir(s)\n\n", nfiles, ndirs);
  87
  88         return 0;
  89 }
  90
  91 static inline int fs_exists_unsupported(const char *filename)
  92 {
  93         return 0;
  94 }
  95
  96 static inline int fs_size_unsupported(const char *filename, loff_t *size)
  97 {
  98         return -1;
  99 }
 100
 101 static inline int fs_read_unsupported(const char *filename, void *buf,
 102                                       loff_t offset, loff_t len,
 103                                       loff_t *actread)
 104 {
 105         return -1;
 106 }
 107
 108 static inline int fs_write_unsupported(const char *filename, void *buf,
 109                                       loff_t offset, loff_t len,
 110                                       loff_t *actwrite)
 111 {
 112         return -1;
 113 }
 114
 115 static inline int fs_ln_unsupported(const char *filename, const char *target)
 116 {
 117         return -1;
 118 }
 119
 120 static inline void fs_close_unsupported(void)
 121 {
 122 }
 123
 124 static inline int fs_uuid_unsupported(char *uuid_str)
 125 {
 126         return -1;
 127 }
 128
 129 static inline int fs_opendir_unsupported(const char *filename,
 130                                          struct fs_dir_stream **dirs)
 131 {
 132         return -EACCES;
 133 }
 134
 135 static inline int fs_unlink_unsupported(const char *filename)
 136 {
 137         return -1;
 138 }
 139
 140 static inline int fs_mkdir_unsupported(const char *dirname)
 141 {
 142         return -1;
 143 }
 144
 145 struct fstype_info {
 146         int fstype;
 147         char *name;
 148         /*
 149          * Is it legal to pass NULL as .probe()'s  fs_dev_desc parameter? This
 150          * should be false in most cases. For "virtual" filesystems which
 151          * aren't based on a U-Boot block device (e.g. sandbox), this can be
 152          * set to true. This should also be true for the dummy entry at the end
 153          * of fstypes[], since that is essentially a "virtual" (non-existent)
 154          * filesystem.
 155          */
 156         bool null_dev_desc_ok;
 157         int (*probe)(struct blk_desc *fs_dev_desc,
 158                      struct disk_partition *fs_partition);
 159         int (*ls)(const char *dirname);
 160         int (*exists)(const char *filename);
 161         int (*size)(const char *filename, loff_t *size);
 162         int (*read)(const char *filename, void *buf, loff_t offset,
 163                     loff_t len, loff_t *actread);
 164         int (*write)(const char *filename, void *buf, loff_t offset,
 165                      loff_t len, loff_t *actwrite);
 166         void (*close)(void);
 167         int (*uuid)(char *uuid_str);
 168         /*
 169          * Open a directory stream.  On success return 0 and directory
 170          * stream pointer via 'dirsp'.  On error, return -errno.  See
 171          * fs_opendir().
 172          */
 173         int (*opendir)(const char *filename, struct fs_dir_stream **dirsp);
 174         /*
 175          * Read next entry from directory stream.  On success return 0
 176          * and directory entry pointer via 'dentp'.  On error return
 177          * -errno.  See fs_readdir().
 178          */
 179         int (*readdir)(struct fs_dir_stream *dirs, struct fs_dirent **dentp);
 180         /* see fs_closedir() */
 181         void (*closedir)(struct fs_dir_stream *dirs);
 182         int (*unlink)(const char *filename);
 183         int (*mkdir)(const char *dirname);
 184         int (*ln)(const char *filename, const char *target);
 185 };
 186
 187 static struct fstype_info fstypes[] = {
 188 #if CONFIG_IS_ENABLED(FS_FAT)
 189         {
 190                 .fstype = FS_TYPE_FAT,
 191                 .name = "fat",
 192                 .null_dev_desc_ok = false,
 193                 .probe = fat_set_blk_dev,
 194                 .close = fat_close,
 195                 .ls = fs_ls_generic,
 196                 .exists = fat_exists,
 197                 .size = fat_size,
 198                 .read = fat_read_file,
 199 #if CONFIG_IS_ENABLED(FAT_WRITE)
 200                 .write = file_fat_write,
 201                 .unlink = fat_unlink,
 202                 .mkdir = fat_mkdir,
 203 #else
 204                 .write = fs_write_unsupported,
 205                 .unlink = fs_unlink_unsupported,
 206                 .mkdir = fs_mkdir_unsupported,
 207 #endif
 208                 .uuid = fat_uuid,
 209                 .opendir = fat_opendir,
 210                 .readdir = fat_readdir,
 211                 .closedir = fat_closedir,
 212                 .ln = fs_ln_unsupported,
 213         },
 214 #endif
 215
 216 #if CONFIG_IS_ENABLED(FS_EXT4)
 217         {
 218                 .fstype = FS_TYPE_EXT,
 219                 .name = "ext4",
 220                 .null_dev_desc_ok = false,
 221                 .probe = ext4fs_probe,
 222                 .close = ext4fs_close,
 223                 .ls = ext4fs_ls,
 224                 .exists = ext4fs_exists,
 225                 .size = ext4fs_size,
 226                 .read = ext4_read_file,
 227 #ifdef CONFIG_EXT4_WRITE
 228                 .write = ext4_write_file,
 229                 .ln = ext4fs_create_link,
 230 #else
 231                 .write = fs_write_unsupported,
 232                 .ln = fs_ln_unsupported,
 233 #endif
 234                 .uuid = ext4fs_uuid,
 235                 .opendir = fs_opendir_unsupported,
 236                 .unlink = fs_unlink_unsupported,
 237                 .mkdir = fs_mkdir_unsupported,
 238         },
 239 #endif
 240 #if IS_ENABLED(CONFIG_SANDBOX) && !IS_ENABLED(CONFIG_XPL_BUILD)
 241         {
 242                 .fstype = FS_TYPE_SANDBOX,
 243                 .name = "sandbox",
 244                 .null_dev_desc_ok = true,
 245                 .probe = sandbox_fs_set_blk_dev,
 246                 .close = sandbox_fs_close,
 247                 .ls = sandbox_fs_ls,
 248                 .exists = sandbox_fs_exists,
 249                 .size = sandbox_fs_size,
 250                 .read = fs_read_sandbox,
 251                 .write = fs_write_sandbox,
 252                 .uuid = fs_uuid_unsupported,
 253                 .opendir = fs_opendir_unsupported,
 254                 .unlink = fs_unlink_unsupported,
 255                 .mkdir = fs_mkdir_unsupported,
 256                 .ln = fs_ln_unsupported,
 257         },
 258 #endif
 259 #if CONFIG_IS_ENABLED(SEMIHOSTING)
 260         {
 261                 .fstype = FS_TYPE_SEMIHOSTING,
 262                 .name = "semihosting",
 263                 .null_dev_desc_ok = true,
 264                 .probe = smh_fs_set_blk_dev,
 265                 .close = fs_close_unsupported,
 266                 .ls = fs_ls_unsupported,
 267                 .exists = fs_exists_unsupported,
 268                 .size = smh_fs_size,
 269                 .read = smh_fs_read,
 270                 .write = smh_fs_write,
 271                 .uuid = fs_uuid_unsupported,
 272                 .opendir = fs_opendir_unsupported,
 273                 .unlink = fs_unlink_unsupported,
 274                 .mkdir = fs_mkdir_unsupported,
 275                 .ln = fs_ln_unsupported,
 276         },
 277 #endif
 278 #ifndef CONFIG_XPL_BUILD
 279 #ifdef CONFIG_CMD_UBIFS
 280         {
 281                 .fstype = FS_TYPE_UBIFS,
 282                 .name = "ubifs",
 283                 .null_dev_desc_ok = true,
 284                 .probe = ubifs_set_blk_dev,
 285                 .close = ubifs_close,
 286                 .ls = ubifs_ls,
 287                 .exists = ubifs_exists,
 288                 .size = ubifs_size,
 289                 .read = ubifs_read,
 290                 .write = fs_write_unsupported,
 291                 .uuid = fs_uuid_unsupported,
 292                 .opendir = fs_opendir_unsupported,
 293                 .unlink = fs_unlink_unsupported,
 294                 .mkdir = fs_mkdir_unsupported,
 295                 .ln = fs_ln_unsupported,
 296         },
 297 #endif
 298 #endif
 299 #ifndef CONFIG_XPL_BUILD
 300 #ifdef CONFIG_FS_BTRFS
 301         {
 302                 .fstype = FS_TYPE_BTRFS,
 303                 .name = "btrfs",
 304                 .null_dev_desc_ok = false,
 305                 .probe = btrfs_probe,
 306                 .close = btrfs_close,
 307                 .ls = btrfs_ls,
 308                 .exists = btrfs_exists,
 309                 .size = btrfs_size,
 310                 .read = btrfs_read,
 311                 .write = fs_write_unsupported,
 312                 .uuid = btrfs_uuid,
 313                 .opendir = fs_opendir_unsupported,
 314                 .unlink = fs_unlink_unsupported,
 315                 .mkdir = fs_mkdir_unsupported,
 316                 .ln = fs_ln_unsupported,
 317         },
 318 #endif
 319 #endif
 320 #if CONFIG_IS_ENABLED(FS_SQUASHFS)
 321         {
 322                 .fstype = FS_TYPE_SQUASHFS,
 323                 .name = "squashfs",
 324                 .null_dev_desc_ok = false,
 325                 .probe = sqfs_probe,
 326                 .opendir = sqfs_opendir,
 327                 .readdir = sqfs_readdir,
 328                 .ls = fs_ls_generic,
 329                 .read = sqfs_read,
 330                 .size = sqfs_size,
 331                 .close = sqfs_close,
 332                 .closedir = sqfs_closedir,
 333                 .exists = sqfs_exists,
 334                 .uuid = fs_uuid_unsupported,
 335                 .write = fs_write_unsupported,
 336                 .ln = fs_ln_unsupported,
 337                 .unlink = fs_unlink_unsupported,
 338                 .mkdir = fs_mkdir_unsupported,
 339         },
 340 #endif
 341 #if IS_ENABLED(CONFIG_FS_EROFS)
 342         {
 343                 .fstype = FS_TYPE_EROFS,
 344                 .name = "erofs",
 345                 .null_dev_desc_ok = false,
 346                 .probe = erofs_probe,
 347                 .opendir = erofs_opendir,
 348                 .readdir = erofs_readdir,
 349                 .ls = fs_ls_generic,
 350                 .read = erofs_read,
 351                 .size = erofs_size,
 352                 .close = erofs_close,
 353                 .closedir = erofs_closedir,
 354                 .exists = erofs_exists,
 355                 .uuid = fs_uuid_unsupported,
 356                 .write = fs_write_unsupported,
 357                 .ln = fs_ln_unsupported,
 358                 .unlink = fs_unlink_unsupported,
 359                 .mkdir = fs_mkdir_unsupported,
 360         },
 361 #endif
 362         {
 363                 .fstype = FS_TYPE_ANY,
 364                 .name = "unsupported",
 365                 .null_dev_desc_ok = true,
 366                 .probe = fs_probe_unsupported,
 367                 .close = fs_close_unsupported,
 368                 .ls = fs_ls_unsupported,
 369                 .exists = fs_exists_unsupported,
 370                 .size = fs_size_unsupported,
 371                 .read = fs_read_unsupported,
 372                 .write = fs_write_unsupported,
 373                 .uuid = fs_uuid_unsupported,
 374                 .opendir = fs_opendir_unsupported,
 375                 .unlink = fs_unlink_unsupported,
 376                 .mkdir = fs_mkdir_unsupported,
 377                 .ln = fs_ln_unsupported,
 378         },
 379 };
 380
 381 static struct fstype_info *fs_get_info(int fstype)
 382 {
 383         struct fstype_info *info;
 384         int i;
 385
 386         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
 387                 if (fstype == info->fstype)
 388                         return info;
 389         }
 390
 391         /* Return the 'unsupported' sentinel */
 392         return info;
 393 }
 394
 395 /**
 396  * fs_get_type() - Get type of current filesystem
 397  *
 398  * Return: filesystem type
 399  *
 400  * Returns filesystem type representing the current filesystem, or
 401  * FS_TYPE_ANY for any unrecognised filesystem.
 402  */
 403 int fs_get_type(void)
 404 {
 405         return fs_type;
 406 }
 407
 408 /**
 409  * fs_get_type_name() - Get type of current filesystem
 410  *
 411  * Return: Pointer to filesystem name
 412  *
 413  * Returns a string describing the current filesystem, or the sentinel
 414  * "unsupported" for any unrecognised filesystem.
 415  */
 416 const char *fs_get_type_name(void)
 417 {
 418         return fs_get_info(fs_type)->name;
 419 }
 420
 421 int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
 422 {
 423         struct fstype_info *info;
 424         int part, i;
 425
 426         part = part_get_info_by_dev_and_name_or_num(ifname, dev_part_str, &fs_dev_desc,
 427                                                     &fs_partition, 1);
 428         if (part < 0)
 429                 return -1;
 430
 431         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 432                 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
 433                                 fstype != info->fstype)
 434                         continue;
 435
 436                 if (!fs_dev_desc && !info->null_dev_desc_ok)
 437                         continue;
 438
 439                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 440                         fs_type = info->fstype;
 441                         fs_dev_part = part;
 442                         return 0;
 443                 }
 444         }
 445
 446         return -1;
 447 }
 448
 449 /* set current blk device w/ blk_desc + partition # */
 450 int fs_set_blk_dev_with_part(struct blk_desc *desc, int part)
 451 {
 452         struct fstype_info *info;
 453         int ret, i;
 454
 455         if (part >= 1)
 456                 ret = part_get_info(desc, part, &fs_partition);
 457         else
 458                 ret = part_get_info_whole_disk(desc, &fs_partition);
 459         if (ret)
 460                 return ret;
 461         fs_dev_desc = desc;
 462
 463         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 464                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 465                         fs_type = info->fstype;
 466                         fs_dev_part = part;
 467                         return 0;
 468                 }
 469         }
 470
 471         return -1;
 472 }
 473
 474 void fs_close(void)
 475 {
 476         struct fstype_info *info = fs_get_info(fs_type);
 477
 478         info->close();
 479
 480         fs_type = FS_TYPE_ANY;
 481 }
 482
 483 int fs_uuid(char *uuid_str)
 484 {
 485         struct fstype_info *info = fs_get_info(fs_type);
 486
 487         return info->uuid(uuid_str);
 488 }
 489
 490 int fs_ls(const char *dirname)
 491 {
 492         int ret;
 493
 494         struct fstype_info *info = fs_get_info(fs_type);
 495
 496         ret = info->ls(dirname);
 497
 498         fs_close();
 499
 500         return ret;
 501 }
 502
 503 int fs_exists(const char *filename)
 504 {
 505         int ret;
 506
 507         struct fstype_info *info = fs_get_info(fs_type);
 508
 509         ret = info->exists(filename);
 510
 511         fs_close();
 512
 513         return ret;
 514 }
 515
 516 int fs_size(const char *filename, loff_t *size)
 517 {
 518         int ret;
 519
 520         struct fstype_info *info = fs_get_info(fs_type);
 521
 522         ret = info->size(filename, size);
 523
 524         fs_close();
 525
 526         return ret;
 527 }
 528
 529 #if CONFIG_IS_ENABLED(LMB)
 530 /* Check if a file may be read to the given address */
 531 static int fs_read_lmb_check(const char *filename, ulong addr, loff_t offset,
 532                              loff_t len, struct fstype_info *info)
 533 {
 534         int ret;
 535         loff_t size;
 536         loff_t read_len;
 537
 538         /* get the actual size of the file */
 539         ret = info->size(filename, &size);
 540         if (ret)
 541                 return ret;
 542         if (offset >= size) {
 543                 /* offset >= EOF, no bytes will be written */
 544                 return 0;
 545         }
 546         read_len = size - offset;
 547
 548         /* limit to 'len' if it is smaller */
 549         if (len && len < read_len)
 550                 read_len = len;
 551
 552         lmb_dump_all();
 553
 554         if (lmb_alloc_addr(addr, read_len) == addr)
 555                 return 0;
 556
 557         log_err("** Reading file would overwrite reserved memory **\n");
 558         return -ENOSPC;
 559 }
 560 #endif
 561
 562 static int _fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
 563                     int do_lmb_check, loff_t *actread)
 564 {
 565         struct fstype_info *info = fs_get_info(fs_type);
 566         void *buf;
 567         int ret;
 568
 569 #if CONFIG_IS_ENABLED(LMB)
 570         if (do_lmb_check) {
 571                 ret = fs_read_lmb_check(filename, addr, offset, len, info);
 572                 if (ret)
 573                         return ret;
 574         }
 575 #endif
 576
 577         /*
 578          * We don't actually know how many bytes are being read, since len==0
 579          * means read the whole file.
 580          */
 581         buf = map_sysmem(addr, len);
 582         ret = info->read(filename, buf, offset, len, actread);
 583         unmap_sysmem(buf);
 584
 585         /* If we requested a specific number of bytes, check we got it */
 586         if (ret == 0 && len && *actread != len)
 587                 log_debug("** %s shorter than offset + len **\n", filename);
 588         fs_close();
 589
 590         return ret;
 591 }
 592
 593 int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
 594             loff_t *actread)
 595 {
 596         return _fs_read(filename, addr, offset, len, 0, actread);
 597 }
 598
 599 int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
 600              loff_t *actwrite)
 601 {
 602         struct fstype_info *info = fs_get_info(fs_type);
 603         void *buf;
 604         int ret;
 605
 606         buf = map_sysmem(addr, len);
 607         ret = info->write(filename, buf, offset, len, actwrite);
 608         unmap_sysmem(buf);
 609
 610         if (ret < 0 && len != *actwrite) {
 611                 log_err("** Unable to write file %s **\n", filename);
 612                 ret = -1;
 613         }
 614         fs_close();
 615
 616         return ret;
 617 }
 618
 619 struct fs_dir_stream *fs_opendir(const char *filename)
 620 {
 621         struct fstype_info *info = fs_get_info(fs_type);
 622         struct fs_dir_stream *dirs = NULL;
 623         int ret;
 624
 625         ret = info->opendir(filename, &dirs);
 626         fs_close();
 627         if (ret) {
 628                 errno = -ret;
 629                 return NULL;
 630         }
 631
 632         dirs->desc = fs_dev_desc;
 633         dirs->part = fs_dev_part;
 634
 635         return dirs;
 636 }
 637
 638 struct fs_dirent *fs_readdir(struct fs_dir_stream *dirs)
 639 {
 640         struct fstype_info *info;
 641         struct fs_dirent *dirent;
 642         int ret;
 643
 644         fs_set_blk_dev_with_part(dirs->desc, dirs->part);
 645         info = fs_get_info(fs_type);
 646
 647         ret = info->readdir(dirs, &dirent);
 648         fs_close();
 649         if (ret) {
 650                 errno = -ret;
 651                 return NULL;
 652         }
 653
 654         return dirent;
 655 }
 656
 657 void fs_closedir(struct fs_dir_stream *dirs)
 658 {
 659         struct fstype_info *info;
 660
 661         if (!dirs)
 662                 return;
 663
 664         fs_set_blk_dev_with_part(dirs->desc, dirs->part);
 665         info = fs_get_info(fs_type);
 666
 667         info->closedir(dirs);
 668         fs_close();
 669 }
 670
 671 int fs_unlink(const char *filename)
 672 {
 673         int ret;
 674
 675         struct fstype_info *info = fs_get_info(fs_type);
 676
 677         ret = info->unlink(filename);
 678
 679         fs_close();
 680
 681         return ret;
 682 }
 683
 684 int fs_mkdir(const char *dirname)
 685 {
 686         int ret;
 687
 688         struct fstype_info *info = fs_get_info(fs_type);
 689
 690         ret = info->mkdir(dirname);
 691
 692         fs_close();
 693
 694         return ret;
 695 }
 696
 697 int fs_ln(const char *fname, const char *target)
 698 {
 699         struct fstype_info *info = fs_get_info(fs_type);
 700         int ret;
 701
 702         ret = info->ln(fname, target);
 703
 704         if (ret < 0) {
 705                 log_err("** Unable to create link %s -> %s **\n", fname, target);
 706                 ret = -1;
 707         }
 708         fs_close();
 709
 710         return ret;
 711 }
 712
 713 int do_size(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 714             int fstype)
 715 {
 716         loff_t size;
 717
 718         if (argc != 4)
 719                 return CMD_RET_USAGE;
 720
 721         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 722                 return 1;
 723
 724         if (fs_size(argv[3], &size) < 0)
 725                 return CMD_RET_FAILURE;
 726
 727         env_set_hex("filesize", size);
 728
 729         return 0;
 730 }
 731
 732 int do_load(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 733             int fstype)
 734 {
 735         unsigned long addr;
 736         const char *addr_str;
 737         const char *filename;
 738         loff_t bytes;
 739         loff_t pos;
 740         loff_t len_read;
 741         int ret;
 742         unsigned long time;
 743         char *ep;
 744
 745         if (argc < 2)
 746                 return CMD_RET_USAGE;
 747         if (argc > 7)
 748                 return CMD_RET_USAGE;
 749
 750         if (fs_set_blk_dev(argv[1], cmd_arg2(argc, argv), fstype)) {
 751                 log_err("Can't set block device\n");
 752                 return 1;
 753         }
 754
 755         if (argc >= 4) {
 756                 addr = hextoul(argv[3], &ep);
 757                 if (ep == argv[3] || *ep != '\0')
 758                         return CMD_RET_USAGE;
 759         } else {
 760                 addr_str = env_get("loadaddr");
 761                 if (addr_str != NULL)
 762                         addr = hextoul(addr_str, NULL);
 763                 else
 764                         addr = CONFIG_SYS_LOAD_ADDR;
 765         }
 766         if (argc >= 5) {
 767                 filename = argv[4];
 768         } else {
 769                 filename = env_get("bootfile");
 770                 if (!filename) {
 771                         puts("** No boot file defined **\n");
 772                         return 1;
 773                 }
 774         }
 775         if (argc >= 6)
 776                 bytes = hextoul(argv[5], NULL);
 777         else
 778                 bytes = 0;
 779         if (argc >= 7)
 780                 pos = hextoul(argv[6], NULL);
 781         else
 782                 pos = 0;
 783
 784         time = get_timer(0);
 785         ret = _fs_read(filename, addr, pos, bytes, 1, &len_read);
 786         time = get_timer(time);
 787         if (ret < 0) {
 788                 log_err("Failed to load '%s'\n", filename);
 789                 return 1;
 790         }
 791
 792         efi_set_bootdev(argv[1], (argc > 2) ? argv[2] : "",
 793                         (argc > 4) ? argv[4] : "", map_sysmem(addr, 0),
 794                         len_read);
 795
 796         printf("%llu bytes read in %lu ms", len_read, time);
 797         if (time > 0) {
 798                 puts(" (");
 799                 print_size(div_u64(len_read, time) * 1000, "/s");
 800                 puts(")");
 801         }
 802         puts("\n");
 803
 804         env_set_hex("fileaddr", addr);
 805         env_set_hex("filesize", len_read);
 806
 807         return 0;
 808 }
 809
 810 int do_ls(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 811           int fstype)
 812 {
 813         if (argc < 2)
 814                 return CMD_RET_USAGE;
 815         if (argc > 4)
 816                 return CMD_RET_USAGE;
 817
 818         if (fs_set_blk_dev(argv[1], cmd_arg2(argc, argv), fstype))
 819                 return 1;
 820
 821         if (fs_ls(argc >= 4 ? argv[3] : "/"))
 822                 return 1;
 823
 824         return 0;
 825 }
 826
 827 int file_exists(const char *dev_type, const char *dev_part, const char *file,
 828                 int fstype)
 829 {
 830         if (fs_set_blk_dev(dev_type, dev_part, fstype))
 831                 return 0;
 832
 833         return fs_exists(file);
 834 }
 835
 836 int do_save(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 837             int fstype)
 838 {
 839         unsigned long addr;
 840         const char *filename;
 841         loff_t bytes;
 842         loff_t pos;
 843         loff_t len;
 844         int ret;
 845         unsigned long time;
 846
 847         if (argc < 6 || argc > 7)
 848                 return CMD_RET_USAGE;
 849
 850         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 851                 return 1;
 852
 853         addr = hextoul(argv[3], NULL);
 854         filename = argv[4];
 855         bytes = hextoul(argv[5], NULL);
 856         if (argc >= 7)
 857                 pos = hextoul(argv[6], NULL);
 858         else
 859                 pos = 0;
 860
 861         time = get_timer(0);
 862         ret = fs_write(filename, addr, pos, bytes, &len);
 863         time = get_timer(time);
 864         if (ret < 0)
 865                 return 1;
 866
 867         printf("%llu bytes written in %lu ms", len, time);
 868         if (time > 0) {
 869                 puts(" (");
 870                 print_size(div_u64(len, time) * 1000, "/s");
 871                 puts(")");
 872         }
 873         puts("\n");
 874
 875         return 0;
 876 }
 877
 878 int do_fs_uuid(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 879                int fstype)
 880 {
 881         int ret;
 882         char uuid[37];
 883         memset(uuid, 0, sizeof(uuid));
 884
 885         if (argc < 3 || argc > 4)
 886                 return CMD_RET_USAGE;
 887
 888         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 889                 return 1;
 890
 891         ret = fs_uuid(uuid);
 892         if (ret)
 893                 return CMD_RET_FAILURE;
 894
 895         if (argc == 4)
 896                 env_set(argv[3], uuid);
 897         else
 898                 printf("%s\n", uuid);
 899
 900         return CMD_RET_SUCCESS;
 901 }
 902
 903 int do_fs_type(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 904 {
 905         struct fstype_info *info;
 906
 907         if (argc < 3 || argc > 4)
 908                 return CMD_RET_USAGE;
 909
 910         if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
 911                 return 1;
 912
 913         info = fs_get_info(fs_type);
 914
 915         if (argc == 4)
 916                 env_set(argv[3], info->name);
 917         else
 918                 printf("%s\n", info->name);
 919
 920         fs_close();
 921
 922         return CMD_RET_SUCCESS;
 923 }
 924
 925 int do_rm(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 926           int fstype)
 927 {
 928         if (argc != 4)
 929                 return CMD_RET_USAGE;
 930
 931         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 932                 return 1;
 933
 934         if (fs_unlink(argv[3]))
 935                 return 1;
 936
 937         return 0;
 938 }
 939
 940 int do_mkdir(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 941              int fstype)
 942 {
 943         int ret;
 944
 945         if (argc != 4)
 946                 return CMD_RET_USAGE;
 947
 948         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 949                 return 1;
 950
 951         ret = fs_mkdir(argv[3]);
 952         if (ret) {
 953                 log_err("** Unable to create a directory \"%s\" **\n", argv[3]);
 954                 return 1;
 955         }
 956
 957         return 0;
 958 }
 959
 960 int do_ln(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 961           int fstype)
 962 {
 963         if (argc != 5)
 964                 return CMD_RET_USAGE;
 965
 966         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 967                 return 1;
 968
 969         if (fs_ln(argv[3], argv[4]))
 970                 return 1;
 971
 972         return 0;
 973 }
 974
 975 int do_fs_types(struct cmd_tbl *cmdtp, int flag, int argc, char * const argv[])
 976 {
 977         struct fstype_info *drv = fstypes;
 978         const int n_ents = ARRAY_SIZE(fstypes);
 979         struct fstype_info *entry;
 980         int i = 0;
 981
 982         puts("Supported filesystems");
 983         for (entry = drv; entry != drv + n_ents; entry++) {
 984                 if (entry->fstype != FS_TYPE_ANY) {
 985                         printf("%c %s", i ? ',' : ':', entry->name);
 986                         i++;
 987                 }
 988         }
 989         if (!i)
 990                 puts(": <none>");
 991         puts("\n");
 992         return CMD_RET_SUCCESS;
 993 }
 994
 995 int fs_read_alloc(const char *fname, ulong size, uint align, void **bufp)
 996 {
 997         loff_t bytes_read;
 998         ulong addr;
 999         char *buf;
1000         int ret;
1001
1002         buf = memalign(align, size + 1);
1003         if (!buf)
1004                 return log_msg_ret("buf", -ENOMEM);
1005         addr = map_to_sysmem(buf);
1006
1007         ret = fs_read(fname, addr, 0, size, &bytes_read);
1008         if (ret) {
1009                 free(buf);
1010                 return log_msg_ret("read", ret);
1011         }
1012         if (size != bytes_read)
1013                 return log_msg_ret("bread", -EIO);
1014         buf[size] = '\0';
1015
1016         *bufp = buf;
1017
1018         return 0;
1019 }
1020
1021 int fs_load_alloc(const char *ifname, const char *dev_part_str,
1022                   const char *fname, ulong max_size, ulong align, void **bufp,
1023                   ulong *sizep)
1024 {
1025         loff_t size;
1026         void *buf;
1027         int ret;
1028
1029         if (fs_set_blk_dev(ifname, dev_part_str, FS_TYPE_ANY))
1030                 return log_msg_ret("set", -ENOMEDIUM);
1031
1032         ret = fs_size(fname, &size);
1033         if (ret)
1034                 return log_msg_ret("sz", -ENOENT);
1035
1036         if (size >= (max_size ?: SZ_1G))
1037                 return log_msg_ret("sz", -E2BIG);
1038
1039         if (fs_set_blk_dev(ifname, dev_part_str, FS_TYPE_ANY))
1040                 return log_msg_ret("set", -ENOMEDIUM);
1041
1042         ret = fs_read_alloc(fname, size, align, &buf);
1043         if (ret)
1044                 return log_msg_ret("al", ret);
1045         *sizep = size;
1046         *bufp = buf;
1047
1048         return 0;
1049 }