1 files changed, 5 insertions, 8 deletions

diff --git a/fs/btrfs/ioctl.c b/fs/btrfs/ioctl.c
index 6dafa857bbb9..d3d9b4abb09b 100644
--- a/fs/btrfs/ioctl.c
+++ b/fs/btrfs/ioctl.c
@@ -187,7 +187,7 @@ static int btrfs_ioctl_setflags(struct file *file, void __user *arg)
         struct btrfs_inode *binode = BTRFS_I(inode);
         struct btrfs_root *root = binode->root;
         struct btrfs_trans_handle *trans;
-        unsigned int fsflags;
+        unsigned int fsflags, old_fsflags;
         int ret;
         const char *comp = NULL;
         u32 binode_flags = binode->flags;
@@ -212,13 +212,10 @@ static int btrfs_ioctl_setflags(struct file *file, void __user *arg)
         inode_lock(inode);
 
         fsflags = btrfs_mask_fsflags_for_type(inode, fsflags);
-        if ((fsflags ^ btrfs_inode_flags_to_fsflags(binode->flags)) &
-            (FS_APPEND_FL | FS_IMMUTABLE_FL)) {
-                if (!capable(CAP_LINUX_IMMUTABLE)) {
-                        ret = -EPERM;
-                        goto out_unlock;
-                }
-        }
+        old_fsflags = btrfs_inode_flags_to_fsflags(binode->flags);
+        ret = vfs_ioc_setflags_prepare(inode, old_fsflags, fsflags);
+        if (ret)
+                goto out_unlock;
 
         if (fsflags & FS_SYNC_FL)
                 binode_flags |= BTRFS_INODE_SYNC;