← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/ssh.py

  • Committer: Parth Malwankar
  • Date: 2010-09-24 12:53:00 UTC
  • mfrom: (5443 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5444.
  • Revision ID: parth.malwankar@gmail.com-20100924125300-70sg2t3q03bcuqpm
merged trunk and moved NEWS entry to correct section

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/ssh.py
336
336
            self._raise_connection_error(host, port=port, orig_error=e,
337
337
                                         msg='Unable to invoke remote bzr')
338
338
 
 
339
_ssh_connection_errors = (EOFError, OSError, IOError, socket.error)
339
340
if paramiko is not None:
340
341
    vendor = ParamikoVendor()
341
342
    register_ssh_vendor('paramiko', vendor)
342
343
    register_ssh_vendor('none', vendor)
343
344
    register_default_ssh_vendor(vendor)
344
 
    _sftp_connection_errors = (EOFError, paramiko.SSHException)
 
345
    _ssh_connection_errors += (paramiko.SSHException,)
345
346
    del vendor
346
 
else:
347
 
    _sftp_connection_errors = (EOFError,)
348
347
 
349
348
 
350
349
class SubprocessVendor(SSHVendor):
375
374
                                                  subsystem='sftp')
376
375
            sock = self._connect(argv)
377
376
            return SFTPClient(SocketAsChannelAdapter(sock))
378
 
        except _sftp_connection_errors, e:
379
 
            self._raise_connection_error(host, port=port, orig_error=e)
380
 
        except (OSError, IOError), e:
381
 
            # If the machine is fast enough, ssh can actually exit
382
 
            # before we try and send it the sftp request, which
383
 
            # raises a Broken Pipe
384
 
            if e.errno not in (errno.EPIPE,):
385
 
                raise
 
377
        except _ssh_connection_errors, e:
386
378
            self._raise_connection_error(host, port=port, orig_error=e)
387
379
 
388
380
    def connect_ssh(self, username, password, host, port, command):
390
382
            argv = self._get_vendor_specific_argv(username, host, port,
391
383
                                                  command=command)
392
384
            return self._connect(argv)
393
 
        except (EOFError), e:
394
 
            self._raise_connection_error(host, port=port, orig_error=e)
395
 
        except (OSError, IOError), e:
396
 
            # If the machine is fast enough, ssh can actually exit
397
 
            # before we try and send it the sftp request, which
398
 
            # raises a Broken Pipe
399
 
            if e.errno not in (errno.EPIPE,):
400
 
                raise
 
385
        except _ssh_connection_errors, e:
401
386
            self._raise_connection_error(host, port=port, orig_error=e)
402
387
 
403
388
    def _get_vendor_specific_argv(self, username, host, port, subsystem=None,
645
630
_subproc_weakrefs = set()
646
631
 
647
632
def _close_ssh_proc(proc):
648
 
    for func in [proc.stdin.close, proc.stdout.close, proc.wait]:
649
 
        try:
650
 
            func()
 
633
    """Carefully close stdin/stdout and reap the SSH process.
 
634
 
 
635
    If the pipes are already closed and/or the process has already been
 
636
    wait()ed on, that's ok, and no error is raised.  The goal is to do our best
 
637
    to clean up (whether or not a clean up was already tried).
 
638
    """
 
639
    dotted_names = ['stdin.close', 'stdout.close', 'wait']
 
640
    for dotted_name in dotted_names:
 
641
        attrs = dotted_name.split('.')
 
642
        try:
 
643
            obj = proc
 
644
            for attr in attrs:
 
645
                obj = getattr(obj, attr)
 
646
        except AttributeError:
 
647
            # It's ok for proc.stdin or proc.stdout to be None.
 
648
            continue
 
649
        try:
 
650
            obj()
651
651
        except OSError:
652
 
            pass
 
652
            # It's ok for the pipe to already be closed, or the process to
 
653
            # already be finished.
 
654
            continue
653
655
 
654
656
 
655
657
class SSHConnection(object):