~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/smart/request.py

  • Committer: Martin Pool
  • Date: 2008-05-08 04:12:06 UTC
  • mto: This revision was merged to the branch mainline in revision 3415.
  • Revision ID: mbp@sourcefrog.net-20080508041206-tkrr8ucmcyrlzkum
Some review cleanups for assertion removal

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007 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
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.
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""Basic server-side logic for dealing with requests."""
32
18
 
33
19
 
34
20
import tempfile
35
 
import thread
36
 
import threading
37
21
 
38
22
from bzrlib import (
39
23
    bzrdir,
40
 
    debug,
41
24
    errors,
42
 
    osutils,
43
25
    registry,
44
26
    revision,
45
 
    trace,
46
27
    urlutils,
47
28
    )
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()
 
29
from bzrlib.bundle.serializer import write_bundle
 
30
from bzrlib.transport import get_transport
 
31
from bzrlib.transport.chroot import ChrootServer
79
32
 
80
33
 
81
34
class SmartServerRequest(object):
82
35
    """Base class for request handlers.
83
 
 
 
36
    
84
37
    To define a new request, subclass this class and override the `do` method
85
38
    (and if appropriate, `do_body` as well).  Request implementors should take
86
39
    care to call `translate_client_path` and `transport_from_client_path` as
87
40
    appropriate when dealing with paths received from the client.
88
41
    """
89
 
    # XXX: rename this class to BaseSmartServerRequestHandler ?  A request
90
 
    # *handler* is a different concept to the request.
91
42
 
92
 
    def __init__(self, backing_transport, root_client_path='/', jail_root=None):
 
43
    def __init__(self, backing_transport, root_client_path='/'):
93
44
        """Constructor.
94
45
 
95
46
        :param backing_transport: the base transport to be used when performing
99
50
            from the client.  Clients will not be able to refer to paths above
100
51
            this root.  If root_client_path is None, then no translation will
101
52
            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
53
        """
105
54
        self._backing_transport = backing_transport
106
 
        if jail_root is None:
107
 
            jail_root = backing_transport
108
 
        self._jail_root = jail_root
109
55
        if root_client_path is not None:
110
56
            if not root_client_path.startswith('/'):
111
57
                root_client_path = '/' + root_client_path
112
58
            if not root_client_path.endswith('/'):
113
59
                root_client_path += '/'
114
60
        self._root_client_path = root_client_path
115
 
        self._body_chunks = []
116
61
 
117
62
    def _check_enabled(self):
118
63
        """Raises DisabledMethod if this method is disabled."""
120
65
 
121
66
    def do(self, *args):
122
67
        """Mandatory extension point for SmartServerRequest subclasses.
123
 
 
 
68
        
124
69
        Subclasses must implement this.
125
 
 
 
70
        
126
71
        This should return a SmartServerResponse if this command expects to
127
72
        receive no body.
128
73
        """
143
88
        """Called if the client sends a body with the request.
144
89
 
145
90
        The do() method is still called, and must have returned None.
146
 
 
 
91
        
147
92
        Must return a SmartServerResponse.
148
93
        """
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
 
 
 
94
        # TODO: if a client erroneously sends a request that shouldn't have a
 
95
        # body, what to do?  Probably SmartServerRequestHandler should catch
 
96
        # this NotImplementedError and translate it into a 'bad request' error
 
97
        # to send to the client.
 
98
        raise NotImplementedError(self.do_body)
 
99
    
171
100
    def translate_client_path(self, client_path):
172
101
        """Translate a path received from a network client into a local
173
102
        relpath.
184
113
            return client_path
185
114
        if not client_path.startswith('/'):
186
115
            client_path = '/' + client_path
187
 
        if client_path + '/' == self._root_client_path:
188
 
            return '.'
189
116
        if client_path.startswith(self._root_client_path):
190
117
            path = client_path[len(self._root_client_path):]
191
118
            relpath = urlutils.joinpath('/', path)
192
119
            if not relpath.startswith('/'):
193
120
                raise ValueError(relpath)
194
 
            return urlutils.escape('.' + relpath)
 
121
            return '.' + relpath
195
122
        else:
196
123
            raise errors.PathNotChild(client_path, self._root_client_path)
197
124
 
208
135
 
209
136
class SmartServerResponse(object):
210
137
    """A response to a client request.
211
 
 
 
138
    
212
139
    This base class should not be used. Instead use
213
140
    SuccessfulSmartServerResponse and FailedSmartServerResponse as appropriate.
214
141
    """
236
163
                other.body_stream is self.body_stream)
237
164
 
238
165
    def __repr__(self):
