diff mbox series

[v5] dio_truncate.c test refactory with LTP API

Message ID 20211119104130.26480-1-acervesato@suse.de
State Accepted
Headers show
Series [v5] dio_truncate.c test refactory with LTP API | expand

Commit Message

Andrea Cervesato Nov. 19, 2021, 10:41 a.m. UTC
From: Andrea Cervesato <andrea.cervesato@suse.com>

Signed-off-by: Andrea Cervesato <andrea.cervesato@suse.com>
---
 testcases/kernel/io/ltp-aiodio/dio_truncate.c | 206 +++++++++---------
 1 file changed, 102 insertions(+), 104 deletions(-)

Comments

Cyril Hrubis Nov. 19, 2021, 12:28 p.m. UTC | #1
Hi!
Pushed with minor changes, thanks.

Apart from simple formatting fixes the most complicated change was
removal of the loop that waits the children at the end of the run().
Since if any of the children returns prematurely the waitpid() for the
pid will fail and the test will exit with TBROK. The easiest solution
was simply to remove the loop since the children exit with 0 now the
library will collect them for us just fine.

Full diff:

diff --git a/testcases/kernel/io/ltp-aiodio/dio_truncate.c b/testcases/kernel/io/ltp-aiodio/dio_truncate.c
index 3681c9bdb..68c10ac7a 100644
--- a/testcases/kernel/io/ltp-aiodio/dio_truncate.c
+++ b/testcases/kernel/io/ltp-aiodio/dio_truncate.c
@@ -10,13 +10,17 @@
  *
  * This test is mixing direct I/O and truncate operations checking if they can
  * be used together at the same time. Multiple children are spawned to read a
- * file that is edited using DIO write/read operations. Algorithm:
+ * file that is written to using direct I/O and truncated in a loop.
+ *
+ * [Algorithm]
+ *
  * - Spawn multiple children which start to read on 'file'
  * - Parent start to fill and truncate 'file' many times with zero char when
  *   children are reading
  * - Parent start to fill and truncate a junk file many times with non-zero char
+ *
  * If no issues occur on direct IO/truncate operations and the file always
- * contains non-zero characters, test PASS. Otherwise, test will FAIL.
+ * contains zero characters, test PASS. Otherwise, test will FAIL.
  */
 
 #define _GNU_SOURCE
@@ -53,7 +57,8 @@ static char *check_zero(char *buf, int size)
 		buf++;
 		size--;
 	}
-	return 0; /* all zeros */
+
+	return 0;
 }
 
 static void dio_read(const char *filename, size_t bs)
@@ -67,19 +72,19 @@ static void dio_read(const char *filename, size_t bs)
 	while ((fd = open(filename, O_RDONLY | O_DIRECT, 0666)) < 0)
 		usleep(100);
 
