/**
 * Copyright (c) 2012 Anup Patel.
 * All rights reserved.
 *
 * 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, 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., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 * @file cmd_vfs.c
 * @author Anup Patel (anup@brainfault.org)
 * @author Himanshu Chauhan (hschauhan@nulltrace.org)
 * @brief Implementation of vfs command
 */

#include <vmm_error.h>
#include <vmm_heap.h>
#include <vmm_stdio.h>
#include <vmm_devtree.h>
#include <vmm_wallclock.h>
#include <vmm_manager.h>
#include <vmm_host_aspace.h>
#include <vmm_guest_aspace.h>
#include <vmm_modules.h>
#include <vmm_cmdmgr.h>
#include <vmm_delay.h>
#include <libs/libfdt.h>
#include <libs/stringlib.h>
#include <libs/vfs.h>

#if CONFIG_CRYPTO_HASH_MD5
#include <libs/md5.h>
#endif

#if CONFIG_CRYPTO_HASH_SHA256
#include <libs/sha256.h>
#endif

#define MODULE_DESC			"Command vfs"
#define MODULE_AUTHOR			"Anup Patel"
#define MODULE_LICENSE			"GPL"
#define MODULE_IPRIORITY		(VFS_IPRIORITY+1)
#define	MODULE_INIT			cmd_vfs_init
#define	MODULE_EXIT			cmd_vfs_exit

#define VFS_MAX_MODULE_SZ		(256 * 1024)
#define VFS_MAX_FDT_SZ			(32 * 1024)
#define VFS_LOAD_BUF_SZ			(4 * 1024)

static void cmd_vfs_usage(struct vmm_chardev *cdev)
{
	vmm_cprintf(cdev, "Usage:\n");
	vmm_cprintf(cdev, "   vfs help\n");
	vmm_cprintf(cdev, "   vfs fslist\n");
	vmm_cprintf(cdev, "   vfs mplist\n");
	vmm_cprintf(cdev, "   vfs mount <bdev_name> <path_to_mount> "
			  "[wait_sec]\n");
	vmm_cprintf(cdev, "   vfs umount <path_to_unmount>\n");
	vmm_cprintf(cdev, "   vfs ls <path_to_dir>\n");
	vmm_cprintf(cdev, "   vfs cat <path_to_file>\n");
#if CONFIG_CRYPTO_HASH_MD5
	vmm_cprintf(cdev, "   vfs md5 <path_to_file>\n");
#endif
#if CONFIG_CRYPTO_HASH_SHA256
	vmm_cprintf(cdev, "   vfs sha256 <path_to_file>\n");
#endif
	vmm_cprintf(cdev, "   vfs run <path_to_file>\n");
	vmm_cprintf(cdev, "   vfs mv <old_path> <new_path>\n");
	vmm_cprintf(cdev, "   vfs rm <path_to_file>\n");
	vmm_cprintf(cdev, "   vfs mkdir <path_to_dir>\n");
	vmm_cprintf(cdev, "   vfs rmdir <path_to_dir>\n");
	vmm_cprintf(cdev, "   vfs module_load <path_to_module_file>\n");
	vmm_cprintf(cdev, "   vfs fdt_load <devtree_path> "
			  "<devtree_root_name> <path_to_fdt_file> "
			  "[<alias0>,<attr_name>,<attr_type>,<value>] "
			  "[<alias1>,<attr_name>,<attr_type>,<value>] ...\n");
	vmm_cprintf(cdev, "   vfs guest_fdt_load <guest_name> "
			  "<path_to_fdt_file> <vcpu_count> "
			  "[<alias0>,<attr_name>,<attr_type>,<value>] "
			  "[<alias1>,<attr_name>,<attr_type>,<value>] ...\n");
	vmm_cprintf(cdev, "   vfs host_load <host_phys_addr> "
			  "<path_to_file> [<file_offset>] [<byte_count>]\n");
	vmm_cprintf(cdev, "   vfs host_load_list <path_to_list_file>\n");
	vmm_cprintf(cdev, "   vfs guest_load <guest_name> <guest_phys_addr> "
			  "<path_to_file> [<file_offset>] [<byte_count>]\n");
	vmm_cprintf(cdev, "   vfs guest_load_list <guest_name> "
			  "<path_to_list_file>\n");
	vmm_cprintf(cdev, "Note:\n");
	vmm_cprintf(cdev, "   <attr_type> = unknown|string|bytes|"
					   "uint32|uint64|"
			  		   "physaddr|physsize|"
					   "virtaddr|virtsize\n");
}

static int cmd_vfs_fslist(struct vmm_chardev *cdev)
{
	int num, count;
	struct filesystem *fs;

	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");
	vmm_cprintf(cdev, " %-9s %-69s\n", "Num", "Name");
	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");
	count = vfs_filesystem_count();
	for (num = 0; num < count; num++) {
		fs = vfs_filesystem_get(num);
		vmm_cprintf(cdev, " %-9d %-69s\n", num, fs->name);
	}
	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");

	return VMM_OK;
}

static int cmd_vfs_mplist(struct vmm_chardev *cdev)
{
	int num, count;
	const char *mode;
	struct mount *m;

	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");
	vmm_cprintf(cdev, " %-15s %-11s %-11s %-39s\n",
			  "BlockDev", "Filesystem", "Mode", "Path");
	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");
	count = vfs_mount_count();
	for (num = 0; num < count; num++) {
		m = vfs_mount_get(num);
		switch (m->m_flags & MOUNT_MASK) {
		case MOUNT_RDONLY:
			mode = "read-only";
			break;
		case MOUNT_RW:
			mode = "read-write";
			break;
		default:
			mode = "unknown";
			break;
		};
		vmm_cprintf(cdev, " %-15s %-11s %-11s %-39s\n",
				  m->m_dev->name, m->m_fs->name,
				  mode, m->m_path);
	}
	vmm_cprintf(cdev, "----------------------------------------"
			  "----------------------------------------\n");

	return VMM_OK;
}

