comparison upmana/mercurial/sshserver.py @ 28:ff154cf3350c ornery-orc

Traipse 'OpenRPG' {100203-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 (Stable) New Features: New Bookmarks Feature New 'boot' command to remote admin New confirmation window for sent nodes Miniatures Layer pop up box allows users to turn off Mini labels, from FlexiRPG New Zoom Mouse plugin added New Images added to Plugin UI Switching to Element Tree New Map efficiency, from FlexiRPG New Status Bar to Update Manager New TrueDebug Class in orpg_log (See documentation for usage) New Portable Mercurial New Tip of the Day, from Core and community New Reference Syntax added for custom PC sheets New Child Reference for gametree New Parent Reference for gametree New Gametree Recursion method, mapping, context sensitivity, and effeciency.. New Features node with bonus nodes and Node Referencing help added New Dieroller structure from Core New DieRoller portability for odd Dice New 7th Sea die roller; ie [7k3] = [7d10.takeHighest(3).open(10)] New 'Mythos' System die roller added New vs. die roller method for WoD; ie [3v3] = [3d10.vs(3)]. Included for Mythos roller also New Warhammer FRPG Die Roller (Special thanks to Puu-san for the support) New EZ_Tree Reference system. Push a button, Traipse the tree, get a reference (Beta!) New Grids act more like Spreadsheets in Use mode, with Auto Calc Fixes: Fix to allow for portability to an OpenSUSE linux OS Fix to mplay_client for Fedora and OpenSUSE Fix to Text based Server Fix to Remote Admin Commands Fix to Pretty Print, from Core Fix to Splitter Nodes not being created Fix to massive amounts of images loading, from Core Fix to Map from gametree not showing to all clients Fix to gametree about menus Fix to Password Manager check on startup Fix to PC Sheets from tool nodes. They now use the tabber_panel Fix to Whiteboard ID to prevent random line or text deleting. Fixes to Server, Remote Server, and Server GUI Fix to Update Manager; cleaner clode for saved repositories Fixes made to Settings Panel and now reactive settings when Ok is pressed Fixes to Alternity roller's attack roll. Uses a simple Tuple instead of a Splice Fix to Use panel of Forms and Tabbers. Now longer enters design mode Fix made Image Fetching. New fetching image and new failed image Fix to whiteboard ID's to prevent non updated clients from ruining the fix. default_manifest.xml renamed to default_upmana.xml
author sirebral
date Wed, 03 Feb 2010 22:16:49 -0600
parents
children
comparison
equal deleted inserted replaced
27:51428d30c59e 28:ff154cf3350c
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()