-	tst_res(TINFO, "child reading file");
+	tst_res(TINFO, "child %i reading file", getpid());
 	while (*run_child) {
 		off_t offset;
 		char *bufoff;
 
-		/* read the file, checking for zeros */
 		offset = SAFE_LSEEK(fd, SEEK_SET, 0);
 		do {
 			r = read(fd, bufptr, 64 * 1024);
 			if (r > 0) {
 				bufoff = check_zero(bufptr, r);
 				if (bufoff) {
-					tst_res(TINFO, "non-zero read at offset %p", offset + bufoff);
+					tst_res(TINFO, "non-zero read at offset %zu",
+						offset + (bufoff - bufptr));
 					free(bufptr);
 					SAFE_CLOSE(fd);
 					return;
@@ -91,8 +96,6 @@ static void dio_read(const char *filename, size_t bs)
 
 	free(bufptr);
 	SAFE_CLOSE(fd);
-
-	tst_res(TPASS, "zero buffer only after truncate");
 }
 
 static void dio_append(const char *path, char pattern, size_t bs, size_t bcount)
@@ -128,7 +131,6 @@ static void run(void)
 	char *filename = "file";
 	int filesize = FILE_SIZE;
 	int num_children = NUM_CHILDREN;
-	int pid[NUM_CHILDREN];
 	int status;
 	int i;
 	int fail = 0;
@@ -136,13 +138,14 @@ static void run(void)
 	*run_child = 1;
 
 	for (i = 0; i < num_children; i++) {
-		pid[i] = SAFE_FORK();
-		if (pid[i] == 0) {
+		if (!SAFE_FORK()) {
 			dio_read(filename, filesize);
 			return;
 		}
 	}
 
+	tst_res(TINFO, "parent writes/truncates the file");
+
 	for (i = 0; i < 100; i++) {
 		dio_append(filename, 0, filesize, 100);
 		SAFE_TRUNCATE(filename, 0);
@@ -161,9 +164,6 @@ static void run(void)
 		tst_res(TPASS, "All bytes read were zeroed");
 
 	*run_child = 0;
-
-	for (i = 0; i < num_children; i++)
-		SAFE_WAITPID(pid[i], &status, 0);
 }
 
 static struct tst_test test = {
diff mbox series

Patch

diff --git a/testcases/kernel/io/ltp-aiodio/dio_truncate.c b/testcases/kernel/io/ltp-aiodio/dio_truncate.c
index 27cf01525..3681c9bdb 100644
--- a/testcases/kernel/io/ltp-aiodio/dio_truncate.c
+++ b/testcases/kernel/io/ltp-aiodio/dio_truncate.c
@@ -1,177 +1,175 @@ 
-
+// SPDX-License-Identifier: GPL-2.0-or-later
 /*
  * Copyright (c) 2004 Daniel McNeil <daniel@osdl.org>
- *               2004 Open Source Development Lab
- *   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 Street, Fifth Floor, Boston, MA 02110-1301 USA
- *
- * Module: .c
+ *				 2004 Open Source Development Lab
+ * Copyright (C) 2021 SUSE LLC Andrea Cervesato <andrea.cervesato@suse.com>
  */
 
-/*
- * Change History:
- *
- * 2/2004  Marty Ridgeway (mridge@us.ibm.com) Changes to adapt to LTP
+/*\
+ * [Description]
  *
+ * This test is mixing direct I/O and truncate operations checking if they can
+ * be used together at the same time. Multiple children are spawned to read a
+ * file that is edited using DIO write/read operations. Algorithm:
+ * - Spawn multiple children which start to read on 'file'
+ * - Parent start to fill and truncate 'file' many times with zero char when
+ *   children are reading
+ * - Parent start to fill and truncate a junk file many times with non-zero char
+ * If no issues occur on direct IO/truncate operations and the file always
+ * contains non-zero characters, test PASS. Otherwise, test will FAIL.
  */
+
 #define _GNU_SOURCE
 
 #include <stdlib.h>
+#include <stdio.h>
+#include <sys/stat.h>
 #include <sys/types.h>
-#include <signal.h>
-#include <errno.h>
 #include <fcntl.h>
-#include <stdio.h>
-#include <unistd.h>
-#include <memory.h>
-#include <string.h>
-#include <limits.h>
+#include "tst_test.h"
 
-#include "test.h"
+#define NUM_CHILDREN 16
+#define FILE_SIZE (64 * 1024)
 
-#define NUM_CHILDREN 8
+static int *run_child;
 
-char *check_zero(unsigned char *buf, int size)
+static char *check_zero(char *buf, int size)
 {
-	unsigned char *p;
+	char *p;
 
 	p = buf;
 
 	while (size > 0) {
 		if (*buf != 0) {
-			fprintf(stderr,
-				"non zero buffer at buf[%d] => 0x%02x,%02x,%02x,%02x\n",
+			tst_res(TINFO,
+				"non zero buffer at buf[%lu] => 0x%02x,%02x,%02x,%02x",
 				buf - p, (unsigned int)buf[0],
 				size > 1 ? (unsigned int)buf[1] : 0,
 				size > 2 ? (unsigned int)buf[2] : 0,
 				size > 3 ? (unsigned int)buf[3] : 0);
-			fprintf(stderr, "buf %p, p %p\n", buf, p);
+			tst_res(TINFO, "buf %p, p %p", buf, p);
 			return buf;
 		}
 		buf++;
 		size--;
 	}
-	return 0;		/* all zeros */
+	return 0; /* all zeros */
 }
 
-int dio_read(char *filename)
+static void dio_read(const char *filename, size_t bs)
 {
 	int fd;
 	int r;
-	void *bufptr = NULL;
+	char *bufptr;
 
-	TEST(posix_memalign(&bufptr, 4096, 64 * 1024));
-	if (TEST_RETURN) {
-		tst_resm(TBROK | TRERRNO, "cannot malloc aligned memory");
-		return -1;
-	}
+	bufptr = SAFE_MEMALIGN(getpagesize(), bs);
 
-	while ((fd = open(filename, O_DIRECT | O_RDONLY)) < 0) {
-	}
-	fprintf(stderr, "dio_truncate: child reading file\n");
-	while (1) {
+	while ((fd = open(filename, O_RDONLY | O_DIRECT, 0666)) < 0)
+		usleep(100);
+
+	tst_res(TINFO, "child reading file");
+	while (*run_child) {
 		off_t offset;
 		char *bufoff;
 
 		/* read the file, checking for zeros */
-		offset = lseek(fd, SEEK_SET, 0);
+		offset = SAFE_LSEEK(fd, SEEK_SET, 0);
 		do {
 			r = read(fd, bufptr, 64 * 1024);
 			if (r > 0) {
-				if ((bufoff = check_zero(bufptr, r))) {
-					fprintf(stderr,
-						"non-zero read at offset %p\n",
-						offset + bufoff);
-					exit(1);
+				bufoff = check_zero(bufptr, r);
+				if (bufoff) {
+					tst_res(TINFO, "non-zero read at offset %p", offset + bufoff);
+					free(bufptr);
+					SAFE_CLOSE(fd);
+					return;
 				}
 				offset += r;
 			}
 		} while (r > 0);
 	}
-	return 0;
+
+	free(bufptr);
+	SAFE_CLOSE(fd);
+
+	tst_res(TPASS, "zero buffer only after truncate");
 }
 
-void dio_append(char *filename, int fill)
+static void dio_append(const char *path, char pattern, size_t bs, size_t bcount)
 {
 	int fd;
-	void *bufptr = NULL;
-	int i;
-	int w;
+	size_t i;
+	char *bufptr;
 
-	fd = open(filename, O_DIRECT | O_WRONLY | O_CREAT, 0666);
+	bufptr = SAFE_MEMALIGN(getpagesize(), bs);
+	memset(bufptr, pattern, bs);
 
-	if (fd < 0) {
-		perror("cannot create file");
-		return;
-	}
+	fd = SAFE_OPEN(path, O_CREAT | O_WRONLY | O_DIRECT, 0666);
 
-	TEST(posix_memalign(&bufptr, 4096, 64 * 1024));
-	if (TEST_RETURN) {
-		tst_resm(TBROK | TRERRNO, "cannot malloc aligned memory");
-		close(fd);
-		return;
-	}
+	for (i = 0; i < bcount; i++)
+		SAFE_WRITE(1, fd, bufptr, bs);
 
-	memset(bufptr, fill, 64 * 1024);
+	free(bufptr);
+	SAFE_CLOSE(fd);
+}
 
-	for (i = 0; i < 1000; i++) {
-		if ((w = write(fd, bufptr, 64 * 1024)) != 64 * 1024) {
-			fprintf(stderr, "write %d returned %d\n", i, w);
-		}
-	}
-	close(fd);
+static void setup(void)
+{
+	run_child = SAFE_MMAP(NULL, sizeof(int), PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANONYMOUS, -1, 0);
+}
+
+static void cleanup(void)
+{
+	SAFE_MUNMAP(run_child, sizeof(int));
 }
 
-int main(void)
+static void run(void)
 {
-	char filename[PATH_MAX];
+	char *filename = "file";
+	int filesize = FILE_SIZE;
+	int num_children = NUM_CHILDREN;
 	int pid[NUM_CHILDREN];
-	int num_children = 1;
+	int status;
 	int i;
+	int fail = 0;
 
-	snprintf(filename, sizeof(filename), "%s/aiodio/file",
-		 getenv("TMP") ? getenv("TMP") : "/tmp");
+	*run_child = 1;
 
 	for (i = 0; i < num_children; i++) {
-		if ((pid[i] = fork()) == 0) {
-			/* child */
-			return dio_read(filename);
-		} else if (pid[i] < 0) {
-			/* error */
-			perror("fork error");
-			break;
-		} else {
-			/* Parent */
-			continue;
+		pid[i] = SAFE_FORK();
+		if (pid[i] == 0) {
+			dio_read(filename, filesize);
+			return;
 		}
 	}
 
-	/*
-	 * Parent creates a zero file using DIO.
-	 * Truncates it to zero
-	 * Create another file with '0xaa'
-	 */
 	for (i = 0; i < 100; i++) {
-		dio_append(filename, 0);
-		truncate(filename, 0);
-		dio_append("junkfile", 0xaa);
-		truncate("junkfile", 0);
-	}
+		dio_append(filename, 0, filesize, 100);
+		SAFE_TRUNCATE(filename, 0);
+		dio_append("junkfile", 0xaa, filesize, 100);
+		SAFE_TRUNCATE("junkfile", 0);
 
-	for (i = 0; i < num_children; i++) {
-		kill(pid[i], SIGTERM);
+		if (SAFE_WAITPID(-1, &status, WNOHANG)) {
+			fail = 1;
+			break;
+		}
 	}
 
-	return 0;
+	if (fail)
+		tst_res(TFAIL, "Non zero bytes read");
+	else
+		tst_res(TPASS, "All bytes read were zeroed");
+
+	*run_child = 0;
+
+	for (i = 0; i < num_children; i++)
+		SAFE_WAITPID(pid[i], &status, 0);
 }
+
+static struct tst_test test = {
+	.test_all = run,
+	.setup = setup,
+	.cleanup = cleanup,
+	.needs_tmpdir = 1,
+	.forks_child = 1,
+};