/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/smart/request.py

  • Committer: Martin
  • Date: 2010-05-03 20:57:39 UTC
  • mto: This revision was merged to the branch mainline in revision 5204.
  • Revision ID: gzlist@googlemail.com-20100503205739-n326zdvevv0rmruh
Retain original stack and error message when translating to ValueError in bencode

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006-2010 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
"""Basic server-side logic for dealing with requests."""
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""Infrastructure for server-side request handlers.
 
18
 
 
19
Interesting module attributes:
 
20
    * The request_handlers registry maps verb names to SmartServerRequest
 
21
      classes.
 
22
    * The jail_info threading.local() object is used to prevent accidental
 
23
      opening of BzrDirs outside of the backing transport, or any other
 
24
      transports placed in jail_info.transports.  The jail_info is reset on
 
25
      every call into a request handler (which can happen an arbitrary number
 
26
      of times during a request).
 
27
"""
 
28
 
 
29
# XXX: The class names are a little confusing: the protocol will instantiate a
 
30
# SmartServerRequestHandler, whose dispatch_command method creates an instance
 
31
# of a SmartServerRequest subclass.
18
32
 
19
33
 
20
34
import tempfile
21
 
 
22
 
from bzrlib import bzrdir, errors, registry, revision
23
 
from bzrlib.bundle.serializer import write_bundle
 
35
import thread
 
36
import threading
 
37
 
 
38
from bzrlib import (
 
39
    bzrdir,
 
40
    debug,
 
41
    errors,
 
42
    osutils,
 
43
    registry,
 
44
    revision,
 
45
    trace,
 
46
    urlutils,
 
47
    )
 
48
from bzrlib.lazy_import import lazy_import
 
49
lazy_import(globals(), """
 
50
from bzrlib.bundle import serializer
 
51
""")
 
52
 
 
53
 
 
54
jail_info = threading.local()
 
55
jail_info.transports = None
 
56
 
 
57
 
 
58
def _install_hook():
 
59
    bzrdir.BzrDir.hooks.install_named_hook(
 
60
        'pre_open', _pre_open_hook, 'checking server jail')
 
61
 
 
62
 
 
63
def _pre_open_hook(transport):
 
64
    allowed_transports = getattr(jail_info, 'transports', None)
 
65
    if allowed_transports is None:
 
66
        return
 
67
    abspath = transport.base
 
68
    for allowed_transport in allowed_transports:
 
69
        try:
 
70
            allowed_transport.relpath(abspath)
 
71
        except errors.PathNotChild:
 
72
            continue
 
73
        else:
 
74
            return
 
75
    raise errors.JailBreak(abspath)
 
76
 
 
77
 
 
78
_install_hook()
24
79
 
25
80
 
26
81
class SmartServerRequest(object):
27
82
    """Base class for request handlers.
 
83
 
 
84
    To define a new request, subclass this class and override the `do` method
 
85
    (and if appropriate, `do_body` as well).  Request implementors should take
 
86
    care to call `translate_client_path` and `transport_from_client_path` as
 
87
    appropriate when dealing with paths received from the client.
28
88
    """
29
 
 
30
 
    def __init__(self, backing_transport):
 
89
    # XXX: rename this class to BaseSmartServerRequestHandler ?  A request
 
90
    # *handler* is a different concept to the request.
 
91
 
 
92
    def __init__(self, backing_transport, root_client_path='/', jail_root=None):
 
93
        """Constructor.
 
94
 
 
95
        :param backing_transport: the base transport to be used when performing
 
96
            this request.
 
97
        :param root_client_path: the client path that maps to the root of
 
98
            backing_transport.  This is used to interpret relpaths received
 
99
            from the client.  Clients will not be able to refer to paths above
 
100
            this root.  If root_client_path is None, then no translation will
 
101
            be performed on client paths.  Default is '/'.
 
102
        :param jail_root: if specified, the root of the BzrDir.open jail to use
 
103
            instead of backing_transport.
 
