comparison mercurial/url.py @ 51992:71658f79758a

clonebundle-digest: add recursion guards for Python 3.8

For Python 3.8 and 3.9, the read/readinto pair can recurse, so make sure
the data is only hashed once.

51991:f0bee30af890

                                self._digest_consumed,
                            )
                        )

            def read(self, amt=None):
                data = super().read(amt)
                self._digest_input(data)
                return data

            def readline(self):
                data = super().readline()
                self._digest_input(data)
                return data

            def readinto(self, dest):
                got = super().readinto(dest)
                self._digest_input(dest[:got])
                return got

            def _close_conn(self):
                self._digest_finished = True
                return super().close()

        response.__class__ = digestresponse
        response._digest = self._digest
        response._digest_consumed = 0
        response._hasher = self._hasher.copy()
        response._digest_finished = False
        return response

    https_response = http_response

51992:71658f79758a

                                self._digest_consumed,
                            )
                        )

            def read(self, amt=None):
                self._digest_recursion_level += 1
                data = super().read(amt)
                self._digest_recursion_level -= 1
                if self._digest_recursion_level == 0:
                    self._digest_input(data)
                return data

            def readline(self):
                self._digest_recursion_level += 1
                data = super().readline()
                self._digest_recursion_level -= 1
                if self._digest_recursion_level == 0:
                    self._digest_input(data)
                return data

            def readinto(self, dest):
                self._digest_recursion_level += 1
                got = super().readinto(dest)
                self._digest_recursion_level -= 1
                if self._digest_recursion_level == 0:
                    self._digest_input(dest[:got])
                return got

            def _close_conn(self):
                self._digest_finished = True
                return super().close()

        response.__class__ = digestresponse
        response._digest = self._digest
        response._digest_consumed = 0
        response._hasher = self._hasher.copy()
        # Python 3.8 / 3.9 recurses internally between read/readinto.
        response._digest_recursion_level = 0
        response._digest_finished = False
        return response

    https_response = http_response