/*
   * Copyright (c) 2012, NVIDIA CORPORATION.  All rights reserved.
   *

   * SPDX-License-Identifier:	GPL-2.0

   */
  
  #include <config.h>

  #include <errno.h>

  #include <common.h>

  #include <mapmem.h>

  #include <part.h>
  #include <ext4fs.h>
  #include <fat.h>
  #include <fs.h>

  #include <sandboxfs.h>

  #include <ubifs_uboot.h>

  #include <btrfs.h>

  #include <asm/io.h>

  #include <div64.h>
  #include <linux/math64.h>

  DECLARE_GLOBAL_DATA_PTR;

  static struct blk_desc *fs_dev_desc;

  static int fs_dev_part;

  static disk_partition_t fs_partition;
  static int fs_type = FS_TYPE_ANY;

  static inline int fs_probe_unsupported(struct blk_desc *fs_dev_desc,

  				      disk_partition_t *fs_partition)

  {
  	printf("** Unrecognized filesystem type **
  ");
  	return -1;
  }

  static inline int fs_ls_unsupported(const char *dirname)
  {

  	return -1;
  }

  /* generic implementation of ls in terms of opendir/readdir/closedir */
  __maybe_unused
  static int fs_ls_generic(const char *dirname)
  {
  	struct fs_dir_stream *dirs;
  	struct fs_dirent *dent;
  	int nfiles = 0, ndirs = 0;
  
  	dirs = fs_opendir(dirname);
  	if (!dirs)
  		return -errno;
  
  	while ((dent = fs_readdir(dirs))) {
  		if (dent->type == FS_DT_DIR) {
  			printf("            %s/
  ", dent->name);
  			ndirs++;
  		} else {
  			printf(" %8lld   %s
  ", dent->size, dent->name);
  			nfiles++;
  		}
  	}
  
  	fs_closedir(dirs);
  
  	printf("
  %d file(s), %d dir(s)
  
  ", nfiles, ndirs);
  
  	return 0;
  }

  static inline int fs_exists_unsupported(const char *filename)
  {
  	return 0;
  }

  static inline int fs_size_unsupported(const char *filename, loff_t *size)

  {
  	return -1;
  }

  static inline int fs_read_unsupported(const char *filename, void *buf,

  				      loff_t offset, loff_t len,
  				      loff_t *actread)

  {

  	return -1;
  }

  static inline int fs_write_unsupported(const char *filename, void *buf,

  				      loff_t offset, loff_t len,
  				      loff_t *actwrite)

  {
  	return -1;
  }

  static inline void fs_close_unsupported(void)
  {
  }

  static inline int fs_uuid_unsupported(char *uuid_str)
  {
  	return -1;
  }

  static inline int fs_opendir_unsupported(const char *filename,
  					 struct fs_dir_stream **dirs)
  {
  	return -EACCES;
  }

  struct fstype_info {

  	int fstype;

  	char *name;

  	/*
  	 * Is it legal to pass NULL as .probe()'s  fs_dev_desc parameter? This
  	 * should be false in most cases. For "virtual" filesystems which
  	 * aren't based on a U-Boot block device (e.g. sandbox), this can be
  	 * set to true. This should also be true for the dumm entry at the end
  	 * of fstypes[], since that is essentially a "virtual" (non-existent)
  	 * filesystem.
  	 */
  	bool null_dev_desc_ok;

  	int (*probe)(struct blk_desc *fs_dev_desc,

  		     disk_partition_t *fs_partition);

  	int (*ls)(const char *dirname);

  	int (*exists)(const char *filename);

  	int (*size)(const char *filename, loff_t *size);
  	int (*read)(const char *filename, void *buf, loff_t offset,
  		    loff_t len, loff_t *actread);
  	int (*write)(const char *filename, void *buf, loff_t offset,
  		     loff_t len, loff_t *actwrite);

  	void (*close)(void);

  	int (*uuid)(char *uuid_str);

  	/*
  	 * Open a directory stream.  On success return 0 and directory
  	 * stream pointer via 'dirsp'.  On error, return -errno.  See
  	 * fs_opendir().
  	 */
  	int (*opendir)(const char *filename, struct fs_dir_stream **dirsp);
  	/*
  	 * Read next entry from directory stream.  On success return 0
  	 * and directory entry pointer via 'dentp'.  On error return
  	 * -errno.  See fs_readdir().
  	 */
  	int (*readdir)(struct fs_dir_stream *dirs, struct fs_dirent **dentp);
  	/* see fs_closedir() */
  	void (*closedir)(struct fs_dir_stream *dirs);

  };
  
  static struct fstype_info fstypes[] = {
  #ifdef CONFIG_FS_FAT

  	{
  		.fstype = FS_TYPE_FAT,

  		.name = "fat",

  		.null_dev_desc_ok = false,

  		.probe = fat_set_blk_dev,
  		.close = fat_close,

  		.ls = fs_ls_generic,

  		.exists = fat_exists,

  		.size = fat_size,

  		.read = fat_read_file,

  #ifdef CONFIG_FAT_WRITE
  		.write = file_fat_write,
  #else

  		.write = fs_write_unsupported,

  #endif

  		.uuid = fs_uuid_unsupported,

  		.opendir = fat_opendir,
  		.readdir = fat_readdir,
  		.closedir = fat_closedir,

  	},

  #endif
  #ifdef CONFIG_FS_EXT4

  	{
  		.fstype = FS_TYPE_EXT,

  		.name = "ext4",

  		.null_dev_desc_ok = false,

  		.probe = ext4fs_probe,
  		.close = ext4fs_close,

  		.ls = ext4fs_ls,

  		.exists = ext4fs_exists,

  		.size = ext4fs_size,

  		.read = ext4_read_file,

  #ifdef CONFIG_CMD_EXT4_WRITE
  		.write = ext4_write_file,
  #else

  		.write = fs_write_unsupported,

  #endif

  		.uuid = ext4fs_uuid,

  		.opendir = fs_opendir_unsupported,

  	},
  #endif

  #ifdef CONFIG_SANDBOX
  	{
  		.fstype = FS_TYPE_SANDBOX,

  		.name = "sandbox",

  		.null_dev_desc_ok = true,

  		.probe = sandbox_fs_set_blk_dev,
  		.close = sandbox_fs_close,
  		.ls = sandbox_fs_ls,

  		.exists = sandbox_fs_exists,

  		.size = sandbox_fs_size,

  		.read = fs_read_sandbox,

  		.write = fs_write_sandbox,

  		.uuid = fs_uuid_unsupported,

  		.opendir = fs_opendir_unsupported,

  	},
  #endif

  #ifdef CONFIG_CMD_UBIFS
  	{
  		.fstype = FS_TYPE_UBIFS,
  		.name = "ubifs",
  		.null_dev_desc_ok = true,
  		.probe = ubifs_set_blk_dev,
  		.close = ubifs_close,
  		.ls = ubifs_ls,
  		.exists = ubifs_exists,
  		.size = ubifs_size,
  		.read = ubifs_read,
  		.write = fs_write_unsupported,
  		.uuid = fs_uuid_unsupported,

  		.opendir = fs_opendir_unsupported,

  	},
  #endif

  #ifdef CONFIG_FS_BTRFS
  	{
  		.fstype = FS_TYPE_BTRFS,
  		.name = "btrfs",
  		.null_dev_desc_ok = false,
  		.probe = btrfs_probe,
  		.close = btrfs_close,
  		.ls = btrfs_ls,
  		.exists = btrfs_exists,
  		.size = btrfs_size,
  		.read = btrfs_read,
  		.write = fs_write_unsupported,
  		.uuid = btrfs_uuid,

  		.opendir = fs_opendir_unsupported,

  	},
  #endif

  	{
  		.fstype = FS_TYPE_ANY,

  		.name = "unsupported",

  		.null_dev_desc_ok = true,

  		.probe = fs_probe_unsupported,
  		.close = fs_close_unsupported,
  		.ls = fs_ls_unsupported,

  		.exists = fs_exists_unsupported,

  		.size = fs_size_unsupported,

  		.read = fs_read_unsupported,

  		.write = fs_write_unsupported,

  		.uuid = fs_uuid_unsupported,

  		.opendir = fs_opendir_unsupported,

  	},
  };

  static struct fstype_info *fs_get_info(int fstype)
  {
  	struct fstype_info *info;
  	int i;
  
  	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
  		if (fstype == info->fstype)
  			return info;
  	}
  
  	/* Return the 'unsupported' sentinel */
  	return info;
  }

  int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
  {

  	struct fstype_info *info;

  	int part, i;

  #ifdef CONFIG_NEEDS_MANUAL_RELOC
  	static int relocated;
  
  	if (!relocated) {

  		for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
  				i++, info++) {

  			info->name += gd->reloc_off;

  			info->probe += gd->reloc_off;
  			info->close += gd->reloc_off;
  			info->ls += gd->reloc_off;
  			info->read += gd->reloc_off;

  			info->write += gd->reloc_off;

  		}

  		relocated = 1;
  	}
  #endif

  	part = blk_get_device_part_str(ifname, dev_part_str, &fs_dev_desc,

  					&fs_partition, 1);
  	if (part < 0)
  		return -1;

  	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
  		if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
  				fstype != info->fstype)

  			continue;

  		if (!fs_dev_desc && !info->null_dev_desc_ok)
  			continue;

  		if (!info->probe(fs_dev_desc, &fs_partition)) {

  			fs_type = info->fstype;

  			fs_dev_part = part;
  			return 0;
  		}
  	}
  
  	return -1;
  }
  
  /* set current blk device w/ blk_desc + partition # */
  int fs_set_blk_dev_with_part(struct blk_desc *desc, int part)
  {
  	struct fstype_info *info;
  	int ret, i;
  
  	if (part >= 1)
  		ret = part_get_info(desc, part, &fs_partition);
  	else
  		ret = part_get_info_whole_disk(desc, &fs_partition);
  	if (ret)
  		return ret;
  	fs_dev_desc = desc;
  
  	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
  		if (!info->probe(fs_dev_desc, &fs_partition)) {
  			fs_type = info->fstype;

  			return 0;
  		}
  	}

  	return -1;
  }
  
  static void fs_close(void)
  {

  	struct fstype_info *info = fs_get_info(fs_type);

  	info->close();

  	fs_type = FS_TYPE_ANY;
  }

  int fs_uuid(char *uuid_str)
  {
  	struct fstype_info *info = fs_get_info(fs_type);
  
  	return info->uuid(uuid_str);
  }

  int fs_ls(const char *dirname)
  {
  	int ret;

  	struct fstype_info *info = fs_get_info(fs_type);
  
  	ret = info->ls(dirname);

  	fs_type = FS_TYPE_ANY;

  	fs_close();
  
  	return ret;
  }

  int fs_exists(const char *filename)
  {
  	int ret;
  
  	struct fstype_info *info = fs_get_info(fs_type);
  
  	ret = info->exists(filename);
  
  	fs_close();
  
  	return ret;
  }

  int fs_size(const char *filename, loff_t *size)

  {
  	int ret;
  
  	struct fstype_info *info = fs_get_info(fs_type);

  	ret = info->size(filename, size);

  
  	fs_close();
  
  	return ret;
  }

  int fs_read(const char *filename, ulong addr, loff_t offset, loff_t len,
  	    loff_t *actread)

  {

  	struct fstype_info *info = fs_get_info(fs_type);

  	void *buf;

  	int ret;

  	/*
  	 * We don't actually know how many bytes are being read, since len==0
  	 * means read the whole file.
  	 */
  	buf = map_sysmem(addr, len);

  	ret = info->read(filename, buf, offset, len, actread);

  	unmap_sysmem(buf);

  	/* If we requested a specific number of bytes, check we got it */

  	if (ret == 0 && len && *actread != len)

  		debug("** %s shorter than offset + len **
  ", filename);

  	fs_close();
  
  	return ret;
  }

  int fs_write(const char *filename, ulong addr, loff_t offset, loff_t len,
  	     loff_t *actwrite)

  {
  	struct fstype_info *info = fs_get_info(fs_type);
  	void *buf;
  	int ret;

  	buf = map_sysmem(addr, len);

  	ret = info->write(filename, buf, offset, len, actwrite);

  	unmap_sysmem(buf);

  	if (ret < 0 && len != *actwrite) {

  		printf("** Unable to write file %s **
  ", filename);
  		ret = -1;
  	}
  	fs_close();
  
  	return ret;
  }

  struct fs_dir_stream *fs_opendir(const char *filename)
  {
  	struct fstype_info *info = fs_get_info(fs_type);
  	struct fs_dir_stream *dirs = NULL;
  	int ret;
  
  	ret = info->opendir(filename, &dirs);
  	fs_close();
  	if (ret) {
  		errno = -ret;
  		return NULL;
  	}
  
  	dirs->desc = fs_dev_desc;
  	dirs->part = fs_dev_part;
  
  	return dirs;
  }
  
  struct fs_dirent *fs_readdir(struct fs_dir_stream *dirs)
  {
  	struct fstype_info *info;
  	struct fs_dirent *dirent;
  	int ret;
  
  	fs_set_blk_dev_with_part(dirs->desc, dirs->part);
  	info = fs_get_info(fs_type);
  
  	ret = info->readdir(dirs, &dirent);
  	fs_close();
  	if (ret) {
  		errno = -ret;
  		return NULL;
  	}
  
  	return dirent;
  }
  
  void fs_closedir(struct fs_dir_stream *dirs)
  {
  	struct fstype_info *info;
  
  	if (!dirs)
  		return;
  
  	fs_set_blk_dev_with_part(dirs->desc, dirs->part);
  	info = fs_get_info(fs_type);
  
  	info->closedir(dirs);
  	fs_close();
  }

  int do_size(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
  		int fstype)
  {

  	loff_t size;

  
  	if (argc != 4)
  		return CMD_RET_USAGE;
  
  	if (fs_set_blk_dev(argv[1], argv[2], fstype))
  		return 1;

  	if (fs_size(argv[3], &size) < 0)

  		return CMD_RET_FAILURE;

  	env_set_hex("filesize", size);

  
  	return 0;
  }

  int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],

  		int fstype)

  {
  	unsigned long addr;
  	const char *addr_str;
  	const char *filename;

  	loff_t bytes;
  	loff_t pos;
  	loff_t len_read;
  	int ret;

  	unsigned long time;

  	char *ep;

  	if (argc < 2)
  		return CMD_RET_USAGE;
  	if (argc > 7)

  		return CMD_RET_USAGE;

  	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))

  		return 1;
  
  	if (argc >= 4) {

  		addr = simple_strtoul(argv[3], &ep, 16);
  		if (ep == argv[3] || *ep != '\0')
  			return CMD_RET_USAGE;

  	} else {

  		addr_str = env_get("loadaddr");

  		if (addr_str != NULL)
  			addr = simple_strtoul(addr_str, NULL, 16);
  		else
  			addr = CONFIG_SYS_LOAD_ADDR;
  	}
  	if (argc >= 5) {
  		filename = argv[4];
  	} else {

  		filename = env_get("bootfile");

  		if (!filename) {
  			puts("** No boot file defined **
  ");
  			return 1;
  		}
  	}
  	if (argc >= 6)

  		bytes = simple_strtoul(argv[5], NULL, 16);

  	else
  		bytes = 0;
  	if (argc >= 7)

  		pos = simple_strtoul(argv[6], NULL, 16);

  	else
  		pos = 0;

  	time = get_timer(0);

  	ret = fs_read(filename, addr, pos, bytes, &len_read);

  	time = get_timer(time);

  	if (ret < 0)

  		return 1;

  	printf("%llu bytes read in %lu ms", len_read, time);

  	if (time > 0) {
  		puts(" (");

  		print_size(div_u64(len_read, time) * 1000, "/s");

  		puts(")");
  	}
  	puts("
  ");

  	env_set_hex("fileaddr", addr);
  	env_set_hex("filesize", len_read);

  
  	return 0;
  }
  
  int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
  	int fstype)
  {
  	if (argc < 2)
  		return CMD_RET_USAGE;

  	if (argc > 4)
  		return CMD_RET_USAGE;

  
  	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
  		return 1;

  	if (fs_ls(argc >= 4 ? argv[3] : "/"))

  		return 1;
  
  	return 0;
  }

  int file_exists(const char *dev_type, const char *dev_part, const char *file,
  		int fstype)
  {
  	if (fs_set_blk_dev(dev_type, dev_part, fstype))
  		return 0;
  
  	return fs_exists(file);
  }

  int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],

  		int fstype)

  {
  	unsigned long addr;
  	const char *filename;

  	loff_t bytes;
  	loff_t pos;
  	loff_t len;
  	int ret;

  	unsigned long time;
  
  	if (argc < 6 || argc > 7)
  		return CMD_RET_USAGE;
  
  	if (fs_set_blk_dev(argv[1], argv[2], fstype))
  		return 1;

  	addr = simple_strtoul(argv[3], NULL, 16);
  	filename = argv[4];

  	bytes = simple_strtoul(argv[5], NULL, 16);

  	if (argc >= 7)

  		pos = simple_strtoul(argv[6], NULL, 16);

  	else
  		pos = 0;
  
  	time = get_timer(0);

  	ret = fs_write(filename, addr, pos, bytes, &len);

  	time = get_timer(time);

  	if (ret < 0)

  		return 1;

  	printf("%llu bytes written in %lu ms", len, time);

  	if (time > 0) {
  		puts(" (");

  		print_size(div_u64(len, time) * 1000, "/s");

  		puts(")");
  	}
  	puts("
  ");
  
  	return 0;
  }

  
  int do_fs_uuid(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
  		int fstype)
  {
  	int ret;
  	char uuid[37];
  	memset(uuid, 0, sizeof(uuid));
  
  	if (argc < 3 || argc > 4)
  		return CMD_RET_USAGE;
  
  	if (fs_set_blk_dev(argv[1], argv[2], fstype))
  		return 1;
  
  	ret = fs_uuid(uuid);
  	if (ret)
  		return CMD_RET_FAILURE;
  
  	if (argc == 4)

  		env_set(argv[3], uuid);

  	else
  		printf("%s
  ", uuid);
  
  	return CMD_RET_SUCCESS;
  }

  
  int do_fs_type(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	struct fstype_info *info;
  
  	if (argc < 3 || argc > 4)
  		return CMD_RET_USAGE;
  
  	if (fs_set_blk_dev(argv[1], argv[2], FS_TYPE_ANY))
  		return 1;
  
  	info = fs_get_info(fs_type);
  
  	if (argc == 4)

  		env_set(argv[3], info->name);

  	else
  		printf("%s
  ", info->name);
  
  	return CMD_RET_SUCCESS;
  }