blob: df2ff06f0fc9a25af82d6ceb1dab88719186e10c [file] [log] [blame]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001"""An extensible library for opening URLs using a variety of protocols
2
3The simplest way to use this module is to call the urlopen function,
4which accepts a string containing a URL or a Request object (described
5below). It opens the URL and returns the results as file-like
6object; the returned object has some extra methods described below.
7
8The OpenerDirector manages a collection of Handler objects that do
9all the actual work. Each Handler implements a particular protocol or
10option. The OpenerDirector is a composite object that invokes the
11Handlers needed to open the requested URL. For example, the
12HTTPHandler performs HTTP GET and POST requests and deals with
13non-error returns. The HTTPRedirectHandler automatically deals with
14HTTP 301, 302, 303 and 307 redirect errors, and the HTTPDigestAuthHandler
15deals with digest authentication.
16
17urlopen(url, data=None) -- Basic usage is the same as original
18urllib. pass the url and optionally data to post to an HTTP URL, and
19get a file-like object back. One difference is that you can also pass
20a Request instance instead of URL. Raises a URLError (subclass of
Andrew Svetlovf7a17b42012-12-25 16:47:37 +020021OSError); for HTTP errors, raises an HTTPError, which can also be
Jeremy Hylton1afc1692008-06-18 20:49:58 +000022treated as a valid response.
23
24build_opener -- Function that creates a new OpenerDirector instance.
25Will install the default handlers. Accepts one or more Handlers as
26arguments, either instances or Handler classes that it will
27instantiate. If one of the argument is a subclass of the default
28handler, the argument will be installed instead of the default.
29
30install_opener -- Installs a new opener as the default opener.
31
32objects of interest:
Senthil Kumaran1107c5d2009-11-15 06:20:55 +000033
Senthil Kumaran47fff872009-12-20 07:10:31 +000034OpenerDirector -- Sets up the User Agent as the Python-urllib client and manages
35the Handler classes, while dealing with requests and responses.
Jeremy Hylton1afc1692008-06-18 20:49:58 +000036
37Request -- An object that encapsulates the state of a request. The
38state can be as simple as the URL. It can also include extra HTTP
39headers, e.g. a User-Agent.
40
41BaseHandler --
42
43internals:
44BaseHandler and parent
45_call_chain conventions
46
47Example usage:
48
Georg Brandl029986a2008-06-23 11:44:14 +000049import urllib.request
Jeremy Hylton1afc1692008-06-18 20:49:58 +000050
51# set up authentication info
Georg Brandl029986a2008-06-23 11:44:14 +000052authinfo = urllib.request.HTTPBasicAuthHandler()
Jeremy Hylton1afc1692008-06-18 20:49:58 +000053authinfo.add_password(realm='PDQ Application',
54 uri='https://mahler:8092/site-updates.py',
55 user='klem',
56 passwd='geheim$parole')
57
Georg Brandl029986a2008-06-23 11:44:14 +000058proxy_support = urllib.request.ProxyHandler({"http" : "http://ahad-haam:3128"})
Jeremy Hylton1afc1692008-06-18 20:49:58 +000059
60# build a new opener that adds authentication and caching FTP handlers
Georg Brandl029986a2008-06-23 11:44:14 +000061opener = urllib.request.build_opener(proxy_support, authinfo,
62 urllib.request.CacheFTPHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +000063
64# install it
Georg Brandl029986a2008-06-23 11:44:14 +000065urllib.request.install_opener(opener)
Jeremy Hylton1afc1692008-06-18 20:49:58 +000066
Georg Brandl029986a2008-06-23 11:44:14 +000067f = urllib.request.urlopen('http://www.python.org/')
Jeremy Hylton1afc1692008-06-18 20:49:58 +000068"""
69
70# XXX issues:
71# If an authentication error handler that tries to perform
72# authentication for some reason but fails, how should the error be
73# signalled? The client needs to know the HTTP error code. But if
74# the handler knows that the problem was, e.g., that it didn't know
75# that hash algo that requested in the challenge, it would be good to
76# pass that information along to the client, too.
77# ftp errors aren't handled cleanly
78# check digest against correct (i.e. non-apache) implementation
79
80# Possible extensions:
81# complex proxies XXX not sure what exactly was meant by this
82# abstract factory for opener
83
84import base64
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +000085import bisect
Jeremy Hylton1afc1692008-06-18 20:49:58 +000086import email
87import hashlib
88import http.client
89import io
90import os
91import posixpath
Jeremy Hylton1afc1692008-06-18 20:49:58 +000092import re
93import socket
Martin Pantere6f06092016-05-16 01:14:20 +000094import string
Jeremy Hylton1afc1692008-06-18 20:49:58 +000095import sys
96import time
Senthil Kumarane24f96a2012-03-13 19:29:33 -070097import tempfile
98import contextlib
Senthil Kumaran38b968b92012-03-14 13:43:53 -070099import warnings
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700100
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000101
Georg Brandl13e89462008-07-01 19:56:00 +0000102from urllib.error import URLError, HTTPError, ContentTooShortError
103from urllib.parse import (
Cheryl Sabella0250de42018-04-25 16:51:54 -0700104 urlparse, urlsplit, urljoin, _unwrap, quote, unquote,
105 _splittype, _splithost, _splitport, _splituser, _splitpasswd,
106 _splitattr, _splitquery, _splitvalue, _splittag, _to_bytes,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100107 unquote_to_bytes, urlunparse)
Georg Brandl13e89462008-07-01 19:56:00 +0000108from urllib.response import addinfourl, addclosehook
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000109
110# check for SSL
111try:
112 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -0400113except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000114 _have_ssl = False
115else:
116 _have_ssl = True
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000117
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800118__all__ = [
119 # Classes
120 'Request', 'OpenerDirector', 'BaseHandler', 'HTTPDefaultErrorHandler',
121 'HTTPRedirectHandler', 'HTTPCookieProcessor', 'ProxyHandler',
122 'HTTPPasswordMgr', 'HTTPPasswordMgrWithDefaultRealm',
R David Murray4c7f9952015-04-16 16:36:18 -0400123 'HTTPPasswordMgrWithPriorAuth', 'AbstractBasicAuthHandler',
124 'HTTPBasicAuthHandler', 'ProxyBasicAuthHandler', 'AbstractDigestAuthHandler',
125 'HTTPDigestAuthHandler', 'ProxyDigestAuthHandler', 'HTTPHandler',
126 'FileHandler', 'FTPHandler', 'CacheFTPHandler', 'DataHandler',
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800127 'UnknownHandler', 'HTTPErrorProcessor',
128 # Functions
129 'urlopen', 'install_opener', 'build_opener',
130 'pathname2url', 'url2pathname', 'getproxies',
131 # Legacy interface
132 'urlretrieve', 'urlcleanup', 'URLopener', 'FancyURLopener',
133]
134
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000135# used in User-Agent header sent
Serhiy Storchaka885bdc42016-02-11 13:10:36 +0200136__version__ = '%d.%d' % sys.version_info[:2]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000137
138_opener = None
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000139def urlopen(url, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Senthil Kumarana5c85b32014-09-19 15:23:30 +0800140 *, cafile=None, capath=None, cadefault=False, context=None):
Raymond Hettinger507343a2015-08-18 00:35:52 -0700141 '''Open the URL url, which can be either a string or a Request object.
142
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000143 *data* must be an object specifying additional data to be sent to
144 the server, or None if no such data is needed. See Request for
145 details.
Raymond Hettinger507343a2015-08-18 00:35:52 -0700146
147 urllib.request module uses HTTP/1.1 and includes a "Connection:close"
148 header in its HTTP requests.
149
150 The optional *timeout* parameter specifies a timeout in seconds for
151 blocking operations like the connection attempt (if not specified, the
152 global default timeout setting will be used). This only works for HTTP,
153 HTTPS and FTP connections.
154
155 If *context* is specified, it must be a ssl.SSLContext instance describing
156 the various SSL options. See HTTPSConnection for more details.
157
158 The optional *cafile* and *capath* parameters specify a set of trusted CA
159 certificates for HTTPS requests. cafile should point to a single file
160 containing a bundle of CA certificates, whereas capath should point to a
161 directory of hashed certificate files. More information can be found in
162 ssl.SSLContext.load_verify_locations().
163
164 The *cadefault* parameter is ignored.
165
Martin Panter29f256902016-06-04 05:06:34 +0000166 This function always returns an object which can work as a context
167 manager and has methods such as
Raymond Hettinger507343a2015-08-18 00:35:52 -0700168
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200169 * geturl() - return the URL of the resource retrieved, commonly used to
Raymond Hettinger507343a2015-08-18 00:35:52 -0700170 determine if a redirect was followed
171
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200172 * info() - return the meta-information of the page, such as headers, in the
Raymond Hettinger507343a2015-08-18 00:35:52 -0700173 form of an email.message_from_string() instance (see Quick Reference to
174 HTTP Headers)
175
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200176 * getcode() - return the HTTP status code of the response. Raises URLError
Raymond Hettinger507343a2015-08-18 00:35:52 -0700177 on errors.
178
Martin Panter29f256902016-06-04 05:06:34 +0000179 For HTTP and HTTPS URLs, this function returns a http.client.HTTPResponse
180 object slightly modified. In addition to the three new methods above, the
181 msg attribute contains the same information as the reason attribute ---
182 the reason phrase returned by the server --- instead of the response
183 headers as it is specified in the documentation for HTTPResponse.
R David Murrayd2367c62016-06-03 20:16:06 -0400184
Martin Panter29f256902016-06-04 05:06:34 +0000185 For FTP, file, and data URLs and requests explicitly handled by legacy
186 URLopener and FancyURLopener classes, this function returns a
187 urllib.response.addinfourl object.
188
189 Note that None may be returned if no handler handles the request (though
Raymond Hettinger507343a2015-08-18 00:35:52 -0700190 the default installed global OpenerDirector uses UnknownHandler to ensure
191 this never happens).
192
193 In addition, if proxy settings are detected (for example, when a *_proxy
194 environment variable like http_proxy is set), ProxyHandler is default
195 installed and makes sure the requests are handled through the proxy.
196
197 '''
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000198 global _opener
Antoine Pitroude9ac6c2012-05-16 21:40:01 +0200199 if cafile or capath or cadefault:
Christian Heimesd0486372016-09-10 23:23:33 +0200200 import warnings
Boštjan Mejak15869582018-11-25 19:32:50 +0100201 warnings.warn("cafile, capath and cadefault are deprecated, use a "
Christian Heimesd0486372016-09-10 23:23:33 +0200202 "custom context instead.", DeprecationWarning, 2)
Senthil Kumarana5c85b32014-09-19 15:23:30 +0800203 if context is not None:
204 raise ValueError(
205 "You can't pass both context and any of cafile, capath, and "
206 "cadefault"
207 )
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000208 if not _have_ssl:
209 raise ValueError('SSL support not available')
Benjamin Petersonb6666972014-12-07 13:46:02 -0500210 context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH,
Christian Heimes67986f92013-11-23 22:43:47 +0100211 cafile=cafile,
212 capath=capath)
Benjamin Petersonb6666972014-12-07 13:46:02 -0500213 https_handler = HTTPSHandler(context=context)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000214 opener = build_opener(https_handler)
Senthil Kumarana5c85b32014-09-19 15:23:30 +0800215 elif context:
216 https_handler = HTTPSHandler(context=context)
217 opener = build_opener(https_handler)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000218 elif _opener is None:
219 _opener = opener = build_opener()
220 else:
221 opener = _opener
222 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000223
224def install_opener(opener):
225 global _opener
226 _opener = opener
227
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700228_url_tempfiles = []
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000229def urlretrieve(url, filename=None, reporthook=None, data=None):
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700230 """
231 Retrieve a URL into a temporary location on disk.
232
233 Requires a URL argument. If a filename is passed, it is used as
234 the temporary file location. The reporthook argument should be
235 a callable that accepts a block number, a read size, and the
236 total file size of the URL target. The data argument should be
237 valid URL encoded data.
238
239 If a filename is passed and the URL points to a local resource,
240 the result is a copy from local file to new file.
241
242 Returns a tuple containing the path to the newly created
243 data file as well as the resulting HTTPMessage object.
244 """
Cheryl Sabella0250de42018-04-25 16:51:54 -0700245 url_type, path = _splittype(url)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700246
247 with contextlib.closing(urlopen(url, data)) as fp:
248 headers = fp.info()
249
250 # Just return the local path and the "headers" for file://
251 # URLs. No sense in performing a copy unless requested.
252 if url_type == "file" and not filename:
253 return os.path.normpath(path), headers
254
255 # Handle temporary file setup.
256 if filename:
257 tfp = open(filename, 'wb')
258 else:
259 tfp = tempfile.NamedTemporaryFile(delete=False)
260 filename = tfp.name
261 _url_tempfiles.append(filename)
262
263 with tfp:
264 result = filename, headers
265 bs = 1024*8
266 size = -1
267 read = 0
268 blocknum = 0
269 if "content-length" in headers:
270 size = int(headers["Content-Length"])
271
272 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800273 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700274
275 while True:
276 block = fp.read(bs)
277 if not block:
278 break
279 read += len(block)
280 tfp.write(block)
281 blocknum += 1
282 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800283 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700284
285 if size >= 0 and read < size:
286 raise ContentTooShortError(
287 "retrieval incomplete: got only %i out of %i bytes"
288 % (read, size), result)
289
290 return result
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000291
292def urlcleanup():
Robert Collins2fee5c92015-08-04 12:52:06 +1200293 """Clean up temporary files from urlretrieve calls."""
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700294 for temp_file in _url_tempfiles:
295 try:
296 os.unlink(temp_file)
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200297 except OSError:
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700298 pass
299
300 del _url_tempfiles[:]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000301 global _opener
302 if _opener:
303 _opener = None
304
305# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000306_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000307def request_host(request):
308 """Return request-host, as defined by RFC 2965.
309
310 Variation from RFC: returned value is lowercased, for convenient
311 comparison.
312
313 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000314 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000315 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000316 if host == "":
317 host = request.get_header("Host", "")
318
319 # remove port, if present
320 host = _cut_port_re.sub("", host, 1)
321 return host.lower()
322
323class Request:
324
325 def __init__(self, url, data=None, headers={},
Senthil Kumarande49d642011-10-16 23:54:44 +0800326 origin_req_host=None, unverifiable=False,
327 method=None):
Senthil Kumaran52380922013-04-25 05:45:48 -0700328 self.full_url = url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000329 self.headers = {}
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200330 self.unredirected_hdrs = {}
331 self._data = None
332 self.data = data
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000333 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000334 for key, value in headers.items():
335 self.add_header(key, value)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000336 if origin_req_host is None:
337 origin_req_host = request_host(self)
338 self.origin_req_host = origin_req_host
339 self.unverifiable = unverifiable
Jason R. Coombs7dc4f4b2013-09-08 12:47:07 -0400340 if method:
341 self.method = method
Senthil Kumaran52380922013-04-25 05:45:48 -0700342
343 @property
344 def full_url(self):
Senthil Kumaran83070752013-05-24 09:14:12 -0700345 if self.fragment:
346 return '{}#{}'.format(self._full_url, self.fragment)
Senthil Kumaran52380922013-04-25 05:45:48 -0700347 return self._full_url
348
349 @full_url.setter
350 def full_url(self, url):
351 # unwrap('<URL:type://host/path>') --> 'type://host/path'
Cheryl Sabella0250de42018-04-25 16:51:54 -0700352 self._full_url = _unwrap(url)
353 self._full_url, self.fragment = _splittag(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000354 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000355
Senthil Kumaran52380922013-04-25 05:45:48 -0700356 @full_url.deleter
357 def full_url(self):
358 self._full_url = None
359 self.fragment = None
360 self.selector = ''
361
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200362 @property
363 def data(self):
364 return self._data
365
366 @data.setter
367 def data(self, data):
368 if data != self._data:
369 self._data = data
370 # issue 16464
371 # if we change data we need to remove content-length header
372 # (cause it's most probably calculated for previous value)
373 if self.has_header("Content-length"):
374 self.remove_header("Content-length")
375
376 @data.deleter
377 def data(self):
R David Murray9cc7d452013-03-20 00:10:51 -0400378 self.data = None
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200379
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000380 def _parse(self):
Cheryl Sabella0250de42018-04-25 16:51:54 -0700381 self.type, rest = _splittype(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000382 if self.type is None:
R David Murrayd8a46962013-04-03 06:58:34 -0400383 raise ValueError("unknown url type: %r" % self.full_url)
Cheryl Sabella0250de42018-04-25 16:51:54 -0700384 self.host, self.selector = _splithost(rest)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000385 if self.host:
386 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000387
388 def get_method(self):
Senthil Kumarande49d642011-10-16 23:54:44 +0800389 """Return a string indicating the HTTP request method."""
Jason R. Coombsaae6a1d2013-09-08 12:54:33 -0400390 default_method = "POST" if self.data is not None else "GET"
391 return getattr(self, 'method', default_method)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000392
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000393 def get_full_url(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700394 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000395
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000396 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000397 if self.type == 'https' and not self._tunnel_host:
398 self._tunnel_host = self.host
399 else:
400 self.type= type
401 self.selector = self.full_url
402 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000403
404 def has_proxy(self):
405 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000406
407 def add_header(self, key, val):
408 # useful for something like authentication
409 self.headers[key.capitalize()] = val
410
411 def add_unredirected_header(self, key, val):
412 # will not be added to a redirected request
413 self.unredirected_hdrs[key.capitalize()] = val
414
415 def has_header(self, header_name):
416 return (header_name in self.headers or
417 header_name in self.unredirected_hdrs)
418
419 def get_header(self, header_name, default=None):
420 return self.headers.get(
421 header_name,
422 self.unredirected_hdrs.get(header_name, default))
423
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200424 def remove_header(self, header_name):
425 self.headers.pop(header_name, None)
426 self.unredirected_hdrs.pop(header_name, None)
427
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000428 def header_items(self):
Serhiy Storchakada084702019-03-27 08:02:28 +0200429 hdrs = {**self.unredirected_hdrs, **self.headers}
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000430 return list(hdrs.items())
431
432class OpenerDirector:
433 def __init__(self):
434 client_version = "Python-urllib/%s" % __version__
435 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000436 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000437 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000438 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000439 self.handle_open = {}
440 self.handle_error = {}
441 self.process_response = {}
442 self.process_request = {}
443
444 def add_handler(self, handler):
445 if not hasattr(handler, "add_parent"):
446 raise TypeError("expected BaseHandler instance, got %r" %
447 type(handler))
448
449 added = False
450 for meth in dir(handler):
451 if meth in ["redirect_request", "do_open", "proxy_open"]:
452 # oops, coincidental match
453 continue
454
455 i = meth.find("_")
456 protocol = meth[:i]
457 condition = meth[i+1:]
458
459 if condition.startswith("error"):
460 j = condition.find("_") + i + 1
461 kind = meth[j+1:]
462 try:
463 kind = int(kind)
464 except ValueError:
465 pass
466 lookup = self.handle_error.get(protocol, {})
467 self.handle_error[protocol] = lookup
468 elif condition == "open":
469 kind = protocol
470 lookup = self.handle_open
471 elif condition == "response":
472 kind = protocol
473 lookup = self.process_response
474 elif condition == "request":
475 kind = protocol
476 lookup = self.process_request
477 else:
478 continue
479
480 handlers = lookup.setdefault(kind, [])
481 if handlers:
482 bisect.insort(handlers, handler)
483 else:
484 handlers.append(handler)
485 added = True
486
487 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000488 bisect.insort(self.handlers, handler)
489 handler.add_parent(self)
490
491 def close(self):
492 # Only exists for backwards compatibility.
493 pass
494
495 def _call_chain(self, chain, kind, meth_name, *args):
496 # Handlers raise an exception if no one else should try to handle
497 # the request, or return None if they can't but another handler
498 # could. Otherwise, they return the response.
499 handlers = chain.get(kind, ())
500 for handler in handlers:
501 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000502 result = func(*args)
503 if result is not None:
504 return result
505
506 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
507 # accept a URL or a Request object
508 if isinstance(fullurl, str):
509 req = Request(fullurl, data)
510 else:
511 req = fullurl
512 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000513 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000514
515 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000516 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000517
518 # pre-process request
519 meth_name = protocol+"_request"
520 for processor in self.process_request.get(protocol, []):
521 meth = getattr(processor, meth_name)
522 req = meth(req)
523
524 response = self._open(req, data)
525
526 # post-process response
527 meth_name = protocol+"_response"
528 for processor in self.process_response.get(protocol, []):
529 meth = getattr(processor, meth_name)
530 response = meth(req, response)
531
532 return response
533
534 def _open(self, req, data=None):
535 result = self._call_chain(self.handle_open, 'default',
536 'default_open', req)
537 if result:
538 return result
539
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000540 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000541 result = self._call_chain(self.handle_open, protocol, protocol +
542 '_open', req)
543 if result:
544 return result
545
546 return self._call_chain(self.handle_open, 'unknown',
547 'unknown_open', req)
548
549 def error(self, proto, *args):
550 if proto in ('http', 'https'):
551 # XXX http[s] protocols are special-cased
552 dict = self.handle_error['http'] # https is not different than http
553 proto = args[2] # YUCK!
554 meth_name = 'http_error_%s' % proto
555 http_err = 1
556 orig_args = args
557 else:
558 dict = self.handle_error
559 meth_name = proto + '_error'
560 http_err = 0
561 args = (dict, proto, meth_name) + args
562 result = self._call_chain(*args)
563 if result:
564 return result
565
566 if http_err:
567 args = (dict, 'default', 'http_error_default') + orig_args
568 return self._call_chain(*args)
569
570# XXX probably also want an abstract factory that knows when it makes
571# sense to skip a superclass in favor of a subclass and when it might
572# make sense to include both
573
574def build_opener(*handlers):
575 """Create an opener object from a list of handlers.
576
577 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000578 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000579
580 If any of the handlers passed as arguments are subclasses of the
581 default handlers, the default handlers will not be used.
582 """
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000583 opener = OpenerDirector()
584 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
585 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100586 FTPHandler, FileHandler, HTTPErrorProcessor,
587 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000588 if hasattr(http.client, "HTTPSConnection"):
589 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000590 skip = set()
591 for klass in default_classes:
592 for check in handlers:
Benjamin Peterson78c85382014-04-01 16:27:30 -0400593 if isinstance(check, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000594 if issubclass(check, klass):
595 skip.add(klass)
596 elif isinstance(check, klass):
597 skip.add(klass)
598 for klass in skip:
599 default_classes.remove(klass)
600
601 for klass in default_classes:
602 opener.add_handler(klass())
603
604 for h in handlers:
Benjamin Peterson5dd3cae2014-04-01 14:20:56 -0400605 if isinstance(h, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000606 h = h()
607 opener.add_handler(h)
608 return opener
609
610class BaseHandler:
611 handler_order = 500
612
613 def add_parent(self, parent):
614 self.parent = parent
615
616 def close(self):
617 # Only exists for backwards compatibility
618 pass
619
620 def __lt__(self, other):
621 if not hasattr(other, "handler_order"):
622 # Try to preserve the old behavior of having custom classes
623 # inserted after default ones (works only for custom user
624 # classes which are not aware of handler_order).
625 return True
626 return self.handler_order < other.handler_order
627
628
629class HTTPErrorProcessor(BaseHandler):
630 """Process HTTP error responses."""
631 handler_order = 1000 # after all other processing
632
633 def http_response(self, request, response):
634 code, msg, hdrs = response.code, response.msg, response.info()
635
636 # According to RFC 2616, "2xx" code indicates that the client's
637 # request was successfully received, understood, and accepted.
638 if not (200 <= code < 300):
639 response = self.parent.error(
640 'http', request, response, code, msg, hdrs)
641
642 return response
643
644 https_response = http_response
645
646class HTTPDefaultErrorHandler(BaseHandler):
647 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000648 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000649
650class HTTPRedirectHandler(BaseHandler):
651 # maximum number of redirections to any single URL
652 # this is needed because of the state that cookies introduce
653 max_repeats = 4
654 # maximum total number of redirections (regardless of URL) before
655 # assuming we're in a loop
656 max_redirections = 10
657
658 def redirect_request(self, req, fp, code, msg, headers, newurl):
659 """Return a Request or None in response to a redirect.
660
661 This is called by the http_error_30x methods when a
662 redirection response is received. If a redirection should
663 take place, return a new Request to allow http_error_30x to
664 perform the redirect. Otherwise, raise HTTPError if no-one
665 else should try to handle this url. Return None if you can't
666 but another Handler might.
667 """
668 m = req.get_method()
669 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
670 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000671 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000672
673 # Strictly (according to RFC 2616), 301 or 302 in response to
674 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000675 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000676 # essentially all clients do redirect in this case, so we do
677 # the same.
Martin Pantere6f06092016-05-16 01:14:20 +0000678
679 # Be conciliant with URIs containing a space. This is mainly
680 # redundant with the more complete encoding done in http_error_302(),
681 # but it is kept for compatibility with other callers.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000682 newurl = newurl.replace(' ', '%20')
Martin Pantere6f06092016-05-16 01:14:20 +0000683
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000684 CONTENT_HEADERS = ("content-length", "content-type")
Jon Dufresne39726282017-05-18 07:35:54 -0700685 newheaders = {k: v for k, v in req.headers.items()
686 if k.lower() not in CONTENT_HEADERS}
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000687 return Request(newurl,
688 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000689 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000690 unverifiable=True)
691
692 # Implementation note: To avoid the server sending us into an
693 # infinite loop, the request object needs to track what URLs we
694 # have already seen. Do this by adding a handler-specific
695 # attribute to the Request object.
696 def http_error_302(self, req, fp, code, msg, headers):
697 # Some servers (incorrectly) return multiple Location headers
698 # (so probably same goes for URI). Use first header.
699 if "location" in headers:
700 newurl = headers["location"]
701 elif "uri" in headers:
702 newurl = headers["uri"]
703 else:
704 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000705
706 # fix a possible malformed URL
707 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700708
709 # For security reasons we don't allow redirection to anything other
710 # than http, https or ftp.
711
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800712 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800713 raise HTTPError(
714 newurl, code,
715 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
716 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700717
Martin Panterce6e0682016-05-16 01:07:13 +0000718 if not urlparts.path and urlparts.netloc:
Facundo Batistaf24802c2008-08-17 03:36:03 +0000719 urlparts = list(urlparts)
720 urlparts[2] = "/"
721 newurl = urlunparse(urlparts)
722
Martin Pantere6f06092016-05-16 01:14:20 +0000723 # http.client.parse_headers() decodes as ISO-8859-1. Recover the
724 # original bytes and percent-encode non-ASCII bytes, and any special
725 # characters such as the space.
726 newurl = quote(
727 newurl, encoding="iso-8859-1", safe=string.punctuation)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000728 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000729
730 # XXX Probably want to forget about the state of the current
731 # request, although that might interact poorly with other
732 # handlers that also use handler-specific request attributes
733 new = self.redirect_request(req, fp, code, msg, headers, newurl)
734 if new is None:
735 return
736
737 # loop detection
738 # .redirect_dict has a key url if url was previously visited.
739 if hasattr(req, 'redirect_dict'):
740 visited = new.redirect_dict = req.redirect_dict
741 if (visited.get(newurl, 0) >= self.max_repeats or
742 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000743 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000744 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000745 else:
746 visited = new.redirect_dict = req.redirect_dict = {}
747 visited[newurl] = visited.get(newurl, 0) + 1
748
749 # Don't close the fp until we are sure that we won't use it
750 # with HTTPError.
751 fp.read()
752 fp.close()
753
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000754 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000755
756 http_error_301 = http_error_303 = http_error_307 = http_error_302
757
758 inf_msg = "The HTTP server returned a redirect error that would " \
759 "lead to an infinite loop.\n" \
760 "The last 30x error message was:\n"
761
762
763def _parse_proxy(proxy):
764 """Return (scheme, user, password, host/port) given a URL or an authority.
765
766 If a URL is supplied, it must have an authority (host:port) component.
767 According to RFC 3986, having an authority component means the URL must
Senthil Kumarand8e24f12014-04-14 16:32:20 -0400768 have two slashes after the scheme.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000769 """
Cheryl Sabella0250de42018-04-25 16:51:54 -0700770 scheme, r_scheme = _splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000771 if not r_scheme.startswith("/"):
772 # authority
773 scheme = None
774 authority = proxy
775 else:
776 # URL
777 if not r_scheme.startswith("//"):
778 raise ValueError("proxy URL with no authority: %r" % proxy)
779 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
780 # and 3.3.), path is empty or starts with '/'
781 end = r_scheme.find("/", 2)
782 if end == -1:
783 end = None
784 authority = r_scheme[2:end]
Cheryl Sabella0250de42018-04-25 16:51:54 -0700785 userinfo, hostport = _splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000786 if userinfo is not None:
Cheryl Sabella0250de42018-04-25 16:51:54 -0700787 user, password = _splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000788 else:
789 user = password = None
790 return scheme, user, password, hostport
791
792class ProxyHandler(BaseHandler):
793 # Proxies must be in front
794 handler_order = 100
795
796 def __init__(self, proxies=None):
797 if proxies is None:
798 proxies = getproxies()
799 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
800 self.proxies = proxies
801 for type, url in proxies.items():
802 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200803 lambda r, proxy=url, type=type, meth=self.proxy_open:
804 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000805
806 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000807 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000808 proxy_type, user, password, hostport = _parse_proxy(proxy)
809 if proxy_type is None:
810 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000811
812 if req.host and proxy_bypass(req.host):
813 return None
814
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000815 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000816 user_pass = '%s:%s' % (unquote(user),
817 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000818 creds = base64.b64encode(user_pass.encode()).decode("ascii")
819 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000820 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000821 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000822 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000823 # let other handlers take care of it
824 return None
825 else:
826 # need to start over, because the other handlers don't
827 # grok the proxy's URL type
828 # e.g. if we have a constructor arg proxies like so:
829 # {'http': 'ftp://proxy.example.com'}, we may end up turning
830 # a request for http://acme.example.com/a into one for
831 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000832 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000833
834class HTTPPasswordMgr:
835
836 def __init__(self):
837 self.passwd = {}
838
839 def add_password(self, realm, uri, user, passwd):
840 # uri could be a single URI or a sequence
841 if isinstance(uri, str):
842 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800843 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000844 self.passwd[realm] = {}
845 for default_port in True, False:
846 reduced_uri = tuple(
Jon Dufresne39726282017-05-18 07:35:54 -0700847 self.reduce_uri(u, default_port) for u in uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000848 self.passwd[realm][reduced_uri] = (user, passwd)
849
850 def find_user_password(self, realm, authuri):
851 domains = self.passwd.get(realm, {})
852 for default_port in True, False:
853 reduced_authuri = self.reduce_uri(authuri, default_port)
854 for uris, authinfo in domains.items():
855 for uri in uris:
856 if self.is_suburi(uri, reduced_authuri):
857 return authinfo
858 return None, None
859
860 def reduce_uri(self, uri, default_port=True):
861 """Accept authority or URI and extract only the authority and path."""
862 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000863 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000864 if parts[1]:
865 # URI
866 scheme = parts[0]
867 authority = parts[1]
868 path = parts[2] or '/'
869 else:
870 # host or host:port
871 scheme = None
872 authority = uri
873 path = '/'
Cheryl Sabella0250de42018-04-25 16:51:54 -0700874 host, port = _splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000875 if default_port and port is None and scheme is not None:
876 dport = {"http": 80,
877 "https": 443,
878 }.get(scheme)
879 if dport is not None:
880 authority = "%s:%d" % (host, dport)
881 return authority, path
882
883 def is_suburi(self, base, test):
884 """Check if test is below base in a URI tree
885
886 Both args must be URIs in reduced form.
887 """
888 if base == test:
889 return True
890 if base[0] != test[0]:
891 return False
892 common = posixpath.commonprefix((base[1], test[1]))
893 if len(common) == len(base[1]):
894 return True
895 return False
896
897
898class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
899
900 def find_user_password(self, realm, authuri):
901 user, password = HTTPPasswordMgr.find_user_password(self, realm,
902 authuri)
903 if user is not None:
904 return user, password
905 return HTTPPasswordMgr.find_user_password(self, None, authuri)
906
907
R David Murray4c7f9952015-04-16 16:36:18 -0400908class HTTPPasswordMgrWithPriorAuth(HTTPPasswordMgrWithDefaultRealm):
909
910 def __init__(self, *args, **kwargs):
911 self.authenticated = {}
912 super().__init__(*args, **kwargs)
913
914 def add_password(self, realm, uri, user, passwd, is_authenticated=False):
915 self.update_authenticated(uri, is_authenticated)
916 # Add a default for prior auth requests
917 if realm is not None:
918 super().add_password(None, uri, user, passwd)
919 super().add_password(realm, uri, user, passwd)
920
921 def update_authenticated(self, uri, is_authenticated=False):
922 # uri could be a single URI or a sequence
923 if isinstance(uri, str):
924 uri = [uri]
925
926 for default_port in True, False:
927 for u in uri:
928 reduced_uri = self.reduce_uri(u, default_port)
929 self.authenticated[reduced_uri] = is_authenticated
930
931 def is_authenticated(self, authuri):
932 for default_port in True, False:
933 reduced_authuri = self.reduce_uri(authuri, default_port)
934 for uri in self.authenticated:
935 if self.is_suburi(uri, reduced_authuri):
936 return self.authenticated[uri]
937
938
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000939class AbstractBasicAuthHandler:
940
941 # XXX this allows for multiple auth-schemes, but will stupidly pick
942 # the last one with a realm specified.
943
944 # allow for double- and single-quoted realm values
945 # (single quotes are a violation of the RFC, but appear in the wild)
946 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800947 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000948
949 # XXX could pre-emptively send auth info already accepted (RFC 2617,
950 # end of section 2, and section 1.2 immediately after "credentials"
951 # production).
952
953 def __init__(self, password_mgr=None):
954 if password_mgr is None:
955 password_mgr = HTTPPasswordMgr()
956 self.passwd = password_mgr
957 self.add_password = self.passwd.add_password
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000958
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000959 def http_error_auth_reqed(self, authreq, host, req, headers):
960 # host may be an authority (without userinfo) or a URL with an
961 # authority
962 # XXX could be multiple headers
963 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000964
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000965 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800966 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800967 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800968 raise ValueError("AbstractBasicAuthHandler does not"
969 " support the following scheme: '%s'" %
970 scheme)
971 else:
972 mo = AbstractBasicAuthHandler.rx.search(authreq)
973 if mo:
974 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800975 if quote not in ['"',"'"]:
976 warnings.warn("Basic Auth Realm was unquoted",
977 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800978 if scheme.lower() == 'basic':
Senthil Kumaran78373762014-08-20 07:53:58 +0530979 return self.retry_http_basic_auth(host, req, realm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000980
981 def retry_http_basic_auth(self, host, req, realm):
982 user, pw = self.passwd.find_user_password(realm, host)
983 if pw is not None:
984 raw = "%s:%s" % (user, pw)
985 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
Senthil Kumaran78373762014-08-20 07:53:58 +0530986 if req.get_header(self.auth_header, None) == auth:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000987 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000988 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000989 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000990 else:
991 return None
992
R David Murray4c7f9952015-04-16 16:36:18 -0400993 def http_request(self, req):
994 if (not hasattr(self.passwd, 'is_authenticated') or
995 not self.passwd.is_authenticated(req.full_url)):
996 return req
997
998 if not req.has_header('Authorization'):
999 user, passwd = self.passwd.find_user_password(None, req.full_url)
1000 credentials = '{0}:{1}'.format(user, passwd).encode()
1001 auth_str = base64.standard_b64encode(credentials).decode()
1002 req.add_unredirected_header('Authorization',
1003 'Basic {}'.format(auth_str.strip()))
1004 return req
1005
1006 def http_response(self, req, response):
1007 if hasattr(self.passwd, 'is_authenticated'):
1008 if 200 <= response.code < 300:
1009 self.passwd.update_authenticated(req.full_url, True)
1010 else:
1011 self.passwd.update_authenticated(req.full_url, False)
1012 return response
1013
1014 https_request = http_request
1015 https_response = http_response
1016
1017
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001018
1019class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
1020
1021 auth_header = 'Authorization'
1022
1023 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001024 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001025 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001026 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001027 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001028
1029
1030class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
1031
1032 auth_header = 'Proxy-authorization'
1033
1034 def http_error_407(self, req, fp, code, msg, headers):
1035 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +00001036 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001037 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
1038 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001039 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001040 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001041 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001042 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001043
1044
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001045# Return n random bytes.
1046_randombytes = os.urandom
1047
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001048
1049class AbstractDigestAuthHandler:
1050 # Digest authentication is specified in RFC 2617.
1051
1052 # XXX The client does not inspect the Authentication-Info header
1053 # in a successful response.
1054
1055 # XXX It should be possible to test this implementation against
1056 # a mock server that just generates a static set of challenges.
1057
1058 # XXX qop="auth-int" supports is shaky
1059
1060 def __init__(self, passwd=None):
1061 if passwd is None:
1062 passwd = HTTPPasswordMgr()
1063 self.passwd = passwd
1064 self.add_password = self.passwd.add_password
1065 self.retried = 0
1066 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001067 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001068
1069 def reset_retry_count(self):
1070 self.retried = 0
1071
1072 def http_error_auth_reqed(self, auth_header, host, req, headers):
1073 authreq = headers.get(auth_header, None)
1074 if self.retried > 5:
1075 # Don't fail endlessly - if we failed once, we'll probably
1076 # fail a second time. Hm. Unless the Password Manager is
1077 # prompting for the information. Crap. This isn't great
1078 # but it's better than the current 'repeat until recursion
1079 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001080 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +00001081 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001082 else:
1083 self.retried += 1
1084 if authreq:
1085 scheme = authreq.split()[0]
1086 if scheme.lower() == 'digest':
1087 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +08001088 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +08001089 raise ValueError("AbstractDigestAuthHandler does not support"
1090 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001091
1092 def retry_http_digest_auth(self, req, auth):
1093 token, challenge = auth.split(' ', 1)
1094 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
1095 auth = self.get_authorization(req, chal)
1096 if auth:
1097 auth_val = 'Digest %s' % auth
1098 if req.headers.get(self.auth_header, None) == auth_val:
1099 return None
1100 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +00001101 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001102 return resp
1103
1104 def get_cnonce(self, nonce):
1105 # The cnonce-value is an opaque
1106 # quoted string value provided by the client and used by both client
1107 # and server to avoid chosen plaintext attacks, to provide mutual
1108 # authentication, and to provide some message integrity protection.
1109 # This isn't a fabulous effort, but it's probably Good Enough.
1110 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001111 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001112 dig = hashlib.sha1(b).hexdigest()
1113 return dig[:16]
1114
1115 def get_authorization(self, req, chal):
1116 try:
1117 realm = chal['realm']
1118 nonce = chal['nonce']
1119 qop = chal.get('qop')
1120 algorithm = chal.get('algorithm', 'MD5')
1121 # mod_digest doesn't send an opaque, even though it isn't
1122 # supposed to be optional
1123 opaque = chal.get('opaque', None)
1124 except KeyError:
1125 return None
1126
1127 H, KD = self.get_algorithm_impls(algorithm)
1128 if H is None:
1129 return None
1130
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001131 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001132 if user is None:
1133 return None
1134
1135 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001136 if req.data is not None:
1137 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001138 else:
1139 entdig = None
1140
1141 A1 = "%s:%s:%s" % (user, realm, pw)
1142 A2 = "%s:%s" % (req.get_method(),
1143 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001144 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001145 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001146 if nonce == self.last_nonce:
1147 self.nonce_count += 1
1148 else:
1149 self.nonce_count = 1
1150 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001151 ncvalue = '%08x' % self.nonce_count
1152 cnonce = self.get_cnonce(nonce)
1153 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1154 respdig = KD(H(A1), noncebit)
1155 elif qop is None:
1156 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1157 else:
1158 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001159 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001160
1161 # XXX should the partial digests be encoded too?
1162
1163 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001164 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001165 respdig)
1166 if opaque:
1167 base += ', opaque="%s"' % opaque
1168 if entdig:
1169 base += ', digest="%s"' % entdig
1170 base += ', algorithm="%s"' % algorithm
1171 if qop:
1172 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1173 return base
1174
1175 def get_algorithm_impls(self, algorithm):
1176 # lambdas assume digest modules are imported at the top level
1177 if algorithm == 'MD5':
1178 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1179 elif algorithm == 'SHA':
1180 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1181 # XXX MD5-sess
Berker Peksage88dd1c2016-03-06 16:16:40 +02001182 else:
1183 raise ValueError("Unsupported digest authentication "
1184 "algorithm %r" % algorithm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001185 KD = lambda s, d: H("%s:%s" % (s, d))
1186 return H, KD
1187
1188 def get_entity_digest(self, data, chal):
1189 # XXX not implemented yet
1190 return None
1191
1192
1193class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1194 """An authentication protocol defined by RFC 2069
1195
1196 Digest authentication improves on basic authentication because it
1197 does not transmit passwords in the clear.
1198 """
1199
1200 auth_header = 'Authorization'
1201 handler_order = 490 # before Basic auth
1202
1203 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001204 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205 retry = self.http_error_auth_reqed('www-authenticate',
1206 host, req, headers)
1207 self.reset_retry_count()
1208 return retry
1209
1210
1211class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1212
1213 auth_header = 'Proxy-Authorization'
1214 handler_order = 490 # before Basic auth
1215
1216 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001217 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001218 retry = self.http_error_auth_reqed('proxy-authenticate',
1219 host, req, headers)
1220 self.reset_retry_count()
1221 return retry
1222
1223class AbstractHTTPHandler(BaseHandler):
1224
1225 def __init__(self, debuglevel=0):
1226 self._debuglevel = debuglevel
1227
1228 def set_http_debuglevel(self, level):
1229 self._debuglevel = level
1230
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001231 def _get_content_length(self, request):
1232 return http.client.HTTPConnection._get_content_length(
1233 request.data,
1234 request.get_method())
1235
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001236 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001237 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001238 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001239 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001240
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001241 if request.data is not None: # POST
1242 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001243 if isinstance(data, str):
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001244 msg = "POST data should be bytes, an iterable of bytes, " \
1245 "or a file object. It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001246 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001247 if not request.has_header('Content-type'):
1248 request.add_unredirected_header(
1249 'Content-type',
1250 'application/x-www-form-urlencoded')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001251 if (not request.has_header('Content-length')
1252 and not request.has_header('Transfer-encoding')):
1253 content_length = self._get_content_length(request)
1254 if content_length is not None:
1255 request.add_unredirected_header(
1256 'Content-length', str(content_length))
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001257 else:
1258 request.add_unredirected_header(
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001259 'Transfer-encoding', 'chunked')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001260
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001261 sel_host = host
1262 if request.has_proxy():
Cheryl Sabella0250de42018-04-25 16:51:54 -07001263 scheme, sel = _splittype(request.selector)
1264 sel_host, sel_path = _splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001265 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001266 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001267 for name, value in self.parent.addheaders:
1268 name = name.capitalize()
1269 if not request.has_header(name):
1270 request.add_unredirected_header(name, value)
1271
1272 return request
1273
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001274 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001275 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001276
1277 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001278 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001279 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001280 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001281 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001282
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001283 # will parse host:port
1284 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran9642eed2016-05-13 01:32:42 -07001285 h.set_debuglevel(self._debuglevel)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001286
1287 headers = dict(req.unredirected_hdrs)
Serhiy Storchaka3f2e6f12018-02-26 16:50:11 +02001288 headers.update({k: v for k, v in req.headers.items()
1289 if k not in headers})
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001290
1291 # TODO(jhylton): Should this be redesigned to handle
1292 # persistent connections?
1293
1294 # We want to make an HTTP/1.1 request, but the addinfourl
1295 # class isn't prepared to deal with a persistent connection.
1296 # It will try to read all remaining data from the socket,
1297 # which will block while the server waits for the next request.
1298 # So make sure the connection gets closed after the (only)
1299 # request.
1300 headers["Connection"] = "close"
Jon Dufresne39726282017-05-18 07:35:54 -07001301 headers = {name.title(): val for name, val in headers.items()}
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001302
1303 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001304 tunnel_headers = {}
1305 proxy_auth_hdr = "Proxy-Authorization"
1306 if proxy_auth_hdr in headers:
1307 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1308 # Proxy-Authorization should not be sent to origin
1309 # server.
1310 del headers[proxy_auth_hdr]
1311 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001312
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001313 try:
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001314 try:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001315 h.request(req.get_method(), req.selector, req.data, headers,
1316 encode_chunked=req.has_header('Transfer-encoding'))
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001317 except OSError as err: # timeout error
1318 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001319 r = h.getresponse()
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001320 except:
1321 h.close()
1322 raise
1323
1324 # If the server does not send us a 'Connection: close' header,
1325 # HTTPConnection assumes the socket should be left open. Manually
1326 # mark the socket to be closed when this response object goes away.
1327 if h.sock:
1328 h.sock.close()
1329 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001330
Senthil Kumaran26430412011-04-13 07:01:19 +08001331 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001332 # This line replaces the .msg attribute of the HTTPResponse
1333 # with .headers, because urllib clients expect the response to
1334 # have the reason in .msg. It would be good to mark this
1335 # attribute is deprecated and get then to use info() or
1336 # .headers.
1337 r.msg = r.reason
1338 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001339
1340
1341class HTTPHandler(AbstractHTTPHandler):
1342
1343 def http_open(self, req):
1344 return self.do_open(http.client.HTTPConnection, req)
1345
1346 http_request = AbstractHTTPHandler.do_request_
1347
1348if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001349
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001350 class HTTPSHandler(AbstractHTTPHandler):
1351
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001352 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1353 AbstractHTTPHandler.__init__(self, debuglevel)
1354 self._context = context
1355 self._check_hostname = check_hostname
1356
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001357 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001358 return self.do_open(http.client.HTTPSConnection, req,
1359 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001360
1361 https_request = AbstractHTTPHandler.do_request_
1362
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001363 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001364
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001365class HTTPCookieProcessor(BaseHandler):
1366 def __init__(self, cookiejar=None):
1367 import http.cookiejar
1368 if cookiejar is None:
1369 cookiejar = http.cookiejar.CookieJar()
1370 self.cookiejar = cookiejar
1371
1372 def http_request(self, request):
1373 self.cookiejar.add_cookie_header(request)
1374 return request
1375
1376 def http_response(self, request, response):
1377 self.cookiejar.extract_cookies(response, request)
1378 return response
1379
1380 https_request = http_request
1381 https_response = http_response
1382
1383class UnknownHandler(BaseHandler):
1384 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001385 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001386 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001387
1388def parse_keqv_list(l):
1389 """Parse list of key=value strings where keys are not duplicated."""
1390 parsed = {}
1391 for elt in l:
1392 k, v = elt.split('=', 1)
1393 if v[0] == '"' and v[-1] == '"':
1394 v = v[1:-1]
1395 parsed[k] = v
1396 return parsed
1397
1398def parse_http_list(s):
1399 """Parse lists as described by RFC 2068 Section 2.
1400
1401 In particular, parse comma-separated lists where the elements of
1402 the list may include quoted-strings. A quoted-string could
1403 contain a comma. A non-quoted string could have quotes in the
1404 middle. Neither commas nor quotes count if they are escaped.
1405 Only double-quotes count, not single-quotes.
1406 """
1407 res = []
1408 part = ''
1409
1410 escape = quote = False
1411 for cur in s:
1412 if escape:
1413 part += cur
1414 escape = False
1415 continue
1416 if quote:
1417 if cur == '\\':
1418 escape = True
1419 continue
1420 elif cur == '"':
1421 quote = False
1422 part += cur
1423 continue
1424
1425 if cur == ',':
1426 res.append(part)
1427 part = ''
1428 continue
1429
1430 if cur == '"':
1431 quote = True
1432
1433 part += cur
1434
1435 # append last part
1436 if part:
1437 res.append(part)
1438
1439 return [part.strip() for part in res]
1440
1441class FileHandler(BaseHandler):
1442 # Use local file or FTP depending on form of URL
1443 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001444 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001445 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1446 req.host != 'localhost'):
Senthil Kumaranbc07ac52014-07-22 00:15:20 -07001447 if not req.host in self.get_names():
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001448 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001449 else:
1450 return self.open_local_file(req)
1451
1452 # names for the localhost
1453 names = None
1454 def get_names(self):
1455 if FileHandler.names is None:
1456 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001457 FileHandler.names = tuple(
1458 socket.gethostbyname_ex('localhost')[2] +
1459 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001460 except socket.gaierror:
1461 FileHandler.names = (socket.gethostbyname('localhost'),)
1462 return FileHandler.names
1463
1464 # not entirely sure what the rules are here
1465 def open_local_file(self, req):
1466 import email.utils
1467 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001468 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001469 filename = req.selector
1470 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001471 try:
1472 stats = os.stat(localfile)
1473 size = stats.st_size
1474 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001475 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001476 headers = email.message_from_string(
1477 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1478 (mtype or 'text/plain', size, modified))
1479 if host:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001480 host, port = _splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001481 if not host or \
1482 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001483 if host:
1484 origurl = 'file://' + host + filename
1485 else:
1486 origurl = 'file://' + filename
1487 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001488 except OSError as exp:
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001489 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001490 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001491
1492def _safe_gethostbyname(host):
1493 try:
1494 return socket.gethostbyname(host)
1495 except socket.gaierror:
1496 return None
1497
1498class FTPHandler(BaseHandler):
1499 def ftp_open(self, req):
1500 import ftplib
1501 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001502 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001503 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001504 raise URLError('ftp error: no host given')
Cheryl Sabella0250de42018-04-25 16:51:54 -07001505 host, port = _splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001506 if port is None:
1507 port = ftplib.FTP_PORT
1508 else:
1509 port = int(port)
1510
1511 # username/password handling
Cheryl Sabella0250de42018-04-25 16:51:54 -07001512 user, host = _splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001513 if user:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001514 user, passwd = _splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001515 else:
1516 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001517 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001518 user = user or ''
1519 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001520
1521 try:
1522 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001523 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001524 raise URLError(msg)
Cheryl Sabella0250de42018-04-25 16:51:54 -07001525 path, attrs = _splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001526 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001527 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001528 dirs, file = dirs[:-1], dirs[-1]
1529 if dirs and not dirs[0]:
1530 dirs = dirs[1:]
1531 try:
1532 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1533 type = file and 'I' or 'D'
1534 for attr in attrs:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001535 attr, value = _splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001536 if attr.lower() == 'type' and \
1537 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1538 type = value.upper()
1539 fp, retrlen = fw.retrfile(file, type)
1540 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001541 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001542 if mtype:
1543 headers += "Content-type: %s\n" % mtype
1544 if retrlen is not None and retrlen >= 0:
1545 headers += "Content-length: %d\n" % retrlen
1546 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001547 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001548 except ftplib.all_errors as exp:
1549 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001550 raise exc.with_traceback(sys.exc_info()[2])
1551
1552 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001553 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1554 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001555
1556class CacheFTPHandler(FTPHandler):
1557 # XXX would be nice to have pluggable cache strategies
1558 # XXX this stuff is definitely not thread safe
1559 def __init__(self):
1560 self.cache = {}
1561 self.timeout = {}
1562 self.soonest = 0
1563 self.delay = 60
1564 self.max_conns = 16
1565
1566 def setTimeout(self, t):
1567 self.delay = t
1568
1569 def setMaxConns(self, m):
1570 self.max_conns = m
1571
1572 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1573 key = user, host, port, '/'.join(dirs), timeout
1574 if key in self.cache:
1575 self.timeout[key] = time.time() + self.delay
1576 else:
1577 self.cache[key] = ftpwrapper(user, passwd, host, port,
1578 dirs, timeout)
1579 self.timeout[key] = time.time() + self.delay
1580 self.check_cache()
1581 return self.cache[key]
1582
1583 def check_cache(self):
1584 # first check for old ones
1585 t = time.time()
1586 if self.soonest <= t:
1587 for k, v in list(self.timeout.items()):
1588 if v < t:
1589 self.cache[k].close()
1590 del self.cache[k]
1591 del self.timeout[k]
1592 self.soonest = min(list(self.timeout.values()))
1593
1594 # then check the size
1595 if len(self.cache) == self.max_conns:
1596 for k, v in list(self.timeout.items()):
1597 if v == self.soonest:
1598 del self.cache[k]
1599 del self.timeout[k]
1600 break
1601 self.soonest = min(list(self.timeout.values()))
1602
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001603 def clear_cache(self):
1604 for conn in self.cache.values():
1605 conn.close()
1606 self.cache.clear()
1607 self.timeout.clear()
1608
Antoine Pitroudf204be2012-11-24 17:59:08 +01001609class DataHandler(BaseHandler):
1610 def data_open(self, req):
1611 # data URLs as specified in RFC 2397.
1612 #
1613 # ignores POSTed data
1614 #
1615 # syntax:
1616 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1617 # mediatype := [ type "/" subtype ] *( ";" parameter )
1618 # data := *urlchar
1619 # parameter := attribute "=" value
1620 url = req.full_url
1621
1622 scheme, data = url.split(":",1)
1623 mediatype, data = data.split(",",1)
1624
1625 # even base64 encoded data URLs might be quoted so unquote in any case:
1626 data = unquote_to_bytes(data)
1627 if mediatype.endswith(";base64"):
1628 data = base64.decodebytes(data)
1629 mediatype = mediatype[:-7]
1630
1631 if not mediatype:
1632 mediatype = "text/plain;charset=US-ASCII"
1633
1634 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1635 (mediatype, len(data)))
1636
1637 return addinfourl(io.BytesIO(data), headers, url)
1638
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001639
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001640# Code move from the old urllib module
1641
1642MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1643
1644# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001645if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001646 from nturl2path import url2pathname, pathname2url
1647else:
1648 def url2pathname(pathname):
1649 """OS-specific conversion from a relative URL of the 'file' scheme
1650 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001651 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001652
1653 def pathname2url(pathname):
1654 """OS-specific conversion from a file system path to a relative URL
1655 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001656 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001657
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001658
1659ftpcache = {}
Senthil Kumarana2a9ddd2017-04-08 23:27:25 -07001660
1661
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001662class URLopener:
1663 """Class to open URLs.
1664 This is a class rather than just a subroutine because we may need
1665 more than one set of global protocol-specific options.
1666 Note -- this is a base class for those who don't want the
1667 automatic handling of errors type 302 (relocated) and 401
1668 (authorization needed)."""
1669
1670 __tempfiles = None
1671
1672 version = "Python-urllib/%s" % __version__
1673
1674 # Constructor
1675 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001676 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001677 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1678 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001679 if proxies is None:
1680 proxies = getproxies()
1681 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1682 self.proxies = proxies
1683 self.key_file = x509.get('key_file')
1684 self.cert_file = x509.get('cert_file')
Raymond Hettingerb7f3c942016-09-09 16:44:53 -07001685 self.addheaders = [('User-Agent', self.version), ('Accept', '*/*')]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001686 self.__tempfiles = []
1687 self.__unlink = os.unlink # See cleanup()
1688 self.tempcache = None
1689 # Undocumented feature: if you assign {} to tempcache,
1690 # it is used to cache files retrieved with
1691 # self.retrieve(). This is not enabled by default
1692 # since it does not work for changing documents (and I
1693 # haven't got the logic to check expiration headers
1694 # yet).
1695 self.ftpcache = ftpcache
1696 # Undocumented feature: you can use a different
1697 # ftp cache by assigning to the .ftpcache member;
1698 # in case you want logically independent URL openers
1699 # XXX This is not threadsafe. Bah.
1700
1701 def __del__(self):
1702 self.close()
1703
1704 def close(self):
1705 self.cleanup()
1706
1707 def cleanup(self):
1708 # This code sometimes runs when the rest of this module
1709 # has already been deleted, so it can't use any globals
1710 # or import anything.
1711 if self.__tempfiles:
1712 for file in self.__tempfiles:
1713 try:
1714 self.__unlink(file)
1715 except OSError:
1716 pass
1717 del self.__tempfiles[:]
1718 if self.tempcache:
1719 self.tempcache.clear()
1720
1721 def addheader(self, *args):
1722 """Add a header to be used by the HTTP interface only
1723 e.g. u.addheader('Accept', 'sound/basic')"""
1724 self.addheaders.append(args)
1725
1726 # External interface
1727 def open(self, fullurl, data=None):
1728 """Use URLopener().open(file) instead of open(file, 'r')."""
Cheryl Sabella0250de42018-04-25 16:51:54 -07001729 fullurl = _unwrap(_to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001730 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001731 if self.tempcache and fullurl in self.tempcache:
1732 filename, headers = self.tempcache[fullurl]
1733 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001734 return addinfourl(fp, headers, fullurl)
Cheryl Sabella0250de42018-04-25 16:51:54 -07001735 urltype, url = _splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001736 if not urltype:
1737 urltype = 'file'
1738 if urltype in self.proxies:
1739 proxy = self.proxies[urltype]
Cheryl Sabella0250de42018-04-25 16:51:54 -07001740 urltype, proxyhost = _splittype(proxy)
1741 host, selector = _splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001742 url = (host, fullurl) # Signal special case to open_*()
1743 else:
1744 proxy = None
1745 name = 'open_' + urltype
1746 self.type = urltype
1747 name = name.replace('-', '_')
1748 if not hasattr(self, name):
1749 if proxy:
1750 return self.open_unknown_proxy(proxy, fullurl, data)
1751 else:
1752 return self.open_unknown(fullurl, data)
1753 try:
1754 if data is None:
1755 return getattr(self, name)(url)
1756 else:
1757 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001758 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001759 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001760 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001761 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001762
1763 def open_unknown(self, fullurl, data=None):
1764 """Overridable interface to open unknown URL type."""
Cheryl Sabella0250de42018-04-25 16:51:54 -07001765 type, url = _splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001766 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001767
1768 def open_unknown_proxy(self, proxy, fullurl, data=None):
1769 """Overridable interface to open unknown URL type."""
Cheryl Sabella0250de42018-04-25 16:51:54 -07001770 type, url = _splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001771 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001772
1773 # External interface
1774 def retrieve(self, url, filename=None, reporthook=None, data=None):
1775 """retrieve(url) returns (filename, headers) for a local object
1776 or (tempfilename, headers) for a remote object."""
Cheryl Sabella0250de42018-04-25 16:51:54 -07001777 url = _unwrap(_to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001778 if self.tempcache and url in self.tempcache:
1779 return self.tempcache[url]
Cheryl Sabella0250de42018-04-25 16:51:54 -07001780 type, url1 = _splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001781 if filename is None and (not type or type == 'file'):
1782 try:
1783 fp = self.open_local_file(url1)
1784 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001785 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001786 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001787 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001788 pass
1789 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001790 try:
1791 headers = fp.info()
1792 if filename:
1793 tfp = open(filename, 'wb')
1794 else:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001795 garbage, path = splittype(url)
1796 garbage, path = splithost(path or "")
1797 path, garbage = splitquery(path or "")
1798 path, garbage = splitattr(path or "")
1799 suffix = os.path.splitext(path)[1]
1800 (fd, filename) = tempfile.mkstemp(suffix)
1801 self.__tempfiles.append(filename)
1802 tfp = os.fdopen(fd, 'wb')
1803 try:
1804 result = filename, headers
1805 if self.tempcache is not None:
1806 self.tempcache[url] = result
1807 bs = 1024*8
1808 size = -1
1809 read = 0
1810 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001811 if "content-length" in headers:
1812 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001813 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001814 reporthook(blocknum, bs, size)
1815 while 1:
1816 block = fp.read(bs)
1817 if not block:
1818 break
1819 read += len(block)
1820 tfp.write(block)
1821 blocknum += 1
1822 if reporthook:
1823 reporthook(blocknum, bs, size)
1824 finally:
1825 tfp.close()
1826 finally:
1827 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001828
1829 # raise exception if actual size does not match content-length header
1830 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001831 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001832 "retrieval incomplete: got only %i out of %i bytes"
1833 % (read, size), result)
1834
1835 return result
1836
1837 # Each method named open_<type> knows how to open that type of URL
1838
1839 def _open_generic_http(self, connection_factory, url, data):
1840 """Make an HTTP connection using connection_class.
1841
1842 This is an internal method that should be called from
1843 open_http() or open_https().
1844
1845 Arguments:
1846 - connection_factory should take a host name and return an
1847 HTTPConnection instance.
1848 - url is the url to retrieval or a host, relative-path pair.
1849 - data is payload for a POST request or None.
1850 """
1851
1852 user_passwd = None
1853 proxy_passwd= None
1854 if isinstance(url, str):
Cheryl Sabella0250de42018-04-25 16:51:54 -07001855 host, selector = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001856 if host:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001857 user_passwd, host = _splituser(host)
Georg Brandl13e89462008-07-01 19:56:00 +00001858 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001859 realhost = host
1860 else:
1861 host, selector = url
1862 # check whether the proxy contains authorization information
Cheryl Sabella0250de42018-04-25 16:51:54 -07001863 proxy_passwd, host = _splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001864 # now we proceed with the url we want to obtain
Cheryl Sabella0250de42018-04-25 16:51:54 -07001865 urltype, rest = _splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001866 url = rest
1867 user_passwd = None
1868 if urltype.lower() != 'http':
1869 realhost = None
1870 else:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001871 realhost, rest = _splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001872 if realhost:
Cheryl Sabella0250de42018-04-25 16:51:54 -07001873 user_passwd, realhost = _splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001874 if user_passwd:
1875 selector = "%s://%s%s" % (urltype, realhost, rest)
1876 if proxy_bypass(realhost):
1877 host = realhost
1878
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001879 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001880
1881 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001882 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001883 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001884 else:
1885 proxy_auth = None
1886
1887 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001888 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001889 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001890 else:
1891 auth = None
1892 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001893 headers = {}
1894 if proxy_auth:
1895 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1896 if auth:
1897 headers["Authorization"] = "Basic %s" % auth
1898 if realhost:
1899 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001900
1901 # Add Connection:close as we don't support persistent connections yet.
1902 # This helps in closing the socket and avoiding ResourceWarning
1903
1904 headers["Connection"] = "close"
1905
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001906 for header, value in self.addheaders:
1907 headers[header] = value
1908
1909 if data is not None:
1910 headers["Content-Type"] = "application/x-www-form-urlencoded"
1911 http_conn.request("POST", selector, data, headers)
1912 else:
1913 http_conn.request("GET", selector, headers=headers)
1914
1915 try:
1916 response = http_conn.getresponse()
1917 except http.client.BadStatusLine:
1918 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001919 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001920
1921 # According to RFC 2616, "2xx" code indicates that the client's
1922 # request was successfully received, understood, and accepted.
1923 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001924 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001925 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001926 else:
1927 return self.http_error(
1928 url, response.fp,
1929 response.status, response.reason, response.msg, data)
1930
1931 def open_http(self, url, data=None):
1932 """Use HTTP protocol."""
1933 return self._open_generic_http(http.client.HTTPConnection, url, data)
1934
1935 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1936 """Handle http errors.
1937
1938 Derived class can override this, or provide specific handlers
1939 named http_error_DDD where DDD is the 3-digit error code."""
1940 # First check if there's a specific handler for this error
1941 name = 'http_error_%d' % errcode
1942 if hasattr(self, name):
1943 method = getattr(self, name)
1944 if data is None:
1945 result = method(url, fp, errcode, errmsg, headers)
1946 else:
1947 result = method(url, fp, errcode, errmsg, headers, data)
1948 if result: return result
1949 return self.http_error_default(url, fp, errcode, errmsg, headers)
1950
1951 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001952 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001953 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001954 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001955
1956 if _have_ssl:
1957 def _https_connection(self, host):
1958 return http.client.HTTPSConnection(host,
1959 key_file=self.key_file,
1960 cert_file=self.cert_file)
1961
1962 def open_https(self, url, data=None):
1963 """Use HTTPS protocol."""
1964 return self._open_generic_http(self._https_connection, url, data)
1965
1966 def open_file(self, url):
1967 """Use local file or FTP depending on form of URL."""
1968 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001969 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001970 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001971 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001972 else:
1973 return self.open_local_file(url)
1974
1975 def open_local_file(self, url):
1976 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001977 import email.utils
1978 import mimetypes
Cheryl Sabella0250de42018-04-25 16:51:54 -07001979 host, file = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001980 localname = url2pathname(file)
1981 try:
1982 stats = os.stat(localname)
1983 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001984 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001985 size = stats.st_size
1986 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1987 mtype = mimetypes.guess_type(url)[0]
1988 headers = email.message_from_string(
1989 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1990 (mtype or 'text/plain', size, modified))
1991 if not host:
1992 urlfile = file
1993 if file[:1] == '/':
1994 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001995 return addinfourl(open(localname, 'rb'), headers, urlfile)
Cheryl Sabella0250de42018-04-25 16:51:54 -07001996 host, port = _splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001997 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07001998 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001999 urlfile = file
2000 if file[:1] == '/':
2001 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08002002 elif file[:2] == './':
2003 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00002004 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002005 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002006
2007 def open_ftp(self, url):
2008 """Use FTP protocol."""
2009 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002010 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002011 import mimetypes
Cheryl Sabella0250de42018-04-25 16:51:54 -07002012 host, path = _splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002013 if not host: raise URLError('ftp error: no host given')
Cheryl Sabella0250de42018-04-25 16:51:54 -07002014 host, port = _splitport(host)
2015 user, host = _splituser(host)
2016 if user: user, passwd = _splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002017 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00002018 host = unquote(host)
2019 user = unquote(user or '')
2020 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002021 host = socket.gethostbyname(host)
2022 if not port:
2023 import ftplib
2024 port = ftplib.FTP_PORT
2025 else:
2026 port = int(port)
Cheryl Sabella0250de42018-04-25 16:51:54 -07002027 path, attrs = _splitattr(path)
Georg Brandl13e89462008-07-01 19:56:00 +00002028 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002029 dirs = path.split('/')
2030 dirs, file = dirs[:-1], dirs[-1]
2031 if dirs and not dirs[0]: dirs = dirs[1:]
2032 if dirs and not dirs[0]: dirs[0] = '/'
2033 key = user, host, port, '/'.join(dirs)
2034 # XXX thread unsafe!
2035 if len(self.ftpcache) > MAXFTPCACHE:
2036 # Prune the cache, rather arbitrarily
Benjamin Peterson3c2dca62014-06-07 15:08:04 -07002037 for k in list(self.ftpcache):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002038 if k != key:
2039 v = self.ftpcache[k]
2040 del self.ftpcache[k]
2041 v.close()
2042 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002043 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002044 self.ftpcache[key] = \
2045 ftpwrapper(user, passwd, host, port, dirs)
2046 if not file: type = 'D'
2047 else: type = 'I'
2048 for attr in attrs:
Cheryl Sabella0250de42018-04-25 16:51:54 -07002049 attr, value = _splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002050 if attr.lower() == 'type' and \
2051 value in ('a', 'A', 'i', 'I', 'd', 'D'):
2052 type = value.upper()
2053 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
2054 mtype = mimetypes.guess_type("ftp:" + url)[0]
2055 headers = ""
2056 if mtype:
2057 headers += "Content-Type: %s\n" % mtype
2058 if retrlen is not None and retrlen >= 0:
2059 headers += "Content-Length: %d\n" % retrlen
2060 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00002061 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002062 except ftperrors() as exp:
2063 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002064
2065 def open_data(self, url, data=None):
2066 """Use "data" URL."""
2067 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002068 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002069 # ignore POSTed data
2070 #
2071 # syntax of data URLs:
2072 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
2073 # mediatype := [ type "/" subtype ] *( ";" parameter )
2074 # data := *urlchar
2075 # parameter := attribute "=" value
2076 try:
2077 [type, data] = url.split(',', 1)
2078 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002079 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002080 if not type:
2081 type = 'text/plain;charset=US-ASCII'
2082 semi = type.rfind(';')
2083 if semi >= 0 and '=' not in type[semi:]:
2084 encoding = type[semi+1:]
2085 type = type[:semi]
2086 else:
2087 encoding = ''
2088 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00002089 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002090 time.gmtime(time.time())))
2091 msg.append('Content-type: %s' % type)
2092 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00002093 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00002094 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002095 else:
Georg Brandl13e89462008-07-01 19:56:00 +00002096 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002097 msg.append('Content-Length: %d' % len(data))
2098 msg.append('')
2099 msg.append(data)
2100 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00002101 headers = email.message_from_string(msg)
2102 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002103 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00002104 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002105
2106
2107class FancyURLopener(URLopener):
2108 """Derived class with handlers for errors we can handle (perhaps)."""
2109
2110 def __init__(self, *args, **kwargs):
2111 URLopener.__init__(self, *args, **kwargs)
2112 self.auth_cache = {}
2113 self.tries = 0
2114 self.maxtries = 10
2115
2116 def http_error_default(self, url, fp, errcode, errmsg, headers):
2117 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00002118 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002119
2120 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
2121 """Error 302 -- relocated (temporarily)."""
2122 self.tries += 1
Martin Pantera0370222016-02-04 06:01:35 +00002123 try:
2124 if self.maxtries and self.tries >= self.maxtries:
2125 if hasattr(self, "http_error_500"):
2126 meth = self.http_error_500
2127 else:
2128 meth = self.http_error_default
2129 return meth(url, fp, 500,
2130 "Internal Server Error: Redirect Recursion",
2131 headers)
2132 result = self.redirect_internal(url, fp, errcode, errmsg,
2133 headers, data)
2134 return result
2135 finally:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002136 self.tries = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002137
2138 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
2139 if 'location' in headers:
2140 newurl = headers['location']
2141 elif 'uri' in headers:
2142 newurl = headers['uri']
2143 else:
2144 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002145 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002146
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002147 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002148 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002149
2150 urlparts = urlparse(newurl)
2151
2152 # For security reasons, we don't allow redirection to anything other
2153 # than http, https and ftp.
2154
2155 # We are using newer HTTPError with older redirect_internal method
2156 # This older method will get deprecated in 3.3
2157
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002158 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002159 raise HTTPError(newurl, errcode,
2160 errmsg +
2161 " Redirection to url '%s' is not allowed." % newurl,
2162 headers, fp)
2163
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002164 return self.open(newurl)
2165
2166 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2167 """Error 301 -- also relocated (permanently)."""
2168 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2169
2170 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2171 """Error 303 -- also relocated (essentially identical to 302)."""
2172 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2173
2174 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2175 """Error 307 -- relocated, but turn POST into error."""
2176 if data is None:
2177 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2178 else:
2179 return self.http_error_default(url, fp, errcode, errmsg, headers)
2180
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002181 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2182 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002183 """Error 401 -- authentication required.
2184 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002185 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002186 URLopener.http_error_default(self, url, fp,
2187 errcode, errmsg, headers)
2188 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002189 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2190 if not match:
2191 URLopener.http_error_default(self, url, fp,
2192 errcode, errmsg, headers)
2193 scheme, realm = match.groups()
2194 if scheme.lower() != 'basic':
2195 URLopener.http_error_default(self, url, fp,
2196 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002197 if not retry:
2198 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2199 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002200 name = 'retry_' + self.type + '_basic_auth'
2201 if data is None:
2202 return getattr(self,name)(url, realm)
2203 else:
2204 return getattr(self,name)(url, realm, data)
2205
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002206 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2207 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002208 """Error 407 -- proxy authentication required.
2209 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002210 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002211 URLopener.http_error_default(self, url, fp,
2212 errcode, errmsg, headers)
2213 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002214 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2215 if not match:
2216 URLopener.http_error_default(self, url, fp,
2217 errcode, errmsg, headers)
2218 scheme, realm = match.groups()
2219 if scheme.lower() != 'basic':
2220 URLopener.http_error_default(self, url, fp,
2221 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002222 if not retry:
2223 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2224 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002225 name = 'retry_proxy_' + self.type + '_basic_auth'
2226 if data is None:
2227 return getattr(self,name)(url, realm)
2228 else:
2229 return getattr(self,name)(url, realm, data)
2230
2231 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Cheryl Sabella0250de42018-04-25 16:51:54 -07002232 host, selector = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002233 newurl = 'http://' + host + selector
2234 proxy = self.proxies['http']
Cheryl Sabella0250de42018-04-25 16:51:54 -07002235 urltype, proxyhost = _splittype(proxy)
2236 proxyhost, proxyselector = _splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002237 i = proxyhost.find('@') + 1
2238 proxyhost = proxyhost[i:]
2239 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2240 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002241 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002242 quote(passwd, safe=''), proxyhost)
2243 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2244 if data is None:
2245 return self.open(newurl)
2246 else:
2247 return self.open(newurl, data)
2248
2249 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Cheryl Sabella0250de42018-04-25 16:51:54 -07002250 host, selector = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002251 newurl = 'https://' + host + selector
2252 proxy = self.proxies['https']
Cheryl Sabella0250de42018-04-25 16:51:54 -07002253 urltype, proxyhost = _splittype(proxy)
2254 proxyhost, proxyselector = _splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002255 i = proxyhost.find('@') + 1
2256 proxyhost = proxyhost[i:]
2257 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2258 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002259 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002260 quote(passwd, safe=''), proxyhost)
2261 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2262 if data is None:
2263 return self.open(newurl)
2264 else:
2265 return self.open(newurl, data)
2266
2267 def retry_http_basic_auth(self, url, realm, data=None):
Cheryl Sabella0250de42018-04-25 16:51:54 -07002268 host, selector = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002269 i = host.find('@') + 1
2270 host = host[i:]
2271 user, passwd = self.get_user_passwd(host, realm, i)
2272 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002273 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002274 quote(passwd, safe=''), host)
2275 newurl = 'http://' + host + selector
2276 if data is None:
2277 return self.open(newurl)
2278 else:
2279 return self.open(newurl, data)
2280
2281 def retry_https_basic_auth(self, url, realm, data=None):
Cheryl Sabella0250de42018-04-25 16:51:54 -07002282 host, selector = _splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002283 i = host.find('@') + 1
2284 host = host[i:]
2285 user, passwd = self.get_user_passwd(host, realm, i)
2286 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002287 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002288 quote(passwd, safe=''), host)
2289 newurl = 'https://' + host + selector
2290 if data is None:
2291 return self.open(newurl)
2292 else:
2293 return self.open(newurl, data)
2294
Florent Xicluna757445b2010-05-17 17:24:07 +00002295 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002296 key = realm + '@' + host.lower()
2297 if key in self.auth_cache:
2298 if clear_cache:
2299 del self.auth_cache[key]
2300 else:
2301 return self.auth_cache[key]
2302 user, passwd = self.prompt_user_passwd(host, realm)
2303 if user or passwd: self.auth_cache[key] = (user, passwd)
2304 return user, passwd
2305
2306 def prompt_user_passwd(self, host, realm):
2307 """Override this in a GUI environment!"""
2308 import getpass
2309 try:
2310 user = input("Enter username for %s at %s: " % (realm, host))
2311 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2312 (user, realm, host))
2313 return user, passwd
2314 except KeyboardInterrupt:
2315 print()
2316 return None, None
2317
2318
2319# Utility functions
2320
2321_localhost = None
2322def localhost():
2323 """Return the IP address of the magic hostname 'localhost'."""
2324 global _localhost
2325 if _localhost is None:
2326 _localhost = socket.gethostbyname('localhost')
2327 return _localhost
2328
2329_thishost = None
2330def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002331 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002332 global _thishost
2333 if _thishost is None:
Senthil Kumarandcdadfe2013-06-01 11:12:17 -07002334 try:
2335 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
2336 except socket.gaierror:
2337 _thishost = tuple(socket.gethostbyname_ex('localhost')[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002338 return _thishost
2339
2340_ftperrors = None
2341def ftperrors():
2342 """Return the set of errors raised by the FTP class."""
2343 global _ftperrors
2344 if _ftperrors is None:
2345 import ftplib
2346 _ftperrors = ftplib.all_errors
2347 return _ftperrors
2348
2349_noheaders = None
2350def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002351 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002352 global _noheaders
2353 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002354 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002355 return _noheaders
2356
2357
2358# Utility classes
2359
2360class ftpwrapper:
2361 """Class used by open_ftp() for cache of open FTP connections."""
2362
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002363 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2364 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002365 self.user = user
2366 self.passwd = passwd
2367 self.host = host
2368 self.port = port
2369 self.dirs = dirs
2370 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002371 self.refcount = 0
2372 self.keepalive = persistent
Victor Stinnerab73e652015-04-07 12:49:27 +02002373 try:
2374 self.init()
2375 except:
2376 self.close()
2377 raise
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002378
2379 def init(self):
2380 import ftplib
2381 self.busy = 0
2382 self.ftp = ftplib.FTP()
2383 self.ftp.connect(self.host, self.port, self.timeout)
2384 self.ftp.login(self.user, self.passwd)
Senthil Kumarancaa00fe2013-06-02 11:59:47 -07002385 _target = '/'.join(self.dirs)
2386 self.ftp.cwd(_target)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002387
2388 def retrfile(self, file, type):
2389 import ftplib
2390 self.endtransfer()
2391 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2392 else: cmd = 'TYPE ' + type; isdir = 0
2393 try:
2394 self.ftp.voidcmd(cmd)
2395 except ftplib.all_errors:
2396 self.init()
2397 self.ftp.voidcmd(cmd)
2398 conn = None
2399 if file and not isdir:
2400 # Try to retrieve as a file
2401 try:
2402 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002403 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002404 except ftplib.error_perm as reason:
2405 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002406 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002407 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002408 if not conn:
2409 # Set transfer mode to ASCII!
2410 self.ftp.voidcmd('TYPE A')
2411 # Try a directory listing. Verify that directory exists.
2412 if file:
2413 pwd = self.ftp.pwd()
2414 try:
2415 try:
2416 self.ftp.cwd(file)
2417 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002418 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002419 finally:
2420 self.ftp.cwd(pwd)
2421 cmd = 'LIST ' + file
2422 else:
2423 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002424 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002425 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002426
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002427 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2428 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002429 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002430 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002431 return (ftpobj, retrlen)
2432
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002433 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002434 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002435
2436 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002437 self.keepalive = False
2438 if self.refcount <= 0:
2439 self.real_close()
2440
2441 def file_close(self):
2442 self.endtransfer()
2443 self.refcount -= 1
2444 if self.refcount <= 0 and not self.keepalive:
2445 self.real_close()
2446
2447 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002448 self.endtransfer()
2449 try:
2450 self.ftp.close()
2451 except ftperrors():
2452 pass
2453
2454# Proxy handling
2455def getproxies_environment():
2456 """Return a dictionary of scheme -> proxy server URL mappings.
2457
2458 Scan the environment for variables named <scheme>_proxy;
2459 this seems to be the standard convention. If you need a
2460 different way, you can pass a proxies dictionary to the
2461 [Fancy]URLopener constructor.
2462
2463 """
2464 proxies = {}
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002465 # in order to prefer lowercase variables, process environment in
2466 # two passes: first matches any, second pass matches lowercase only
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002467 for name, value in os.environ.items():
2468 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002469 if value and name[-6:] == '_proxy':
2470 proxies[name[:-6]] = value
Senthil Kumaran4cbb23f2016-07-30 23:24:16 -07002471 # CVE-2016-1000110 - If we are running as CGI script, forget HTTP_PROXY
2472 # (non-all-lowercase) as it may be set from the web server by a "Proxy:"
2473 # header from the client
Senthil Kumaran17742f22016-07-30 23:39:06 -07002474 # If "proxy" is lowercase, it will still be used thanks to the next block
Senthil Kumaran4cbb23f2016-07-30 23:24:16 -07002475 if 'REQUEST_METHOD' in os.environ:
2476 proxies.pop('http', None)
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002477 for name, value in os.environ.items():
2478 if name[-6:] == '_proxy':
2479 name = name.lower()
2480 if value:
2481 proxies[name[:-6]] = value
2482 else:
2483 proxies.pop(name[:-6], None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002484 return proxies
2485
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002486def proxy_bypass_environment(host, proxies=None):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002487 """Test if proxies should not be used for a particular host.
2488
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002489 Checks the proxy dict for the value of no_proxy, which should
2490 be a list of comma separated DNS suffixes, or '*' for all hosts.
2491
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002492 """
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002493 if proxies is None:
2494 proxies = getproxies_environment()
2495 # don't bypass, if no_proxy isn't specified
2496 try:
2497 no_proxy = proxies['no']
2498 except KeyError:
2499 return 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002500 # '*' is special case for always bypass
2501 if no_proxy == '*':
2502 return 1
2503 # strip port off host
Cheryl Sabella0250de42018-04-25 16:51:54 -07002504 hostonly, port = _splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002505 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002506 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2507 for name in no_proxy_list:
Martin Panteraa279822016-04-30 01:03:40 +00002508 if name:
Xiang Zhang959ff7f2017-01-09 11:47:55 +08002509 name = name.lstrip('.') # ignore leading dots
Martin Panteraa279822016-04-30 01:03:40 +00002510 name = re.escape(name)
2511 pattern = r'(.+\.)?%s$' % name
2512 if (re.match(pattern, hostonly, re.I)
2513 or re.match(pattern, host, re.I)):
2514 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002515 # otherwise, don't bypass
2516 return 0
2517
2518
Ronald Oussorene72e1612011-03-14 18:15:25 -04002519# This code tests an OSX specific data structure but is testable on all
2520# platforms
2521def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2522 """
2523 Return True iff this host shouldn't be accessed using a proxy
2524
2525 This function uses the MacOSX framework SystemConfiguration
2526 to fetch the proxy information.
2527
2528 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2529 { 'exclude_simple': bool,
2530 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2531 }
2532 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002533 from fnmatch import fnmatch
2534
Cheryl Sabella0250de42018-04-25 16:51:54 -07002535 hostonly, port = _splitport(host)
Ronald Oussorene72e1612011-03-14 18:15:25 -04002536
2537 def ip2num(ipAddr):
2538 parts = ipAddr.split('.')
2539 parts = list(map(int, parts))
2540 if len(parts) != 4:
2541 parts = (parts + [0, 0, 0, 0])[:4]
2542 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2543
2544 # Check for simple host names:
2545 if '.' not in host:
2546 if proxy_settings['exclude_simple']:
2547 return True
2548
2549 hostIP = None
2550
2551 for value in proxy_settings.get('exceptions', ()):
2552 # Items in the list are strings like these: *.local, 169.254/16
2553 if not value: continue
2554
2555 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2556 if m is not None:
2557 if hostIP is None:
2558 try:
2559 hostIP = socket.gethostbyname(hostonly)
2560 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002561 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002562 continue
2563
2564 base = ip2num(m.group(1))
2565 mask = m.group(2)
2566 if mask is None:
2567 mask = 8 * (m.group(1).count('.') + 1)
2568 else:
2569 mask = int(mask[1:])
2570 mask = 32 - mask
2571
2572 if (hostIP >> mask) == (base >> mask):
2573 return True
2574
2575 elif fnmatch(host, value):
2576 return True
2577
2578 return False
2579
2580
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002581if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002582 from _scproxy import _get_proxy_settings, _get_proxies
2583
2584 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002585 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002586 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002587
2588 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002589 """Return a dictionary of scheme -> proxy server URL mappings.
2590
Ronald Oussoren84151202010-04-18 20:46:11 +00002591 This function uses the MacOSX framework SystemConfiguration
2592 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002593 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002594 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002595
Ronald Oussoren84151202010-04-18 20:46:11 +00002596
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002597
2598 def proxy_bypass(host):
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002599 """Return True, if host should be bypassed.
2600
2601 Checks proxy settings gathered from the environment, if specified,
2602 or from the MacOSX framework SystemConfiguration.
2603
2604 """
2605 proxies = getproxies_environment()
2606 if proxies:
2607 return proxy_bypass_environment(host, proxies)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002608 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002609 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002610
2611 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002612 return getproxies_environment() or getproxies_macosx_sysconf()
2613
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002614
2615elif os.name == 'nt':
2616 def getproxies_registry():
2617 """Return a dictionary of scheme -> proxy server URL mappings.
2618
2619 Win32 uses the registry to store proxies.
2620
2621 """
2622 proxies = {}
2623 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002624 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002625 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002626 # Std module, so should be around - but you never know!
2627 return proxies
2628 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002629 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002630 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002631 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002632 'ProxyEnable')[0]
2633 if proxyEnable:
2634 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002635 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002636 'ProxyServer')[0])
2637 if '=' in proxyServer:
2638 # Per-protocol settings
2639 for p in proxyServer.split(';'):
2640 protocol, address = p.split('=', 1)
2641 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002642 if not re.match('^([^/:]+)://', address):
2643 address = '%s://%s' % (protocol, address)
2644 proxies[protocol] = address
2645 else:
2646 # Use one setting for all protocols
2647 if proxyServer[:5] == 'http:':
2648 proxies['http'] = proxyServer
2649 else:
2650 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002651 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002652 proxies['ftp'] = 'ftp://%s' % proxyServer
2653 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002654 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002655 # Either registry key not found etc, or the value in an
2656 # unexpected format.
2657 # proxies already set up to be empty so nothing to do
2658 pass
2659 return proxies
2660
2661 def getproxies():
2662 """Return a dictionary of scheme -> proxy server URL mappings.
2663
2664 Returns settings gathered from the environment, if specified,
2665 or the registry.
2666
2667 """
2668 return getproxies_environment() or getproxies_registry()
2669
2670 def proxy_bypass_registry(host):
2671 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002672 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002673 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002674 # Std modules, so should be around - but you never know!
2675 return 0
2676 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002677 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002678 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002679 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002680 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002681 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002682 'ProxyOverride')[0])
2683 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002684 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002685 return 0
2686 if not proxyEnable or not proxyOverride:
2687 return 0
2688 # try to make a host list from name and IP address.
Cheryl Sabella0250de42018-04-25 16:51:54 -07002689 rawHost, port = _splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002690 host = [rawHost]
2691 try:
2692 addr = socket.gethostbyname(rawHost)
2693 if addr != rawHost:
2694 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002695 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002696 pass
2697 try:
2698 fqdn = socket.getfqdn(rawHost)
2699 if fqdn != rawHost:
2700 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002701 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002702 pass
2703 # make a check value list from the registry entry: replace the
2704 # '<local>' string by the localhost entry and the corresponding
2705 # canonical entry.
2706 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002707 # now check if we match one of the registry values.
2708 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002709 if test == '<local>':
2710 if '.' not in rawHost:
2711 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002712 test = test.replace(".", r"\.") # mask dots
2713 test = test.replace("*", r".*") # change glob sequence
2714 test = test.replace("?", r".") # change glob char
2715 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002716 if re.match(test, val, re.I):
2717 return 1
2718 return 0
2719
2720 def proxy_bypass(host):
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002721 """Return True, if host should be bypassed.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002722
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002723 Checks proxy settings gathered from the environment, if specified,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002724 or the registry.
2725
2726 """
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002727 proxies = getproxies_environment()
2728 if proxies:
2729 return proxy_bypass_environment(host, proxies)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002730 else:
2731 return proxy_bypass_registry(host)
2732
2733else:
2734 # By default use environment variables
2735 getproxies = getproxies_environment
2736 proxy_bypass = proxy_bypass_environment