← 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: 2007-11-04 18:51:39 UTC
  • mfrom: (2961.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20071104185139-kaio3sneodg2kp71
Authentication ring implementation (read-only)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/ssh.py
132
132
 
133
133
    def _get_vendor_by_inspection(self):
134
134
        """Return the vendor or None by checking for known SSH implementations."""
135
 
        for args in (['ssh', '-V'], ['plink', '-V']):
 
135
        for args in [['ssh', '-V'], ['plink', '-V']]:
136
136
            version = self._get_ssh_version_string(args)
137
137
            vendor = self._get_vendor_by_version_string(version, args)
138
138
            if vendor is not None:
172
172
    signal.signal(signal.SIGINT, signal.SIG_IGN)
173
173
 
174
174
 
175
 
class SocketAsChannelAdapter(object):
 
175
class LoopbackSFTP(object):
176
176
    """Simple wrapper for a socket that pretends to be a paramiko Channel."""
177
177
 
178
178
    def __init__(self, sock):
179
179
        self.__socket = sock
180
180
 
181
 
    def get_name(self):
182
 
        return "bzr SocketAsChannelAdapter"
183
 
    
184
181
    def send(self, data):
185
182
        return self.__socket.send(data)
186
183
 
187
184
    def recv(self, n):
188
 
        try:
189
 
            return self.__socket.recv(n)
190
 
        except socket.error, e:
191
 
            if e.args[0] in (errno.EPIPE, errno.ECONNRESET, errno.ECONNABORTED,
192
 
                             errno.EBADF):
193
 
                # Connection has closed.  Paramiko expects an empty string in
194
 
                # this case, not an exception.
195
 
                return ''
196
 
            raise
 
185
        return self.__socket.recv(n)
197
186
 
198
187
    def recv_ready(self):
199
 
        # TODO: jam 20051215 this function is necessary to support the
200
 
        # pipelined() function. In reality, it probably should use
201
 
        # poll() or select() to actually return if there is data
202
 
        # available, otherwise we probably don't get any benefit
203
188
        return True
204
189
 
205
190
    def close(self):
252
237
            sock.connect((host, port))
253
238
        except socket.error, e:
254
239
            self._raise_connection_error(host, port=port, orig_error=e)
255
 
        return SFTPClient(SocketAsChannelAdapter(sock))
 
240
        return SFTPClient(LoopbackSFTP(sock))
256
241
 
257
242
register_ssh_vendor('loopback', LoopbackVendor())
258
243
 
341
326
    register_ssh_vendor('paramiko', vendor)
342
327
    register_ssh_vendor('none', vendor)
343
328
    register_default_ssh_vendor(vendor)
344
 
    _sftp_connection_errors = (EOFError, paramiko.SSHException)
345
329
    del vendor
346
 
else:
347
 
    _sftp_connection_errors = (EOFError,)
348
330
 
349
331
 
350
332
class SubprocessVendor(SSHVendor):
362
344
            argv = self._get_vendor_specific_argv(username, host, port,
363
345
                                                  subsystem='sftp')
364
346
            sock = self._connect(argv)
365
 
            return SFTPClient(SocketAsChannelAdapter(sock))
366
 
        except _sftp_connection_errors, e:
 
347
            return SFTPClient(sock)
 
348
        except (EOFError, paramiko.SSHException), e:
367
349
            self._raise_connection_error(host, port=port, orig_error=e)
368
350
        except (OSError, IOError), e:
369
351
            # If the machine is fast enough, ssh can actually exit
402
384
 
403
385
    def _get_vendor_specific_argv(self, username, host, port, subsystem=None,
404
386
                                  command=None):
 
387
        assert subsystem is not None or command is not None, (
 
388
            'Must specify a command or subsystem')
 
389
        if subsystem is not None:
 
390
            assert command is None, (
 
391
                'subsystem and command are mutually exclusive')
405
392
        args = ['ssh',
406
393
                '-oForwardX11=no', '-oForwardAgent=no',
407
394
                '-oClearAllForwardings=yes', '-oProtocol=2',
424
411
 
425
412
    def _get_vendor_specific_argv(self, username, host, port, subsystem=None,
426
413
                                  command=None):
 
414
        assert subsystem is not None or command is not None, (
 
415
            'Must specify a command or subsystem')
 
416
        if subsystem is not None:
 
417
            assert command is None, (
 
418
                'subsystem and command are mutually exclusive')
427
419
        args = ['ssh', '-x']
428
420
        if port is not None:
429
421
            args.extend(['-p', str(port)])
443
435
 
444
436
    def _get_vendor_specific_argv(self, username, host, port, subsystem=None,
445
437
                                  command=None):
446
 
        args = ['plink', '-x', '-a', '-ssh', '-2', '-batch']
 
438
        assert subsystem is not None or command is not None, (
 
439
            'Must specify a command or subsystem')
 
440
        if subsystem is not None:
 
441
            assert command is None, (
 
442
                'subsystem and command are mutually exclusive')
 
443
        args = ['plink', '-x', '-a', '-ssh', '-2']
447
444
        if port is not None:
448
445
            args.extend(['-P', str(port)])
449
446
        if username is not None:
601
598
    def send(self, data):
602
599
        return os.write(self.proc.stdin.fileno(), data)
603
600
 
 
601
    def recv_ready(self):
 
602
        # TODO: jam 20051215 this function is necessary to support the
 
603
        # pipelined() function. In reality, it probably should use
 
604
        # poll() or select() to actually return if there is data
 
605
        # available, otherwise we probably don't get any benefit
 
606
        return True
 
607
 
604
608
    def recv(self, count):
605
609
        return os.read(self.proc.stdout.fileno(), count)
606
610