| durin42 |
| hg-reviewers |
The default is version 1 only. So we don't need to pass this argument
when declaring commands.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| else: | else: | ||||
| raise error.ProgrammingError('unhandled transport version: %d' % | raise error.ProgrammingError('unhandled transport version: %d' % | ||||
| transportversion) | transportversion) | ||||
| return func | return func | ||||
| return register | return register | ||||
| # TODO define a more appropriate permissions type to use for this. | # TODO define a more appropriate permissions type to use for this. | ||||
| @wireprotocommand('batch', 'cmds *', permission='pull', | @wireprotocommand('batch', 'cmds *', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def batch(repo, proto, cmds, others): | def batch(repo, proto, cmds, others): | ||||
| unescapearg = wireprototypes.unescapebatcharg | unescapearg = wireprototypes.unescapebatcharg | ||||
| repo = repo.filtered("served") | repo = repo.filtered("served") | ||||
| res = [] | res = [] | ||||
| for pair in cmds.split(';'): | for pair in cmds.split(';'): | ||||
| op, args = pair.split(' ', 1) | op, args = pair.split(' ', 1) | ||||
| vals = {} | vals = {} | ||||
| for a in args.split(','): | for a in args.split(','): | ||||
| # raw bytes (for backwards compatibility). | # raw bytes (for backwards compatibility). | ||||
| assert isinstance(result, (wireprototypes.bytesresponse, bytes)) | assert isinstance(result, (wireprototypes.bytesresponse, bytes)) | ||||
| if isinstance(result, wireprototypes.bytesresponse): | if isinstance(result, wireprototypes.bytesresponse): | ||||
| result = result.data | result = result.data | ||||
| res.append(wireprototypes.escapebatcharg(result)) | res.append(wireprototypes.escapebatcharg(result)) | ||||
| return wireprototypes.bytesresponse(';'.join(res)) | return wireprototypes.bytesresponse(';'.join(res)) | ||||
| @wireprotocommand('between', 'pairs', transportpolicy=POLICY_V1_ONLY, | @wireprotocommand('between', 'pairs', permission='pull') | ||||
| permission='pull') | |||||
| def between(repo, proto, pairs): | def between(repo, proto, pairs): | ||||
| pairs = [wireprototypes.decodelist(p, '-') for p in pairs.split(" ")] | pairs = [wireprototypes.decodelist(p, '-') for p in pairs.split(" ")] | ||||
| r = [] | r = [] | ||||
| for b in repo.between(pairs): | for b in repo.between(pairs): | ||||
| r.append(wireprototypes.encodelist(b) + "\n") | r.append(wireprototypes.encodelist(b) + "\n") | ||||
| return wireprototypes.bytesresponse(''.join(r)) | return wireprototypes.bytesresponse(''.join(r)) | ||||
| @wireprotocommand('branchmap', permission='pull', | @wireprotocommand('branchmap', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def branchmap(repo, proto): | def branchmap(repo, proto): | ||||
| branchmap = repo.branchmap() | branchmap = repo.branchmap() | ||||
| heads = [] | heads = [] | ||||
| for branch, nodes in branchmap.iteritems(): | for branch, nodes in branchmap.iteritems(): | ||||
| branchname = urlreq.quote(encoding.fromlocal(branch)) | branchname = urlreq.quote(encoding.fromlocal(branch)) | ||||
| branchnodes = wireprototypes.encodelist(nodes) | branchnodes = wireprototypes.encodelist(nodes) | ||||
| heads.append('%s %s' % (branchname, branchnodes)) | heads.append('%s %s' % (branchname, branchnodes)) | ||||
| return wireprototypes.bytesresponse('\n'.join(heads)) | return wireprototypes.bytesresponse('\n'.join(heads)) | ||||
| @wireprotocommand('branches', 'nodes', transportpolicy=POLICY_V1_ONLY, | @wireprotocommand('branches', 'nodes', permission='pull') | ||||
| permission='pull') | |||||
| def branches(repo, proto, nodes): | def branches(repo, proto, nodes): | ||||
| nodes = wireprototypes.decodelist(nodes) | nodes = wireprototypes.decodelist(nodes) | ||||
| r = [] | r = [] | ||||
| for b in repo.branches(nodes): | for b in repo.branches(nodes): | ||||
| r.append(wireprototypes.encodelist(b) + "\n") | r.append(wireprototypes.encodelist(b) + "\n") | ||||
| return wireprototypes.bytesresponse(''.join(r)) | return wireprototypes.bytesresponse(''.join(r)) | ||||
| @wireprotocommand('clonebundles', '', permission='pull', | @wireprotocommand('clonebundles', '', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def clonebundles(repo, proto): | def clonebundles(repo, proto): | ||||
| """Server command for returning info for available bundles to seed clones. | """Server command for returning info for available bundles to seed clones. | ||||
| Clients will parse this response and determine what bundle to fetch. | Clients will parse this response and determine what bundle to fetch. | ||||
| Extensions may wrap this command to filter or dynamically emit data | Extensions may wrap this command to filter or dynamically emit data | ||||
| depending on the request. e.g. you could advertise URLs for the closest | depending on the request. e.g. you could advertise URLs for the closest | ||||
| data center given the client's IP address. | data center given the client's IP address. | ||||
| capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo, role='server')) | capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo, role='server')) | ||||
| caps.append('bundle2=' + urlreq.quote(capsblob)) | caps.append('bundle2=' + urlreq.quote(capsblob)) | ||||
| caps.append('unbundle=%s' % ','.join(bundle2.bundlepriority)) | caps.append('unbundle=%s' % ','.join(bundle2.bundlepriority)) | ||||
| return proto.addcapabilities(repo, caps) | return proto.addcapabilities(repo, caps) | ||||
| # If you are writing an extension and consider wrapping this function. Wrap | # If you are writing an extension and consider wrapping this function. Wrap | ||||
| # `_capabilities` instead. | # `_capabilities` instead. | ||||
| @wireprotocommand('capabilities', permission='pull', | @wireprotocommand('capabilities', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def capabilities(repo, proto): | def capabilities(repo, proto): | ||||
| caps = _capabilities(repo, proto) | caps = _capabilities(repo, proto) | ||||
| return wireprototypes.bytesresponse(' '.join(sorted(caps))) | return wireprototypes.bytesresponse(' '.join(sorted(caps))) | ||||
| @wireprotocommand('changegroup', 'roots', transportpolicy=POLICY_V1_ONLY, | @wireprotocommand('changegroup', 'roots', permission='pull') | ||||
| permission='pull') | |||||
| def changegroup(repo, proto, roots): | def changegroup(repo, proto, roots): | ||||
| nodes = wireprototypes.decodelist(roots) | nodes = wireprototypes.decodelist(roots) | ||||
| outgoing = discovery.outgoing(repo, missingroots=nodes, | outgoing = discovery.outgoing(repo, missingroots=nodes, | ||||
| missingheads=repo.heads()) | missingheads=repo.heads()) | ||||
| cg = changegroupmod.makechangegroup(repo, outgoing, '01', 'serve') | cg = changegroupmod.makechangegroup(repo, outgoing, '01', 'serve') | ||||
| gen = iter(lambda: cg.read(32768), '') | gen = iter(lambda: cg.read(32768), '') | ||||
| return wireprototypes.streamres(gen=gen) | return wireprototypes.streamres(gen=gen) | ||||
| @wireprotocommand('changegroupsubset', 'bases heads', | @wireprotocommand('changegroupsubset', 'bases heads', | ||||
| transportpolicy=POLICY_V1_ONLY, | |||||
| permission='pull') | permission='pull') | ||||
| def changegroupsubset(repo, proto, bases, heads): | def changegroupsubset(repo, proto, bases, heads): | ||||
| bases = wireprototypes.decodelist(bases) | bases = wireprototypes.decodelist(bases) | ||||
| heads = wireprototypes.decodelist(heads) | heads = wireprototypes.decodelist(heads) | ||||
| outgoing = discovery.outgoing(repo, missingroots=bases, | outgoing = discovery.outgoing(repo, missingroots=bases, | ||||
| missingheads=heads) | missingheads=heads) | ||||
| cg = changegroupmod.makechangegroup(repo, outgoing, '01', 'serve') | cg = changegroupmod.makechangegroup(repo, outgoing, '01', 'serve') | ||||
| gen = iter(lambda: cg.read(32768), '') | gen = iter(lambda: cg.read(32768), '') | ||||
| return wireprototypes.streamres(gen=gen) | return wireprototypes.streamres(gen=gen) | ||||
| @wireprotocommand('debugwireargs', 'one two *', | @wireprotocommand('debugwireargs', 'one two *', | ||||
| permission='pull', transportpolicy=POLICY_V1_ONLY) | permission='pull') | ||||
| def debugwireargs(repo, proto, one, two, others): | def debugwireargs(repo, proto, one, two, others): | ||||
| # only accept optional args from the known set | # only accept optional args from the known set | ||||
| opts = options('debugwireargs', ['three', 'four'], others) | opts = options('debugwireargs', ['three', 'four'], others) | ||||
| return wireprototypes.bytesresponse(repo.debugwireargs( | return wireprototypes.bytesresponse(repo.debugwireargs( | ||||
| one, two, **pycompat.strkwargs(opts))) | one, two, **pycompat.strkwargs(opts))) | ||||
| def find_pullbundle(repo, proto, opts, clheads, heads, common): | def find_pullbundle(repo, proto, opts, clheads, heads, common): | ||||
| """Return a file object for the first matching pullbundle. | """Return a file object for the first matching pullbundle. | ||||
| repo.ui.debug('sending pullbundle "%s"\n' % path) | repo.ui.debug('sending pullbundle "%s"\n' % path) | ||||
| try: | try: | ||||
| return repo.vfs.open(path) | return repo.vfs.open(path) | ||||
| except IOError: | except IOError: | ||||
| repo.ui.debug('pullbundle "%s" not accessible\n' % path) | repo.ui.debug('pullbundle "%s" not accessible\n' % path) | ||||
| continue | continue | ||||
| return None | return None | ||||
| @wireprotocommand('getbundle', '*', permission='pull', | @wireprotocommand('getbundle', '*', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def getbundle(repo, proto, others): | def getbundle(repo, proto, others): | ||||
| opts = options('getbundle', wireprototypes.GETBUNDLE_ARGUMENTS.keys(), | opts = options('getbundle', wireprototypes.GETBUNDLE_ARGUMENTS.keys(), | ||||
| others) | others) | ||||
| for k, v in opts.iteritems(): | for k, v in opts.iteritems(): | ||||
| keytype = wireprototypes.GETBUNDLE_ARGUMENTS[k] | keytype = wireprototypes.GETBUNDLE_ARGUMENTS[k] | ||||
| if keytype == 'nodes': | if keytype == 'nodes': | ||||
| opts[k] = wireprototypes.decodelist(v) | opts[k] = wireprototypes.decodelist(v) | ||||
| elif keytype == 'csv': | elif keytype == 'csv': | ||||
| bundler.addpart(bundle2.bundlepart('error:abort', | bundler.addpart(bundle2.bundlepart('error:abort', | ||||
| manargs, advargs)) | manargs, advargs)) | ||||
| chunks = bundler.getchunks() | chunks = bundler.getchunks() | ||||
| prefercompressed = False | prefercompressed = False | ||||
| return wireprototypes.streamres( | return wireprototypes.streamres( | ||||
| gen=chunks, prefer_uncompressed=not prefercompressed) | gen=chunks, prefer_uncompressed=not prefercompressed) | ||||
| @wireprotocommand('heads', permission='pull', transportpolicy=POLICY_V1_ONLY) | @wireprotocommand('heads', permission='pull') | ||||
| def heads(repo, proto): | def heads(repo, proto): | ||||
| h = repo.heads() | h = repo.heads() | ||||
| return wireprototypes.bytesresponse(wireprototypes.encodelist(h) + '\n') | return wireprototypes.bytesresponse(wireprototypes.encodelist(h) + '\n') | ||||
| @wireprotocommand('hello', permission='pull', transportpolicy=POLICY_V1_ONLY) | @wireprotocommand('hello', permission='pull') | ||||
| def hello(repo, proto): | def hello(repo, proto): | ||||
| """Called as part of SSH handshake to obtain server info. | """Called as part of SSH handshake to obtain server info. | ||||
| Returns a list of lines describing interesting things about the | Returns a list of lines describing interesting things about the | ||||
| server, in an RFC822-like format. | server, in an RFC822-like format. | ||||
| Currently, the only one defined is ``capabilities``, which consists of a | Currently, the only one defined is ``capabilities``, which consists of a | ||||
| line of space separated tokens describing server abilities: | line of space separated tokens describing server abilities: | ||||
| capabilities: <token0> <token1> <token2> | capabilities: <token0> <token1> <token2> | ||||
| """ | """ | ||||
| caps = capabilities(repo, proto).data | caps = capabilities(repo, proto).data | ||||
| return wireprototypes.bytesresponse('capabilities: %s\n' % caps) | return wireprototypes.bytesresponse('capabilities: %s\n' % caps) | ||||
| @wireprotocommand('listkeys', 'namespace', permission='pull', | @wireprotocommand('listkeys', 'namespace', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def listkeys(repo, proto, namespace): | def listkeys(repo, proto, namespace): | ||||
| d = sorted(repo.listkeys(encoding.tolocal(namespace)).items()) | d = sorted(repo.listkeys(encoding.tolocal(namespace)).items()) | ||||
| return wireprototypes.bytesresponse(pushkeymod.encodekeys(d)) | return wireprototypes.bytesresponse(pushkeymod.encodekeys(d)) | ||||
| @wireprotocommand('lookup', 'key', permission='pull', | @wireprotocommand('lookup', 'key', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def lookup(repo, proto, key): | def lookup(repo, proto, key): | ||||
| try: | try: | ||||
| k = encoding.tolocal(key) | k = encoding.tolocal(key) | ||||
| n = repo.lookup(k) | n = repo.lookup(k) | ||||
| r = hex(n) | r = hex(n) | ||||
| success = 1 | success = 1 | ||||
| except Exception as inst: | except Exception as inst: | ||||
| r = stringutil.forcebytestr(inst) | r = stringutil.forcebytestr(inst) | ||||
| success = 0 | success = 0 | ||||
| return wireprototypes.bytesresponse('%d %s\n' % (success, r)) | return wireprototypes.bytesresponse('%d %s\n' % (success, r)) | ||||
| @wireprotocommand('known', 'nodes *', permission='pull', | @wireprotocommand('known', 'nodes *', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def known(repo, proto, nodes, others): | def known(repo, proto, nodes, others): | ||||
| v = ''.join(b and '1' or '0' | v = ''.join(b and '1' or '0' | ||||
| for b in repo.known(wireprototypes.decodelist(nodes))) | for b in repo.known(wireprototypes.decodelist(nodes))) | ||||
| return wireprototypes.bytesresponse(v) | return wireprototypes.bytesresponse(v) | ||||
| @wireprotocommand('protocaps', 'caps', permission='pull', | @wireprotocommand('protocaps', 'caps', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def protocaps(repo, proto, caps): | def protocaps(repo, proto, caps): | ||||
| if proto.name == wireprototypes.SSHV1: | if proto.name == wireprototypes.SSHV1: | ||||
| proto._protocaps = set(caps.split(' ')) | proto._protocaps = set(caps.split(' ')) | ||||
| return wireprototypes.bytesresponse('OK') | return wireprototypes.bytesresponse('OK') | ||||
| @wireprotocommand('pushkey', 'namespace key old new', permission='push', | @wireprotocommand('pushkey', 'namespace key old new', permission='push') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def pushkey(repo, proto, namespace, key, old, new): | def pushkey(repo, proto, namespace, key, old, new): | ||||
| # compatibility with pre-1.8 clients which were accidentally | # compatibility with pre-1.8 clients which were accidentally | ||||
| # sending raw binary nodes rather than utf-8-encoded hex | # sending raw binary nodes rather than utf-8-encoded hex | ||||
| if len(new) == 20 and stringutil.escapestr(new) != new: | if len(new) == 20 and stringutil.escapestr(new) != new: | ||||
| # looks like it could be a binary node | # looks like it could be a binary node | ||||
| try: | try: | ||||
| new.decode('utf-8') | new.decode('utf-8') | ||||
| new = encoding.tolocal(new) # but cleanly decodes as UTF-8 | new = encoding.tolocal(new) # but cleanly decodes as UTF-8 | ||||
| except UnicodeDecodeError: | except UnicodeDecodeError: | ||||
| pass # binary, leave unmodified | pass # binary, leave unmodified | ||||
| else: | else: | ||||
| new = encoding.tolocal(new) # normal path | new = encoding.tolocal(new) # normal path | ||||
| with proto.mayberedirectstdio() as output: | with proto.mayberedirectstdio() as output: | ||||
| r = repo.pushkey(encoding.tolocal(namespace), encoding.tolocal(key), | r = repo.pushkey(encoding.tolocal(namespace), encoding.tolocal(key), | ||||
| encoding.tolocal(old), new) or False | encoding.tolocal(old), new) or False | ||||
| output = output.getvalue() if output else '' | output = output.getvalue() if output else '' | ||||
| return wireprototypes.bytesresponse('%d\n%s' % (int(r), output)) | return wireprototypes.bytesresponse('%d\n%s' % (int(r), output)) | ||||
| @wireprotocommand('stream_out', permission='pull', | @wireprotocommand('stream_out', permission='pull') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def stream(repo, proto): | def stream(repo, proto): | ||||
| '''If the server supports streaming clone, it advertises the "stream" | '''If the server supports streaming clone, it advertises the "stream" | ||||
| capability with a value representing the version and flags of the repo | capability with a value representing the version and flags of the repo | ||||
| it is serving. Client checks to see if it understands the format. | it is serving. Client checks to see if it understands the format. | ||||
| ''' | ''' | ||||
| return wireprototypes.streamreslegacy( | return wireprototypes.streamreslegacy( | ||||
| streamclone.generatev1wireproto(repo)) | streamclone.generatev1wireproto(repo)) | ||||
| @wireprotocommand('unbundle', 'heads', permission='push', | @wireprotocommand('unbundle', 'heads', permission='push') | ||||
| transportpolicy=POLICY_V1_ONLY) | |||||
| def unbundle(repo, proto, heads): | def unbundle(repo, proto, heads): | ||||
| their_heads = wireprototypes.decodelist(heads) | their_heads = wireprototypes.decodelist(heads) | ||||
| with proto.mayberedirectstdio() as output: | with proto.mayberedirectstdio() as output: | ||||
| try: | try: | ||||
| exchange.check_heads(repo, their_heads, 'preparing changes') | exchange.check_heads(repo, their_heads, 'preparing changes') | ||||
| cleanup = lambda: None | cleanup = lambda: None | ||||
| try: | try: | ||||