File: its_sysfs.py

package info (click to toggle)
linux 6.18.5-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,742,096 kB
  • sloc: ansic: 26,781,576; asm: 272,087; sh: 148,750; python: 79,244; makefile: 57,741; perl: 36,527; xml: 19,542; cpp: 5,911; yacc: 4,939; lex: 2,950; awk: 1,607; sed: 30; ruby: 25
file content (65 lines) | stat: -rwxr-xr-x 2,391 bytes parent folder | download | duplicates (16)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2025 Intel Corporation
#
# Test for Indirect Target Selection(ITS) mitigation sysfs status.

import sys, os, re
this_dir = os.path.dirname(os.path.realpath(__file__))
sys.path.insert(0, this_dir + '/../../kselftest')
import ksft

from common import *

bug = "indirect_target_selection"
mitigation = get_sysfs(bug)

ITS_MITIGATION_ALIGNED_THUNKS	= "Mitigation: Aligned branch/return thunks"
ITS_MITIGATION_RETPOLINE_STUFF	= "Mitigation: Retpolines, Stuffing RSB"
ITS_MITIGATION_VMEXIT_ONLY		= "Mitigation: Vulnerable, KVM: Not affected"
ITS_MITIGATION_VULNERABLE       = "Vulnerable"

def check_mitigation():
    if mitigation == ITS_MITIGATION_ALIGNED_THUNKS:
        if cmdline_has(f'{bug}=stuff') and sysfs_has("spectre_v2", "Retpolines"):
            bug_check_fail(bug, ITS_MITIGATION_ALIGNED_THUNKS, ITS_MITIGATION_RETPOLINE_STUFF)
            return
        if cmdline_has(f'{bug}=vmexit') and cpuinfo_has('its_native_only'):
            bug_check_fail(bug, ITS_MITIGATION_ALIGNED_THUNKS, ITS_MITIGATION_VMEXIT_ONLY)
            return
        bug_check_pass(bug, ITS_MITIGATION_ALIGNED_THUNKS)
        return

    if mitigation == ITS_MITIGATION_RETPOLINE_STUFF:
        if cmdline_has(f'{bug}=stuff') and sysfs_has("spectre_v2", "Retpolines"):
            bug_check_pass(bug, ITS_MITIGATION_RETPOLINE_STUFF)
            return
        if sysfs_has('retbleed', 'Stuffing'):
            bug_check_pass(bug, ITS_MITIGATION_RETPOLINE_STUFF)
            return
        bug_check_fail(bug, ITS_MITIGATION_RETPOLINE_STUFF, ITS_MITIGATION_ALIGNED_THUNKS)

    if mitigation == ITS_MITIGATION_VMEXIT_ONLY:
        if cmdline_has(f'{bug}=vmexit') and cpuinfo_has('its_native_only'):
            bug_check_pass(bug, ITS_MITIGATION_VMEXIT_ONLY)
            return
        bug_check_fail(bug, ITS_MITIGATION_VMEXIT_ONLY, ITS_MITIGATION_ALIGNED_THUNKS)

    if mitigation == ITS_MITIGATION_VULNERABLE:
        if sysfs_has("spectre_v2", "Vulnerable"):
            bug_check_pass(bug, ITS_MITIGATION_VULNERABLE)
        else:
            bug_check_fail(bug, "Mitigation", ITS_MITIGATION_VULNERABLE)

    bug_status_unknown(bug, mitigation)
    return

ksft.print_header()
ksft.set_plan(1)
ksft.print_msg(f'{bug}: {mitigation} ...')

if not basic_checks_sufficient(bug, mitigation):
    check_mitigation()

ksft.finished()