← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Vincent Ladeuil
  • Date: 2009-05-04 14:48:21 UTC
  • mto: (4349.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4350.
  • Revision ID: v.ladeuil+lp@free.fr-20090504144821-39dvqkikmd3zqkdg
Handle servers proposing several authentication schemes.

* bzrlib/transport/http/_urllib2_wrappers.py:
(AbstractAuthHandler.auth_required): Several schemes can be
proposed by the server, try to match each one in turn.
(BasicAuthHandler.auth_match): Delete dead code.

* bzrlib/tests/test_http.py:
(load_tests): Separate proxy and http authentication tests as they
require different server setups.
(TestAuth.create_transport_readonly_server): Simplified by using
parameter provided by load_tests.
(TestAuth.test_changing_nonce): Adapt to new parametrization.
(TestProxyAuth.create_transport_readonly_server): Deleted.

* bzrlib/tests/http_utils.py:
(DigestAndBasicAuthRequestHandler, HTTPBasicAndDigestAuthServer,
ProxyBasicAndDigestAuthServer): Add a test server proposing both
basic and digest auth schemes but accepting only digest as valid.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/local.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
20
20
"""
21
21
 
22
22
import os
23
 
from stat import ST_MODE, S_ISDIR, S_IMODE
 
23
from stat import ST_MODE, S_ISDIR, ST_SIZE, S_IMODE
24
24
import sys
25
25
 
26
26
from bzrlib.lazy_import import lazy_import
33
33
    osutils,
34
34
    urlutils,
35
35
    symbol_versioning,
 
36
    transport,
36
37
    )
 
38
from bzrlib.trace import mutter
37
39
from bzrlib.transport import LateReadError
38
40
""")
39
41
 
40
 
from bzrlib import transport
41
 
 
42
 
 
43
 
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
44
 
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
45
 
 
46
 
 
47
 
class LocalTransport(transport.Transport):
 
42
from bzrlib.transport import Transport, Server
 
43
 
 
44
 
 
45
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY
 
46
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY
 
47
 
 
48
 
 
49
class LocalTransport(Transport):
48
50
    """This is the transport agent for local filesystem access."""
49
51
 
50
52
    def __init__(self, base):
97
99
         - relative_reference is url escaped.