104
        """
31
105
        self._backing_transport = backing_transport
 
106
        if jail_root is None:
 
107
            jail_root = backing_transport
 
108
        self._jail_root = jail_root
 
109
        if root_client_path is not None:
 
110
            if not root_client_path.startswith('/'):
 
111
                root_client_path = '/' + root_client_path
 
112
            if not root_client_path.endswith('/'):
 
113
                root_client_path += '/'
 
114
        self._root_client_path = root_client_path
 
115
        self._body_chunks = []
32
116
 
33
117
    def _check_enabled(self):
34
118
        """Raises DisabledMethod if this method is disabled."""
36
120
 
37
121
    def do(self, *args):
38
122
        """Mandatory extension point for SmartServerRequest subclasses.
39
 
        
 
123
 
40
124
        Subclasses must implement this.
41
 
        
 
125
 
42
126
        This should return a SmartServerResponse if this command expects to
43
127
        receive no body.
44
128
        """
57
141
 
58
142
    def do_body(self, body_bytes):
59
143
        """Called if the client sends a body with the request.
60
 
        
 
144
 
 
145
        The do() method is still called, and must have returned None.
 
146
 
61
147
        Must return a SmartServerResponse.
62
148
        """
63
 
        # TODO: if a client erroneously sends a request that shouldn't have a
64
 
        # body, what to do?  Probably SmartServerRequestHandler should catch
65
 
        # this NotImplementedError and translate it into a 'bad request' error
66
 
        # to send to the client.
67
 
        raise NotImplementedError(self.do_body)
 
149
        if body_bytes != '':
 
150
            raise errors.SmartProtocolError('Request does not expect a body')
 
151
 
 
152
    def do_chunk(self, chunk_bytes):
 
153
        """Called with each body chunk if the request has a streamed body.
 
154
 
 
155
        The do() method is still called, and must have returned None.
 
156
        """
 
157
        self._body_chunks.append(chunk_bytes)
 
158
 
 
159
    def do_end(self):
 
160
        """Called when the end of the request has been received."""
 
161
        body_bytes = ''.join(self._body_chunks)
 
162
        self._body_chunks = None
 
163
        return self.do_body(body_bytes)
 
164
 
 
165
    def setup_jail(self):
 
166
        jail_info.transports = [self._jail_root]
 
167
 
 
168
    def teardown_jail(self):
 
169
        jail_info.transports = None
 
170
 
 
171
    def translate_client_path(self, client_path):
 
172
        """Translate a path received from a network client into a local
 
173
        relpath.
 
174
 
 
175
        All paths received from the client *must* be translated.
 
176
 
 
177
        :param client_path: the path from the client.
 
178
        :returns: a relpath that may be used with self._backing_transport
 
179
            (unlike the untranslated client_path, which must not be used with
 
180
            the backing transport).
 
181
        """
 
182
        if self._root_client_path is None:
 
183
            # no translation necessary!
 
184
            return client_path
 
185
        if not client_path.startswith('/'):
 
186
            client_path = '/' + client_path
 
187
        if client_path + '/' == self._root_client_path:
 
188
            return '.'
 
189
        if client_path.startswith(self._root_client_path):
 
190
            path = client_path[len(self._root_client_path):]
 
191
            relpath = urlutils.joinpath('/', path)
 
192
            if not relpath.startswith('/'):
 
193
                raise ValueError(relpath)
 
194
            return urlutils.escape('.' + relpath)
 
195
        else:
 
196
            raise errors.PathNotChild(client_path, self._root_client_path)
 
197
 
 
198
    def transport_from_client_path(self, client_path):
 
199
        """Get a backing transport corresponding to the location referred to by
 
200
        a network client.
 
201
 
 
202
        :seealso: translate_client_path
 
203
        :returns: a transport cloned from self._backing_transport
 
204
        """
 
205
        relpath = self.translate_client_path(client_path)
 
206
        return self._backing_transport.clone(relpath)
68
207
 
69
208
 
70
209
class SmartServerResponse(object):
71
 
    """Response generated by SmartServerRequestHandler."""
72
 
 
73
 
    def __init__(self, args, body=None):
 
210
    """A response to a client request.
 
211
 
 
212
    This base class should not be used. Instead use
 
213
    SuccessfulSmartServerResponse and FailedSmartServerResponse as appropriate.
 
214
    """
 
215
 
 
216
    def __init__(self, args, body=None, body_stream=None):
 
217
        """Constructor.
 
