← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/sftp.py

  • Committer: Robert Collins
  • Date: 2006-01-05 22:30:59 UTC
  • mto: (1534.1.4 integration)
  • mto: This revision was merged to the branch mainline in revision 1536.
  • Revision ID: robertc@robertcollins.net-20060105223059-a8b64f7b47cf12fb
 * bzrlib.osutils.safe_unicode now exists to provide parameter coercion
   for functions that need unicode strings. (Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/sftp.py
23
23
import stat
24
24
import sys
25
25
import urllib
 
26
import urlparse
 
27
import time
 
28
import random
 
29
import subprocess
 
30
import weakref
26
31
 
27
32
from bzrlib.errors import (FileExists, 
28
 
                           TransportNotPossible, NoSuchFile, NonRelativePath,
29
 
                           TransportError)
30
 
from bzrlib.config import config_dir
 
33
                           TransportNotPossible, NoSuchFile, PathNotChild,
 
34
                           TransportError,
 
35
                           LockError)
 
36
from bzrlib.config import config_dir, ensure_config_dir_exists
31
37
from bzrlib.trace import mutter, warning, error
32
38
from bzrlib.transport import Transport, register_transport
 
39
from bzrlib.osutils import pathjoin, fancy_rename
 
40
import bzrlib.ui
33
41
 
34
42
try:
35
43
    import paramiko
36
44
except ImportError:
37
45
    error('The SFTP transport requires paramiko.')
38
46
    raise
 
47
else:
 
48
    from paramiko.sftp import (SFTP_FLAG_WRITE, SFTP_FLAG_CREATE,
 
49
                               SFTP_FLAG_EXCL, SFTP_FLAG_TRUNC,
 
50
                               CMD_HANDLE, CMD_OPEN)
 
51
    from paramiko.sftp_attr import SFTPAttributes
 
52
    from paramiko.sftp_file import SFTPFile
 
53
    from paramiko.sftp_client import SFTPClient
 
54
 
 
55
if 'sftp' not in urlparse.uses_netloc: urlparse.uses_netloc.append('sftp')
 
56
 
 
57
 
 
58
_close_fds = True
 
59
if sys.platform == 'win32':
 
60
    # close_fds not supported on win32
 
61
    _close_fds = False
 
62
 
 
63
_ssh_vendor = None
 
64
def _get_ssh_vendor():
 
65
    """Find out what version of SSH is on the system."""
 
66
    global _ssh_vendor
 
67
    if _ssh_vendor is not None:
 
68
        return _ssh_vendor
 
69
 
 
70
    _ssh_vendor = 'none'
 
71
 
 
72
    try:
 
73
        p = subprocess.Popen(['ssh', '-V'],
 
74
                             close_fds=_close_fds,
 
75
                             stdin=subprocess.PIPE,
 
76
                             stdout=subprocess.PIPE,
 
77
                             stderr=subprocess.PIPE)
 
78
        returncode = p.returncode
 
79
        stdout, stderr = p.communicate()
 
80
    except OSError:
 
81
        returncode = -1
 
82
        stdout = stderr = ''
 
83
    if 'OpenSSH' in stderr:
 
84
        mutter('ssh implementation is OpenSSH')
 
85
        _ssh_vendor = 'openssh'
 
86
    elif 'SSH Secure Shell' in stderr:
 
87
        mutter('ssh implementation is SSH Corp.')
 
88
        _ssh_vendor = 'ssh'
 
89
 
 
90
    if _ssh_vendor != 'none':
 
91
        return _ssh_vendor
 
92
 
 
93
    # XXX: 20051123 jamesh
 
94
    # A check for putty's plink or lsh would go here.
 
95
 
 
96
    mutter('falling back to paramiko implementation')
 
97
    return _ssh_vendor
 
98
 
 
99
 
 
100
class SFTPSubprocess:
 
101
    """A socket-like object that talks to an ssh subprocess via pipes."""
 
102
    def __init__(self, hostname, port=None, user=None):
 
103
        vendor = _get_ssh_vendor()
 
104
        assert vendor in ['openssh', 'ssh']
 
105
        if vendor == 'openssh':
 
106
            args = ['ssh',
 
107
                    '-oForwardX11=no', '-oForwardAgent=no',
 
108
                    '-oClearAllForwardings=yes', '-oProtocol=2',
 
109
                    '-oNoHostAuthenticationForLocalhost=yes']
 
110
            if port is not None:
 
111
                args.extend(['-p', str(port)])
 
112
            if user is not None:
 
113
                args.extend(['-l', user])
 
114
            args.extend(['-s', hostname, 'sftp'])
 
115
        elif vendor == 'ssh':
 
116
            args = ['ssh', '-x']
 
