Refactor and fix SSH service restart on OSMount worker SSH env config…

…uration

On newer Ubuntu (starting with 24.04) Server images, `sshd` service name has
been changed to `ssh`. This patch handles this rename and refactors it into
the base OSMount class, in order for it to be picked up for all Linux OSMount
minions.

coriolis/osmorphing/osmount/base.py

@@ -17,7 +17,6 @@
 from coriolis import exception
 from coriolis import utils
 
-
 LOG = logging.getLogger(__name__)
 
 MAJOR_COLUMN_INDEX = 4
@@ -96,7 +95,16 @@ def setup(self):
             self._connect()
 
     def _allow_ssh_env_vars(self):
-        pass
+        self._exec_cmd('sudo sed -i -e "\$aAcceptEnv *" /etc/ssh/sshd_config')
+        try:
+            utils.restart_service(self._ssh, "sshd")
+        except exception.CoriolisException:
+            # Newer Ubuntu distros renamed this service to `ssh`
+            LOG.warning(
+                "Could not restart service sshd. Attempting to restart ssh "
+                "service")
+            utils.restart_service(self._ssh, 'ssh')
+        return True
 
     def _exec_cmd(self, cmd, timeout=None):
         if not timeout:

coriolis/osmorphing/osmount/redhat.py

@@ -23,8 +23,3 @@ def setup(self):
         self._exec_cmd("sudo -E yum install -y lvm2 psmisc")
         self._exec_cmd("sudo modprobe dm-mod")
         self._exec_cmd("sudo rm -f /etc/lvm/devices/system.devices")
-
-    def _allow_ssh_env_vars(self):
-        self._exec_cmd('sudo sed -i -e "\$aAcceptEnv *" /etc/ssh/sshd_config')
-        utils.restart_service(self._ssh, "sshd")
-        return True

coriolis/osmorphing/osmount/ubuntu.py

@@ -36,8 +36,3 @@ def setup(self):
             "install lvm2 psmisc -y")
 
         self._exec_cmd("sudo modprobe dm-mod")
-
-    def _allow_ssh_env_vars(self):
-        self._exec_cmd('sudo sed -i -e "\$aAcceptEnv *" /etc/ssh/sshd_config')
-        utils.restart_service(self._ssh, "sshd")
-        return True