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
|
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Bazaar plugin to run the smart server on Launchpad.
Cribbed from bzrlib.builtins.cmd_serve from Bazaar 0.16.
"""
__metaclass__ = type
__all__ = ['cmd_launchpad_server']
import resource
import sys
from bzrlib.commands import Command, register_command
from bzrlib.option import Option
from bzrlib import lockdir, ui
from bzrlib.smart import medium, server
from bzrlib.transport import get_transport
class cmd_launchpad_server(Command):
"""Run a Bazaar server that maps Launchpad branch URLs to the internal
file-system format.
"""
aliases = ['lp-serve']
takes_options = [
Option('inet',
help='serve on stdin/out for use from inetd or sshd'),
Option('port',
help='listen for connections on nominated port of the form '
'[hostname:]portnumber. Passing 0 as the port number will'
' result in a dynamically allocated port. Default port is'
' 4155.',
type=str),
Option('upload-directory',
help='upload branches to this directory. Defaults to '
'config.codehosting.hosted_branches_root.',
type=unicode),
Option('mirror-directory',
help='serve branches from this directory. Defaults to '
'config.codehosting.mirrored_branches_root.'),
Option('branchfs-endpoint',
help='the url of the internal XML-RPC server. Defaults to '
'config.codehosting.branchfs_endpoint.',
type=unicode),
]
takes_args = ['user_id']
def get_smart_server(self, transport, port, inet):
"""Construct a smart server."""
if inet:
smart_server = medium.SmartServerPipeStreamMedium(
sys.stdin, sys.stdout, transport)
else:
host = medium.BZR_DEFAULT_INTERFACE
if port is None:
port = medium.BZR_DEFAULT_PORT
else:
if ':' in port:
host, port = port.split(':')
port = int(port)
smart_server = server.SmartTCPServer(
transport, host=host, port=port)
print 'listening on port: ', smart_server.port
sys.stdout.flush()
return smart_server
def run_server(self, smart_server):
"""Run the given smart server."""
# for the duration of this server, no UI output is permitted.
# note that this may cause problems with blackbox tests. This should
# be changed with care though, as we dont want to use bandwidth
# sending progress over stderr to smart server clients!
old_factory = ui.ui_factory
try:
ui.ui_factory = ui.SilentUIFactory()
smart_server.serve()
finally:
ui.ui_factory = old_factory
def run(self, user_id, port=None, upload_directory=None,
mirror_directory=None, branchfs_endpoint_url=None, inet=False):
from lp.codehosting.bzrutils import install_oops_handler
from lp.codehosting.vfs import get_lp_server, hooks
install_oops_handler(user_id)
four_gig = int(4e9)
resource.setrlimit(resource.RLIMIT_AS, (four_gig, four_gig))
seen_new_branch = hooks.SetProcTitleHook()
lp_server = get_lp_server(
int(user_id), branchfs_endpoint_url,
upload_directory, mirror_directory, seen_new_branch.seen)
lp_server.start_server()
old_lockdir_timeout = lockdir._DEFAULT_TIMEOUT_SECONDS
try:
lp_transport = get_transport(lp_server.get_url())
smart_server = self.get_smart_server(lp_transport, port, inet)
lockdir._DEFAULT_TIMEOUT_SECONDS = 0
self.run_server(smart_server)
finally:
lockdir._DEFAULT_TIMEOUT_SECONDS = old_lockdir_timeout
lp_server.stop_server()
register_command(cmd_launchpad_server)
|