comparison mercurial/sshpeer.py @ 21073:83ce71ef7804

sshpeer: add implementation of _calltwowaystream

This implements the call needed by bundle2. The error handling is a bit flaky
right now, but bundle2 error handling in general is still flaky anyway. Bundle2
is still disabled by default, I do not expect this to be a problem.

21072:0879352d67d8

        r = self._recv()
        if r:
            return '', r
        return self._recv(), ''


    def _recv(self):
        l = self.pipei.readline()
        if l == '\n':
            err = []

21073:83ce71ef7804

        r = self._recv()
        if r:
            return '', r
        return self._recv(), ''

    def _calltwowaystream(self, cmd, fp, **args):
        r = self._call(cmd, **args)
        if r:
            # XXX needs to be made better
            raise util.Abort('unexpected remote reply: %s' % r)
        while True:
            d = fp.read(4096)
            if not d:
                break
            self._send(d)
        self._send("", flush=True)
        return self.pipei

    def _recv(self):
        l = self.pipei.readline()
        if l == '\n':
            err = []