218
 
 
219
        :param args: tuple of response arguments.
 
220
        :param body: string of a response body.
 
221
        :param body_stream: iterable of bytestrings to be streamed to the
 
222
            client.
 
223
        """
74
224
        self.args = args
 
225
        if body is not None and body_stream is not None:
 
226
            raise errors.BzrError(
 
227
                "'body' and 'body_stream' are mutually exclusive.")
75
228
        self.body = body
 
229
        self.body_stream = body_stream
76
230
 
77
231
    def __eq__(self, other):
78
232
        if other is None:
79
233
            return False
80
 
        return other.args == self.args and other.body == self.body
 
234
        return (other.args == self.args and
 
235
                other.body == self.body and
 
236
                other.body_stream is self.body_stream)
81
237
 
82
238
    def __repr__(self):
83
 
        return "<SmartServerResponse args=%r body=%r>" % (self.args, self.body)
 
239
        return "<%s args=%r body=%r>" % (self.__class__.__name__,
 
240
            self.args, self.body)
 
241
 
 
242
 
 
243
class FailedSmartServerResponse(SmartServerResponse):
 
244
    """A SmartServerResponse for a request which failed."""
 
245
 
 
246
    def is_successful(self):
 
247
        """FailedSmartServerResponse are not successful."""
 
248
        return False
 
249
 
 
250
 
 
251
class SuccessfulSmartServerResponse(SmartServerResponse):
 
252
    """A SmartServerResponse for a successfully completed request."""
 
253
 
 
254
    def is_successful(self):
 
255
        """SuccessfulSmartServerResponse are successful."""
 
256
        return True
84
257
 
85
258
 
86
259
class SmartServerRequestHandler(object):
87
260
    """Protocol logic for smart server.
88
 
    
 
261
 
89
262
    This doesn't handle serialization at all, it just processes requests and
90
263
    creates responses.
91
264
    """
100
273
    # TODO: Better way of representing the body for commands that take it,
101
274
    # and allow it to be streamed into the server.
102
275
 
103
 
    def __init__(self, backing_transport, commands):
 
276
    def __init__(self, backing_transport, commands, root_client_path,
 
277
        jail_root=None):
104
278
        """Constructor.
105
279
 
106
280
        :param backing_transport: a Transport to handle requests for.
108
282
            subclasses. e.g. bzrlib.transport.smart.vfs.vfs_commands.
109
283
        """
110
284
        self._backing_transport = backing_transport
 
285
        self._root_client_path = root_client_path
111
286
        self._commands = commands
112
 
        self._body_bytes = ''
 
287
        if jail_root is None:
 
288
            jail_root = backing_transport
 
289
        self._jail_root = jail_root
113
290
        self.response = None
114
291
        self.finished_reading = False
115
292
        self._command = None
 
293
        if 'hpss' in debug.debug_flags:
 
294
            self._request_start_time = osutils.timer_func()
 
295
            self._thread_id = thread.get_ident()
 
296
 
 
297
    def _trace(self, action, message, extra_bytes=None, include_time=False):
 
298
        # It is a bit of a shame that this functionality overlaps with that of 
 
299
        # ProtocolThreeRequester._trace. However, there is enough difference
 
300
        # that just putting it in a helper doesn't help a lot. And some state
 
301
        # is taken from the instance.
 
302
        if include_time:
 
303
            t = '%5.3fs ' % (osutils.timer_func() - self._request_start_time)
 
304
        else:
 
305
            t = ''
 
306
        if extra_bytes is None:
 
307
            extra = ''
 
308
        else:
 
309
            extra = ' ' + repr(extra_bytes[:40])
 
310
            if len(extra) > 33:
 
311
                extra = extra[:29] + extra[-1] + '...'
 
312
        trace.mutter('%12s: [%s] %s%s%s'
 
313
                     % (action, self._thread_id, t, message, extra))
116
314
 
117
315
    def accept_body(self, bytes):
118
316
        """Accept body data."""
119
 
 
120
 
        # TODO: This should be overriden for each command that desired body data
121
 
        # to handle the right format of that data, i.e. plain bytes, a bundle,
122
 
        # etc.  The deserialisation into that format should be done in the
123
 
        # Protocol object.
124
 
 
125
 
        # default fallback is to accumulate bytes.
126
 
        self._body_bytes += bytes
127
 
        
 
317
        if self._command is None:
 
318
            # no active command object, so ignore the event.
 
319
            return
 
320
        self._run_handler_code(self._command.do_chunk, (bytes,), {})
 
321
        if 'hpss' in debug.debug_flags:
 
322
            self._trace('accept body',
 
323
                        '%d bytes' % (len(bytes),), bytes)
 
324
 
128
325
    def end_of_body(self):
129
326
        """No more body data will be received."""
130
 
        self._run_handler_code(self._command.do_body, (self._body_bytes,), {})
 
327
        self._run_handler_code(self._command.do_end, (), {})
131
328
        # cannot read after this.
132
329
        self.finished_reading = True
133
 
 
134
 
    def dispatch_command(self, cmd, args):
135
 
        """Deprecated compatibility method.""" # XXX XXX
136
 
        try:
137
 
            command = self._commands.get(cmd)
138
 
        except LookupError:
139
 
            raise errors.SmartProtocolError("bad request %r" % (cmd,))
140
 
        self._command = command(self._backing_transport)
141
 
        self._run_handler_code(self._command.execute, args, {})
 
330
        if 'hpss' in debug.debug_flags:
 
331
            self._trace('end of body', '', include_time=True)
142
332
 
143
333
    def _run_handler_code(self, callable, args, kwargs):
144
334
        """Run some handler specific code 'callable'.
