Merge pull request #4 from gsalvatella/feat/login-shell

Providing a login shell in ssh connections (thanks to @gsalvatella for the contribution)

Author: MarcDorval

connection/SshTarget.py

@@ -87,7 +87,8 @@ def __send_key(self, original_pwd):
 
     def write(self, text):
         if self is not None:
-            self.stdin, self.stdout, self.stderr = self.exec_command(text, environment={'ENV': '/etc/profile'})
+            cmd = b"/bin/sh --login -c " + text  # Provide a login shell to set the environment and locate scripts
+            self.stdin, self.stdout, self.stderr = self.exec_command(cmd, environment={'ENV': '/etc/profile'})
 
     def read(self):
         self.result = str(self.stdout.read(), "utf-8").strip()

connection/wfx_connection.py

@@ -8,6 +8,7 @@
 # Standard library imports
 import logging
 import os
+import shlex
 import sys
 import time
 
@@ -319,7 +320,7 @@ def write(self, text):
                 for line in text.rstrip().split('\n'):
                     print(str.format("%-8s S>>|  " % self.nickname), end='')
                     print(line)
-            self.link.write(bytes(text.rstrip() + '\n', 'utf-8'))
+            self.link.write(bytes(shlex.quote(text).rstrip() + '\n', 'utf-8'))
 
     def read(self):
         if self.link is not None: