1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3#please run as root
4
5# Kselftest framework requirement - SKIP code is 4.
6ksft_skip=4
7
8mnt=./huge
9exitcode=0
10
11#get huge pagesize and freepages from /proc/meminfo
12while read -r name size unit; do
13	if [ "$name" = "HugePages_Free:" ]; then
14		freepgs="$size"
15	fi
16	if [ "$name" = "Hugepagesize:" ]; then
17		hpgsize_KB="$size"
18	fi
19done < /proc/meminfo
20
21# Simple hugetlbfs tests have a hardcoded minimum requirement of
22# huge pages totaling 256MB (262144KB) in size.  The userfaultfd
23# hugetlb test requires a minimum of 2 * nr_cpus huge pages.  Take
24# both of these requirements into account and attempt to increase
25# number of huge pages available.
26nr_cpus=$(nproc)
27hpgsize_MB=$((hpgsize_KB / 1024))
28half_ufd_size_MB=$((((nr_cpus * hpgsize_MB + 127) / 128) * 128))
29needmem_KB=$((half_ufd_size_MB * 2 * 1024))
30
31#set proper nr_hugepages
32if [ -n "$freepgs" ] && [ -n "$hpgsize_KB" ]; then
33	nr_hugepgs=$(cat /proc/sys/vm/nr_hugepages)
34	needpgs=$((needmem_KB / hpgsize_KB))
35	tries=2
36	while [ "$tries" -gt 0 ] && [ "$freepgs" -lt "$needpgs" ]; do
37		lackpgs=$((needpgs - freepgs))
38		echo 3 > /proc/sys/vm/drop_caches
39		if ! echo $((lackpgs + nr_hugepgs)) > /proc/sys/vm/nr_hugepages; then
40			echo "Please run this test as root"
41			exit $ksft_skip
42		fi
43		while read -r name size unit; do
44			if [ "$name" = "HugePages_Free:" ]; then
45				freepgs=$size
46			fi
47		done < /proc/meminfo
48		tries=$((tries - 1))
49	done
50	if [ "$freepgs" -lt "$needpgs" ]; then
51		printf "Not enough huge pages available (%d < %d)\n" \
52		       "$freepgs" "$needpgs"
53		exit 1
54	fi
55else
56	echo "no hugetlbfs support in kernel?"
57	exit 1
58fi
59
60#filter 64bit architectures
61ARCH64STR="arm64 ia64 mips64 parisc64 ppc64 ppc64le riscv64 s390x sh64 sparc64 x86_64"
62if [ -z "$ARCH" ]; then
63	ARCH=$(uname -m 2>/dev/null | sed -e 's/aarch64.*/arm64/')
64fi
65VADDR64=0
66echo "$ARCH64STR" | grep "$ARCH" && VADDR64=1
67
68# Usage: run_test [test binary] [arbitrary test arguments...]
69run_test() {
70	local title="running $*"
71	local sep=$(echo -n "$title" | tr "[:graph:][:space:]" -)
72	printf "%s\n%s\n%s\n" "$sep" "$title" "$sep"
73
74	"$@"
75	local ret=$?
76	if [ $ret -eq 0 ]; then
77		echo "[PASS]"
78	elif [ $ret -eq $ksft_skip ]; then
79		echo "[SKIP]"
80		exitcode=$ksft_skip
81	else
82		echo "[FAIL]"
83		exitcode=1
84	fi
85}
86
87mkdir "$mnt"
88mount -t hugetlbfs none "$mnt"
89
90run_test ./hugepage-mmap
91
92shmmax=$(cat /proc/sys/kernel/shmmax)
93shmall=$(cat /proc/sys/kernel/shmall)
94echo 268435456 > /proc/sys/kernel/shmmax
95echo 4194304 > /proc/sys/kernel/shmall
96run_test ./hugepage-shm
97echo "$shmmax" > /proc/sys/kernel/shmmax
98echo "$shmall" > /proc/sys/kernel/shmall
99
100run_test ./map_hugetlb
101
102run_test ./hugepage-mremap "$mnt"/huge_mremap
103rm -f "$mnt"/huge_mremap
104
105run_test ./hugepage-vmemmap
106
107run_test ./hugetlb-madvise "$mnt"/madvise-test
108rm -f "$mnt"/madvise-test
109
110echo "NOTE: The above hugetlb tests provide minimal coverage.  Use"
111echo "      https://github.com/libhugetlbfs/libhugetlbfs.git for"
112echo "      hugetlb regression testing."
113
114run_test ./map_fixed_noreplace
115
116# get_user_pages_fast() benchmark
117run_test ./gup_test -u
118# pin_user_pages_fast() benchmark
119run_test ./gup_test -a
120# Dump pages 0, 19, and 4096, using pin_user_pages:
121run_test ./gup_test -ct -F 0x1 0 19 0x1000
122
123uffd_mods=("" ":dev")
124for mod in "${uffd_mods[@]}"; do
125	run_test ./userfaultfd anon${mod} 20 16
126	# Hugetlb tests require source and destination huge pages. Pass in half
127	# the size ($half_ufd_size_MB), which is used for *each*.
128	run_test ./userfaultfd hugetlb${mod} "$half_ufd_size_MB" 32
129	run_test ./userfaultfd hugetlb_shared${mod} "$half_ufd_size_MB" 32 "$mnt"/uffd-test
130	rm -f "$mnt"/uffd-test
131	run_test ./userfaultfd shmem${mod} 20 16
132done
133
134#cleanup
135umount "$mnt"
136rm -rf "$mnt"
137echo "$nr_hugepgs" > /proc/sys/vm/nr_hugepages
138
139run_test ./compaction_test
140
141run_test sudo -u nobody ./on-fault-limit
142
143run_test ./map_populate
144
145run_test ./mlock-random-test
146
147run_test ./mlock2-tests
148
149run_test ./mrelease_test
150
151run_test ./mremap_test
152
153run_test ./thuge-gen
154
155if [ $VADDR64 -ne 0 ]; then
156	run_test ./virtual_address_range
157
158	# virtual address 128TB switch test
159	run_test ./va_128TBswitch.sh
160fi # VADDR64
161
162# vmalloc stability smoke test
163run_test ./test_vmalloc.sh smoke
164
165run_test ./mremap_dontunmap
166
167run_test ./test_hmm.sh smoke
168
169# MADV_POPULATE_READ and MADV_POPULATE_WRITE tests
170run_test ./madv_populate
171
172run_test ./memfd_secret
173
174# KSM MADV_MERGEABLE test with 10 identical pages
175run_test ./ksm_tests -M -p 10
176# KSM unmerge test
177run_test ./ksm_tests -U
178# KSM test with 10 zero pages and use_zero_pages = 0
179run_test ./ksm_tests -Z -p 10 -z 0
180# KSM test with 10 zero pages and use_zero_pages = 1
181run_test ./ksm_tests -Z -p 10 -z 1
182# KSM test with 2 NUMA nodes and merge_across_nodes = 1
183run_test ./ksm_tests -N -m 1
184# KSM test with 2 NUMA nodes and merge_across_nodes = 0
185run_test ./ksm_tests -N -m 0
186
187# protection_keys tests
188if [ -x ./protection_keys_32 ]
189then
190	run_test ./protection_keys_32
191fi
192
193if [ -x ./protection_keys_64 ]
194then
195	run_test ./protection_keys_64
196fi
197
198run_test ./soft-dirty
199
200exit $exitcode
201