From 6e8d60a8c9ab22c489c0707ffd854bf1db9852f4 Mon Sep 17 00:00:00 2001

From: Boris Burkov <boris@bur.io>

Date: Fri, 21 Aug 2020 00:40:08 -0700

Subject: [PATCH] btrfs-progs: receive: process setflags ioctl commands

In send stream v2, send can emit a command for setting inode flags via

the setflags ioctl. Pass the flags attribute through to the ioctl call

in receive.

Reviewed-by: Nikolay Borisov <nborisov@suse.com>

Signed-off-by: Boris Burkov <boris@bur.io>

---

 cmds/receive-dump.c  |  6 ++++++

 cmds/receive.c       | 25 +++++++++++++++++++++++++

 common/send-stream.c |  7 +++++++

 common/send-stream.h |  1 +

 4 files changed, 39 insertions(+)

diff --git a/cmds/receive-dump.c b/cmds/receive-dump.c

index fa397bcf..df5991e1 100644

--- a/cmds/receive-dump.c

+++ b/cmds/receive-dump.c

@@ -339,6 +339,11 @@ static int print_fallocate(const char *path, int mode, u64 offset, u64 len,

 			  mode, offset, len);

 }

+static int print_setflags(const char *path, int flags, void *user)

+{

+	return PRINT_DUMP(user, path, "setflags", "flags=%d", flags);

+}

+

 struct btrfs_send_ops btrfs_print_send_ops = {

 	.subvol = print_subvol,

 	.snapshot = print_snapshot,

@@ -363,4 +368,5 @@ struct btrfs_send_ops btrfs_print_send_ops = {

 	.update_extent = print_update_extent,

 	.encoded_write = print_encoded_write,

 	.fallocate = print_fallocate,

+	.setflags = print_setflags,

 };

diff --git a/cmds/receive.c b/cmds/receive.c

index 4893d693..7f76a04f 100644

--- a/cmds/receive.c

+++ b/cmds/receive.c

@@ -38,6 +38,7 @@

 #include <sys/types.h>

 #include <sys/uio.h>

 #include <sys/xattr.h>

+#include <linux/fs.h>

 #include <uuid/uuid.h>

 #include <lzo/lzo1x.h>

@@ -1284,6 +1285,29 @@ static int process_fallocate(const char *path, int mode, u64 offset, u64 len,

 	return 0;

 }

+static int process_setflags(const char *path, int flags, void *user)

+{

+	int ret;

+	struct btrfs_receive *rctx = user;

+	char full_path[PATH_MAX];

+

+	ret = path_cat_out(full_path, rctx->full_subvol_path, path);

+	if (ret < 0) {

+		error("setflags: path invalid: %s", path);

+		return ret;

+	}

+	ret = open_inode_for_write(rctx, full_path);

+	if (ret < 0)

+		return ret;

+	ret = ioctl(rctx->write_fd, FS_IOC_SETFLAGS, &flags);

+	if (ret < 0) {

+		ret = -errno;

+		error("setflags: setflags ioctl on %s failed: %m", path);

+		return ret;

+	}

+	return 0;

+}

+

 static struct btrfs_send_ops send_ops = {

 	.subvol = process_subvol,

 	.snapshot = process_snapshot,

@@ -1308,6 +1332,7 @@ static struct btrfs_send_ops send_ops = {

 	.update_extent = process_update_extent,

 	.encoded_write = process_encoded_write,

 	.fallocate = process_fallocate,

+	.setflags = process_setflags,

 };

 static int do_receive(struct btrfs_receive *rctx, const char *tomnt,

diff --git a/common/send-stream.c b/common/send-stream.c

index 2d0aa624..21295cbb 100644

--- a/common/send-stream.c

+++ b/common/send-stream.c

@@ -374,6 +374,7 @@ static int read_and_process_cmd(struct btrfs_send_stream *sctx)

 	int len;

 	int xattr_len;

 	int fallocate_mode;

+	int setflags_flags;

 	ret = read_cmd(sctx);

 	if (ret)

@@ -546,8 +547,14 @@ static int read_and_process_cmd(struct btrfs_send_stream *sctx)

 		ret = sctx->ops->fallocate(path, fallocate_mode, offset, tmp,

 					   sctx->user);

 		break;

+	case BTRFS_SEND_C_SETFLAGS:

+		TLV_GET_STRING(sctx, BTRFS_SEND_A_PATH, &path);

+		TLV_GET_U32(sctx, BTRFS_SEND_A_SETFLAGS_FLAGS, &setflags_flags);

+		ret = sctx->ops->setflags(path, setflags_flags, sctx->user);

+		break;

 	}

+

 tlv_get_failed:

 out:

 	free(path);

diff --git a/common/send-stream.h b/common/send-stream.h

index 61a88d3d..3189f889 100644

--- a/common/send-stream.h

+++ b/common/send-stream.h

@@ -59,6 +59,7 @@ struct btrfs_send_ops {

 			     u32 encryption, void *user);

 	int (*fallocate)(const char *path, int mode, u64 offset, u64 len,

 			 void *user);

+	int (*setflags)(const char *path, int flags, void *user);

 };

 int btrfs_read_and_process_send_stream(int fd,

-- 

2.35.1