static int cmd_vfs_mount(struct vmm_chardev *cdev,
			 const char *dev, const char *path,
			 long *pwait)
{
	int rc;
	bool found;
	int fd, num, count;
	struct vmm_blockdev *bdev;
	struct filesystem *fs;
	long wait = (pwait) ? *pwait : 0;

	do {
		bdev = vmm_blockdev_find(dev);
		if (bdev || (wait <= 0))
			break;
		vmm_msleep(1000);
		wait -= 1;
	} while (wait > 0);
	if (!bdev) {
		vmm_cprintf(cdev, "Block device %s not found\n", dev);
		return VMM_ENODEV;
	}

	if (strcmp(path, "/") != 0) {
		fd = vfs_opendir(path);
		if (fd < 0) {
			vmm_cprintf(cdev, "Directory %s not found\n", path);
			return fd;
		} else {
			vfs_closedir(fd);
		}
	}

	found = FALSE;
	count = vfs_filesystem_count();
	vmm_cprintf(cdev, "Trying:");
	for (num = 0; num < count; num++) {
		fs = vfs_filesystem_get(num);
		vmm_cprintf(cdev, " %s", fs->name);
		rc = vfs_mount(path, fs->name, dev, MOUNT_RW);
		if (!rc) {
			found = TRUE;
			vmm_cprintf(cdev, "\n");
			break;
		}
	}

	if (!found) {
		vmm_cprintf(cdev, "\nMount failed\n");
		return VMM_ENOSYS;
	}

	vmm_cprintf(cdev, "Mounted %s using %s at %s\n", dev, fs->name, path);

	return VMM_OK;
}

static int cmd_vfs_umount(struct vmm_chardev *cdev, const char *path)
{
	int rc;

	rc = vfs_unmount(path);
	if (rc) {
		vmm_cprintf(cdev, "Unmount failed\n");
	} else {
		vmm_cprintf(cdev, "Unmount successful\n");
	}

	return rc;
}

static int cmd_vfs_ls(struct vmm_chardev *cdev, const char *path)
{
	char type[11];
	char dpath[VFS_MAX_PATH];
	int fd, rc, plen, total_ent;
	struct vmm_timeinfo ti;
	struct stat st;
	struct dirent d;

	fd = vfs_opendir(path);
	if (fd < 0) {
		vmm_cprintf(cdev, "Failed to opendir %s\n", path);
		return fd;
	}

	if ((plen = strlcpy(dpath, path, sizeof(dpath))) >= sizeof(dpath)) {
		rc = VMM_EOVERFLOW;
		goto closedir_fail;
	}

	if (path[plen-1] != '/') {
		if ((plen = strlcat(dpath, "/", sizeof(dpath))) >=
		    sizeof(dpath)) {
			rc = VMM_EOVERFLOW;
			goto closedir_fail;
		}
	}

	total_ent = 0;
	while (!vfs_readdir(fd, &d)) {
		dpath[plen] = '\0';
		if (strlcat(dpath, d.d_name, sizeof(dpath)) >= sizeof(dpath)) {
			rc = VMM_EOVERFLOW;
			goto closedir_fail;
		}
		rc = vfs_stat(dpath, &st);
		if (rc) {
			vmm_cprintf(cdev, "Failed to get %s stat\n", dpath);
			goto closedir_fail;
		}
		strcpy(type, "----------");
		if (st.st_mode & S_IFDIR) {
			type[0]= 'd';
		} else if (st.st_mode & S_IFCHR) {
			type[0]= 'c';
		} else if (st.st_mode & S_IFBLK) {
			type[0]= 'b';
		} else if (st.st_mode & S_IFLNK) {
			type[0]= 'l';
		}
		if (st.st_mode & S_IRUSR) {
			type[1] = 'r';
		}
		if (st.st_mode & S_IWUSR) {
			type[2] = 'w';
		}
		if (st.st_mode & S_IXUSR) {
			type[3] = 'x';
		}
		if (st.st_mode & S_IRGRP) {
			type[4] = 'r';
		}
		if (st.st_mode & S_IWGRP) {
			type[5] = 'w';
		}
		if (st.st_mode & S_IXGRP) {
			type[6] = 'x';
		}
		if (st.st_mode & S_IROTH) {
			type[7] = 'r';
		}
		if (st.st_mode & S_IWOTH) {
			type[8] = 'w';
		}
		if (st.st_mode & S_IXOTH) {
			type[9] = 'x';
		}
		vmm_cprintf(cdev, "%10s ", type);
		vmm_cprintf(cdev, "%10lld ", st.st_size);
		vmm_wallclock_mkinfo(st.st_mtime, 0, &ti);
		switch (ti.tm_mon) {
		case 0:
			vmm_cprintf(cdev, "%s ", "Jan");
			break;
		case 1:
			vmm_cprintf(cdev, "%s ", "Feb");
			break;
		case 2:
			vmm_cprintf(cdev, "%s ", "Mar");
			break;
		case 3:
			vmm_cprintf(cdev, "%s ", "Apr");
			break;
		case 4:
			vmm_cprintf(cdev, "%s ", "May");
			break;
		case 5:
			vmm_cprintf(cdev, "%s ", "Jun");
			break;
		case 6:
			vmm_cprintf(cdev, "%s ", "Jul");
			break;
		case 7:
			vmm_cprintf(cdev, "%s ", "Aug");
			break;
		case 8:
			vmm_cprintf(cdev, "%s ", "Sep");
			break;
		case 9:
			vmm_cprintf(cdev, "%s ", "Oct");
			break;
		case 10:
			vmm_cprintf(cdev, "%s ", "Nov");
			break;
		case 11:
			vmm_cprintf(cdev, "%s ", "Dec");
			break;
		default:
			break;
		};
		vmm_cprintf(cdev, "%02d %ld %02d:%02d:%02d ",
				  ti.tm_mday, ti.tm_year + 1900,
				  ti.tm_hour, ti.tm_min, ti.tm_sec);
		if (type[0] == 'd')
			vmm_cprintf(cdev, "%s/\n", d.d_name);
		else
			vmm_cprintf(cdev, "%s\n", d.d_name);

		total_ent++;
	}
	vmm_cprintf(cdev, "total %d\n", total_ent);
	rc = vfs_closedir(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to closedir %s\n", path);
		return rc;
	}

	return VMM_OK;

closedir_fail:
	vfs_closedir(fd);
	return rc;
}

