Enable auto login for qemu SUT

This patch add the possibility to handle qemu images with auto login and to handle the first PS1 shown after boot. Signed-off-by: Andrea Cervesato <andrea.cervesato@suse.com>

Author: acerv

ltp/qemu_sut.py

@@ -35,6 +35,8 @@ def __init__(self) -> None:
  self._virtfs = None
  self._serial_type = None
  self._opts = None
+ self._user = None
+ self._prompt = None
 
  @property
  def name(self) -> str:
@@ -45,6 +47,7 @@ def config_help(self) -> dict:
  return {
  "image": "qcow2 image location",
  "image_overlay": "image_overlay: image copy location",
+ "user": "username used to login. If empty, it won't be used (default: 'root')",
  "password": "root password (default: root)",
  "system": "system architecture (default: x86_64)",
  "ram": "RAM of the VM (default: 2G)",
@@ -53,6 +56,7 @@ def config_help(self) -> dict:
  "virtfs": "directory to mount inside VM",
  "ro_image": "path of the image that will exposed as read only",
  "options": "user defined options",
+ "first_prompt": "first shell prompt string (default: '#')",
  }
 
  def _get_transport(self) -> str:
@@ -131,20 +135,21 @@ def _get_command(self) -> str:
  return cmd
 
  def _login(self, timeout: float, iobuffer: IOBuffer) -> None:
- self._wait_for("login:", timeout, iobuffer)
- self._write_stdin("root\n")
+ if self._user:
+ self._wait_for("login:", timeout, iobuffer)
+ self._write_stdin(f"{self._user}\n")
 
  if self._password:
  self._wait_for("Password:", 5, iobuffer)
  self._write_stdin(f"{self._password}\n")
 
  time.sleep(0.2)
 
- self._wait_for("#", 5, iobuffer)
+ self._wait_for(self._prompt, timeout, iobuffer)
  time.sleep(0.2)
 
  self._write_stdin("stty -echo; stty cols 1024\n")
- self._wait_for("#", 5, None)
+ self._wait_for(self._prompt, 5, None)
 
  _, retcode, _ = self._exec("export PS1=''", 5, None)
  if retcode != 0:
@@ -179,12 +184,14 @@ def setup(self, **kwargs: dict) -> None:
  self._image = kwargs.get("image", None)
  self._image_overlay = kwargs.get("image_overlay", None)
  self._ro_image = kwargs.get("ro_image", None)
+ self._user = kwargs.get("user", "root")
  self._password = kwargs.get("password", "root")
  self._ram = kwargs.get("ram", "2G")
  self._smp = kwargs.get("smp", "2")
  self._virtfs = kwargs.get("virtfs", None)
  self._serial_type = kwargs.get("serial", "isa")
  self._opts = kwargs.get("options", None)
+ self._prompt = kwargs.get("first_prompt", "#")
 
  system = kwargs.get("system", "x86_64")
  self._qemu_cmd = f"qemu-system-{system}"
@@ -216,3 +223,6 @@ def setup(self, **kwargs: dict) -> None:
 
  if self._serial_type not in ["isa", "virtio"]:
  raise SUTError("Serial protocol must be isa or virtio")
+
+ if not self._prompt:
+ raise SUTError("first_prompt is not defined")