[37776ea] | 1 | #!/usr/local/bin/python |
---|
| 2 | |
---|
| 3 | import sys, os |
---|
| 4 | import re |
---|
| 5 | |
---|
| 6 | import tempfile |
---|
| 7 | import subprocess |
---|
| 8 | import logging |
---|
| 9 | import time |
---|
| 10 | import signal |
---|
| 11 | |
---|
| 12 | from service_error import service_error |
---|
| 13 | |
---|
| 14 | class proxy_segment: |
---|
| 15 | class ssh_cmd_timeout(RuntimeError): pass |
---|
| 16 | |
---|
| 17 | def __init__(self, log=None, keyfile=None, debug=False): |
---|
| 18 | self.log = log or logging.getLogger(\ |
---|
| 19 | 'fedd.access.proxy_segment') |
---|
| 20 | self.ssh_privkey_file = keyfile |
---|
| 21 | self.debug = debug |
---|
| 22 | self.ssh_exec="/usr/bin/ssh" |
---|
| 23 | self.scp_exec = "/usr/bin/scp" |
---|
| 24 | self.ssh_cmd_timeout = proxy_segment.ssh_cmd_timeout |
---|
| 25 | |
---|
| 26 | def scp_file(self, file, user, host, dest=""): |
---|
| 27 | """ |
---|
| 28 | scp a file to the remote host. If debug is set the action is only |
---|
| 29 | logged. |
---|
| 30 | """ |
---|
| 31 | |
---|
| 32 | scp_cmd = [self.scp_exec, '-o', 'IdentitiesOnly yes', |
---|
[e777dab] | 33 | '-o', 'StrictHostKeyChecking no', '-o', 'ForwardX11 no','-i', |
---|
[37776ea] | 34 | self.ssh_privkey_file, file, |
---|
| 35 | "%s@%s:%s" % (user, host, dest)] |
---|
| 36 | rv = 0 |
---|
| 37 | |
---|
| 38 | try: |
---|
| 39 | dnull = open("/dev/null", "w") |
---|
[d3c8759] | 40 | except EnvironmentError: |
---|
[37776ea] | 41 | self.log.debug("[ssh_file]: failed to open " + \ |
---|
| 42 | "/dev/null for redirect") |
---|
| 43 | dnull = Null |
---|
| 44 | |
---|
| 45 | self.log.debug("[scp_file]: %s" % " ".join(scp_cmd)) |
---|
| 46 | if not self.debug: |
---|
| 47 | rv = subprocess.call(scp_cmd, stdout=dnull, |
---|
| 48 | stderr=dnull, close_fds=True) |
---|
| 49 | |
---|
| 50 | return rv == 0 |
---|
| 51 | |
---|
| 52 | def ssh_cmd(self, user, host, cmd, wname=None, timeout=None): |
---|
| 53 | """ |
---|
| 54 | Run a remote command on host as user. If debug is set, the action |
---|
| 55 | is only logged. Commands are run without stdin, to avoid stray |
---|
| 56 | SIGTTINs. |
---|
| 57 | """ |
---|
| 58 | sh_str = ("%s -n -o 'IdentitiesOnly yes' -o " + \ |
---|
[e777dab] | 59 | "'StrictHostKeyChecking no' -o 'ForwardX11 no' " + |
---|
| 60 | "-i %s %s@%s %s") % \ |
---|
[37776ea] | 61 | (self.ssh_exec, self.ssh_privkey_file, |
---|
| 62 | user, host, cmd) |
---|
| 63 | |
---|
| 64 | try: |
---|
| 65 | dnull = open("/dev/null", "w") |
---|
[d3c8759] | 66 | except EnvironmentError: |
---|
[37776ea] | 67 | self.log.debug("[ssh_cmd]: failed to open /dev/null " + \ |
---|
| 68 | "for redirect") |
---|
| 69 | dnull = Null |
---|
| 70 | |
---|
| 71 | self.log.debug("[ssh_cmd]: %s" % sh_str) |
---|
| 72 | if not self.debug: |
---|
| 73 | if dnull: |
---|
| 74 | sub = subprocess.Popen(sh_str, shell=True, stdout=dnull, |
---|
| 75 | stderr=dnull, close_fds=True) |
---|
| 76 | else: |
---|
| 77 | sub = subprocess.Popen(sh_str, shell=True, close_fds=True) |
---|
| 78 | if timeout: |
---|
| 79 | i = 0 |
---|
| 80 | rv = sub.poll() |
---|
| 81 | while i < timeout: |
---|
| 82 | if rv is not None: break |
---|
| 83 | else: |
---|
| 84 | time.sleep(1) |
---|
| 85 | rv = sub.poll() |
---|
| 86 | i += 1 |
---|
| 87 | else: |
---|
| 88 | self.log.debug("Process exceeded runtime: %s" % sh_str) |
---|
| 89 | os.kill(sub.pid, signal.SIGKILL) |
---|
| 90 | raise self.ssh_cmd_timeout(); |
---|
| 91 | return rv == 0 |
---|
| 92 | else: |
---|
| 93 | return sub.wait() == 0 |
---|
| 94 | else: |
---|
| 95 | if timeout == 0: |
---|
| 96 | self.log.debug("debug timeout raised on %s " % sh_str) |
---|
| 97 | raise self.ssh_cmd_timeout() |
---|
| 98 | else: |
---|
| 99 | return True |
---|
| 100 | |
---|