From: Masatake YAMATO <yamato@redhat.com>
Date: Mon, 29 Jan 2024 16:02:07 +0000 (+0900)
Subject: tests: (test_mkfds::mmap) new factory
X-Git-Tag: v2.42-start~545^2~1
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=4436e79547986529c3ac34151399df9d1448308e;p=thirdparty%2Futil-linux.git

tests: (test_mkfds::mmap) new factory

Signed-off-by: Masatake YAMATO <yamato@redhat.com>
---

diff --git a/tests/helpers/test_mkfds.c b/tests/helpers/test_mkfds.c
index 642f8ec9a6..b0d34c6050 100644
--- a/tests/helpers/test_mkfds.c
+++ b/tests/helpers/test_mkfds.c
@@ -56,6 +56,7 @@
 #include <sys/signalfd.h>
 #include <sys/socket.h>
 #include <sys/shm.h>
+#include <sys/stat.h>
 #include <sys/syscall.h>
 #include <sys/timerfd.h>
 #include <sys/types.h>
@@ -3167,6 +3168,51 @@ static void free_pty(const struct factory * factory _U_, void *data)
 	free(data);
 }
 
+struct mmap_data {
+	char *addr;
+	size_t len;
+};
+
+static void *make_mmap(const struct factory *factory, struct fdesc fdescs[] _U_,
+		       int argc, char ** argv)
+{
+	struct arg file = decode_arg("file", factory->params, argc, argv);
+	const char *sfile = ARG_STRING(file);
+
+	int fd = open(sfile, O_RDONLY);
+	if (fd < 0)
+		err(EXIT_FAILURE, "failed in opening %s", sfile);
+	free_arg(&file);
+
+	struct stat sb;
+	if (fstat(fd, &sb) < 0) {
+		int e = errno;
+		close(fd);
+		errno = e;
+		err(EXIT_FAILURE, "failed in fstat()");
+	}
+	char *addr = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
+	if (addr == MAP_FAILED) {
+		int e = errno;
+		close(fd);
+		errno = e;
+		err(EXIT_FAILURE, "failed in mmap()");
+	}
+	close(fd);
+
+	struct mmap_data *data = xmalloc(sizeof(*data));
+	data->addr = addr;
+	data->len = sb.st_size;
+
+	return data;
+}
+
+static void free_mmap(const struct factory * factory _U_, void *data)
+{
+	munmap(((struct mmap_data *)data)->addr,
+	       ((struct mmap_data *)data)->len);
+}
+
 #define PARAM_END { .name = NULL, }
 static const struct factory factories[] = {
 	{
@@ -3999,6 +4045,24 @@ static const struct factory factories[] = {
 			PARAM_END
 		}
 	},
+	{
+		.name = "mmap",
+		.desc = "do mmap the given file",
+		.priv = false,
+		.N    = 0,
+		.EX_N = 0,
+		.make = make_mmap,
+		.free = free_mmap,
+		.params = (struct parameter []) {
+			{
+				.name = "file",
+				.type = PTYPE_STRING,
+				.desc = "file to be opened",
+				.defv.string = "/etc/passwd",
+			},
+			PARAM_END
+		},
+	},
 };
 
 static int count_parameters(const struct factory *factory)