160
350
        # XXX: most of this error conversion is VFS-related, and thus ought to
161
351
        # be in SmartServerVFSRequestHandler somewhere.
162
352
        try:
163
 
            return callable(*args, **kwargs)
164
 
        except errors.NoSuchFile, e:
165
 
            return SmartServerResponse(('NoSuchFile', e.path))
166
 
        except errors.FileExists, e:
167
 
            return SmartServerResponse(('FileExists', e.path))
168
 
        except errors.DirectoryNotEmpty, e:
169
 
            return SmartServerResponse(('DirectoryNotEmpty', e.path))
170
 
        except errors.ShortReadvError, e:
171
 
            return SmartServerResponse(('ShortReadvError',
172
 
                e.path, str(e.offset), str(e.length), str(e.actual)))
173
 
        except UnicodeError, e:
174
 
            # If it is a DecodeError, than most likely we are starting
175
 
            # with a plain string
176
 
            str_or_unicode = e.object
177
 
            if isinstance(str_or_unicode, unicode):
178
 
                # XXX: UTF-8 might have \x01 (our seperator byte) in it.  We
179
 
                # should escape it somehow.
180
 
                val = 'u:' + str_or_unicode.encode('utf-8')
181
 
            else:
182
 
                val = 's:' + str_or_unicode.encode('base64')
183
 
            # This handles UnicodeEncodeError or UnicodeDecodeError
184
 
            return SmartServerResponse((e.__class__.__name__,
185
 
                    e.encoding, val, str(e.start), str(e.end), e.reason))
186
 
        except errors.TransportNotPossible, e:
187
 
            if e.msg == "readonly transport":
188
 
                return SmartServerResponse(('ReadOnlyError', ))
189
 
            else:
190
 
                raise
 
353
            self._command.setup_jail()
 
354
            try:
 
355
                return callable(*args, **kwargs)
 
356
            finally:
 
357
                self._command.teardown_jail()
 
358
        except (KeyboardInterrupt, SystemExit):
 
359
            raise
 
360
        except Exception, err:
 
361
            err_struct = _translate_error(err)
 
362
            return FailedSmartServerResponse(err_struct)
 
363
 
 
364
    def headers_received(self, headers):
 
365
        # Just a no-op at the moment.
 
366
        if 'hpss' in debug.debug_flags:
 
367
            self._trace('headers', repr(headers))
 
368
 
 
369
    def args_received(self, args):
 
370
        cmd = args[0]
 
371
        args = args[1:]
 
372
        try:
 
373
            command = self._commands.get(cmd)
 
374
        except LookupError:
 
375
            if 'hpss' in debug.debug_flags:
 
