1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3#
4# Copyright (C) 2022 Adam Sindelar (Meta) <adam@wowsignal.io>
5#
6# This is a test for mmap behavior with 5-level paging. This script wraps the
7# real test to check that the kernel is configured to support at least 5
8# pagetable levels.
9
10# 1 means the test failed
11exitcode=1
12
13# Kselftest framework requirement - SKIP code is 4.
14ksft_skip=4
15
16fail()
17{
18	echo "$1"
19	exit $exitcode
20}
21
22check_supported_x86_64()
23{
24	local config="/proc/config.gz"
25	[[ -f "${config}" ]] || config="/boot/config-$(uname -r)"
26	[[ -f "${config}" ]] || fail "Cannot find kernel config in /proc or /boot"
27
28	# gzip -dcfq automatically handles both compressed and plaintext input.
29	# See man 1 gzip under '-f'.
30	local pg_table_levels=$(gzip -dcfq "${config}" | grep PGTABLE_LEVELS | cut -d'=' -f 2)
31
32	local cpu_supports_pl5=$(awk '/^flags/ {if (/la57/) {print 0;}
33		else {print 1}; exit}' /proc/cpuinfo 2>/dev/null)
34
35	if [[ "${pg_table_levels}" -lt 5 ]]; then
36		echo "$0: PGTABLE_LEVELS=${pg_table_levels}, must be >= 5 to run this test"
37		exit $ksft_skip
38	elif [[ "${cpu_supports_pl5}" -ne 0 ]]; then
39		echo "$0: CPU does not have the necessary la57 flag to support page table level 5"
40		exit $ksft_skip
41	fi
42}
43
44check_supported_ppc64()
45{
46	local config="/proc/config.gz"
47	[[ -f "${config}" ]] || config="/boot/config-$(uname -r)"
48	[[ -f "${config}" ]] || fail "Cannot find kernel config in /proc or /boot"
49
50	local pg_table_levels=$(gzip -dcfq "${config}" | grep PGTABLE_LEVELS | cut -d'=' -f 2)
51	if [[ "${pg_table_levels}" -lt 5 ]]; then
52		echo "$0: PGTABLE_LEVELS=${pg_table_levels}, must be >= 5 to run this test"
53		exit $ksft_skip
54	fi
55
56	local mmu_support=$(grep -m1 "mmu" /proc/cpuinfo | awk '{print $3}')
57	if [[ "$mmu_support" != "radix" ]]; then
58		echo "$0: System does not use Radix MMU, required for 5-level paging"
59		exit $ksft_skip
60	fi
61
62	local hugepages_total=$(awk '/HugePages_Total/ {print $2}' /proc/meminfo)
63	if [[ "${hugepages_total}" -eq 0 ]]; then
64		echo "$0: HugePages are not enabled, required for some tests"
65		exit $ksft_skip
66	fi
67}
68
69check_test_requirements()
70{
71	# The test supports x86_64 and powerpc64. We currently have no useful
72	# eligibility check for powerpc64, and the test itself will reject other
73	# architectures.
74	case `uname -m` in
75		"x86_64")
76			check_supported_x86_64
77		;;
78		"ppc64le"|"ppc64")
79			check_supported_ppc64
80		;;
81		*)
82			return 0
83		;;
84	esac
85}
86
87check_test_requirements
88./va_high_addr_switch --run-hugetlb
89