nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

Avoid downloading of too big files

Browse Source
This commit is contained in:
Némunaire 2013-01-04 18:05:41 +01:00
parent 309615cdc7
commit bd4a9e17eb
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/web.py
View File

@ -110,7 +110,13 @@ def getURLContent(url, timeout=15):
try: try:
res = conn.getresponse() res = conn.getresponse()
data = res.read() size = int(res.getheader("Content-Length", 5000))
cntype = res.getheader("Content-Type")
if size > 10000 or cntype[:4] != "text":
return None
data = res.read(size)
except http.client.BadStatusLine: except http.client.BadStatusLine:
return None return None
finally: finally: