File: test_rokmod.py

package info (click to toggle)
setools 4.6.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,600 kB
  • sloc: python: 24,485; makefile: 14
file content (99 lines) | stat: -rw-r--r-- 4,232 bytes parent folder | download
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# Copyright 2020, 2025, Microsoft Corporation
#
# SPDX-License-Identifier: GPL-2.0-only
#
import os

import pytest
import setools


@pytest.mark.obj_args("tests/library/checker/rokmod.conf")
class TestReadOnlyKernelModules:

    def test_invalid_option(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test invalid option"""
        with pytest.raises(setools.exception.InvalidCheckOption):
            config = {"INVALID": "option"}
            check = setools.checker.rokmod.ReadOnlyKernelModules(
                compiled_policy, "test_invalid_option", config)

    def test_all_mods(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test all modules are returned for no-option test.."""
        config = dict[str, str]()
        check = setools.checker.rokmod.ReadOnlyKernelModules(
            compiled_policy, "test_all_mods", config)
        result = check._collect_kernel_mods()

        # because of unconfined, nonkmod is loadable
        expected = set(("rokmod", "kmodfile1", "kmodfile2", "nonkmod", "exempt_file"))
        assert expected == set(result.keys())

    def test_exempt_load_domain(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for exempting a loading domain."""
        config = {"exempt_load_domain": "unconfined"}
        check = setools.checker.rokmod.ReadOnlyKernelModules(
            compiled_policy, "test_exempt_load_domain", config)
        result = check._collect_kernel_mods()

        expected = set(("kmodfile1", "kmodfile2", "rokmod"))
        assert expected == set(result.keys())

    def test_exempt_file(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for exempting a file."""
        config = {"exempt_file": "exempt_file"}
        check = setools.checker.rokmod.ReadOnlyKernelModules(
            compiled_policy, "test_exempt_file", config)
        result = check._collect_kernel_mods()

        expected = set(("rokmod", "kmodfile1", "kmodfile2", "nonkmod"))
        assert expected == result.keys()

    def test_exempt_file_attr(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for exempting a file by attribute."""
        config = {"exempt_file": "exempt_files_attr"}
        check = setools.checker.rokmod.ReadOnlyKernelModules(
            compiled_policy, "test_exempt_file_attr", config)
        result = check._collect_kernel_mods()

        expected = set(("rokmod", "nonkmod", "exempt_file"))
        assert expected == result.keys()

    def test_fail(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for failing."""
        with open(os.devnull, "w", encoding="utf-8") as fd:
            config = {"exempt_load_domain": "unconfined",
                      "exempt_write_domain": "unconfined"}
            check = setools.checker.rokmod.ReadOnlyKernelModules(
                compiled_policy, "test_fail", config)
            check.output = fd
            result = check.run()

            expected = [compiled_policy.lookup_type("kmodfile1"),
                        compiled_policy.lookup_type("kmodfile2")]
            assert expected == result

    def test_pass(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for passing."""
        with open(os.devnull, "w", encoding="utf-8") as fd:
            config = {"exempt_load_domain": "unconfined",
                      "exempt_write_domain": "domain1  domain2  unconfined"}
            check = setools.checker.rokmod.ReadOnlyKernelModules(
                compiled_policy, "test_pass", config)
            check.output = fd
            result = check.run()

            assert not result

    def test_pass2(self, compiled_policy: setools.SELinuxPolicy) -> None:
        """Test for passing with alternate exemptions."""
        with open(os.devnull, "w", encoding="utf-8") as fd:
            config = {"exempt_load_domain": "unconfined",
                      "exempt_file": "kmodfile2",
                      "exempt_write_domain": "domain1  unconfined"}
            check = setools.checker.rokmod.ReadOnlyKernelModules(
                compiled_policy, "test_pass2", config)
            check.output = fd
            result = check.run()

            assert not result