Issue #15633: httplib.HTTPResponse is now mark closed when the server sends less than the advertised Content-Length.

This commit is contained in:
Antoine Pitrou 2013-02-02 22:49:34 +01:00
parent 1efd9824d8
commit d66c0ee76e
3 changed files with 27 additions and 7 deletions

View file

@ -547,7 +547,11 @@ class HTTPResponse:
if self.length is None:
s = self.fp.read()
else:
s = self._safe_read(self.length)
try:
s = self._safe_read(self.length)
except IncompleteRead:
self.close()
raise
self.length = 0
self.close() # we read everything
return s
@ -561,13 +565,14 @@ class HTTPResponse:
# connection, and the user is reading more bytes than will be provided
# (for example, reading in 1k chunks)
s = self.fp.read(amt)
if not s:
# Ideally, we would raise IncompleteRead if the content-length
# wasn't satisfied, but it might break compatibility.
self.close()
if self.length is not None:
self.length -= len(s)
if not self.length:
self.close()
else:
if not s:
self.close()
return s