98
100
        """
99
101
        if relative_reference in ('.', ''):
100
 
            # _local_base normally has a trailing slash; strip it so that stat
101
 
            # on a transport pointing to a symlink reads the link not the
102
 
            # referent but be careful of / and c:\
103
 
            return osutils.split(self._local_base)[0]
 
102
            return self._local_base
104
103
        return self._local_base + urlutils.unescape(relative_reference)
105
104
 
106
105
    def abspath(self, relpath):
161
160
            transport._file_streams[canonical_url].flush()
162
161
        try:
163
162
            path = self._abspath(relpath)
164
 
            return osutils.open_file(path, 'rb')
 
163
            return open(path, 'rb')
165
164
        except (IOError, OSError),e:
166
165
            if e.errno == errno.EISDIR:
167
166
                return LateReadError(relpath)
205
204
        except (IOError, OSError),e:
206
205
            self._translate_error(e, path)
207
206
        try:
208
 
            if bytes:
209
 
                fp.write(bytes)
 
207
            fp.write(bytes)
210
208
            fp.commit()
211
209
        finally:
212
210
            fp.close()
287
285
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
288
286
                             create_parent_dir=False, dir_mode=None):
289
287
        def writer(fd):
290
 
            if bytes:
291
 
                os.write(fd, bytes)
 
288
            os.write(fd, bytes)
292
289
        self._put_non_atomic_helper(relpath, writer, mode=mode,
293
290
                                    create_parent_dir=create_parent_dir,
294
291
                                    dir_mode=dir_mode)
330
327
        # initialise the file
331
328
        self.put_bytes_non_atomic(relpath, "", mode=mode)
332
329
        abspath = self._abspath(relpath)
333
 
        handle = osutils.open_file(abspath, 'wb')
 
330
        handle = open(abspath, 'wb')
334
331
        if mode is not None:
335
332
            self._check_mode_and_size(abspath, handle.fileno(), mode)
336
333
        transport._file_streams[self.abspath(relpath)] = handle
373
370
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
374
371
        try:
375
372
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
376
 
            if bytes:
377
 
                os.write(fd, bytes)
 
373
            os.write(fd, bytes)
378
374
        finally:
379
375
            os.close(fd)
380
376
        return result
400
396
 
401
397
    def rename(self, rel_from, rel_to):
402
398
        path_from = self._abspath(rel_from)
403
 
        path_to = self._abspath(rel_to)
404
399
        try:
405
400
            # *don't* call bzrlib.osutils.rename, because we want to
406
 
            # detect conflicting names on rename, and osutils.rename tries to
407
 
            # mask cross-platform differences there
408
 
            os.rename(path_from, path_to)
 
401
            # detect errors on rename
 
402
            os.rename(path_from, self._abspath(rel_to))
409
403
        except (IOError, OSError),e:
410
404
            # TODO: What about path_to?
411
405
            self._translate_error(e, path_from)
484
478
        path = relpath
485
479
        try:
486
480
            path = self._abspath(relpath)
487
 
            return os.lstat(path)
 
481
            return os.stat(path)
488
482
        except (IOError, OSError),e:
489
483
            self._translate_error(e, path)
490
484
 
518
512
        except (IOError, OSError),e:
519
513
            self._translate_error(e, path)
520
514
 
521
 
    if osutils.host_os_dereferences_symlinks():
522
 
        def readlink(self, relpath):
523
 
            """See Transport.readlink."""
524
 
            return osutils.readlink(self._abspath(relpath))
525
 
 
526
 
    if osutils.hardlinks_good():
527
 
        def hardlink(self, source, link_name):
528
 
            """See Transport.link."""
529
 
            try:
530
 
                os.link(self._abspath(source), self._abspath(link_name))
531
 
            except (IOError, OSError), e:
532
 
                self._translate_error(e, source)
533
 
 
534
 
    if osutils.has_symlinks():
535
 
        def symlink(self, source, link_name):
536
 
            """See Transport.symlink."""
537
 
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
538
 
            source_rel = urlutils.file_relpath(
539
 
                urlutils.strip_trailing_slash(abs_link_dirpath),
540
 
                urlutils.strip_trailing_slash(self.abspath(source))
541
 
            )
542
 
 
543
 
            try:
544
 
                os.symlink(source_rel, self._abspath(link_name))
545
 
            except (IOError, OSError), e:
546
 
                self._translate_error(e, source_rel)
547
 
 
548
515
    def _can_roundtrip_unix_modebits(self):
549
516
        if sys.platform == 'win32':
550
517
            # anyone else?
584
551
            return EmulatedWin32LocalTransport(abspath)
585
552
 
586
553
 
 
554
class LocalURLServer(Server):
 
555
    """A pretend server for local transports, using file:// urls.
 
556
 
 
557
    Of course no actual server is required to access the local filesystem, so
 
558
    this just exists to tell the test code how to get to it.
 
559
    """
 
560
 
 
561
    def setUp(self):
 
562
        """Setup the server to service requests.
 
563
 
 
564
        :param decorated_transport: ignored by this implementation.
 
565
        """
 
566
 
 
567
    def get_url(self):
 
568
        """See Transport.Server.get_url."""
 
569
        return urlutils.local_path_to_url('')
 
570
 
 
571
 
587
572
def get_test_permutations():
588
573
    """Return the permutations to be used in testing."""
589
 
    from bzrlib.tests import test_server
590
 
    return [(LocalTransport, test_server.LocalURLServer),]
 
574
    return [
 
575
            (LocalTransport, LocalURLServer),
 
576
            ]