239
 
        return "<%s args=%r body=%r>" % (self.__class__.__name__,
 
166
        status = {True: 'OK', False: 'ERR'}[self.is_successful()]
 
167
        return "<SmartServerResponse status=%s args=%r body=%r>" % (status,
240
168
            self.args, self.body)
241
169
 
242
170
 
258
186
 
259
187
class SmartServerRequestHandler(object):
260
188
    """Protocol logic for smart server.
261
 
 
 
189
    
262
190
    This doesn't handle serialization at all, it just processes requests and
263
191
    creates responses.
264
192
    """
273
201
    # TODO: Better way of representing the body for commands that take it,
274
202
    # and allow it to be streamed into the server.
275
203
 
276
 
    def __init__(self, backing_transport, commands, root_client_path,
277
 
        jail_root=None):
 
204
    def __init__(self, backing_transport, commands, root_client_path):
278
205
        """Constructor.
279
206
 
280
207
        :param backing_transport: a Transport to handle requests for.
284
211
        self._backing_transport = backing_transport
285
212
        self._root_client_path = root_client_path
286
213
        self._commands = commands
287
 
        if jail_root is None:
288
 
            jail_root = backing_transport
289
 
        self._jail_root = jail_root
 
214
        self._body_bytes = ''
290
215
        self.response = None
291
216
        self.finished_reading = False
292
217
        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))
314
218
 
315
219
    def accept_body(self, bytes):
316
220
        """Accept body data."""
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
 
 
 
221
 
 
222
        # TODO: This should be overriden for each command that desired body data
 
223
        # to handle the right format of that data, i.e. plain bytes, a bundle,
 
224
        # etc.  The deserialisation into that format should be done in the
 
225
        # Protocol object.
 
226
 
 
227
        # default fallback is to accumulate bytes.
 
228
        self._body_bytes += bytes
 
229
        
325
230
    def end_of_body(self):
326
231
        """No more body data will be received."""
327
 
        self._run_handler_code(self._command.do_end, (), {})
 
232
        self._run_handler_code(self._command.do_body, (self._body_bytes,), {})
328
233
        # cannot read after this.
329
234
        self.finished_reading = True
330
 
        if 'hpss' in debug.debug_flags:
331
 
            self._trace('end of body', '', include_time=True)
 
235
 
 
236
    def dispatch_command(self, cmd, args):
 
237
        """Deprecated compatibility method.""" # XXX XXX
 
238
        try:
 
239
            command = self._commands.get(cmd)
 
240
        except LookupError:
 
241
            raise errors.SmartProtocolError("bad request %r" % (cmd,))
 
242
        self._command = command(self._backing_transport, self._root_client_path)
 
243
        self._run_handler_code(self._command.execute, args, {})
332
244
 
333
245
    def _run_handler_code(self, callable, args, kwargs):
334
246
        """Run some handler specific code 'callable'.
350
262
        # XXX: most of this error conversion is VFS-related, and thus ought to
351
263
        # be in SmartServerVFSRequestHandler somewhere.
352
264
        try:
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))
 
265
            return callable(*args, **kwargs)
 
266
        except errors.NoSuchFile, e:
 
267
            return FailedSmartServerResponse(('NoSuchFile', e.path))
 
268
        except errors.FileExists, e:
 
269
            return FailedSmartServerResponse(('FileExists', e.path))
 
270
        except errors.DirectoryNotEmpty, e:
 
271
            return FailedSmartServerResponse(('DirectoryNotEmpty', e.path))
 
272
        except errors.ShortReadvError, e:
 
273
            return FailedSmartServerResponse(('ShortReadvError',
 
274
                e.path, str(e.offset), str(e.length), str(e.actual)))
 
275
        except UnicodeError, e:
 
276
            # If it is a DecodeError, than most likely we are starting
 
277
            # with a plain string
 
278
            str_or_unicode = e.object
 
279
            if isinstance(str_or_unicode, unicode):
 
280
                # XXX: UTF-8 might have \x01 (our seperator byte) in it.  We
 
281
                # should escape it somehow.
 
282
                val = 'u:' + str_or_unicode.encode('utf-8')
 
283
            else:
 
284
                val = 's:' + str_or_unicode.encode('base64')
 
285
            # This handles UnicodeEncodeError or UnicodeDecodeError
 
286
            return FailedSmartServerResponse((e.__class__.__name__,
 
287
                    e.encoding, val, str(e.start), str(e.end), e.reason))
 
288
        except errors.TransportNotPossible, e:
 
289
            if e.msg == "readonly transport":
 
290
                return FailedSmartServerResponse(('ReadOnlyError', ))
 
291
            else:
 
292
                raise
452
293
 
453
294
 
454
295
class HelloRequest(SmartServerRequest):
470
311
        repo = control.open_repository()
471
312
        tmpf = tempfile.TemporaryFile()
472
313
        base_revision = revision.NULL_REVISION
473
 
        serializer.write_bundle(repo, revision_id, base_revision, tmpf)
 
314
        write_bundle(repo, revision_id, base_revision, tmpf)
474
315
        tmpf.seek(0)
475
316
        return SuccessfulSmartServerResponse((), tmpf.read())
476
317
 
490
331
request_handlers.register_lazy(
491
332
    'append', 'bzrlib.smart.vfs', 'AppendRequest')