376
                self._trace('hpss unknown request', 
 
377
                            cmd, repr(args)[1:-1])
 
378
            raise errors.UnknownSmartMethod(cmd)
 
379
        if 'hpss' in debug.debug_flags:
 
380
            from bzrlib.smart import vfs
 
381
            if issubclass(command, vfs.VfsRequest):
 
382
                action = 'hpss vfs req'
 
383
            else:
 
384
                action = 'hpss request'
 
385
            self._trace(action, 
 
386
                        '%s %s' % (cmd, repr(args)[1:-1]))
 
387
        self._command = command(
 
388
            self._backing_transport, self._root_client_path, self._jail_root)
 
389
        self._run_handler_code(self._command.execute, args, {})
 
390
 
 
391
    def end_received(self):
 
392
        if self._command is None:
 
393
            # no active command object, so ignore the event.
 
394
            return
 
395
        self._run_handler_code(self._command.do_end, (), {})
 
396
        if 'hpss' in debug.debug_flags:
 
397
            self._trace('end', '', include_time=True)
 
398
 
 
399
    def post_body_error_received(self, error_args):
 
400
        # Just a no-op at the moment.
 
401
        pass
 
402
 
 
403
 
 
404
def _translate_error(err):
 
405
    if isinstance(err, errors.NoSuchFile):
 
406
        return ('NoSuchFile', err.path)
 
407
    elif isinstance(err, errors.FileExists):
 
408
        return ('FileExists', err.path)
 
409
    elif isinstance(err, errors.DirectoryNotEmpty):
 
410
        return ('DirectoryNotEmpty', err.path)
 
411
    elif isinstance(err, errors.IncompatibleRepositories):
 
412
        return ('IncompatibleRepositories', str(err.source), str(err.target),
 
413
            str(err.details))
 
414
    elif isinstance(err, errors.ShortReadvError):
 
415
        return ('ShortReadvError', err.path, str(err.offset), str(err.length),
 
416
                str(err.actual))
 
417
    elif isinstance(err, errors.UnstackableRepositoryFormat):
 
418
        return (('UnstackableRepositoryFormat', str(err.format), err.url))
 
419
    elif isinstance(err, errors.UnstackableBranchFormat):
 
420
        return ('UnstackableBranchFormat', str(err.format), err.url)
 
421
    elif isinstance(err, errors.NotStacked):
 
422
        return ('NotStacked',)
 
423
    elif isinstance(err, UnicodeError):
 
424
        # If it is a DecodeError, than most likely we are starting
 
425
        # with a plain string
 
426
        str_or_unicode = err.object
 
427
        if isinstance(str_or_unicode, unicode):
 
428
            # XXX: UTF-8 might have \x01 (our protocol v1 and v2 seperator
 
429
            # byte) in it, so this encoding could cause broken responses.
 
430
            # Newer clients use protocol v3, so will be fine.
 
431
            val = 'u:' + str_or_unicode.encode('utf-8')
 
432
        else:
 
433
            val = 's:' + str_or_unicode.encode('base64')
 
434
        # This handles UnicodeEncodeError or UnicodeDecodeError
 
435
        return (err.__class__.__name__, err.encoding, val, str(err.start),
 
436
                str(err.end), err.reason)
 
437
    elif isinstance(err, errors.TransportNotPossible):
 
438
        if err.msg == "readonly transport":
 
439
            return ('ReadOnlyError', )
 
440
    elif isinstance(err, errors.ReadError):
 
441
        # cannot read the file
 
442
        return ('ReadError', err.path)
 
443
    elif isinstance(err, errors.PermissionDenied):
 
444
        return ('PermissionDenied', err.path, err.extra)
 
445
    elif isinstance(err, errors.TokenMismatch):
 
446
        return ('TokenMismatch', err.given_token, err.lock_token)
 
447
    elif isinstance(err, errors.LockContention):
 
448
        return ('LockContention',)
 
449
    # Unserialisable error.  Log it, and return a generic error
 
450
    trace.log_exception_quietly()
 
451
    return ('error', str(err))
191
452
 
192
453
 
193
454
class HelloRequest(SmartServerRequest):
194
 
    """Answer a version request with my version."""
 
