Small fixes in security tests

Signed-off-by: Katarzyna Treder <katarzyna.treder@h-partners.com>
This commit is contained in:
Katarzyna Treder
2024-10-07 13:21:44 +02:00
parent e7f14f7d00
commit 20ee2fda1f
16 changed files with 78 additions and 13 deletions

View File

@@ -20,6 +20,7 @@ from api.cas.cli import add_core_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
get_fuzz_config,
run_cmd_and_validate,
@@ -68,6 +69,9 @@ def test_fuzzy_add_core_cache_id(
)
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = [str(cache.cache_id).encode("ascii")]
fuzz_config = get_fuzz_config("cache_id.yml")

View File

@@ -24,6 +24,7 @@ from tests.security.fuzzy.kernel.common.common import (
get_fuzz_config,
run_cmd_and_validate,
)
from test_utils.os_utils import Udev
core_id_min = 0
core_id_max = 4095
@@ -68,6 +69,9 @@ def test_fuzzy_add_core_core_id(
)
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
fuzz_config = get_fuzz_config("core_id.yml")
PeachFuzzer.generate_config(fuzz_config)

View File

@@ -20,6 +20,7 @@ from api.cas.cli import add_core_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
run_cmd_and_validate,
get_device_fuzz_config,
@@ -67,6 +68,9 @@ def test_fuzzy_add_core_device(
)
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = [
disk.path

View File

@@ -9,6 +9,7 @@ from api.cas.cli import script_try_add_cmd, remove_detached_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskTypeSet, DiskType
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
run_cmd_and_validate,
get_fuzz_config,
@@ -41,6 +42,9 @@ def test_fuzzy_script_add_core_try_add_cache_id():
command_template=base_cmd, count=TestRun.usr.fuzzy_iter_count
)
with TestRun.step("Disable udev"):
Udev.disable()
for index, cmd in TestRun.iteration(
enumerate(commands), f"Run command {TestRun.usr.fuzzy_iter_count} times"
):

View File

@@ -9,6 +9,7 @@ from api.cas.cli import script_try_add_cmd, remove_detached_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskTypeSet, DiskType
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
run_cmd_and_validate,
get_fuzz_config,
@@ -43,6 +44,9 @@ def test_fuzzy_script_add_core_try_add_core_id():
command_template=base_cmd, count=TestRun.usr.fuzzy_iter_count
)
with TestRun.step("Disable udev"):
Udev.disable()
for index, cmd in TestRun.iteration(
enumerate(commands), f"Run command {TestRun.usr.fuzzy_iter_count} times"
):

View File

@@ -7,6 +7,7 @@
from api.cas.cli import script_try_add_cmd, remove_detached_cmd
from core.test_run import TestRun
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
run_cmd_and_validate,
get_device_fuzz_config,
@@ -36,6 +37,9 @@ def test_fuzzy_script_add_core_try_add_core_device():
command_template=base_cmd, count=TestRun.usr.fuzzy_iter_count
)
with TestRun.step("Disable udev"):
Udev.disable()
for index, cmd in TestRun.iteration(
enumerate(commands), f"Run command {TestRun.usr.fuzzy_iter_count} times"
):

View File

@@ -19,6 +19,7 @@ from api.cas.cli import start_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from test_utils.size import Unit, Size
from tests.security.fuzzy.kernel.common.common import (
get_fuzz_config,
@@ -49,6 +50,9 @@ def test_fuzzy_start_cache_flags(cache_mode, cache_line_size, unaligned_io, use_
cache_disk = TestRun.disks["cache"]
cache_disk.create_partitions([Size(400, Unit.MebiByte)])
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Start and stop cache"):
# Reload kernel modules
cache = casadm.start_cache(
@@ -93,7 +97,7 @@ def test_fuzzy_start_cache_flags(cache_mode, cache_line_size, unaligned_io, use_
any_alphanumeric_pattern = r"\w+"
base_cmd = re.sub(
pattern=f"{incompatible_param} {any_alphanumeric_pattern}",
sub="",
repl="",
string=base_cmd,
)
base_cmd = f"{base_cmd.strip()} {param}"

View File

@@ -18,6 +18,7 @@ from api.cas.cli import remove_core_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -56,6 +57,9 @@ def test_fuzzy_remove_core_cache_id(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = [str(core.cache_id).encode("ascii")]
PeachFuzzer.generate_config(get_fuzz_config("cache_id.yml"))

View File

@@ -18,6 +18,7 @@ from api.cas.cli import remove_core_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -56,6 +57,9 @@ def test_fuzzy_remove_core_core_id(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = [str(core.core_id).encode("ascii")]
PeachFuzzer.generate_config(get_fuzz_config("core_id.yml"))

View File

@@ -17,6 +17,7 @@ from api.cas.cli import remove_core_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -55,6 +56,9 @@ def test_fuzzy_remove_core_flag(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = ["", "-f", "--force"]
valid_values = [v.encode("ascii") for v in valid_values]

View File

@@ -20,6 +20,7 @@ from api.cas.init_config import InitConfig
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -58,6 +59,9 @@ def test_fuzzy_remove_inactive_cache_id(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Create init config from running configuration."):
InitConfig.create_init_config_from_running_configuration()
@@ -93,3 +97,6 @@ def test_fuzzy_remove_inactive_cache_id(
cache.stop(no_data_flush=True)
core_disk.unplug()
casadm.load_cache(device=cache_disk.partitions[0])
with TestRun.step("Plug core device"):
core_disk.plug_all()

View File

@@ -19,6 +19,7 @@ from api.cas.init_config import InitConfig
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -57,6 +58,9 @@ def test_fuzzy_remove_inactive_core_id(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Create init config from running configuration."):
InitConfig.create_init_config_from_running_configuration()
@@ -92,3 +96,6 @@ def test_fuzzy_remove_inactive_core_id(
cache.stop(no_data_flush=True)
core_disk.unplug()
casadm.load_cache(device=cache_disk.partitions[0])
with TestRun.step("Plug core device"):
core_disk.plug_all()

View File

@@ -18,6 +18,7 @@ from api.cas.cli import stop_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -56,6 +57,9 @@ def test_fuzzy_stop_cache_cache_id(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = [str(core.cache_id).encode("ascii")]
PeachFuzzer.generate_config(get_fuzz_config("cache_id.yml"))

View File

@@ -18,6 +18,7 @@ from api.cas.cli import stop_cmd
from core.test_run import TestRun
from storage_devices.disk import DiskType, DiskTypeSet, DiskTypeLowerThan
from test_tools.peach_fuzzer.peach_fuzzer import PeachFuzzer
from test_utils.os_utils import Udev
from tests.security.fuzzy.kernel.common.common import (
prepare_cas_instance,
get_fuzz_config,
@@ -56,6 +57,9 @@ def test_fuzzy_stop_cache_flag(
cleaning_policy=cleaning_policy,
)
with TestRun.step("Disable udev"):
Udev.disable()
with TestRun.step("Prepare PeachFuzzer"):
valid_values = ["", "-n", "--no-data-flush"]
valid_values = [v.encode("ascii") for v in valid_values]

View File

@@ -87,7 +87,7 @@ def test_fuzzy_get_param_name(cache_mode, cache_line_size, unaligned_io, use_io_
if param == str(ParamName.seq_cutoff):
cmd += f" --core-id {core.core_id}"
cmd = base_cmd.replace("{param}", param)
cmd = cmd.replace("{param}", param)
run_cmd_and_validate(
cmd=get_cmd(cmd, param.encode("ascii")),