117
            if port is not None:
 
118
                args.extend(['-p', str(port)])
 
119
            if user is not None:
 
120
                args.extend(['-l', user])
 
121
            args.extend(['-s', 'sftp', hostname])
 
122
 
 
123
        self.proc = subprocess.Popen(args, close_fds=_close_fds,
 
124
                                     stdin=subprocess.PIPE,
 
125
                                     stdout=subprocess.PIPE)
 
126
 
 
127
    def send(self, data):
 
128
        return os.write(self.proc.stdin.fileno(), data)
 
129
 
 
130
    def recv_ready(self):
 
131
        # TODO: jam 20051215 this function is necessary to support the
 
132
        # pipelined() function. In reality, it probably should use
 
133
        # poll() or select() to actually return if there is data
 
134
        # available, otherwise we probably don't get any benefit
 
135
        return True
 
136
 
 
137
    def recv(self, count):
 
138
        return os.read(self.proc.stdout.fileno(), count)
 
139
 
 
140
    def close(self):
 
141
        self.proc.stdin.close()
 
142
        self.proc.stdout.close()
 
143
        self.proc.wait()
39
144
 
40
145
 
41
146
SYSTEM_HOSTKEYS = {}
42
147
BZR_HOSTKEYS = {}
43
148
 
 
149
# This is a weakref dictionary, so that we can reuse connections
 
150
# that are still active. Long term, it might be nice to have some
 
151
# sort of expiration policy, such as disconnect if inactive for
 
152
# X seconds. But that requires a lot more fanciness.
 
153
_connected_hosts = weakref.WeakValueDictionary()
 
154
 
44
155
def load_host_keys():
45
156
    """
46
157
    Load system host keys (probably doesn't work on windows) and any
51
162
        SYSTEM_HOSTKEYS = paramiko.util.load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
52
163
    except Exception, e:
53
164
        mutter('failed to load system host keys: ' + str(e))
54
 
    bzr_hostkey_path = os.path.join(config_dir(), 'ssh_host_keys')
 
165
    bzr_hostkey_path = pathjoin(config_dir(), 'ssh_host_keys')
55
166
    try:
56
167
        BZR_HOSTKEYS = paramiko.util.load_host_keys(bzr_hostkey_path)
57
168
    except Exception, e:
63
174
    Save "discovered" host keys in $(config)/ssh_host_keys/.
64
175
    """
65
176
    global SYSTEM_HOSTKEYS, BZR_HOSTKEYS
66
 
    bzr_hostkey_path = os.path.join(config_dir(), 'ssh_host_keys')
67
 
    if not os.path.isdir(config_dir()):
68
 
        os.mkdir(config_dir())
 
177
    bzr_hostkey_path = pathjoin(config_dir(), 'ssh_host_keys')
 
178
    ensure_config_dir_exists()
 
179
 
69
180
    try:
70
181
        f = open(bzr_hostkey_path, 'w')
71
182
        f.write('# SSH host keys collected by bzr\n')
77
188
        mutter('failed to save bzr host keys: ' + str(e))
78
189
 
79
190
 
80
 
 
81
 
class SFTPTransportError (TransportError):
82
 
    pass
83
 
 
 
191
class SFTPLock(object):
 
192
    """This fakes a lock in a remote location."""
 
193
    __slots__ = ['path', 'lock_path', 'lock_file', 'transport']
 
194
    def __init__(self, path, transport):
 
195
        assert isinstance(transport, SFTPTransport)
 
196
 
 
197
        self.lock_file = None
 
198
        self.path = path
 
199
        self.lock_path = path + '.write-lock'
 
200
        self.transport = transport
 
201
        try:
 
202
            abspath = transport._abspath(self.lock_path)
 
203
            self.lock_file = transport._sftp_open_exclusive(abspath)
 
204
        except FileExists:
 
205
            raise LockError('File %r already locked' % (self.path,))
 
206
 
 
207
    def __del__(self):
 
208
        """Should this warn, or actually try to cleanup?"""
 
209
        if self.lock_file:
 
210
            warn("SFTPLock %r not explicitly unlocked" % (self.path,))
 
211
            self.unlock()
 
212
 
 
213
    def unlock(self):
 
214
        if not self.lock_file:
 
215
            return
 
216
        self.lock_file.close()
 
217
        self.lock_file = None
 
218
        try:
 
219
            self.transport.delete(self.lock_path)
 
220
        except (NoSuchFile,):
 
221
            # What specific errors should we catch here?
 
222
            pass
84
223
 
85
224
class SFTPTransport (Transport):
86
225
    """
87
226
    Transport implementation for SFTP access.
88
227
    """
 
228
    _do_prefetch = False # Right now Paramiko's prefetch support causes things to hang
