← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/ssh.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-05-16 17:33:27 UTC
  • mfrom: (5755.2.10 2.4-max-entries-gc-602614)
  • Revision ID: pqm@pqm.ubuntu.com-20110516173327-5ehst0ttceohsf5w
(jameinel) Add bzr.groupcompress.max_bytes_to_index to limit peak memory
 when delta-compressing large files (bug #602614) (John A Meinel)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/ssh.py
1
 
# Copyright (C) 2006-2010 Robey Pointer <robey@lag.net>
 
1
# Copyright (C) 2006-2011 Robey Pointer <robey@lag.net>
2
2
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
3
3
#
4
4
# This program is free software; you can redistribute it and/or modify
126
126
        elif 'SSH Secure Shell' in version:
127
127
            trace.mutter('ssh implementation is SSH Corp.')
128
128
            vendor = SSHCorpSubprocessVendor()
 
129
        elif 'lsh' in version:
 
130
            trace.mutter('ssh implementation is GNU lsh.')
 
131
            vendor = LSHSubprocessVendor()
129
132
        # As plink user prompts are not handled currently, don't auto-detect
130
133
        # it by inspection below, but keep this vendor detection for if a path
131
134
        # is given in BZR_SSH. See https://bugs.launchpad.net/bugs/414743
356
359
        # whatever) chunks.
357
360
        try:
358
361
            my_sock, subproc_sock = socket.socketpair()
 
362
            osutils.set_fd_cloexec(my_sock)
359
363
        except (AttributeError, socket.error):
360
364
            # This platform doesn't support socketpair(), so just use ordinary
361
365
            # pipes instead.
362
366
            stdin = stdout = subprocess.PIPE
363
 
            sock = None
 
367
            my_sock, subproc_sock = None, None
364
368
        else:
365
369
            stdin = stdout = subproc_sock
366
 
            sock = my_sock
367
370
        proc = subprocess.Popen(argv, stdin=stdin, stdout=stdout,
368
371
                                **os_specific_subprocess_params())
369
 
        return SSHSubprocessConnection(proc, sock=sock)
 
372
        if subproc_sock is not None:
 
373
            subproc_sock.close()
 
374
        return SSHSubprocessConnection(proc, sock=my_sock)
370
375
 
371
376
    def connect_sftp(self, username, password, host, port):
372
377
        try:
403
408
                                  command=None):
404
409
        args = [self.executable_path,
405
410
                '-oForwardX11=no', '-oForwardAgent=no',
406
 
                '-oClearAllForwardings=yes', '-oProtocol=2',
 
411
                '-oClearAllForwardings=yes',
407
412
                '-oNoHostAuthenticationForLocalhost=yes']
408
413
        if port is not None:
409
414
            args.extend(['-p', str(port)])
439
444
register_ssh_vendor('sshcorp', SSHCorpSubprocessVendor())
440
445
 
441
446
 
 
447
class LSHSubprocessVendor(SubprocessVendor):
 
448
    """SSH vendor that uses the 'lsh' executable from GNU"""
 
449
 
 
450
    executable_path = 'lsh'
 
451
 
 
452
    def _get_vendor_specific_argv(self, username, host, port, subsystem=None,
 
453
                                  command=None):
 
454
        args = [self.executable_path]
 
455
        if port is not None:
 
456
            args.extend(['-p', str(port)])
 
457
        if username is not None:
 
458
            args.extend(['-l', username])
 
459
        if subsystem is not None:
 
460
            args.extend(['--subsystem', subsystem, host])
 
461
        else:
 
462
            args.extend([host] + command)
 
463
        return args
 
464
 
 
465
register_ssh_vendor('lsh', LSHSubprocessVendor())
 
466
 
 
467
 
442
468
class PLinkSubprocessVendor(SubprocessVendor):
443
469
    """SSH vendor that uses the 'plink' executable from Putty."""
444
470
 
629
655
import weakref
630
656
_subproc_weakrefs = set()
631
657
 
632
 
def _close_ssh_proc(proc):
 
658
def _close_ssh_proc(proc, sock):
633
659
    """Carefully close stdin/stdout and reap the SSH process.
634
660
 
635
661
    If the pipes are already closed and/or the process has already been
636
662
    wait()ed on, that's ok, and no error is raised.  The goal is to do our best
637
663
    to clean up (whether or not a clean up was already tried).
638
664
    """
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()
 
665
    funcs = []
 
666
    for closeable in (proc.stdin, proc.stdout, sock):
 
667
        # We expect that either proc (a subprocess.Popen) will have stdin and
 
668
        # stdout streams to close, or that we will have been passed a socket to
 
669
        # close, with the option not in use being None.
 
670
        if closeable is not None:
 
671
            funcs.append(closeable.close)
 
672
    funcs.append(proc.wait)
 
673
    for func in funcs:
 
674
        try:
 
675
            func()
651
676
        except OSError:
652
677
            # It's ok for the pipe to already be closed, or the process to
653
678
            # already be finished.
692
717
        # to avoid leaving processes lingering indefinitely.
693
718
        def terminate(ref):
694
719
            _subproc_weakrefs.remove(ref)
695
 
            _close_ssh_proc(proc)
 
720
            _close_ssh_proc(proc, sock)
696
721
        _subproc_weakrefs.add(weakref.ref(self, terminate))
697
722
 
698
723
    def send(self, data):
708
733
            return os.read(self.proc.stdout.fileno(), count)
709
734
 
710
735
    def close(self):
711
 
        _close_ssh_proc(self.proc)
 
736
        _close_ssh_proc(self.proc, self._sock)
712
737
 
713
738
    def get_sock_or_pipes(self):
714
739
        if self._sock is not None: