1 // SPDX-License-Identifier: LGPL-2.1-or-later
2 /*
3 * Copyright (C) 2005-2006 David Gibson & Adam Litke, IBM Corporation.
4 * Author: David Gibson & Adam Litke
5 */
6
7 /*\
8 * [Description]
9 *
10 * The test checks that mlocking hugetlb areas works with all combinations
11 * of MAP_PRIVATE and MAP_SHARED with and without MAP_LOCKED specified.
12 */
13
14 #include "hugetlb.h"
15
16 #define MNTPOINT "hugetlbfs/"
17
18 static int fd = -1;
19 static unsigned long hpage_size;
20
21 static struct tcase {
22 int flags;
23 char *flags_str;
24 } tcases[] = {
25 {MAP_PRIVATE, "MAP_PRIVATE"},
26 {MAP_SHARED, "MAP_SHARED"},
27 {MAP_PRIVATE | MAP_LOCKED, "MAP_PRIVATE | MAP_LOCKED"},
28 {MAP_SHARED | MAP_LOCKED, "MAP_SHARED | MAP_LOCKED"},
29 };
30
run_test(unsigned int i)31 static void run_test(unsigned int i)
32 {
33 int ret;
34 void *p;
35 struct tcase *tc = &tcases[i];
36
37 fd = tst_creat_unlinked(MNTPOINT, 0);
38 p = SAFE_MMAP(0, hpage_size, PROT_READ|PROT_WRITE, tc->flags, fd, 0);
39
40 ret = mlock(p, hpage_size);
41 if (ret) {
42 tst_res(TFAIL|TERRNO, "mlock() failed (flags %s)", tc->flags_str);
43 goto cleanup;
44 }
45
46 ret = munlock(p, hpage_size);
47 if (ret)
48 tst_res(TFAIL|TERRNO, "munlock() failed (flags %s)", tc->flags_str);
49 else
50 tst_res(TPASS, "mlock/munlock with %s hugetlb mmap", tc->flags_str);
51
52 cleanup:
53 SAFE_MUNMAP(p, hpage_size);
54 SAFE_CLOSE(fd);
55 }
56
setup(void)57 static void setup(void)
58 {
59 struct rlimit limit_info;
60
61 hpage_size = tst_get_hugepage_size();
62
63 SAFE_GETRLIMIT(RLIMIT_MEMLOCK, &limit_info);
64 if (limit_info.rlim_cur < hpage_size) {
65 limit_info.rlim_max = hpage_size;
66 limit_info.rlim_cur = hpage_size;
67 SAFE_SETRLIMIT(RLIMIT_MEMLOCK, &limit_info);
68 }
69 }
70
cleanup(void)71 static void cleanup(void)
72 {
73 if (fd >= 0)
74 SAFE_CLOSE(fd);
75 }
76
77 static struct tst_test test = {
78 .tcnt = ARRAY_SIZE(tcases),
79 .needs_root = 1,
80 .mntpoint = MNTPOINT,
81 .needs_hugetlbfs = 1,
82 .needs_tmpdir = 1,
83 .setup = setup,
84 .cleanup = cleanup,
85 .test = run_test,
86 .hugepages = {1, TST_NEEDS},
87 };
88