89
229
 
90
 
    _url_matcher = re.compile(r'^sftp://([^:@]*(:[^@]*)?@)?(.*?)(:\d+)?(/.*)?$')
91
 
    
92
230
    def __init__(self, base, clone_from=None):
93
231
        assert base.startswith('sftp://')
 
232
        self._parse_url(base)
 
233
        base = self._unparse_url()
94
234
        super(SFTPTransport, self).__init__(base)
95
 
        self._parse_url(base)
96
235
        if clone_from is None:
97
236
            self._sftp_connect()
98
237
        else:
148
287
                basepath.append(p)
149
288
 
150
289
        path = '/'.join(basepath)
151
 
        if len(path) and path[0] != '/':
152
 
            path = '/' + path
 
290
        # could still be a "relative" path here, but relative on the sftp server
153
291
        return path
154
292
 
155
293
    def relpath(self, abspath):
156
 
        # FIXME: this is identical to HttpTransport -- share it
157
 
        if not abspath.startswith(self.base):
158
 
            raise NonRelativePath('path %r is not under base URL %r'
159
 
                           % (abspath, self.base))
160
 
        pl = len(self.base)
161
 
        return abspath[pl:].lstrip('/')
 
294
        username, password, host, port, path = self._split_url(abspath)
 
295
        error = []
 
296
        if (username != self._username):
 
297
            error.append('username mismatch')
 
298
        if (host != self._host):
 
299
            error.append('host mismatch')
 
300
        if (port != self._port):
 
301
            error.append('port mismatch')
 
302
        if (not path.startswith(self._path)):
 
303
            error.append('path mismatch')
 
304
        if error:
 
305
            extra = ': ' + ', '.join(error)
 
306
            raise PathNotChild(abspath, self.base, extra=extra)
 
307
        pl = len(self._path)
 
308
        return path[pl:].lstrip('/')
162
309
 
163
310
    def has(self, relpath):
164
311
        """
178
325
        """
179
326
        try:
180
327
            path = self._abspath(relpath)
181
 
            return self._sftp.file(path)
182
 
        except (IOError, paramiko.SSHException), x:
183
 
            raise NoSuchFile('Error retrieving %s: %s' % (path, str(x)), x)
 
328
            f = self._sftp.file(path, mode='rb')
 
329
            if self._do_prefetch and hasattr(f, 'prefetch'):
 
330
                f.prefetch()
 
331
            return f
 
332
        except (IOError, paramiko.SSHException), e:
 
333
            self._translate_io_exception(e, path, ': error retrieving')
184
334
 
185
335
    def get_partial(self, relpath, start, length=None):
186
336
        """
194
344
                 Some implementations may return objects which can be read
195
345
                 past this length, but this is not guaranteed.
196
346
        """
 
347
        # TODO: implement get_partial_multi to help with knit support
197
348
        f = self.get(relpath)
198
349
        f.seek(start)
 
350
        if self._do_prefetch and hasattr(f, 'prefetch'):
 
351
            f.prefetch()
199
352
        return f
200
353
 
201
 
    def put(self, relpath, f):
 
354
    def put(self, relpath, f, mode=None):
202
355
        """
203
356
        Copy the file-like or string object into the location.
204
357
 
205
358
        :param relpath: Location to put the contents, relative to base.
206
359
        :param f:       File-like or string object.
 
360
        :param mode: The final mode for the file
207
361
        """
208
 
        # FIXME: should do something atomic or locking here, this is unsafe
209
 
        try:
210
 
            path = self._abspath(relpath)
211
 
            fout = self._sftp.file(path, 'wb')
212
 
        except IOError, e:
213
 
            self._translate_io_exception(e, relpath)
214
 
        except (IOError, paramiko.SSHException), x:
215
 
            raise SFTPTransportError('Unable to write file %r' % (path,), x)
216
 
        try:
217
 
            self._pump(f, fout)
218
 
        finally:
 
362
        final_path = self._abspath(relpath)
 
363
        self._put(final_path, f, mode=mode)
 
364
 
 
365
    def _put(self, abspath, f, mode=None):
 
366
        """Helper function so both put() and copy_abspaths can reuse the code"""
 
367
        tmp_abspath = '%s.tmp.%.9f.%d.%d' % (abspath, time.time(),
 
368
                        os.getpid(), random.randint(0,0x7FFFFFFF))
 
369
        fout = self._sftp_open_exclusive(tmp_abspath, mode=mode)
 
370
        closed = False
 
371
        try:
 
372
            try:
 
373
                fout.set_pipelined(True)
 
374
                self._pump(f, fout)
 
375
            except (IOError, paramiko.SSHException), e:
 
