From 936390a108d21abba8f41e63e5fd7cccd6bec187 Mon Sep 17 00:00:00 2001
From: mcasquer <mcasquer@redhat.com>
Date: Tue, 15 Oct 2024 11:17:15 +0200
Subject: [PATCH] vm_poisoned_migration_prevention: adds new test case

Creates a new test case that checks the migration prevention
when there's poisoned memory.

Signed-off-by: mcasquer <mcasquer@redhat.com>
---
 .../cfg/vm_poisoned_migration_prevention.cfg  | 16 ++++++
 .../tests/vm_poisoned_migration_prevention.py | 52 +++++++++++++++++++
 2 files changed, 68 insertions(+)
 create mode 100644 qemu/tests/cfg/vm_poisoned_migration_prevention.cfg
 create mode 100644 qemu/tests/vm_poisoned_migration_prevention.py

diff --git a/qemu/tests/cfg/vm_poisoned_migration_prevention.cfg b/qemu/tests/cfg/vm_poisoned_migration_prevention.cfg
new file mode 100644
index 0000000000..b955de05fd
--- /dev/null
+++ b/qemu/tests/cfg/vm_poisoned_migration_prevention.cfg
@@ -0,0 +1,16 @@
+- vm_poisoned_migration_prevention:
+    only Linux
+    required_qemu = [9.0.0,)
+    type = vm_poisoned_migration_prevention
+    virt_test_type = qemu
+    smp = 8
+    vcpu_maxcpus = ${smp}
+    slots_mem = 4
+    maxmem_mem = 32G
+    mem = 4096
+    kill_vm_on_error = yes
+    mig_timeout = 1200
+    migration_protocol = "tcp"
+    # The user needs to provide here a valid path to a file that creates HWPoison in the host
+    hwpoison_file = ""
+    target_path = "/var/tmp/hwpoison"
diff --git a/qemu/tests/vm_poisoned_migration_prevention.py b/qemu/tests/vm_poisoned_migration_prevention.py
new file mode 100644
index 0000000000..2403931187
--- /dev/null
+++ b/qemu/tests/vm_poisoned_migration_prevention.py
@@ -0,0 +1,52 @@
+import re
+from shutil import rmtree
+
+from avocado.utils import process
+from virttest import error_context, qemu_monitor
+
+
+@error_context.context_aware
+def run(test, params, env):
+    """
+    Handle migration prevention after VM memory poisoning
+    1) Boot VMs in source and destination hosts
+    2) Download the (internal) hwpoison tool
+    3) Executes the tool to poison some memory
+    4) Checks in the host dmesg the failure traces
+    5) Do migration and check the error message is correct
+
+    :param test: QEMU test object
+    :param params: Dictionary with the test parameters
+    :param env: Dictionary with test environment
+    """
+    hwpoison_file = params.get("hwpoison_file", "")
+    target_path = params.get("target_path")
+    vm = env.get_vm(params["main_vm"])
+    vm.verify_alive()
+    vm.wait_for_login()
+
+    process.system(f"mkdir -p {target_path}", shell=True)
+    qemu_pid = vm.get_pid()
+
+    # This tool is only for internal use, replace the values if needed
+    error_context.base_context("Download the hwpoison tool", test.log.info)
+    process.system(f"wget {hwpoison_file} -P {target_path} ", shell=True)
+
+    error_context.context("Execute the hwpoison tool", test.log.info)
+    process.system(f"python3 {target_path}/hwpoison.py -p {qemu_pid}", shell=True)
+
+    error_context.context("Check the memory failure in dmesg log", test.log.debug)
+    trace_found = re.search("Memory failure", str(process.system_output("dmesg")))
+
+    if not trace_found:
+        test.fail("No memory failure traces found in dmesg")
+
+    mig_timeout = params.get_numeric("mig_timeout", 1200, float)
+    mig_protocol = params.get("migration_protocol", "tcp")
+    try:
+        vm.migrate(mig_timeout, mig_protocol, env=env, not_wait_for_migration=True)
+    except qemu_monitor.MonitorError as e:
+        error_context.context(f"The monitor error: {e}", test.log.info)
+
+    test.log.debug("Cleaning the tool file")
+    rmtree(target_path, ignore_errors=True)