/*
 * Copyright (c) 2011-2014 The DragonFly Project.  All rights reserved.
 *
 * This code is derived from software contributed to The DragonFly Project
 * by Matthew Dillon <dillon@dragonflybsd.org>
 * by Venkatesh Srinivas <vsrinivas@dragonflybsd.org>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name of The DragonFly Project nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific, prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */
/*
 * Ioctl Functions.
 *
 * WARNING! The ioctl functions which manipulate the connection state need
 *	    to be able to run without deadlock on the volume's chain lock.
 *	    Most of these functions use a separate lock.
 */

#include "hammer2.h"

static int hammer2_ioctl_version_get(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_recluster(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_remote_scan(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_remote_add(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_remote_del(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_remote_rep(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_socket_get(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_socket_set(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_pfs_get(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_pfs_lookup(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_pfs_create(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_pfs_snapshot(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_pfs_delete(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_inode_get(hammer2_inode_t *ip, void *data);
static int hammer2_ioctl_inode_set(hammer2_inode_t *ip, void *data);
//static int hammer2_ioctl_inode_comp_set(hammer2_inode_t *ip, void *data);
//static int hammer2_ioctl_inode_comp_rec_set(hammer2_inode_t *ip, void *data);
//static int hammer2_ioctl_inode_comp_rec_set2(hammer2_inode_t *ip, void *data);

int
hammer2_ioctl(hammer2_inode_t *ip, u_long com, void *data, int fflag,
	      struct ucred *cred)
{
	int error;

	/*
	 * Standard root cred checks, will be selectively ignored below
	 * for ioctls that do not require root creds.
	 */
	error = priv_check_cred(cred, PRIV_HAMMER_IOCTL, 0);

	switch(com) {
	case HAMMER2IOC_VERSION_GET:
		error = hammer2_ioctl_version_get(ip, data);
		break;
	case HAMMER2IOC_RECLUSTER:
		if (error == 0)
			error = hammer2_ioctl_recluster(ip, data);
		break;
	case HAMMER2IOC_REMOTE_SCAN:
		if (error == 0)
			error = hammer2_ioctl_remote_scan(ip, data);
		break;
	case HAMMER2IOC_REMOTE_ADD:
		if (error == 0)
			error = hammer2_ioctl_remote_add(ip, data);
		break;
	case HAMMER2IOC_REMOTE_DEL:
		if (error == 0)
			error = hammer2_ioctl_remote_del(ip, data);
		break;
	case HAMMER2IOC_REMOTE_REP:
		if (error == 0)
			error = hammer2_ioctl_remote_rep(ip, data);
		break;
	case HAMMER2IOC_SOCKET_GET:
		if (error == 0)
			error = hammer2_ioctl_socket_get(ip, data);
		break;
	case HAMMER2IOC_SOCKET_SET:
		if (error == 0)
			error = hammer2_ioctl_socket_set(ip, data);
		break;
	case HAMMER2IOC_PFS_GET:
		if (error == 0)
			error = hammer2_ioctl_pfs_get(ip, data);
		break;
	case HAMMER2IOC_PFS_LOOKUP:
		if (error == 0)
			error = hammer2_ioctl_pfs_lookup(ip, data);
		break;
	case HAMMER2IOC_PFS_CREATE:
		if (error == 0)
			error = hammer2_ioctl_pfs_create(ip, data);
		break;
	case HAMMER2IOC_PFS_DELETE:
		if (error == 0)
			error = hammer2_ioctl_pfs_delete(ip, data);
		break;
	case HAMMER2IOC_PFS_SNAPSHOT:
		if (error == 0)
			error = hammer2_ioctl_pfs_snapshot(ip, data);
		break;
	case HAMMER2IOC_INODE_GET:
		error = hammer2_ioctl_inode_get(ip, data);
		break;
	case HAMMER2IOC_INODE_SET:
		if (error == 0)
			error = hammer2_ioctl_inode_set(ip, data);
		break;
	/*case HAMMER2IOC_INODE_COMP_SET:
		error = hammer2_ioctl_inode_comp_set(ip, data);
		break;
	case HAMMER2IOC_INODE_COMP_REC_SET:
	 	error = hammer2_ioctl_inode_comp_rec_set(ip, data);
	 	break;
	case HAMMER2IOC_INODE_COMP_REC_SET2:
		error = hammer2_ioctl_inode_comp_rec_set2(ip, data);
		break;*/
	default:
		error = EOPNOTSUPP;
		break;
	}
	return (error);
}

/*
 * Retrieve version and basic info
 */
static int
hammer2_ioctl_version_get(hammer2_inode_t *ip, void *data)
{
	hammer2_mount_t *hmp = ip->pmp->iroot->cluster.focus->hmp;
	hammer2_ioc_version_t *version = data;

	version->version = hmp->voldata.version;
	return 0;
}

static int
hammer2_ioctl_recluster(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_recluster_t *recl = data;
	struct file *fp;

	fp = holdfp(curproc->p_fd, recl->fd, -1);
	if (fp) {
		kprintf("reconnect to cluster\n");
		hammer2_cluster_reconnect(ip->pmp, fp);
		return 0;
	} else {
		return EINVAL;
	}
}

/*
 * Retrieve information about a remote
 */
static int
hammer2_ioctl_remote_scan(hammer2_inode_t *ip, void *data)
{
	hammer2_mount_t *hmp = ip->pmp->iroot->cluster.focus->hmp;
	hammer2_ioc_remote_t *remote = data;
	int copyid = remote->copyid;

	if (copyid < 0 || copyid >= HAMMER2_COPYID_COUNT)
		return (EINVAL);

	hammer2_voldata_lock(hmp);
	remote->copy1 = hmp->voldata.copyinfo[copyid];
	hammer2_voldata_unlock(hmp);

	/*
	 * Adjust nextid (GET only)
	 */
	while (++copyid < HAMMER2_COPYID_COUNT &&
	       hmp->voldata.copyinfo[copyid].copyid == 0) {
		;
	}
	if (copyid == HAMMER2_COPYID_COUNT)
		remote->nextid = -1;
	else
		remote->nextid = copyid;

	return(0);
}

/*
 * Add new remote entry
 */
static int
hammer2_ioctl_remote_add(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_remote_t *remote = data;
	hammer2_pfsmount_t *pmp = ip->pmp;
	hammer2_mount_t *hmp;
	int copyid = remote->copyid;
	int error = 0;

	if (copyid >= HAMMER2_COPYID_COUNT)
		return (EINVAL);

	hmp = pmp->iroot->cluster.focus->hmp; /* XXX */
	hammer2_voldata_lock(hmp);
	if (copyid < 0) {
		for (copyid = 1; copyid < HAMMER2_COPYID_COUNT; ++copyid) {
			if (hmp->voldata.copyinfo[copyid].copyid == 0)
				break;
		}
		if (copyid == HAMMER2_COPYID_COUNT) {
			error = ENOSPC;
			goto failed;
		}
	}
	hammer2_voldata_modify(hmp);
	remote->copy1.copyid = copyid;
	hmp->voldata.copyinfo[copyid] = remote->copy1;
	hammer2_volconf_update(pmp, copyid);
failed:
	hammer2_voldata_unlock(hmp);
	return (error);
}

/*
 * Delete existing remote entry
 */
static int
hammer2_ioctl_remote_del(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_remote_t *remote = data;
	hammer2_pfsmount_t *pmp = ip->pmp;
	hammer2_mount_t *hmp;
	int copyid = remote->copyid;
	int error = 0;

	hmp = pmp->iroot->cluster.focus->hmp; /* XXX */
	if (copyid >= HAMMER2_COPYID_COUNT)
		return (EINVAL);
	remote->copy1.path[sizeof(remote->copy1.path) - 1] = 0;
	hammer2_voldata_lock(hmp);
	if (copyid < 0) {
		for (copyid = 1; copyid < HAMMER2_COPYID_COUNT; ++copyid) {
			if (hmp->voldata.copyinfo[copyid].copyid == 0)
				continue;
			if (strcmp(remote->copy1.path,
			    hmp->voldata.copyinfo[copyid].path) == 0) {
				break;
			}
		}
		if (copyid == HAMMER2_COPYID_COUNT) {
			error = ENOENT;
			goto failed;
		}
	}
	hammer2_voldata_modify(hmp);
	hmp->voldata.copyinfo[copyid].copyid = 0;
	hammer2_volconf_update(pmp, copyid);
failed:
	hammer2_voldata_unlock(hmp);
	return (error);
}

/*
 * Replace existing remote entry
 */
static int
hammer2_ioctl_remote_rep(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_remote_t *remote = data;
	hammer2_mount_t *hmp;
	int copyid = remote->copyid;

	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */

	if (copyid < 0 || copyid >= HAMMER2_COPYID_COUNT)
		return (EINVAL);

	hammer2_voldata_lock(hmp);
	hammer2_voldata_modify(hmp);
	/*hammer2_volconf_update(pmp, copyid);*/
	hammer2_voldata_unlock(hmp);

	return(0);
}

/*
 * Retrieve communications socket
 */
static int
hammer2_ioctl_socket_get(hammer2_inode_t *ip, void *data)
{
	return (EOPNOTSUPP);
}

/*
 * Set communications socket for connection
 */
static int
hammer2_ioctl_socket_set(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_remote_t *remote = data;
	hammer2_mount_t *hmp;
	int copyid = remote->copyid;

	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */
	if (copyid < 0 || copyid >= HAMMER2_COPYID_COUNT)
		return (EINVAL);

	hammer2_voldata_lock(hmp);
	hammer2_voldata_unlock(hmp);

	return(0);
}

/*
 * Used to scan and retrieve PFS information.  PFS's are directories under
 * the super-root.
 *
 * To scan PFSs pass name_key=0.  The function will scan for the next
 * PFS and set all fields, as well as set name_next to the next key.
 * When no PFSs remain, name_next is set to (hammer2_key_t)-1.
 *
 * To retrieve the PFS associated with the file descriptor, pass
 * name_key set to (hammer2_key_t)-1.
 */
static int
hammer2_ioctl_pfs_get(hammer2_inode_t *ip, void *data)
{
	const hammer2_inode_data_t *ipdata;
	hammer2_mount_t *hmp;
	hammer2_ioc_pfs_t *pfs;
	hammer2_cluster_t *cparent;
	hammer2_cluster_t *rcluster;
	hammer2_cluster_t *cluster;
	hammer2_key_t key_next;
	int error;
	int ddflag;

	error = 0;
	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */
	pfs = data;
	cparent = hammer2_inode_lock_ex(hmp->spmp->iroot);
	rcluster = hammer2_inode_lock_ex(ip->pmp->iroot);

	/*
	 * Search for the first key or specific key.  Remember that keys
	 * can be returned in any order.
	 */
	if (pfs->name_key == 0) {
		cluster = hammer2_cluster_lookup(cparent, &key_next,
						 0, (hammer2_key_t)-1,
						 0, &ddflag);
	} else if (pfs->name_key == (hammer2_key_t)-1) {
		ipdata = &hammer2_cluster_data(rcluster)->ipdata;
		cluster = hammer2_cluster_lookup(cparent, &key_next,
						 ipdata->name_key,
						 ipdata->name_key,
						 0, &ddflag);
		ipdata = NULL;	/* safety */
	} else {
		cluster = hammer2_cluster_lookup(cparent, &key_next,
						 pfs->name_key, pfs->name_key,
						 0, &ddflag);
	}
	hammer2_inode_unlock_ex(ip->pmp->iroot, rcluster);

	while (cluster &&
	       hammer2_cluster_type(cluster) != HAMMER2_BREF_TYPE_INODE) {
		cluster = hammer2_cluster_next(cparent, cluster, &key_next,
					       key_next, (hammer2_key_t)-1,
					       0);
	}
	if (cluster) {
		/*
		 * Load the data being returned by the ioctl.
		 */
		ipdata = &hammer2_cluster_data(cluster)->ipdata;
		pfs->name_key = ipdata->name_key;
		pfs->pfs_type = ipdata->pfs_type;
		pfs->pfs_clid = ipdata->pfs_clid;
		pfs->pfs_fsid = ipdata->pfs_fsid;
		KKASSERT(ipdata->name_len < sizeof(pfs->name));
		bcopy(ipdata->filename, pfs->name, ipdata->name_len);
		pfs->name[ipdata->name_len] = 0;
		ipdata = NULL;	/* safety */

		/*
		 * Calculate the next field
		 */
		do {
			cluster = hammer2_cluster_next(cparent, cluster,
						       &key_next,
						       0, (hammer2_key_t)-1,
						       0);
		} while (cluster &&
			 hammer2_cluster_type(cluster) !=
			  HAMMER2_BREF_TYPE_INODE);
		if (cluster) {
			ipdata = &hammer2_cluster_data(cluster)->ipdata;
			pfs->name_next = ipdata->name_key;
			hammer2_cluster_unlock(cluster);
		} else {
			pfs->name_next = (hammer2_key_t)-1;
		}
	} else {
		pfs->name_next = (hammer2_key_t)-1;
		error = ENOENT;
	}
	hammer2_inode_unlock_ex(hmp->spmp->iroot, cparent);

	return (error);
}

/*
 * Find a specific PFS by name
 */
static int
hammer2_ioctl_pfs_lookup(hammer2_inode_t *ip, void *data)
{
	const hammer2_inode_data_t *ipdata;
	hammer2_mount_t *hmp;
	hammer2_ioc_pfs_t *pfs;
	hammer2_cluster_t *cparent;
	hammer2_cluster_t *cluster;
	hammer2_key_t key_next;
	hammer2_key_t lhc;
	int error;
	int ddflag;
	size_t len;

	error = 0;
	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */
	pfs = data;
	cparent = hammer2_inode_lock_sh(hmp->spmp->iroot);

	pfs->name[sizeof(pfs->name) - 1] = 0;
	len = strlen(pfs->name);
	lhc = hammer2_dirhash(pfs->name, len);

	cluster = hammer2_cluster_lookup(cparent, &key_next,
					 lhc, lhc + HAMMER2_DIRHASH_LOMASK,
					 HAMMER2_LOOKUP_SHARED, &ddflag);
	while (cluster) {
		if (hammer2_cluster_type(cluster) == HAMMER2_BREF_TYPE_INODE) {
			ipdata = &hammer2_cluster_data(cluster)->ipdata;
			if (ipdata->name_len == len &&
			    bcmp(ipdata->filename, pfs->name, len) == 0) {
				break;
			}
			ipdata = NULL;	/* safety */
		}
		cluster = hammer2_cluster_next(cparent, cluster, &key_next,
					   key_next,
					   lhc + HAMMER2_DIRHASH_LOMASK,
					   HAMMER2_LOOKUP_SHARED);
	}

	/*
	 * Load the data being returned by the ioctl.
	 */
	if (cluster) {
		ipdata = &hammer2_cluster_data(cluster)->ipdata;
		pfs->name_key = ipdata->name_key;
		pfs->pfs_type = ipdata->pfs_type;
		pfs->pfs_clid = ipdata->pfs_clid;
		pfs->pfs_fsid = ipdata->pfs_fsid;
		ipdata = NULL;

		hammer2_cluster_unlock(cluster);
	} else {
		error = ENOENT;
	}
	hammer2_inode_unlock_sh(hmp->spmp->iroot, cparent);

	return (error);
}

/*
 * Create a new PFS under the super-root
 */
static int
hammer2_ioctl_pfs_create(hammer2_inode_t *ip, void *data)
{
	hammer2_inode_data_t *nipdata;
	hammer2_mount_t *hmp;
	hammer2_ioc_pfs_t *pfs;
	hammer2_inode_t *nip;
	hammer2_cluster_t *ncluster;
	hammer2_trans_t trans;
	int error;

	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */
	pfs = data;
	nip = NULL;

	if (pfs->name[0] == 0)
		return(EINVAL);
	pfs->name[sizeof(pfs->name) - 1] = 0;	/* ensure 0-termination */

	hammer2_trans_init(&trans, ip->pmp, HAMMER2_TRANS_NEWINODE);
	nip = hammer2_inode_create(&trans, hmp->spmp->iroot, NULL, NULL,
				     pfs->name, strlen(pfs->name),
				     &ncluster, &error);
	if (error == 0) {
		nipdata = hammer2_cluster_modify_ip(&trans, nip, ncluster,
						  HAMMER2_MODIFY_ASSERTNOCOPY);
		nipdata->pfs_type = pfs->pfs_type;
		nipdata->pfs_clid = pfs->pfs_clid;
		nipdata->pfs_fsid = pfs->pfs_fsid;

		/*
		 * Do not allow compression on PFS's with the special name
		 * "boot", the boot loader can't decompress (yet).
		 */
		if (strcmp(pfs->name, "boot") == 0)
			nipdata->comp_algo = HAMMER2_COMP_AUTOZERO;
		hammer2_cluster_modsync(ncluster);
		hammer2_inode_unlock_ex(nip, ncluster);
	}
	hammer2_trans_done(&trans);

	return (error);
}

/*
 * Destroy an existing PFS under the super-root
 */
static int
hammer2_ioctl_pfs_delete(hammer2_inode_t *ip, void *data)
{
	hammer2_mount_t *hmp;
	hammer2_ioc_pfs_t *pfs = data;
	hammer2_trans_t trans;
	int error;

	hmp = ip->pmp->iroot->cluster.focus->hmp; /* XXX */
	hammer2_trans_init(&trans, ip->pmp, 0);
	error = hammer2_unlink_file(&trans, hmp->spmp->iroot,
				    pfs->name, strlen(pfs->name),
				    2, NULL, NULL);
	hammer2_trans_done(&trans);

	return (error);
}

static int
hammer2_ioctl_pfs_snapshot(hammer2_inode_t *ip, void *data)
{
	hammer2_ioc_pfs_t *pfs = data;
	hammer2_trans_t trans;
	hammer2_cluster_t *cparent;
	int error;

	if (pfs->name[0] == 0)
		return(EINVAL);
	if (pfs->name[sizeof(pfs->name)-1] != 0)
		return(EINVAL);

	hammer2_vfs_sync(ip->pmp->mp, MNT_WAIT);

	hammer2_trans_init(&trans, ip->pmp, HAMMER2_TRANS_NEWINODE);
	cparent = hammer2_inode_lock_ex(ip);
	error = hammer2_cluster_snapshot(&trans, cparent, pfs);
	hammer2_inode_unlock_ex(ip, cparent);
	hammer2_trans_done(&trans);

	return (error);
}

/*
 * Retrieve the raw inode structure
 */
static int
hammer2_ioctl_inode_get(hammer2_inode_t *ip, void *data)
{
	const hammer2_inode_data_t *ipdata;
	hammer2_ioc_inode_t *ino;
	hammer2_cluster_t *cparent;

	ino = data;

	cparent = hammer2_inode_lock_sh(ip);
	ipdata = &hammer2_cluster_data(cparent)->ipdata;
	ino->ip_data = *ipdata;
	ino->kdata = ip;
	hammer2_inode_unlock_sh(ip, cparent);

	return (0);
}

/*
 * Set various parameters in an inode which cannot be set through
 * normal filesystem VNOPS.
 */
static int
hammer2_ioctl_inode_set(hammer2_inode_t *ip, void *data)
{
	const hammer2_inode_data_t *ripdata;
	hammer2_inode_data_t *wipdata;
	hammer2_ioc_inode_t *ino = data;
	hammer2_cluster_t *cparent;
	hammer2_trans_t trans;
	int error = 0;
	int dosync = 0;

	hammer2_trans_init(&trans, ip->pmp, 0);
	cparent = hammer2_inode_lock_ex(ip);
	ripdata = &hammer2_cluster_data(cparent)->ipdata;

	if (ino->ip_data.comp_algo != ripdata->comp_algo) {
		wipdata = hammer2_cluster_modify_ip(&trans, ip, cparent, 0);
		wipdata->comp_algo = ino->ip_data.comp_algo;
		ripdata = wipdata; /* safety */
		dosync = 1;
	}
	ino->kdata = ip;
	
	/* Ignore these flags for now...*/
	if (ino->flags & HAMMER2IOC_INODE_FLAG_IQUOTA) {
	}
	if (ino->flags & HAMMER2IOC_INODE_FLAG_DQUOTA) {
	}
	if (ino->flags & HAMMER2IOC_INODE_FLAG_COPIES) {
	}
	if (dosync)
		hammer2_cluster_modsync(cparent);
	hammer2_trans_done(&trans);
	hammer2_inode_unlock_ex(ip, cparent);

	return (error);
}