376
                self._translate_io_exception(e, tmp_abspath)
 
377
            if mode is not None:
 
378
                self._sftp.chmod(tmp_abspath, mode)
219
379
            fout.close()
 
380
            closed = True
 
381
            self._rename(tmp_abspath, abspath)
 
382
        except Exception, e:
 
383
            # If we fail, try to clean up the temporary file
 
384
            # before we throw the exception
 
385
            # but don't let another exception mess things up
 
386
            # Write out the traceback, because otherwise
 
387
            # the catch and throw destroys it
 
388
            import traceback
 
389
            mutter(traceback.format_exc())
 
390
            try:
 
391
                if not closed:
 
392
                    fout.close()
 
393
                self._sftp.remove(tmp_abspath)
 
394
            except:
 
395
                # raise the saved except
 
396
                raise e
 
397
            # raise the original with its traceback if we can.
 
398
            raise
220
399
 
221
400
    def iter_files_recursive(self):
222
401
        """Walk the relative paths of all files in this transport."""
230
409
            else:
231
410
                yield relpath
232
411
 
233
 
    def mkdir(self, relpath):
 
412
    def mkdir(self, relpath, mode=None):
234
413
        """Create a directory at the given path."""
235
414
        try:
236
415
            path = self._abspath(relpath)
 
416
            # In the paramiko documentation, it says that passing a mode flag 
 
417
            # will filtered against the server umask.
 
418
            # StubSFTPServer does not do this, which would be nice, because it is
 
419
            # what we really want :)
 
420
            # However, real servers do use umask, so we really should do it that way
237
421
            self._sftp.mkdir(path)
238
 
        except IOError, e:
239
 
            self._translate_io_exception(e, relpath)
240
 
        except (IOError, paramiko.SSHException), x:
241
 
            raise SFTPTransportError('Unable to mkdir %r' % (path,), x)
242
 
 
243
 
    def _translate_io_exception(self, e, relpath):
 
422
            if mode is not None:
 
423
                self._sftp.chmod(path, mode=mode)
 
424
        except (paramiko.SSHException, IOError), e:
 
425
            self._translate_io_exception(e, path, ': unable to mkdir',
 
426
                failure_exc=FileExists)
 
427
 
 
428
    def _translate_io_exception(self, e, path, more_info='', failure_exc=NoSuchFile):
 
429
        """Translate a paramiko or IOError into a friendlier exception.
 
430
 
 
431
        :param e: The original exception
 
432
        :param path: The path in question when the error is raised
 
433
        :param more_info: Extra information that can be included,
 
434
                          such as what was going on
 
435
        :param failure_exc: Paramiko has the super fun ability to raise completely
 
436
                           opaque errors that just set "e.args = ('Failure',)" with
 
437
                           no more information.
 
438
                           This sometimes means FileExists, but it also sometimes
 
439
                           means NoSuchFile
 
440
        """
244
441
        # paramiko seems to generate detailless errors.
245
 
        if (e.errno == errno.ENOENT or
246
 
            e.args == ('No such file or directory',) or
247
 
            e.args == ('No such file',)):
248
 
            raise NoSuchFile(relpath)
249
 
        if (e.args == ('mkdir failed',)):
250
 
            raise FileExists(relpath)
251
 
        # strange but true, for the paramiko server.
252
 
        if (e.args == ('Failure',)):
253
 
            raise FileExists(relpath)
254
 
        raise
 
442
        self._translate_error(e, path, raise_generic=False)
 
443
        if hasattr(e, 'args'):
 
444
            if (e.args == ('No such file or directory',) or
 
445
                e.args == ('No such file',)):
 
446
                raise NoSuchFile(path, str(e) + more_info)
 
447
            if (e.args == ('mkdir failed',)):
 
448
                raise FileExists(path, str(e) + more_info)
 
449
            # strange but true, for the paramiko server.
 
450
            if (e.args == ('Failure',)):
 
451
                raise failure_exc(path, str(e) + more_info)
 
452
            mutter('Raising exception with args %s', e.args)
 
453
        if hasattr(e, 'errno'):
 
454
            mutter('Raising exception with errno %s', e.errno)
 
455
        raise e
255
456
 
256
457
    def append(self, relpath, f):
257
458
        """
262
463
            path = self._abspath(relpath)
263
464
            fout = self._sftp.file(path, 'ab')
264
465
            self._pump(f, fout)
265
 
        except (IOError, paramiko.SSHException), x:
266
 
            raise SFTPTransportError('Unable to append file %r' % (path,), x)
 
466
        except (IOError, paramiko.SSHException), e:
 
467
            self._translate_io_exception(e, relpath, ': unable to append')
267
468
 
268
469
    def copy(self, rel_from, rel_to):
269
470
        """Copy the item at rel_from to the location at rel_to"""
270
471
        path_from = self._abspath(rel_from)
271
472
        path_to = self._abspath(rel_to)
 
473
        self._copy_abspaths(path_from, path_to)
 
474
 
 
475
    def _copy_abspaths(self, path_from, path_to, mode=None):
 
476
        """Copy files given an absolute path
 
