blob: fc8ef7f91b09fbb8ce59827ff6301c0153eef9d1 [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
94import sys
95import time
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000096import collections
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
136__version__ = sys.version[:3]
137
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):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000141 global _opener
Antoine Pitroude9ac6c2012-05-16 21:40:01 +0200142 if cafile or capath or cadefault:
Senthil Kumarana5c85b32014-09-19 15:23:30 +0800143 if context is not None:
144 raise ValueError(
145 "You can't pass both context and any of cafile, capath, and "
146 "cadefault"
147 )
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000148 if not _have_ssl:
149 raise ValueError('SSL support not available')
Benjamin Petersonb6666972014-12-07 13:46:02 -0500150 context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH,
Christian Heimes67986f92013-11-23 22:43:47 +0100151 cafile=cafile,
152 capath=capath)
Benjamin Petersonb6666972014-12-07 13:46:02 -0500153 https_handler = HTTPSHandler(context=context)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000154 opener = build_opener(https_handler)
Senthil Kumarana5c85b32014-09-19 15:23:30 +0800155 elif context:
156 https_handler = HTTPSHandler(context=context)
157 opener = build_opener(https_handler)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000158 elif _opener is None:
159 _opener = opener = build_opener()
160 else:
161 opener = _opener
162 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000163
164def install_opener(opener):
165 global _opener
166 _opener = opener
167
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700168_url_tempfiles = []
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000169def urlretrieve(url, filename=None, reporthook=None, data=None):
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700170 """
171 Retrieve a URL into a temporary location on disk.
172
173 Requires a URL argument. If a filename is passed, it is used as
174 the temporary file location. The reporthook argument should be
175 a callable that accepts a block number, a read size, and the
176 total file size of the URL target. The data argument should be
177 valid URL encoded data.
178
179 If a filename is passed and the URL points to a local resource,
180 the result is a copy from local file to new file.
181
182 Returns a tuple containing the path to the newly created
183 data file as well as the resulting HTTPMessage object.
184 """
185 url_type, path = splittype(url)
186
187 with contextlib.closing(urlopen(url, data)) as fp:
188 headers = fp.info()
189
190 # Just return the local path and the "headers" for file://
191 # URLs. No sense in performing a copy unless requested.
192 if url_type == "file" and not filename:
193 return os.path.normpath(path), headers
194
195 # Handle temporary file setup.
196 if filename:
197 tfp = open(filename, 'wb')
198 else:
199 tfp = tempfile.NamedTemporaryFile(delete=False)
200 filename = tfp.name
201 _url_tempfiles.append(filename)
202
203 with tfp:
204 result = filename, headers
205 bs = 1024*8
206 size = -1
207 read = 0
208 blocknum = 0
209 if "content-length" in headers:
210 size = int(headers["Content-Length"])
211
212 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800213 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700214
215 while True:
216 block = fp.read(bs)
217 if not block:
218 break
219 read += len(block)
220 tfp.write(block)
221 blocknum += 1
222 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800223 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700224
225 if size >= 0 and read < size:
226 raise ContentTooShortError(
227 "retrieval incomplete: got only %i out of %i bytes"
228 % (read, size), result)
229
230 return result
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000231
232def urlcleanup():
Robert Collins2fee5c92015-08-04 12:52:06 +1200233 """Clean up temporary files from urlretrieve calls."""
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700234 for temp_file in _url_tempfiles:
235 try:
236 os.unlink(temp_file)
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200237 except OSError:
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700238 pass
239
240 del _url_tempfiles[:]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000241 global _opener
242 if _opener:
243 _opener = None
244
245# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000246_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000247def request_host(request):
248 """Return request-host, as defined by RFC 2965.
249
250 Variation from RFC: returned value is lowercased, for convenient
251 comparison.
252
253 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000254 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000255 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000256 if host == "":
257 host = request.get_header("Host", "")
258
259 # remove port, if present
260 host = _cut_port_re.sub("", host, 1)
261 return host.lower()
262
263class Request:
264
265 def __init__(self, url, data=None, headers={},
Senthil Kumarande49d642011-10-16 23:54:44 +0800266 origin_req_host=None, unverifiable=False,
267 method=None):
Senthil Kumaran52380922013-04-25 05:45:48 -0700268 self.full_url = url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000269 self.headers = {}
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200270 self.unredirected_hdrs = {}
271 self._data = None
272 self.data = data
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000273 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000274 for key, value in headers.items():
275 self.add_header(key, value)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000276 if origin_req_host is None:
277 origin_req_host = request_host(self)
278 self.origin_req_host = origin_req_host
279 self.unverifiable = unverifiable
Jason R. Coombs7dc4f4b2013-09-08 12:47:07 -0400280 if method:
281 self.method = method
Senthil Kumaran52380922013-04-25 05:45:48 -0700282
283 @property
284 def full_url(self):
Senthil Kumaran83070752013-05-24 09:14:12 -0700285 if self.fragment:
286 return '{}#{}'.format(self._full_url, self.fragment)
Senthil Kumaran52380922013-04-25 05:45:48 -0700287 return self._full_url
288
289 @full_url.setter
290 def full_url(self, url):
291 # unwrap('<URL:type://host/path>') --> 'type://host/path'
292 self._full_url = unwrap(url)
293 self._full_url, self.fragment = splittag(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000294 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000295
Senthil Kumaran52380922013-04-25 05:45:48 -0700296 @full_url.deleter
297 def full_url(self):
298 self._full_url = None
299 self.fragment = None
300 self.selector = ''
301
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200302 @property
303 def data(self):
304 return self._data
305
306 @data.setter
307 def data(self, data):
308 if data != self._data:
309 self._data = data
310 # issue 16464
311 # if we change data we need to remove content-length header
312 # (cause it's most probably calculated for previous value)
313 if self.has_header("Content-length"):
314 self.remove_header("Content-length")
315
316 @data.deleter
317 def data(self):
R David Murray9cc7d452013-03-20 00:10:51 -0400318 self.data = None
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200319
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000320 def _parse(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700321 self.type, rest = splittype(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000322 if self.type is None:
R David Murrayd8a46962013-04-03 06:58:34 -0400323 raise ValueError("unknown url type: %r" % self.full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000324 self.host, self.selector = splithost(rest)
325 if self.host:
326 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000327
328 def get_method(self):
Senthil Kumarande49d642011-10-16 23:54:44 +0800329 """Return a string indicating the HTTP request method."""
Jason R. Coombsaae6a1d2013-09-08 12:54:33 -0400330 default_method = "POST" if self.data is not None else "GET"
331 return getattr(self, 'method', default_method)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000332
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000333 def get_full_url(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700334 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000335
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000336 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000337 if self.type == 'https' and not self._tunnel_host:
338 self._tunnel_host = self.host
339 else:
340 self.type= type
341 self.selector = self.full_url
342 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000343
344 def has_proxy(self):
345 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000346
347 def add_header(self, key, val):
348 # useful for something like authentication
349 self.headers[key.capitalize()] = val
350
351 def add_unredirected_header(self, key, val):
352 # will not be added to a redirected request
353 self.unredirected_hdrs[key.capitalize()] = val
354
355 def has_header(self, header_name):
356 return (header_name in self.headers or
357 header_name in self.unredirected_hdrs)
358
359 def get_header(self, header_name, default=None):
360 return self.headers.get(
361 header_name,
362 self.unredirected_hdrs.get(header_name, default))
363
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200364 def remove_header(self, header_name):
365 self.headers.pop(header_name, None)
366 self.unredirected_hdrs.pop(header_name, None)
367
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000368 def header_items(self):
369 hdrs = self.unredirected_hdrs.copy()
370 hdrs.update(self.headers)
371 return list(hdrs.items())
372
373class OpenerDirector:
374 def __init__(self):
375 client_version = "Python-urllib/%s" % __version__
376 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000377 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000378 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000379 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000380 self.handle_open = {}
381 self.handle_error = {}
382 self.process_response = {}
383 self.process_request = {}
384
385 def add_handler(self, handler):
386 if not hasattr(handler, "add_parent"):
387 raise TypeError("expected BaseHandler instance, got %r" %
388 type(handler))
389
390 added = False
391 for meth in dir(handler):
392 if meth in ["redirect_request", "do_open", "proxy_open"]:
393 # oops, coincidental match
394 continue
395
396 i = meth.find("_")
397 protocol = meth[:i]
398 condition = meth[i+1:]
399
400 if condition.startswith("error"):
401 j = condition.find("_") + i + 1
402 kind = meth[j+1:]
403 try:
404 kind = int(kind)
405 except ValueError:
406 pass
407 lookup = self.handle_error.get(protocol, {})
408 self.handle_error[protocol] = lookup
409 elif condition == "open":
410 kind = protocol
411 lookup = self.handle_open
412 elif condition == "response":
413 kind = protocol
414 lookup = self.process_response
415 elif condition == "request":
416 kind = protocol
417 lookup = self.process_request
418 else:
419 continue
420
421 handlers = lookup.setdefault(kind, [])
422 if handlers:
423 bisect.insort(handlers, handler)
424 else:
425 handlers.append(handler)
426 added = True
427
428 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000429 bisect.insort(self.handlers, handler)
430 handler.add_parent(self)
431
432 def close(self):
433 # Only exists for backwards compatibility.
434 pass
435
436 def _call_chain(self, chain, kind, meth_name, *args):
437 # Handlers raise an exception if no one else should try to handle
438 # the request, or return None if they can't but another handler
439 # could. Otherwise, they return the response.
440 handlers = chain.get(kind, ())
441 for handler in handlers:
442 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000443 result = func(*args)
444 if result is not None:
445 return result
446
447 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
448 # accept a URL or a Request object
449 if isinstance(fullurl, str):
450 req = Request(fullurl, data)
451 else:
452 req = fullurl
453 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000454 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000455
456 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000457 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000458
459 # pre-process request
460 meth_name = protocol+"_request"
461 for processor in self.process_request.get(protocol, []):
462 meth = getattr(processor, meth_name)
463 req = meth(req)
464
465 response = self._open(req, data)
466
467 # post-process response
468 meth_name = protocol+"_response"
469 for processor in self.process_response.get(protocol, []):
470 meth = getattr(processor, meth_name)
471 response = meth(req, response)
472
473 return response
474
475 def _open(self, req, data=None):
476 result = self._call_chain(self.handle_open, 'default',
477 'default_open', req)
478 if result:
479 return result
480
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000481 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000482 result = self._call_chain(self.handle_open, protocol, protocol +
483 '_open', req)
484 if result:
485 return result
486
487 return self._call_chain(self.handle_open, 'unknown',
488 'unknown_open', req)
489
490 def error(self, proto, *args):
491 if proto in ('http', 'https'):
492 # XXX http[s] protocols are special-cased
493 dict = self.handle_error['http'] # https is not different than http
494 proto = args[2] # YUCK!
495 meth_name = 'http_error_%s' % proto
496 http_err = 1
497 orig_args = args
498 else:
499 dict = self.handle_error
500 meth_name = proto + '_error'
501 http_err = 0
502 args = (dict, proto, meth_name) + args
503 result = self._call_chain(*args)
504 if result:
505 return result
506
507 if http_err:
508 args = (dict, 'default', 'http_error_default') + orig_args
509 return self._call_chain(*args)
510
511# XXX probably also want an abstract factory that knows when it makes
512# sense to skip a superclass in favor of a subclass and when it might
513# make sense to include both
514
515def build_opener(*handlers):
516 """Create an opener object from a list of handlers.
517
518 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000519 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000520
521 If any of the handlers passed as arguments are subclasses of the
522 default handlers, the default handlers will not be used.
523 """
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000524 opener = OpenerDirector()
525 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
526 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100527 FTPHandler, FileHandler, HTTPErrorProcessor,
528 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000529 if hasattr(http.client, "HTTPSConnection"):
530 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000531 skip = set()
532 for klass in default_classes:
533 for check in handlers:
Benjamin Peterson78c85382014-04-01 16:27:30 -0400534 if isinstance(check, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000535 if issubclass(check, klass):
536 skip.add(klass)
537 elif isinstance(check, klass):
538 skip.add(klass)
539 for klass in skip:
540 default_classes.remove(klass)
541
542 for klass in default_classes:
543 opener.add_handler(klass())
544
545 for h in handlers:
Benjamin Peterson5dd3cae2014-04-01 14:20:56 -0400546 if isinstance(h, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000547 h = h()
548 opener.add_handler(h)
549 return opener
550
551class BaseHandler:
552 handler_order = 500
553
554 def add_parent(self, parent):
555 self.parent = parent
556
557 def close(self):
558 # Only exists for backwards compatibility
559 pass
560
561 def __lt__(self, other):
562 if not hasattr(other, "handler_order"):
563 # Try to preserve the old behavior of having custom classes
564 # inserted after default ones (works only for custom user
565 # classes which are not aware of handler_order).
566 return True
567 return self.handler_order < other.handler_order
568
569
570class HTTPErrorProcessor(BaseHandler):
571 """Process HTTP error responses."""
572 handler_order = 1000 # after all other processing
573
574 def http_response(self, request, response):
575 code, msg, hdrs = response.code, response.msg, response.info()
576
577 # According to RFC 2616, "2xx" code indicates that the client's
578 # request was successfully received, understood, and accepted.
579 if not (200 <= code < 300):
580 response = self.parent.error(
581 'http', request, response, code, msg, hdrs)
582
583 return response
584
585 https_response = http_response
586
587class HTTPDefaultErrorHandler(BaseHandler):
588 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000589 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000590
591class HTTPRedirectHandler(BaseHandler):
592 # maximum number of redirections to any single URL
593 # this is needed because of the state that cookies introduce
594 max_repeats = 4
595 # maximum total number of redirections (regardless of URL) before
596 # assuming we're in a loop
597 max_redirections = 10
598
599 def redirect_request(self, req, fp, code, msg, headers, newurl):
600 """Return a Request or None in response to a redirect.
601
602 This is called by the http_error_30x methods when a
603 redirection response is received. If a redirection should
604 take place, return a new Request to allow http_error_30x to
605 perform the redirect. Otherwise, raise HTTPError if no-one
606 else should try to handle this url. Return None if you can't
607 but another Handler might.
608 """
609 m = req.get_method()
610 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
611 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000612 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000613
614 # Strictly (according to RFC 2616), 301 or 302 in response to
615 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000616 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000617 # essentially all clients do redirect in this case, so we do
618 # the same.
619 # be conciliant with URIs containing a space
620 newurl = newurl.replace(' ', '%20')
621 CONTENT_HEADERS = ("content-length", "content-type")
622 newheaders = dict((k, v) for k, v in req.headers.items()
623 if k.lower() not in CONTENT_HEADERS)
624 return Request(newurl,
625 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000626 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000627 unverifiable=True)
628
629 # Implementation note: To avoid the server sending us into an
630 # infinite loop, the request object needs to track what URLs we
631 # have already seen. Do this by adding a handler-specific
632 # attribute to the Request object.
633 def http_error_302(self, req, fp, code, msg, headers):
634 # Some servers (incorrectly) return multiple Location headers
635 # (so probably same goes for URI). Use first header.
636 if "location" in headers:
637 newurl = headers["location"]
638 elif "uri" in headers:
639 newurl = headers["uri"]
640 else:
641 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000642
643 # fix a possible malformed URL
644 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700645
646 # For security reasons we don't allow redirection to anything other
647 # than http, https or ftp.
648
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800649 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800650 raise HTTPError(
651 newurl, code,
652 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
653 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700654
Facundo Batistaf24802c2008-08-17 03:36:03 +0000655 if not urlparts.path:
656 urlparts = list(urlparts)
657 urlparts[2] = "/"
658 newurl = urlunparse(urlparts)
659
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000660 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000661
662 # XXX Probably want to forget about the state of the current
663 # request, although that might interact poorly with other
664 # handlers that also use handler-specific request attributes
665 new = self.redirect_request(req, fp, code, msg, headers, newurl)
666 if new is None:
667 return
668
669 # loop detection
670 # .redirect_dict has a key url if url was previously visited.
671 if hasattr(req, 'redirect_dict'):
672 visited = new.redirect_dict = req.redirect_dict
673 if (visited.get(newurl, 0) >= self.max_repeats or
674 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000675 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000676 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000677 else:
678 visited = new.redirect_dict = req.redirect_dict = {}
679 visited[newurl] = visited.get(newurl, 0) + 1
680
681 # Don't close the fp until we are sure that we won't use it
682 # with HTTPError.
683 fp.read()
684 fp.close()
685
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000686 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000687
688 http_error_301 = http_error_303 = http_error_307 = http_error_302
689
690 inf_msg = "The HTTP server returned a redirect error that would " \
691 "lead to an infinite loop.\n" \
692 "The last 30x error message was:\n"
693
694
695def _parse_proxy(proxy):
696 """Return (scheme, user, password, host/port) given a URL or an authority.
697
698 If a URL is supplied, it must have an authority (host:port) component.
699 According to RFC 3986, having an authority component means the URL must
Senthil Kumarand8e24f12014-04-14 16:32:20 -0400700 have two slashes after the scheme.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000701 """
Georg Brandl13e89462008-07-01 19:56:00 +0000702 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000703 if not r_scheme.startswith("/"):
704 # authority
705 scheme = None
706 authority = proxy
707 else:
708 # URL
709 if not r_scheme.startswith("//"):
710 raise ValueError("proxy URL with no authority: %r" % proxy)
711 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
712 # and 3.3.), path is empty or starts with '/'
713 end = r_scheme.find("/", 2)
714 if end == -1:
715 end = None
716 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000717 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000718 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000719 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000720 else:
721 user = password = None
722 return scheme, user, password, hostport
723
724class ProxyHandler(BaseHandler):
725 # Proxies must be in front
726 handler_order = 100
727
728 def __init__(self, proxies=None):
729 if proxies is None:
730 proxies = getproxies()
731 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
732 self.proxies = proxies
733 for type, url in proxies.items():
734 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200735 lambda r, proxy=url, type=type, meth=self.proxy_open:
736 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000737
738 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000739 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000740 proxy_type, user, password, hostport = _parse_proxy(proxy)
741 if proxy_type is None:
742 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000743
744 if req.host and proxy_bypass(req.host):
745 return None
746
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000747 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000748 user_pass = '%s:%s' % (unquote(user),
749 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000750 creds = base64.b64encode(user_pass.encode()).decode("ascii")
751 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000752 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000753 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000754 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000755 # let other handlers take care of it
756 return None
757 else:
758 # need to start over, because the other handlers don't
759 # grok the proxy's URL type
760 # e.g. if we have a constructor arg proxies like so:
761 # {'http': 'ftp://proxy.example.com'}, we may end up turning
762 # a request for http://acme.example.com/a into one for
763 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000764 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000765
766class HTTPPasswordMgr:
767
768 def __init__(self):
769 self.passwd = {}
770
771 def add_password(self, realm, uri, user, passwd):
772 # uri could be a single URI or a sequence
773 if isinstance(uri, str):
774 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800775 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000776 self.passwd[realm] = {}
777 for default_port in True, False:
778 reduced_uri = tuple(
779 [self.reduce_uri(u, default_port) for u in uri])
780 self.passwd[realm][reduced_uri] = (user, passwd)
781
782 def find_user_password(self, realm, authuri):
783 domains = self.passwd.get(realm, {})
784 for default_port in True, False:
785 reduced_authuri = self.reduce_uri(authuri, default_port)
786 for uris, authinfo in domains.items():
787 for uri in uris:
788 if self.is_suburi(uri, reduced_authuri):
789 return authinfo
790 return None, None
791
792 def reduce_uri(self, uri, default_port=True):
793 """Accept authority or URI and extract only the authority and path."""
794 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000795 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000796 if parts[1]:
797 # URI
798 scheme = parts[0]
799 authority = parts[1]
800 path = parts[2] or '/'
801 else:
802 # host or host:port
803 scheme = None
804 authority = uri
805 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000806 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000807 if default_port and port is None and scheme is not None:
808 dport = {"http": 80,
809 "https": 443,
810 }.get(scheme)
811 if dport is not None:
812 authority = "%s:%d" % (host, dport)
813 return authority, path
814
815 def is_suburi(self, base, test):
816 """Check if test is below base in a URI tree
817
818 Both args must be URIs in reduced form.
819 """
820 if base == test:
821 return True
822 if base[0] != test[0]:
823 return False
824 common = posixpath.commonprefix((base[1], test[1]))
825 if len(common) == len(base[1]):
826 return True
827 return False
828
829
830class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
831
832 def find_user_password(self, realm, authuri):
833 user, password = HTTPPasswordMgr.find_user_password(self, realm,
834 authuri)
835 if user is not None:
836 return user, password
837 return HTTPPasswordMgr.find_user_password(self, None, authuri)
838
839
R David Murray4c7f9952015-04-16 16:36:18 -0400840class HTTPPasswordMgrWithPriorAuth(HTTPPasswordMgrWithDefaultRealm):
841
842 def __init__(self, *args, **kwargs):
843 self.authenticated = {}
844 super().__init__(*args, **kwargs)
845
846 def add_password(self, realm, uri, user, passwd, is_authenticated=False):
847 self.update_authenticated(uri, is_authenticated)
848 # Add a default for prior auth requests
849 if realm is not None:
850 super().add_password(None, uri, user, passwd)
851 super().add_password(realm, uri, user, passwd)
852
853 def update_authenticated(self, uri, is_authenticated=False):
854 # uri could be a single URI or a sequence
855 if isinstance(uri, str):
856 uri = [uri]
857
858 for default_port in True, False:
859 for u in uri:
860 reduced_uri = self.reduce_uri(u, default_port)
861 self.authenticated[reduced_uri] = is_authenticated
862
863 def is_authenticated(self, authuri):
864 for default_port in True, False:
865 reduced_authuri = self.reduce_uri(authuri, default_port)
866 for uri in self.authenticated:
867 if self.is_suburi(uri, reduced_authuri):
868 return self.authenticated[uri]
869
870
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000871class AbstractBasicAuthHandler:
872
873 # XXX this allows for multiple auth-schemes, but will stupidly pick
874 # the last one with a realm specified.
875
876 # allow for double- and single-quoted realm values
877 # (single quotes are a violation of the RFC, but appear in the wild)
878 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800879 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000880
881 # XXX could pre-emptively send auth info already accepted (RFC 2617,
882 # end of section 2, and section 1.2 immediately after "credentials"
883 # production).
884
885 def __init__(self, password_mgr=None):
886 if password_mgr is None:
887 password_mgr = HTTPPasswordMgr()
888 self.passwd = password_mgr
889 self.add_password = self.passwd.add_password
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000890
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000891 def http_error_auth_reqed(self, authreq, host, req, headers):
892 # host may be an authority (without userinfo) or a URL with an
893 # authority
894 # XXX could be multiple headers
895 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000896
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000897 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800898 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800899 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800900 raise ValueError("AbstractBasicAuthHandler does not"
901 " support the following scheme: '%s'" %
902 scheme)
903 else:
904 mo = AbstractBasicAuthHandler.rx.search(authreq)
905 if mo:
906 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800907 if quote not in ['"',"'"]:
908 warnings.warn("Basic Auth Realm was unquoted",
909 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800910 if scheme.lower() == 'basic':
Senthil Kumaran78373762014-08-20 07:53:58 +0530911 return self.retry_http_basic_auth(host, req, realm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000912
913 def retry_http_basic_auth(self, host, req, realm):
914 user, pw = self.passwd.find_user_password(realm, host)
915 if pw is not None:
916 raw = "%s:%s" % (user, pw)
917 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
Senthil Kumaran78373762014-08-20 07:53:58 +0530918 if req.get_header(self.auth_header, None) == auth:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000919 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000920 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000921 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000922 else:
923 return None
924
R David Murray4c7f9952015-04-16 16:36:18 -0400925 def http_request(self, req):
926 if (not hasattr(self.passwd, 'is_authenticated') or
927 not self.passwd.is_authenticated(req.full_url)):
928 return req
929
930 if not req.has_header('Authorization'):
931 user, passwd = self.passwd.find_user_password(None, req.full_url)
932 credentials = '{0}:{1}'.format(user, passwd).encode()
933 auth_str = base64.standard_b64encode(credentials).decode()
934 req.add_unredirected_header('Authorization',
935 'Basic {}'.format(auth_str.strip()))
936 return req
937
938 def http_response(self, req, response):
939 if hasattr(self.passwd, 'is_authenticated'):
940 if 200 <= response.code < 300:
941 self.passwd.update_authenticated(req.full_url, True)
942 else:
943 self.passwd.update_authenticated(req.full_url, False)
944 return response
945
946 https_request = http_request
947 https_response = http_response
948
949
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000950
951class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
952
953 auth_header = 'Authorization'
954
955 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000956 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000957 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000958 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000959 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000960
961
962class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
963
964 auth_header = 'Proxy-authorization'
965
966 def http_error_407(self, req, fp, code, msg, headers):
967 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000968 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000969 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
970 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000971 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000972 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000973 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000974 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000975
976
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800977# Return n random bytes.
978_randombytes = os.urandom
979
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000980
981class AbstractDigestAuthHandler:
982 # Digest authentication is specified in RFC 2617.
983
984 # XXX The client does not inspect the Authentication-Info header
985 # in a successful response.
986
987 # XXX It should be possible to test this implementation against
988 # a mock server that just generates a static set of challenges.
989
990 # XXX qop="auth-int" supports is shaky
991
992 def __init__(self, passwd=None):
993 if passwd is None:
994 passwd = HTTPPasswordMgr()
995 self.passwd = passwd
996 self.add_password = self.passwd.add_password
997 self.retried = 0
998 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000999 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001000
1001 def reset_retry_count(self):
1002 self.retried = 0
1003
1004 def http_error_auth_reqed(self, auth_header, host, req, headers):
1005 authreq = headers.get(auth_header, None)
1006 if self.retried > 5:
1007 # Don't fail endlessly - if we failed once, we'll probably
1008 # fail a second time. Hm. Unless the Password Manager is
1009 # prompting for the information. Crap. This isn't great
1010 # but it's better than the current 'repeat until recursion
1011 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001012 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +00001013 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001014 else:
1015 self.retried += 1
1016 if authreq:
1017 scheme = authreq.split()[0]
1018 if scheme.lower() == 'digest':
1019 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +08001020 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +08001021 raise ValueError("AbstractDigestAuthHandler does not support"
1022 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001023
1024 def retry_http_digest_auth(self, req, auth):
1025 token, challenge = auth.split(' ', 1)
1026 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
1027 auth = self.get_authorization(req, chal)
1028 if auth:
1029 auth_val = 'Digest %s' % auth
1030 if req.headers.get(self.auth_header, None) == auth_val:
1031 return None
1032 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +00001033 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001034 return resp
1035
1036 def get_cnonce(self, nonce):
1037 # The cnonce-value is an opaque
1038 # quoted string value provided by the client and used by both client
1039 # and server to avoid chosen plaintext attacks, to provide mutual
1040 # authentication, and to provide some message integrity protection.
1041 # This isn't a fabulous effort, but it's probably Good Enough.
1042 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001043 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001044 dig = hashlib.sha1(b).hexdigest()
1045 return dig[:16]
1046
1047 def get_authorization(self, req, chal):
1048 try:
1049 realm = chal['realm']
1050 nonce = chal['nonce']
1051 qop = chal.get('qop')
1052 algorithm = chal.get('algorithm', 'MD5')
1053 # mod_digest doesn't send an opaque, even though it isn't
1054 # supposed to be optional
1055 opaque = chal.get('opaque', None)
1056 except KeyError:
1057 return None
1058
1059 H, KD = self.get_algorithm_impls(algorithm)
1060 if H is None:
1061 return None
1062
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001063 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001064 if user is None:
1065 return None
1066
1067 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001068 if req.data is not None:
1069 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001070 else:
1071 entdig = None
1072
1073 A1 = "%s:%s:%s" % (user, realm, pw)
1074 A2 = "%s:%s" % (req.get_method(),
1075 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001076 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001077 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001078 if nonce == self.last_nonce:
1079 self.nonce_count += 1
1080 else:
1081 self.nonce_count = 1
1082 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001083 ncvalue = '%08x' % self.nonce_count
1084 cnonce = self.get_cnonce(nonce)
1085 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1086 respdig = KD(H(A1), noncebit)
1087 elif qop is None:
1088 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1089 else:
1090 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001091 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001092
1093 # XXX should the partial digests be encoded too?
1094
1095 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001096 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001097 respdig)
1098 if opaque:
1099 base += ', opaque="%s"' % opaque
1100 if entdig:
1101 base += ', digest="%s"' % entdig
1102 base += ', algorithm="%s"' % algorithm
1103 if qop:
1104 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1105 return base
1106
1107 def get_algorithm_impls(self, algorithm):
1108 # lambdas assume digest modules are imported at the top level
1109 if algorithm == 'MD5':
1110 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1111 elif algorithm == 'SHA':
1112 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1113 # XXX MD5-sess
Berker Peksage88dd1c2016-03-06 16:16:40 +02001114 else:
1115 raise ValueError("Unsupported digest authentication "
1116 "algorithm %r" % algorithm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001117 KD = lambda s, d: H("%s:%s" % (s, d))
1118 return H, KD
1119
1120 def get_entity_digest(self, data, chal):
1121 # XXX not implemented yet
1122 return None
1123
1124
1125class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1126 """An authentication protocol defined by RFC 2069
1127
1128 Digest authentication improves on basic authentication because it
1129 does not transmit passwords in the clear.
1130 """
1131
1132 auth_header = 'Authorization'
1133 handler_order = 490 # before Basic auth
1134
1135 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001136 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001137 retry = self.http_error_auth_reqed('www-authenticate',
1138 host, req, headers)
1139 self.reset_retry_count()
1140 return retry
1141
1142
1143class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1144
1145 auth_header = 'Proxy-Authorization'
1146 handler_order = 490 # before Basic auth
1147
1148 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001149 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001150 retry = self.http_error_auth_reqed('proxy-authenticate',
1151 host, req, headers)
1152 self.reset_retry_count()
1153 return retry
1154
1155class AbstractHTTPHandler(BaseHandler):
1156
1157 def __init__(self, debuglevel=0):
1158 self._debuglevel = debuglevel
1159
1160 def set_http_debuglevel(self, level):
1161 self._debuglevel = level
1162
1163 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001164 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001165 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001166 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001167
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001168 if request.data is not None: # POST
1169 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001170 if isinstance(data, str):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001171 msg = "POST data should be bytes or an iterable of bytes. " \
1172 "It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001173 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001174 if not request.has_header('Content-type'):
1175 request.add_unredirected_header(
1176 'Content-type',
1177 'application/x-www-form-urlencoded')
1178 if not request.has_header('Content-length'):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001179 try:
1180 mv = memoryview(data)
1181 except TypeError:
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001182 if isinstance(data, collections.Iterable):
Georg Brandl61536042011-02-03 07:46:41 +00001183 raise ValueError("Content-Length should be specified "
1184 "for iterable data of type %r %r" % (type(data),
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001185 data))
1186 else:
1187 request.add_unredirected_header(
Senthil Kumaran1e991f22010-12-24 04:03:59 +00001188 'Content-length', '%d' % (len(mv) * mv.itemsize))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001189
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001190 sel_host = host
1191 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001192 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001193 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001194 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001195 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001196 for name, value in self.parent.addheaders:
1197 name = name.capitalize()
1198 if not request.has_header(name):
1199 request.add_unredirected_header(name, value)
1200
1201 return request
1202
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001203 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001204 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205
1206 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001207 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001208 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001209 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001210 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001211
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001212 # will parse host:port
1213 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001214
1215 headers = dict(req.unredirected_hdrs)
1216 headers.update(dict((k, v) for k, v in req.headers.items()
1217 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001218
1219 # TODO(jhylton): Should this be redesigned to handle
1220 # persistent connections?
1221
1222 # We want to make an HTTP/1.1 request, but the addinfourl
1223 # class isn't prepared to deal with a persistent connection.
1224 # It will try to read all remaining data from the socket,
1225 # which will block while the server waits for the next request.
1226 # So make sure the connection gets closed after the (only)
1227 # request.
1228 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001229 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001230
1231 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001232 tunnel_headers = {}
1233 proxy_auth_hdr = "Proxy-Authorization"
1234 if proxy_auth_hdr in headers:
1235 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1236 # Proxy-Authorization should not be sent to origin
1237 # server.
1238 del headers[proxy_auth_hdr]
1239 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001240
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001241 try:
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001242 try:
1243 h.request(req.get_method(), req.selector, req.data, headers)
1244 except OSError as err: # timeout error
1245 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001246 r = h.getresponse()
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001247 except:
1248 h.close()
1249 raise
1250
1251 # If the server does not send us a 'Connection: close' header,
1252 # HTTPConnection assumes the socket should be left open. Manually
1253 # mark the socket to be closed when this response object goes away.
1254 if h.sock:
1255 h.sock.close()
1256 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001257
Senthil Kumaran26430412011-04-13 07:01:19 +08001258 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001259 # This line replaces the .msg attribute of the HTTPResponse
1260 # with .headers, because urllib clients expect the response to
1261 # have the reason in .msg. It would be good to mark this
1262 # attribute is deprecated and get then to use info() or
1263 # .headers.
1264 r.msg = r.reason
1265 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001266
1267
1268class HTTPHandler(AbstractHTTPHandler):
1269
1270 def http_open(self, req):
1271 return self.do_open(http.client.HTTPConnection, req)
1272
1273 http_request = AbstractHTTPHandler.do_request_
1274
1275if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001276
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001277 class HTTPSHandler(AbstractHTTPHandler):
1278
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001279 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1280 AbstractHTTPHandler.__init__(self, debuglevel)
1281 self._context = context
1282 self._check_hostname = check_hostname
1283
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001284 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001285 return self.do_open(http.client.HTTPSConnection, req,
1286 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001287
1288 https_request = AbstractHTTPHandler.do_request_
1289
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001290 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001291
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001292class HTTPCookieProcessor(BaseHandler):
1293 def __init__(self, cookiejar=None):
1294 import http.cookiejar
1295 if cookiejar is None:
1296 cookiejar = http.cookiejar.CookieJar()
1297 self.cookiejar = cookiejar
1298
1299 def http_request(self, request):
1300 self.cookiejar.add_cookie_header(request)
1301 return request
1302
1303 def http_response(self, request, response):
1304 self.cookiejar.extract_cookies(response, request)
1305 return response
1306
1307 https_request = http_request
1308 https_response = http_response
1309
1310class UnknownHandler(BaseHandler):
1311 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001312 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001313 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001314
1315def parse_keqv_list(l):
1316 """Parse list of key=value strings where keys are not duplicated."""
1317 parsed = {}
1318 for elt in l:
1319 k, v = elt.split('=', 1)
1320 if v[0] == '"' and v[-1] == '"':
1321 v = v[1:-1]
1322 parsed[k] = v
1323 return parsed
1324
1325def parse_http_list(s):
1326 """Parse lists as described by RFC 2068 Section 2.
1327
1328 In particular, parse comma-separated lists where the elements of
1329 the list may include quoted-strings. A quoted-string could
1330 contain a comma. A non-quoted string could have quotes in the
1331 middle. Neither commas nor quotes count if they are escaped.
1332 Only double-quotes count, not single-quotes.
1333 """
1334 res = []
1335 part = ''
1336
1337 escape = quote = False
1338 for cur in s:
1339 if escape:
1340 part += cur
1341 escape = False
1342 continue
1343 if quote:
1344 if cur == '\\':
1345 escape = True
1346 continue
1347 elif cur == '"':
1348 quote = False
1349 part += cur
1350 continue
1351
1352 if cur == ',':
1353 res.append(part)
1354 part = ''
1355 continue
1356
1357 if cur == '"':
1358 quote = True
1359
1360 part += cur
1361
1362 # append last part
1363 if part:
1364 res.append(part)
1365
1366 return [part.strip() for part in res]
1367
1368class FileHandler(BaseHandler):
1369 # Use local file or FTP depending on form of URL
1370 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001371 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001372 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1373 req.host != 'localhost'):
Senthil Kumaranbc07ac52014-07-22 00:15:20 -07001374 if not req.host in self.get_names():
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001375 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001376 else:
1377 return self.open_local_file(req)
1378
1379 # names for the localhost
1380 names = None
1381 def get_names(self):
1382 if FileHandler.names is None:
1383 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001384 FileHandler.names = tuple(
1385 socket.gethostbyname_ex('localhost')[2] +
1386 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001387 except socket.gaierror:
1388 FileHandler.names = (socket.gethostbyname('localhost'),)
1389 return FileHandler.names
1390
1391 # not entirely sure what the rules are here
1392 def open_local_file(self, req):
1393 import email.utils
1394 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001395 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001396 filename = req.selector
1397 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001398 try:
1399 stats = os.stat(localfile)
1400 size = stats.st_size
1401 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001402 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001403 headers = email.message_from_string(
1404 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1405 (mtype or 'text/plain', size, modified))
1406 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001407 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001408 if not host or \
1409 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001410 if host:
1411 origurl = 'file://' + host + filename
1412 else:
1413 origurl = 'file://' + filename
1414 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001415 except OSError as exp:
Georg Brandl029986a2008-06-23 11:44:14 +00001416 # users shouldn't expect OSErrors coming from urlopen()
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001417 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001418 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001419
1420def _safe_gethostbyname(host):
1421 try:
1422 return socket.gethostbyname(host)
1423 except socket.gaierror:
1424 return None
1425
1426class FTPHandler(BaseHandler):
1427 def ftp_open(self, req):
1428 import ftplib
1429 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001430 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001431 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001432 raise URLError('ftp error: no host given')
1433 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001434 if port is None:
1435 port = ftplib.FTP_PORT
1436 else:
1437 port = int(port)
1438
1439 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001440 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001441 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001442 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001443 else:
1444 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001445 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001446 user = user or ''
1447 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001448
1449 try:
1450 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001451 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001452 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001453 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001454 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001455 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001456 dirs, file = dirs[:-1], dirs[-1]
1457 if dirs and not dirs[0]:
1458 dirs = dirs[1:]
1459 try:
1460 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1461 type = file and 'I' or 'D'
1462 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001463 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001464 if attr.lower() == 'type' and \
1465 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1466 type = value.upper()
1467 fp, retrlen = fw.retrfile(file, type)
1468 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001469 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001470 if mtype:
1471 headers += "Content-type: %s\n" % mtype
1472 if retrlen is not None and retrlen >= 0:
1473 headers += "Content-length: %d\n" % retrlen
1474 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001475 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001476 except ftplib.all_errors as exp:
1477 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001478 raise exc.with_traceback(sys.exc_info()[2])
1479
1480 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001481 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1482 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001483
1484class CacheFTPHandler(FTPHandler):
1485 # XXX would be nice to have pluggable cache strategies
1486 # XXX this stuff is definitely not thread safe
1487 def __init__(self):
1488 self.cache = {}
1489 self.timeout = {}
1490 self.soonest = 0
1491 self.delay = 60
1492 self.max_conns = 16
1493
1494 def setTimeout(self, t):
1495 self.delay = t
1496
1497 def setMaxConns(self, m):
1498 self.max_conns = m
1499
1500 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1501 key = user, host, port, '/'.join(dirs), timeout
1502 if key in self.cache:
1503 self.timeout[key] = time.time() + self.delay
1504 else:
1505 self.cache[key] = ftpwrapper(user, passwd, host, port,
1506 dirs, timeout)
1507 self.timeout[key] = time.time() + self.delay
1508 self.check_cache()
1509 return self.cache[key]
1510
1511 def check_cache(self):
1512 # first check for old ones
1513 t = time.time()
1514 if self.soonest <= t:
1515 for k, v in list(self.timeout.items()):
1516 if v < t:
1517 self.cache[k].close()
1518 del self.cache[k]
1519 del self.timeout[k]
1520 self.soonest = min(list(self.timeout.values()))
1521
1522 # then check the size
1523 if len(self.cache) == self.max_conns:
1524 for k, v in list(self.timeout.items()):
1525 if v == self.soonest:
1526 del self.cache[k]
1527 del self.timeout[k]
1528 break
1529 self.soonest = min(list(self.timeout.values()))
1530
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001531 def clear_cache(self):
1532 for conn in self.cache.values():
1533 conn.close()
1534 self.cache.clear()
1535 self.timeout.clear()
1536
Antoine Pitroudf204be2012-11-24 17:59:08 +01001537class DataHandler(BaseHandler):
1538 def data_open(self, req):
1539 # data URLs as specified in RFC 2397.
1540 #
1541 # ignores POSTed data
1542 #
1543 # syntax:
1544 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1545 # mediatype := [ type "/" subtype ] *( ";" parameter )
1546 # data := *urlchar
1547 # parameter := attribute "=" value
1548 url = req.full_url
1549
1550 scheme, data = url.split(":",1)
1551 mediatype, data = data.split(",",1)
1552
1553 # even base64 encoded data URLs might be quoted so unquote in any case:
1554 data = unquote_to_bytes(data)
1555 if mediatype.endswith(";base64"):
1556 data = base64.decodebytes(data)
1557 mediatype = mediatype[:-7]
1558
1559 if not mediatype:
1560 mediatype = "text/plain;charset=US-ASCII"
1561
1562 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1563 (mediatype, len(data)))
1564
1565 return addinfourl(io.BytesIO(data), headers, url)
1566
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001567
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001568# Code move from the old urllib module
1569
1570MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1571
1572# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001573if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001574 from nturl2path import url2pathname, pathname2url
1575else:
1576 def url2pathname(pathname):
1577 """OS-specific conversion from a relative URL of the 'file' scheme
1578 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001579 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001580
1581 def pathname2url(pathname):
1582 """OS-specific conversion from a file system path to a relative URL
1583 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001584 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001585
1586# This really consists of two pieces:
1587# (1) a class which handles opening of all sorts of URLs
1588# (plus assorted utilities etc.)
1589# (2) a set of functions for parsing URLs
1590# XXX Should these be separated out into different modules?
1591
1592
1593ftpcache = {}
1594class URLopener:
1595 """Class to open URLs.
1596 This is a class rather than just a subroutine because we may need
1597 more than one set of global protocol-specific options.
1598 Note -- this is a base class for those who don't want the
1599 automatic handling of errors type 302 (relocated) and 401
1600 (authorization needed)."""
1601
1602 __tempfiles = None
1603
1604 version = "Python-urllib/%s" % __version__
1605
1606 # Constructor
1607 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001608 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001609 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1610 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001611 if proxies is None:
1612 proxies = getproxies()
1613 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1614 self.proxies = proxies
1615 self.key_file = x509.get('key_file')
1616 self.cert_file = x509.get('cert_file')
1617 self.addheaders = [('User-Agent', self.version)]
1618 self.__tempfiles = []
1619 self.__unlink = os.unlink # See cleanup()
1620 self.tempcache = None
1621 # Undocumented feature: if you assign {} to tempcache,
1622 # it is used to cache files retrieved with
1623 # self.retrieve(). This is not enabled by default
1624 # since it does not work for changing documents (and I
1625 # haven't got the logic to check expiration headers
1626 # yet).
1627 self.ftpcache = ftpcache
1628 # Undocumented feature: you can use a different
1629 # ftp cache by assigning to the .ftpcache member;
1630 # in case you want logically independent URL openers
1631 # XXX This is not threadsafe. Bah.
1632
1633 def __del__(self):
1634 self.close()
1635
1636 def close(self):
1637 self.cleanup()
1638
1639 def cleanup(self):
1640 # This code sometimes runs when the rest of this module
1641 # has already been deleted, so it can't use any globals
1642 # or import anything.
1643 if self.__tempfiles:
1644 for file in self.__tempfiles:
1645 try:
1646 self.__unlink(file)
1647 except OSError:
1648 pass
1649 del self.__tempfiles[:]
1650 if self.tempcache:
1651 self.tempcache.clear()
1652
1653 def addheader(self, *args):
1654 """Add a header to be used by the HTTP interface only
1655 e.g. u.addheader('Accept', 'sound/basic')"""
1656 self.addheaders.append(args)
1657
1658 # External interface
1659 def open(self, fullurl, data=None):
1660 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001661 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001662 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001663 if self.tempcache and fullurl in self.tempcache:
1664 filename, headers = self.tempcache[fullurl]
1665 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001666 return addinfourl(fp, headers, fullurl)
1667 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001668 if not urltype:
1669 urltype = 'file'
1670 if urltype in self.proxies:
1671 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001672 urltype, proxyhost = splittype(proxy)
1673 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001674 url = (host, fullurl) # Signal special case to open_*()
1675 else:
1676 proxy = None
1677 name = 'open_' + urltype
1678 self.type = urltype
1679 name = name.replace('-', '_')
1680 if not hasattr(self, name):
1681 if proxy:
1682 return self.open_unknown_proxy(proxy, fullurl, data)
1683 else:
1684 return self.open_unknown(fullurl, data)
1685 try:
1686 if data is None:
1687 return getattr(self, name)(url)
1688 else:
1689 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001690 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001691 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001692 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001693 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001694
1695 def open_unknown(self, fullurl, data=None):
1696 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001697 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001698 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001699
1700 def open_unknown_proxy(self, proxy, fullurl, data=None):
1701 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001702 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001703 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001704
1705 # External interface
1706 def retrieve(self, url, filename=None, reporthook=None, data=None):
1707 """retrieve(url) returns (filename, headers) for a local object
1708 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001709 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001710 if self.tempcache and url in self.tempcache:
1711 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001712 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001713 if filename is None and (not type or type == 'file'):
1714 try:
1715 fp = self.open_local_file(url1)
1716 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001717 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001718 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001719 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001720 pass
1721 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001722 try:
1723 headers = fp.info()
1724 if filename:
1725 tfp = open(filename, 'wb')
1726 else:
1727 import tempfile
1728 garbage, path = splittype(url)
1729 garbage, path = splithost(path or "")
1730 path, garbage = splitquery(path or "")
1731 path, garbage = splitattr(path or "")
1732 suffix = os.path.splitext(path)[1]
1733 (fd, filename) = tempfile.mkstemp(suffix)
1734 self.__tempfiles.append(filename)
1735 tfp = os.fdopen(fd, 'wb')
1736 try:
1737 result = filename, headers
1738 if self.tempcache is not None:
1739 self.tempcache[url] = result
1740 bs = 1024*8
1741 size = -1
1742 read = 0
1743 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001744 if "content-length" in headers:
1745 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001746 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001747 reporthook(blocknum, bs, size)
1748 while 1:
1749 block = fp.read(bs)
1750 if not block:
1751 break
1752 read += len(block)
1753 tfp.write(block)
1754 blocknum += 1
1755 if reporthook:
1756 reporthook(blocknum, bs, size)
1757 finally:
1758 tfp.close()
1759 finally:
1760 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001761
1762 # raise exception if actual size does not match content-length header
1763 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001764 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001765 "retrieval incomplete: got only %i out of %i bytes"
1766 % (read, size), result)
1767
1768 return result
1769
1770 # Each method named open_<type> knows how to open that type of URL
1771
1772 def _open_generic_http(self, connection_factory, url, data):
1773 """Make an HTTP connection using connection_class.
1774
1775 This is an internal method that should be called from
1776 open_http() or open_https().
1777
1778 Arguments:
1779 - connection_factory should take a host name and return an
1780 HTTPConnection instance.
1781 - url is the url to retrieval or a host, relative-path pair.
1782 - data is payload for a POST request or None.
1783 """
1784
1785 user_passwd = None
1786 proxy_passwd= None
1787 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001788 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001789 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001790 user_passwd, host = splituser(host)
1791 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001792 realhost = host
1793 else:
1794 host, selector = url
1795 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001796 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001797 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001798 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001799 url = rest
1800 user_passwd = None
1801 if urltype.lower() != 'http':
1802 realhost = None
1803 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001804 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001805 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001806 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001807 if user_passwd:
1808 selector = "%s://%s%s" % (urltype, realhost, rest)
1809 if proxy_bypass(realhost):
1810 host = realhost
1811
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001812 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001813
1814 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001815 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001816 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001817 else:
1818 proxy_auth = None
1819
1820 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001821 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001822 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001823 else:
1824 auth = None
1825 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001826 headers = {}
1827 if proxy_auth:
1828 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1829 if auth:
1830 headers["Authorization"] = "Basic %s" % auth
1831 if realhost:
1832 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001833
1834 # Add Connection:close as we don't support persistent connections yet.
1835 # This helps in closing the socket and avoiding ResourceWarning
1836
1837 headers["Connection"] = "close"
1838
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001839 for header, value in self.addheaders:
1840 headers[header] = value
1841
1842 if data is not None:
1843 headers["Content-Type"] = "application/x-www-form-urlencoded"
1844 http_conn.request("POST", selector, data, headers)
1845 else:
1846 http_conn.request("GET", selector, headers=headers)
1847
1848 try:
1849 response = http_conn.getresponse()
1850 except http.client.BadStatusLine:
1851 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001852 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001853
1854 # According to RFC 2616, "2xx" code indicates that the client's
1855 # request was successfully received, understood, and accepted.
1856 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001857 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001858 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001859 else:
1860 return self.http_error(
1861 url, response.fp,
1862 response.status, response.reason, response.msg, data)
1863
1864 def open_http(self, url, data=None):
1865 """Use HTTP protocol."""
1866 return self._open_generic_http(http.client.HTTPConnection, url, data)
1867
1868 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1869 """Handle http errors.
1870
1871 Derived class can override this, or provide specific handlers
1872 named http_error_DDD where DDD is the 3-digit error code."""
1873 # First check if there's a specific handler for this error
1874 name = 'http_error_%d' % errcode
1875 if hasattr(self, name):
1876 method = getattr(self, name)
1877 if data is None:
1878 result = method(url, fp, errcode, errmsg, headers)
1879 else:
1880 result = method(url, fp, errcode, errmsg, headers, data)
1881 if result: return result
1882 return self.http_error_default(url, fp, errcode, errmsg, headers)
1883
1884 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001885 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001886 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001887 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001888
1889 if _have_ssl:
1890 def _https_connection(self, host):
1891 return http.client.HTTPSConnection(host,
1892 key_file=self.key_file,
1893 cert_file=self.cert_file)
1894
1895 def open_https(self, url, data=None):
1896 """Use HTTPS protocol."""
1897 return self._open_generic_http(self._https_connection, url, data)
1898
1899 def open_file(self, url):
1900 """Use local file or FTP depending on form of URL."""
1901 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001902 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001903 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001904 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001905 else:
1906 return self.open_local_file(url)
1907
1908 def open_local_file(self, url):
1909 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001910 import email.utils
1911 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001912 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001913 localname = url2pathname(file)
1914 try:
1915 stats = os.stat(localname)
1916 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001917 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001918 size = stats.st_size
1919 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1920 mtype = mimetypes.guess_type(url)[0]
1921 headers = email.message_from_string(
1922 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1923 (mtype or 'text/plain', size, modified))
1924 if not host:
1925 urlfile = file
1926 if file[:1] == '/':
1927 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001928 return addinfourl(open(localname, 'rb'), headers, urlfile)
1929 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001930 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07001931 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001932 urlfile = file
1933 if file[:1] == '/':
1934 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08001935 elif file[:2] == './':
1936 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00001937 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001938 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001939
1940 def open_ftp(self, url):
1941 """Use FTP protocol."""
1942 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001943 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001944 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001945 host, path = splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001946 if not host: raise URLError('ftp error: no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001947 host, port = splitport(host)
1948 user, host = splituser(host)
1949 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001950 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001951 host = unquote(host)
1952 user = unquote(user or '')
1953 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001954 host = socket.gethostbyname(host)
1955 if not port:
1956 import ftplib
1957 port = ftplib.FTP_PORT
1958 else:
1959 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001960 path, attrs = splitattr(path)
1961 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001962 dirs = path.split('/')
1963 dirs, file = dirs[:-1], dirs[-1]
1964 if dirs and not dirs[0]: dirs = dirs[1:]
1965 if dirs and not dirs[0]: dirs[0] = '/'
1966 key = user, host, port, '/'.join(dirs)
1967 # XXX thread unsafe!
1968 if len(self.ftpcache) > MAXFTPCACHE:
1969 # Prune the cache, rather arbitrarily
Benjamin Peterson3c2dca62014-06-07 15:08:04 -07001970 for k in list(self.ftpcache):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001971 if k != key:
1972 v = self.ftpcache[k]
1973 del self.ftpcache[k]
1974 v.close()
1975 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001976 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001977 self.ftpcache[key] = \
1978 ftpwrapper(user, passwd, host, port, dirs)
1979 if not file: type = 'D'
1980 else: type = 'I'
1981 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001982 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001983 if attr.lower() == 'type' and \
1984 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1985 type = value.upper()
1986 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1987 mtype = mimetypes.guess_type("ftp:" + url)[0]
1988 headers = ""
1989 if mtype:
1990 headers += "Content-Type: %s\n" % mtype
1991 if retrlen is not None and retrlen >= 0:
1992 headers += "Content-Length: %d\n" % retrlen
1993 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001994 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001995 except ftperrors() as exp:
1996 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001997
1998 def open_data(self, url, data=None):
1999 """Use "data" URL."""
2000 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07002001 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002002 # ignore POSTed data
2003 #
2004 # syntax of data URLs:
2005 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
2006 # mediatype := [ type "/" subtype ] *( ";" parameter )
2007 # data := *urlchar
2008 # parameter := attribute "=" value
2009 try:
2010 [type, data] = url.split(',', 1)
2011 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002012 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002013 if not type:
2014 type = 'text/plain;charset=US-ASCII'
2015 semi = type.rfind(';')
2016 if semi >= 0 and '=' not in type[semi:]:
2017 encoding = type[semi+1:]
2018 type = type[:semi]
2019 else:
2020 encoding = ''
2021 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00002022 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002023 time.gmtime(time.time())))
2024 msg.append('Content-type: %s' % type)
2025 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00002026 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00002027 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002028 else:
Georg Brandl13e89462008-07-01 19:56:00 +00002029 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002030 msg.append('Content-Length: %d' % len(data))
2031 msg.append('')
2032 msg.append(data)
2033 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00002034 headers = email.message_from_string(msg)
2035 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002036 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00002037 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002038
2039
2040class FancyURLopener(URLopener):
2041 """Derived class with handlers for errors we can handle (perhaps)."""
2042
2043 def __init__(self, *args, **kwargs):
2044 URLopener.__init__(self, *args, **kwargs)
2045 self.auth_cache = {}
2046 self.tries = 0
2047 self.maxtries = 10
2048
2049 def http_error_default(self, url, fp, errcode, errmsg, headers):
2050 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00002051 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002052
2053 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
2054 """Error 302 -- relocated (temporarily)."""
2055 self.tries += 1
Martin Pantera0370222016-02-04 06:01:35 +00002056 try:
2057 if self.maxtries and self.tries >= self.maxtries:
2058 if hasattr(self, "http_error_500"):
2059 meth = self.http_error_500
2060 else:
2061 meth = self.http_error_default
2062 return meth(url, fp, 500,
2063 "Internal Server Error: Redirect Recursion",
2064 headers)
2065 result = self.redirect_internal(url, fp, errcode, errmsg,
2066 headers, data)
2067 return result
2068 finally:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002069 self.tries = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002070
2071 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
2072 if 'location' in headers:
2073 newurl = headers['location']
2074 elif 'uri' in headers:
2075 newurl = headers['uri']
2076 else:
2077 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002078 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002079
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002080 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002081 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002082
2083 urlparts = urlparse(newurl)
2084
2085 # For security reasons, we don't allow redirection to anything other
2086 # than http, https and ftp.
2087
2088 # We are using newer HTTPError with older redirect_internal method
2089 # This older method will get deprecated in 3.3
2090
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002091 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002092 raise HTTPError(newurl, errcode,
2093 errmsg +
2094 " Redirection to url '%s' is not allowed." % newurl,
2095 headers, fp)
2096
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002097 return self.open(newurl)
2098
2099 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2100 """Error 301 -- also relocated (permanently)."""
2101 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2102
2103 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2104 """Error 303 -- also relocated (essentially identical to 302)."""
2105 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2106
2107 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2108 """Error 307 -- relocated, but turn POST into error."""
2109 if data is None:
2110 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2111 else:
2112 return self.http_error_default(url, fp, errcode, errmsg, headers)
2113
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002114 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2115 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002116 """Error 401 -- authentication required.
2117 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002118 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002119 URLopener.http_error_default(self, url, fp,
2120 errcode, errmsg, headers)
2121 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002122 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2123 if not match:
2124 URLopener.http_error_default(self, url, fp,
2125 errcode, errmsg, headers)
2126 scheme, realm = match.groups()
2127 if scheme.lower() != 'basic':
2128 URLopener.http_error_default(self, url, fp,
2129 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002130 if not retry:
2131 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2132 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002133 name = 'retry_' + self.type + '_basic_auth'
2134 if data is None:
2135 return getattr(self,name)(url, realm)
2136 else:
2137 return getattr(self,name)(url, realm, data)
2138
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002139 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2140 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002141 """Error 407 -- proxy authentication required.
2142 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002143 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002144 URLopener.http_error_default(self, url, fp,
2145 errcode, errmsg, headers)
2146 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002147 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2148 if not match:
2149 URLopener.http_error_default(self, url, fp,
2150 errcode, errmsg, headers)
2151 scheme, realm = match.groups()
2152 if scheme.lower() != 'basic':
2153 URLopener.http_error_default(self, url, fp,
2154 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002155 if not retry:
2156 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2157 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002158 name = 'retry_proxy_' + self.type + '_basic_auth'
2159 if data is None:
2160 return getattr(self,name)(url, realm)
2161 else:
2162 return getattr(self,name)(url, realm, data)
2163
2164 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002165 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002166 newurl = 'http://' + host + selector
2167 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002168 urltype, proxyhost = splittype(proxy)
2169 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002170 i = proxyhost.find('@') + 1
2171 proxyhost = proxyhost[i:]
2172 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2173 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002174 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002175 quote(passwd, safe=''), proxyhost)
2176 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2177 if data is None:
2178 return self.open(newurl)
2179 else:
2180 return self.open(newurl, data)
2181
2182 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002183 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002184 newurl = 'https://' + host + selector
2185 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002186 urltype, proxyhost = splittype(proxy)
2187 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002188 i = proxyhost.find('@') + 1
2189 proxyhost = proxyhost[i:]
2190 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2191 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002192 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002193 quote(passwd, safe=''), proxyhost)
2194 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2195 if data is None:
2196 return self.open(newurl)
2197 else:
2198 return self.open(newurl, data)
2199
2200 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002201 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002202 i = host.find('@') + 1
2203 host = host[i:]
2204 user, passwd = self.get_user_passwd(host, realm, i)
2205 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002206 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002207 quote(passwd, safe=''), host)
2208 newurl = 'http://' + host + selector
2209 if data is None:
2210 return self.open(newurl)
2211 else:
2212 return self.open(newurl, data)
2213
2214 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002215 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002216 i = host.find('@') + 1
2217 host = host[i:]
2218 user, passwd = self.get_user_passwd(host, realm, i)
2219 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002220 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002221 quote(passwd, safe=''), host)
2222 newurl = 'https://' + host + selector
2223 if data is None:
2224 return self.open(newurl)
2225 else:
2226 return self.open(newurl, data)
2227
Florent Xicluna757445b2010-05-17 17:24:07 +00002228 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002229 key = realm + '@' + host.lower()
2230 if key in self.auth_cache:
2231 if clear_cache:
2232 del self.auth_cache[key]
2233 else:
2234 return self.auth_cache[key]
2235 user, passwd = self.prompt_user_passwd(host, realm)
2236 if user or passwd: self.auth_cache[key] = (user, passwd)
2237 return user, passwd
2238
2239 def prompt_user_passwd(self, host, realm):
2240 """Override this in a GUI environment!"""
2241 import getpass
2242 try:
2243 user = input("Enter username for %s at %s: " % (realm, host))
2244 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2245 (user, realm, host))
2246 return user, passwd
2247 except KeyboardInterrupt:
2248 print()
2249 return None, None
2250
2251
2252# Utility functions
2253
2254_localhost = None
2255def localhost():
2256 """Return the IP address of the magic hostname 'localhost'."""
2257 global _localhost
2258 if _localhost is None:
2259 _localhost = socket.gethostbyname('localhost')
2260 return _localhost
2261
2262_thishost = None
2263def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002264 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002265 global _thishost
2266 if _thishost is None:
Senthil Kumarandcdadfe2013-06-01 11:12:17 -07002267 try:
2268 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
2269 except socket.gaierror:
2270 _thishost = tuple(socket.gethostbyname_ex('localhost')[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002271 return _thishost
2272
2273_ftperrors = None
2274def ftperrors():
2275 """Return the set of errors raised by the FTP class."""
2276 global _ftperrors
2277 if _ftperrors is None:
2278 import ftplib
2279 _ftperrors = ftplib.all_errors
2280 return _ftperrors
2281
2282_noheaders = None
2283def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002284 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002285 global _noheaders
2286 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002287 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002288 return _noheaders
2289
2290
2291# Utility classes
2292
2293class ftpwrapper:
2294 """Class used by open_ftp() for cache of open FTP connections."""
2295
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002296 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2297 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002298 self.user = user
2299 self.passwd = passwd
2300 self.host = host
2301 self.port = port
2302 self.dirs = dirs
2303 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002304 self.refcount = 0
2305 self.keepalive = persistent
Victor Stinnerab73e652015-04-07 12:49:27 +02002306 try:
2307 self.init()
2308 except:
2309 self.close()
2310 raise
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002311
2312 def init(self):
2313 import ftplib
2314 self.busy = 0
2315 self.ftp = ftplib.FTP()
2316 self.ftp.connect(self.host, self.port, self.timeout)
2317 self.ftp.login(self.user, self.passwd)
Senthil Kumarancaa00fe2013-06-02 11:59:47 -07002318 _target = '/'.join(self.dirs)
2319 self.ftp.cwd(_target)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002320
2321 def retrfile(self, file, type):
2322 import ftplib
2323 self.endtransfer()
2324 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2325 else: cmd = 'TYPE ' + type; isdir = 0
2326 try:
2327 self.ftp.voidcmd(cmd)
2328 except ftplib.all_errors:
2329 self.init()
2330 self.ftp.voidcmd(cmd)
2331 conn = None
2332 if file and not isdir:
2333 # Try to retrieve as a file
2334 try:
2335 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002336 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002337 except ftplib.error_perm as reason:
2338 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002339 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002340 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002341 if not conn:
2342 # Set transfer mode to ASCII!
2343 self.ftp.voidcmd('TYPE A')
2344 # Try a directory listing. Verify that directory exists.
2345 if file:
2346 pwd = self.ftp.pwd()
2347 try:
2348 try:
2349 self.ftp.cwd(file)
2350 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002351 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002352 finally:
2353 self.ftp.cwd(pwd)
2354 cmd = 'LIST ' + file
2355 else:
2356 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002357 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002358 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002359
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002360 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2361 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002362 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002363 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002364 return (ftpobj, retrlen)
2365
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002366 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002367 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002368
2369 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002370 self.keepalive = False
2371 if self.refcount <= 0:
2372 self.real_close()
2373
2374 def file_close(self):
2375 self.endtransfer()
2376 self.refcount -= 1
2377 if self.refcount <= 0 and not self.keepalive:
2378 self.real_close()
2379
2380 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002381 self.endtransfer()
2382 try:
2383 self.ftp.close()
2384 except ftperrors():
2385 pass
2386
2387# Proxy handling
2388def getproxies_environment():
2389 """Return a dictionary of scheme -> proxy server URL mappings.
2390
2391 Scan the environment for variables named <scheme>_proxy;
2392 this seems to be the standard convention. If you need a
2393 different way, you can pass a proxies dictionary to the
2394 [Fancy]URLopener constructor.
2395
2396 """
2397 proxies = {}
2398 for name, value in os.environ.items():
2399 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002400 if value and name[-6:] == '_proxy':
2401 proxies[name[:-6]] = value
2402 return proxies
2403
2404def proxy_bypass_environment(host):
2405 """Test if proxies should not be used for a particular host.
2406
2407 Checks the environment for a variable named no_proxy, which should
2408 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2409 """
2410 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2411 # '*' is special case for always bypass
2412 if no_proxy == '*':
2413 return 1
2414 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002415 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002416 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002417 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2418 for name in no_proxy_list:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002419 if name and (hostonly.endswith(name) or host.endswith(name)):
2420 return 1
2421 # otherwise, don't bypass
2422 return 0
2423
2424
Ronald Oussorene72e1612011-03-14 18:15:25 -04002425# This code tests an OSX specific data structure but is testable on all
2426# platforms
2427def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2428 """
2429 Return True iff this host shouldn't be accessed using a proxy
2430
2431 This function uses the MacOSX framework SystemConfiguration
2432 to fetch the proxy information.
2433
2434 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2435 { 'exclude_simple': bool,
2436 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2437 }
2438 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002439 from fnmatch import fnmatch
2440
2441 hostonly, port = splitport(host)
2442
2443 def ip2num(ipAddr):
2444 parts = ipAddr.split('.')
2445 parts = list(map(int, parts))
2446 if len(parts) != 4:
2447 parts = (parts + [0, 0, 0, 0])[:4]
2448 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2449
2450 # Check for simple host names:
2451 if '.' not in host:
2452 if proxy_settings['exclude_simple']:
2453 return True
2454
2455 hostIP = None
2456
2457 for value in proxy_settings.get('exceptions', ()):
2458 # Items in the list are strings like these: *.local, 169.254/16
2459 if not value: continue
2460
2461 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2462 if m is not None:
2463 if hostIP is None:
2464 try:
2465 hostIP = socket.gethostbyname(hostonly)
2466 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002467 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002468 continue
2469
2470 base = ip2num(m.group(1))
2471 mask = m.group(2)
2472 if mask is None:
2473 mask = 8 * (m.group(1).count('.') + 1)
2474 else:
2475 mask = int(mask[1:])
2476 mask = 32 - mask
2477
2478 if (hostIP >> mask) == (base >> mask):
2479 return True
2480
2481 elif fnmatch(host, value):
2482 return True
2483
2484 return False
2485
2486
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002487if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002488 from _scproxy import _get_proxy_settings, _get_proxies
2489
2490 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002491 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002492 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002493
2494 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002495 """Return a dictionary of scheme -> proxy server URL mappings.
2496
Ronald Oussoren84151202010-04-18 20:46:11 +00002497 This function uses the MacOSX framework SystemConfiguration
2498 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002499 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002500 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002501
Ronald Oussoren84151202010-04-18 20:46:11 +00002502
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002503
2504 def proxy_bypass(host):
2505 if getproxies_environment():
2506 return proxy_bypass_environment(host)
2507 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002508 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002509
2510 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002511 return getproxies_environment() or getproxies_macosx_sysconf()
2512
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002513
2514elif os.name == 'nt':
2515 def getproxies_registry():
2516 """Return a dictionary of scheme -> proxy server URL mappings.
2517
2518 Win32 uses the registry to store proxies.
2519
2520 """
2521 proxies = {}
2522 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002523 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002524 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002525 # Std module, so should be around - but you never know!
2526 return proxies
2527 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002528 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002529 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002530 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002531 'ProxyEnable')[0]
2532 if proxyEnable:
2533 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002534 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002535 'ProxyServer')[0])
2536 if '=' in proxyServer:
2537 # Per-protocol settings
2538 for p in proxyServer.split(';'):
2539 protocol, address = p.split('=', 1)
2540 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002541 if not re.match('^([^/:]+)://', address):
2542 address = '%s://%s' % (protocol, address)
2543 proxies[protocol] = address
2544 else:
2545 # Use one setting for all protocols
2546 if proxyServer[:5] == 'http:':
2547 proxies['http'] = proxyServer
2548 else:
2549 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002550 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002551 proxies['ftp'] = 'ftp://%s' % proxyServer
2552 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002553 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002554 # Either registry key not found etc, or the value in an
2555 # unexpected format.
2556 # proxies already set up to be empty so nothing to do
2557 pass
2558 return proxies
2559
2560 def getproxies():
2561 """Return a dictionary of scheme -> proxy server URL mappings.
2562
2563 Returns settings gathered from the environment, if specified,
2564 or the registry.
2565
2566 """
2567 return getproxies_environment() or getproxies_registry()
2568
2569 def proxy_bypass_registry(host):
2570 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002571 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002572 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002573 # Std modules, so should be around - but you never know!
2574 return 0
2575 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002576 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002577 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002578 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002579 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002580 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002581 'ProxyOverride')[0])
2582 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002583 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002584 return 0
2585 if not proxyEnable or not proxyOverride:
2586 return 0
2587 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002588 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002589 host = [rawHost]
2590 try:
2591 addr = socket.gethostbyname(rawHost)
2592 if addr != rawHost:
2593 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002594 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002595 pass
2596 try:
2597 fqdn = socket.getfqdn(rawHost)
2598 if fqdn != rawHost:
2599 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002600 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002601 pass
2602 # make a check value list from the registry entry: replace the
2603 # '<local>' string by the localhost entry and the corresponding
2604 # canonical entry.
2605 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002606 # now check if we match one of the registry values.
2607 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002608 if test == '<local>':
2609 if '.' not in rawHost:
2610 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002611 test = test.replace(".", r"\.") # mask dots
2612 test = test.replace("*", r".*") # change glob sequence
2613 test = test.replace("?", r".") # change glob char
2614 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002615 if re.match(test, val, re.I):
2616 return 1
2617 return 0
2618
2619 def proxy_bypass(host):
2620 """Return a dictionary of scheme -> proxy server URL mappings.
2621
2622 Returns settings gathered from the environment, if specified,
2623 or the registry.
2624
2625 """
2626 if getproxies_environment():
2627 return proxy_bypass_environment(host)
2628 else:
2629 return proxy_bypass_registry(host)
2630
2631else:
2632 # By default use environment variables
2633 getproxies = getproxies_environment
2634 proxy_bypass = proxy_bypass_environment