1 // SPDX-License-Identifier: GPL-2.0
3 * Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved.
15 #include <sandboxfs.h>
16 #include <ubifs_uboot.h>
20 #include <linux/math64.h>
21 #include <efi_loader.h>
23 DECLARE_GLOBAL_DATA_PTR;
25 static struct blk_desc *fs_dev_desc;
26 static int fs_dev_part;
27 static disk_partition_t fs_partition;
28 static int fs_type = FS_TYPE_ANY;
30 static inline int fs_probe_unsupported(struct blk_desc *fs_dev_desc,
31 disk_partition_t *fs_partition)
33 printf("** Unrecognized filesystem type **\n");
37 static inline int fs_ls_unsupported(const char *dirname)
42 /* generic implementation of ls in terms of opendir/readdir/closedir */
44 static int fs_ls_generic(const char *dirname)
46 struct fs_dir_stream *dirs;
47 struct fs_dirent *dent;
48 int nfiles = 0, ndirs = 0;
50 dirs = fs_opendir(dirname);
54 while ((dent = fs_readdir(dirs))) {
55 if (dent->type == FS_DT_DIR) {
56 printf(" %s/\n", dent->name);
59 printf(" %8lld %s\n", dent->size, dent->name);
66 printf("\n%d file(s), %d dir(s)\n\n", nfiles, ndirs);
71 static inline int fs_exists_unsupported(const char *filename)
76 static inline int fs_size_unsupported(const char *filename, loff_t *size)
81 static inline int fs_read_unsupported(const char *filename, void *buf,
82 loff_t offset, loff_t len,
88 static inline int fs_write_unsupported(const char *filename, void *buf,
89 loff_t offset, loff_t len,
95 static inline int fs_ln_unsupported(const char *filename, const char *target)
100 static inline void fs_close_unsupported(void)
104 static inline int fs_uuid_unsupported(char *uuid_str)
109 static inline int fs_opendir_unsupported(const char *filename,
110 struct fs_dir_stream **dirs)
115 static inline int fs_unlink_unsupported(const char *filename)
120 static inline int fs_mkdir_unsupported(const char *dirname)
129 * Is it legal to pass NULL as .probe()'s fs_dev_desc parameter? This
130 * should be false in most cases. For "virtual" filesystems which
131 * aren't based on a U-Boot block device (e.g. sandbox), this can be
132 * set to true. This should also be true for the dummy entry at the end
133 * of fstypes[], since that is essentially a "virtual" (non-existent)
136 bool null_dev_desc_ok;
137 int (*probe)(struct blk_desc *fs_dev_desc,
138 disk_partition_t *fs_partition);
139 int (*ls)(const char *dirname);
140 int (*exists)(const char *filename);
141 int (*size)(const char *filename, loff_t *size);
142 int (*read)(const char *filename, void *buf, loff_t offset,
143 loff_t len, loff_t *actread);
144 int (*write)(const char *filename, void *buf, loff_t offset,
145 loff_t len, loff_t *actwrite);
147 int (*uuid)(char *uuid_str);
149 * Open a directory stream. On success return 0 and directory
150 * stream pointer via 'dirsp'. On error, return -errno. See
153 int (*opendir)(const char *filename, struct fs_dir_stream **dirsp);
155 * Read next entry from directory stream. On success return 0
156 * and directory entry pointer via 'dentp'. On error return
157 * -errno. See fs_readdir().
159 int (*readdir)(struct fs_dir_stream *dirs, struct fs_dirent **dentp);
160 /* see fs_closedir() */
161 void (*closedir)(struct fs_dir_stream *dirs);
162 int (*unlink)(const char *filename);
163 int (*mkdir)(const char *dirname);
164 int (*ln)(const char *filename, const char *target);
167 static struct fstype_info fstypes[] = {
170 .fstype = FS_TYPE_FAT,
172 .null_dev_desc_ok = false,
173 .probe = fat_set_blk_dev,
176 .exists = fat_exists,
178 .read = fat_read_file,
179 #if CONFIG_IS_ENABLED(FAT_WRITE)
180 .write = file_fat_write,
181 .unlink = fat_unlink,
184 .write = fs_write_unsupported,
185 .unlink = fs_unlink_unsupported,
186 .mkdir = fs_mkdir_unsupported,
188 .uuid = fs_uuid_unsupported,
189 .opendir = fat_opendir,
190 .readdir = fat_readdir,
191 .closedir = fat_closedir,
192 .ln = fs_ln_unsupported,
196 #if CONFIG_IS_ENABLED(FS_EXT4)
198 .fstype = FS_TYPE_EXT,
200 .null_dev_desc_ok = false,
201 .probe = ext4fs_probe,
202 .close = ext4fs_close,
204 .exists = ext4fs_exists,
206 .read = ext4_read_file,
207 #ifdef CONFIG_CMD_EXT4_WRITE
208 .write = ext4_write_file,
209 .ln = ext4fs_create_link,
211 .write = fs_write_unsupported,
212 .ln = fs_ln_unsupported,
215 .opendir = fs_opendir_unsupported,
216 .unlink = fs_unlink_unsupported,
217 .mkdir = fs_mkdir_unsupported,
220 #ifdef CONFIG_SANDBOX
222 .fstype = FS_TYPE_SANDBOX,
224 .null_dev_desc_ok = true,
225 .probe = sandbox_fs_set_blk_dev,
226 .close = sandbox_fs_close,
228 .exists = sandbox_fs_exists,
229 .size = sandbox_fs_size,
230 .read = fs_read_sandbox,
231 .write = fs_write_sandbox,
232 .uuid = fs_uuid_unsupported,
233 .opendir = fs_opendir_unsupported,
234 .unlink = fs_unlink_unsupported,
235 .mkdir = fs_mkdir_unsupported,
236 .ln = fs_ln_unsupported,
239 #ifdef CONFIG_CMD_UBIFS
241 .fstype = FS_TYPE_UBIFS,
243 .null_dev_desc_ok = true,
244 .probe = ubifs_set_blk_dev,
245 .close = ubifs_close,
247 .exists = ubifs_exists,
250 .write = fs_write_unsupported,
251 .uuid = fs_uuid_unsupported,
252 .opendir = fs_opendir_unsupported,
253 .unlink = fs_unlink_unsupported,
254 .mkdir = fs_mkdir_unsupported,
255 .ln = fs_ln_unsupported,
258 #ifdef CONFIG_FS_BTRFS
260 .fstype = FS_TYPE_BTRFS,
262 .null_dev_desc_ok = false,
263 .probe = btrfs_probe,
264 .close = btrfs_close,
266 .exists = btrfs_exists,
269 .write = fs_write_unsupported,
271 .opendir = fs_opendir_unsupported,
272 .unlink = fs_unlink_unsupported,
273 .mkdir = fs_mkdir_unsupported,
274 .ln = fs_ln_unsupported,
278 .fstype = FS_TYPE_ANY,
279 .name = "unsupported",
280 .null_dev_desc_ok = true,
281 .probe = fs_probe_unsupported,
282 .close = fs_close_unsupported,
283 .ls = fs_ls_unsupported,
284 .exists = fs_exists_unsupported,
285 .size = fs_size_unsupported,
286 .read = fs_read_unsupported,
287 .write = fs_write_unsupported,
288 .uuid = fs_uuid_unsupported,
289 .opendir = fs_opendir_unsupported,
290 .unlink = fs_unlink_unsupported,
291 .mkdir = fs_mkdir_unsupported,
292 .ln = fs_ln_unsupported,
296 static struct fstype_info *fs_get_info(int fstype)
298 struct fstype_info *info;
301 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
302 if (fstype == info->fstype)
306 /* Return the 'unsupported' sentinel */
311 * fs_get_type_name() - Get type of current filesystem
313 * Return: Pointer to filesystem name
315 * Returns a string describing the current filesystem, or the sentinel
316 * "unsupported" for any unrecognised filesystem.
318 const char *fs_get_type_name(void)
320 return fs_get_info(fs_type)->name;
323 int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
325 struct fstype_info *info;
327 #ifdef CONFIG_NEEDS_MANUAL_RELOC
328 static int relocated;
331 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
333 info->name += gd->reloc_off;
334 info->probe += gd->reloc_off;
335 info->close += gd->reloc_off;
336 info->ls += gd->reloc_off;
337 info->read += gd->reloc_off;
338 info->write += gd->reloc_off;
344 part = blk_get_device_part_str(ifname, dev_part_str, &fs_dev_desc,
349 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
350 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
351 fstype != info->fstype)
354 if (!fs_dev_desc && !info->null_dev_desc_ok)
357 if (!info->probe(fs_dev_desc, &fs_partition)) {
358 fs_type = info->fstype;
367 /* set current blk device w/ blk_desc + partition # */
368 int fs_set_blk_dev_with_part(struct blk_desc *desc, int part)
370 struct fstype_info *info;
374 ret = part_get_info(desc, part, &fs_partition);
376 ret = part_get_info_whole_disk(desc, &fs_partition);
381 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
382 if (!info->probe(fs_dev_desc, &fs_partition)) {
383 fs_type = info->fstype;
392 static void fs_close(void)
394 struct fstype_info *info = fs_get_info(fs_type);
398 fs_type = FS_TYPE_ANY;
401 int fs_uuid(char *uuid_str)
403 struct fstype_info *info = fs_get_info(fs_type);
405 return info->uuid(uuid_str);
408 int fs_ls(const char *dirname)
412 struct fstype_info *info = fs_get_info(fs_type);
414 ret = info->ls(dirname);
416 fs_type = FS_TYPE_ANY;
422 int fs_exists(const char *filename)
426 struct fstype_info *info = fs_get_info(fs_type);
428 ret = info->exists(filename);
435 int fs_size(const char *filename, loff_t *size)
439 struct fstype_info *info = fs_get_info(fs_type);
441 ret = info->size(filename, size);
449 /* Check if a file may be read to the given address */
450 static int fs_read_lmb_check(const char *filename, ulong addr, loff_t offset,
451 loff_t len, struct fstype_info *info)
458 /* get the actual size of the file */
459 ret = info->size(filename, &size);
462 if (offset >= size) {
463 /* offset >= EOF, no bytes will be written */
466 read_len = size - offset;
468 /* limit to 'len' if it is smaller */
469 if (len && len < read_len)
472 lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);
475 if (lmb_alloc_addr(&lmb, addr, read_len) == addr)
478 printf("** Reading file would overwrite reserved memory **\n");
483 static int _fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
484 int do_lmb_check, loff_t *actread)
486 struct fstype_info *info = fs_get_info(fs_type);
492 ret = fs_read_lmb_check(filename, addr, offset, len, info);
499 * We don't actually know how many bytes are being read, since len==0
500 * means read the whole file.
502 buf = map_sysmem(addr, len);
503 ret = info->read(filename, buf, offset, len, actread);
506 /* If we requested a specific number of bytes, check we got it */
507 if (ret == 0 && len && *actread != len)
508 debug("** %s shorter than offset + len **\n", filename);
514 int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
517 return _fs_read(filename, addr, offset, len, 0, actread);
520 int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
523 struct fstype_info *info = fs_get_info(fs_type);
527 buf = map_sysmem(addr, len);
528 ret = info->write(filename, buf, offset, len, actwrite);
531 if (ret < 0 && len != *actwrite) {
532 printf("** Unable to write file %s **\n", filename);
540 struct fs_dir_stream *fs_opendir(const char *filename)
542 struct fstype_info *info = fs_get_info(fs_type);
543 struct fs_dir_stream *dirs = NULL;
546 ret = info->opendir(filename, &dirs);
553 dirs->desc = fs_dev_desc;
554 dirs->part = fs_dev_part;
559 struct fs_dirent *fs_readdir(struct fs_dir_stream *dirs)
561 struct fstype_info *info;
562 struct fs_dirent *dirent;
565 fs_set_blk_dev_with_part(dirs->desc, dirs->part);
566 info = fs_get_info(fs_type);
568 ret = info->readdir(dirs, &dirent);
578 void fs_closedir(struct fs_dir_stream *dirs)
580 struct fstype_info *info;
585 fs_set_blk_dev_with_part(dirs->desc, dirs->part);
586 info = fs_get_info(fs_type);
588 info->closedir(dirs);
592 int fs_unlink(const char *filename)
596 struct fstype_info *info = fs_get_info(fs_type);
598 ret = info->unlink(filename);
600 fs_type = FS_TYPE_ANY;
606 int fs_mkdir(const char *dirname)
610 struct fstype_info *info = fs_get_info(fs_type);
612 ret = info->mkdir(dirname);
614 fs_type = FS_TYPE_ANY;
620 int fs_ln(const char *fname, const char *target)
622 struct fstype_info *info = fs_get_info(fs_type);
625 ret = info->ln(fname, target);
628 printf("** Unable to create link %s -> %s **\n", fname, target);
636 int do_size(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
642 return CMD_RET_USAGE;
644 if (fs_set_blk_dev(argv[1], argv[2], fstype))
647 if (fs_size(argv[3], &size) < 0)
648 return CMD_RET_FAILURE;
650 env_set_hex("filesize", size);
655 int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
659 const char *addr_str;
660 const char *filename;
669 return CMD_RET_USAGE;
671 return CMD_RET_USAGE;
673 if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
677 addr = simple_strtoul(argv[3], &ep, 16);
678 if (ep == argv[3] || *ep != '\0')
679 return CMD_RET_USAGE;
681 addr_str = env_get("loadaddr");
682 if (addr_str != NULL)
683 addr = simple_strtoul(addr_str, NULL, 16);
685 addr = CONFIG_SYS_LOAD_ADDR;
690 filename = env_get("bootfile");
692 puts("** No boot file defined **\n");
697 bytes = simple_strtoul(argv[5], NULL, 16);
701 pos = simple_strtoul(argv[6], NULL, 16);
705 #ifdef CONFIG_CMD_BOOTEFI
706 efi_set_bootdev(argv[1], (argc > 2) ? argv[2] : "",
707 (argc > 4) ? argv[4] : "");
710 ret = _fs_read(filename, addr, pos, bytes, 1, &len_read);
711 time = get_timer(time);
715 printf("%llu bytes read in %lu ms", len_read, time);
718 print_size(div_u64(len_read, time) * 1000, "/s");
723 env_set_hex("fileaddr", addr);
724 env_set_hex("filesize", len_read);
729 int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
733 return CMD_RET_USAGE;
735 return CMD_RET_USAGE;
737 if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
740 if (fs_ls(argc >= 4 ? argv[3] : "/"))
746 int file_exists(const char *dev_type, const char *dev_part, const char *file,
749 if (fs_set_blk_dev(dev_type, dev_part, fstype))
752 return fs_exists(file);
755 int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
759 const char *filename;
766 if (argc < 6 || argc > 7)
767 return CMD_RET_USAGE;
769 if (fs_set_blk_dev(argv[1], argv[2], fstype))
772 addr = simple_strtoul(argv[3], NULL, 16);
774 bytes = simple_strtoul(argv[5], NULL, 16);
776 pos = simple_strtoul(argv[6], NULL, 16);
781 ret = fs_write(filename, addr, pos, bytes, &len);
782 time = get_timer(time);
786 printf("%llu bytes written in %lu ms", len, time);
789 print_size(div_u64(len, time) * 1000, "/s");
797 int do_fs_uuid(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
802 memset(uuid, 0, sizeof(uuid));
804 if (argc < 3 || argc > 4)
805 return CMD_RET_USAGE;
807 if (fs_set_blk_dev(argv[1], argv[2], fstype))
812 return CMD_RET_FAILURE;
815 env_set(argv[3], uuid);
817 printf("%s\n", uuid);
819 return CMD_RET_SUCCESS;
822 int do_fs_type(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
824 struct fstype_info *info;
826 if (argc < 3 || argc > 4)
827 return CMD_RET_USAGE;
829 if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
832 info = fs_get_info(fs_type);
835 env_set(argv[3], info->name);
837 printf("%s\n", info->name);
841 return CMD_RET_SUCCESS;
844 int do_rm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
848 return CMD_RET_USAGE;
850 if (fs_set_blk_dev(argv[1], argv[2], fstype))
853 if (fs_unlink(argv[3]))
859 int do_mkdir(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
865 return CMD_RET_USAGE;
867 if (fs_set_blk_dev(argv[1], argv[2], fstype))
870 ret = fs_mkdir(argv[3]);
872 printf("** Unable to create a directory \"%s\" **\n", argv[3]);
879 int do_ln(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
883 return CMD_RET_USAGE;
885 if (fs_set_blk_dev(argv[1], argv[2], fstype))
888 if (fs_ln(argv[3], argv[4]))