477
 
 
478
        :param path_from: Path on remote server to read
 
479
        :param path_to: Path on remote server to write
 
480
        :return: None
 
481
 
 
482
        TODO: Should the destination location be atomically created?
 
483
              This has not been specified
 
484
        TODO: This should use some sort of remote copy, rather than
 
485
              pulling the data locally, and then writing it remotely
 
486
        """
272
487
        try:
273
488
            fin = self._sftp.file(path_from, 'rb')
274
489
            try:
275
 
                fout = self._sftp.file(path_to, 'wb')
276
 
                try:
277
 
                    fout.set_pipelined(True)
278
 
                    self._pump(fin, fout)
279
 
                finally:
280
 
                    fout.close()
 
490
                self._put(path_to, fin, mode=mode)
281
491
            finally:
282
492
                fin.close()
283
 
        except (IOError, paramiko.SSHException), x:
284
 
            raise SFTPTransportError('Unable to copy %r to %r' % (path_from, path_to), x)
 
493
        except (IOError, paramiko.SSHException), e:
 
494
            self._translate_io_exception(e, path_from, ': unable copy to: %r' % path_to)
 
495
 
 
496
    def copy_to(self, relpaths, other, mode=None, pb=None):
 
497
        """Copy a set of entries from self into another Transport.
 
498
 
 
499
        :param relpaths: A list/generator of entries to be copied.
 
500
        """
 
501
        if isinstance(other, SFTPTransport) and other._sftp is self._sftp:
 
502
            # Both from & to are on the same remote filesystem
 
503
            # We can use a remote copy, instead of pulling locally, and pushing
 
504
 
 
505
            total = self._get_total(relpaths)
 
506
            count = 0
 
507
            for path in relpaths:
 
508
                path_from = self._abspath(relpath)
 
509
                path_to = other._abspath(relpath)
 
510
                self._update_pb(pb, 'copy-to', count, total)
 
511
                self._copy_abspaths(path_from, path_to, mode=mode)
 
512
                count += 1
 
513
            return count
 
514
        else:
 
515
            return super(SFTPTransport, self).copy_to(relpaths, other, mode=mode, pb=pb)
 
516
 
 
517
    def _rename(self, abs_from, abs_to):
 
518
        """Do a fancy rename on the remote server.
 
519
        
 
520
        Using the implementation provided by osutils.
 
521
        """
 
522
        try:
 
523
            fancy_rename(abs_from, abs_to,
 
524
                    rename_func=self._sftp.rename,
 
525
                    unlink_func=self._sftp.remove)
 
526
        except (IOError, paramiko.SSHException), e:
 
527
            self._translate_io_exception(e, abs_from, ': unable to rename to %r' % (abs_to))
285
528
 
286
529
    def move(self, rel_from, rel_to):
287
530
        """Move the item at rel_from to the location at rel_to"""
288
531
        path_from = self._abspath(rel_from)
289
532
        path_to = self._abspath(rel_to)
290
 
        try:
291
 
            self._sftp.rename(path_from, path_to)
292
 
        except (IOError, paramiko.SSHException), x:
293
 
            raise SFTPTransportError('Unable to move %r to %r' % (path_from, path_to), x)
 
533
        self._rename(path_from, path_to)
294
534
 
295
535
    def delete(self, relpath):
296
536
        """Delete the item at relpath"""
297
537
        path = self._abspath(relpath)
298
538
        try:
299
539
            self._sftp.remove(path)
300
 
        except (IOError, paramiko.SSHException), x:
301
 
            raise SFTPTransportError('Unable to delete %r' % (path,), x)
 
540
        except (IOError, paramiko.SSHException), e:
 
541
            self._translate_io_exception(e, path, ': unable to delete')
302
542
            
303
543
    def listable(self):
304
544
        """Return True if this store supports listing."""
312
552
        path = self._abspath(relpath)
313
553
        try:
314
554
            return self._sftp.listdir(path)
315
 
        except (IOError, paramiko.SSHException), x:
316
 
            raise SFTPTransportError('Unable to list folder %r' % (path,), x)
 
555
        except (IOError, paramiko.SSHException), e:
 
556
            self._translate_io_exception(e, path, ': failed to list_dir')
317
557
 
318
558
    def stat(self, relpath):
319
559
        """Return the stat information for a file."""
320
560
        path = self._abspath(relpath)
321
561
        try:
322
562
            return self._sftp.stat(path)
323
 
        except (IOError, paramiko.SSHException), x:
324
 
            raise SFTPTransportError('Unable to stat %r' % (path,), x)
 
563
        except (IOError, paramiko.SSHException), e:
 
564
            self._translate_io_exception(e, path, ': unable to stat')
325
565
 
326
566
    def lock_read(self, relpath):
327
567
        """