455
    """Answer a version request with the highest protocol version this server
 
456
    supports.
 
457
    """
195
458
 
196
459
    def do(self):
197
 
        return SmartServerResponse(('ok', '1'))
 
460
        return SuccessfulSmartServerResponse(('ok', '2'))
198
461
 
199
462
 
200
463
class GetBundleRequest(SmartServerRequest):
 
464
    """Get a bundle of from the null revision to the specified revision."""
201
465
 
202
466
    def do(self, path, revision_id):
203
467
        # open transport relative to our base
204
 
        t = self._backing_transport.clone(path)
 
468
        t = self.transport_from_client_path(path)
205
469
        control, extra_path = bzrdir.BzrDir.open_containing_from_transport(t)
206
470
        repo = control.open_repository()
207
471
        tmpf = tempfile.TemporaryFile()
208
472
        base_revision = revision.NULL_REVISION
209
 
        write_bundle(repo, revision_id, base_revision, tmpf)
 
473
        serializer.write_bundle(repo, revision_id, base_revision, tmpf)
210
474
        tmpf.seek(0)
211
 
        return SmartServerResponse((), tmpf.read())
212
 
 
213
 
 
214
 
# This exists solely to help RemoteObjectHacking.  It should be removed
215
 
# eventually.  It should not be considered part of the real smart server
216
 
# protocol!
217
 
class ProbeDontUseRequest(SmartServerRequest):
218
 
 
219
 
    def do(self, path):
220
 
        from bzrlib.bzrdir import BzrDirFormat
221
 
        t = self._backing_transport.clone(path)
222
 
        default_format = BzrDirFormat.get_default_format()
223
 
        real_bzrdir = default_format.open(t, _found=True)
224
 
        try:
225
 
            real_bzrdir._format.probe_transport(t)
226
 
        except (errors.NotBranchError, errors.UnknownFormatError):
227
 
            answer = 'no'
228
 
        else:
229
 
            answer = 'yes'
230
 
        return SmartServerResponse((answer,))
 
475
        return SuccessfulSmartServerResponse((), tmpf.read())
231
476
 
232
477
 
233
478
class SmartServerIsReadonly(SmartServerRequest):
238
483
            answer = 'yes'
239
484
        else:
240
485
            answer = 'no'
241
 
        return SmartServerResponse((answer,))
 
486
        return SuccessfulSmartServerResponse((answer,))
242
487
 
243
488
 
244
489
request_handlers = registry.Registry()
245
490
request_handlers.register_lazy(
246
491
    'append', 'bzrlib.smart.vfs', 'AppendRequest')
247
492
request_handlers.register_lazy(
248
 
    'Branch.get_config_file', 'bzrlib.smart.branch', 'SmartServerBranchGetConfigFile')
 
493
    'Branch.get_config_file', 'bzrlib.smart.branch',
 
494
    'SmartServerBranchGetConfigFile')
 
495
request_handlers.register_lazy(
 
496
    'Branch.get_parent', 'bzrlib.smart.branch', 'SmartServerBranchGetParent')
 
497
request_handlers.register_lazy(
 
498
    'Branch.get_tags_bytes', 'bzrlib.smart.branch',
 
499
    'SmartServerBranchGetTagsBytes')
 
500
request_handlers.register_lazy(
 
501
    'Branch.set_tags_bytes', 'bzrlib.smart.branch',
 
502
    'SmartServerBranchSetTagsBytes')
 
503
request_handlers.register_lazy(
 
504
    'Branch.get_stacked_on_url', 'bzrlib.smart.branch', 'SmartServerBranchRequestGetStackedOnURL')
249
505
request_handlers.register_lazy(
250
506
    'Branch.last_revision_info', 'bzrlib.smart.branch', 'SmartServerBranchRequestLastRevisionInfo')
251
507
request_handlers.register_lazy(
252
508
    'Branch.lock_write', 'bzrlib.smart.branch', 'SmartServerBranchRequestLockWrite')
253
 
request_handlers.register_lazy(
254
 
    'Branch.revision_history', 'bzrlib.smart.branch', 'SmartServerRequestRevisionHistory')
255
 
request_handlers.register_lazy(
256
 
    'Branch.set_last_revision', 'bzrlib.smart.branch', 'SmartServerBranchRequestSetLastRevision')
 
