comparison mercurial/sshpeer.py @ 37411:3e1688711efd

wireproto: turn client capabilities into sets, sorted on the wire

Differential Revision: https://phab.mercurial-scm.org/D3169

37410:a6651f5e2c78

def _clientcapabilities():
    """Return list of capabilities of this client.

    Returns a list of capabilities that are supported by this client.
    """
    protoparams = []
    comps = [e.wireprotosupport().name for e in
             util.compengines.supportedwireengines(util.CLIENTROLE)]
    protoparams.append('comp=%s' % ','.join(comps))
    return protoparams

def _performhandshake(ui, stdin, stdout, stderr):
    def badresponse():
        # Flush any output on stderr.

    # Finally, if supported by the server, notify it about our own
    # capabilities.
    if 'protocaps' in peer.capabilities():
        try:
            peer._call("protocaps", caps=' '.join(_clientcapabilities()))
        except IOError:
            peer._cleanup()
            raise error.RepoError(_('capability exchange failed'))

    return peer

37411:3e1688711efd

def _clientcapabilities():
    """Return list of capabilities of this client.

    Returns a list of capabilities that are supported by this client.
    """
    protoparams = set()
    comps = [e.wireprotosupport().name for e in
             util.compengines.supportedwireengines(util.CLIENTROLE)]
    protoparams.add('comp=%s' % ','.join(comps))
    return protoparams

def _performhandshake(ui, stdin, stdout, stderr):
    def badresponse():
        # Flush any output on stderr.

    # Finally, if supported by the server, notify it about our own
    # capabilities.
    if 'protocaps' in peer.capabilities():
        try:
            peer._call("protocaps",
                       caps=' '.join(sorted(_clientcapabilities())))
        except IOError:
            peer._cleanup()
            raise error.RepoError(_('capability exchange failed'))

    return peer