328
568
        Lock the given file for shared (read) access.
329
 
        :return: A lock object, which should be passed to Transport.unlock()
 
569
        :return: A lock object, which has an unlock() member function
330
570
        """
331
571
        # FIXME: there should be something clever i can do here...
332
572
        class BogusLock(object):
341
581
        Lock the given file for exclusive (write) access.
342
582
        WARNING: many transports do not support this, so trying avoid using it
343
583
 
344
 
        :return: A lock object, which should be passed to Transport.unlock()
 
584
        :return: A lock object, which has an unlock() member function
345
585
        """
346
 
        # FIXME: there should be something clever i can do here...
347
 
        class BogusLock(object):
348
 
            def __init__(self, path):
349
 
                self.path = path
350
 
            def unlock(self):
351
 
                pass
352
 
        return BogusLock(relpath)
 
586
        # This is a little bit bogus, but basically, we create a file
 
587
        # which should not already exist, and if it does, we assume
 
588
        # that there is a lock, and if it doesn't, the we assume
 
589
        # that we have taken the lock.
 
590
        return SFTPLock(relpath, self)
353
591
 
354
592
 
355
593
    def _unparse_url(self, path=None):
356
594
        if path is None:
357
595
            path = self._path
358
 
        if self._port == 22:
359
 
            return 'sftp://%s@%s%s' % (self._username, self._host, path)
360
 
        return 'sftp://%s@%s:%d%s' % (self._username, self._host, self._port, path)
 
596
        path = urllib.quote(path)
 
597
        if path.startswith('/'):
 
598
            path = '/%2F' + path[1:]
 
599
        else:
 
600
            path = '/' + path
 
601
        netloc = urllib.quote(self._host)
 
602
        if self._username is not None:
 
603
            netloc = '%s@%s' % (urllib.quote(self._username), netloc)
 
604
        if self._port is not None:
 
605
            netloc = '%s:%d' % (netloc, self._port)
 
606
 
 
607
        return urlparse.urlunparse(('sftp', netloc, path, '', '', ''))
 
608
 
 
609
    def _split_url(self, url):
 
610
        if isinstance(url, unicode):
 
611
            url = url.encode('utf-8')
 
612
        (scheme, netloc, path, params,
 
613
         query, fragment) = urlparse.urlparse(url, allow_fragments=False)
 
614
        assert scheme == 'sftp'
 
615
        username = password = host = port = None
 
616
        if '@' in netloc:
 
617
            username, host = netloc.split('@', 1)
 
618
            if ':' in username:
 
619
                username, password = username.split(':', 1)
 
620
                password = urllib.unquote(password)
 
621
            username = urllib.unquote(username)
 
622
        else:
 
623
            host = netloc
 
624
 
 
625
        if ':' in host:
 
626
            host, port = host.rsplit(':', 1)
 
627
            try:
 
628
                port = int(port)
 
629
            except ValueError:
 
630
                # TODO: Should this be ConnectionError?
 
631
                raise TransportError('%s: invalid port number' % port)
 
632
        host = urllib.unquote(host)
 
633
 
 
634
        path = urllib.unquote(path)
 
635
 
 
636
        # the initial slash should be removed from the path, and treated
 
637
        # as a homedir relative path (the path begins with a double slash
 
638
        # if it is absolute).
 
639
        # see draft-ietf-secsh-scp-sftp-ssh-uri-03.txt
 
640
        if path.startswith('/'):
 
641
            path = path[1:]
 
642
 
 
643
        return (username, password, host, port, path)
361
644
 
362
645
    def _parse_url(self, url):
363
 
        assert url[:7] == 'sftp://'
364
 
        m = self._url_matcher.match(url)
365
 
        if m is None:
366
 
            raise SFTPTransportError('Unable to parse SFTP URL %r' % (url,))
367
 
        self._username, self._password, self._host, self._port, self._path = m.groups()
368
 
        if self._username is None:
369
 
            self._username = getpass.getuser()
370
 
        else:
371
 
            self._username = self._username[:-1]
372
 
        if self._password:
373
 
            self._password = self._password[1:]
374
 
            self._username = self._username[len(self._password)+1:]
375
 
        if self._port is None:
376
 
            self._port = 22
377
 
        else:
378
 
            self._port = int(self._port[1:])
