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() - Get type of current filesystem
313 * Return: filesystem type
315 * Returns filesystem type representing the current filesystem, or
316 * FS_TYPE_ANY for any unrecognised filesystem.
318 int fs_get_type(void)
324 * fs_get_type_name() - Get type of current filesystem
326 * Return: Pointer to filesystem name
328 * Returns a string describing the current filesystem, or the sentinel
329 * "unsupported" for any unrecognised filesystem.
331 const char *fs_get_type_name(void)
333 return fs_get_info(fs_type)->name;
336 int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
338 struct fstype_info *info;
340 #ifdef CONFIG_NEEDS_MANUAL_RELOC
341 static int relocated;
344 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
346 info->name += gd->reloc_off;
347 info->probe += gd->reloc_off;
348 info->close += gd->reloc_off;
349 info->ls += gd->reloc_off;
350 info->read += gd->reloc_off;
351 info->write += gd->reloc_off;
357 part = blk_get_device_part_str(ifname, dev_part_str, &fs_dev_desc,
362 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
363 if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
364 fstype != info->fstype)
367 if (!fs_dev_desc && !info->null_dev_desc_ok)
370 if (!info->probe(fs_dev_desc, &fs_partition)) {
371 fs_type = info->fstype;
380 /* set current blk device w/ blk_desc + partition # */
381 int fs_set_blk_dev_with_part(struct blk_desc *desc, int part)
383 struct fstype_info *info;
387 ret = part_get_info(desc, part, &fs_partition);
389 ret = part_get_info_whole_disk(desc, &fs_partition);
394 for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
395 if (!info->probe(fs_dev_desc, &fs_partition)) {
396 fs_type = info->fstype;
407 struct fstype_info *info = fs_get_info(fs_type);
411 fs_type = FS_TYPE_ANY;
414 int fs_uuid(char *uuid_str)
416 struct fstype_info *info = fs_get_info(fs_type);
418 return info->uuid(uuid_str);
421 int fs_ls(const char *dirname)
425 struct fstype_info *info = fs_get_info(fs_type);
427 ret = info->ls(dirname);
434 int fs_exists(const char *filename)
438 struct fstype_info *info = fs_get_info(fs_type);
440 ret = info->exists(filename);
447 int fs_size(const char *filename, loff_t *size)
451 struct fstype_info *info = fs_get_info(fs_type);
453 ret = info->size(filename, size);
461 /* Check if a file may be read to the given address */
462 static int fs_read_lmb_check(const char *filename, ulong addr, loff_t offset,
463 loff_t len, struct fstype_info *info)
470 /* get the actual size of the file */
471 ret = info->size(filename, &size);
474 if (offset >= size) {
475 /* offset >= EOF, no bytes will be written */
478 read_len = size - offset;
480 /* limit to 'len' if it is smaller */
481 if (len && len < read_len)
484 lmb_init_and_reserve(&lmb, gd->bd, (void *)gd->fdt_blob);
487 if (lmb_alloc_addr(&lmb, addr, read_len) == addr)
490 printf("** Reading file would overwrite reserved memory **\n");
495 static int _fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
496 int do_lmb_check, loff_t *actread)
498 struct fstype_info *info = fs_get_info(fs_type);
504 ret = fs_read_lmb_check(filename, addr, offset, len, info);
511 * We don't actually know how many bytes are being read, since len==0
512 * means read the whole file.
514 buf = map_sysmem(addr, len);
515 ret = info->read(filename, buf, offset, len, actread);
518 /* If we requested a specific number of bytes, check we got it */
519 if (ret == 0 && len && *actread != len)
520 debug("** %s shorter than offset + len **\n", filename);
526 int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
529 return _fs_read(filename, addr, offset, len, 0, actread);
532 int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
535 struct fstype_info *info = fs_get_info(fs_type);
539 buf = map_sysmem(addr, len);
540 ret = info->write(filename, buf, offset, len, actwrite);
543 if (ret < 0 && len != *actwrite) {
544 printf("** Unable to write file %s **\n", filename);
552 struct fs_dir_stream *fs_opendir(const char *filename)
554 struct fstype_info *info = fs_get_info(fs_type);
555 struct fs_dir_stream *dirs = NULL;
558 ret = info->opendir(filename, &dirs);
565 dirs->desc = fs_dev_desc;
566 dirs->part = fs_dev_part;
571 struct fs_dirent *fs_readdir(struct fs_dir_stream *dirs)
573 struct fstype_info *info;
574 struct fs_dirent *dirent;
577 fs_set_blk_dev_with_part(dirs->desc, dirs->part);
578 info = fs_get_info(fs_type);
580 ret = info->readdir(dirs, &dirent);
590 void fs_closedir(struct fs_dir_stream *dirs)
592 struct fstype_info *info;
597 fs_set_blk_dev_with_part(dirs->desc, dirs->part);
598 info = fs_get_info(fs_type);
600 info->closedir(dirs);
604 int fs_unlink(const char *filename)
608 struct fstype_info *info = fs_get_info(fs_type);
610 ret = info->unlink(filename);
617 int fs_mkdir(const char *dirname)
621 struct fstype_info *info = fs_get_info(fs_type);
623 ret = info->mkdir(dirname);
630 int fs_ln(const char *fname, const char *target)
632 struct fstype_info *info = fs_get_info(fs_type);
635 ret = info->ln(fname, target);
638 printf("** Unable to create link %s -> %s **\n", fname, target);
646 int do_size(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
652 return CMD_RET_USAGE;
654 if (fs_set_blk_dev(argv[1], argv[2], fstype))
657 if (fs_size(argv[3], &size) < 0)
658 return CMD_RET_FAILURE;
660 env_set_hex("filesize", size);
665 int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
669 const char *addr_str;
670 const char *filename;
679 return CMD_RET_USAGE;
681 return CMD_RET_USAGE;
683 if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
687 addr = simple_strtoul(argv[3], &ep, 16);
688 if (ep == argv[3] || *ep != '\0')
689 return CMD_RET_USAGE;
691 addr_str = env_get("loadaddr");
692 if (addr_str != NULL)
693 addr = simple_strtoul(addr_str, NULL, 16);
695 addr = CONFIG_SYS_LOAD_ADDR;
700 filename = env_get("bootfile");
702 puts("** No boot file defined **\n");
707 bytes = simple_strtoul(argv[5], NULL, 16);
711 pos = simple_strtoul(argv[6], NULL, 16);
715 #ifdef CONFIG_CMD_BOOTEFI
716 efi_set_bootdev(argv[1], (argc > 2) ? argv[2] : "",
717 (argc > 4) ? argv[4] : "");
720 ret = _fs_read(filename, addr, pos, bytes, 1, &len_read);
721 time = get_timer(time);
725 printf("%llu bytes read in %lu ms", len_read, time);
728 print_size(div_u64(len_read, time) * 1000, "/s");
733 env_set_hex("fileaddr", addr);
734 env_set_hex("filesize", len_read);
739 int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
743 return CMD_RET_USAGE;
745 return CMD_RET_USAGE;
747 if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
750 if (fs_ls(argc >= 4 ? argv[3] : "/"))
756 int file_exists(const char *dev_type, const char *dev_part, const char *file,
759 if (fs_set_blk_dev(dev_type, dev_part, fstype))
762 return fs_exists(file);
765 int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
769 const char *filename;
776 if (argc < 6 || argc > 7)
777 return CMD_RET_USAGE;
779 if (fs_set_blk_dev(argv[1], argv[2], fstype))
782 addr = simple_strtoul(argv[3], NULL, 16);
784 bytes = simple_strtoul(argv[5], NULL, 16);
786 pos = simple_strtoul(argv[6], NULL, 16);
791 ret = fs_write(filename, addr, pos, bytes, &len);
792 time = get_timer(time);
796 printf("%llu bytes written in %lu ms", len, time);
799 print_size(div_u64(len, time) * 1000, "/s");
807 int do_fs_uuid(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
812 memset(uuid, 0, sizeof(uuid));
814 if (argc < 3 || argc > 4)
815 return CMD_RET_USAGE;
817 if (fs_set_blk_dev(argv[1], argv[2], fstype))
822 return CMD_RET_FAILURE;
825 env_set(argv[3], uuid);
827 printf("%s\n", uuid);
829 return CMD_RET_SUCCESS;
832 int do_fs_type(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
834 struct fstype_info *info;
836 if (argc < 3 || argc > 4)
837 return CMD_RET_USAGE;
839 if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
842 info = fs_get_info(fs_type);
845 env_set(argv[3], info->name);
847 printf("%s\n", info->name);
851 return CMD_RET_SUCCESS;
854 int do_rm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
858 return CMD_RET_USAGE;
860 if (fs_set_blk_dev(argv[1], argv[2], fstype))
863 if (fs_unlink(argv[3]))
869 int do_mkdir(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
875 return CMD_RET_USAGE;
877 if (fs_set_blk_dev(argv[1], argv[2], fstype))
880 ret = fs_mkdir(argv[3]);
882 printf("** Unable to create a directory \"%s\" **\n", argv[3]);
889 int do_ln(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
893 return CMD_RET_USAGE;
895 if (fs_set_blk_dev(argv[1], argv[2], fstype))
898 if (fs_ln(argv[3], argv[4]))