diff mbox series

Add vmsplice03 test

Message ID 20191126135434.3196-1-jcronenberg@suse.de
State Superseded
Headers show
Series Add vmsplice03 test | expand

Commit Message

Jorik Cronenberg Nov. 26, 2019, 1:54 p.m. UTC
From: jcronenberg <jcronenberg@suse.de>

Add a test for splicing from a pipe to user memory.

Signed-off-by: Jorik Cronenberg <jcronenberg@suse.de>
---
 testcases/kernel/syscalls/vmsplice/.gitignore |  1 +
 .../kernel/syscalls/vmsplice/vmsplice03.c     | 70 +++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 testcases/kernel/syscalls/vmsplice/vmsplice03.c

Comments

Cyril Hrubis Dec. 3, 2019, 12:15 p.m. UTC | #1
Hi!
> diff --git a/testcases/kernel/syscalls/vmsplice/vmsplice03.c b/testcases/kernel/syscalls/vmsplice/vmsplice03.c
> new file mode 100644
> index 000000000..54dace1b6
> --- /dev/null
> +++ b/testcases/kernel/syscalls/vmsplice/vmsplice03.c
> @@ -0,0 +1,70 @@
> +// SPDX-License-Identifier: GPL-2.0-or-later
> +/*
> + * Copyright (c) 2019 SUSE LLC
> + * Author: Jorik Cronenberg <jcronenberg@suse.de>
> + *
> + * Test vmsplice() from a pipe into user memory
> + */
> +
> +#define _GNU_SOURCE
> +
> +#include "tst_test.h"
> +#include "lapi/fcntl.h"
> +#include "lapi/vmsplice.h"
> +
> +
> +#define TEST_BLOCK_SIZE (64*1024)	/* 64K */
> +
> +static char buffer[TEST_BLOCK_SIZE];
> +static off64_t offset;
> +
> +static void vmsplice_test(void)
> +{
> +	int written, i;
> +	int pipes[2];
> +	struct iovec iov;
> +	char arr_write[TEST_BLOCK_SIZE];
> +	char *arr_read;
> +
> +	iov.iov_base = arr_write;
> +	iov.iov_len = TEST_BLOCK_SIZE;
> +	offset = sizeof(struct iovec);

It may be a good idea to memset the arr_write[] array with well defined
data, e.g. 0 just to be sure that kernel really filled it.

> +	SAFE_PIPE(pipes);
> +	SAFE_WRITE(1, pipes[1], buffer, TEST_BLOCK_SIZE);
> +	written = vmsplice(pipes[0], &iov, 1, 0);
> +	arr_read = (char *)&iov;

This only works by accident because the arr_write is put on the stack
after the iovec structure and there is no padding in between them.

What you really should do is to use the arr_write pointer in the
comparsion instead. The struc iovec is just a description of an buffer
that starts in memory at arr_write address and is iov_len long because
the kernel has to know its lenght.

> +	if (written < 0)
> +		tst_brk(TBROK | TERRNO, "vmsplice() failed");
> +	else if (written == 0)

There is no need for an else branch after tst_brk() because tst_brk()
will exit the test immediately.

> +		tst_res(TFAIL, "vmsplice() didn't write anything");
> +	else {
> +		for (i = 0; i < written; i++) {
> +			if (arr_read[i+offset] != buffer[i]) {
> +				tst_res(TFAIL,
> +					"Wrong data in user memory at %i", i);
> +				break;
> +			}
> +		}
> +		if (i == written)
> +			tst_res(TPASS, "Spliced correctly into user memory");
> +	}
> +
> +	SAFE_CLOSE(pipes[1]);
> +	SAFE_CLOSE(pipes[0]);
> +}
> +
> +static void setup(void)
> +{
> +	int i;
> +
> +	for (i = 0; i < TEST_BLOCK_SIZE; i++)
> +		buffer[i] = i & 0xff;
> +}
> +
> +static struct tst_test test = {
> +	.setup = setup,
> +	.test_all = vmsplice_test,
> +	.min_kver = "2.6.23",
> +};

Otherwise it looks good.
diff mbox series

Patch

diff --git a/testcases/kernel/syscalls/vmsplice/.gitignore b/testcases/kernel/syscalls/vmsplice/.gitignore
index 2cc74a956..03922073c 100644
--- a/testcases/kernel/syscalls/vmsplice/.gitignore
+++ b/testcases/kernel/syscalls/vmsplice/.gitignore
@@ -1,2 +1,3 @@ 
 /vmsplice01
 /vmsplice02
+/vmsplice03
diff --git a/testcases/kernel/syscalls/vmsplice/vmsplice03.c b/testcases/kernel/syscalls/vmsplice/vmsplice03.c
new file mode 100644
index 000000000..54dace1b6
--- /dev/null
+++ b/testcases/kernel/syscalls/vmsplice/vmsplice03.c
@@ -0,0 +1,70 @@ 
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Copyright (c) 2019 SUSE LLC
+ * Author: Jorik Cronenberg <jcronenberg@suse.de>
+ *
+ * Test vmsplice() from a pipe into user memory
+ */
+
+#define _GNU_SOURCE
+
+#include "tst_test.h"
+#include "lapi/fcntl.h"
+#include "lapi/vmsplice.h"
+
+
+#define TEST_BLOCK_SIZE (64*1024)	/* 64K */
+
+static char buffer[TEST_BLOCK_SIZE];
+static off64_t offset;
+
+static void vmsplice_test(void)
+{
+	int written, i;
+	int pipes[2];
+	struct iovec iov;
+	char arr_write[TEST_BLOCK_SIZE];
+	char *arr_read;
+
+	iov.iov_base = arr_write;
+	iov.iov_len = TEST_BLOCK_SIZE;
+	offset = sizeof(struct iovec);
+
+	SAFE_PIPE(pipes);
+	SAFE_WRITE(1, pipes[1], buffer, TEST_BLOCK_SIZE);
+	written = vmsplice(pipes[0], &iov, 1, 0);
+	arr_read = (char *)&iov;
+
+	if (written < 0)
+		tst_brk(TBROK | TERRNO, "vmsplice() failed");
+	else if (written == 0)
+		tst_res(TFAIL, "vmsplice() didn't write anything");
+	else {
+		for (i = 0; i < written; i++) {
+			if (arr_read[i+offset] != buffer[i]) {
+				tst_res(TFAIL,
+					"Wrong data in user memory at %i", i);
+				break;
+			}
+		}
+		if (i == written)
+			tst_res(TPASS, "Spliced correctly into user memory");
+	}
+
+	SAFE_CLOSE(pipes[1]);
+	SAFE_CLOSE(pipes[0]);
+}
+
+static void setup(void)
+{
+	int i;
+
+	for (i = 0; i < TEST_BLOCK_SIZE; i++)
+		buffer[i] = i & 0xff;
+}
+
+static struct tst_test test = {
+	.setup = setup,
+	.test_all = vmsplice_test,
+	.min_kver = "2.6.23",
+};