/*
   * (C) Copyright 2008
   * Stefan Roese, DENX Software Engineering, sr@denx.de.
   *
   * See file CREDITS for list of people who contributed to this
   * project.
   *
   * This program is free software; you can redistribute it and/or
   * modify it under the terms of the GNU General Public License as
   * published by the Free Software Foundation; either version 2 of
   * the License, or (at your option) any later version.
   *
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with this program; if not, write to the Free Software
   * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
   * MA 02111-1307 USA
   *
   */
  
  
  /*
   * UBIFS command support
   */
  
  #undef DEBUG
  
  #include <common.h>
  #include <config.h>
  #include <command.h>

  #include "../fs/ubifs/ubifs.h"

  static int ubifs_initialized;
  static int ubifs_mounted;

  extern struct super_block *ubifs_sb;

  /* Prototypes */
  int ubifs_init(void);
  int ubifs_mount(char *vol_name);

  void ubifs_umount(struct ubifs_info *c);

  int ubifs_ls(char *dir_name);
  int ubifs_load(char *filename, u32 addr, u32 size);

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

  {
  	char *vol_name;
  	int ret;

  	if (argc != 2)

  		return CMD_RET_USAGE;

  	vol_name = argv[1];
  	debug("Using volume %s
  ", vol_name);
  
  	if (ubifs_initialized == 0) {
  		ubifs_init();
  		ubifs_initialized = 1;
  	}
  
  	ret = ubifs_mount(vol_name);
  	if (ret)
  		return -1;
  
  	ubifs_mounted = 1;
  
  	return 0;
  }

  int ubifs_is_mounted(void)
  {
  	return ubifs_mounted;
  }
  
  void cmd_ubifs_umount(void)
  {
  
  	if (ubifs_sb) {
  		printf("Unmounting UBIFS volume %s!
  ",
  		       ((struct ubifs_info *)(ubifs_sb->s_fs_info))->vi.name);
  		ubifs_umount(ubifs_sb->s_fs_info);
  	}
  
  	ubifs_sb = NULL;
  	ubifs_mounted = 0;
  	ubifs_initialized = 0;
  }

  int do_ubifs_umount(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	if (argc != 1)

  		return CMD_RET_USAGE;

  
  	if (ubifs_initialized == 0) {
  		printf("No UBIFS volume mounted!
  ");
  		return -1;
  	}

  	cmd_ubifs_umount();

  
  	return 0;
  }

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

  {
  	char *filename = "/";
  	int ret;
  
  	if (!ubifs_mounted) {

  		printf("UBIFS not mounted, use ubifsmount to mount volume first!
  ");

  		return -1;
  	}
  
  	if (argc == 2)
  		filename = argv[1];
  	debug("Using filename %s
  ", filename);
  
  	ret = ubifs_ls(filename);
  	if (ret)
  		printf("%s not found!
  ", filename);
  
  	return ret;
  }

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

  {
  	char *filename;

  	char *endp;

  	int ret;
  	u32 addr;
  	u32 size = 0;
  
  	if (!ubifs_mounted) {
  		printf("UBIFS not mounted, use ubifs mount to mount volume first!
  ");
  		return -1;
  	}

  	if (argc < 3)

  		return CMD_RET_USAGE;

  	addr = simple_strtoul(argv[1], &endp, 16);

  	if (endp == argv[1])

  		return CMD_RET_USAGE;

  	filename = argv[2];

  	if (argc == 4) {
  		size = simple_strtoul(argv[3], &endp, 16);

  		if (endp == argv[3])

  			return CMD_RET_USAGE;

  	}

  	debug("Loading file '%s' to address 0x%08x (size %d)
  ", filename, addr, size);
  
  	ret = ubifs_load(filename, addr, size);
  	if (ret)
  		printf("%s not found!
  ", filename);
  
  	return ret;
  }
  
  U_BOOT_CMD(
  	ubifsmount, 2, 0, do_ubifs_mount,

  	"mount UBIFS volume",

  	"<volume-name>
  "
  	"    - mount 'volume-name' volume"

  );

  U_BOOT_CMD(

  	ubifsumount, 1, 0, do_ubifs_umount,
  	"unmount UBIFS volume",
  	"    - unmount current volume"
  );
  
  U_BOOT_CMD(

  	ubifsls, 2, 0, do_ubifs_ls,

  	"list files in a directory",
  	"[directory]
  "
  	"    - list files in a 'directory' (default '/')"
  );

  U_BOOT_CMD(
  	ubifsload, 4, 0, do_ubifs_load,

  	"load file from an UBIFS filesystem",
  	"<addr> <filename> [bytes]
  "
  	"    - load file 'filename' to address 'addr'"
  );