Merge pull request #3 from startersclan/fix/python-fix-regression-in-miniclient.http-get-to-read-full-response-body-if-content-length-or-transfer-encoding-http-headers-are-absent

Fix (python): Fix regression in `miniclient.http_get()` to read full response body if `Content-Length` or `Transfer-Encoding` HTTP Headers are absent

Author: leojonathanoh

stats/miniclient.py

@@ -51,14 +51,18 @@ def http_get(host, port = 80, document = "/"):
 			if "Content-Length" in headers:
 				content_length = headers["Content-Length"]
 				body = http.read()
-			elif headers["Transfer-Encoding"] == "chunked":
+			elif "Transfer-Encoding" in headers and headers["Transfer-Encoding"] == "chunked":
 				while 1:
 					chunk_length = int(http.readline(), 16)
 					if chunk_length != 0:
 						body += http.read(chunk_length)
 					http.readline() # CRLF
 					if chunk_length == 0:
 						break
+			else:
+				# No Content-Length nor Transfer-Encoding header. Read until EOF
+				body = http.read()
+
 			http.shutdown() # be nice, tell the http server we're done sending the request
 			http.close() # all done
 			return body
@@ -123,14 +127,18 @@ def http_postSnapshot(host, port = 80, document = "/", snapshot = ""):
 			if "Content-Length" in headers:
 				content_length = headers["Content-Length"]
 				body = http.read()
-			elif headers["Transfer-Encoding"] == "chunked":
+			elif "Transfer-Encoding" in headers and headers["Transfer-Encoding"] == "chunked":
 				while 1:
 					chunk_length = int(http.readline(), 16)
 					if chunk_length != 0:
 						body += http.read(chunk_length)
 					http.readline() # CRLF
 					if chunk_length == 0:
 						break
+			else:
+				# No Content-Length nor Transfer-Encoding header. Read until EOF
+				body = http.read()
+
 			http.shutdown() # be nice, tell the http server we're done sending the request
 			http.close() # all done
 			return body