Mercurial > traipse_dev

comparison upmana/mercurial/sshserver.py @ 135:dcf4fbe09b70 beta

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Traipse Beta 'OpenRPG' {091010-00} Traipse is a distribution of OpenRPG that is designed to be easy to setup and go. Traipse also makes it easy for developers to work on code without fear of sacrifice. 'Ornery-Orc' continues the trend of 'Grumpy' and adds fixes to the code. 'Ornery-Orc's main goal is to offer more advanced features and enhance the productivity of the user. Update Summary (Beta) Added Bookmarks Fix to Remote Admin Commands Minor fix to text based Server Fix to Pretty Print, from Core Fix to Splitter Nodes not being created Fix to massive amounts of images loading, from Core Added 'boot' command to remote admin Added confirmation window for sent nodes Minor changes to allow for portability to an OpenSUSE linux OS Miniatures Layer pop up box allows users to turn off Mini labels, from FlexiRPG Zoom Mouse plugin added Images added to Plugin UI Switching to Element Tree Map efficiency, from FlexiRPG Added Status Bar to Update Manager default_manifest.xml renamed to default_upmana.xml Cleaner clode for saved repositories New TrueDebug Class in orpg_log (See documentation for usage) Mercurial's hgweb folder is ported to upmana **Pretty important update that can help remove thousands of dead children from your gametree. **Children, <forms />, <group_atts />, <horizontal />, <cols />, <rows />, <height />, etc... are all tags now. Check your gametree and look for dead children!! **New Gamtree Recusion method, mapping, and context sensitivity. !!Alpha - Watch out for infinite loops!!
author sirebral
date Tue, 10 Nov 2009 14:11:28 -0600
parents
children
comparison
equal deleted inserted replaced
101:394ebb3b6a0f 135:dcf4fbe09b70
1 # sshserver.py - ssh protocol server support for mercurial
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
5 #
6 # This software may be used and distributed according to the terms of the
7 # GNU General Public License version 2, incorporated herein by reference.
8
9 from i18n import _
10 from node import bin, hex
11 import streamclone, util, hook
12 import os, sys, tempfile, urllib
13
14 class sshserver(object):
15 def __init__(self, ui, repo):
16 self.ui = ui
17 self.repo = repo
18 self.lock = None
19 self.fin = sys.stdin
20 self.fout = sys.stdout
21
22 hook.redirect(True)
23 sys.stdout = sys.stderr
24
25 # Prevent insertion/deletion of CRs
26 util.set_binary(self.fin)
27 util.set_binary(self.fout)
28
29 def getarg(self):
30 argline = self.fin.readline()[:-1]
31 arg, l = argline.split()
32 val = self.fin.read(int(l))
33 return arg, val
34
35 def respond(self, v):
36 self.fout.write("%d\n" % len(v))
37 self.fout.write(v)
38 self.fout.flush()
39
40 def serve_forever(self):
41 try:
42 while self.serve_one(): pass
43 finally:
44 if self.lock is not None:
45 self.lock.release()
46 sys.exit(0)
47
48 def serve_one(self):
49 cmd = self.fin.readline()[:-1]
50 if cmd:
51 impl = getattr(self, 'do_' + cmd, None)
52 if impl: impl()
53 else: self.respond("")
54 return cmd != ''
55
56 def do_lookup(self):
57 arg, key = self.getarg()
58 assert arg == 'key'
59 try:
60 r = hex(self.repo.lookup(key))
61 success = 1
62 except Exception,inst:
63 r = str(inst)
64 success = 0
65 self.respond("%s %s\n" % (success, r))
66
67 def do_branchmap(self):
68 branchmap = self.repo.branchmap()
69 heads = []
70 for branch, nodes in branchmap.iteritems():
71 branchname = urllib.quote(branch)
72 branchnodes = [hex(node) for node in nodes]
73 heads.append('%s %s' % (branchname, ' '.join(branchnodes)))
74 self.respond('\n'.join(heads))
75
76 def do_heads(self):
77 h = self.repo.heads()
78 self.respond(" ".join(map(hex, h)) + "\n")
79
80 def do_hello(self):
81 '''the hello command returns a set of lines describing various
82 interesting things about the server, in an RFC822-like format.
83 Currently the only one defined is "capabilities", which
84 consists of a line in the form:
85
86 capabilities: space separated list of tokens
87 '''
88
89 caps = ['unbundle', 'lookup', 'changegroupsubset', 'branchmap']
90 if self.ui.configbool('server', 'uncompressed'):
91 caps.append('stream=%d' % self.repo.changelog.version)
92 self.respond("capabilities: %s\n" % (' '.join(caps),))
93
94 def do_lock(self):
95 '''DEPRECATED - allowing remote client to lock repo is not safe'''
96
97 self.lock = self.repo.lock()
98 self.respond("")
99
100 def do_unlock(self):
101 '''DEPRECATED'''
102
103 if self.lock:
104 self.lock.release()
105 self.lock = None
106 self.respond("")
107
108 def do_branches(self):
109 arg, nodes = self.getarg()
110 nodes = map(bin, nodes.split(" "))
111 r = []
112 for b in self.repo.branches(nodes):
113 r.append(" ".join(map(hex, b)) + "\n")
114 self.respond("".join(r))
115
116 def do_between(self):
117 arg, pairs = self.getarg()
118 pairs = [map(bin, p.split("-")) for p in pairs.split(" ")]
119 r = []
120 for b in self.repo.between(pairs):
121 r.append(" ".join(map(hex, b)) + "\n")
122 self.respond("".join(r))
123
124 def do_changegroup(self):
125 nodes = []
126 arg, roots = self.getarg()
127 nodes = map(bin, roots.split(" "))
128
129 cg = self.repo.changegroup(nodes, 'serve')
130 while True:
131 d = cg.read(4096)
132 if not d:
133 break
134 self.fout.write(d)
135
136 self.fout.flush()
137
138 def do_changegroupsubset(self):
139 argmap = dict([self.getarg(), self.getarg()])
140 bases = [bin(n) for n in argmap['bases'].split(' ')]
141 heads = [bin(n) for n in argmap['heads'].split(' ')]
142
143 cg = self.repo.changegroupsubset(bases, heads, 'serve')
144 while True:
145 d = cg.read(4096)
146 if not d:
147 break
148 self.fout.write(d)
149
150 self.fout.flush()
151
152 def do_addchangegroup(self):
153 '''DEPRECATED'''
154
155 if not self.lock:
156 self.respond("not locked")
157 return
158
159 self.respond("")
160 r = self.repo.addchangegroup(self.fin, 'serve', self.client_url())
161 self.respond(str(r))
162
163 def client_url(self):
164 client = os.environ.get('SSH_CLIENT', '').split(' ', 1)[0]
165 return 'remote:ssh:' + client
166
167 def do_unbundle(self):
168 their_heads = self.getarg()[1].split()
169
170 def check_heads():
171 heads = map(hex, self.repo.heads())
172 return their_heads == [hex('force')] or their_heads == heads
173
174 # fail early if possible
175 if not check_heads():
176 self.respond(_('unsynced changes'))
177 return
178
179 self.respond('')
180
181 # write bundle data to temporary file because it can be big
182 tempname = fp = None
183 try:
184 fd, tempname = tempfile.mkstemp(prefix='hg-unbundle-')
185 fp = os.fdopen(fd, 'wb+')
186
187 count = int(self.fin.readline())
188 while count:
189 fp.write(self.fin.read(count))
190 count = int(self.fin.readline())
191
192 was_locked = self.lock is not None
193 if not was_locked:
194 self.lock = self.repo.lock()
195 try:
196 if not check_heads():
197 # someone else committed/pushed/unbundled while we
198 # were transferring data
199 self.respond(_('unsynced changes'))
200 return
201 self.respond('')
202
203 # push can proceed
204
205 fp.seek(0)
206 r = self.repo.addchangegroup(fp, 'serve', self.client_url())
207 self.respond(str(r))
208 finally:
209 if not was_locked:
210 self.lock.release()
211 self.lock = None
212 finally:
213 if fp is not None:
214 fp.close()
215 if tempname is not None:
216 os.unlink(tempname)
217
218 def do_stream_out(self):
219 try:
220 for chunk in streamclone.stream_out(self.repo):
221 self.fout.write(chunk)
222 self.fout.flush()
223 except streamclone.StreamException, inst:
224 self.fout.write(str(inst))
225 self.fout.flush()