Browse Source

[utils] Percent-encode redirect URL of Location header (Closes #6457)

totalwebcasting
Sergey M․ 9 years ago
parent
commit
5a4d9ddb21
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      youtube_dl/utils.py

+ 11
- 0
youtube_dl/utils.py View File

@ -715,6 +715,17 @@ class YoutubeDLHandler(compat_urllib_request.HTTPHandler):
gz = io.BytesIO(self.deflate(resp.read()))
resp = self.addinfourl_wrapper(gz, old_resp.headers, old_resp.url, old_resp.code)
resp.msg = old_resp.msg
# Percent-encode redirect URL of Location HTTP header to satisfy RFC 3986
if 300 <= resp.code < 400:
location = resp.headers.get('Location')
if location:
# As of RFC 2616 default charset is iso-8859-1 that is respected by python 3
if sys.version_info >= (3, 0):
location = location.encode('iso-8859-1').decode('utf-8')
location_escaped = escape_url(location)
if location != location_escaped:
del resp.headers['Location']
resp.headers['Location'] = location_escaped
return resp
https_request = http_request


Loading…
Cancel
Save