← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/http/wsgi.py

  • Committer: Robert Collins
  • Date: 2007-11-18 19:56:39 UTC
  • mfrom: (3006 +trunk)
  • mto: This revision was merged to the branch mainline in revision 3038.
  • Revision ID: robertc@robertcollins.net-20071118195639-m6zf3d5ljjw88kkn
Merge bzr.dev.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/http/wsgi.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
"""WSGI application for bzr HTTP smart server.
18
18
 
20
20
    http://www.python.org/dev/peps/pep-0333/
21
21
"""
22
22
 
23
 
from __future__ import absolute_import
24
 
 
25
23
from cStringIO import StringIO
26
24
 
27
 
from bzrlib.smart import medium
 
25
from bzrlib.smart import protocol
28
26
from bzrlib.transport import chroot, get_transport
29
27
from bzrlib.urlutils import local_path_to_url
30
 
 
31
 
 
32
 
def make_app(root, prefix, path_var='REQUEST_URI', readonly=True,
33
 
    load_plugins=True, enable_logging=True):
 
28
    
 
29
 
 
30
def make_app(root, prefix, path_var, readonly=True):
34
31
    """Convenience function to construct a WSGI bzr smart server.
35
 
 
 
32
    
36
33
    :param root: a local path that requests will be relative to.
37
34
    :param prefix: See RelpathSetter.
38
35
    :param path_var: See RelpathSetter.
42
39
        base_transport = get_transport('readonly+' + local_url)
43
40
    else:
44
41
        base_transport = get_transport(local_url)
45
 
    if load_plugins:
46
 
        from bzrlib.plugin import load_plugins
47
 
        load_plugins()
48
 
    if enable_logging:
49
 
        import bzrlib.trace
50
 
        bzrlib.trace.enable_default_logging()
51
 
    app = SmartWSGIApp(base_transport, prefix)
52
 
    app = RelpathSetter(app, '', path_var)
 
42
    app = SmartWSGIApp(base_transport)
 
43
    app = RelpathSetter(app, prefix, path_var)
53
44
    return app
54
45
 
55
46
 
56
47
class RelpathSetter(object):
57
48
    """WSGI middleware to set 'bzrlib.relpath' in the environ.
58
 
 
 
49
    
59
50
    Different servers can invoke a SmartWSGIApp in different ways.  This
60
51
    middleware allows an adminstrator to configure how to the SmartWSGIApp will
61
52
    determine what path it should be serving for a given request for many common
67
58
    prefix="/some/prefix/" and path_var="REQUEST_URI" will set that request's
68
59
    'bzrlib.relpath' variable to "repo/branch".
69
60
    """
70
 
 
 
61
    
71
62
    def __init__(self, app, prefix='', path_var='REQUEST_URI'):
72
63
        """Constructor.
73
64
 
94
85
class SmartWSGIApp(object):
95
86
    """A WSGI application for the bzr smart server."""
96
87
 
97
 
    def __init__(self, backing_transport, root_client_path='/'):
 
88
    def __init__(self, backing_transport):
98
89
        """Constructor.
99
90
 
100
91
        :param backing_transport: a transport.  Requests will be processed
101
92
            relative to this transport.
102
 
        :param root_client_path: the client path that maps to the root of
103
 
            backing_transport.  This is used to interpret relpaths received from
104
 
            the client.
105
93
        """
106
 
        # Use a ChrootServer so that this web application won't
 
94
        # Use a ChrootTransportDecorator so that this web application won't
107
95
        # accidentally let people access locations they shouldn't.
108
96
        # e.g. consider a smart server request for "get /etc/passwd" or
109
97
        # something.
110
98
        self.chroot_server = chroot.ChrootServer(backing_transport)
111
 
        self.chroot_server.start_server()
 
99
        self.chroot_server.setUp()
112
100
        self.backing_transport = get_transport(self.chroot_server.get_url())
113
 
        self.root_client_path = root_client_path
114
101
        # While the chroot server can technically be torn down at this point,
115
 
        # as all it does is remove the scheme registration from transport's