static int cmd_vfs_file_parent_dir(struct vmm_chardev *cdev,
				   const char *path,
				   char *out, size_t out_sz)
{
	size_t out_len;
	int rc = VMM_OK;
	struct stat st;

	rc = vfs_stat(path, &st);
	if (rc) {
		vmm_cprintf(cdev, "Failed to stat %s\n", path);
		return rc;
	}

	if (!(st.st_mode & S_IFREG)) {
		vmm_cprintf(cdev, "Cannot read %s\n", path);
		return VMM_EINVALID;
	}

	out_len = strlcpy(out, path, out_sz);
	while (out_len) {
		if (out[out_len - 1] == '/') {
			out[out_len - 1] = '\0';
			break;
		}
		out_len--;
	}

	return VMM_OK;
}

static int cmd_vfs_file_open_read(struct vmm_chardev *cdev,
				  const char *curdir,
				  const char *path,
				  int *fdp, u32 *len)
{
	int rc = VMM_OK;
	int fd = -1;
	struct stat st;
	char *npath = NULL;

	while (*path == ' ') {
		path++;
	}

	if (*path == '.' && *(path + 1) == '/' && curdir) {
		npath = vmm_zalloc(VFS_MAX_PATH * 2);
		if (!npath) {
			vmm_cprintf(cdev,
				    "Failed to alloc new path buffer\n");
			return VMM_ENOMEM;
		}
		path++;
		strlcpy(npath, curdir, VFS_MAX_PATH * 2);
		strlcat(npath, path, VFS_MAX_PATH * 2);
		fd = vfs_open(npath, O_RDONLY, 0);
		vmm_free(npath);
	} else {
		fd = vfs_open(path, O_RDONLY, 0);
	}
	if (fd < 0) {
		vmm_cprintf(cdev, "Failed to open %s\n", path);
		return fd;
	}

	rc = vfs_fstat(fd, &st);
	if (rc) {
		vfs_close(fd);
		vmm_cprintf(cdev, "Failed to stat %s\n", path);
		return rc;
	}

	if (!(st.st_mode & S_IFREG)) {
		vfs_close(fd);
		vmm_cprintf(cdev, "Cannot read %s\n", path);
		return VMM_EINVALID;
	}

	*len = st.st_size;
	*fdp = fd;

	return VMM_OK;
}

static size_t cmd_vfs_file_buf_read(int fd, char buf[], u32 len)
{
	memset(buf, 0, VFS_LOAD_BUF_SZ);

	len = (len < VFS_LOAD_BUF_SZ) ? len : VFS_LOAD_BUF_SZ;
	return vfs_read(fd, buf, len);
}

static int cmd_vfs_run(struct vmm_chardev *cdev, const char *path)
{
	int fd, rc;
	u32 line, len;
	loff_t buf_off;
	size_t end, buf_rd, buf_valid;
	char *buf = NULL;

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		return rc;
	}

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		vfs_close(fd);
		return VMM_ENOMEM;
	}

	line = 0;
	buf_off = 0;
	while (len) {
		buf_rd = cmd_vfs_file_buf_read(fd, buf, VFS_LOAD_BUF_SZ);
		if (buf_rd < 1) {
			break;
		}

		for (end = 0; end < buf_rd; end++) {
			if (buf[end] == '\n' ||
			    buf[end] == '\r' ||
			    buf[end] == '\0')
				break;
		}

		if (end == buf_rd) {
			vmm_cprintf(cdev, "line%d: failed to find end-of-line"
				    " (%zd bytes)\n", line, buf_rd);
			break;
		}

		buf[end] = '\0';
		buf_valid = end + 1;

		if (buf_valid < buf_rd) {
			vfs_lseek(fd, buf_off + buf_valid, SEEK_SET);
		}

		for (end = 0; end < buf_valid; end++) {
			if (buf[end] == '#') {
				buf[end] = '\0';
				break;
			}
		}

		rc = vmm_cmdmgr_execute_cmdstr(cdev, buf, NULL);
		if (rc) {
			vmm_cprintf(cdev, "line%d: failed (error %d)\n",
				    line, rc);
			break;
		}

		buf_off += buf_valid;
		len -= buf_valid;
		line++;
	}

	vmm_free(buf);
	rc = vfs_close(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}

