put a timeout on executing things via ssh

tes/sshwrapper/__init__.py

@@ -3,6 +3,7 @@ This module handles SSH Connections
 """
 import subprocess
 
+TIMEOUT=10
 class SshAgentException(Exception):
     """
     Raised if an SSH Agent isn't running (Normally the SPA client should invoke the agent
@@ -270,10 +271,15 @@ class Ssh:
                                    '-p', sshport, '-l', user, host, cmd],
                                   stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                                   stdin=subprocess.PIPE, env=env)
-        if stdin is not None:
-            (stdout, stderr) = exec_p.communicate(stdin.encode())
-        else:
+        try:
+            if stdin is not None:
+                (stdout, stderr) = exec_p.communicate(stdin.encode(),timeout=TIMEOUT)
+            else:
+                (stdout, stderr) = exec_p.communicate(timeout=TIMEOUT)
+        except subprocess.TimeoutExpired:
+            exec_p.kill()
             (stdout, stderr) = exec_p.communicate()
+
         return {'stdout':stdout, 'stderr':stderr}
 
     @staticmethod