[LTP] [PATCH 19/29] Hugetlb: Migrating libhugetlbfs mmap-gettest

[Tarun Sahu]

Migrating the libhugetlbfs/testcases/mmap-gettest.c test

Test Description: This baseline test validates that a mapping of a
certain size can be created, correctly.  Once created, all the pages are
filled with a pattern and rechecked to test for corruption. The mapping is
then released.  This process is repeated for a specified number of
iterations.

Signed-off-by: Tarun Sahu <tsahu@linux.ibm.com>
---
 runtest/hugetlb                               |   1 +
 testcases/kernel/mem/.gitignore               |   1 +
 .../kernel/mem/hugetlb/hugemmap/hugemmap22.c  | 170 ++++++++++++++++++
 3 files changed, 172 insertions(+)
 create mode 100644 testcases/kernel/mem/hugetlb/hugemmap/hugemmap22.c

diff --git a/runtest/hugetlb b/runtest/hugetlb
index 449fad56a..5d06c8679 100644
--- a/runtest/hugetlb
+++ b/runtest/hugetlb
@@ -23,6 +23,7 @@ hugemmap18 hugemmap18
 hugemmap19 hugemmap19
 hugemmap20 hugemmap20
 hugemmap21 hugemmap21 -T 2 -s 5
+hugemmap22 hugemmap22 -T 2 -s 5
 hugemmap05_1 hugemmap05 -m
 hugemmap05_2 hugemmap05 -s
 hugemmap05_3 hugemmap05 -s -m
diff --git a/testcases/kernel/mem/.gitignore b/testcases/kernel/mem/.gitignore
index 74edfa392..0fd01dbce 100644
--- a/testcases/kernel/mem/.gitignore
+++ b/testcases/kernel/mem/.gitignore
@@ -22,6 +22,7 @@
 /hugetlb/hugemmap/hugemmap19
 /hugetlb/hugemmap/hugemmap20
 /hugetlb/hugemmap/hugemmap21
+/hugetlb/hugemmap/hugemmap22
 /hugetlb/hugeshmat/hugeshmat01
 /hugetlb/hugeshmat/hugeshmat02
 /hugetlb/hugeshmat/hugeshmat03
