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 <common.h>
  13 #include <env.h>
  14 #include <lmb.h>
  15 #include <log.h>
  16 #include <malloc.h>
  17 #include <mapmem.h>
  18 #include <part.h>
  19 #include <ext4fs.h>
  20 #include <fat.h>
  21 #include <fs.h>
  22 #include <sandboxfs.h>
  23 #include <semihostingfs.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_CMD_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 #ifdef CONFIG_SANDBOX
 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 #ifdef CONFIG_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_SPL_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_SPL_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 #ifdef CONFIG_NEEDS_MANUAL_RELOC
 426         static int relocated;
 427
 428         if (!relocated) {
 429                 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
 430                                 i++, info++) {
 431                         info->name += gd->reloc_off;
 432                         info->probe += gd->reloc_off;
 433                         info->close += gd->reloc_off;
 434                         info->ls += gd->reloc_off;
 435                         info->read += gd->reloc_off;
 436                         info->write += gd->reloc_off;
 437                 }
 438                 relocated = 1;
 439         }
 440 #endif
 441
 442         part = part_get_info_by_dev_and_name_or_num(ifname, dev_part_str, &fs_dev_desc,
 443                                                     &fs_partition, 1);
 444         if (part < 0)
 445                 return -1;
 446
 447         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 448                 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
 449                                 fstype != info->fstype)
 450                         continue;
 451
 452                 if (!fs_dev_desc && !info->null_dev_desc_ok)
 453                         continue;
 454
 455                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 456                         fs_type = info->fstype;
 457                         fs_dev_part = part;
 458                         return 0;
 459                 }
 460         }
 461
 462         return -1;
 463 }
 464
 465 /* set current blk device w/ blk_desc + partition # */
 466 int fs_set_blk_dev_with_part(struct blk_desc *desc, int part)
 467 {
 468         struct fstype_info *info;
 469         int ret, i;
 470
 471         if (part >= 1)
 472                 ret = part_get_info(desc, part, &fs_partition);
 473         else
 474                 ret = part_get_info_whole_disk(desc, &fs_partition);
 475         if (ret)
 476                 return ret;
 477         fs_dev_desc = desc;
 478
 479         for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
 480                 if (!info->probe(fs_dev_desc, &fs_partition)) {
 481                         fs_type = info->fstype;
 482                         fs_dev_part = part;
 483                         return 0;
 484                 }
 485         }
 486
 487         return -1;
 488 }
 489
 490 void fs_close(void)
 491 {
 492         struct fstype_info *info = fs_get_info(fs_type);
 493
 494         info->close();
 495
 496         fs_type = FS_TYPE_ANY;
 497 }
 498
 499 int fs_uuid(char *uuid_str)
 500 {
 501         struct fstype_info *info = fs_get_info(fs_type);
 502
 503         return info->uuid(uuid_str);
 504 }
 505
 506 int fs_ls(const char *dirname)
 507 {
 508         int ret;
 509
 510         struct fstype_info *info = fs_get_info(fs_type);
 511
 512         ret = info->ls(dirname);
 513
 514         fs_close();
 515
 516         return ret;
 517 }
 518
 519 int fs_exists(const char *filename)
 520 {
 521         int ret;
 522
 523         struct fstype_info *info = fs_get_info(fs_type);
 524
 525         ret = info->exists(filename);
 526
 527         fs_close();
 528
 529         return ret;
 530 }
 531
 532 int fs_size(const char *filename, loff_t *size)
 533 {
 534         int ret;
 535
 536         struct fstype_info *info = fs_get_info(fs_type);
 537
 538         ret = info->size(filename, size);
 539
 540         fs_close();
 541
 542         return ret;
 543 }
 544
 545 #ifdef CONFIG_LMB
 546 /* Check if a file may be read to the given address */
 547 static int fs_read_lmb_check(const char *filename, ulong addr, loff_t offset,
 548                              loff_t len, struct fstype_info *info)
 549 {
 550         struct lmb lmb;
 551         int ret;
 552         loff_t size;
 553         loff_t read_len;
 554
 555         /* get the actual size of the file */
 556         ret = info->size(filename, &size);
 557         if (ret)
 558                 return ret;
 559         if (offset >= size) {
 560                 /* offset >= EOF, no bytes will be written */
 561                 return 0;
 562         }
 563         read_len = size - offset;
 564
 565         /* limit to 'len' if it is smaller */
 566         if (len && len < read_len)
 567                 read_len = len;
 568
 569         lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);
 570         lmb_dump_all(&lmb);
 571
 572         if (lmb_alloc_addr(&lmb, addr, read_len) == addr)
 573                 return 0;
 574
 575         log_err("** Reading file would overwrite reserved memory **\n");
 576         return -ENOSPC;
 577 }
 578 #endif
 579
 580 static int _fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
 581                     int do_lmb_check, loff_t *actread)
 582 {
 583         struct fstype_info *info = fs_get_info(fs_type);
 584         void *buf;
 585         int ret;
 586
 587 #ifdef CONFIG_LMB
 588         if (do_lmb_check) {
 589                 ret = fs_read_lmb_check(filename, addr, offset, len, info);
 590                 if (ret)
 591                         return ret;
 592         }
 593 #endif
 594
 595         /*
 596          * We don't actually know how many bytes are being read, since len==0
 597          * means read the whole file.
 598          */
 599         buf = map_sysmem(addr, len);
 600         ret = info->read(filename, buf, offset, len, actread);
 601         unmap_sysmem(buf);
 602
 603         /* If we requested a specific number of bytes, check we got it */
 604         if (ret == 0 && len && *actread != len)
 605                 log_debug("** %s shorter than offset + len **\n", filename);
 606         fs_close();
 607
 608         return ret;
 609 }
 610
 611 int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
 612             loff_t *actread)
 613 {
 614         return _fs_read(filename, addr, offset, len, 0, actread);
 615 }
 616
 617 int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
 618              loff_t *actwrite)
 619 {
 620         struct fstype_info *info = fs_get_info(fs_type);
 621         void *buf;
 622         int ret;
 623
 624         buf = map_sysmem(addr, len);
 625         ret = info->write(filename, buf, offset, len, actwrite);
 626         unmap_sysmem(buf);
 627
 628         if (ret < 0 && len != *actwrite) {
 629                 log_err("** Unable to write file %s **\n", filename);
 630                 ret = -1;
 631         }
 632         fs_close();
 633
 634         return ret;
 635 }
 636
 637 struct fs_dir_stream *fs_opendir(const char *filename)
 638 {
 639         struct fstype_info *info = fs_get_info(fs_type);
 640         struct fs_dir_stream *dirs = NULL;
 641         int ret;
 642
 643         ret = info->opendir(filename, &dirs);
 644         fs_close();
 645         if (ret) {
 646                 errno = -ret;
 647                 return NULL;
 648         }
 649
 650         dirs->desc = fs_dev_desc;
 651         dirs->part = fs_dev_part;
 652
 653         return dirs;
 654 }
 655
 656 struct fs_dirent *fs_readdir(struct fs_dir_stream *dirs)
 657 {
 658         struct fstype_info *info;
 659         struct fs_dirent *dirent;
 660         int ret;
 661
 662         fs_set_blk_dev_with_part(dirs->desc, dirs->part);
 663         info = fs_get_info(fs_type);
 664
 665         ret = info->readdir(dirs, &dirent);
 666         fs_close();
 667         if (ret) {
 668                 errno = -ret;
 669                 return NULL;
 670         }
 671
 672         return dirent;
 673 }
 674
 675 void fs_closedir(struct fs_dir_stream *dirs)
 676 {
 677         struct fstype_info *info;
 678
 679         if (!dirs)
 680                 return;
 681
 682         fs_set_blk_dev_with_part(dirs->desc, dirs->part);
 683         info = fs_get_info(fs_type);
 684
 685         info->closedir(dirs);
 686         fs_close();
 687 }
 688
 689 int fs_unlink(const char *filename)
 690 {
 691         int ret;
 692
 693         struct fstype_info *info = fs_get_info(fs_type);
 694
 695         ret = info->unlink(filename);
 696
 697         fs_close();
 698
 699         return ret;
 700 }
 701
 702 int fs_mkdir(const char *dirname)
 703 {
 704         int ret;
 705
 706         struct fstype_info *info = fs_get_info(fs_type);
 707
 708         ret = info->mkdir(dirname);
 709
 710         fs_close();
 711
 712         return ret;
 713 }
 714
 715 int fs_ln(const char *fname, const char *target)
 716 {
 717         struct fstype_info *info = fs_get_info(fs_type);
 718         int ret;
 719
 720         ret = info->ln(fname, target);
 721
 722         if (ret < 0) {
 723                 log_err("** Unable to create link %s -> %s **\n", fname, target);
 724                 ret = -1;
 725         }
 726         fs_close();
 727
 728         return ret;
 729 }
 730
 731 int do_size(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 732             int fstype)
 733 {
 734         loff_t size;
 735
 736         if (argc != 4)
 737                 return CMD_RET_USAGE;
 738
 739         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 740                 return 1;
 741
 742         if (fs_size(argv[3], &size) < 0)
 743                 return CMD_RET_FAILURE;
 744
 745         env_set_hex("filesize", size);
 746
 747         return 0;
 748 }
 749
 750 int do_load(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 751             int fstype)
 752 {
 753         unsigned long addr;
 754         const char *addr_str;
 755         const char *filename;
 756         loff_t bytes;
 757         loff_t pos;
 758         loff_t len_read;
 759         int ret;
 760         unsigned long time;
 761         char *ep;
 762
 763         if (argc < 2)
 764                 return CMD_RET_USAGE;
 765         if (argc > 7)
 766                 return CMD_RET_USAGE;
 767
 768         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype)) {
 769                 log_err("Can't set block device\n");
 770                 return 1;
 771         }
 772
 773         if (argc >= 4) {
 774                 addr = hextoul(argv[3], &ep);
 775                 if (ep == argv[3] || *ep != '\0')
 776                         return CMD_RET_USAGE;
 777         } else {
 778                 addr_str = env_get("loadaddr");
 779                 if (addr_str != NULL)
 780                         addr = hextoul(addr_str, NULL);
 781                 else
 782                         addr = CONFIG_SYS_LOAD_ADDR;
 783         }
 784         if (argc >= 5) {
 785                 filename = argv[4];
 786         } else {
 787                 filename = env_get("bootfile");
 788                 if (!filename) {
 789                         puts("** No boot file defined **\n");
 790                         return 1;
 791                 }
 792         }
 793         if (argc >= 6)
 794                 bytes = hextoul(argv[5], NULL);
 795         else
 796                 bytes = 0;
 797         if (argc >= 7)
 798                 pos = hextoul(argv[6], NULL);
 799         else
 800                 pos = 0;
 801
 802         time = get_timer(0);
 803         ret = _fs_read(filename, addr, pos, bytes, 1, &len_read);
 804         time = get_timer(time);
 805         if (ret < 0) {
 806                 log_err("Failed to load '%s'\n", filename);
 807                 return 1;
 808         }
 809
 810         if (IS_ENABLED(CONFIG_CMD_BOOTEFI))
 811                 efi_set_bootdev(argv[1], (argc > 2) ? argv[2] : "",
 812                                 (argc > 4) ? argv[4] : "", map_sysmem(addr, 0),
 813                                 len_read);
 814
 815         printf("%llu bytes read in %lu ms", len_read, time);
 816         if (time > 0) {
 817                 puts(" (");
 818                 print_size(div_u64(len_read, time) * 1000, "/s");
 819                 puts(")");
 820         }
 821         puts("\n");
 822
 823         env_set_hex("fileaddr", addr);
 824         env_set_hex("filesize", len_read);
 825
 826         return 0;
 827 }
 828
 829 int do_ls(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 830           int fstype)
 831 {
 832         if (argc < 2)
 833                 return CMD_RET_USAGE;
 834         if (argc > 4)
 835                 return CMD_RET_USAGE;
 836
 837         if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
 838                 return 1;
 839
 840         if (fs_ls(argc >= 4 ? argv[3] : "/"))
 841                 return 1;
 842
 843         return 0;
 844 }
 845
 846 int file_exists(const char *dev_type, const char *dev_part, const char *file,
 847                 int fstype)
 848 {
 849         if (fs_set_blk_dev(dev_type, dev_part, fstype))
 850                 return 0;
 851
 852         return fs_exists(file);
 853 }
 854
 855 int do_save(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 856             int fstype)
 857 {
 858         unsigned long addr;
 859         const char *filename;
 860         loff_t bytes;
 861         loff_t pos;
 862         loff_t len;
 863         int ret;
 864         unsigned long time;
 865
 866         if (argc < 6 || argc > 7)
 867                 return CMD_RET_USAGE;
 868
 869         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 870                 return 1;
 871
 872         addr = hextoul(argv[3], NULL);
 873         filename = argv[4];
 874         bytes = hextoul(argv[5], NULL);
 875         if (argc >= 7)
 876                 pos = hextoul(argv[6], NULL);
 877         else
 878                 pos = 0;
 879
 880         time = get_timer(0);
 881         ret = fs_write(filename, addr, pos, bytes, &len);
 882         time = get_timer(time);
 883         if (ret < 0)
 884                 return 1;
 885
 886         printf("%llu bytes written in %lu ms", len, time);
 887         if (time > 0) {
 888                 puts(" (");
 889                 print_size(div_u64(len, time) * 1000, "/s");
 890                 puts(")");
 891         }
 892         puts("\n");
 893
 894         return 0;
 895 }
 896
 897 int do_fs_uuid(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 898                int fstype)
 899 {
 900         int ret;
 901         char uuid[37];
 902         memset(uuid, 0, sizeof(uuid));
 903
 904         if (argc < 3 || argc > 4)
 905                 return CMD_RET_USAGE;
 906
 907         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 908                 return 1;
 909
 910         ret = fs_uuid(uuid);
 911         if (ret)
 912                 return CMD_RET_FAILURE;
 913
 914         if (argc == 4)
 915                 env_set(argv[3], uuid);
 916         else
 917                 printf("%s\n", uuid);
 918
 919         return CMD_RET_SUCCESS;
 920 }
 921
 922 int do_fs_type(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 923 {
 924         struct fstype_info *info;
 925
 926         if (argc < 3 || argc > 4)
 927                 return CMD_RET_USAGE;
 928
 929         if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
 930                 return 1;
 931
 932         info = fs_get_info(fs_type);
 933
 934         if (argc == 4)
 935                 env_set(argv[3], info->name);
 936         else
 937                 printf("%s\n", info->name);
 938
 939         fs_close();
 940
 941         return CMD_RET_SUCCESS;
 942 }
 943
 944 int do_rm(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 945           int fstype)
 946 {
 947         if (argc != 4)
 948                 return CMD_RET_USAGE;
 949
 950         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 951                 return 1;
 952
 953         if (fs_unlink(argv[3]))
 954                 return 1;
 955
 956         return 0;
 957 }
 958
 959 int do_mkdir(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 960              int fstype)
 961 {
 962         int ret;
 963
 964         if (argc != 4)
 965                 return CMD_RET_USAGE;
 966
 967         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 968                 return 1;
 969
 970         ret = fs_mkdir(argv[3]);
 971         if (ret) {
 972                 log_err("** Unable to create a directory \"%s\" **\n", argv[3]);
 973                 return 1;
 974         }
 975
 976         return 0;
 977 }
 978
 979 int do_ln(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[],
 980           int fstype)
 981 {
 982         if (argc != 5)
 983                 return CMD_RET_USAGE;
 984
 985         if (fs_set_blk_dev(argv[1], argv[2], fstype))
 986                 return 1;
 987
 988         if (fs_ln(argv[3], argv[4]))
 989                 return 1;
 990
 991         return 0;
 992 }
 993
 994 int do_fs_types(struct cmd_tbl *cmdtp, int flag, int argc, char * const argv[])
 995 {
 996         struct fstype_info *drv = fstypes;
 997         const int n_ents = ARRAY_SIZE(fstypes);
 998         struct fstype_info *entry;
 999         int i = 0;
1000
1001         puts("Supported filesystems");
1002         for (entry = drv; entry != drv + n_ents; entry++) {
1003                 if (entry->fstype != FS_TYPE_ANY) {
1004                         printf("%c %s", i ? ',' : ':', entry->name);
1005                         i++;
1006                 }
1007         }
1008         if (!i)
1009                 puts(": <none>");
1010         puts("\n");
1011         return CMD_RET_SUCCESS;
1012 }
1013
1014 int fs_read_alloc(const char *fname, ulong size, uint align, void **bufp)
1015 {
1016         loff_t bytes_read;
1017         ulong addr;
1018         char *buf;
1019         int ret;
1020
1021         buf = memalign(align, size + 1);
1022         if (!buf)
1023                 return log_msg_ret("buf", -ENOMEM);
1024         addr = map_to_sysmem(buf);
1025
1026         ret = fs_read(fname, addr, 0, size, &bytes_read);
1027         if (ret) {
1028                 free(buf);
1029                 return log_msg_ret("read", ret);
1030         }
1031         if (size != bytes_read)
1032                 return log_msg_ret("bread", -EIO);
1033         buf[size] = '\0';
1034
1035         *bufp = buf;
1036
1037         return 0;
1038 }
1039
1040 int fs_load_alloc(const char *ifname, const char *dev_part_str,
1041                   const char *fname, ulong max_size, ulong align, void **bufp,
1042                   ulong *sizep)
1043 {
1044         loff_t size;
1045         void *buf;
1046         int ret;
1047
1048         if (fs_set_blk_dev(ifname, dev_part_str, FS_TYPE_ANY))
1049                 return log_msg_ret("set", -ENOMEDIUM);
1050
1051         ret = fs_size(fname, &size);
1052         if (ret)
1053                 return log_msg_ret("sz", -ENOENT);
1054
1055         if (size >= (max_size ?: SZ_1G))
1056                 return log_msg_ret("sz", -E2BIG);
1057
1058         if (fs_set_blk_dev(ifname, dev_part_str, FS_TYPE_ANY))
1059                 return log_msg_ret("set", -ENOMEDIUM);
1060
1061         ret = fs_read_alloc(fname, size, align, &buf);
1062         if (ret)
1063                 return log_msg_ret("al", ret);
1064         *sizep = size;
1065         *bufp = buf;
1066
1067         return 0;
1068 }