~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/chroot.py

  • Committer: Andrew Bennetts
  • Date: 2009-09-16 04:37:33 UTC
  • mto: (4634.43.10 bzr-ssh-homedir-take-3)
  • mto: This revision was merged to the branch mainline in revision 4702.
  • Revision ID: andrew.bennetts@canonical.com-20090916043733-mh7tdvkhynz9ya5c
Reimplement ChrootServer/ChrootTransport using bzrlib.transport.pathfilter.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006-2009 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
17
17
"""Implementation of Transport that prevents access to locations above a set
18
18
root.
19
19
"""
20
 
from urlparse import urlparse
21
20
 
22
 
from bzrlib import errors, urlutils
23
21
from bzrlib.transport import (
24
22
    get_transport,
 
23
    pathfilter,
25
24
    register_transport,
26
25
    Server,
27
26
    Transport,
28
27
    unregister_transport,
29
28
    )
30
 
from bzrlib.transport.decorator import TransportDecorator, DecoratorServer
31
 
from bzrlib.transport.memory import MemoryTransport
32
 
 
33
 
 
34
 
class ChrootServer(Server):
 
29
 
 
30
 
 
31
class ChrootServer(pathfilter.PathFilteringServer):
35
32
    """User space 'chroot' facility.
36
33
 
37
34
    The server's get_url returns the url for a chroot transport mapped to the
39
36
    directories of the backing transport are not visible. The chroot url will
40
37
    not allow '..' sequences to result in requests to the chroot affecting
41
38
    directories outside the backing transport.
 
39
 
 
40
    PathFilteringServer does all the path sanitation needed to enforce a
 
41
    chroot, so this is a simple subclass of PathFilteringServer that ignores
 
42
    filter_func.
42
43
    """
43
44
 
44
45
    def __init__(self, backing_transport):
45
 
        self.backing_transport = backing_transport
 
46
        pathfilter.PathFilteringServer.__init__(self, backing_transport, None)
46
47
 
47
48
    def _factory(self, url):
48
49
        return ChrootTransport(self, url)
49
50
 
50
 
    def get_url(self):
51
 
        return self.scheme
52
 
 
53
51
    def setUp(self):
54
52
        self.scheme = 'chroot-%d:///' % id(self)
55
53
        register_transport(self.scheme, self._factory)
56
54
 
57
 
    def tearDown(self):
58
 
        unregister_transport(self.scheme, self._factory)
59
 
 
60
 
 
61
 
class ChrootTransport(Transport):
 
55
 
 
56
class ChrootTransport(pathfilter.PathFilteringTransport):
62
57
    """A ChrootTransport.
63
58
 
64
59
    Please see ChrootServer for details.
65
60
    """
66
61
 
67
 
    def __init__(self, server, base):
68
 
        self.server = server
69
 
        if not base.endswith('/'):
70
 
            base += '/'
71
 
        Transport.__init__(self, base)
72
 
        self.base_path = self.base[len(self.server.scheme)-1:]
73
 
        self.scheme = self.server.scheme
74
 
 
75
 
    def _call(self, methodname, relpath, *args):
76
 
        method = getattr(self.server.backing_transport, methodname)
77
 
        return method(self._safe_relpath(relpath), *args)
78
 
 
79
 
    def _safe_relpath(self, relpath):
80
 
        safe_relpath = self._combine_paths(self.base_path, relpath)
81
 
        if not safe_relpath.startswith('/'):
82
 
            raise ValueError(safe_relpath)
83
 
        return safe_relpath[1:]
84
 
 
85
 
    # Transport methods
86
 
    def abspath(self, relpath):
87
 
        return self.scheme + self._safe_relpath(relpath)
88
 
 
89
 
    def append_file(self, relpath, f, mode=None):
90
 
        return self._call('append_file', relpath, f, mode)
91
 
 
92
 
    def _can_roundtrip_unix_modebits(self):
93
 
        return self.server.backing_transport._can_roundtrip_unix_modebits()
94
 
 
95
 
    def clone(self, relpath):
96
 
        return ChrootTransport(self.server, self.abspath(relpath))
97
 
 
98
 
    def delete(self, relpath):
99
 
        return self._call('delete', relpath)
100
 
 
101
 
    def delete_tree(self, relpath):
102
 
        return self._call('delete_tree', relpath)
103
 
 
104
 
    def external_url(self):
105
 
        """See bzrlib.transport.Transport.external_url."""
106
 
        # Chroots, like MemoryTransport depend on in-process
107
 
        # state and thus the base cannot simply be handed out.
108
 
        # See the base class docstring for more details and
109
 
        # possible directions. For now we return the chrooted
110
 
        # url.
111
 
        return self.server.backing_transport.external_url()
112
 
 
113
 
    def get(self, relpath):
114
 
        return self._call('get', relpath)
115
 
 
116
 
    def has(self, relpath):
117
 
        return self._call('has', relpath)
118
 
 
119
 
    def is_readonly(self):
120
 
        return self.server.backing_transport.is_readonly()
121
 
 
122
 
    def iter_files_recursive(self):
123
 
        backing_transport = self.server.backing_transport.clone(
124
 
            self._safe_relpath('.'))
125
 
        return backing_transport.iter_files_recursive()
126
 
 
127
 
    def listable(self):
128
 
        return self.server.backing_transport.listable()
129
 
 
130
 
    def list_dir(self, relpath):
131
 
        return self._call('list_dir', relpath)
132
 
 
133
 
    def lock_read(self, relpath):
134
 
        return self._call('lock_read', relpath)
135
 
 
136
 
    def lock_write(self, relpath):
137
 
        return self._call('lock_write', relpath)
138
 
 
139
 
    def mkdir(self, relpath, mode=None):
140
 
        return self._call('mkdir', relpath, mode)
141
 
 
142
 
    def open_write_stream(self, relpath, mode=None):
143
 
        return self._call('open_write_stream', relpath, mode)
144
 
 
145
 
    def put_file(self, relpath, f, mode=None):
146
 
        return self._call('put_file', relpath, f, mode)
147
 
 
148
 
    def rename(self, rel_from, rel_to):
149
 
        return self._call('rename', rel_from, self._safe_relpath(rel_to))
150
 
 
151
 
    def rmdir(self, relpath):
152
 
        return self._call('rmdir', relpath)
153
 
 
154
 
    def stat(self, relpath):
155
 
        return self._call('stat', relpath)
 
62
    def _filter(self, relpath):
 
63
        # A simplified version of PathFilteringTransport's _filter that omits
 
64
        # the call to self.server.filter_func.
 
65
        return self._relpath_from_server_root(relpath)
156
66
 
157
67
 
158
68
class TestingChrootServer(ChrootServer):
159
69
 
160
70
    def __init__(self):
161
71
        """TestingChrootServer is not usable until setUp is called."""
 
72
        ChrootServer.__init__(self, None)
162
73
 
163
74
    def setUp(self, backing_server=None):
164
75
        """Setup the Chroot on backing_server."""
171
82
 
172
83
def get_test_permutations():
173
84
    """Return the permutations to be used in testing."""
174
 
    return [(ChrootTransport, TestingChrootServer),
175
 
            ]
 
85
    return [(ChrootTransport, TestingChrootServer)]