diff --git a/testcases/kernel/mem/hugetlb/hugemmap/hugemmap22.c b/testcases/kernel/mem/hugetlb/hugemmap/hugemmap22.c
new file mode 100644
index 000000000..5bc7b9389
--- /dev/null
+++ b/testcases/kernel/mem/hugetlb/hugemmap/hugemmap22.c
@@ -0,0 +1,170 @@
+// SPDX-License-Identifier: LGPL-2.1-or-later
+/*
+ * Copyright (C) 2005-2006 David Gibson & Adam Litke, IBM Corporation.
+ *
+ * Test Name: mmap gettest
+ *
+ * Test Description: This baseline test validates that a mapping of a
+ * certain size can be created, correctly.  Once created, all the pages are
+ * filled with a pattern and rechecked to test for corruption. The mapping is
+ * then released.  This process is repeated for a specified number of
+ * iterations.
+ *
+ * HISTORY
+ *  Written by David Gibson & Adam Litke
+ *  Migrated from libhugetlbfs by Tarun Sahu
+ *
+ */
+
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <sys/mount.h>
+#include <limits.h>
+#include <sys/param.h>
+#include <sys/types.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/shm.h>
+#include <sys/wait.h>
+
+#include "hugetlb.h"
+
+#define BUF_SZ 256
+
+/* Setup Configuration */
+static unsigned long nr_hugepages;	/* Number of huge pages to allocate */
+static char *iterations_opt;
+static int iter;
+static char *verbose;
+static char hfile[MAXPATHLEN];
+static unsigned long hpage_size;
+static int fha = -1;
+
+static int do_one(size_t size)
+{
+	char *ma;
+	size_t i, j;
+	char pattern = 'A';
+
+	fha = SAFE_OPEN(hfile, O_RDWR | O_CREAT, 0600);
+	SAFE_UNLINK(hfile);
+
+	/* Map the files with MAP_PRIVATE */
+	ma = SAFE_MMAP(NULL, size, (PROT_READ|PROT_WRITE), MAP_SHARED, fha, 0);
+
+	/* Make sure the page is zeroed */
+	for (i = 0; i < nr_hugepages; i++) {
+		if (verbose)
+			tst_res(TINFO, "Verifying %p\n", (ma+(i*hpage_size)));
+		for (j = 0; j < hpage_size; j++) {
+			if (*(ma+(i*hpage_size)+j) != 0) {
+				tst_res(TFAIL, "Verifying the mmap area failed. "
+				     "Got %c, expected 0",
+				     *(ma+(i*hpage_size)+j));
+				goto fail;
+			}
+		}
+	}
+	/* Fill each file with a pattern */
+	for (i = 0; i < nr_hugepages; i++) {
+		pattern = 65+(i%26);
+		if (verbose)
+			tst_res(TINFO, "Touching %p with %c\n", ma+(i*hpage_size), pattern);
+		memset(ma+(i*hpage_size), pattern, hpage_size);
+	}
+
+	/* Verify the pattern */
+	for (i = 0; i < nr_hugepages; i++) {
+		pattern = 65+(i%26);
+		if (verbose)
+			tst_res(TINFO, "Verifying %p\n", (ma+(i*hpage_size)));
+		for (j = 0; j < hpage_size; j++) {
+			if (*(ma+(i*hpage_size)+j) != pattern) {
+				tst_res(TFAIL, "Verifying the mmap area failed. "
+				     "Got %c, expected %c",
+				     *(ma+(i*hpage_size)+j), pattern);
+				goto fail;
+			}
+		}
+	}
+
+	/* Munmap the area */
+	SAFE_MUNMAP(ma, size);
+
+	/* Close and delete the file */
+	SAFE_CLOSE(fha);
+	return 0;
+fail:
+	tst_brk(TBROK, "Once failed, No point in continuing the test");
+	return 0;
+}
+
+static void run_test(void)
+{
+	size_t size;
+	int i;
+
+	size = nr_hugepages * hpage_size;
+
+	for (i = 0; i < iter; i++) {
+		if (verbose)
+			tst_res(TINFO, "Iteration %d\n", i);
+		do_one(size);
+	}
+
+	tst_res(TPASS, "Successful");
+}
+
+static void setup(void)
+{
+	int nr_hpages, ret;
+
+	if (tst_hugepages == 0)
+		tst_brk(TCONF, "Not enough hugepages for testing.");
+
+	if (!Hopt)
+		Hopt = tst_get_tmpdir();
+	SAFE_MOUNT("none", Hopt, "hugetlbfs", 0, NULL);
+
+	snprintf(hfile, sizeof(hfile), "%s/ltp_mmapfile%d", Hopt, getpid());
+
+	hpage_size = SAFE_READ_MEMINFO("Hugepagesize:")*1024;
+
+	if (!(iterations_opt) || !(nr_opt))
+		tst_brk(TCONF, "Usage: -T <# iterations> -s <# pages>\n");
+
+	ret = tst_parse_int(iterations_opt, &iter, 1, INT_MAX);
+	if (ret || iter <= 0)
+		tst_brk(TCONF, "Invalid iteration argument");
+	ret = tst_parse_int(nr_opt, &nr_hpages, 1, INT_MAX);
+	if (ret || nr_hpages <= 0)
+		tst_brk(TCONF, "Invalid pages argument");
+	nr_hugepages = nr_hpages;
+}
+
+static void cleanup(void)
+{
+	if (fha >= 0)
+		SAFE_CLOSE(fha);
+	umount2(Hopt, MNT_DETACH);
+}
+
+static struct tst_test test = {
+	.needs_root = 1,
+	.needs_tmpdir = 1,
+	.options = (struct tst_option[]) {
+		{"v", &verbose, "Turns on verbose mode"},
+		{"T:", &iterations_opt, "Number of Iterations"},
+		{"H:", &Hopt,   "Location of hugetlbfs, i.e.  -H /var/hugetlbfs"},
+		{"s:", &nr_opt, "Set the number of the been allocated hugepages"},
+		{}
+	},
+	.setup = setup,
+	.cleanup = cleanup,
+	.test_all = run_test,
+	.hugepages = {2, TST_REQUEST},
+};
-- 
2.31.1