ssh: add support for known_hosts

Reviewed-by: Andrea Cervesato <andrea.cervesato@suse.com>
Signed-off-by: jingfelix <jingfelix@outlook.com>
Signed-off-by: Andrea Cervesato <andrea.cervesato@suse.com>
Closes: #42

Author: jingfelix

libkirk/ssh.py

@@ -74,6 +74,7 @@ def __init__(self) -> None:
         self._password = None
         self._key_file = None
         self._sudo = False
+        self._known_hosts = None
         self._session_sem = None
         self._stop = False
         self._conn = None
@@ -94,6 +95,7 @@ def config_help(self) -> dict:
             "key_file": "private key location",
             "reset_cmd": "command to reset the remote SUT",
             "sudo": "use sudo to access to root shell (default: 0)",
+            "known_hosts": "path to custom known_hosts file (optional)",
         }
 
     async def _reset(self, iobuffer: IOBuffer = None) -> None:
@@ -161,6 +163,7 @@ def setup(self, **kwargs: dict) -> None:
         self._user = kwargs.get("user", "root")
         self._password = kwargs.get("password", None)
         self._key_file = kwargs.get("key_file", None)
+        self._known_hosts = kwargs.get("known_hosts", None)
 
         try:
             self._port = int(kwargs.get("port", "22"))
@@ -196,13 +199,15 @@ async def communicate(self, iobuffer: IOBuffer = None) -> None:
                     host=self._host,
                     port=self._port,
                     username=self._user,
-                    client_keys=[priv_key])
+                    client_keys=[priv_key],
+                    known_hosts=self._known_hosts)
             else:
                 self._conn = await asyncssh.connect(
                     host=self._host,
                     port=self._port,
                     username=self._user,
-                    password=self._password)
+                    password=self._password,
+                    known_hosts=self._known_hosts)
 
             # read maximum number of sessions and limit `run_command`
             # concurrent calls to that by using a semaphore
@@ -225,7 +230,8 @@ async def stop(self, iobuffer: IOBuffer = None) -> None:
         self._stop = True
         try:
             if self._channels:
-                self._logger.info("Killing %d process(es)", len(self._channels))
+                self._logger.info("Killing %d process(es)",
+                                  len(self._channels))
 
                 for proc in self._channels:
                     proc.kill()