509
request_handlers.register_lazy( 'Branch.revision_history',
 
510
    'bzrlib.smart.branch', 'SmartServerRequestRevisionHistory')
 
511
request_handlers.register_lazy( 'Branch.set_config_option',
 
512
    'bzrlib.smart.branch', 'SmartServerBranchRequestSetConfigOption')
 
513
request_handlers.register_lazy( 'Branch.set_last_revision',
 
514
    'bzrlib.smart.branch', 'SmartServerBranchRequestSetLastRevision')
 
515
request_handlers.register_lazy(
 
516
    'Branch.set_last_revision_info', 'bzrlib.smart.branch',
 
517
    'SmartServerBranchRequestSetLastRevisionInfo')
 
518
request_handlers.register_lazy(
 
519
    'Branch.set_last_revision_ex', 'bzrlib.smart.branch',
 
520
    'SmartServerBranchRequestSetLastRevisionEx')
 
521
request_handlers.register_lazy(
 
522
    'Branch.set_parent_location', 'bzrlib.smart.branch',
 
523
    'SmartServerBranchRequestSetParentLocation')
257
524
request_handlers.register_lazy(
258
525
    'Branch.unlock', 'bzrlib.smart.branch', 'SmartServerBranchRequestUnlock')
259
526
request_handlers.register_lazy(
260
 
    'BzrDir.find_repository', 'bzrlib.smart.bzrdir', 'SmartServerRequestFindRepository')
261
 
request_handlers.register_lazy(
262
 
    'BzrDirFormat.initialize', 'bzrlib.smart.bzrdir', 'SmartServerRequestInitializeBzrDir')
263
 
request_handlers.register_lazy(
264
 
    'BzrDir.open_branch', 'bzrlib.smart.bzrdir', 'SmartServerRequestOpenBranch')
 
527
    'BzrDir.cloning_metadir', 'bzrlib.smart.bzrdir',
 
528
    'SmartServerBzrDirRequestCloningMetaDir')
 
529
request_handlers.register_lazy(
 
530
    'BzrDir.create_branch', 'bzrlib.smart.bzrdir',
 
531
    'SmartServerRequestCreateBranch')
 
532
request_handlers.register_lazy(
 
533
    'BzrDir.create_repository', 'bzrlib.smart.bzrdir',
 
534
    'SmartServerRequestCreateRepository')
 
535
request_handlers.register_lazy(
 
536
    'BzrDir.find_repository', 'bzrlib.smart.bzrdir',
 
537
    'SmartServerRequestFindRepositoryV1')
 
538
request_handlers.register_lazy(
 
539
    'BzrDir.find_repositoryV2', 'bzrlib.smart.bzrdir',
 
540
    'SmartServerRequestFindRepositoryV2')
 
541
request_handlers.register_lazy(
 
542
    'BzrDir.find_repositoryV3', 'bzrlib.smart.bzrdir',
 
543
    'SmartServerRequestFindRepositoryV3')
 
544
request_handlers.register_lazy(
 
545
    'BzrDir.get_config_file', 'bzrlib.smart.bzrdir',
 
546
    'SmartServerBzrDirRequestConfigFile')
 
547
request_handlers.register_lazy(
 
548
    'BzrDirFormat.initialize', 'bzrlib.smart.bzrdir',
 
549
    'SmartServerRequestInitializeBzrDir')
 
550
request_handlers.register_lazy(
 
551
    'BzrDirFormat.initialize_ex_1.16', 'bzrlib.smart.bzrdir',
 
552
    'SmartServerRequestBzrDirInitializeEx')
 
553
request_handlers.register_lazy(
 
554
    'BzrDir.open', 'bzrlib.smart.bzrdir', 'SmartServerRequestOpenBzrDir')
 
555
request_handlers.register_lazy(
 
556
    'BzrDir.open_2.1', 'bzrlib.smart.bzrdir', 'SmartServerRequestOpenBzrDir_2_1')
 
557
request_handlers.register_lazy(
 
558
    'BzrDir.open_branch', 'bzrlib.smart.bzrdir',
 
559
    'SmartServerRequestOpenBranch')
 