379
 
        if (self._path is None) or (self._path == ''):
380
 
            self._path = '/'
 
646
        (self._username, self._password,
 
647
         self._host, self._port, self._path) = self._split_url(url)
381
648
 
382
649
    def _sftp_connect(self):
 
650
        """Connect to the remote sftp server.
 
651
        After this, self._sftp should have a valid connection (or
 
652
        we raise an TransportError 'could not connect').
 
653
 
 
654
        TODO: Raise a more reasonable ConnectionFailed exception
 
655
        """
 
656
        global _connected_hosts
 
657
 
 
658
        idx = (self._host, self._port, self._username)
 
659
        try:
 
660
            self._sftp = _connected_hosts[idx]
 
661
            return
 
662
        except KeyError:
 
663
            pass
 
664
        
 
665
        vendor = _get_ssh_vendor()
 
666
        if vendor != 'none':
 
667
            sock = SFTPSubprocess(self._host, self._port, self._username)
 
668
            self._sftp = SFTPClient(sock)
 
669
        else:
 
670
            self._paramiko_connect()
 
671
 
 
672
        _connected_hosts[idx] = self._sftp
 
673
 
 
674
    def _paramiko_connect(self):
383
675
        global SYSTEM_HOSTKEYS, BZR_HOSTKEYS
384
676
        
385
677
        load_host_keys()
386
 
        
 
678
 
387
679
        try:
388
 
            t = paramiko.Transport((self._host, self._port))
 
680
            t = paramiko.Transport((self._host, self._port or 22))
389
681
            t.start_client()
390
 
        except paramiko.SSHException:
391
 
            raise SFTPTransportError('Unable to reach SSH host %s:%d' % (self._host, self._port))
 
682
        except paramiko.SSHException, e:
 
683
            raise ConnectionError('Unable to reach SSH host %s:%d' %
 
684
                                  (self._host, self._port), e)
392
685
            
393
686
        server_key = t.get_remote_server_key()
394
687
        server_key_hex = paramiko.util.hexify(server_key.get_fingerprint())
409
702
            save_host_keys()
410
703
        if server_key != our_server_key:
411
704
            filename1 = os.path.expanduser('~/.ssh/known_hosts')
412
 
            filename2 = os.path.join(config_dir(), 'ssh_host_keys')
