1 files changed, 185 insertions, 0 deletions

diff --git a/fs/btrfs/ioctl.c b/fs/btrfs/ioctl.c
index 0e92e5763005..55a7283a9e18 100644
--- a/fs/btrfs/ioctl.c
+++ b/fs/btrfs/ioctl.c
@@ -3390,6 +3390,183 @@ out:
         return ret;
 }
 
+static long btrfs_ioctl_quota_ctl(struct btrfs_root *root, void __user *arg)
+{
+        struct btrfs_ioctl_quota_ctl_args *sa;
+        struct btrfs_trans_handle *trans = NULL;
+        int ret;
+        int err;
+
+        if (!capable(CAP_SYS_ADMIN))
+                return -EPERM;
+
+        if (root->fs_info->sb->s_flags & MS_RDONLY)
+                return -EROFS;
+
+        sa = memdup_user(arg, sizeof(*sa));
+        if (IS_ERR(sa))
+                return PTR_ERR(sa);
+
+        if (sa->cmd != BTRFS_QUOTA_CTL_RESCAN) {
+                trans = btrfs_start_transaction(root, 2);
+                if (IS_ERR(trans)) {
+                        ret = PTR_ERR(trans);
+                        goto out;
+                }
+        }
+
+        switch (sa->cmd) {
+        case BTRFS_QUOTA_CTL_ENABLE:
+                ret = btrfs_quota_enable(trans, root->fs_info);
+                break;
+        case BTRFS_QUOTA_CTL_DISABLE:
+                ret = btrfs_quota_disable(trans, root->fs_info);
+                break;
+        case BTRFS_QUOTA_CTL_RESCAN:
+                ret = btrfs_quota_rescan(root->fs_info);
+                break;
+        default:
+                ret = -EINVAL;
+                break;
+        }
+
+        if (copy_to_user(arg, sa, sizeof(*sa)))
+                ret = -EFAULT;
+
+        if (trans) {
+                err = btrfs_commit_transaction(trans, root);
+                if (err && !ret)
+                        ret = err;
+        }
+
+out:
+        kfree(sa);
+        return ret;
+}
+
+static long btrfs_ioctl_qgroup_assign(struct btrfs_root *root, void __user *arg)
+{
+        struct btrfs_ioctl_qgroup_assign_args *sa;
+        struct btrfs_trans_handle *trans;
+        int ret;
+        int err;
+
+        if (!capable(CAP_SYS_ADMIN))
+                return -EPERM;
+
+        if (root->fs_info->sb->s_flags & MS_RDONLY)
+                return -EROFS;
+
+        sa = memdup_user(arg, sizeof(*sa));
+        if (IS_ERR(sa))
+                return PTR_ERR(sa);
+
+        trans = btrfs_join_transaction(root);
+        if (IS_ERR(trans)) {
+                ret = PTR_ERR(trans);
+                goto out;
+        }
+
+        /* FIXME: check if the IDs really exist */
+        if (sa->assign) {
+                ret = btrfs_add_qgroup_relation(trans, root->fs_info,
+                                                sa->src, sa->dst);
+        } else {
+                ret = btrfs_del_qgroup_relation(trans, root->fs_info,
+                                                sa->src, sa->dst);
+        }
+
+        err = btrfs_end_transaction(trans, root);
+        if (err && !ret)
+                ret = err;
+
+out:
+        kfree(sa);
+        return ret;
+}
+
+static long btrfs_ioctl_qgroup_create(struct btrfs_root *root, void __user *arg)
+{
+        struct btrfs_ioctl_qgroup_create_args *sa;
+        struct btrfs_trans_handle *trans;
+        int ret;
+        int err;
+
+        if (!capable(CAP_SYS_ADMIN))
+                return -EPERM;
+
+        if (root->fs_info->sb->s_flags & MS_RDONLY)
+                return -EROFS;
+
+        sa = memdup_user(arg, sizeof(*sa));
+        if (IS_ERR(sa))
+                return PTR_ERR(sa);
+
+        trans = btrfs_join_transaction(root);
+        if (IS_ERR(trans)) {
+                ret = PTR_ERR(trans);
+                goto out;
+        }
+
+        /* FIXME: check if the IDs really exist */
+        if (sa->create) {
+                ret = btrfs_create_qgroup(trans, root->fs_info, sa->qgroupid,
+                                          NULL);
+        } else {
+                ret = btrfs_remove_qgroup(trans, root->fs_info, sa->qgroupid);
+        }
+
+        err = btrfs_end_transaction(trans, root);
+        if (err && !ret)
+                ret = err;
+
+out:
+        kfree(sa);
+        return ret;
+}
+
+static long btrfs_ioctl_qgroup_limit(struct btrfs_root *root, void __user *arg)
+{
+        struct btrfs_ioctl_qgroup_limit_args *sa;
+        struct btrfs_trans_handle *trans;
+        int ret;
+        int err;
+        u64 qgroupid;
+
+        if (!capable(CAP_SYS_ADMIN))
+                return -EPERM;
+
+        if (root->fs_info->sb->s_flags & MS_RDONLY)
+                return -EROFS;
+
+        sa = memdup_user(arg, sizeof(*sa));
+        if (IS_ERR(sa))
+                return PTR_ERR(sa);
+
+        trans = btrfs_join_transaction(root);
+        if (IS_ERR(trans)) {
+                ret = PTR_ERR(trans);
+                goto out;
+        }
+
+        qgroupid = sa->qgroupid;
+        if (!qgroupid) {
+                /* take the current subvol as qgroup */
+                qgroupid = root->root_key.objectid;
+        }
+
+        /* FIXME: check if the IDs really exist */
+        ret = btrfs_limit_qgroup(trans, root->fs_info, qgroupid, &sa->lim);
+
+        err = btrfs_end_transaction(trans, root);
+        if (err && !ret)
+                ret = err;
+
+out:
+        kfree(sa);
+        return ret;
+}
+
 long btrfs_ioctl(struct file *file, unsigned int
                 cmd, unsigned long arg)
 {
@@ -3476,6 +3653,14 @@ long btrfs_ioctl(struct file *file, unsigned int
                 return btrfs_ioctl_get_dev_stats(root, argp, 0);
         case BTRFS_IOC_GET_AND_RESET_DEV_STATS:
                 return btrfs_ioctl_get_dev_stats(root, argp, 1);
+        case BTRFS_IOC_QUOTA_CTL:
+                return btrfs_ioctl_quota_ctl(root, argp);
+        case BTRFS_IOC_QGROUP_ASSIGN:
+                return btrfs_ioctl_qgroup_assign(root, argp);
+        case BTRFS_IOC_QGROUP_CREATE:
+                return btrfs_ioctl_qgroup_create(root, argp);
+        case BTRFS_IOC_QGROUP_LIMIT:
+                return btrfs_ioctl_qgroup_limit(root, argp);
         }
 
         return -ENOTTY;