comparison upmana/mercurial/sshserver.py @ 121:496dbf12a6cb alpha

Traipse Alpha 'OpenRPG' {091030-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 (Cleaning up for Beta): Adds Bookmarks (Alpha) with cool Smiley Star and Plus Symbol images! Changes made to the map for increased portability. SnowDog has changes planned in Core, though. Added an initial push to the BCG. Not much to see, just shows off how it is re-writing Main code. Fix to remote admin commands Minor fix to texted based server, works in /System/ folder Some Core changes to gametree to correctly disply Pretty Print, thanks David! Fix to Splitter Nodes not being created. Added images to Plugin Control panel for Autostart feature Fix to massive amounts of images loading; from Core fix to gsclient so with_statement imports Added 'boot' command to remote admin Prep work in Pass tool for remote admin rankings and different passwords, ei, Server, Admin, Moderator, etc. Remote Admin Commands more organized, more prep work. Added Confirmation window for sent nodes. Minor changes to allow for portability to an OpenSUSE linux OS (hopefully without breaking) {091028} Made changes to gametree to start working with Element Tree, mostly from Core Minor changes to Map to start working with Element Tree, from Core Preliminary changes to map efficiency, from FlexiRPG Miniatures Layer pop up box allows users to turn off Mini labels, from FlexiRPG Changes to main.py to start working with Element Tree {091029} Changes made to server to start working with Element Tree. Changes made to Meta Server Lib. Prepping test work for a multi meta network page. Minor bug fixed with mini to gametree Zoom Mouse plugin added. {091030} Getting ready for Beta. Server needs debugging so Alpha remains bugged. Plugin UI code cleaned. Auto start works with a graphic, pop-up asks to enable or disable plugin. Update Manager now has a partially working Status Bar. Status Bar captures terminal text, so Merc out put is visible. Manifest.xml file, will be renamed, is now much cleaner. Debug Console has a clear button and a Report Bug button. Prep work for a Term2Win class in Debug Console. Known: Current Alpha fails in Windows.
author sirebral
date Fri, 30 Oct 2009 22:21:40 -0500
parents
children
comparison
equal deleted inserted replaced
120:d86e762a994f 121:496dbf12a6cb
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()