492
333
request_handlers.register_lazy(
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')
 
334
    'Branch.get_config_file', 'bzrlib.smart.branch', 'SmartServerBranchGetConfigFile')
505
335
request_handlers.register_lazy(
506
336
    'Branch.last_revision_info', 'bzrlib.smart.branch', 'SmartServerBranchRequestLastRevisionInfo')
507
337
request_handlers.register_lazy(
508
338
    'Branch.lock_write', 'bzrlib.smart.branch', 'SmartServerBranchRequestLockWrite')
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')
 
339
request_handlers.register_lazy(
 
340
    'Branch.revision_history', 'bzrlib.smart.branch', 'SmartServerRequestRevisionHistory')
 
341
request_handlers.register_lazy(
 
342
    'Branch.set_last_revision', 'bzrlib.smart.branch', 'SmartServerBranchRequestSetLastRevision')
515
343
request_handlers.register_lazy(
516
344
    'Branch.set_last_revision_info', 'bzrlib.smart.branch',
517
345
    'SmartServerBranchRequestSetLastRevisionInfo')
518
346
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')
524
 
request_handlers.register_lazy(
525
347
    'Branch.unlock', 'bzrlib.smart.branch', 'SmartServerBranchRequestUnlock')
526
348
request_handlers.register_lazy(
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')
 
349
    'BzrDir.find_repository', 'bzrlib.smart.bzrdir', 'SmartServerRequestFindRepositoryV1')
 
350
request_handlers.register_lazy(
 
351
    'BzrDir.find_repositoryV2', 'bzrlib.smart.bzrdir', 'SmartServerRequestFindRepositoryV2')
 
352
request_handlers.register_lazy(
 
353
    'BzrDirFormat.initialize', 'bzrlib.smart.bzrdir', 'SmartServerRequestInitializeBzrDir')
 
354
request_handlers.register_lazy(
 
355
    'BzrDir.open_branch', 'bzrlib.smart.bzrdir', 'SmartServerRequestOpenBranch')
566
356
request_handlers.register_lazy(
567
357
    'delete', 'bzrlib.smart.vfs', 'DeleteRequest')
568
358
request_handlers.register_lazy(
589
379
    'readv', 'bzrlib.smart.vfs', 'ReadvRequest')
590
380
request_handlers.register_lazy(
591
381
    'rename', 'bzrlib.smart.vfs', 'RenameRequest')
592
 
request_handlers.register_lazy(
593
 
    'PackRepository.autopack', 'bzrlib.smart.packrepository',
594
 
    'SmartServerPackRepositoryAutopack')
595
382
request_handlers.register_lazy('Repository.gather_stats',
596
383
                               'bzrlib.smart.repository',
597
384
                               'SmartServerRepositoryGatherStats')
599
386
                               'bzrlib.smart.repository',
600
387
                               'SmartServerRepositoryGetParentMap')
601
388
request_handlers.register_lazy(
 
389
    'Repository.stream_knit_data_for_revisions',
 
390
    'bzrlib.smart.repository',
 
391
    'SmartServerRepositoryStreamKnitDataForRevisions')
 
392
request_handlers.register_lazy(
 
393
    'Repository.stream_revisions_chunked',
 
394
    'bzrlib.smart.repository',
 
395
    'SmartServerRepositoryStreamRevisionsChunked')
 
396
request_handlers.register_lazy(
602
397
    'Repository.get_revision_graph', 'bzrlib.smart.repository', 'SmartServerRepositoryGetRevisionGraph')
603
398
request_handlers.register_lazy(
604
399
    'Repository.has_revision', 'bzrlib.smart.repository', 'SmartServerRequestHasRevision')
605
400
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(
612
401
    'Repository.is_shared', 'bzrlib.smart.repository', 'SmartServerRepositoryIsShared')
613
402
request_handlers.register_lazy(
614
403
    'Repository.lock_write', 'bzrlib.smart.repository', 'SmartServerRepositoryLockWrite')
615
404
request_handlers.register_lazy(
616
 
    'Repository.set_make_working_trees', 'bzrlib.smart.repository',
617
 
    'SmartServerRepositorySetMakeWorkingTrees')
618
 
request_handlers.register_lazy(
619
405
    'Repository.unlock', 'bzrlib.smart.repository', 'SmartServerRepositoryUnlock')
620
406
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
407
    'Repository.tarball', 'bzrlib.smart.repository',
631
408
    'SmartServerRepositoryTarball')
632
409
request_handlers.register_lazy(
635
412
    'stat', 'bzrlib.smart.vfs', 'StatRequest')
636
413
request_handlers.register_lazy(
637
414
    'Transport.is_readonly', 'bzrlib.smart.request', 'SmartServerIsReadonly')
 
415
request_handlers.register_lazy(
 
416
    'BzrDir.open', 'bzrlib.smart.bzrdir', 'SmartServerRequestOpenBzrDir')