1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""WSGI application for bzr HTTP smart server.
For more information about WSGI, see PEP 333:
http://www.python.org/dev/peps/pep-0333/
"""
from cStringIO import StringIO
from bzrlib.smart import protocol
from bzrlib.transport import chroot, get_transport
from bzrlib.urlutils import local_path_to_url
def make_app(root, prefix, path_var, readonly=True):
"""Convenience function to construct a WSGI bzr smart server.
:param root: a local path that requests will be relative to.
:param prefix: See RelpathSetter.
:param path_var: See RelpathSetter.
"""
local_url = local_path_to_url(root)
if readonly:
base_transport = get_transport('readonly+' + local_url)
else:
base_transport = get_transport(local_url)
app = SmartWSGIApp(base_transport)
app = RelpathSetter(app, prefix, path_var)
return app
class RelpathSetter(object):
"""WSGI middleware to set 'bzrlib.relpath' in the environ.
Different servers can invoke a SmartWSGIApp in different ways. This
middleware allows an adminstrator to configure how to the SmartWSGIApp will
determine what path it should be serving for a given request for many common
situations.
For example, a request for "/some/prefix/repo/branch/.bzr/smart" received by
a typical Apache and mod_fastcgi configuration will set `REQUEST_URI` to
"/some/prefix/repo/branch/.bzr/smart". A RelpathSetter with
prefix="/some/prefix/" and path_var="REQUEST_URI" will set that request's
'bzrlib.relpath' variable to "repo/branch".
"""
def __init__(self, app, prefix='', path_var='REQUEST_URI'):
"""Constructor.
:param app: WSGI app to wrap, e.g. a SmartWSGIApp instance.
:param path_var: the variable in the WSGI environ to calculate the
'bzrlib.relpath' variable from.
:param prefix: a prefix to strip from the variable specified in
path_var before setting 'bzrlib.relpath'.
"""
self.app = app
self.prefix = prefix
self.path_var = path_var
def __call__(self, environ, start_response):
path = environ[self.path_var]
suffix = '/.bzr/smart'
if not (path.startswith(self.prefix) and path.endswith(suffix)):
start_response('404 Not Found', [])
return []
environ['bzrlib.relpath'] = path[len(self.prefix):-len(suffix)]
return self.app(environ, start_response)
class SmartWSGIApp(object):
"""A WSGI application for the bzr smart server."""
def __init__(self, backing_transport):
"""Constructor.
:param backing_transport: a transport. Requests will be processed
relative to this transport.
"""
# Use a ChrootTransportDecorator so that this web application won't
# accidentally let people access locations they shouldn't.
# e.g. consider a smart server request for "get /etc/passwd" or
# something.
self.chroot_server = chroot.ChrootServer(backing_transport)
self.chroot_server.setUp()
self.backing_transport = get_transport(self.chroot_server.get_url())
# While the chroot server can technically be torn down at this point,
# as all it does is remove the scheme registration from transport's
# protocol dictionary, we don't *just in case* there are parts of
# bzrlib that will invoke 'get_transport' on urls rather than cloning
# around the existing transport.
#self.chroot_server.tearDown()
def __call__(self, environ, start_response):
"""WSGI application callable."""
if environ['REQUEST_METHOD'] != 'POST':
start_response('405 Method not allowed', [('Allow', 'POST')])
return []
relpath = environ['bzrlib.relpath']
transport = self.backing_transport.clone(relpath)
out_buffer = StringIO()
request_data_length = int(environ['CONTENT_LENGTH'])
request_data_bytes = environ['wsgi.input'].read(request_data_length)
smart_protocol_request = self.make_request(
transport, out_buffer.write, request_data_bytes)
if smart_protocol_request.next_read_size() != 0:
# The request appears to be incomplete, or perhaps it's just a
# newer version we don't understand. Regardless, all we can do
# is return an error response in the format of our version of the
# protocol.
response_data = 'error\x01incomplete request\n'
else:
response_data = out_buffer.getvalue()
headers = [('Content-type', 'application/octet-stream')]
headers.append(("Content-Length", str(len(response_data))))
start_response('200 OK', headers)
return [response_data]
def make_request(self, transport, write_func, request_bytes):
# XXX: This duplicates the logic in
# SmartServerStreamMedium._build_protocol.
if request_bytes.startswith(protocol.REQUEST_VERSION_TWO):
protocol_class = protocol.SmartServerRequestProtocolTwo
request_bytes = request_bytes[len(protocol.REQUEST_VERSION_TWO):]
else:
protocol_class = protocol.SmartServerRequestProtocolOne
server_protocol = protocol_class(transport, write_func)
server_protocol.accept_bytes(request_bytes)
return server_protocol
|