fs/nilfs2/ioctl.c - LeafOS-Devices/android_kernel_samsung_gta4xl - Gitiles

 /*
  * ioctl.c - NILFS ioctl operations.
  *
  * Copyright (C) 2007, 2008 Nippon Telegraph and Telephone Corporation.
  *
  * 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  *
  * Written by Koji Sato <koji@osrg.net>.
  */

 #include <linux/fs.h>
 #include <linux/wait.h>
 #include <linux/smp_lock.h>	/* lock_kernel(), unlock_kernel() */
 #include <linux/capability.h>	/* capable() */
 #include <linux/uaccess.h>	/* copy_from_user(), copy_to_user() */
 #include <linux/nilfs2_fs.h>
 #include "nilfs.h"
 #include "segment.h"
 #include "bmap.h"
 #include "cpfile.h"
 #include "sufile.h"
 #include "dat.h"


 static int nilfs_ioctl_wrap_copy(struct the_nilfs *nilfs,
 				 struct nilfs_argv *argv, int dir,
 				 ssize_t (*dofunc)(struct the_nilfs *,
 						   __u64 *, int,
 						   void *, size_t, size_t))
 {
 	void *buf;
 	void __user *base = (void __user *)(unsigned long)argv->v_base;
 	size_t maxmembs, total, n;
 	ssize_t nr;
 	int ret, i;
 	__u64 pos, ppos;

 	if (argv->v_nmembs == 0)
 		return 0;

 	if (argv->v_size > PAGE_SIZE)
 		return -EINVAL;

 	buf = (void *)__get_free_pages(GFP_NOFS, 0);
 	if (unlikely(!buf))
 		return -ENOMEM;
 	maxmembs = PAGE_SIZE / argv->v_size;

 	ret = 0;
 	total = 0;
 	pos = argv->v_index;
 	for (i = 0; i < argv->v_nmembs; i += n) {
 		n = (argv->v_nmembs - i < maxmembs) ?
 			argv->v_nmembs - i : maxmembs;
 		if ((dir & _IOC_WRITE) &&
 		    copy_from_user(buf, base + argv->v_size * i,
 				   argv->v_size * n)) {
 			ret = -EFAULT;
 			break;
 		}
 		ppos = pos;
 		nr = dofunc(nilfs, &pos, argv->v_flags, buf, argv->v_size,
 			       n);
 		if (nr < 0) {
 			ret = nr;
 			break;
 		}
 		if ((dir & _IOC_READ) &&
 		    copy_to_user(base + argv->v_size * i, buf,
 				 argv->v_size * nr)) {
 			ret = -EFAULT;
 			break;
 		}
 		total += nr;
 		if ((size_t)nr < n)
 			break;
 		if (pos == ppos)
 			pos += n;
 	}
 	argv->v_nmembs = total;

 	free_pages((unsigned long)buf, 0);
 	return ret;
 }

 static int nilfs_ioctl_change_cpmode(struct inode *inode, struct file *filp,
 				     unsigned int cmd, void __user *argp)
 {
 	struct inode *cpfile = NILFS_SB(inode->i_sb)->s_nilfs->ns_cpfile;
 	struct nilfs_transaction_info ti;
 	struct nilfs_cpmode cpmode;
 	int ret;

 	if (!capable(CAP_SYS_ADMIN))
 		return -EPERM;
 	if (copy_from_user(&cpmode, argp, sizeof(cpmode)))
 		return -EFAULT;

 	nilfs_transaction_begin(inode->i_sb, &ti, 0);
 	ret = nilfs_cpfile_change_cpmode(
 		cpfile, cpmode.cm_cno, cpmode.cm_mode);
 	if (unlikely(ret < 0)) {
 		nilfs_transaction_abort(inode->i_sb);
 		return ret;
 	}
 	nilfs_transaction_commit(inode->i_sb); /* never fails */
 	return ret;
 }

 static int
 nilfs_ioctl_delete_checkpoint(struct inode *inode, struct file *filp,
 			      unsigned int cmd, void __user *argp)
 {
 	struct inode *cpfile = NILFS_SB(inode->i_sb)->s_nilfs->ns_cpfile;
 	struct nilfs_transaction_info ti;
 	__u64 cno;
 	int ret;

 	if (!capable(CAP_SYS_ADMIN))
 		return -EPERM;
 	if (copy_from_user(&cno, argp, sizeof(cno)))
 		return -EFAULT;

 	nilfs_transaction_begin(inode->i_sb, &ti, 0);
 	ret = nilfs_cpfile_delete_checkpoint(cpfile, cno);
 	if (unlikely(ret < 0)) {
 		nilfs_transaction_abort(inode->i_sb);
 		return ret;
 	}
 	nilfs_transaction_commit(inode->i_sb); /* never fails */
 	return ret;
 }

 static ssize_t
 nilfs_ioctl_do_get_cpinfo(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			  void *buf, size_t size, size_t nmembs)
 {
 	return nilfs_cpfile_get_cpinfo(nilfs->ns_cpfile, posp, flags, buf,
 				       nmembs);
 }

 static int nilfs_ioctl_get_cpinfo(struct inode *inode, struct file *filp,
 				  unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_argv argv;
 	int ret;

 	if (copy_from_user(&argv, argp, sizeof(argv)))
 		return -EFAULT;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
 				    nilfs_ioctl_do_get_cpinfo);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &argv, sizeof(argv)))
 		ret = -EFAULT;
 	return ret;
 }

 static int nilfs_ioctl_get_cpstat(struct inode *inode, struct file *filp,
 				  unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_cpstat cpstat;
 	int ret;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_cpfile_get_stat(nilfs->ns_cpfile, &cpstat);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &cpstat, sizeof(cpstat)))
 		ret = -EFAULT;
 	return ret;
 }

 static ssize_t
 nilfs_ioctl_do_get_suinfo(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			  void *buf, size_t size, size_t nmembs)
 {
 	return nilfs_sufile_get_suinfo(nilfs->ns_sufile, *posp, buf, nmembs);
 }

 static int nilfs_ioctl_get_suinfo(struct inode *inode, struct file *filp,
 				  unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_argv argv;
 	int ret;

 	if (copy_from_user(&argv, argp, sizeof(argv)))
 		return -EFAULT;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
 				    nilfs_ioctl_do_get_suinfo);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &argv, sizeof(argv)))
 		ret = -EFAULT;
 	return ret;
 }

 static int nilfs_ioctl_get_sustat(struct inode *inode, struct file *filp,
 				  unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_sustat sustat;
 	int ret;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_sufile_get_stat(nilfs->ns_sufile, &sustat);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &sustat, sizeof(sustat)))
 		ret = -EFAULT;
 	return ret;
 }

 static ssize_t
 nilfs_ioctl_do_get_vinfo(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			 void *buf, size_t size, size_t nmembs)
 {
 	return nilfs_dat_get_vinfo(nilfs_dat_inode(nilfs), buf, nmembs);
 }

 static int nilfs_ioctl_get_vinfo(struct inode *inode, struct file *filp,
 				 unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_argv argv;
 	int ret;

 	if (copy_from_user(&argv, argp, sizeof(argv)))
 		return -EFAULT;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
 				    nilfs_ioctl_do_get_vinfo);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &argv, sizeof(argv)))
 		ret = -EFAULT;
 	return ret;
 }

 static ssize_t
 nilfs_ioctl_do_get_bdescs(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			  void *buf, size_t size, size_t nmembs)
 {
 	struct inode *dat = nilfs_dat_inode(nilfs);
 	struct nilfs_bmap *bmap = NILFS_I(dat)->i_bmap;
 	struct nilfs_bdesc *bdescs = buf;
 	int ret, i;

 	for (i = 0; i < nmembs; i++) {
 		ret = nilfs_bmap_lookup_at_level(bmap,
 						 bdescs[i].bd_offset,
 						 bdescs[i].bd_level + 1,
 						 &bdescs[i].bd_blocknr);
 		if (ret < 0) {
 			if (ret != -ENOENT)
 				return ret;
 			bdescs[i].bd_blocknr = 0;
 		}
 	}
 	return nmembs;
 }

 static int nilfs_ioctl_get_bdescs(struct inode *inode, struct file *filp,
 				  unsigned int cmd, void __user *argp)
 {
 	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
 	struct nilfs_argv argv;
 	int ret;

 	if (copy_from_user(&argv, argp, sizeof(argv)))
 		return -EFAULT;

 	down_read(&nilfs->ns_segctor_sem);
 	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
 				    nilfs_ioctl_do_get_bdescs);
 	up_read(&nilfs->ns_segctor_sem);
 	if (ret < 0)
 		return ret;

 	if (copy_to_user(argp, &argv, sizeof(argv)))
 		ret = -EFAULT;
 	return ret;
 }

 static int nilfs_ioctl_move_inode_block(struct inode *inode,
 					struct nilfs_vdesc *vdesc,
 					struct list_head *buffers)
 {
 	struct buffer_head *bh;
 	int ret;

 	if (vdesc->vd_flags == 0)
 		ret = nilfs_gccache_submit_read_data(
 			inode, vdesc->vd_offset, vdesc->vd_blocknr,
 			vdesc->vd_vblocknr, &bh);
 	else
 		ret = nilfs_gccache_submit_read_node(
 			inode, vdesc->vd_blocknr, vdesc->vd_vblocknr, &bh);

 	if (unlikely(ret < 0)) {
 		if (ret == -ENOENT)
 			printk(KERN_CRIT
 			       "%s: invalid virtual block address (%s): "
 			       "ino=%llu, cno=%llu, offset=%llu, "
 			       "blocknr=%llu, vblocknr=%llu\n",
 			       __func__, vdesc->vd_flags ? "node" : "data",
 			       (unsigned long long)vdesc->vd_ino,
 			       (unsigned long long)vdesc->vd_cno,
 			       (unsigned long long)vdesc->vd_offset,
 			       (unsigned long long)vdesc->vd_blocknr,
 			       (unsigned long long)vdesc->vd_vblocknr);
 		return ret;
 	}
 	bh->b_private = vdesc;
 	list_add_tail(&bh->b_assoc_buffers, buffers);
 	return 0;
 }

 static ssize_t
 nilfs_ioctl_do_move_blocks(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			   void *buf, size_t size, size_t nmembs)
 {
 	struct inode *inode;
 	struct nilfs_vdesc *vdesc;
 	struct buffer_head *bh, *n;
 	LIST_HEAD(buffers);
 	ino_t ino;
 	__u64 cno;
 	int i, ret;

 	for (i = 0, vdesc = buf; i < nmembs; ) {
 		ino = vdesc->vd_ino;
 		cno = vdesc->vd_cno;
 		inode = nilfs_gc_iget(nilfs, ino, cno);
 		if (unlikely(inode == NULL)) {
 			ret = -ENOMEM;
 			goto failed;
 		}
 		do {
 			ret = nilfs_ioctl_move_inode_block(inode, vdesc,
 							   &buffers);
 			if (unlikely(ret < 0))
 				goto failed;
 			vdesc++;
 		} while (++i < nmembs &&
 			 vdesc->vd_ino == ino && vdesc->vd_cno == cno);
 	}

 	list_for_each_entry_safe(bh, n, &buffers, b_assoc_buffers) {
 		ret = nilfs_gccache_wait_and_mark_dirty(bh);
 		if (unlikely(ret < 0)) {
 			if (ret == -EEXIST) {
 				vdesc = bh->b_private;
 				printk(KERN_CRIT
 				       "%s: conflicting %s buffer: "
 				       "ino=%llu, cno=%llu, offset=%llu, "
 				       "blocknr=%llu, vblocknr=%llu\n",
 				       __func__,
 				       vdesc->vd_flags ? "node" : "data",
 				       (unsigned long long)vdesc->vd_ino,
 				       (unsigned long long)vdesc->vd_cno,
 				       (unsigned long long)vdesc->vd_offset,
 				       (unsigned long long)vdesc->vd_blocknr,
 				       (unsigned long long)vdesc->vd_vblocknr);
 			}
 			goto failed;
 		}
 		list_del_init(&bh->b_assoc_buffers);
 		bh->b_private = NULL;
 		brelse(bh);
 	}
 	return nmembs;

  failed:
 	list_for_each_entry_safe(bh, n, &buffers, b_assoc_buffers) {
 		list_del_init(&bh->b_assoc_buffers);
 		bh->b_private = NULL;
 		brelse(bh);
 	}
 	return ret;
 }

 static inline int nilfs_ioctl_move_blocks(struct the_nilfs *nilfs,
 					  struct nilfs_argv *argv,
 					  int dir)
 {
 	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
 				     nilfs_ioctl_do_move_blocks);
 }

 static ssize_t
 nilfs_ioctl_do_delete_checkpoints(struct the_nilfs *nilfs, __u64 *posp,
 				  int flags, void *buf, size_t size,
 				  size_t nmembs)
 {
 	struct inode *cpfile = nilfs->ns_cpfile;
 	struct nilfs_period *periods = buf;
 	int ret, i;

 	for (i = 0; i < nmembs; i++) {
 		ret = nilfs_cpfile_delete_checkpoints(
 			cpfile, periods[i].p_start, periods[i].p_end);
 		if (ret < 0)
 			return ret;
 	}
 	return nmembs;
 }

 static inline int nilfs_ioctl_delete_checkpoints(struct the_nilfs *nilfs,
 						 struct nilfs_argv *argv,
 						 int dir)
 {
 	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
 				     nilfs_ioctl_do_delete_checkpoints);
 }

 static ssize_t
 nilfs_ioctl_do_free_vblocknrs(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			      void *buf, size_t size, size_t nmembs)
 {
 	int ret = nilfs_dat_freev(nilfs_dat_inode(nilfs), buf, nmembs);

 	return (ret < 0) ? ret : nmembs;
 }

 static inline int nilfs_ioctl_free_vblocknrs(struct the_nilfs *nilfs,
 					     struct nilfs_argv *argv,
 					     int dir)
 {
 	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
 				     nilfs_ioctl_do_free_vblocknrs);
 }

 static ssize_t
 nilfs_ioctl_do_mark_blocks_dirty(struct the_nilfs *nilfs, __u64 *posp,
 				 int flags, void *buf, size_t size,
 				 size_t nmembs)
 {
 	struct inode *dat = nilfs_dat_inode(nilfs);
 	struct nilfs_bmap *bmap = NILFS_I(dat)->i_bmap;
 	struct nilfs_bdesc *bdescs = buf;
 	int ret, i;

 	for (i = 0; i < nmembs; i++) {
 		/* XXX: use macro or inline func to check liveness */
 		ret = nilfs_bmap_lookup_at_level(bmap,
 						 bdescs[i].bd_offset,
 						 bdescs[i].bd_level + 1,
 						 &bdescs[i].bd_blocknr);
 		if (ret < 0) {
 			if (ret != -ENOENT)
 				return ret;
 			bdescs[i].bd_blocknr = 0;
 		}
 		if (bdescs[i].bd_blocknr != bdescs[i].bd_oblocknr)
 			/* skip dead block */
 			continue;
 		if (bdescs[i].bd_level == 0) {
 			ret = nilfs_mdt_mark_block_dirty(dat,
 							 bdescs[i].bd_offset);
 			if (ret < 0) {
 				WARN_ON(ret == -ENOENT);
 				return ret;
 			}
 		} else {
 			ret = nilfs_bmap_mark(bmap, bdescs[i].bd_offset,
 					      bdescs[i].bd_level);
 			if (ret < 0) {
 				WARN_ON(ret == -ENOENT);
 				return ret;
 			}
 		}
 	}
 	return nmembs;
 }

 static inline int nilfs_ioctl_mark_blocks_dirty(struct the_nilfs *nilfs,
 						struct nilfs_argv *argv,
 						int dir)
 {
 	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
 				     nilfs_ioctl_do_mark_blocks_dirty);
 }

 static ssize_t
 nilfs_ioctl_do_free_segments(struct the_nilfs *nilfs, __u64 *posp, int flags,
 			     void *buf, size_t size, size_t nmembs)
 {
 	struct nilfs_sb_info *sbi = nilfs_get_writer(nilfs);
 	int ret;

 	if (unlikely(!sbi))
 		return -EROFS;
 	ret = nilfs_segctor_add_segments_to_be_freed(
 		NILFS_SC(sbi), buf, nmembs);
 	nilfs_put_writer(nilfs);

 	return (ret < 0) ? ret : nmembs;
 }

 static inline int nilfs_ioctl_free_segments(struct the_nilfs *nilfs,
 					     struct nilfs_argv *argv,
 					     int dir)
 {
 	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
 				     nilfs_ioctl_do_free_segments);
 }

 int nilfs_ioctl_prepare_clean_segments(struct the_nilfs *nilfs,
 				       void __user *argp)
 {
 	struct nilfs_argv argv[5];
 	const char *msg;
 	int dir, ret;

 	if (copy_from_user(argv, argp, sizeof(argv)))
 		return -EFAULT;

 	dir = _IOC_WRITE;
 	ret = nilfs_ioctl_move_blocks(nilfs, &argv[0], dir);
 	if (ret < 0) {
 		msg = "cannot read source blocks";
 		goto failed;
 	}
 	ret = nilfs_ioctl_delete_checkpoints(nilfs, &argv[1], dir);
 	if (ret < 0) {
 		/*
 		 * can safely abort because checkpoints can be removed
 		 * independently.
 		 */
 		msg = "cannot delete checkpoints";
 		goto failed;
 	}
 	ret = nilfs_ioctl_free_vblocknrs(nilfs, &argv[2], dir);
 	if (ret < 0) {
 		/*
 		 * can safely abort because DAT file is updated atomically
 		 * using a copy-on-write technique.
 		 */
 		msg = "cannot delete virtual blocks from DAT file";
 		goto failed;
 	}
 	ret = nilfs_ioctl_mark_blocks_dirty(nilfs, &argv[3], dir);
 	if (ret < 0) {
 		/*
 		 * can safely abort because the operation is nondestructive.
 		 */
 		msg = "cannot mark copying blocks dirty";
 		goto failed;
 	}
 	ret = nilfs_ioctl_free_segments(nilfs, &argv[4], dir);
 	if (ret < 0) {
 		/*
 		 * can safely abort because this operation is atomic.
 		 */
 		msg = "cannot set segments to be freed";
 		goto failed;
 	}
 	return 0;

  failed:
 	nilfs_remove_all_gcinode(nilfs);
 	printk(KERN_ERR "NILFS: GC failed during preparation: %s: err=%d\n",
 	       msg, ret);
 	return ret;
 }

 static int nilfs_ioctl_clean_segments(struct inode *inode, struct file *filp,
 				      unsigned int cmd, void __user *argp)
 {
 	if (!capable(CAP_SYS_ADMIN))
 		return -EPERM;
 	return nilfs_clean_segments(inode->i_sb, argp);
 }

 static int nilfs_ioctl_sync(struct inode *inode, struct file *filp,
 			    unsigned int cmd, void __user *argp)
 {
 	__u64 cno;
 	int ret;

 	ret = nilfs_construct_segment(inode->i_sb);
 	if (ret < 0)
 		return ret;

 	if (argp != NULL) {
 		cno = NILFS_SB(inode->i_sb)->s_nilfs->ns_cno - 1;
 		if (copy_to_user(argp, &cno, sizeof(cno)))
 			return -EFAULT;
 	}
 	return 0;
 }

 long nilfs_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
 {
 	struct inode *inode = filp->f_dentry->d_inode;
 	void __user *argp = (void * __user *)arg;

 	switch (cmd) {
 	case NILFS_IOCTL_CHANGE_CPMODE:
 		return nilfs_ioctl_change_cpmode(inode, filp, cmd, argp);
 	case NILFS_IOCTL_DELETE_CHECKPOINT:
 		return nilfs_ioctl_delete_checkpoint(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_CPINFO:
 		return nilfs_ioctl_get_cpinfo(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_CPSTAT:
 		return nilfs_ioctl_get_cpstat(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_SUINFO:
 		return nilfs_ioctl_get_suinfo(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_SUSTAT:
 		return nilfs_ioctl_get_sustat(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_VINFO:
 		/* XXX: rename to ??? */
 		return nilfs_ioctl_get_vinfo(inode, filp, cmd, argp);
 	case NILFS_IOCTL_GET_BDESCS:
 		return nilfs_ioctl_get_bdescs(inode, filp, cmd, argp);
 	case NILFS_IOCTL_CLEAN_SEGMENTS:
 		return nilfs_ioctl_clean_segments(inode, filp, cmd, argp);
 	case NILFS_IOCTL_SYNC:
 		return nilfs_ioctl_sync(inode, filp, cmd, argp);
 	default:
 		return -ENOTTY;
 	}
 }
	/*
	* ioctl.c - NILFS ioctl operations.
	*
	* Copyright (C) 2007, 2008 Nippon Telegraph and Telephone Corporation.
	*
	* 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	*
	* Written by Koji Sato <koji@osrg.net>.
	*/

	#include <linux/fs.h>
	#include <linux/wait.h>
	#include <linux/smp_lock.h> /* lock_kernel(), unlock_kernel() */
	#include <linux/capability.h> /* capable() */
	#include <linux/uaccess.h> /* copy_from_user(), copy_to_user() */
	#include <linux/nilfs2_fs.h>
	#include "nilfs.h"
	#include "segment.h"
	#include "bmap.h"
	#include "cpfile.h"
	#include "sufile.h"
	#include "dat.h"


	static int nilfs_ioctl_wrap_copy(struct the_nilfs *nilfs,
	struct nilfs_argv *argv, int dir,
	ssize_t (dofunc)(struct the_nilfs ,
	__u64 *, int,
	void *, size_t, size_t))
	{
	void *buf;
	void __user base = (void __user )(unsigned long)argv->v_base;
	size_t maxmembs, total, n;
	ssize_t nr;
	int ret, i;
	__u64 pos, ppos;

	if (argv->v_nmembs == 0)
	return 0;

	if (argv->v_size > PAGE_SIZE)
	return -EINVAL;

	buf = (void *)__get_free_pages(GFP_NOFS, 0);
	if (unlikely(!buf))
	return -ENOMEM;
	maxmembs = PAGE_SIZE / argv->v_size;

	ret = 0;
	total = 0;
	pos = argv->v_index;
	for (i = 0; i < argv->v_nmembs; i += n) {
	n = (argv->v_nmembs - i < maxmembs) ?
	argv->v_nmembs - i : maxmembs;
	if ((dir & _IOC_WRITE) &&
	copy_from_user(buf, base + argv->v_size * i,
	argv->v_size * n)) {
	ret = -EFAULT;
	break;
	}
	ppos = pos;
	nr = dofunc(nilfs, &pos, argv->v_flags, buf, argv->v_size,
	n);
	if (nr < 0) {
	ret = nr;
	break;
	}
	if ((dir & _IOC_READ) &&
	copy_to_user(base + argv->v_size * i, buf,
	argv->v_size * nr)) {
	ret = -EFAULT;
	break;
	}
	total += nr;
	if ((size_t)nr < n)
	break;
	if (pos == ppos)
	pos += n;
	}
	argv->v_nmembs = total;

	free_pages((unsigned long)buf, 0);
	return ret;
	}

	static int nilfs_ioctl_change_cpmode(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct inode *cpfile = NILFS_SB(inode->i_sb)->s_nilfs->ns_cpfile;
	struct nilfs_transaction_info ti;
	struct nilfs_cpmode cpmode;
	int ret;

	if (!capable(CAP_SYS_ADMIN))
	return -EPERM;
	if (copy_from_user(&cpmode, argp, sizeof(cpmode)))
	return -EFAULT;

	nilfs_transaction_begin(inode->i_sb, &ti, 0);
	ret = nilfs_cpfile_change_cpmode(
	cpfile, cpmode.cm_cno, cpmode.cm_mode);
	if (unlikely(ret < 0)) {
	nilfs_transaction_abort(inode->i_sb);
	return ret;
	}
	nilfs_transaction_commit(inode->i_sb); /* never fails */
	return ret;
	}

	static int
	nilfs_ioctl_delete_checkpoint(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct inode *cpfile = NILFS_SB(inode->i_sb)->s_nilfs->ns_cpfile;
	struct nilfs_transaction_info ti;
	__u64 cno;
	int ret;

	if (!capable(CAP_SYS_ADMIN))
	return -EPERM;
	if (copy_from_user(&cno, argp, sizeof(cno)))
	return -EFAULT;

	nilfs_transaction_begin(inode->i_sb, &ti, 0);
	ret = nilfs_cpfile_delete_checkpoint(cpfile, cno);
	if (unlikely(ret < 0)) {
	nilfs_transaction_abort(inode->i_sb);
	return ret;
	}
	nilfs_transaction_commit(inode->i_sb); /* never fails */
	return ret;
	}

	static ssize_t
	nilfs_ioctl_do_get_cpinfo(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	return nilfs_cpfile_get_cpinfo(nilfs->ns_cpfile, posp, flags, buf,
	nmembs);
	}

	static int nilfs_ioctl_get_cpinfo(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_argv argv;
	int ret;

	if (copy_from_user(&argv, argp, sizeof(argv)))
	return -EFAULT;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
	nilfs_ioctl_do_get_cpinfo);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &argv, sizeof(argv)))
	ret = -EFAULT;
	return ret;
	}

	static int nilfs_ioctl_get_cpstat(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_cpstat cpstat;
	int ret;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_cpfile_get_stat(nilfs->ns_cpfile, &cpstat);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &cpstat, sizeof(cpstat)))
	ret = -EFAULT;
	return ret;
	}

	static ssize_t
	nilfs_ioctl_do_get_suinfo(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	return nilfs_sufile_get_suinfo(nilfs->ns_sufile, *posp, buf, nmembs);
	}

	static int nilfs_ioctl_get_suinfo(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_argv argv;
	int ret;

	if (copy_from_user(&argv, argp, sizeof(argv)))
	return -EFAULT;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
	nilfs_ioctl_do_get_suinfo);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &argv, sizeof(argv)))
	ret = -EFAULT;
	return ret;
	}

	static int nilfs_ioctl_get_sustat(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_sustat sustat;
	int ret;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_sufile_get_stat(nilfs->ns_sufile, &sustat);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &sustat, sizeof(sustat)))
	ret = -EFAULT;
	return ret;
	}

	static ssize_t
	nilfs_ioctl_do_get_vinfo(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	return nilfs_dat_get_vinfo(nilfs_dat_inode(nilfs), buf, nmembs);
	}

	static int nilfs_ioctl_get_vinfo(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_argv argv;
	int ret;

	if (copy_from_user(&argv, argp, sizeof(argv)))
	return -EFAULT;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
	nilfs_ioctl_do_get_vinfo);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &argv, sizeof(argv)))
	ret = -EFAULT;
	return ret;
	}

	static ssize_t
	nilfs_ioctl_do_get_bdescs(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	struct inode *dat = nilfs_dat_inode(nilfs);
	struct nilfs_bmap *bmap = NILFS_I(dat)->i_bmap;
	struct nilfs_bdesc *bdescs = buf;
	int ret, i;

	for (i = 0; i < nmembs; i++) {
	ret = nilfs_bmap_lookup_at_level(bmap,
	bdescs[i].bd_offset,
	bdescs[i].bd_level + 1,
	&bdescs[i].bd_blocknr);
	if (ret < 0) {
	if (ret != -ENOENT)
	return ret;
	bdescs[i].bd_blocknr = 0;
	}
	}
	return nmembs;
	}

	static int nilfs_ioctl_get_bdescs(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	struct the_nilfs *nilfs = NILFS_SB(inode->i_sb)->s_nilfs;
	struct nilfs_argv argv;
	int ret;

	if (copy_from_user(&argv, argp, sizeof(argv)))
	return -EFAULT;

	down_read(&nilfs->ns_segctor_sem);
	ret = nilfs_ioctl_wrap_copy(nilfs, &argv, _IOC_DIR(cmd),
	nilfs_ioctl_do_get_bdescs);
	up_read(&nilfs->ns_segctor_sem);
	if (ret < 0)
	return ret;

	if (copy_to_user(argp, &argv, sizeof(argv)))
	ret = -EFAULT;
	return ret;
	}

	static int nilfs_ioctl_move_inode_block(struct inode *inode,
	struct nilfs_vdesc *vdesc,
	struct list_head *buffers)
	{
	struct buffer_head *bh;
	int ret;

	if (vdesc->vd_flags == 0)
	ret = nilfs_gccache_submit_read_data(
	inode, vdesc->vd_offset, vdesc->vd_blocknr,
	vdesc->vd_vblocknr, &bh);
	else
	ret = nilfs_gccache_submit_read_node(
	inode, vdesc->vd_blocknr, vdesc->vd_vblocknr, &bh);

	if (unlikely(ret < 0)) {
	if (ret == -ENOENT)
	printk(KERN_CRIT
	"%s: invalid virtual block address (%s): "
	"ino=%llu, cno=%llu, offset=%llu, "
	"blocknr=%llu, vblocknr=%llu\n",
	__func__, vdesc->vd_flags ? "node" : "data",
	(unsigned long long)vdesc->vd_ino,
	(unsigned long long)vdesc->vd_cno,
	(unsigned long long)vdesc->vd_offset,
	(unsigned long long)vdesc->vd_blocknr,
	(unsigned long long)vdesc->vd_vblocknr);
	return ret;
	}
	bh->b_private = vdesc;
	list_add_tail(&bh->b_assoc_buffers, buffers);
	return 0;
	}

	static ssize_t
	nilfs_ioctl_do_move_blocks(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	struct inode *inode;
	struct nilfs_vdesc *vdesc;
	struct buffer_head bh, n;
	LIST_HEAD(buffers);
	ino_t ino;
	__u64 cno;
	int i, ret;

	for (i = 0, vdesc = buf; i < nmembs; ) {
	ino = vdesc->vd_ino;
	cno = vdesc->vd_cno;
	inode = nilfs_gc_iget(nilfs, ino, cno);
	if (unlikely(inode == NULL)) {
	ret = -ENOMEM;
	goto failed;
	}
	do {
	ret = nilfs_ioctl_move_inode_block(inode, vdesc,
	&buffers);
	if (unlikely(ret < 0))
	goto failed;
	vdesc++;
	} while (++i < nmembs &&
	vdesc->vd_ino == ino && vdesc->vd_cno == cno);
	}

	list_for_each_entry_safe(bh, n, &buffers, b_assoc_buffers) {
	ret = nilfs_gccache_wait_and_mark_dirty(bh);
	if (unlikely(ret < 0)) {
	if (ret == -EEXIST) {
	vdesc = bh->b_private;
	printk(KERN_CRIT
	"%s: conflicting %s buffer: "
	"ino=%llu, cno=%llu, offset=%llu, "
	"blocknr=%llu, vblocknr=%llu\n",
	__func__,
	vdesc->vd_flags ? "node" : "data",
	(unsigned long long)vdesc->vd_ino,
	(unsigned long long)vdesc->vd_cno,
	(unsigned long long)vdesc->vd_offset,
	(unsigned long long)vdesc->vd_blocknr,
	(unsigned long long)vdesc->vd_vblocknr);
	}
	goto failed;
	}
	list_del_init(&bh->b_assoc_buffers);
	bh->b_private = NULL;
	brelse(bh);
	}
	return nmembs;

	failed:
	list_for_each_entry_safe(bh, n, &buffers, b_assoc_buffers) {
	list_del_init(&bh->b_assoc_buffers);
	bh->b_private = NULL;
	brelse(bh);
	}
	return ret;
	}

	static inline int nilfs_ioctl_move_blocks(struct the_nilfs *nilfs,
	struct nilfs_argv *argv,
	int dir)
	{
	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
	nilfs_ioctl_do_move_blocks);
	}

	static ssize_t
	nilfs_ioctl_do_delete_checkpoints(struct the_nilfs nilfs, __u64 posp,
	int flags, void *buf, size_t size,
	size_t nmembs)
	{
	struct inode *cpfile = nilfs->ns_cpfile;
	struct nilfs_period *periods = buf;
	int ret, i;

	for (i = 0; i < nmembs; i++) {
	ret = nilfs_cpfile_delete_checkpoints(
	cpfile, periods[i].p_start, periods[i].p_end);
	if (ret < 0)
	return ret;
	}
	return nmembs;
	}

	static inline int nilfs_ioctl_delete_checkpoints(struct the_nilfs *nilfs,
	struct nilfs_argv *argv,
	int dir)
	{
	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
	nilfs_ioctl_do_delete_checkpoints);
	}

	static ssize_t
	nilfs_ioctl_do_free_vblocknrs(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	int ret = nilfs_dat_freev(nilfs_dat_inode(nilfs), buf, nmembs);

	return (ret < 0) ? ret : nmembs;
	}

	static inline int nilfs_ioctl_free_vblocknrs(struct the_nilfs *nilfs,
	struct nilfs_argv *argv,
	int dir)
	{
	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
	nilfs_ioctl_do_free_vblocknrs);
	}

	static ssize_t
	nilfs_ioctl_do_mark_blocks_dirty(struct the_nilfs nilfs, __u64 posp,
	int flags, void *buf, size_t size,
	size_t nmembs)
	{
	struct inode *dat = nilfs_dat_inode(nilfs);
	struct nilfs_bmap *bmap = NILFS_I(dat)->i_bmap;
	struct nilfs_bdesc *bdescs = buf;
	int ret, i;

	for (i = 0; i < nmembs; i++) {
	/* XXX: use macro or inline func to check liveness */
	ret = nilfs_bmap_lookup_at_level(bmap,
	bdescs[i].bd_offset,
	bdescs[i].bd_level + 1,
	&bdescs[i].bd_blocknr);
	if (ret < 0) {
	if (ret != -ENOENT)
	return ret;
	bdescs[i].bd_blocknr = 0;
	}
	if (bdescs[i].bd_blocknr != bdescs[i].bd_oblocknr)
	/* skip dead block */
	continue;
	if (bdescs[i].bd_level == 0) {
	ret = nilfs_mdt_mark_block_dirty(dat,
	bdescs[i].bd_offset);
	if (ret < 0) {
	WARN_ON(ret == -ENOENT);
	return ret;
	}
	} else {
	ret = nilfs_bmap_mark(bmap, bdescs[i].bd_offset,
	bdescs[i].bd_level);
	if (ret < 0) {
	WARN_ON(ret == -ENOENT);
	return ret;
	}
	}
	}
	return nmembs;
	}

	static inline int nilfs_ioctl_mark_blocks_dirty(struct the_nilfs *nilfs,
	struct nilfs_argv *argv,
	int dir)
	{
	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
	nilfs_ioctl_do_mark_blocks_dirty);
	}

	static ssize_t
	nilfs_ioctl_do_free_segments(struct the_nilfs nilfs, __u64 posp, int flags,
	void *buf, size_t size, size_t nmembs)
	{
	struct nilfs_sb_info *sbi = nilfs_get_writer(nilfs);
	int ret;

	if (unlikely(!sbi))
	return -EROFS;
	ret = nilfs_segctor_add_segments_to_be_freed(
	NILFS_SC(sbi), buf, nmembs);
	nilfs_put_writer(nilfs);

	return (ret < 0) ? ret : nmembs;
	}

	static inline int nilfs_ioctl_free_segments(struct the_nilfs *nilfs,
	struct nilfs_argv *argv,
	int dir)
	{
	return nilfs_ioctl_wrap_copy(nilfs, argv, dir,
	nilfs_ioctl_do_free_segments);
	}

	int nilfs_ioctl_prepare_clean_segments(struct the_nilfs *nilfs,
	void __user *argp)
	{
	struct nilfs_argv argv[5];
	const char *msg;
	int dir, ret;

	if (copy_from_user(argv, argp, sizeof(argv)))
	return -EFAULT;

	dir = _IOC_WRITE;
	ret = nilfs_ioctl_move_blocks(nilfs, &argv[0], dir);
	if (ret < 0) {
	msg = "cannot read source blocks";
	goto failed;
	}
	ret = nilfs_ioctl_delete_checkpoints(nilfs, &argv[1], dir);
	if (ret < 0) {
	/*
	* can safely abort because checkpoints can be removed
	* independently.
	*/
	msg = "cannot delete checkpoints";
	goto failed;
	}
	ret = nilfs_ioctl_free_vblocknrs(nilfs, &argv[2], dir);
	if (ret < 0) {
	/*
	* can safely abort because DAT file is updated atomically
	* using a copy-on-write technique.
	*/
	msg = "cannot delete virtual blocks from DAT file";
	goto failed;
	}
	ret = nilfs_ioctl_mark_blocks_dirty(nilfs, &argv[3], dir);
	if (ret < 0) {
	/*
	* can safely abort because the operation is nondestructive.
	*/
	msg = "cannot mark copying blocks dirty";
	goto failed;
	}
	ret = nilfs_ioctl_free_segments(nilfs, &argv[4], dir);
	if (ret < 0) {
	/*
	* can safely abort because this operation is atomic.
	*/
	msg = "cannot set segments to be freed";
	goto failed;
	}
	return 0;

	failed:
	nilfs_remove_all_gcinode(nilfs);
	printk(KERN_ERR "NILFS: GC failed during preparation: %s: err=%d\n",
	msg, ret);
	return ret;
	}

	static int nilfs_ioctl_clean_segments(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	if (!capable(CAP_SYS_ADMIN))
	return -EPERM;
	return nilfs_clean_segments(inode->i_sb, argp);
	}

	static int nilfs_ioctl_sync(struct inode inode, struct file filp,
	unsigned int cmd, void __user *argp)
	{
	__u64 cno;
	int ret;

	ret = nilfs_construct_segment(inode->i_sb);
	if (ret < 0)
	return ret;

	if (argp != NULL) {
	cno = NILFS_SB(inode->i_sb)->s_nilfs->ns_cno - 1;
	if (copy_to_user(argp, &cno, sizeof(cno)))
	return -EFAULT;
	}
	return 0;
	}

	long nilfs_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
	{
	struct inode *inode = filp->f_dentry->d_inode;
	void __user argp = (void __user *)arg;

	switch (cmd) {
	case NILFS_IOCTL_CHANGE_CPMODE:
	return nilfs_ioctl_change_cpmode(inode, filp, cmd, argp);
	case NILFS_IOCTL_DELETE_CHECKPOINT:
	return nilfs_ioctl_delete_checkpoint(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_CPINFO:
	return nilfs_ioctl_get_cpinfo(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_CPSTAT:
	return nilfs_ioctl_get_cpstat(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_SUINFO:
	return nilfs_ioctl_get_suinfo(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_SUSTAT:
	return nilfs_ioctl_get_sustat(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_VINFO:
	/* XXX: rename to ??? */
	return nilfs_ioctl_get_vinfo(inode, filp, cmd, argp);
	case NILFS_IOCTL_GET_BDESCS:
	return nilfs_ioctl_get_bdescs(inode, filp, cmd, argp);
	case NILFS_IOCTL_CLEAN_SEGMENTS:
	return nilfs_ioctl_clean_segments(inode, filp, cmd, argp);
	case NILFS_IOCTL_SYNC:
	return nilfs_ioctl_sync(inode, filp, cmd, argp);
	default:
	return -ENOTTY;
	}
	}