mercurial-crew-with-dirclash: mercurial/sshserver.py annotate

annotate mercurial/sshserver.py @ 2802:fdc232d8a193

Move repo.verify

author	Matt Mackall <mpm@selenic.com>
date	Mon, 07 Aug 2006 16:27:09 -0500
parents	109a22f5434a
children	345bac2bc4ec

rev	line source
2399 cc90dcbdf053 fix comment. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2397 diff changeset	1 # sshserver.py - ssh protocol server support for mercurial
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	2 #
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	3 # Copyright 2005 Matt Mackall <mpm@selenic.com>
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	4 #
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	5 # This software may be used and distributed according to the terms
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	6 # of the GNU General Public License, incorporated herein by reference.
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	7
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	8 from demandload import demandload
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	9 from i18n import gettext as _
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	10 from node import *
2612 ffb895f16925 add support for streaming clone. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2439 diff changeset	11 demandload(globals(), "os streamclone sys tempfile util")
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	12
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	13 class sshserver(object):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	14 def __init__(self, ui, repo):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	15 self.ui = ui
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	16 self.repo = repo
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	17 self.lock = None
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	18 self.fin = sys.stdin
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	19 self.fout = sys.stdout
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	20
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	21 sys.stdout = sys.stderr
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	22
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	23 # Prevent insertion/deletion of CRs
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	24 util.set_binary(self.fin)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	25 util.set_binary(self.fout)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	26
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	27 def getarg(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	28 argline = self.fin.readline()[:-1]
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	29 arg, l = argline.split()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	30 val = self.fin.read(int(l))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	31 return arg, val
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	32
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	33 def respond(self, v):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	34 self.fout.write("%d\n" % len(v))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	35 self.fout.write(v)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	36 self.fout.flush()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	37
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	38 def serve_forever(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	39 while self.serve_one(): pass
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	40 sys.exit(0)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	41
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	42 def serve_one(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	43 cmd = self.fin.readline()[:-1]
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	44 if cmd:
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	45 impl = getattr(self, 'do_' + cmd, None)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	46 if impl: impl()
2397 e9d402506514 merge change to ssh protocol. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2396 diff changeset	47 else: self.respond("")
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	48 return cmd != ''
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	49
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	50 def do_heads(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	51 h = self.repo.heads()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	52 self.respond(" ".join(map(hex, h)) + "\n")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	53
2419 b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	54 def do_hello(self):
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	55 '''the hello command returns a set of lines describing various
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	56 interesting things about the server, in an RFC822-like format.
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	57 Currently the only one defined is "capabilities", which
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	58 consists of a line in the form:
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	59
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	60 capabilities: space separated list of tokens
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	61 '''
b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	62
2621 5a5852a417b1 clone: disable stream support on server side by default. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2612 diff changeset	63 caps = ['unbundle']
2622 064aef9162cc rename stream hgrc option to compressed. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2621 diff changeset	64 if self.ui.configbool('server', 'uncompressed'):
2621 5a5852a417b1 clone: disable stream support on server side by default. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2612 diff changeset	65 caps.append('stream=%d' % self.repo.revlogversion)
5a5852a417b1 clone: disable stream support on server side by default. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2612 diff changeset	66 self.respond("capabilities: %s\n" % (' '.join(caps),))
2419 b17eebc911ae Initial implementation of hello command for ssh Matt Mackall <mpm@selenic.com> parents: 2399 diff changeset	67
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	68 def do_lock(self):
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	69 '''DEPRECATED - allowing remote client to lock repo is not safe'''
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	70
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	71 self.lock = self.repo.lock()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	72 self.respond("")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	73
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	74 def do_unlock(self):
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	75 '''DEPRECATED'''
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	76
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	77 if self.lock:
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	78 self.lock.release()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	79 self.lock = None
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	80 self.respond("")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	81
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	82 def do_branches(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	83 arg, nodes = self.getarg()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	84 nodes = map(bin, nodes.split(" "))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	85 r = []
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	86 for b in self.repo.branches(nodes):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	87 r.append(" ".join(map(hex, b)) + "\n")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	88 self.respond("".join(r))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	89
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	90 def do_between(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	91 arg, pairs = self.getarg()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	92 pairs = [map(bin, p.split("-")) for p in pairs.split(" ")]
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	93 r = []
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	94 for b in self.repo.between(pairs):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	95 r.append(" ".join(map(hex, b)) + "\n")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	96 self.respond("".join(r))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	97
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	98 def do_changegroup(self):
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	99 nodes = []
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	100 arg, roots = self.getarg()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	101 nodes = map(bin, roots.split(" "))
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	102
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	103 cg = self.repo.changegroup(nodes, 'serve')
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	104 while True:
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	105 d = cg.read(4096)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	106 if not d:
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	107 break
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	108 self.fout.write(d)
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	109
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	110 self.fout.flush()
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	111
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	112 def do_addchangegroup(self):
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	113 '''DEPRECATED'''
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	114
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	115 if not self.lock:
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	116 self.respond("not locked")
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	117 return
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	118
8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	119 self.respond("")
2673 109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	120 r = self.repo.addchangegroup(self.fin, 'serve', self.client_url())
2396 8d44649df03b refactor ssh server. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	121 self.respond(str(r))
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	122
2673 109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	123 def client_url(self):
109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	124 client = os.environ.get('SSH_CLIENT', '').split(' ', 1)[0]
109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	125 return 'remote:ssh:' + client
109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	126
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	127 def do_unbundle(self):
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	128 their_heads = self.getarg()[1].split()
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	129
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	130 def check_heads():
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	131 heads = map(hex, self.repo.heads())
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	132 return their_heads == [hex('force')] or their_heads == heads
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	133
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	134 # fail early if possible
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	135 if not check_heads():
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	136 self.respond(_('unsynced changes'))
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	137 return
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	138
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	139 self.respond('')
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	140
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	141 # write bundle data to temporary file because it can be big
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	142
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	143 try:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	144 fd, tempname = tempfile.mkstemp(prefix='hg-unbundle-')
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	145 fp = os.fdopen(fd, 'wb+')
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	146
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	147 count = int(self.fin.readline())
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	148 while count:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	149 fp.write(self.fin.read(count))
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	150 count = int(self.fin.readline())
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	151
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	152 was_locked = self.lock is not None
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	153 if not was_locked:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	154 self.lock = self.repo.lock()
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	155 try:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	156 if not check_heads():
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	157 # someone else committed/pushed/unbundled while we
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	158 # were transferring data
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	159 self.respond(_('unsynced changes'))
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	160 return
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	161 self.respond('')
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	162
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	163 # push can proceed
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	164
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	165 fp.seek(0)
2673 109a22f5434a hooks: add url to changegroup, incoming, prechangegroup, pretxnchangegroup hooks Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2622 diff changeset	166 r = self.repo.addchangegroup(fp, 'serve', self.client_url())
2439 e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	167 self.respond(str(r))
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	168 finally:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	169 if not was_locked:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	170 self.lock.release()
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	171 self.lock = None
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	172 finally:
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	173 fp.close()
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	174 os.unlink(tempname)
e8c4f3d3df8c extend network protocol to stop clients from locking servers Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2419 diff changeset	175
2612 ffb895f16925 add support for streaming clone. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2439 diff changeset	176 def do_stream_out(self):
ffb895f16925 add support for streaming clone. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2439 diff changeset	177 streamclone.stream_out(self.repo, self.fout)

Mercurial > hg > mercurial-crew-with-dirclash

annotate mercurial/sshserver.py @ 2802:fdc232d8a193