560
request_handlers.register_lazy(
 
561
    'BzrDir.open_branchV2', 'bzrlib.smart.bzrdir',
 
562
    'SmartServerRequestOpenBranchV2')
 
563
request_handlers.register_lazy(
 
564
    'BzrDir.open_branchV3', 'bzrlib.smart.bzrdir',
 
565
    'SmartServerRequestOpenBranchV3')
265
566
request_handlers.register_lazy(
266
567
    'delete', 'bzrlib.smart.vfs', 'DeleteRequest')
267
568
request_handlers.register_lazy(
288
589
    'readv', 'bzrlib.smart.vfs', 'ReadvRequest')
289
590
request_handlers.register_lazy(
290
591
    'rename', 'bzrlib.smart.vfs', 'RenameRequest')
 
592
request_handlers.register_lazy(
 
593
    'PackRepository.autopack', 'bzrlib.smart.packrepository',
 
594
    'SmartServerPackRepositoryAutopack')
291
595
request_handlers.register_lazy('Repository.gather_stats',
292
596
                               'bzrlib.smart.repository',
293
597
                               'SmartServerRepositoryGatherStats')
 
598
request_handlers.register_lazy('Repository.get_parent_map',
 
599
                               'bzrlib.smart.repository',
 
600
                               'SmartServerRepositoryGetParentMap')
294
601
request_handlers.register_lazy(
295
602
    'Repository.get_revision_graph', 'bzrlib.smart.repository', 'SmartServerRepositoryGetRevisionGraph')
296
603
request_handlers.register_lazy(
297
604
    'Repository.has_revision', 'bzrlib.smart.repository', 'SmartServerRequestHasRevision')
298
605
request_handlers.register_lazy(
 
606
    'Repository.insert_stream', 'bzrlib.smart.repository', 'SmartServerRepositoryInsertStream')
 
607
request_handlers.register_lazy(
 
608
    'Repository.insert_stream_1.19', 'bzrlib.smart.repository', 'SmartServerRepositoryInsertStream_1_19')
 
609
request_handlers.register_lazy(
 
610
    'Repository.insert_stream_locked', 'bzrlib.smart.repository', 'SmartServerRepositoryInsertStreamLocked')
 
611
request_handlers.register_lazy(
299
612
    'Repository.is_shared', 'bzrlib.smart.repository', 'SmartServerRepositoryIsShared')
300
613
request_handlers.register_lazy(
301
614
    'Repository.lock_write', 'bzrlib.smart.repository', 'SmartServerRepositoryLockWrite')
302
615
request_handlers.register_lazy(
 
616
    'Repository.set_make_working_trees', 'bzrlib.smart.repository',
 
617
    'SmartServerRepositorySetMakeWorkingTrees')
 
618
request_handlers.register_lazy(
303
619
    'Repository.unlock', 'bzrlib.smart.repository', 'SmartServerRepositoryUnlock')
304
620
request_handlers.register_lazy(
 
621
    'Repository.get_rev_id_for_revno', 'bzrlib.smart.repository',
 
622
    'SmartServerRepositoryGetRevIdForRevno')
 
623
request_handlers.register_lazy(
 
624
    'Repository.get_stream', 'bzrlib.smart.repository',
 
625
    'SmartServerRepositoryGetStream')
 
626
request_handlers.register_lazy(
 
627
    'Repository.get_stream_1.19', 'bzrlib.smart.repository',
 
628
    'SmartServerRepositoryGetStream_1_19')
 
629
request_handlers.register_lazy(
 
630
    'Repository.tarball', 'bzrlib.smart.repository',
 
631
    'SmartServerRepositoryTarball')
 
632
request_handlers.register_lazy(
305
633
    'rmdir', 'bzrlib.smart.vfs', 'RmdirRequest')
306
634
request_handlers.register_lazy(
307
635
    'stat', 'bzrlib.smart.vfs', 'StatRequest')
308
636
request_handlers.register_lazy(
309
637
    'Transport.is_readonly', 'bzrlib.smart.request', 'SmartServerIsReadonly')
310
 
request_handlers.register_lazy(
311
 
    'probe_dont_use', 'bzrlib.smart.request', 'ProbeDontUseRequest')