[PATCH] httpconnection: make sure to clear progress of httpsendfile at EOF

Yuya Nishihara yuya at tcha.org
Sat Mar 18 07:45:48 UTC 2017


# HG changeset patch
# User Yuya Nishihara <yuya at tcha.org>
# Date 1489820534 -32400
#      Sat Mar 18 16:02:14 2017 +0900
# Node ID 6d6c4d50c59b87e01936095f34720e79b812d05b
# Parent  da7d19324b1e3d2ce0636bf73e76bf8f3d68ed27
httpconnection: make sure to clear progress of httpsendfile at EOF

read() should never raise EOFError. If it did, UnboundLocalError would occur
due to unassigned 'ret' variable.

This issue was originally reported to TortoiseHg:
https://bitbucket.org/tortoisehg/thg/issues/4707/

diff --git a/mercurial/httpconnection.py b/mercurial/httpconnection.py
--- a/mercurial/httpconnection.py
+++ b/mercurial/httpconnection.py
@@ -44,10 +44,10 @@ class httpsendfile(object):
         self._total = self.length // 1024 * 2
 
     def read(self, *args, **kwargs):
-        try:
-            ret = self._data.read(*args, **kwargs)
-        except EOFError:
+        ret = self._data.read(*args, **kwargs)
+        if not ret:
             self.ui.progress(_('sending'), None)
+            return ret
         self._pos += len(ret)
         # We pass double the max for total because we currently have
         # to send the bundle twice in the case of a server that


More information about the Mercurial-devel mailing list