116
 
        # protocol dictionary, we don't *just in case* there are parts of
 
102
        # as all it does is remove the scheme registration from transport's 
 
103
        # protocol dictionary, we don't *just in case* there are parts of 
117
104
        # bzrlib that will invoke 'get_transport' on urls rather than cloning
118
105
        # around the existing transport.
119
 
        #self.chroot_server.stop_server()
 
106
        #self.chroot_server.tearDown()
120
107
 
121
108
    def __call__(self, environ, start_response):
122
109
        """WSGI application callable."""
125
112
            return []
126
113
 
127
114
        relpath = environ['bzrlib.relpath']
128
 
 
129
 
        if not relpath.startswith('/'):
130
 
            relpath = '/' + relpath
131
 
        if not relpath.endswith('/'):
132
 
            relpath += '/'
133
 
 
134
 
        # Compare the HTTP path (relpath) and root_client_path, and calculate
135
 
        # new relpath and root_client_path accordingly, to be used to build the
136
 
        # request.
137
 
        if relpath.startswith(self.root_client_path):
138
 
            # The relpath traverses all of the mandatory root client path.
139
 
            # Remove the root_client_path from the relpath, and set
140
 
            # adjusted_tcp to None to tell the request handler that no further
141
 
            # path translation is required.
142
 
            adjusted_rcp = None
143
 
            adjusted_relpath = relpath[len(self.root_client_path):]
144
 
        elif self.root_client_path.startswith(relpath):
145
 
            # The relpath traverses some of the mandatory root client path.
146
 
            # Subtract the relpath from the root_client_path, and set the
147
 
            # relpath to '.'.
148
 
            adjusted_rcp = '/' + self.root_client_path[len(relpath):]
149
 
            adjusted_relpath = '.'
150
 
        else:
151
 
            adjusted_rcp = self.root_client_path
152
 
            adjusted_relpath = relpath
153
 
 
154
 
        if adjusted_relpath.startswith('/'):
155
 
            adjusted_relpath = adjusted_relpath[1:]
156
 
        if adjusted_relpath.startswith('/'):
157
 
            raise AssertionError(adjusted_relpath)
158
 
 
159
 
        transport = self.backing_transport.clone(adjusted_relpath)
 
115
        transport = self.backing_transport.clone(relpath)
160
116
        out_buffer = StringIO()
161
117
        request_data_length = int(environ['CONTENT_LENGTH'])
162
118
        request_data_bytes = environ['wsgi.input'].read(request_data_length)
163
119
        smart_protocol_request = self.make_request(
164
 
            transport, out_buffer.write, request_data_bytes, adjusted_rcp)
 
120
            transport, out_buffer.write, request_data_bytes)
165
121
        if smart_protocol_request.next_read_size() != 0:
166
122
            # The request appears to be incomplete, or perhaps it's just a
167
123
            # newer version we don't understand.  Regardless, all we can do
175
131
        start_response('200 OK', headers)
176
132
        return [response_data]
177
133
 
178
 
    def make_request(self, transport, write_func, request_bytes, rcp):
179
 
        protocol_factory, unused_bytes = medium._get_protocol_factory_for_bytes(
180
 
            request_bytes)
181
 
        server_protocol = protocol_factory(
182
 
            transport, write_func, rcp, self.backing_transport)
183
 
        server_protocol.accept_bytes(unused_bytes)
 
134
    def make_request(self, transport, write_func, request_bytes):
 
135
        # XXX: This duplicates the logic in
 
136
        # SmartServerStreamMedium._build_protocol.
 
137
        if request_bytes.startswith(protocol.REQUEST_VERSION_TWO):
 
138
            protocol_class = protocol.SmartServerRequestProtocolTwo
 
139
            request_bytes = request_bytes[len(protocol.REQUEST_VERSION_TWO):]
 
140
        else:
 
141
            protocol_class = protocol.SmartServerRequestProtocolOne
 
142
        server_protocol = protocol_class(transport, write_func)
 
143
        server_protocol.accept_bytes(request_bytes)
184
144
        return server_protocol