413
 
            raise SFTPTransportError('Host keys for %s do not match!  %s != %s' % \
 
705
            filename2 = pathjoin(config_dir(), 'ssh_host_keys')
 
706
            raise TransportError('Host keys for %s do not match!  %s != %s' % \
414
707
                (self._host, our_server_key_hex, server_key_hex),
415
708
                ['Try editing %s or %s' % (filename1, filename2)])
416
709
 
417
 
        self._sftp_auth(t, self._username, self._host)
 
710
        self._sftp_auth(t)
418
711
        
419
712
        try:
420
713
            self._sftp = t.open_sftp_client()
421
 
        except paramiko.SSHException:
422
 
            raise BzrError('Unable to find path %s on SFTP server %s' % \
423
 
                (self._path, self._host))
424
 
 
425
 
    def _sftp_auth(self, transport, username, host):
426
 
        agent = paramiko.Agent()
427
 
        for key in agent.get_keys():
428
 
            mutter('Trying SSH agent key %s' % paramiko.util.hexify(key.get_fingerprint()))
429
 
            try:
430
 
                transport.auth_publickey(self._username, key)
431
 
                return
432
 
            except paramiko.SSHException, e:
433
 
                pass
 
714
        except paramiko.SSHException, e:
 
715
            raise ConnectionError('Unable to start sftp client %s:%d' %
 
716
                                  (self._host, self._port), e)
 
717
 
 
718
    def _sftp_auth(self, transport):
 
719
        # paramiko requires a username, but it might be none if nothing was supplied
 
720
        # use the local username, just in case.
 
721
        # We don't override self._username, because if we aren't using paramiko,
 
722
        # the username might be specified in ~/.ssh/config and we don't want to
 
723
        # force it to something else
 
724
        # Also, it would mess up the self.relpath() functionality
 
725
        username = self._username or getpass.getuser()
 
726
 
 
727
        # Paramiko tries to open a socket.AF_UNIX in order to connect
 
728
        # to ssh-agent. That attribute doesn't exist on win32 (it does in cygwin)
 
729
        # so we get an AttributeError exception. For now, just don't try to
 
730
        # connect to an agent if we are on win32
 
731
        if sys.platform != 'win32':
 
732
            agent = paramiko.Agent()
 
733
            for key in agent.get_keys():
 
734
                mutter('Trying SSH agent key %s' % paramiko.util.hexify(key.get_fingerprint()))
 
735
                try:
 
736
                    transport.auth_publickey(username, key)
 
737
                    return
 
738
                except paramiko.SSHException, e:
 
739
                    pass
434
740
        
435
741
        # okay, try finding id_rsa or id_dss?  (posix only)
436
 
        if self._try_pkey_auth(transport, paramiko.RSAKey, 'id_rsa'):
437
 
            return
438
 
        if self._try_pkey_auth(transport, paramiko.DSSKey, 'id_dsa'):
439
 
            return
 
742
        if self._try_pkey_auth(transport, paramiko.RSAKey, username, 'id_rsa'):
 
743
            return
 
744
        if self._try_pkey_auth(transport, paramiko.DSSKey, username, 'id_dsa'):
 
745
            return
 
746
 
440
747
 
441
748
        if self._password:
442
749
            try:
443
 
                transport.auth_password(self._username, self._password)
 
750
                transport.auth_password(username, self._password)
444
751
                return
445
752
            except paramiko.SSHException, e:
446
753
                pass
447
754
 
 
755
            # FIXME: Don't keep a password held in memory if you can help it
 
756
            #self._password = None
 
757
 
448
758
        # give up and ask for a password
449
 
        password = getpass.getpass('SSH %s@%s password: ' % (self._username, self._host))
 
759
        password = bzrlib.ui.ui_factory.get_password(
 
760
                prompt='SSH %(user)s@%(host)s password',
 
761
                user=username, host=self._host)
450
762
        try:
451
 
            transport.auth_password(self._username, password)
452
 
        except paramiko.SSHException:
453
 
            raise SFTPTransportError('Unable to authenticate to SSH host as %s@%s' % \
454
 
                (self._username, self._host))
 
763
            transport.auth_password(username, password)
 
764
        except paramiko.SSHException, e:
 
765
            raise ConnectionError('Unable to authenticate to SSH host as %s@%s' %
 
766
                                  (username, self._host), e)
455
767
 
456
 
    def _try_pkey_auth(self, transport, pkey_class, filename):
 
768
    def _try_pkey_auth(self, transport, pkey_class, username, filename):
457
769
        filename = os.path.expanduser('~/.ssh/' + filename)
458
770
        try:
459
771
            key = pkey_class.from_private_key_file(filename)
460
 
            transport.auth_publickey(self._username, key)
 
772
            transport.auth_publickey(username, key)
461
773
            return True
462
774
        except paramiko.PasswordRequiredException:
463
 
            password = getpass.getpass('SSH %s password: ' % (os.path.basename(filename),))
 
775
            password = bzrlib.ui.ui_factory.get_password(
 
776
                    prompt='SSH %(filename)s password',
 
777
                    filename=filename)
464
778
            try:
465
779
                key = pkey_class.from_private_key_file(filename, password)
466
 
                transport.auth_publickey(self._username, key)
 
780
                transport.auth_publickey(username, key)
467
781
                return True
468
782
            except paramiko.SSHException:
469
783
                mutter('SSH authentication via %s key failed.' % (os.path.basename(filename),))
472
786
        except IOError:
473
787
            pass
474
788
        return False
 
789
 
 
790
    def _sftp_open_exclusive(self, abspath, mode=None):
 
791
        """Open a remote path exclusively.
 
792
 
 
793
        SFTP supports O_EXCL (SFTP_FLAG_EXCL), which fails if
 
794
        the file already exists. However it does not expose this
 
795
        at the higher level of SFTPClient.open(), so we have to
 
796
        sneak away with it.
 
797
 
 
798
        WARNING: This breaks the SFTPClient abstraction, so it
 
799
        could easily break against an updated version of paramiko.
 
800
 
 
801
        :param abspath: The remote absolute path where the file should be opened
 
802
        :param mode: The mode permissions bits for the new file
 
803
        """
 
804
        path = self._sftp._adjust_cwd(abspath)
 
805
        attr = SFTPAttributes()
 
806
        if mode is not None:
 
807
            attr.st_mode = mode
 
808
        omode = (SFTP_FLAG_WRITE | SFTP_FLAG_CREATE 
 
809
                | SFTP_FLAG_TRUNC | SFTP_FLAG_EXCL)
 
810
        try:
 
811
            t, msg = self._sftp._request(CMD_OPEN, path, omode, attr)
 
812
            if t != CMD_HANDLE:
 
813
                raise TransportError('Expected an SFTP handle')
 
814
            handle = msg.get_string()
 
815
            return SFTPFile(self._sftp, handle, 'wb', -1)
 
816
        except (paramiko.SSHException, IOError), e:
 
817
            self._translate_io_exception(e, abspath, ': unable to open',
 
818
                failure_exc=FileExists)
 
819