diff options
author | Kristjan Valur Jonsson <sweskman@gmail.com> | 2011-03-30 11:54:13 +0000 |
---|---|---|
committer | Kristjan Valur Jonsson <sweskman@gmail.com> | 2011-03-30 11:54:13 +0000 |
commit | d05595697d70bef1c5cf66484e1d1102d510152e (patch) | |
tree | 6226ead31564a8c7fb1177a261ab3c7402c8c89a /Lib/urllib/request.py | |
parent | 35722a93768e942be31270fe60acff138d878b26 (diff) | |
parent | d6c1d57195d907a8187442cd2f3e9d9575d6b4a8 (diff) | |
download | cpython-git-d05595697d70bef1c5cf66484e1d1102d510152e.tar.gz |
Merge 3.1
Diffstat (limited to 'Lib/urllib/request.py')
-rw-r--r-- | Lib/urllib/request.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/Lib/urllib/request.py b/Lib/urllib/request.py index 087e9a606c..220dfe4b19 100644 --- a/Lib/urllib/request.py +++ b/Lib/urllib/request.py @@ -528,6 +528,17 @@ class HTTPRedirectHandler(BaseHandler): # fix a possible malformed URL urlparts = urlparse(newurl) + + # For security reasons we don't allow redirection to anything other + # than http, https or ftp. + + if not urlparts.scheme in ('http', 'https', 'ftp'): + raise HTTPError(newurl, code, + msg + + " - Redirection to url '%s' is not allowed" % + newurl, + headers, fp) + if not urlparts.path: urlparts = list(urlparts) urlparts[2] = "/" @@ -1864,8 +1875,24 @@ class FancyURLopener(URLopener): return void = fp.read() fp.close() + # In case the server sent a relative URL, join with original: newurl = urljoin(self.type + ":" + url, newurl) + + urlparts = urlparse(newurl) + + # For security reasons, we don't allow redirection to anything other + # than http, https and ftp. + + # We are using newer HTTPError with older redirect_internal method + # This older method will get deprecated in 3.3 + + if not urlparts.scheme in ('http', 'https', 'ftp'): + raise HTTPError(newurl, errcode, + errmsg + + " Redirection to url '%s' is not allowed." % newurl, + headers, fp) + return self.open(newurl) def http_error_301(self, url, fp, errcode, errmsg, headers, data=None): |