#if CONFIG_CRYPTO_HASH_MD5
static int cmd_vfs_md5(struct vmm_chardev *cdev, const char *path)
{
	int fd, rc, i;
	u32 len;
	size_t buf_rd;
	u8 *buf = NULL;
	struct md5_context md5c;
	u8 digest[16];

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		return rc;
	}

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		vfs_close(fd);
		return VMM_ENOMEM;
	}

	md5_init(&md5c);

	while (len) {
		buf_rd = cmd_vfs_file_buf_read(fd, (char *)buf, len);
		if (buf_rd < 1) {
			break;
		}
		md5_update(&md5c, buf, buf_rd);
	}

	md5_final(digest, &md5c);

	vmm_cprintf(cdev, "MD5 Digest: ");
	for (i = 0; i < 16; i++)
		vmm_cprintf(cdev, "%x", digest[i]);
	vmm_cprintf(cdev, "\n");

	vmm_free(buf);
	rc = vfs_close(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}
#endif

#if CONFIG_CRYPTO_HASH_SHA256
static int cmd_vfs_sha256(struct vmm_chardev *cdev, const char *path)
{
	int fd, rc, i;
	u32 len;
	size_t buf_rd;
	u8 *buf = NULL;
	struct sha256_context sha256c;
	sha256_digest_t digest;

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		return rc;
	}

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		vfs_close(fd);
		return VMM_ENOMEM;
	}

	sha256_init(&sha256c);

	while (len) {
		buf_rd = cmd_vfs_file_buf_read(fd, (char *)buf, len);
		if (buf_rd < 1) {
			break;
		}
		sha256_update(&sha256c, buf, buf_rd);
	}

	sha256_final(digest, &sha256c);

	vmm_cprintf(cdev, "SHA-256 Digest: ");
	for (i = 0; i < SHA256_DIGEST_LEN; i++)
		vmm_cprintf(cdev, "%x", digest[i]);
	vmm_cprintf(cdev, "\n");

	vmm_free(buf);
	rc = vfs_close(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}
#endif

static int cmd_vfs_cat(struct vmm_chardev *cdev, const char *path)
{
	int fd, rc;
	u32 i, off, len;
	bool found_non_printable;
	size_t buf_rd;
	char *buf = NULL;

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		return rc;
	}

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		vfs_close(fd);
		return VMM_ENOMEM;
	}

	off = 0;
	while (len) {
		buf_rd = cmd_vfs_file_buf_read(fd, buf, len);
		if (buf_rd < 1) {
			break;
		}

		found_non_printable = FALSE;
		for (i = 0; i < buf_rd; i++) {
			if (!vmm_isprintable(buf[i])) {
				found_non_printable = TRUE;
				break;
			}
			vmm_cputc(cdev, buf[i]);
		}
		if (found_non_printable) {
			vmm_cprintf(cdev, "\nFound non-printable char %d "
					  "at offset %d\n", buf[i], (off + i));
			break;
		}

		off += buf_rd;
		len -= buf_rd;
	}

	vmm_free(buf);
	rc = vfs_close(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}

static int cmd_vfs_mv(struct vmm_chardev *cdev,
			const char *old_path, const char *new_path)
{
	int rc;
	struct stat st;

	rc = vfs_stat(old_path, &st);
	if (rc) {
		vmm_cprintf(cdev, "Path %s does not exist.\n", old_path);
		return rc;
	}

	rc = vfs_rename(old_path, new_path);
	if (rc) {
		vmm_cprintf(cdev, "Failed to rename.\n");
		return rc;
	}

	return VMM_OK;
}

static int cmd_vfs_rm(struct vmm_chardev *cdev, const char *path)
{
	int rc;
	struct stat st;

	rc = vfs_stat(path, &st);
	if (rc) {
		vmm_cprintf(cdev, "Path %s does not exist.\n", path);
		return rc;
	}

	if (!(st.st_mode & S_IFREG)) {
		vmm_cprintf(cdev, "Path %s should be regular file.\n", path);
		return VMM_EINVALID;
	}

	return vfs_unlink(path);
}

static int cmd_vfs_mkdir(struct vmm_chardev *cdev, const char *path)
{
	int rc;
	struct stat st;

	rc = vfs_stat(path, &st);
	if (!rc) {
		vmm_cprintf(cdev, "Path %s already exist.\n", path);
		return VMM_EEXIST;
	}

	return vfs_mkdir(path, S_IRWXU|S_IRWXG|S_IRWXO);
}

static int cmd_vfs_rmdir(struct vmm_chardev *cdev, const char *path)
{
	int rc;
	struct stat st;

	rc = vfs_stat(path, &st);
	if (rc) {
		vmm_cprintf(cdev, "Path %s does not exist.\n", path);
		return rc;
	}

	if (!(st.st_mode & S_IFDIR)) {
		vmm_cprintf(cdev, "Path %s should be directory.\n", path);
		return VMM_EINVALID;
	}

	return vfs_rmdir(path);
}

static int cmd_vfs_module_load(struct vmm_chardev *cdev, const char *path)
{
	int fd, rc = VMM_OK;
	u32 len = 0;
	size_t module_rd;
	void *module_data;

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		goto fail;
	}

	if (!len) {
		vmm_cprintf(cdev, "File %s has zero bytes.\n", path);
		rc = VMM_EINVALID;
		goto fail_closefd;
	}

	if (len > VFS_MAX_MODULE_SZ) {
		vmm_cprintf(cdev, "File %s has size %"PRId32" bytes (> %d bytes).\n",
			    path, len, VFS_MAX_MODULE_SZ);
		rc = VMM_EINVALID;
		goto fail_closefd;
	}

	module_data = vmm_zalloc(VFS_MAX_MODULE_SZ);
	if (!module_data) {
		rc = VMM_ENOMEM;
		goto fail_closefd;
	}

	module_rd = vfs_read(fd, module_data, VFS_MAX_MODULE_SZ);
	if (module_rd < len) {
		rc = VMM_EIO;
		goto fail_freedata;
	}

	if ((rc = vmm_modules_load((virtual_addr_t)module_data,
				   (virtual_size_t)len))) {
		goto fail_freedata;
	} else {
		vmm_cprintf(cdev, "Loaded module succesfully\n");
	}

fail_freedata:
	vmm_free(module_data);
fail_closefd:
	vfs_close(fd);
fail:
	return rc;
}

static int cmd_vfs_fdt_load(struct vmm_chardev *cdev,
			    const char *devtree_path,
			    const char *devtree_root_name,
			    const char *path,
			    int aliasc, char **aliasv)
{
	int a, fd, rc = VMM_OK;
	u32 len = 0;
	char *astr;
	const char *aname, *apath, *aattr, *atype;
	size_t fdt_rd;
	void *fdt_data, *val = NULL;
	u32 val_type, val_len = 0;
	struct vmm_devtree_node *root, *anode, *node;
	struct vmm_devtree_node *parent;
	struct fdt_fileinfo fdt;

	parent = vmm_devtree_getnode(devtree_path);
	if (!parent) {
		vmm_cprintf(cdev, "Devtree path %s does not exist.\n",
			    devtree_path);
		return VMM_EINVALID;
	}

	root = vmm_devtree_getchild(parent, devtree_root_name);
	if (root) {
		vmm_devtree_dref_node(root);
		vmm_cprintf(cdev, "Devtree path %s/%s already exist.\n",
			    devtree_path, devtree_root_name);
		rc = VMM_EINVALID;
		goto fail;
	}

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		goto fail;
	}

	if (!len) {
		vmm_cprintf(cdev, "File %s has zero bytes.\n", path);
		rc = VMM_EINVALID;
		goto fail_closefd;
	}

	if (len > VFS_MAX_FDT_SZ) {
		vmm_cprintf(cdev, "File %s has size %"PRId32" bytes (> %d bytes).\n",
			    path, len, VFS_MAX_FDT_SZ);
		rc = VMM_EINVALID;
		goto fail_closefd;
	}

	fdt_data = vmm_zalloc(VFS_MAX_FDT_SZ);
	if (!fdt_data) {
		rc = VMM_ENOMEM;
		goto fail_closefd;
	}

	fdt_rd = vfs_read(fd, fdt_data, VFS_MAX_FDT_SZ);
	if (fdt_rd < len) {
		rc = VMM_EIO;
		goto fail_freedata;
	}

	rc = libfdt_parse_fileinfo((virtual_addr_t)fdt_data, &fdt);
	if (rc) {
		goto fail_freedata;
	}

	root = NULL;
	rc = libfdt_parse_devtree(&fdt, &root, devtree_root_name, parent);
	if (rc) {
		goto fail_freedata;
	}

	anode = vmm_devtree_getchild(root, VMM_DEVTREE_ALIASES_NODE_NAME);

	for (a = 0; a < aliasc; a++) {
		if (!anode) {
			vmm_cprintf(cdev, "Error: %s node not available\n",
				    VMM_DEVTREE_ALIASES_NODE_NAME);
			continue;
		}

		astr = aliasv[a];

		aname = astr;
		while (*astr != '\0' && *astr != ',') {
			astr++;
		}
		if (*astr == ',') {
			*astr = '\0';
			astr++;
		}

		if (*astr == '\0') {
			continue;
		}
		aattr = astr;
		while (*astr != '\0' && *astr != ',') {
			astr++;
		}
		if (*astr == ',') {
			*astr = '\0';
			astr++;
		}

		if (*astr == '\0') {
			continue;
		}
		atype = astr;
		while (*astr != '\0' && *astr != ',') {
			astr++;
		}
		if (*astr == ',') {
			*astr = '\0';
			astr++;
		}

		if (*astr == '\0') {
			continue;
		}

		if (vmm_devtree_read_string(anode, aname, &apath)) {
			vmm_cprintf(cdev, "Error: Failed to read %s attribute "
				    "of %s node\n", aname,
				    VMM_DEVTREE_ALIASES_NODE_NAME);
			continue;
		}

		node = vmm_devtree_getchild(root, apath);
		if (!node) {
			vmm_cprintf(cdev, "Error: %s node not found under "
				    "%s/%s\n", apath, devtree_path,
				    devtree_root_name);
			continue;
		}

		if (!strcmp(atype, "unknown")) {
			val = NULL;
			val_len = 0;
			val_type = VMM_DEVTREE_MAX_ATTRTYPE;
		} else if (!strcmp(atype, "string")) {
			val_len = strlen(astr) + 1;
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			strcpy(val, astr);
			val_type = VMM_DEVTREE_ATTRTYPE_STRING;
		} else if (!strcmp(atype, "bytes")) {
			val_len = 1;
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((u8 *)val) = strtoul(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_BYTEARRAY;
		} else if (!strcmp(atype, "uint32")) {
			val_len = 4;
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((u32 *)val) = strtoul(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_UINT32;
		} else if (!strcmp(atype, "uint64")) {
			val_len = 8;
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((u64 *)val) = strtoull(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_UINT64;
		} else if (!strcmp(atype, "physaddr")) {
			val_len = sizeof(physical_addr_t);
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((physical_addr_t *)val) = strtoull(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_PHYSADDR;
		} else if (!strcmp(atype, "physsize")) {
			val_len = sizeof(physical_size_t);
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((physical_size_t *)val) = strtoull(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_PHYSSIZE;
		} else if (!strcmp(atype, "virtaddr")) {
			val_len = sizeof(virtual_addr_t);
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((virtual_addr_t *)val) = strtoull(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_VIRTADDR;
		} else if (!strcmp(atype, "virtsize")) {
			val_len = sizeof(virtual_size_t);
			val = vmm_zalloc(val_len);
			if (!val) {
				vmm_cprintf(cdev, "Error: vmm_zalloc(%d) "
					    "failed\n", val_len);
				goto next_iter;
			}
			*((virtual_size_t *)val) = strtoull(astr, NULL, 0);
			val_type = VMM_DEVTREE_ATTRTYPE_VIRTSIZE;
		} else {
			vmm_cprintf(cdev, "Error: Invalid attribute type %s\n",
				    atype);
			goto next_iter;
		}

		if (val && (val_len > 0)) {
			vmm_devtree_setattr(node, aattr, val,
					    val_type, val_len, FALSE);
			vmm_free(val);
		}

next_iter:
		vmm_devtree_dref_node(node);
	}

	vmm_devtree_dref_node(anode);

fail_freedata:
	vmm_free(fdt_data);
fail_closefd:
	vfs_close(fd);
fail:
	vmm_devtree_dref_node(parent);
	return rc;
}

static int cmd_vfs_guest_fdt_load(struct vmm_chardev *cdev,
				  const char *guest_name,
				  const char *path,
				  u32 vcpu_count,
				  int aliasc, char **aliasv)
{
	int rc;
	u32 i;
	char name[VMM_FIELD_SHORT_NAME_SIZE];
	struct vmm_devtree_node *guests_node, *guest_node;
	struct vmm_devtree_node *vcpu_tmpl_node, *vcpus_node, *vcpu_node;

	if (!vcpu_count) {
		vmm_cprintf(cdev, "VCPU count should be non-zero\n");
		return VMM_EINVALID;
	}

	guests_node = vmm_devtree_getnode(VMM_DEVTREE_PATH_SEPARATOR_STRING
					  VMM_DEVTREE_GUESTINFO_NODE_NAME);
	if (!guests_node) {
		vmm_cprintf(cdev, "%s DT node not found\n",
			    VMM_DEVTREE_GUESTINFO_NODE_NAME);
		return VMM_ENOTAVAIL;
	}

	rc = cmd_vfs_fdt_load(cdev, VMM_DEVTREE_PATH_SEPARATOR_STRING
			      VMM_DEVTREE_GUESTINFO_NODE_NAME, guest_name,
			      path, aliasc, aliasv);
	if (VMM_OK != rc) {
		vmm_cprintf(cdev, "Failed to load fdt (error %d)\n", rc);
		vmm_devtree_dref_node(guests_node);
		return rc;
	}

	guest_node = vmm_devtree_getchild(guests_node, guest_name);
	vmm_devtree_dref_node(guests_node);
	if (!guest_node) {
		vmm_cprintf(cdev, "Failed to get guest DT node\n");
		return VMM_ENOTAVAIL;
	}

	vcpu_tmpl_node = vmm_devtree_getchild(guest_node,
					  VMM_DEVTREE_VCPU_TEMPLATE_NODE_NAME);
	if (!vcpu_tmpl_node) {
		vmm_cprintf(cdev, "Failed to get vcpu template DT node\n");
		vmm_devtree_dref_node(guest_node);
		vmm_devtree_delnode(guest_node);
		return VMM_ENOTAVAIL;
	}

	vcpus_node = vmm_devtree_addnode(guest_node,
					 VMM_DEVTREE_VCPUS_NODE_NAME);
	if (!vcpus_node) {
		return VMM_EFAIL;
		vmm_cprintf(cdev, "Failed to add vcpus DT node\n");
		vmm_devtree_dref_node(vcpu_tmpl_node);
		vmm_devtree_dref_node(guest_node);
		vmm_devtree_delnode(guest_node);
	}

	for (i = 0; i < vcpu_count; i++) {
		vmm_snprintf(name, sizeof(name), "vcpu%d", i);
		rc  = vmm_devtree_copynode(vcpus_node, name, vcpu_tmpl_node);
		if (rc) {
			vmm_cprintf(cdev, "Failed to add %s DT node\n", name);
			vmm_devtree_dref_node(vcpu_tmpl_node);
			vmm_devtree_dref_node(guest_node);
			vmm_devtree_delnode(guest_node);
			return rc;
		}

		if (i) {
			continue;
		}

		vcpu_node = vmm_devtree_getchild(vcpus_node, name);
		if (!vcpu_node) {
			vmm_cprintf(cdev, "Failed to find %s DT node\n", name);
			vmm_devtree_dref_node(vcpu_tmpl_node);
			vmm_devtree_dref_node(guest_node);
			vmm_devtree_delnode(guest_node);
			return rc;
		}

		/* Remove "poweroff" DT prop (if present) for first VCPU */
		vmm_devtree_delattr(vcpu_node,
				    VMM_DEVTREE_VCPU_POWEROFF_ATTR_NAME);
		vmm_devtree_dref_node(vcpu_node);
	}

	vmm_devtree_dref_node(vcpu_tmpl_node);
	vmm_devtree_dref_node(guest_node);

	return VMM_OK;
}

static int cmd_vfs_load(struct vmm_chardev *cdev,
			struct vmm_guest *guest,
			physical_addr_t pa,
			const char *curdir,
			const char *path, u32 off, u32 len)
{
	int fd, rc;
	loff_t rd_off;
	physical_addr_t wr_pa;
	size_t buf_wr, buf_rd, buf_count, wr_count;
	char *buf = NULL;

	rc = cmd_vfs_file_open_read(cdev, curdir, path, &fd, &len);
	if (VMM_OK != rc) {
		return rc;
	}

	if (off >= len) {
		vfs_close(fd);
		vmm_cprintf(cdev, "Offset greater than file size\n");
		return VMM_EINVALID;
	}

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		vfs_close(fd);
		return VMM_ENOMEM;
	}

	len = ((len - off) < len) ? (len - off) : len;

	rd_off = 0;
	wr_count = 0;
	wr_pa = pa;
	while (len) {
		buf_rd = (len < VFS_LOAD_BUF_SZ) ? len : VFS_LOAD_BUF_SZ;
		buf_count = vfs_read(fd, buf, buf_rd);
		if (buf_count < 1) {
			vmm_cprintf(cdev, "Failed to read "
					  "%zu bytes @ 0x%llx from %s\n",
					   buf_rd, rd_off, path);
			break;
		}
		rd_off += buf_count;
		if (guest) {
			buf_wr = vmm_guest_memory_write(guest, wr_pa,
							buf, buf_count, FALSE);
		} else {
			buf_wr = vmm_host_memory_write(wr_pa,
						       buf, buf_count, FALSE);
		}
		if (buf_wr != buf_count) {
			vmm_cprintf(cdev, "Failed to write "
					  "%zu bytes @ 0x%"PRIPADDR" (%s)\n",
					  buf_count, wr_pa,
					  (guest) ? (guest->name) : "host");
			break;
		}
		len -= buf_wr;
		wr_count += buf_wr;
		wr_pa += buf_wr;
	}

	vmm_cprintf(cdev, "%s: Loaded 0x%"PRIPADDR" with %zu bytes\n",
			  (guest) ? (guest->name) : "host",
			  pa, wr_count);

	vmm_free(buf);
	rc = vfs_close(fd);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}

static const char cmd_vfs_esclist[] = {'\n', '\r', ' '};

static int cmd_vfs_in_esclist(char c)
{
	u32 escpos = 0;

	for (escpos = 0; escpos < sizeof (cmd_vfs_esclist); ++escpos) {
		if (cmd_vfs_esclist[escpos] == c) {
			return 1;
		}
	}
	return 0;
}

static char *cmd_vfs_next_token(char **bufp, u32 *len)
{
	u32 pos = 0;
	char *buf = *bufp;
	char *token = NULL;

	while ((pos < *len) && cmd_vfs_in_esclist(buf[pos])) {
		++pos;
	}
	token = buf;

	while ((pos < *len) && !cmd_vfs_in_esclist(buf[pos])) {
		if (!vmm_isprintable(buf[pos])) {
			*len -= pos;
			return NULL;
		}
		++pos;
	}
	buf[pos++] = '\0';

	if (pos < *len) {
		*bufp = buf + pos;
		*len -= pos;
	} else {
		*len = 0;
		*bufp = NULL;
	}

	return token;
}

static int cmd_vfs_load_list(struct vmm_chardev *cdev,
			     struct vmm_guest *guest,
			     const char *path)
{
	u32 len;
	int fd, rc;
	physical_addr_t pa;
	char *buf = NULL;
	char *buf_save = NULL;
	char *token = NULL;
	char *curdir = NULL;

	if (NULL == (curdir = vmm_zalloc(VFS_MAX_PATH))) {
		vmm_cprintf(cdev, "Failed to alloc current dir buffer\n");
		return VMM_ENOMEM;
	}

	rc = cmd_vfs_file_parent_dir(cdev, path, curdir, VFS_MAX_PATH);
	if (VMM_OK != rc) {
		vmm_free(curdir);
		vmm_cprintf(cdev, "Failed to find parent dir of %s\n", path);
		return rc;
	}

	rc = cmd_vfs_file_open_read(cdev, NULL, path, &fd, &len);
	if (VMM_OK != rc) {
		vmm_free(curdir);
		vmm_cprintf(cdev, "Failed to open %s\n", path);
		return rc;
	}

	if (VFS_LOAD_BUF_SZ <= len) {
		vmm_cprintf(cdev, "List file exceeds limit of %d chars\n",
			    VFS_LOAD_BUF_SZ);
	}

	vmm_cprintf(cdev, "%s: Parsing %s\n",
		    (guest) ? (guest->name) : "host", path);

	if (NULL == (buf = vmm_malloc(VFS_LOAD_BUF_SZ))) {
		vfs_close(fd);
		vmm_free(curdir);
		vmm_cprintf(cdev, "Failed to allocate buffer\n");
		return VMM_ENOMEM;
	}
	buf_save = buf;

	len = cmd_vfs_file_buf_read(fd, buf, len);
	if (len < 0) {
		vmm_free(buf_save);
		vfs_close(fd);
		vmm_free(curdir);
		vmm_cprintf(cdev, "Failed to read %s, error %d\n", path, len);
		return len;
	}

	while (buf) {
		token = cmd_vfs_next_token(&buf, &len);
		if (!token || !buf) {
			vmm_cprintf(cdev, "Failed to read address\n");
			break;
		}
		pa = (physical_addr_t)strtoull(token, NULL, 0);

		token = cmd_vfs_next_token(&buf, &len);
		vmm_cprintf(cdev, "%s: Loading 0x%"PRIPADDR" with file %s\n",
			    (guest) ? (guest->name) : "host",
			    pa, token);

		rc = cmd_vfs_load(cdev, guest, pa,
				  curdir, token, 0, 0xFFFFFFFF);
		if (rc) {
			vmm_cprintf(cdev, "error %d\n", rc);
			break;
		}
	}

	vmm_free(buf_save);
	rc = vfs_close(fd);
	vmm_free(curdir);
	if (rc) {
		vmm_cprintf(cdev, "Failed to close %s\n", path);
		return rc;
	}

	return VMM_OK;
}

static int cmd_vfs_exec(struct vmm_chardev *cdev, int argc, char **argv)
{
	u32 off, len;
	physical_addr_t pa;
	struct vmm_guest *guest;
	if (argc < 2) {
		cmd_vfs_usage(cdev);
		return VMM_EFAIL;
	}
	if (strcmp(argv[1], "help") == 0) {
		cmd_vfs_usage(cdev);
		return VMM_OK;
	} else if ((strcmp(argv[1], "fslist") == 0) && (argc == 2)) {
		return cmd_vfs_fslist(cdev);
	} else if ((strcmp(argv[1], "mplist") == 0) && (argc == 2)) {
		return cmd_vfs_mplist(cdev);
	} else if (strcmp(argv[1], "mount") == 0) {
		if (argc == 4) {
			return cmd_vfs_mount(cdev, argv[2], argv[3], NULL);
		} else if (argc == 5) {
			long wait = strtol(argv[4], NULL, 10);
			return cmd_vfs_mount(cdev, argv[2], argv[3], &wait);
		}
	} else if ((strcmp(argv[1], "umount") == 0) && (argc == 3)) {
		return cmd_vfs_umount(cdev, argv[2]);
	} else if ((strcmp(argv[1], "ls") == 0) && (argc == 3)) {
		return cmd_vfs_ls(cdev, argv[2]);
	} else if ((strcmp(argv[1], "run") == 0) && (argc == 3)) {
		return cmd_vfs_run(cdev, argv[2]);
#if CONFIG_CRYPTO_HASH_MD5
	} else if ((strcmp(argv[1], "md5") == 0) && (argc == 3)) {
		return cmd_vfs_md5(cdev, argv[2]);
#endif
#if CONFIG_CRYPTO_HASH_SHA256
	} else if ((strcmp(argv[1], "sha256") == 0) && (argc == 3)) {
		return cmd_vfs_sha256(cdev, argv[2]);
#endif
	} else if ((strcmp(argv[1], "cat") == 0) && (argc == 3)) {
		return cmd_vfs_cat(cdev, argv[2]);
	} else if ((strcmp(argv[1], "mv") == 0) && (argc == 4)) {
		return cmd_vfs_mv(cdev, argv[2], argv[3]);
	} else if ((strcmp(argv[1], "rm") == 0) && (argc == 3)) {
		return cmd_vfs_rm(cdev, argv[2]);
	} else if ((strcmp(argv[1], "mkdir") == 0) && (argc == 3)) {
		return cmd_vfs_mkdir(cdev, argv[2]);
	} else if ((strcmp(argv[1], "rmdir") == 0) && (argc == 3)) {
		return cmd_vfs_rmdir(cdev, argv[2]);
	} else if ((strcmp(argv[1], "module_load") == 0) && (argc == 3)) {
		return cmd_vfs_module_load(cdev, argv[2]);
	} else if ((strcmp(argv[1], "fdt_load") == 0) && (argc >= 5)) {
		return cmd_vfs_fdt_load(cdev, argv[2], argv[3], argv[4],
					argc-5, (argc-5) ? &argv[5] : NULL);
	} else if ((strcmp(argv[1], "guest_fdt_load") == 0) && (argc >= 5)) {
		off = (argc > 4) ? strtoul(argv[4], NULL, 0) : 0;
		return cmd_vfs_guest_fdt_load(cdev, argv[2], argv[3], off,
					argc-5, (argc-5) ? &argv[5] : NULL);
	} else if ((strcmp(argv[1], "host_load") == 0) && (argc > 3)) {
		pa = (physical_addr_t)strtoull(argv[2], NULL, 0);
		off = (argc > 4) ? strtoul(argv[4], NULL, 0) : 0;
		len = (argc > 5) ? strtoul(argv[5], NULL, 0) : 0xFFFFFFFF;
		return cmd_vfs_load(cdev, NULL, pa, NULL, argv[3], off, len);
	} else if ((strcmp(argv[1], "host_load_list") == 0) && (argc == 3)) {
		return cmd_vfs_load_list(cdev, NULL, argv[2]);
	} else if ((strcmp(argv[1], "guest_load") == 0) && (argc > 4)) {
		guest = vmm_manager_guest_find(argv[2]);
		if (!guest) {
			vmm_cprintf(cdev, "Failed to find guest %s\n",
				    argv[2]);
			return VMM_ENOTAVAIL;
		}
		pa = (physical_addr_t)strtoull(argv[3], NULL, 0);
		off = (argc > 5) ? strtoul(argv[5], NULL, 0) : 0;
		len = (argc > 6) ? strtoul(argv[6], NULL, 0) : 0xFFFFFFFF;
		return cmd_vfs_load(cdev, guest, pa, NULL, argv[4], off, len);
	} else if ((strcmp(argv[1], "guest_load_list") == 0) && (argc == 4)) {
		guest = vmm_manager_guest_find(argv[2]);
		if (!guest) {
			vmm_cprintf(cdev, "Failed to find guest %s\n",
				    argv[2]);
			return VMM_ENOTAVAIL;
		}
		return cmd_vfs_load_list(cdev, guest, argv[3]);
	}
	cmd_vfs_usage(cdev);
	return VMM_EFAIL;
}

static struct vmm_cmd cmd_vfs = {
	.name = "vfs",
	.desc = "vfs related commands",
	.usage = cmd_vfs_usage,
	.exec = cmd_vfs_exec,
};

static int __init cmd_vfs_init(void)
{
	return vmm_cmdmgr_register_cmd(&cmd_vfs);
}

static void __exit cmd_vfs_exit(void)
{
	vmm_cmdmgr_unregister_cmd(&cmd_vfs);
}

VMM_DECLARE_MODULE(MODULE_DESC,
			MODULE_AUTHOR,
			MODULE_LICENSE,
			MODULE_IPRIORITY,
			MODULE_INIT,
			MODULE_EXIT);
