blob: 3f8dcfb151d744ec1a278ceb35592bf8cc96bbdf [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 (
104 urlparse, urlsplit, urljoin, unwrap, quote, unquote,
105 splittype, splithost, splitport, splituser, splitpasswd,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100106 splitattr, splitquery, splitvalue, splittag, to_bytes,
107 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
201 warnings.warn("cafile, cpath and cadefault are deprecated, use a "
202 "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 """
245 url_type, path = splittype(url)
246
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'
352 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):
Senthil Kumaran52380922013-04-25 05:45:48 -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)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000384 self.host, self.selector = splithost(rest)
385 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):
429 hdrs = self.unredirected_hdrs.copy()
430 hdrs.update(self.headers)
431 return list(hdrs.items())
432
433class OpenerDirector:
434 def __init__(self):
435 client_version = "Python-urllib/%s" % __version__
436 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000437 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000438 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000439 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000440 self.handle_open = {}
441 self.handle_error = {}
442 self.process_response = {}
443 self.process_request = {}
444
445 def add_handler(self, handler):
446 if not hasattr(handler, "add_parent"):
447 raise TypeError("expected BaseHandler instance, got %r" %
448 type(handler))
449
450 added = False
451 for meth in dir(handler):
452 if meth in ["redirect_request", "do_open", "proxy_open"]:
453 # oops, coincidental match
454 continue
455
456 i = meth.find("_")
457 protocol = meth[:i]
458 condition = meth[i+1:]
459
460 if condition.startswith("error"):
461 j = condition.find("_") + i + 1
462 kind = meth[j+1:]
463 try:
464 kind = int(kind)
465 except ValueError:
466 pass
467 lookup = self.handle_error.get(protocol, {})
468 self.handle_error[protocol] = lookup
469 elif condition == "open":
470 kind = protocol
471 lookup = self.handle_open
472 elif condition == "response":
473 kind = protocol
474 lookup = self.process_response
475 elif condition == "request":
476 kind = protocol
477 lookup = self.process_request
478 else:
479 continue
480
481 handlers = lookup.setdefault(kind, [])
482 if handlers:
483 bisect.insort(handlers, handler)
484 else:
485 handlers.append(handler)
486 added = True
487
488 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000489 bisect.insort(self.handlers, handler)
490 handler.add_parent(self)
491
492 def close(self):
493 # Only exists for backwards compatibility.
494 pass
495
496 def _call_chain(self, chain, kind, meth_name, *args):
497 # Handlers raise an exception if no one else should try to handle
498 # the request, or return None if they can't but another handler
499 # could. Otherwise, they return the response.
500 handlers = chain.get(kind, ())
501 for handler in handlers:
502 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000503 result = func(*args)
504 if result is not None:
505 return result
506
507 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
508 # accept a URL or a Request object
509 if isinstance(fullurl, str):
510 req = Request(fullurl, data)
511 else:
512 req = fullurl
513 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000514 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000515
516 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000517 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000518
519 # pre-process request
520 meth_name = protocol+"_request"
521 for processor in self.process_request.get(protocol, []):
522 meth = getattr(processor, meth_name)
523 req = meth(req)
524
525 response = self._open(req, data)
526
527 # post-process response
528 meth_name = protocol+"_response"
529 for processor in self.process_response.get(protocol, []):
530 meth = getattr(processor, meth_name)
531 response = meth(req, response)
532
533 return response
534
535 def _open(self, req, data=None):
536 result = self._call_chain(self.handle_open, 'default',
537 'default_open', req)
538 if result:
539 return result
540
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000541 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000542 result = self._call_chain(self.handle_open, protocol, protocol +
543 '_open', req)
544 if result:
545 return result
546
547 return self._call_chain(self.handle_open, 'unknown',
548 'unknown_open', req)
549
550 def error(self, proto, *args):
551 if proto in ('http', 'https'):
552 # XXX http[s] protocols are special-cased
553 dict = self.handle_error['http'] # https is not different than http
554 proto = args[2] # YUCK!
555 meth_name = 'http_error_%s' % proto
556 http_err = 1
557 orig_args = args
558 else:
559 dict = self.handle_error
560 meth_name = proto + '_error'
561 http_err = 0
562 args = (dict, proto, meth_name) + args
563 result = self._call_chain(*args)
564 if result:
565 return result
566
567 if http_err:
568 args = (dict, 'default', 'http_error_default') + orig_args
569 return self._call_chain(*args)
570
571# XXX probably also want an abstract factory that knows when it makes
572# sense to skip a superclass in favor of a subclass and when it might
573# make sense to include both
574
575def build_opener(*handlers):
576 """Create an opener object from a list of handlers.
577
578 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000579 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000580
581 If any of the handlers passed as arguments are subclasses of the
582 default handlers, the default handlers will not be used.
583 """
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000584 opener = OpenerDirector()
585 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
586 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100587 FTPHandler, FileHandler, HTTPErrorProcessor,
588 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000589 if hasattr(http.client, "HTTPSConnection"):
590 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000591 skip = set()
592 for klass in default_classes:
593 for check in handlers:
Benjamin Peterson78c85382014-04-01 16:27:30 -0400594 if isinstance(check, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000595 if issubclass(check, klass):
596 skip.add(klass)
597 elif isinstance(check, klass):
598 skip.add(klass)
599 for klass in skip:
600 default_classes.remove(klass)
601
602 for klass in default_classes:
603 opener.add_handler(klass())
604
605 for h in handlers:
Benjamin Peterson5dd3cae2014-04-01 14:20:56 -0400606 if isinstance(h, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000607 h = h()
608 opener.add_handler(h)
609 return opener
610
611class BaseHandler:
612 handler_order = 500
613
614 def add_parent(self, parent):
615 self.parent = parent
616
617 def close(self):
618 # Only exists for backwards compatibility
619 pass
620
621 def __lt__(self, other):
622 if not hasattr(other, "handler_order"):
623 # Try to preserve the old behavior of having custom classes
624 # inserted after default ones (works only for custom user
625 # classes which are not aware of handler_order).
626 return True
627 return self.handler_order < other.handler_order
628
629
630class HTTPErrorProcessor(BaseHandler):
631 """Process HTTP error responses."""
632 handler_order = 1000 # after all other processing
633
634 def http_response(self, request, response):
635 code, msg, hdrs = response.code, response.msg, response.info()
636
637 # According to RFC 2616, "2xx" code indicates that the client's
638 # request was successfully received, understood, and accepted.
639 if not (200 <= code < 300):
640 response = self.parent.error(
641 'http', request, response, code, msg, hdrs)
642
643 return response
644
645 https_response = http_response
646
647class HTTPDefaultErrorHandler(BaseHandler):
648 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000649 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000650
651class HTTPRedirectHandler(BaseHandler):
652 # maximum number of redirections to any single URL
653 # this is needed because of the state that cookies introduce
654 max_repeats = 4
655 # maximum total number of redirections (regardless of URL) before
656 # assuming we're in a loop
657 max_redirections = 10
658
659 def redirect_request(self, req, fp, code, msg, headers, newurl):
660 """Return a Request or None in response to a redirect.
661
662 This is called by the http_error_30x methods when a
663 redirection response is received. If a redirection should
664 take place, return a new Request to allow http_error_30x to
665 perform the redirect. Otherwise, raise HTTPError if no-one
666 else should try to handle this url. Return None if you can't
667 but another Handler might.
668 """
669 m = req.get_method()
670 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
671 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000672 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000673
674 # Strictly (according to RFC 2616), 301 or 302 in response to
675 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000676 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000677 # essentially all clients do redirect in this case, so we do
678 # the same.
Martin Pantere6f06092016-05-16 01:14:20 +0000679
680 # Be conciliant with URIs containing a space. This is mainly
681 # redundant with the more complete encoding done in http_error_302(),
682 # but it is kept for compatibility with other callers.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000683 newurl = newurl.replace(' ', '%20')
Martin Pantere6f06092016-05-16 01:14:20 +0000684
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000685 CONTENT_HEADERS = ("content-length", "content-type")
686 newheaders = dict((k, v) for k, v in req.headers.items()
687 if k.lower() not in CONTENT_HEADERS)
688 return Request(newurl,
689 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000690 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000691 unverifiable=True)
692
693 # Implementation note: To avoid the server sending us into an
694 # infinite loop, the request object needs to track what URLs we
695 # have already seen. Do this by adding a handler-specific
696 # attribute to the Request object.
697 def http_error_302(self, req, fp, code, msg, headers):
698 # Some servers (incorrectly) return multiple Location headers
699 # (so probably same goes for URI). Use first header.
700 if "location" in headers:
701 newurl = headers["location"]
702 elif "uri" in headers:
703 newurl = headers["uri"]
704 else:
705 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000706
707 # fix a possible malformed URL
708 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700709
710 # For security reasons we don't allow redirection to anything other
711 # than http, https or ftp.
712
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800713 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800714 raise HTTPError(
715 newurl, code,
716 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
717 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700718
Martin Panterce6e0682016-05-16 01:07:13 +0000719 if not urlparts.path and urlparts.netloc:
Facundo Batistaf24802c2008-08-17 03:36:03 +0000720 urlparts = list(urlparts)
721 urlparts[2] = "/"
722 newurl = urlunparse(urlparts)
723
Martin Pantere6f06092016-05-16 01:14:20 +0000724 # http.client.parse_headers() decodes as ISO-8859-1. Recover the
725 # original bytes and percent-encode non-ASCII bytes, and any special
726 # characters such as the space.
727 newurl = quote(
728 newurl, encoding="iso-8859-1", safe=string.punctuation)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000729 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000730
731 # XXX Probably want to forget about the state of the current
732 # request, although that might interact poorly with other
733 # handlers that also use handler-specific request attributes
734 new = self.redirect_request(req, fp, code, msg, headers, newurl)
735 if new is None:
736 return
737
738 # loop detection
739 # .redirect_dict has a key url if url was previously visited.
740 if hasattr(req, 'redirect_dict'):
741 visited = new.redirect_dict = req.redirect_dict
742 if (visited.get(newurl, 0) >= self.max_repeats or
743 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000744 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000745 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000746 else:
747 visited = new.redirect_dict = req.redirect_dict = {}
748 visited[newurl] = visited.get(newurl, 0) + 1
749
750 # Don't close the fp until we are sure that we won't use it
751 # with HTTPError.
752 fp.read()
753 fp.close()
754
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000755 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000756
757 http_error_301 = http_error_303 = http_error_307 = http_error_302
758
759 inf_msg = "The HTTP server returned a redirect error that would " \
760 "lead to an infinite loop.\n" \
761 "The last 30x error message was:\n"
762
763
764def _parse_proxy(proxy):
765 """Return (scheme, user, password, host/port) given a URL or an authority.
766
767 If a URL is supplied, it must have an authority (host:port) component.
768 According to RFC 3986, having an authority component means the URL must
Senthil Kumarand8e24f12014-04-14 16:32:20 -0400769 have two slashes after the scheme.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000770 """
Georg Brandl13e89462008-07-01 19:56:00 +0000771 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000772 if not r_scheme.startswith("/"):
773 # authority
774 scheme = None
775 authority = proxy
776 else:
777 # URL
778 if not r_scheme.startswith("//"):
779 raise ValueError("proxy URL with no authority: %r" % proxy)
780 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
781 # and 3.3.), path is empty or starts with '/'
782 end = r_scheme.find("/", 2)
783 if end == -1:
784 end = None
785 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000786 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000787 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000788 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000789 else:
790 user = password = None
791 return scheme, user, password, hostport
792
793class ProxyHandler(BaseHandler):
794 # Proxies must be in front
795 handler_order = 100
796
797 def __init__(self, proxies=None):
798 if proxies is None:
799 proxies = getproxies()
800 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
801 self.proxies = proxies
802 for type, url in proxies.items():
803 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200804 lambda r, proxy=url, type=type, meth=self.proxy_open:
805 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000806
807 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000808 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000809 proxy_type, user, password, hostport = _parse_proxy(proxy)
810 if proxy_type is None:
811 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000812
813 if req.host and proxy_bypass(req.host):
814 return None
815
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000816 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000817 user_pass = '%s:%s' % (unquote(user),
818 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000819 creds = base64.b64encode(user_pass.encode()).decode("ascii")
820 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000821 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000822 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000823 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000824 # let other handlers take care of it
825 return None
826 else:
827 # need to start over, because the other handlers don't
828 # grok the proxy's URL type
829 # e.g. if we have a constructor arg proxies like so:
830 # {'http': 'ftp://proxy.example.com'}, we may end up turning
831 # a request for http://acme.example.com/a into one for
832 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000833 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000834
835class HTTPPasswordMgr:
836
837 def __init__(self):
838 self.passwd = {}
839
840 def add_password(self, realm, uri, user, passwd):
841 # uri could be a single URI or a sequence
842 if isinstance(uri, str):
843 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800844 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000845 self.passwd[realm] = {}
846 for default_port in True, False:
847 reduced_uri = tuple(
848 [self.reduce_uri(u, default_port) for u in uri])
849 self.passwd[realm][reduced_uri] = (user, passwd)
850
851 def find_user_password(self, realm, authuri):
852 domains = self.passwd.get(realm, {})
853 for default_port in True, False:
854 reduced_authuri = self.reduce_uri(authuri, default_port)
855 for uris, authinfo in domains.items():
856 for uri in uris:
857 if self.is_suburi(uri, reduced_authuri):
858 return authinfo
859 return None, None
860
861 def reduce_uri(self, uri, default_port=True):
862 """Accept authority or URI and extract only the authority and path."""
863 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000864 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000865 if parts[1]:
866 # URI
867 scheme = parts[0]
868 authority = parts[1]
869 path = parts[2] or '/'
870 else:
871 # host or host:port
872 scheme = None
873 authority = uri
874 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000875 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000876 if default_port and port is None and scheme is not None:
877 dport = {"http": 80,
878 "https": 443,
879 }.get(scheme)
880 if dport is not None:
881 authority = "%s:%d" % (host, dport)
882 return authority, path
883
884 def is_suburi(self, base, test):
885 """Check if test is below base in a URI tree
886
887 Both args must be URIs in reduced form.
888 """
889 if base == test:
890 return True
891 if base[0] != test[0]:
892 return False
893 common = posixpath.commonprefix((base[1], test[1]))
894 if len(common) == len(base[1]):
895 return True
896 return False
897
898
899class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
900
901 def find_user_password(self, realm, authuri):
902 user, password = HTTPPasswordMgr.find_user_password(self, realm,
903 authuri)
904 if user is not None:
905 return user, password
906 return HTTPPasswordMgr.find_user_password(self, None, authuri)
907
908
R David Murray4c7f9952015-04-16 16:36:18 -0400909class HTTPPasswordMgrWithPriorAuth(HTTPPasswordMgrWithDefaultRealm):
910
911 def __init__(self, *args, **kwargs):
912 self.authenticated = {}
913 super().__init__(*args, **kwargs)
914
915 def add_password(self, realm, uri, user, passwd, is_authenticated=False):
916 self.update_authenticated(uri, is_authenticated)
917 # Add a default for prior auth requests
918 if realm is not None:
919 super().add_password(None, uri, user, passwd)
920 super().add_password(realm, uri, user, passwd)
921
922 def update_authenticated(self, uri, is_authenticated=False):
923 # uri could be a single URI or a sequence
924 if isinstance(uri, str):
925 uri = [uri]
926
927 for default_port in True, False:
928 for u in uri:
929 reduced_uri = self.reduce_uri(u, default_port)
930 self.authenticated[reduced_uri] = is_authenticated
931
932 def is_authenticated(self, authuri):
933 for default_port in True, False:
934 reduced_authuri = self.reduce_uri(authuri, default_port)
935 for uri in self.authenticated:
936 if self.is_suburi(uri, reduced_authuri):
937 return self.authenticated[uri]
938
939
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000940class AbstractBasicAuthHandler:
941
942 # XXX this allows for multiple auth-schemes, but will stupidly pick
943 # the last one with a realm specified.
944
945 # allow for double- and single-quoted realm values
946 # (single quotes are a violation of the RFC, but appear in the wild)
947 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800948 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000949
950 # XXX could pre-emptively send auth info already accepted (RFC 2617,
951 # end of section 2, and section 1.2 immediately after "credentials"
952 # production).
953
954 def __init__(self, password_mgr=None):
955 if password_mgr is None:
956 password_mgr = HTTPPasswordMgr()
957 self.passwd = password_mgr
958 self.add_password = self.passwd.add_password
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000959
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000960 def http_error_auth_reqed(self, authreq, host, req, headers):
961 # host may be an authority (without userinfo) or a URL with an
962 # authority
963 # XXX could be multiple headers
964 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000965
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000966 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800967 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800968 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800969 raise ValueError("AbstractBasicAuthHandler does not"
970 " support the following scheme: '%s'" %
971 scheme)
972 else:
973 mo = AbstractBasicAuthHandler.rx.search(authreq)
974 if mo:
975 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800976 if quote not in ['"',"'"]:
977 warnings.warn("Basic Auth Realm was unquoted",
978 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800979 if scheme.lower() == 'basic':
Senthil Kumaran78373762014-08-20 07:53:58 +0530980 return self.retry_http_basic_auth(host, req, realm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000981
982 def retry_http_basic_auth(self, host, req, realm):
983 user, pw = self.passwd.find_user_password(realm, host)
984 if pw is not None:
985 raw = "%s:%s" % (user, pw)
986 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
Senthil Kumaran78373762014-08-20 07:53:58 +0530987 if req.get_header(self.auth_header, None) == auth:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000988 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000989 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000990 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000991 else:
992 return None
993
R David Murray4c7f9952015-04-16 16:36:18 -0400994 def http_request(self, req):
995 if (not hasattr(self.passwd, 'is_authenticated') or
996 not self.passwd.is_authenticated(req.full_url)):
997 return req
998
999 if not req.has_header('Authorization'):
1000 user, passwd = self.passwd.find_user_password(None, req.full_url)
1001 credentials = '{0}:{1}'.format(user, passwd).encode()
1002 auth_str = base64.standard_b64encode(credentials).decode()
1003 req.add_unredirected_header('Authorization',
1004 'Basic {}'.format(auth_str.strip()))
1005 return req
1006
1007 def http_response(self, req, response):
1008 if hasattr(self.passwd, 'is_authenticated'):
1009 if 200 <= response.code < 300:
1010 self.passwd.update_authenticated(req.full_url, True)
1011 else:
1012 self.passwd.update_authenticated(req.full_url, False)
1013 return response
1014
1015 https_request = http_request
1016 https_response = http_response
1017
1018
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001019
1020class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
1021
1022 auth_header = 'Authorization'
1023
1024 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001025 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001026 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001027 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001028 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001029
1030
1031class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
1032
1033 auth_header = 'Proxy-authorization'
1034
1035 def http_error_407(self, req, fp, code, msg, headers):
1036 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +00001037 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001038 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
1039 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001040 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001041 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001042 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +00001043 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001044
1045
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001046# Return n random bytes.
1047_randombytes = os.urandom
1048
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001049
1050class AbstractDigestAuthHandler:
1051 # Digest authentication is specified in RFC 2617.
1052
1053 # XXX The client does not inspect the Authentication-Info header
1054 # in a successful response.
1055
1056 # XXX It should be possible to test this implementation against
1057 # a mock server that just generates a static set of challenges.
1058
1059 # XXX qop="auth-int" supports is shaky
1060
1061 def __init__(self, passwd=None):
1062 if passwd is None:
1063 passwd = HTTPPasswordMgr()
1064 self.passwd = passwd
1065 self.add_password = self.passwd.add_password
1066 self.retried = 0
1067 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001068 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001069
1070 def reset_retry_count(self):
1071 self.retried = 0
1072
1073 def http_error_auth_reqed(self, auth_header, host, req, headers):
1074 authreq = headers.get(auth_header, None)
1075 if self.retried > 5:
1076 # Don't fail endlessly - if we failed once, we'll probably
1077 # fail a second time. Hm. Unless the Password Manager is
1078 # prompting for the information. Crap. This isn't great
1079 # but it's better than the current 'repeat until recursion
1080 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001081 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +00001082 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001083 else:
1084 self.retried += 1
1085 if authreq:
1086 scheme = authreq.split()[0]
1087 if scheme.lower() == 'digest':
1088 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +08001089 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +08001090 raise ValueError("AbstractDigestAuthHandler does not support"
1091 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001092
1093 def retry_http_digest_auth(self, req, auth):
1094 token, challenge = auth.split(' ', 1)
1095 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
1096 auth = self.get_authorization(req, chal)
1097 if auth:
1098 auth_val = 'Digest %s' % auth
1099 if req.headers.get(self.auth_header, None) == auth_val:
1100 return None
1101 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +00001102 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001103 return resp
1104
1105 def get_cnonce(self, nonce):
1106 # The cnonce-value is an opaque
1107 # quoted string value provided by the client and used by both client
1108 # and server to avoid chosen plaintext attacks, to provide mutual
1109 # authentication, and to provide some message integrity protection.
1110 # This isn't a fabulous effort, but it's probably Good Enough.
1111 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001112 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001113 dig = hashlib.sha1(b).hexdigest()
1114 return dig[:16]
1115
1116 def get_authorization(self, req, chal):
1117 try:
1118 realm = chal['realm']
1119 nonce = chal['nonce']
1120 qop = chal.get('qop')
1121 algorithm = chal.get('algorithm', 'MD5')
1122 # mod_digest doesn't send an opaque, even though it isn't
1123 # supposed to be optional
1124 opaque = chal.get('opaque', None)
1125 except KeyError:
1126 return None
1127
1128 H, KD = self.get_algorithm_impls(algorithm)
1129 if H is None:
1130 return None
1131
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001132 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001133 if user is None:
1134 return None
1135
1136 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001137 if req.data is not None:
1138 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001139 else:
1140 entdig = None
1141
1142 A1 = "%s:%s:%s" % (user, realm, pw)
1143 A2 = "%s:%s" % (req.get_method(),
1144 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001145 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001146 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001147 if nonce == self.last_nonce:
1148 self.nonce_count += 1
1149 else:
1150 self.nonce_count = 1
1151 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001152 ncvalue = '%08x' % self.nonce_count
1153 cnonce = self.get_cnonce(nonce)
1154 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1155 respdig = KD(H(A1), noncebit)
1156 elif qop is None:
1157 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1158 else:
1159 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001160 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001161
1162 # XXX should the partial digests be encoded too?
1163
1164 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001165 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001166 respdig)
1167 if opaque:
1168 base += ', opaque="%s"' % opaque
1169 if entdig:
1170 base += ', digest="%s"' % entdig
1171 base += ', algorithm="%s"' % algorithm
1172 if qop:
1173 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1174 return base
1175
1176 def get_algorithm_impls(self, algorithm):
1177 # lambdas assume digest modules are imported at the top level
1178 if algorithm == 'MD5':
1179 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1180 elif algorithm == 'SHA':
1181 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1182 # XXX MD5-sess
Berker Peksage88dd1c2016-03-06 16:16:40 +02001183 else:
1184 raise ValueError("Unsupported digest authentication "
1185 "algorithm %r" % algorithm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001186 KD = lambda s, d: H("%s:%s" % (s, d))
1187 return H, KD
1188
1189 def get_entity_digest(self, data, chal):
1190 # XXX not implemented yet
1191 return None
1192
1193
1194class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1195 """An authentication protocol defined by RFC 2069
1196
1197 Digest authentication improves on basic authentication because it
1198 does not transmit passwords in the clear.
1199 """
1200
1201 auth_header = 'Authorization'
1202 handler_order = 490 # before Basic auth
1203
1204 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001205 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001206 retry = self.http_error_auth_reqed('www-authenticate',
1207 host, req, headers)
1208 self.reset_retry_count()
1209 return retry
1210
1211
1212class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1213
1214 auth_header = 'Proxy-Authorization'
1215 handler_order = 490 # before Basic auth
1216
1217 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001218 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001219 retry = self.http_error_auth_reqed('proxy-authenticate',
1220 host, req, headers)
1221 self.reset_retry_count()
1222 return retry
1223
1224class AbstractHTTPHandler(BaseHandler):
1225
1226 def __init__(self, debuglevel=0):
1227 self._debuglevel = debuglevel
1228
1229 def set_http_debuglevel(self, level):
1230 self._debuglevel = level
1231
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001232 def _get_content_length(self, request):
1233 return http.client.HTTPConnection._get_content_length(
1234 request.data,
1235 request.get_method())
1236
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001237 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001238 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001239 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001240 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001241
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001242 if request.data is not None: # POST
1243 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001244 if isinstance(data, str):
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001245 msg = "POST data should be bytes, an iterable of bytes, " \
1246 "or a file object. It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001247 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001248 if not request.has_header('Content-type'):
1249 request.add_unredirected_header(
1250 'Content-type',
1251 'application/x-www-form-urlencoded')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001252 if (not request.has_header('Content-length')
1253 and not request.has_header('Transfer-encoding')):
1254 content_length = self._get_content_length(request)
1255 if content_length is not None:
1256 request.add_unredirected_header(
1257 'Content-length', str(content_length))
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001258 else:
1259 request.add_unredirected_header(
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001260 'Transfer-encoding', 'chunked')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001261
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001262 sel_host = host
1263 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001264 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001265 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001266 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001267 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001268 for name, value in self.parent.addheaders:
1269 name = name.capitalize()
1270 if not request.has_header(name):
1271 request.add_unredirected_header(name, value)
1272
1273 return request
1274
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001275 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001276 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001277
1278 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001279 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001280 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001281 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001282 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001283
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001284 # will parse host:port
1285 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran9642eed2016-05-13 01:32:42 -07001286 h.set_debuglevel(self._debuglevel)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001287
1288 headers = dict(req.unredirected_hdrs)
1289 headers.update(dict((k, v) for k, v in req.headers.items()
1290 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001291
1292 # TODO(jhylton): Should this be redesigned to handle
1293 # persistent connections?
1294
1295 # We want to make an HTTP/1.1 request, but the addinfourl
1296 # class isn't prepared to deal with a persistent connection.
1297 # It will try to read all remaining data from the socket,
1298 # which will block while the server waits for the next request.
1299 # So make sure the connection gets closed after the (only)
1300 # request.
1301 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001302 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001303
1304 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001305 tunnel_headers = {}
1306 proxy_auth_hdr = "Proxy-Authorization"
1307 if proxy_auth_hdr in headers:
1308 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1309 # Proxy-Authorization should not be sent to origin
1310 # server.
1311 del headers[proxy_auth_hdr]
1312 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001313
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001314 try:
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001315 try:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001316 h.request(req.get_method(), req.selector, req.data, headers,
1317 encode_chunked=req.has_header('Transfer-encoding'))
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001318 except OSError as err: # timeout error
1319 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001320 r = h.getresponse()
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001321 except:
1322 h.close()
1323 raise
1324
1325 # If the server does not send us a 'Connection: close' header,
1326 # HTTPConnection assumes the socket should be left open. Manually
1327 # mark the socket to be closed when this response object goes away.
1328 if h.sock:
1329 h.sock.close()
1330 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001331
Senthil Kumaran26430412011-04-13 07:01:19 +08001332 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001333 # This line replaces the .msg attribute of the HTTPResponse
1334 # with .headers, because urllib clients expect the response to
1335 # have the reason in .msg. It would be good to mark this
1336 # attribute is deprecated and get then to use info() or
1337 # .headers.
1338 r.msg = r.reason
1339 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001340
1341
1342class HTTPHandler(AbstractHTTPHandler):
1343
1344 def http_open(self, req):
1345 return self.do_open(http.client.HTTPConnection, req)
1346
1347 http_request = AbstractHTTPHandler.do_request_
1348
1349if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001350
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001351 class HTTPSHandler(AbstractHTTPHandler):
1352
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001353 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1354 AbstractHTTPHandler.__init__(self, debuglevel)
1355 self._context = context
1356 self._check_hostname = check_hostname
1357
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001358 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001359 return self.do_open(http.client.HTTPSConnection, req,
1360 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001361
1362 https_request = AbstractHTTPHandler.do_request_
1363
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001364 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001365
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001366class HTTPCookieProcessor(BaseHandler):
1367 def __init__(self, cookiejar=None):
1368 import http.cookiejar
1369 if cookiejar is None:
1370 cookiejar = http.cookiejar.CookieJar()
1371 self.cookiejar = cookiejar
1372
1373 def http_request(self, request):
1374 self.cookiejar.add_cookie_header(request)
1375 return request
1376
1377 def http_response(self, request, response):
1378 self.cookiejar.extract_cookies(response, request)
1379 return response
1380
1381 https_request = http_request
1382 https_response = http_response
1383
1384class UnknownHandler(BaseHandler):
1385 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001386 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001387 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001388
1389def parse_keqv_list(l):
1390 """Parse list of key=value strings where keys are not duplicated."""
1391 parsed = {}
1392 for elt in l:
1393 k, v = elt.split('=', 1)
1394 if v[0] == '"' and v[-1] == '"':
1395 v = v[1:-1]
1396 parsed[k] = v
1397 return parsed
1398
1399def parse_http_list(s):
1400 """Parse lists as described by RFC 2068 Section 2.
1401
1402 In particular, parse comma-separated lists where the elements of
1403 the list may include quoted-strings. A quoted-string could
1404 contain a comma. A non-quoted string could have quotes in the
1405 middle. Neither commas nor quotes count if they are escaped.
1406 Only double-quotes count, not single-quotes.
1407 """
1408 res = []
1409 part = ''
1410
1411 escape = quote = False
1412 for cur in s:
1413 if escape:
1414 part += cur
1415 escape = False
1416 continue
1417 if quote:
1418 if cur == '\\':
1419 escape = True
1420 continue
1421 elif cur == '"':
1422 quote = False
1423 part += cur
1424 continue
1425
1426 if cur == ',':
1427 res.append(part)
1428 part = ''
1429 continue
1430
1431 if cur == '"':
1432 quote = True
1433
1434 part += cur
1435
1436 # append last part
1437 if part:
1438 res.append(part)
1439
1440 return [part.strip() for part in res]
1441
1442class FileHandler(BaseHandler):
1443 # Use local file or FTP depending on form of URL
1444 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001445 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001446 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1447 req.host != 'localhost'):
Senthil Kumaranbc07ac52014-07-22 00:15:20 -07001448 if not req.host in self.get_names():
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001449 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001450 else:
1451 return self.open_local_file(req)
1452
1453 # names for the localhost
1454 names = None
1455 def get_names(self):
1456 if FileHandler.names is None:
1457 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001458 FileHandler.names = tuple(
1459 socket.gethostbyname_ex('localhost')[2] +
1460 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001461 except socket.gaierror:
1462 FileHandler.names = (socket.gethostbyname('localhost'),)
1463 return FileHandler.names
1464
1465 # not entirely sure what the rules are here
1466 def open_local_file(self, req):
1467 import email.utils
1468 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001469 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001470 filename = req.selector
1471 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001472 try:
1473 stats = os.stat(localfile)
1474 size = stats.st_size
1475 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001476 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001477 headers = email.message_from_string(
1478 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1479 (mtype or 'text/plain', size, modified))
1480 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001481 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001482 if not host or \
1483 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001484 if host:
1485 origurl = 'file://' + host + filename
1486 else:
1487 origurl = 'file://' + filename
1488 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001489 except OSError as exp:
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001490 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001491 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001492
1493def _safe_gethostbyname(host):
1494 try:
1495 return socket.gethostbyname(host)
1496 except socket.gaierror:
1497 return None
1498
1499class FTPHandler(BaseHandler):
1500 def ftp_open(self, req):
1501 import ftplib
1502 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001503 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001504 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001505 raise URLError('ftp error: no host given')
1506 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001507 if port is None:
1508 port = ftplib.FTP_PORT
1509 else:
1510 port = int(port)
1511
1512 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001513 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001514 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001515 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001516 else:
1517 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001518 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001519 user = user or ''
1520 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001521
1522 try:
1523 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001524 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001525 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001526 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001527 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001528 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001529 dirs, file = dirs[:-1], dirs[-1]
1530 if dirs and not dirs[0]:
1531 dirs = dirs[1:]
1532 try:
1533 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1534 type = file and 'I' or 'D'
1535 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001536 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001537 if attr.lower() == 'type' and \
1538 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1539 type = value.upper()
1540 fp, retrlen = fw.retrfile(file, type)
1541 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001542 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001543 if mtype:
1544 headers += "Content-type: %s\n" % mtype
1545 if retrlen is not None and retrlen >= 0:
1546 headers += "Content-length: %d\n" % retrlen
1547 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001548 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001549 except ftplib.all_errors as exp:
1550 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001551 raise exc.with_traceback(sys.exc_info()[2])
1552
1553 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001554 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1555 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001556
1557class CacheFTPHandler(FTPHandler):
1558 # XXX would be nice to have pluggable cache strategies
1559 # XXX this stuff is definitely not thread safe
1560 def __init__(self):
1561 self.cache = {}
1562 self.timeout = {}
1563 self.soonest = 0
1564 self.delay = 60
1565 self.max_conns = 16
1566
1567 def setTimeout(self, t):
1568 self.delay = t
1569
1570 def setMaxConns(self, m):
1571 self.max_conns = m
1572
1573 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1574 key = user, host, port, '/'.join(dirs), timeout
1575 if key in self.cache:
1576 self.timeout[key] = time.time() + self.delay
1577 else:
1578 self.cache[key] = ftpwrapper(user, passwd, host, port,
1579 dirs, timeout)
1580 self.timeout[key] = time.time() + self.delay
1581 self.check_cache()
1582 return self.cache[key]
1583
1584 def check_cache(self):
1585 # first check for old ones
1586 t = time.time()
1587 if self.soonest <= t:
1588 for k, v in list(self.timeout.items()):
1589 if v < t:
1590 self.cache[k].close()
1591 del self.cache[k]
1592 del self.timeout[k]
1593 self.soonest = min(list(self.timeout.values()))
1594
1595 # then check the size
1596 if len(self.cache) == self.max_conns:
1597 for k, v in list(self.timeout.items()):
1598 if v == self.soonest:
1599 del self.cache[k]
1600 del self.timeout[k]
1601 break
1602 self.soonest = min(list(self.timeout.values()))
1603
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001604 def clear_cache(self):
1605 for conn in self.cache.values():
1606 conn.close()
1607 self.cache.clear()
1608 self.timeout.clear()
1609
Antoine Pitroudf204be2012-11-24 17:59:08 +01001610class DataHandler(BaseHandler):
1611 def data_open(self, req):
1612 # data URLs as specified in RFC 2397.
1613 #
1614 # ignores POSTed data
1615 #
1616 # syntax:
1617 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1618 # mediatype := [ type "/" subtype ] *( ";" parameter )
1619 # data := *urlchar
1620 # parameter := attribute "=" value
1621 url = req.full_url
1622
1623 scheme, data = url.split(":",1)
1624 mediatype, data = data.split(",",1)
1625
1626 # even base64 encoded data URLs might be quoted so unquote in any case:
1627 data = unquote_to_bytes(data)
1628 if mediatype.endswith(";base64"):
1629 data = base64.decodebytes(data)
1630 mediatype = mediatype[:-7]
1631
1632 if not mediatype:
1633 mediatype = "text/plain;charset=US-ASCII"
1634
1635 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1636 (mediatype, len(data)))
1637
1638 return addinfourl(io.BytesIO(data), headers, url)
1639
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001640
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001641# Code move from the old urllib module
1642
1643MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1644
1645# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001646if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001647 from nturl2path import url2pathname, pathname2url
1648else:
1649 def url2pathname(pathname):
1650 """OS-specific conversion from a relative URL of the 'file' scheme
1651 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001652 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001653
1654 def pathname2url(pathname):
1655 """OS-specific conversion from a file system path to a relative URL
1656 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001657 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001658
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001659
1660ftpcache = {}
Senthil Kumarana2a9ddd2017-04-08 23:27:25 -07001661
1662
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001663class URLopener:
1664 """Class to open URLs.
1665 This is a class rather than just a subroutine because we may need
1666 more than one set of global protocol-specific options.
1667 Note -- this is a base class for those who don't want the
1668 automatic handling of errors type 302 (relocated) and 401
1669 (authorization needed)."""
1670
1671 __tempfiles = None
1672
1673 version = "Python-urllib/%s" % __version__
1674
1675 # Constructor
1676 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001677 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001678 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1679 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001680 if proxies is None:
1681 proxies = getproxies()
1682 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1683 self.proxies = proxies
1684 self.key_file = x509.get('key_file')
1685 self.cert_file = x509.get('cert_file')
Raymond Hettingerb7f3c942016-09-09 16:44:53 -07001686 self.addheaders = [('User-Agent', self.version), ('Accept', '*/*')]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001687 self.__tempfiles = []
1688 self.__unlink = os.unlink # See cleanup()
1689 self.tempcache = None
1690 # Undocumented feature: if you assign {} to tempcache,
1691 # it is used to cache files retrieved with
1692 # self.retrieve(). This is not enabled by default
1693 # since it does not work for changing documents (and I
1694 # haven't got the logic to check expiration headers
1695 # yet).
1696 self.ftpcache = ftpcache
1697 # Undocumented feature: you can use a different
1698 # ftp cache by assigning to the .ftpcache member;
1699 # in case you want logically independent URL openers
1700 # XXX This is not threadsafe. Bah.
1701
1702 def __del__(self):
1703 self.close()
1704
1705 def close(self):
1706 self.cleanup()
1707
1708 def cleanup(self):
1709 # This code sometimes runs when the rest of this module
1710 # has already been deleted, so it can't use any globals
1711 # or import anything.
1712 if self.__tempfiles:
1713 for file in self.__tempfiles:
1714 try:
1715 self.__unlink(file)
1716 except OSError:
1717 pass
1718 del self.__tempfiles[:]
1719 if self.tempcache:
1720 self.tempcache.clear()
1721
1722 def addheader(self, *args):
1723 """Add a header to be used by the HTTP interface only
1724 e.g. u.addheader('Accept', 'sound/basic')"""
1725 self.addheaders.append(args)
1726
1727 # External interface
1728 def open(self, fullurl, data=None):
1729 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001730 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001731 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001732 if self.tempcache and fullurl in self.tempcache:
1733 filename, headers = self.tempcache[fullurl]
1734 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001735 return addinfourl(fp, headers, fullurl)
1736 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001737 if not urltype:
1738 urltype = 'file'
1739 if urltype in self.proxies:
1740 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001741 urltype, proxyhost = splittype(proxy)
1742 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001743 url = (host, fullurl) # Signal special case to open_*()
1744 else:
1745 proxy = None
1746 name = 'open_' + urltype
1747 self.type = urltype
1748 name = name.replace('-', '_')
1749 if not hasattr(self, name):
1750 if proxy:
1751 return self.open_unknown_proxy(proxy, fullurl, data)
1752 else:
1753 return self.open_unknown(fullurl, data)
1754 try:
1755 if data is None:
1756 return getattr(self, name)(url)
1757 else:
1758 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001759 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001760 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001761 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001762 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001763
1764 def open_unknown(self, fullurl, data=None):
1765 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001766 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001767 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001768
1769 def open_unknown_proxy(self, proxy, fullurl, data=None):
1770 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001771 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001772 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001773
1774 # External interface
1775 def retrieve(self, url, filename=None, reporthook=None, data=None):
1776 """retrieve(url) returns (filename, headers) for a local object
1777 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001778 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001779 if self.tempcache and url in self.tempcache:
1780 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001781 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001782 if filename is None and (not type or type == 'file'):
1783 try:
1784 fp = self.open_local_file(url1)
1785 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001786 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001787 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001788 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001789 pass
1790 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001791 try:
1792 headers = fp.info()
1793 if filename:
1794 tfp = open(filename, 'wb')
1795 else:
1796 import tempfile
1797 garbage, path = splittype(url)
1798 garbage, path = splithost(path or "")
1799 path, garbage = splitquery(path or "")
1800 path, garbage = splitattr(path or "")
1801 suffix = os.path.splitext(path)[1]
1802 (fd, filename) = tempfile.mkstemp(suffix)
1803 self.__tempfiles.append(filename)
1804 tfp = os.fdopen(fd, 'wb')
1805 try:
1806 result = filename, headers
1807 if self.tempcache is not None:
1808 self.tempcache[url] = result
1809 bs = 1024*8
1810 size = -1
1811 read = 0
1812 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001813 if "content-length" in headers:
1814 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001815 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001816 reporthook(blocknum, bs, size)
1817 while 1:
1818 block = fp.read(bs)
1819 if not block:
1820 break
1821 read += len(block)
1822 tfp.write(block)
1823 blocknum += 1
1824 if reporthook:
1825 reporthook(blocknum, bs, size)
1826 finally:
1827 tfp.close()
1828 finally:
1829 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001830
1831 # raise exception if actual size does not match content-length header
1832 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001833 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001834 "retrieval incomplete: got only %i out of %i bytes"
1835 % (read, size), result)
1836
1837 return result
1838
1839 # Each method named open_<type> knows how to open that type of URL
1840
1841 def _open_generic_http(self, connection_factory, url, data):
1842 """Make an HTTP connection using connection_class.
1843
1844 This is an internal method that should be called from
1845 open_http() or open_https().
1846
1847 Arguments:
1848 - connection_factory should take a host name and return an
1849 HTTPConnection instance.
1850 - url is the url to retrieval or a host, relative-path pair.
1851 - data is payload for a POST request or None.
1852 """
1853
1854 user_passwd = None
1855 proxy_passwd= None
1856 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001857 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001858 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001859 user_passwd, host = splituser(host)
1860 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001861 realhost = host
1862 else:
1863 host, selector = url
1864 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001865 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001866 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001867 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001868 url = rest
1869 user_passwd = None
1870 if urltype.lower() != 'http':
1871 realhost = None
1872 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001873 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001874 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001875 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001876 if user_passwd:
1877 selector = "%s://%s%s" % (urltype, realhost, rest)
1878 if proxy_bypass(realhost):
1879 host = realhost
1880
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001881 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001882
1883 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001884 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001885 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001886 else:
1887 proxy_auth = None
1888
1889 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001890 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001891 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001892 else:
1893 auth = None
1894 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001895 headers = {}
1896 if proxy_auth:
1897 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1898 if auth:
1899 headers["Authorization"] = "Basic %s" % auth
1900 if realhost:
1901 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001902
1903 # Add Connection:close as we don't support persistent connections yet.
1904 # This helps in closing the socket and avoiding ResourceWarning
1905
1906 headers["Connection"] = "close"
1907
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001908 for header, value in self.addheaders:
1909 headers[header] = value
1910
1911 if data is not None:
1912 headers["Content-Type"] = "application/x-www-form-urlencoded"
1913 http_conn.request("POST", selector, data, headers)
1914 else:
1915 http_conn.request("GET", selector, headers=headers)
1916
1917 try:
1918 response = http_conn.getresponse()
1919 except http.client.BadStatusLine:
1920 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001921 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001922
1923 # According to RFC 2616, "2xx" code indicates that the client's
1924 # request was successfully received, understood, and accepted.
1925 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001926 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001927 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001928 else:
1929 return self.http_error(
1930 url, response.fp,
1931 response.status, response.reason, response.msg, data)
1932
1933 def open_http(self, url, data=None):
1934 """Use HTTP protocol."""
1935 return self._open_generic_http(http.client.HTTPConnection, url, data)
1936
1937 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1938 """Handle http errors.
1939
1940 Derived class can override this, or provide specific handlers
1941 named http_error_DDD where DDD is the 3-digit error code."""
1942 # First check if there's a specific handler for this error
1943 name = 'http_error_%d' % errcode
1944 if hasattr(self, name):
1945 method = getattr(self, name)
1946 if data is None:
1947 result = method(url, fp, errcode, errmsg, headers)
1948 else:
1949 result = method(url, fp, errcode, errmsg, headers, data)
1950 if result: return result
1951 return self.http_error_default(url, fp, errcode, errmsg, headers)
1952
1953 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001954 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001955 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001956 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001957
1958 if _have_ssl:
1959 def _https_connection(self, host):
1960 return http.client.HTTPSConnection(host,
1961 key_file=self.key_file,
1962 cert_file=self.cert_file)
1963
1964 def open_https(self, url, data=None):
1965 """Use HTTPS protocol."""
1966 return self._open_generic_http(self._https_connection, url, data)
1967
1968 def open_file(self, url):
1969 """Use local file or FTP depending on form of URL."""
1970 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001971 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001972 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001973 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001974 else:
1975 return self.open_local_file(url)
1976
1977 def open_local_file(self, url):
1978 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001979 import email.utils
1980 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001981 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001982 localname = url2pathname(file)
1983 try:
1984 stats = os.stat(localname)
1985 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001986 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001987 size = stats.st_size
1988 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1989 mtype = mimetypes.guess_type(url)[0]
1990 headers = email.message_from_string(
1991 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1992 (mtype or 'text/plain', size, modified))
1993 if not host:
1994 urlfile = file
1995 if file[:1] == '/':
1996 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001997 return addinfourl(open(localname, 'rb'), headers, urlfile)
1998 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001999 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07002000 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002001 urlfile = file
2002 if file[:1] == '/':
2003 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08002004 elif file[:2] == './':
2005 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00002006 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002007 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002008
2009 def open_ftp(self, url):
2010 """Use FTP protocol."""
2011 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002012 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002013 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00002014 host, path = splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002015 if not host: raise URLError('ftp error: no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00002016 host, port = splitport(host)
2017 user, host = splituser(host)
2018 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002019 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00002020 host = unquote(host)
2021 user = unquote(user or '')
2022 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002023 host = socket.gethostbyname(host)
2024 if not port:
2025 import ftplib
2026 port = ftplib.FTP_PORT
2027 else:
2028 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00002029 path, attrs = splitattr(path)
2030 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002031 dirs = path.split('/')
2032 dirs, file = dirs[:-1], dirs[-1]
2033 if dirs and not dirs[0]: dirs = dirs[1:]
2034 if dirs and not dirs[0]: dirs[0] = '/'
2035 key = user, host, port, '/'.join(dirs)
2036 # XXX thread unsafe!
2037 if len(self.ftpcache) > MAXFTPCACHE:
2038 # Prune the cache, rather arbitrarily
Benjamin Peterson3c2dca62014-06-07 15:08:04 -07002039 for k in list(self.ftpcache):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002040 if k != key:
2041 v = self.ftpcache[k]
2042 del self.ftpcache[k]
2043 v.close()
2044 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002045 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002046 self.ftpcache[key] = \
2047 ftpwrapper(user, passwd, host, port, dirs)
2048 if not file: type = 'D'
2049 else: type = 'I'
2050 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00002051 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002052 if attr.lower() == 'type' and \
2053 value in ('a', 'A', 'i', 'I', 'd', 'D'):
2054 type = value.upper()
2055 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
2056 mtype = mimetypes.guess_type("ftp:" + url)[0]
2057 headers = ""
2058 if mtype:
2059 headers += "Content-Type: %s\n" % mtype
2060 if retrlen is not None and retrlen >= 0:
2061 headers += "Content-Length: %d\n" % retrlen
2062 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00002063 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002064 except ftperrors() as exp:
2065 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002066
2067 def open_data(self, url, data=None):
2068 """Use "data" URL."""
2069 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002070 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002071 # ignore POSTed data
2072 #
2073 # syntax of data URLs:
2074 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
2075 # mediatype := [ type "/" subtype ] *( ";" parameter )
2076 # data := *urlchar
2077 # parameter := attribute "=" value
2078 try:
2079 [type, data] = url.split(',', 1)
2080 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002081 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002082 if not type:
2083 type = 'text/plain;charset=US-ASCII'
2084 semi = type.rfind(';')
2085 if semi >= 0 and '=' not in type[semi:]:
2086 encoding = type[semi+1:]
2087 type = type[:semi]
2088 else:
2089 encoding = ''
2090 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00002091 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002092 time.gmtime(time.time())))
2093 msg.append('Content-type: %s' % type)
2094 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00002095 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00002096 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002097 else:
Georg Brandl13e89462008-07-01 19:56:00 +00002098 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002099 msg.append('Content-Length: %d' % len(data))
2100 msg.append('')
2101 msg.append(data)
2102 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00002103 headers = email.message_from_string(msg)
2104 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002105 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00002106 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002107
2108
2109class FancyURLopener(URLopener):
2110 """Derived class with handlers for errors we can handle (perhaps)."""
2111
2112 def __init__(self, *args, **kwargs):
2113 URLopener.__init__(self, *args, **kwargs)
2114 self.auth_cache = {}
2115 self.tries = 0
2116 self.maxtries = 10
2117
2118 def http_error_default(self, url, fp, errcode, errmsg, headers):
2119 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00002120 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002121
2122 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
2123 """Error 302 -- relocated (temporarily)."""
2124 self.tries += 1
Martin Pantera0370222016-02-04 06:01:35 +00002125 try:
2126 if self.maxtries and self.tries >= self.maxtries:
2127 if hasattr(self, "http_error_500"):
2128 meth = self.http_error_500
2129 else:
2130 meth = self.http_error_default
2131 return meth(url, fp, 500,
2132 "Internal Server Error: Redirect Recursion",
2133 headers)
2134 result = self.redirect_internal(url, fp, errcode, errmsg,
2135 headers, data)
2136 return result
2137 finally:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002138 self.tries = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002139
2140 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
2141 if 'location' in headers:
2142 newurl = headers['location']
2143 elif 'uri' in headers:
2144 newurl = headers['uri']
2145 else:
2146 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002147 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002148
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002149 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002150 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002151
2152 urlparts = urlparse(newurl)
2153
2154 # For security reasons, we don't allow redirection to anything other
2155 # than http, https and ftp.
2156
2157 # We are using newer HTTPError with older redirect_internal method
2158 # This older method will get deprecated in 3.3
2159
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002160 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002161 raise HTTPError(newurl, errcode,
2162 errmsg +
2163 " Redirection to url '%s' is not allowed." % newurl,
2164 headers, fp)
2165
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002166 return self.open(newurl)
2167
2168 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2169 """Error 301 -- also relocated (permanently)."""
2170 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2171
2172 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2173 """Error 303 -- also relocated (essentially identical to 302)."""
2174 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2175
2176 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2177 """Error 307 -- relocated, but turn POST into error."""
2178 if data is None:
2179 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2180 else:
2181 return self.http_error_default(url, fp, errcode, errmsg, headers)
2182
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002183 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2184 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002185 """Error 401 -- authentication required.
2186 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002187 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002188 URLopener.http_error_default(self, url, fp,
2189 errcode, errmsg, headers)
2190 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002191 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2192 if not match:
2193 URLopener.http_error_default(self, url, fp,
2194 errcode, errmsg, headers)
2195 scheme, realm = match.groups()
2196 if scheme.lower() != 'basic':
2197 URLopener.http_error_default(self, url, fp,
2198 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002199 if not retry:
2200 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2201 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002202 name = 'retry_' + self.type + '_basic_auth'
2203 if data is None:
2204 return getattr(self,name)(url, realm)
2205 else:
2206 return getattr(self,name)(url, realm, data)
2207
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002208 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2209 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002210 """Error 407 -- proxy authentication required.
2211 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002212 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002213 URLopener.http_error_default(self, url, fp,
2214 errcode, errmsg, headers)
2215 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002216 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2217 if not match:
2218 URLopener.http_error_default(self, url, fp,
2219 errcode, errmsg, headers)
2220 scheme, realm = match.groups()
2221 if scheme.lower() != 'basic':
2222 URLopener.http_error_default(self, url, fp,
2223 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002224 if not retry:
2225 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2226 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002227 name = 'retry_proxy_' + self.type + '_basic_auth'
2228 if data is None:
2229 return getattr(self,name)(url, realm)
2230 else:
2231 return getattr(self,name)(url, realm, data)
2232
2233 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002234 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002235 newurl = 'http://' + host + selector
2236 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002237 urltype, proxyhost = splittype(proxy)
2238 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002239 i = proxyhost.find('@') + 1
2240 proxyhost = proxyhost[i:]
2241 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2242 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002243 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002244 quote(passwd, safe=''), proxyhost)
2245 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2246 if data is None:
2247 return self.open(newurl)
2248 else:
2249 return self.open(newurl, data)
2250
2251 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002252 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002253 newurl = 'https://' + host + selector
2254 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002255 urltype, proxyhost = splittype(proxy)
2256 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002257 i = proxyhost.find('@') + 1
2258 proxyhost = proxyhost[i:]
2259 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2260 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002261 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002262 quote(passwd, safe=''), proxyhost)
2263 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2264 if data is None:
2265 return self.open(newurl)
2266 else:
2267 return self.open(newurl, data)
2268
2269 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002270 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002271 i = host.find('@') + 1
2272 host = host[i:]
2273 user, passwd = self.get_user_passwd(host, realm, i)
2274 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002275 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002276 quote(passwd, safe=''), host)
2277 newurl = 'http://' + host + selector
2278 if data is None:
2279 return self.open(newurl)
2280 else:
2281 return self.open(newurl, data)
2282
2283 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002284 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002285 i = host.find('@') + 1
2286 host = host[i:]
2287 user, passwd = self.get_user_passwd(host, realm, i)
2288 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002289 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002290 quote(passwd, safe=''), host)
2291 newurl = 'https://' + host + selector
2292 if data is None:
2293 return self.open(newurl)
2294 else:
2295 return self.open(newurl, data)
2296
Florent Xicluna757445b2010-05-17 17:24:07 +00002297 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002298 key = realm + '@' + host.lower()
2299 if key in self.auth_cache:
2300 if clear_cache:
2301 del self.auth_cache[key]
2302 else:
2303 return self.auth_cache[key]
2304 user, passwd = self.prompt_user_passwd(host, realm)
2305 if user or passwd: self.auth_cache[key] = (user, passwd)
2306 return user, passwd
2307
2308 def prompt_user_passwd(self, host, realm):
2309 """Override this in a GUI environment!"""
2310 import getpass
2311 try:
2312 user = input("Enter username for %s at %s: " % (realm, host))
2313 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2314 (user, realm, host))
2315 return user, passwd
2316 except KeyboardInterrupt:
2317 print()
2318 return None, None
2319
2320
2321# Utility functions
2322
2323_localhost = None
2324def localhost():
2325 """Return the IP address of the magic hostname 'localhost'."""
2326 global _localhost
2327 if _localhost is None:
2328 _localhost = socket.gethostbyname('localhost')
2329 return _localhost
2330
2331_thishost = None
2332def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002333 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002334 global _thishost
2335 if _thishost is None:
Senthil Kumarandcdadfe2013-06-01 11:12:17 -07002336 try:
2337 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
2338 except socket.gaierror:
2339 _thishost = tuple(socket.gethostbyname_ex('localhost')[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002340 return _thishost
2341
2342_ftperrors = None
2343def ftperrors():
2344 """Return the set of errors raised by the FTP class."""
2345 global _ftperrors
2346 if _ftperrors is None:
2347 import ftplib
2348 _ftperrors = ftplib.all_errors
2349 return _ftperrors
2350
2351_noheaders = None
2352def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002353 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002354 global _noheaders
2355 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002356 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002357 return _noheaders
2358
2359
2360# Utility classes
2361
2362class ftpwrapper:
2363 """Class used by open_ftp() for cache of open FTP connections."""
2364
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002365 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2366 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002367 self.user = user
2368 self.passwd = passwd
2369 self.host = host
2370 self.port = port
2371 self.dirs = dirs
2372 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002373 self.refcount = 0
2374 self.keepalive = persistent
Victor Stinnerab73e652015-04-07 12:49:27 +02002375 try:
2376 self.init()
2377 except:
2378 self.close()
2379 raise
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002380
2381 def init(self):
2382 import ftplib
2383 self.busy = 0
2384 self.ftp = ftplib.FTP()
2385 self.ftp.connect(self.host, self.port, self.timeout)
2386 self.ftp.login(self.user, self.passwd)
Senthil Kumarancaa00fe2013-06-02 11:59:47 -07002387 _target = '/'.join(self.dirs)
2388 self.ftp.cwd(_target)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002389
2390 def retrfile(self, file, type):
2391 import ftplib
2392 self.endtransfer()
2393 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2394 else: cmd = 'TYPE ' + type; isdir = 0
2395 try:
2396 self.ftp.voidcmd(cmd)
2397 except ftplib.all_errors:
2398 self.init()
2399 self.ftp.voidcmd(cmd)
2400 conn = None
2401 if file and not isdir:
2402 # Try to retrieve as a file
2403 try:
2404 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002405 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002406 except ftplib.error_perm as reason:
2407 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002408 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002409 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002410 if not conn:
2411 # Set transfer mode to ASCII!
2412 self.ftp.voidcmd('TYPE A')
2413 # Try a directory listing. Verify that directory exists.
2414 if file:
2415 pwd = self.ftp.pwd()
2416 try:
2417 try:
2418 self.ftp.cwd(file)
2419 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002420 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002421 finally:
2422 self.ftp.cwd(pwd)
2423 cmd = 'LIST ' + file
2424 else:
2425 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002426 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002427 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002428
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002429 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2430 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002431 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002432 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002433 return (ftpobj, retrlen)
2434
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002435 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002436 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002437
2438 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002439 self.keepalive = False
2440 if self.refcount <= 0:
2441 self.real_close()
2442
2443 def file_close(self):
2444 self.endtransfer()
2445 self.refcount -= 1
2446 if self.refcount <= 0 and not self.keepalive:
2447 self.real_close()
2448
2449 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002450 self.endtransfer()
2451 try:
2452 self.ftp.close()
2453 except ftperrors():
2454 pass
2455
2456# Proxy handling
2457def getproxies_environment():
2458 """Return a dictionary of scheme -> proxy server URL mappings.
2459
2460 Scan the environment for variables named <scheme>_proxy;
2461 this seems to be the standard convention. If you need a
2462 different way, you can pass a proxies dictionary to the
2463 [Fancy]URLopener constructor.
2464
2465 """
2466 proxies = {}
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002467 # in order to prefer lowercase variables, process environment in
2468 # two passes: first matches any, second pass matches lowercase only
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002469 for name, value in os.environ.items():
2470 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002471 if value and name[-6:] == '_proxy':
2472 proxies[name[:-6]] = value
Senthil Kumaran4cbb23f2016-07-30 23:24:16 -07002473 # CVE-2016-1000110 - If we are running as CGI script, forget HTTP_PROXY
2474 # (non-all-lowercase) as it may be set from the web server by a "Proxy:"
2475 # header from the client
Senthil Kumaran17742f22016-07-30 23:39:06 -07002476 # If "proxy" is lowercase, it will still be used thanks to the next block
Senthil Kumaran4cbb23f2016-07-30 23:24:16 -07002477 if 'REQUEST_METHOD' in os.environ:
2478 proxies.pop('http', None)
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002479 for name, value in os.environ.items():
2480 if name[-6:] == '_proxy':
2481 name = name.lower()
2482 if value:
2483 proxies[name[:-6]] = value
2484 else:
2485 proxies.pop(name[:-6], None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002486 return proxies
2487
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002488def proxy_bypass_environment(host, proxies=None):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002489 """Test if proxies should not be used for a particular host.
2490
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002491 Checks the proxy dict for the value of no_proxy, which should
2492 be a list of comma separated DNS suffixes, or '*' for all hosts.
2493
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002494 """
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002495 if proxies is None:
2496 proxies = getproxies_environment()
2497 # don't bypass, if no_proxy isn't specified
2498 try:
2499 no_proxy = proxies['no']
2500 except KeyError:
2501 return 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002502 # '*' is special case for always bypass
2503 if no_proxy == '*':
2504 return 1
2505 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002506 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002507 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002508 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2509 for name in no_proxy_list:
Martin Panteraa279822016-04-30 01:03:40 +00002510 if name:
Xiang Zhang959ff7f2017-01-09 11:47:55 +08002511 name = name.lstrip('.') # ignore leading dots
Martin Panteraa279822016-04-30 01:03:40 +00002512 name = re.escape(name)
2513 pattern = r'(.+\.)?%s$' % name
2514 if (re.match(pattern, hostonly, re.I)
2515 or re.match(pattern, host, re.I)):
2516 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002517 # otherwise, don't bypass
2518 return 0
2519
2520
Ronald Oussorene72e1612011-03-14 18:15:25 -04002521# This code tests an OSX specific data structure but is testable on all
2522# platforms
2523def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2524 """
2525 Return True iff this host shouldn't be accessed using a proxy
2526
2527 This function uses the MacOSX framework SystemConfiguration
2528 to fetch the proxy information.
2529
2530 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2531 { 'exclude_simple': bool,
2532 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2533 }
2534 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002535 from fnmatch import fnmatch
2536
2537 hostonly, port = splitport(host)
2538
2539 def ip2num(ipAddr):
2540 parts = ipAddr.split('.')
2541 parts = list(map(int, parts))
2542 if len(parts) != 4:
2543 parts = (parts + [0, 0, 0, 0])[:4]
2544 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2545
2546 # Check for simple host names:
2547 if '.' not in host:
2548 if proxy_settings['exclude_simple']:
2549 return True
2550
2551 hostIP = None
2552
2553 for value in proxy_settings.get('exceptions', ()):
2554 # Items in the list are strings like these: *.local, 169.254/16
2555 if not value: continue
2556
2557 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2558 if m is not None:
2559 if hostIP is None:
2560 try:
2561 hostIP = socket.gethostbyname(hostonly)
2562 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002563 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002564 continue
2565
2566 base = ip2num(m.group(1))
2567 mask = m.group(2)
2568 if mask is None:
2569 mask = 8 * (m.group(1).count('.') + 1)
2570 else:
2571 mask = int(mask[1:])
2572 mask = 32 - mask
2573
2574 if (hostIP >> mask) == (base >> mask):
2575 return True
2576
2577 elif fnmatch(host, value):
2578 return True
2579
2580 return False
2581
2582
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002583if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002584 from _scproxy import _get_proxy_settings, _get_proxies
2585
2586 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002587 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002588 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002589
2590 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002591 """Return a dictionary of scheme -> proxy server URL mappings.
2592
Ronald Oussoren84151202010-04-18 20:46:11 +00002593 This function uses the MacOSX framework SystemConfiguration
2594 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002595 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002596 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002597
Ronald Oussoren84151202010-04-18 20:46:11 +00002598
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002599
2600 def proxy_bypass(host):
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002601 """Return True, if host should be bypassed.
2602
2603 Checks proxy settings gathered from the environment, if specified,
2604 or from the MacOSX framework SystemConfiguration.
2605
2606 """
2607 proxies = getproxies_environment()
2608 if proxies:
2609 return proxy_bypass_environment(host, proxies)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002610 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002611 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002612
2613 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002614 return getproxies_environment() or getproxies_macosx_sysconf()
2615
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002616
2617elif os.name == 'nt':
2618 def getproxies_registry():
2619 """Return a dictionary of scheme -> proxy server URL mappings.
2620
2621 Win32 uses the registry to store proxies.
2622
2623 """
2624 proxies = {}
2625 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002626 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002627 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002628 # Std module, so should be around - but you never know!
2629 return proxies
2630 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002631 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002632 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002633 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002634 'ProxyEnable')[0]
2635 if proxyEnable:
2636 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002637 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002638 'ProxyServer')[0])
2639 if '=' in proxyServer:
2640 # Per-protocol settings
2641 for p in proxyServer.split(';'):
2642 protocol, address = p.split('=', 1)
2643 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002644 if not re.match('^([^/:]+)://', address):
2645 address = '%s://%s' % (protocol, address)
2646 proxies[protocol] = address
2647 else:
2648 # Use one setting for all protocols
2649 if proxyServer[:5] == 'http:':
2650 proxies['http'] = proxyServer
2651 else:
2652 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002653 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002654 proxies['ftp'] = 'ftp://%s' % proxyServer
2655 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002656 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002657 # Either registry key not found etc, or the value in an
2658 # unexpected format.
2659 # proxies already set up to be empty so nothing to do
2660 pass
2661 return proxies
2662
2663 def getproxies():
2664 """Return a dictionary of scheme -> proxy server URL mappings.
2665
2666 Returns settings gathered from the environment, if specified,
2667 or the registry.
2668
2669 """
2670 return getproxies_environment() or getproxies_registry()
2671
2672 def proxy_bypass_registry(host):
2673 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002674 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002675 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002676 # Std modules, so should be around - but you never know!
2677 return 0
2678 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002679 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002680 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002681 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002682 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002683 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002684 'ProxyOverride')[0])
2685 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002686 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002687 return 0
2688 if not proxyEnable or not proxyOverride:
2689 return 0
2690 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002691 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002692 host = [rawHost]
2693 try:
2694 addr = socket.gethostbyname(rawHost)
2695 if addr != rawHost:
2696 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002697 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002698 pass
2699 try:
2700 fqdn = socket.getfqdn(rawHost)
2701 if fqdn != rawHost:
2702 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002703 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002704 pass
2705 # make a check value list from the registry entry: replace the
2706 # '<local>' string by the localhost entry and the corresponding
2707 # canonical entry.
2708 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002709 # now check if we match one of the registry values.
2710 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002711 if test == '<local>':
2712 if '.' not in rawHost:
2713 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002714 test = test.replace(".", r"\.") # mask dots
2715 test = test.replace("*", r".*") # change glob sequence
2716 test = test.replace("?", r".") # change glob char
2717 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002718 if re.match(test, val, re.I):
2719 return 1
2720 return 0
2721
2722 def proxy_bypass(host):
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002723 """Return True, if host should be bypassed.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002724
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002725 Checks proxy settings gathered from the environment, if specified,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002726 or the registry.
2727
2728 """
Senthil Kumarana7c0ff22016-04-25 08:16:23 -07002729 proxies = getproxies_environment()
2730 if proxies:
2731 return proxy_bypass_environment(host, proxies)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002732 else:
2733 return proxy_bypass_registry(host)
2734
2735else:
2736 # By default use environment variables
2737 getproxies = getproxies_environment
2738 proxy_bypass = proxy_bypass_environment