← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/chroot.py

  • Committer: Robert Collins
  • Date: 2007-05-07 17:49:30 UTC
  • mto: This revision was merged to the branch mainline in revision 2485.
  • Revision ID: robertc@robertcollins.net-20070507174930-gksqv90l5emjrcyq
Add the performance roadmap rationale.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/chroot.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Implementation of Transport that prevents access to locations above a set
18
18
root.
19
19
"""
 
20
from urlparse import urlparse
20
21
 
 
22
from bzrlib import errors, urlutils
21
23
from bzrlib.transport import (
22
 
    pathfilter,
 
24
    get_transport,
23
25
    register_transport,
 
26
    Server,
 
27
    Transport,
 
28
    unregister_transport,
24
29
    )
25
 
 
26
 
 
27
 
class ChrootServer(pathfilter.PathFilteringServer):
 
30
from bzrlib.transport.decorator import TransportDecorator, DecoratorServer
 
31
from bzrlib.transport.memory import MemoryTransport
 
32
 
 
33
 
 
34
class ChrootServer(Server):
28
35
    """User space 'chroot' facility.
29
 
 
 
36
    
30
37
    The server's get_url returns the url for a chroot transport mapped to the
31
38
    backing transport. The url is of the form chroot-xxx:/// so parent
32
39
    directories of the backing transport are not visible. The chroot url will
33
40
    not allow '..' sequences to result in requests to the chroot affecting
34
41
    directories outside the backing transport.
35
 
 
36
 
    PathFilteringServer does all the path sanitation needed to enforce a
37
 
    chroot, so this is a simple subclass of PathFilteringServer that ignores
38
 
    filter_func.
39
42
    """
40
43
 
41
44
    def __init__(self, backing_transport):
42
 
        pathfilter.PathFilteringServer.__init__(self, backing_transport, None)
 
45
        self.backing_transport = backing_transport
43
46
 
44
47
    def _factory(self, url):
 
48
        assert url.startswith(self.scheme)
45
49
        return ChrootTransport(self, url)
46
50
 
47
 
    def start_server(self):
 
51
    def get_url(self):
 
52
        return self.scheme
 
53
 
 
54
    def setUp(self):
48
55
        self.scheme = 'chroot-%d:///' % id(self)
49
56
        register_transport(self.scheme, self._factory)
50
57
 
51
 
 
52
 
class ChrootTransport(pathfilter.PathFilteringTransport):
 
58
    def tearDown(self):
 
59
        unregister_transport(self.scheme, self._factory)
 
60
 
 
61
 
 
62
class ChrootTransport(Transport):
53
63
    """A ChrootTransport.
54
64
 
55
65
    Please see ChrootServer for details.
56
66
    """
57
67
 
58
 
    def _filter(self, relpath):
59
 
        # A simplified version of PathFilteringTransport's _filter that omits
60
 
        # the call to self.server.filter_func.
61
 
        return self._relpath_from_server_root(relpath)
 
68
    def __init__(self, server, base):
 
69
        self.server = server
 
70
        if not base.endswith('/'):
 
71
            base += '/'
 
72
        Transport.__init__(self, base)
 
73
        self.base_path = self.base[len(self.server.scheme)-1:]
 
74
        self.scheme = self.server.scheme
 
75
 
 
76
    def _call(self, methodname, relpath, *args):
 
77
        method = getattr(self.server.backing_transport, methodname)
 
78
        return method(self._safe_relpath(relpath), *args)
 
79
 
 
80
    def _safe_relpath(self, relpath):
 
81
        safe_relpath = self._combine_paths(self.base_path, relpath)
 
82
        assert safe_relpath.startswith('/')
 
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 clone(self, relpath):
 
93
        return ChrootTransport(self.server, self.abspath(relpath))
 
94
 
 
95
    def delete(self, relpath):
 
96
        return self._call('delete', relpath)
 
97
 
 
98
    def delete_tree(self, relpath):
 
99
        return self._call('delete_tree', relpath)
 
100
 
 
101
    def get(self, relpath):
 
102
        return self._call('get', relpath)
 
103
 
 
104
    def has(self, relpath):
 
105
        return self._call('has', relpath)
 
106
 
 
107
    def iter_files_recursive(self):
 
108
        backing_transport = self.server.backing_transport.clone(
 
109
            self._safe_relpath('.'))
 
110
        return backing_transport.iter_files_recursive()
 
111
 
 
112
    def listable(self):
 
113
        return self.server.backing_transport.listable()
 
114
 
 
115
    def list_dir(self, relpath):
 
116
        return self._call('list_dir', relpath)
 
117
 
 
118
    def lock_read(self, relpath):
 
119
        return self._call('lock_read', relpath)
 
120
 
 
121
    def lock_write(self, relpath):
 
122
        return self._call('lock_write', relpath)
 
123
 
 
124
    def mkdir(self, relpath, mode=None):
 
125
        return self._call('mkdir', relpath, mode)
 
126
 
 
127
    def put_file(self, relpath, f, mode=None):
 
128
        return self._call('put_file', relpath, f, mode)
 
129
 
 
130
    def rename(self, rel_from, rel_to):
 
131
        return self._call('rename', rel_from, self._safe_relpath(rel_to))
 
132
 
 
133
    def rmdir(self, relpath):
 
134
        return self._call('rmdir', relpath)
 
135
 
 
136
    def stat(self, relpath):
 
137
        return self._call('stat', relpath)
 
138
 
 
139
 
 
140
class TestingChrootServer(ChrootServer):
 
141
 
 
142
    def __init__(self):
 
143
        """TestingChrootServer is not usable until setUp is called."""
 
144
 
 
145
    def setUp(self, backing_server=None):
 
146
        """Setup the Chroot on backing_server."""
 
147
        if backing_server is not None:
 
148
            self.backing_transport = get_transport(backing_server.get_url())
 
149
        else:
 
150
            self.backing_transport = get_transport('.')
 
151
        ChrootServer.setUp(self)
62
152
 
63
153
 
64
154
def get_test_permutations():
65
155
    """Return the permutations to be used in testing."""
66
 
    from bzrlib.tests import test_server
67
 
    return [(ChrootTransport, test_server.TestingChrootServer)]
 
156
    return [(ChrootTransport, TestingChrootServer),
 
157
            ]