blob: e0c8116373299819dc4cc10e0f9bbd3a5c4ef35a [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',
123 'AbstractBasicAuthHandler', 'HTTPBasicAuthHandler', 'ProxyBasicAuthHandler',
124 'AbstractDigestAuthHandler', 'HTTPDigestAuthHandler', 'ProxyDigestAuthHandler',
Antoine Pitroudf204be2012-11-24 17:59:08 +0100125 'HTTPHandler', 'FileHandler', 'FTPHandler', 'CacheFTPHandler', 'DataHandler',
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800126 'UnknownHandler', 'HTTPErrorProcessor',
127 # Functions
128 'urlopen', 'install_opener', 'build_opener',
129 'pathname2url', 'url2pathname', 'getproxies',
130 # Legacy interface
131 'urlretrieve', 'urlcleanup', 'URLopener', 'FancyURLopener',
132]
133
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000134# used in User-Agent header sent
135__version__ = sys.version[:3]
136
137_opener = None
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000138def urlopen(url, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Senthil Kumaran8b7e1612014-09-19 15:23:30 +0800139 *, cafile=None, capath=None, cadefault=False, context=None):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000140 global _opener
Antoine Pitroude9ac6c2012-05-16 21:40:01 +0200141 if cafile or capath or cadefault:
Senthil Kumaran8b7e1612014-09-19 15:23:30 +0800142 if context is not None:
143 raise ValueError(
144 "You can't pass both context and any of cafile, capath, and "
145 "cadefault"
146 )
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000147 if not _have_ssl:
148 raise ValueError('SSL support not available')
Christian Heimes67986f92013-11-23 22:43:47 +0100149 context = ssl._create_stdlib_context(cert_reqs=ssl.CERT_REQUIRED,
150 cafile=cafile,
151 capath=capath)
Antoine Pitrou9a8d6932013-04-01 18:55:35 +0200152 https_handler = HTTPSHandler(context=context, check_hostname=True)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000153 opener = build_opener(https_handler)
Senthil Kumaran8b7e1612014-09-19 15:23:30 +0800154 elif context:
155 https_handler = HTTPSHandler(context=context)
156 opener = build_opener(https_handler)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000157 elif _opener is None:
158 _opener = opener = build_opener()
159 else:
160 opener = _opener
161 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000162
163def install_opener(opener):
164 global _opener
165 _opener = opener
166
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700167_url_tempfiles = []
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000168def urlretrieve(url, filename=None, reporthook=None, data=None):
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700169 """
170 Retrieve a URL into a temporary location on disk.
171
172 Requires a URL argument. If a filename is passed, it is used as
173 the temporary file location. The reporthook argument should be
174 a callable that accepts a block number, a read size, and the
175 total file size of the URL target. The data argument should be
176 valid URL encoded data.
177
178 If a filename is passed and the URL points to a local resource,
179 the result is a copy from local file to new file.
180
181 Returns a tuple containing the path to the newly created
182 data file as well as the resulting HTTPMessage object.
183 """
184 url_type, path = splittype(url)
185
186 with contextlib.closing(urlopen(url, data)) as fp:
187 headers = fp.info()
188
189 # Just return the local path and the "headers" for file://
190 # URLs. No sense in performing a copy unless requested.
191 if url_type == "file" and not filename:
192 return os.path.normpath(path), headers
193
194 # Handle temporary file setup.
195 if filename:
196 tfp = open(filename, 'wb')
197 else:
198 tfp = tempfile.NamedTemporaryFile(delete=False)
199 filename = tfp.name
200 _url_tempfiles.append(filename)
201
202 with tfp:
203 result = filename, headers
204 bs = 1024*8
205 size = -1
206 read = 0
207 blocknum = 0
208 if "content-length" in headers:
209 size = int(headers["Content-Length"])
210
211 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800212 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700213
214 while True:
215 block = fp.read(bs)
216 if not block:
217 break
218 read += len(block)
219 tfp.write(block)
220 blocknum += 1
221 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800222 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700223
224 if size >= 0 and read < size:
225 raise ContentTooShortError(
226 "retrieval incomplete: got only %i out of %i bytes"
227 % (read, size), result)
228
229 return result
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000230
231def urlcleanup():
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700232 for temp_file in _url_tempfiles:
233 try:
234 os.unlink(temp_file)
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200235 except OSError:
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700236 pass
237
238 del _url_tempfiles[:]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000239 global _opener
240 if _opener:
241 _opener = None
242
243# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000244_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000245def request_host(request):
246 """Return request-host, as defined by RFC 2965.
247
248 Variation from RFC: returned value is lowercased, for convenient
249 comparison.
250
251 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000252 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000253 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000254 if host == "":
255 host = request.get_header("Host", "")
256
257 # remove port, if present
258 host = _cut_port_re.sub("", host, 1)
259 return host.lower()
260
261class Request:
262
263 def __init__(self, url, data=None, headers={},
Senthil Kumarande49d642011-10-16 23:54:44 +0800264 origin_req_host=None, unverifiable=False,
265 method=None):
Senthil Kumaran52380922013-04-25 05:45:48 -0700266 self.full_url = url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000267 self.headers = {}
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200268 self.unredirected_hdrs = {}
269 self._data = None
270 self.data = data
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000271 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000272 for key, value in headers.items():
273 self.add_header(key, value)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000274 if origin_req_host is None:
275 origin_req_host = request_host(self)
276 self.origin_req_host = origin_req_host
277 self.unverifiable = unverifiable
Jason R. Coombs7dc4f4b2013-09-08 12:47:07 -0400278 if method:
279 self.method = method
Senthil Kumaran52380922013-04-25 05:45:48 -0700280
281 @property
282 def full_url(self):
Senthil Kumaran83070752013-05-24 09:14:12 -0700283 if self.fragment:
284 return '{}#{}'.format(self._full_url, self.fragment)
Senthil Kumaran52380922013-04-25 05:45:48 -0700285 return self._full_url
286
287 @full_url.setter
288 def full_url(self, url):
289 # unwrap('<URL:type://host/path>') --> 'type://host/path'
290 self._full_url = unwrap(url)
291 self._full_url, self.fragment = splittag(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000292 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000293
Senthil Kumaran52380922013-04-25 05:45:48 -0700294 @full_url.deleter
295 def full_url(self):
296 self._full_url = None
297 self.fragment = None
298 self.selector = ''
299
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200300 @property
301 def data(self):
302 return self._data
303
304 @data.setter
305 def data(self, data):
306 if data != self._data:
307 self._data = data
308 # issue 16464
309 # if we change data we need to remove content-length header
310 # (cause it's most probably calculated for previous value)
311 if self.has_header("Content-length"):
312 self.remove_header("Content-length")
313
314 @data.deleter
315 def data(self):
R David Murray9cc7d452013-03-20 00:10:51 -0400316 self.data = None
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200317
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000318 def _parse(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700319 self.type, rest = splittype(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000320 if self.type is None:
R David Murrayd8a46962013-04-03 06:58:34 -0400321 raise ValueError("unknown url type: %r" % self.full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000322 self.host, self.selector = splithost(rest)
323 if self.host:
324 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000325
326 def get_method(self):
Senthil Kumarande49d642011-10-16 23:54:44 +0800327 """Return a string indicating the HTTP request method."""
Jason R. Coombsaae6a1d2013-09-08 12:54:33 -0400328 default_method = "POST" if self.data is not None else "GET"
329 return getattr(self, 'method', default_method)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000330
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000331 def get_full_url(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700332 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000333
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000334 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000335 if self.type == 'https' and not self._tunnel_host:
336 self._tunnel_host = self.host
337 else:
338 self.type= type
339 self.selector = self.full_url
340 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000341
342 def has_proxy(self):
343 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000344
345 def add_header(self, key, val):
346 # useful for something like authentication
347 self.headers[key.capitalize()] = val
348
349 def add_unredirected_header(self, key, val):
350 # will not be added to a redirected request
351 self.unredirected_hdrs[key.capitalize()] = val
352
353 def has_header(self, header_name):
354 return (header_name in self.headers or
355 header_name in self.unredirected_hdrs)
356
357 def get_header(self, header_name, default=None):
358 return self.headers.get(
359 header_name,
360 self.unredirected_hdrs.get(header_name, default))
361
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200362 def remove_header(self, header_name):
363 self.headers.pop(header_name, None)
364 self.unredirected_hdrs.pop(header_name, None)
365
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000366 def header_items(self):
367 hdrs = self.unredirected_hdrs.copy()
368 hdrs.update(self.headers)
369 return list(hdrs.items())
370
371class OpenerDirector:
372 def __init__(self):
373 client_version = "Python-urllib/%s" % __version__
374 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000375 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000376 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000377 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000378 self.handle_open = {}
379 self.handle_error = {}
380 self.process_response = {}
381 self.process_request = {}
382
383 def add_handler(self, handler):
384 if not hasattr(handler, "add_parent"):
385 raise TypeError("expected BaseHandler instance, got %r" %
386 type(handler))
387
388 added = False
389 for meth in dir(handler):
390 if meth in ["redirect_request", "do_open", "proxy_open"]:
391 # oops, coincidental match
392 continue
393
394 i = meth.find("_")
395 protocol = meth[:i]
396 condition = meth[i+1:]
397
398 if condition.startswith("error"):
399 j = condition.find("_") + i + 1
400 kind = meth[j+1:]
401 try:
402 kind = int(kind)
403 except ValueError:
404 pass
405 lookup = self.handle_error.get(protocol, {})
406 self.handle_error[protocol] = lookup
407 elif condition == "open":
408 kind = protocol
409 lookup = self.handle_open
410 elif condition == "response":
411 kind = protocol
412 lookup = self.process_response
413 elif condition == "request":
414 kind = protocol
415 lookup = self.process_request
416 else:
417 continue
418
419 handlers = lookup.setdefault(kind, [])
420 if handlers:
421 bisect.insort(handlers, handler)
422 else:
423 handlers.append(handler)
424 added = True
425
426 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000427 bisect.insort(self.handlers, handler)
428 handler.add_parent(self)
429
430 def close(self):
431 # Only exists for backwards compatibility.
432 pass
433
434 def _call_chain(self, chain, kind, meth_name, *args):
435 # Handlers raise an exception if no one else should try to handle
436 # the request, or return None if they can't but another handler
437 # could. Otherwise, they return the response.
438 handlers = chain.get(kind, ())
439 for handler in handlers:
440 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000441 result = func(*args)
442 if result is not None:
443 return result
444
445 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
446 # accept a URL or a Request object
447 if isinstance(fullurl, str):
448 req = Request(fullurl, data)
449 else:
450 req = fullurl
451 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000452 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000453
454 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000455 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000456
457 # pre-process request
458 meth_name = protocol+"_request"
459 for processor in self.process_request.get(protocol, []):
460 meth = getattr(processor, meth_name)
461 req = meth(req)
462
463 response = self._open(req, data)
464
465 # post-process response
466 meth_name = protocol+"_response"
467 for processor in self.process_response.get(protocol, []):
468 meth = getattr(processor, meth_name)
469 response = meth(req, response)
470
471 return response
472
473 def _open(self, req, data=None):
474 result = self._call_chain(self.handle_open, 'default',
475 'default_open', req)
476 if result:
477 return result
478
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000479 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000480 result = self._call_chain(self.handle_open, protocol, protocol +
481 '_open', req)
482 if result:
483 return result
484
485 return self._call_chain(self.handle_open, 'unknown',
486 'unknown_open', req)
487
488 def error(self, proto, *args):
489 if proto in ('http', 'https'):
490 # XXX http[s] protocols are special-cased
491 dict = self.handle_error['http'] # https is not different than http
492 proto = args[2] # YUCK!
493 meth_name = 'http_error_%s' % proto
494 http_err = 1
495 orig_args = args
496 else:
497 dict = self.handle_error
498 meth_name = proto + '_error'
499 http_err = 0
500 args = (dict, proto, meth_name) + args
501 result = self._call_chain(*args)
502 if result:
503 return result
504
505 if http_err:
506 args = (dict, 'default', 'http_error_default') + orig_args
507 return self._call_chain(*args)
508
509# XXX probably also want an abstract factory that knows when it makes
510# sense to skip a superclass in favor of a subclass and when it might
511# make sense to include both
512
513def build_opener(*handlers):
514 """Create an opener object from a list of handlers.
515
516 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000517 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000518
519 If any of the handlers passed as arguments are subclasses of the
520 default handlers, the default handlers will not be used.
521 """
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000522 opener = OpenerDirector()
523 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
524 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100525 FTPHandler, FileHandler, HTTPErrorProcessor,
526 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000527 if hasattr(http.client, "HTTPSConnection"):
528 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000529 skip = set()
530 for klass in default_classes:
531 for check in handlers:
Benjamin Peterson78c85382014-04-01 16:27:30 -0400532 if isinstance(check, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000533 if issubclass(check, klass):
534 skip.add(klass)
535 elif isinstance(check, klass):
536 skip.add(klass)
537 for klass in skip:
538 default_classes.remove(klass)
539
540 for klass in default_classes:
541 opener.add_handler(klass())
542
543 for h in handlers:
Benjamin Peterson5dd3cae2014-04-01 14:20:56 -0400544 if isinstance(h, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000545 h = h()
546 opener.add_handler(h)
547 return opener
548
549class BaseHandler:
550 handler_order = 500
551
552 def add_parent(self, parent):
553 self.parent = parent
554
555 def close(self):
556 # Only exists for backwards compatibility
557 pass
558
559 def __lt__(self, other):
560 if not hasattr(other, "handler_order"):
561 # Try to preserve the old behavior of having custom classes
562 # inserted after default ones (works only for custom user
563 # classes which are not aware of handler_order).
564 return True
565 return self.handler_order < other.handler_order
566
567
568class HTTPErrorProcessor(BaseHandler):
569 """Process HTTP error responses."""
570 handler_order = 1000 # after all other processing
571
572 def http_response(self, request, response):
573 code, msg, hdrs = response.code, response.msg, response.info()
574
575 # According to RFC 2616, "2xx" code indicates that the client's
576 # request was successfully received, understood, and accepted.
577 if not (200 <= code < 300):
578 response = self.parent.error(
579 'http', request, response, code, msg, hdrs)
580
581 return response
582
583 https_response = http_response
584
585class HTTPDefaultErrorHandler(BaseHandler):
586 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000587 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000588
589class HTTPRedirectHandler(BaseHandler):
590 # maximum number of redirections to any single URL
591 # this is needed because of the state that cookies introduce
592 max_repeats = 4
593 # maximum total number of redirections (regardless of URL) before
594 # assuming we're in a loop
595 max_redirections = 10
596
597 def redirect_request(self, req, fp, code, msg, headers, newurl):
598 """Return a Request or None in response to a redirect.
599
600 This is called by the http_error_30x methods when a
601 redirection response is received. If a redirection should
602 take place, return a new Request to allow http_error_30x to
603 perform the redirect. Otherwise, raise HTTPError if no-one
604 else should try to handle this url. Return None if you can't
605 but another Handler might.
606 """
607 m = req.get_method()
608 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
609 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000610 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000611
612 # Strictly (according to RFC 2616), 301 or 302 in response to
613 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000614 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000615 # essentially all clients do redirect in this case, so we do
616 # the same.
617 # be conciliant with URIs containing a space
618 newurl = newurl.replace(' ', '%20')
619 CONTENT_HEADERS = ("content-length", "content-type")
620 newheaders = dict((k, v) for k, v in req.headers.items()
621 if k.lower() not in CONTENT_HEADERS)
622 return Request(newurl,
623 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000624 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000625 unverifiable=True)
626
627 # Implementation note: To avoid the server sending us into an
628 # infinite loop, the request object needs to track what URLs we
629 # have already seen. Do this by adding a handler-specific
630 # attribute to the Request object.
631 def http_error_302(self, req, fp, code, msg, headers):
632 # Some servers (incorrectly) return multiple Location headers
633 # (so probably same goes for URI). Use first header.
634 if "location" in headers:
635 newurl = headers["location"]
636 elif "uri" in headers:
637 newurl = headers["uri"]
638 else:
639 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000640
641 # fix a possible malformed URL
642 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700643
644 # For security reasons we don't allow redirection to anything other
645 # than http, https or ftp.
646
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800647 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800648 raise HTTPError(
649 newurl, code,
650 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
651 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700652
Facundo Batistaf24802c2008-08-17 03:36:03 +0000653 if not urlparts.path:
654 urlparts = list(urlparts)
655 urlparts[2] = "/"
656 newurl = urlunparse(urlparts)
657
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000658 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000659
660 # XXX Probably want to forget about the state of the current
661 # request, although that might interact poorly with other
662 # handlers that also use handler-specific request attributes
663 new = self.redirect_request(req, fp, code, msg, headers, newurl)
664 if new is None:
665 return
666
667 # loop detection
668 # .redirect_dict has a key url if url was previously visited.
669 if hasattr(req, 'redirect_dict'):
670 visited = new.redirect_dict = req.redirect_dict
671 if (visited.get(newurl, 0) >= self.max_repeats or
672 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000673 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000674 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000675 else:
676 visited = new.redirect_dict = req.redirect_dict = {}
677 visited[newurl] = visited.get(newurl, 0) + 1
678
679 # Don't close the fp until we are sure that we won't use it
680 # with HTTPError.
681 fp.read()
682 fp.close()
683
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000684 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000685
686 http_error_301 = http_error_303 = http_error_307 = http_error_302
687
688 inf_msg = "The HTTP server returned a redirect error that would " \
689 "lead to an infinite loop.\n" \
690 "The last 30x error message was:\n"
691
692
693def _parse_proxy(proxy):
694 """Return (scheme, user, password, host/port) given a URL or an authority.
695
696 If a URL is supplied, it must have an authority (host:port) component.
697 According to RFC 3986, having an authority component means the URL must
Senthil Kumarand8e24f12014-04-14 16:32:20 -0400698 have two slashes after the scheme.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000699 """
Georg Brandl13e89462008-07-01 19:56:00 +0000700 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000701 if not r_scheme.startswith("/"):
702 # authority
703 scheme = None
704 authority = proxy
705 else:
706 # URL
707 if not r_scheme.startswith("//"):
708 raise ValueError("proxy URL with no authority: %r" % proxy)
709 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
710 # and 3.3.), path is empty or starts with '/'
711 end = r_scheme.find("/", 2)
712 if end == -1:
713 end = None
714 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000715 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000716 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000717 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000718 else:
719 user = password = None
720 return scheme, user, password, hostport
721
722class ProxyHandler(BaseHandler):
723 # Proxies must be in front
724 handler_order = 100
725
726 def __init__(self, proxies=None):
727 if proxies is None:
728 proxies = getproxies()
729 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
730 self.proxies = proxies
731 for type, url in proxies.items():
732 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200733 lambda r, proxy=url, type=type, meth=self.proxy_open:
734 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000735
736 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000737 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000738 proxy_type, user, password, hostport = _parse_proxy(proxy)
739 if proxy_type is None:
740 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000741
742 if req.host and proxy_bypass(req.host):
743 return None
744
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000745 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000746 user_pass = '%s:%s' % (unquote(user),
747 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000748 creds = base64.b64encode(user_pass.encode()).decode("ascii")
749 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000750 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000751 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000752 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000753 # let other handlers take care of it
754 return None
755 else:
756 # need to start over, because the other handlers don't
757 # grok the proxy's URL type
758 # e.g. if we have a constructor arg proxies like so:
759 # {'http': 'ftp://proxy.example.com'}, we may end up turning
760 # a request for http://acme.example.com/a into one for
761 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000762 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000763
764class HTTPPasswordMgr:
765
766 def __init__(self):
767 self.passwd = {}
768
769 def add_password(self, realm, uri, user, passwd):
770 # uri could be a single URI or a sequence
771 if isinstance(uri, str):
772 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800773 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000774 self.passwd[realm] = {}
775 for default_port in True, False:
776 reduced_uri = tuple(
777 [self.reduce_uri(u, default_port) for u in uri])
778 self.passwd[realm][reduced_uri] = (user, passwd)
779
780 def find_user_password(self, realm, authuri):
781 domains = self.passwd.get(realm, {})
782 for default_port in True, False:
783 reduced_authuri = self.reduce_uri(authuri, default_port)
784 for uris, authinfo in domains.items():
785 for uri in uris:
786 if self.is_suburi(uri, reduced_authuri):
787 return authinfo
788 return None, None
789
790 def reduce_uri(self, uri, default_port=True):
791 """Accept authority or URI and extract only the authority and path."""
792 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000793 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000794 if parts[1]:
795 # URI
796 scheme = parts[0]
797 authority = parts[1]
798 path = parts[2] or '/'
799 else:
800 # host or host:port
801 scheme = None
802 authority = uri
803 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000804 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000805 if default_port and port is None and scheme is not None:
806 dport = {"http": 80,
807 "https": 443,
808 }.get(scheme)
809 if dport is not None:
810 authority = "%s:%d" % (host, dport)
811 return authority, path
812
813 def is_suburi(self, base, test):
814 """Check if test is below base in a URI tree
815
816 Both args must be URIs in reduced form.
817 """
818 if base == test:
819 return True
820 if base[0] != test[0]:
821 return False
822 common = posixpath.commonprefix((base[1], test[1]))
823 if len(common) == len(base[1]):
824 return True
825 return False
826
827
828class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
829
830 def find_user_password(self, realm, authuri):
831 user, password = HTTPPasswordMgr.find_user_password(self, realm,
832 authuri)
833 if user is not None:
834 return user, password
835 return HTTPPasswordMgr.find_user_password(self, None, authuri)
836
837
838class AbstractBasicAuthHandler:
839
840 # XXX this allows for multiple auth-schemes, but will stupidly pick
841 # the last one with a realm specified.
842
843 # allow for double- and single-quoted realm values
844 # (single quotes are a violation of the RFC, but appear in the wild)
845 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800846 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000847
848 # XXX could pre-emptively send auth info already accepted (RFC 2617,
849 # end of section 2, and section 1.2 immediately after "credentials"
850 # production).
851
852 def __init__(self, password_mgr=None):
853 if password_mgr is None:
854 password_mgr = HTTPPasswordMgr()
855 self.passwd = password_mgr
856 self.add_password = self.passwd.add_password
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000857
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000858 def http_error_auth_reqed(self, authreq, host, req, headers):
859 # host may be an authority (without userinfo) or a URL with an
860 # authority
861 # XXX could be multiple headers
862 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000863
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000864 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800865 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800866 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800867 raise ValueError("AbstractBasicAuthHandler does not"
868 " support the following scheme: '%s'" %
869 scheme)
870 else:
871 mo = AbstractBasicAuthHandler.rx.search(authreq)
872 if mo:
873 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800874 if quote not in ['"',"'"]:
875 warnings.warn("Basic Auth Realm was unquoted",
876 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800877 if scheme.lower() == 'basic':
Senthil Kumaran78373762014-08-20 07:53:58 +0530878 return self.retry_http_basic_auth(host, req, realm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000879
880 def retry_http_basic_auth(self, host, req, realm):
881 user, pw = self.passwd.find_user_password(realm, host)
882 if pw is not None:
883 raw = "%s:%s" % (user, pw)
884 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
Senthil Kumaran78373762014-08-20 07:53:58 +0530885 if req.get_header(self.auth_header, None) == auth:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000886 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000887 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000888 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000889 else:
890 return None
891
892
893class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
894
895 auth_header = 'Authorization'
896
897 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000898 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000899 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000900 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000901 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000902
903
904class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
905
906 auth_header = 'Proxy-authorization'
907
908 def http_error_407(self, req, fp, code, msg, headers):
909 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000910 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000911 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
912 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000913 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000914 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000915 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000916 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000917
918
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800919# Return n random bytes.
920_randombytes = os.urandom
921
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000922
923class AbstractDigestAuthHandler:
924 # Digest authentication is specified in RFC 2617.
925
926 # XXX The client does not inspect the Authentication-Info header
927 # in a successful response.
928
929 # XXX It should be possible to test this implementation against
930 # a mock server that just generates a static set of challenges.
931
932 # XXX qop="auth-int" supports is shaky
933
934 def __init__(self, passwd=None):
935 if passwd is None:
936 passwd = HTTPPasswordMgr()
937 self.passwd = passwd
938 self.add_password = self.passwd.add_password
939 self.retried = 0
940 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000941 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000942
943 def reset_retry_count(self):
944 self.retried = 0
945
946 def http_error_auth_reqed(self, auth_header, host, req, headers):
947 authreq = headers.get(auth_header, None)
948 if self.retried > 5:
949 # Don't fail endlessly - if we failed once, we'll probably
950 # fail a second time. Hm. Unless the Password Manager is
951 # prompting for the information. Crap. This isn't great
952 # but it's better than the current 'repeat until recursion
953 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000954 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000955 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000956 else:
957 self.retried += 1
958 if authreq:
959 scheme = authreq.split()[0]
960 if scheme.lower() == 'digest':
961 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800962 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800963 raise ValueError("AbstractDigestAuthHandler does not support"
964 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000965
966 def retry_http_digest_auth(self, req, auth):
967 token, challenge = auth.split(' ', 1)
968 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
969 auth = self.get_authorization(req, chal)
970 if auth:
971 auth_val = 'Digest %s' % auth
972 if req.headers.get(self.auth_header, None) == auth_val:
973 return None
974 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000975 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000976 return resp
977
978 def get_cnonce(self, nonce):
979 # The cnonce-value is an opaque
980 # quoted string value provided by the client and used by both client
981 # and server to avoid chosen plaintext attacks, to provide mutual
982 # authentication, and to provide some message integrity protection.
983 # This isn't a fabulous effort, but it's probably Good Enough.
984 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800985 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000986 dig = hashlib.sha1(b).hexdigest()
987 return dig[:16]
988
989 def get_authorization(self, req, chal):
990 try:
991 realm = chal['realm']
992 nonce = chal['nonce']
993 qop = chal.get('qop')
994 algorithm = chal.get('algorithm', 'MD5')
995 # mod_digest doesn't send an opaque, even though it isn't
996 # supposed to be optional
997 opaque = chal.get('opaque', None)
998 except KeyError:
999 return None
1000
1001 H, KD = self.get_algorithm_impls(algorithm)
1002 if H is None:
1003 return None
1004
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001005 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001006 if user is None:
1007 return None
1008
1009 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001010 if req.data is not None:
1011 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001012 else:
1013 entdig = None
1014
1015 A1 = "%s:%s:%s" % (user, realm, pw)
1016 A2 = "%s:%s" % (req.get_method(),
1017 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001018 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001019 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001020 if nonce == self.last_nonce:
1021 self.nonce_count += 1
1022 else:
1023 self.nonce_count = 1
1024 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001025 ncvalue = '%08x' % self.nonce_count
1026 cnonce = self.get_cnonce(nonce)
1027 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1028 respdig = KD(H(A1), noncebit)
1029 elif qop is None:
1030 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1031 else:
1032 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001033 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001034
1035 # XXX should the partial digests be encoded too?
1036
1037 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001038 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001039 respdig)
1040 if opaque:
1041 base += ', opaque="%s"' % opaque
1042 if entdig:
1043 base += ', digest="%s"' % entdig
1044 base += ', algorithm="%s"' % algorithm
1045 if qop:
1046 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1047 return base
1048
1049 def get_algorithm_impls(self, algorithm):
1050 # lambdas assume digest modules are imported at the top level
1051 if algorithm == 'MD5':
1052 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1053 elif algorithm == 'SHA':
1054 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1055 # XXX MD5-sess
1056 KD = lambda s, d: H("%s:%s" % (s, d))
1057 return H, KD
1058
1059 def get_entity_digest(self, data, chal):
1060 # XXX not implemented yet
1061 return None
1062
1063
1064class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1065 """An authentication protocol defined by RFC 2069
1066
1067 Digest authentication improves on basic authentication because it
1068 does not transmit passwords in the clear.
1069 """
1070
1071 auth_header = 'Authorization'
1072 handler_order = 490 # before Basic auth
1073
1074 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001075 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001076 retry = self.http_error_auth_reqed('www-authenticate',
1077 host, req, headers)
1078 self.reset_retry_count()
1079 return retry
1080
1081
1082class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1083
1084 auth_header = 'Proxy-Authorization'
1085 handler_order = 490 # before Basic auth
1086
1087 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001088 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001089 retry = self.http_error_auth_reqed('proxy-authenticate',
1090 host, req, headers)
1091 self.reset_retry_count()
1092 return retry
1093
1094class AbstractHTTPHandler(BaseHandler):
1095
1096 def __init__(self, debuglevel=0):
1097 self._debuglevel = debuglevel
1098
1099 def set_http_debuglevel(self, level):
1100 self._debuglevel = level
1101
1102 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001103 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001104 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001105 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001106
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001107 if request.data is not None: # POST
1108 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001109 if isinstance(data, str):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001110 msg = "POST data should be bytes or an iterable of bytes. " \
1111 "It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001112 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001113 if not request.has_header('Content-type'):
1114 request.add_unredirected_header(
1115 'Content-type',
1116 'application/x-www-form-urlencoded')
1117 if not request.has_header('Content-length'):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001118 try:
1119 mv = memoryview(data)
1120 except TypeError:
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001121 if isinstance(data, collections.Iterable):
Georg Brandl61536042011-02-03 07:46:41 +00001122 raise ValueError("Content-Length should be specified "
1123 "for iterable data of type %r %r" % (type(data),
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001124 data))
1125 else:
1126 request.add_unredirected_header(
Senthil Kumaran1e991f22010-12-24 04:03:59 +00001127 'Content-length', '%d' % (len(mv) * mv.itemsize))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001128
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001129 sel_host = host
1130 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001131 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001132 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001133 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001134 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001135 for name, value in self.parent.addheaders:
1136 name = name.capitalize()
1137 if not request.has_header(name):
1138 request.add_unredirected_header(name, value)
1139
1140 return request
1141
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001142 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001143 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001144
1145 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001146 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001147 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001148 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001149 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001150
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001151 # will parse host:port
1152 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001153
1154 headers = dict(req.unredirected_hdrs)
1155 headers.update(dict((k, v) for k, v in req.headers.items()
1156 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001157
1158 # TODO(jhylton): Should this be redesigned to handle
1159 # persistent connections?
1160
1161 # We want to make an HTTP/1.1 request, but the addinfourl
1162 # class isn't prepared to deal with a persistent connection.
1163 # It will try to read all remaining data from the socket,
1164 # which will block while the server waits for the next request.
1165 # So make sure the connection gets closed after the (only)
1166 # request.
1167 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001168 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001169
1170 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001171 tunnel_headers = {}
1172 proxy_auth_hdr = "Proxy-Authorization"
1173 if proxy_auth_hdr in headers:
1174 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1175 # Proxy-Authorization should not be sent to origin
1176 # server.
1177 del headers[proxy_auth_hdr]
1178 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001179
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001180 try:
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001181 try:
1182 h.request(req.get_method(), req.selector, req.data, headers)
1183 except OSError as err: # timeout error
1184 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001185 r = h.getresponse()
Serhiy Storchakaf54c3502014-09-06 21:41:39 +03001186 except:
1187 h.close()
1188 raise
1189
1190 # If the server does not send us a 'Connection: close' header,
1191 # HTTPConnection assumes the socket should be left open. Manually
1192 # mark the socket to be closed when this response object goes away.
1193 if h.sock:
1194 h.sock.close()
1195 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001196
Senthil Kumaran26430412011-04-13 07:01:19 +08001197 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001198 # This line replaces the .msg attribute of the HTTPResponse
1199 # with .headers, because urllib clients expect the response to
1200 # have the reason in .msg. It would be good to mark this
1201 # attribute is deprecated and get then to use info() or
1202 # .headers.
1203 r.msg = r.reason
1204 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205
1206
1207class HTTPHandler(AbstractHTTPHandler):
1208
1209 def http_open(self, req):
1210 return self.do_open(http.client.HTTPConnection, req)
1211
1212 http_request = AbstractHTTPHandler.do_request_
1213
1214if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001215
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001216 class HTTPSHandler(AbstractHTTPHandler):
1217
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001218 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1219 AbstractHTTPHandler.__init__(self, debuglevel)
1220 self._context = context
1221 self._check_hostname = check_hostname
1222
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001223 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001224 return self.do_open(http.client.HTTPSConnection, req,
1225 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001226
1227 https_request = AbstractHTTPHandler.do_request_
1228
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001229 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001230
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001231class HTTPCookieProcessor(BaseHandler):
1232 def __init__(self, cookiejar=None):
1233 import http.cookiejar
1234 if cookiejar is None:
1235 cookiejar = http.cookiejar.CookieJar()
1236 self.cookiejar = cookiejar
1237
1238 def http_request(self, request):
1239 self.cookiejar.add_cookie_header(request)
1240 return request
1241
1242 def http_response(self, request, response):
1243 self.cookiejar.extract_cookies(response, request)
1244 return response
1245
1246 https_request = http_request
1247 https_response = http_response
1248
1249class UnknownHandler(BaseHandler):
1250 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001251 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001252 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001253
1254def parse_keqv_list(l):
1255 """Parse list of key=value strings where keys are not duplicated."""
1256 parsed = {}
1257 for elt in l:
1258 k, v = elt.split('=', 1)
1259 if v[0] == '"' and v[-1] == '"':
1260 v = v[1:-1]
1261 parsed[k] = v
1262 return parsed
1263
1264def parse_http_list(s):
1265 """Parse lists as described by RFC 2068 Section 2.
1266
1267 In particular, parse comma-separated lists where the elements of
1268 the list may include quoted-strings. A quoted-string could
1269 contain a comma. A non-quoted string could have quotes in the
1270 middle. Neither commas nor quotes count if they are escaped.
1271 Only double-quotes count, not single-quotes.
1272 """
1273 res = []
1274 part = ''
1275
1276 escape = quote = False
1277 for cur in s:
1278 if escape:
1279 part += cur
1280 escape = False
1281 continue
1282 if quote:
1283 if cur == '\\':
1284 escape = True
1285 continue
1286 elif cur == '"':
1287 quote = False
1288 part += cur
1289 continue
1290
1291 if cur == ',':
1292 res.append(part)
1293 part = ''
1294 continue
1295
1296 if cur == '"':
1297 quote = True
1298
1299 part += cur
1300
1301 # append last part
1302 if part:
1303 res.append(part)
1304
1305 return [part.strip() for part in res]
1306
1307class FileHandler(BaseHandler):
1308 # Use local file or FTP depending on form of URL
1309 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001310 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001311 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1312 req.host != 'localhost'):
Senthil Kumaranbc07ac52014-07-22 00:15:20 -07001313 if not req.host in self.get_names():
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001314 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001315 else:
1316 return self.open_local_file(req)
1317
1318 # names for the localhost
1319 names = None
1320 def get_names(self):
1321 if FileHandler.names is None:
1322 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001323 FileHandler.names = tuple(
1324 socket.gethostbyname_ex('localhost')[2] +
1325 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001326 except socket.gaierror:
1327 FileHandler.names = (socket.gethostbyname('localhost'),)
1328 return FileHandler.names
1329
1330 # not entirely sure what the rules are here
1331 def open_local_file(self, req):
1332 import email.utils
1333 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001334 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001335 filename = req.selector
1336 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001337 try:
1338 stats = os.stat(localfile)
1339 size = stats.st_size
1340 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001341 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001342 headers = email.message_from_string(
1343 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1344 (mtype or 'text/plain', size, modified))
1345 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001346 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001347 if not host or \
1348 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001349 if host:
1350 origurl = 'file://' + host + filename
1351 else:
1352 origurl = 'file://' + filename
1353 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001354 except OSError as exp:
Georg Brandl029986a2008-06-23 11:44:14 +00001355 # users shouldn't expect OSErrors coming from urlopen()
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001356 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001357 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001358
1359def _safe_gethostbyname(host):
1360 try:
1361 return socket.gethostbyname(host)
1362 except socket.gaierror:
1363 return None
1364
1365class FTPHandler(BaseHandler):
1366 def ftp_open(self, req):
1367 import ftplib
1368 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001369 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001370 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001371 raise URLError('ftp error: no host given')
1372 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001373 if port is None:
1374 port = ftplib.FTP_PORT
1375 else:
1376 port = int(port)
1377
1378 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001379 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001380 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001381 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001382 else:
1383 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001384 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001385 user = user or ''
1386 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001387
1388 try:
1389 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001390 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001391 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001392 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001393 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001394 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001395 dirs, file = dirs[:-1], dirs[-1]
1396 if dirs and not dirs[0]:
1397 dirs = dirs[1:]
1398 try:
1399 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1400 type = file and 'I' or 'D'
1401 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001402 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001403 if attr.lower() == 'type' and \
1404 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1405 type = value.upper()
1406 fp, retrlen = fw.retrfile(file, type)
1407 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001408 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001409 if mtype:
1410 headers += "Content-type: %s\n" % mtype
1411 if retrlen is not None and retrlen >= 0:
1412 headers += "Content-length: %d\n" % retrlen
1413 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001414 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001415 except ftplib.all_errors as exp:
1416 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001417 raise exc.with_traceback(sys.exc_info()[2])
1418
1419 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001420 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1421 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001422
1423class CacheFTPHandler(FTPHandler):
1424 # XXX would be nice to have pluggable cache strategies
1425 # XXX this stuff is definitely not thread safe
1426 def __init__(self):
1427 self.cache = {}
1428 self.timeout = {}
1429 self.soonest = 0
1430 self.delay = 60
1431 self.max_conns = 16
1432
1433 def setTimeout(self, t):
1434 self.delay = t
1435
1436 def setMaxConns(self, m):
1437 self.max_conns = m
1438
1439 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1440 key = user, host, port, '/'.join(dirs), timeout
1441 if key in self.cache:
1442 self.timeout[key] = time.time() + self.delay
1443 else:
1444 self.cache[key] = ftpwrapper(user, passwd, host, port,
1445 dirs, timeout)
1446 self.timeout[key] = time.time() + self.delay
1447 self.check_cache()
1448 return self.cache[key]
1449
1450 def check_cache(self):
1451 # first check for old ones
1452 t = time.time()
1453 if self.soonest <= t:
1454 for k, v in list(self.timeout.items()):
1455 if v < t:
1456 self.cache[k].close()
1457 del self.cache[k]
1458 del self.timeout[k]
1459 self.soonest = min(list(self.timeout.values()))
1460
1461 # then check the size
1462 if len(self.cache) == self.max_conns:
1463 for k, v in list(self.timeout.items()):
1464 if v == self.soonest:
1465 del self.cache[k]
1466 del self.timeout[k]
1467 break
1468 self.soonest = min(list(self.timeout.values()))
1469
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001470 def clear_cache(self):
1471 for conn in self.cache.values():
1472 conn.close()
1473 self.cache.clear()
1474 self.timeout.clear()
1475
Antoine Pitroudf204be2012-11-24 17:59:08 +01001476class DataHandler(BaseHandler):
1477 def data_open(self, req):
1478 # data URLs as specified in RFC 2397.
1479 #
1480 # ignores POSTed data
1481 #
1482 # syntax:
1483 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1484 # mediatype := [ type "/" subtype ] *( ";" parameter )
1485 # data := *urlchar
1486 # parameter := attribute "=" value
1487 url = req.full_url
1488
1489 scheme, data = url.split(":",1)
1490 mediatype, data = data.split(",",1)
1491
1492 # even base64 encoded data URLs might be quoted so unquote in any case:
1493 data = unquote_to_bytes(data)
1494 if mediatype.endswith(";base64"):
1495 data = base64.decodebytes(data)
1496 mediatype = mediatype[:-7]
1497
1498 if not mediatype:
1499 mediatype = "text/plain;charset=US-ASCII"
1500
1501 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1502 (mediatype, len(data)))
1503
1504 return addinfourl(io.BytesIO(data), headers, url)
1505
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001506
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001507# Code move from the old urllib module
1508
1509MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1510
1511# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001512if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001513 from nturl2path import url2pathname, pathname2url
1514else:
1515 def url2pathname(pathname):
1516 """OS-specific conversion from a relative URL of the 'file' scheme
1517 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001518 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001519
1520 def pathname2url(pathname):
1521 """OS-specific conversion from a file system path to a relative URL
1522 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001523 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001524
1525# This really consists of two pieces:
1526# (1) a class which handles opening of all sorts of URLs
1527# (plus assorted utilities etc.)
1528# (2) a set of functions for parsing URLs
1529# XXX Should these be separated out into different modules?
1530
1531
1532ftpcache = {}
1533class URLopener:
1534 """Class to open URLs.
1535 This is a class rather than just a subroutine because we may need
1536 more than one set of global protocol-specific options.
1537 Note -- this is a base class for those who don't want the
1538 automatic handling of errors type 302 (relocated) and 401
1539 (authorization needed)."""
1540
1541 __tempfiles = None
1542
1543 version = "Python-urllib/%s" % __version__
1544
1545 # Constructor
1546 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001547 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001548 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1549 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001550 if proxies is None:
1551 proxies = getproxies()
1552 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1553 self.proxies = proxies
1554 self.key_file = x509.get('key_file')
1555 self.cert_file = x509.get('cert_file')
1556 self.addheaders = [('User-Agent', self.version)]
1557 self.__tempfiles = []
1558 self.__unlink = os.unlink # See cleanup()
1559 self.tempcache = None
1560 # Undocumented feature: if you assign {} to tempcache,
1561 # it is used to cache files retrieved with
1562 # self.retrieve(). This is not enabled by default
1563 # since it does not work for changing documents (and I
1564 # haven't got the logic to check expiration headers
1565 # yet).
1566 self.ftpcache = ftpcache
1567 # Undocumented feature: you can use a different
1568 # ftp cache by assigning to the .ftpcache member;
1569 # in case you want logically independent URL openers
1570 # XXX This is not threadsafe. Bah.
1571
1572 def __del__(self):
1573 self.close()
1574
1575 def close(self):
1576 self.cleanup()
1577
1578 def cleanup(self):
1579 # This code sometimes runs when the rest of this module
1580 # has already been deleted, so it can't use any globals
1581 # or import anything.
1582 if self.__tempfiles:
1583 for file in self.__tempfiles:
1584 try:
1585 self.__unlink(file)
1586 except OSError:
1587 pass
1588 del self.__tempfiles[:]
1589 if self.tempcache:
1590 self.tempcache.clear()
1591
1592 def addheader(self, *args):
1593 """Add a header to be used by the HTTP interface only
1594 e.g. u.addheader('Accept', 'sound/basic')"""
1595 self.addheaders.append(args)
1596
1597 # External interface
1598 def open(self, fullurl, data=None):
1599 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001600 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001601 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001602 if self.tempcache and fullurl in self.tempcache:
1603 filename, headers = self.tempcache[fullurl]
1604 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001605 return addinfourl(fp, headers, fullurl)
1606 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001607 if not urltype:
1608 urltype = 'file'
1609 if urltype in self.proxies:
1610 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001611 urltype, proxyhost = splittype(proxy)
1612 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001613 url = (host, fullurl) # Signal special case to open_*()
1614 else:
1615 proxy = None
1616 name = 'open_' + urltype
1617 self.type = urltype
1618 name = name.replace('-', '_')
1619 if not hasattr(self, name):
1620 if proxy:
1621 return self.open_unknown_proxy(proxy, fullurl, data)
1622 else:
1623 return self.open_unknown(fullurl, data)
1624 try:
1625 if data is None:
1626 return getattr(self, name)(url)
1627 else:
1628 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001629 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001630 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001631 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001632 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001633
1634 def open_unknown(self, fullurl, data=None):
1635 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001636 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001637 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001638
1639 def open_unknown_proxy(self, proxy, fullurl, data=None):
1640 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001641 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001642 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001643
1644 # External interface
1645 def retrieve(self, url, filename=None, reporthook=None, data=None):
1646 """retrieve(url) returns (filename, headers) for a local object
1647 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001648 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001649 if self.tempcache and url in self.tempcache:
1650 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001651 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001652 if filename is None and (not type or type == 'file'):
1653 try:
1654 fp = self.open_local_file(url1)
1655 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001656 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001657 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001658 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001659 pass
1660 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001661 try:
1662 headers = fp.info()
1663 if filename:
1664 tfp = open(filename, 'wb')
1665 else:
1666 import tempfile
1667 garbage, path = splittype(url)
1668 garbage, path = splithost(path or "")
1669 path, garbage = splitquery(path or "")
1670 path, garbage = splitattr(path or "")
1671 suffix = os.path.splitext(path)[1]
1672 (fd, filename) = tempfile.mkstemp(suffix)
1673 self.__tempfiles.append(filename)
1674 tfp = os.fdopen(fd, 'wb')
1675 try:
1676 result = filename, headers
1677 if self.tempcache is not None:
1678 self.tempcache[url] = result
1679 bs = 1024*8
1680 size = -1
1681 read = 0
1682 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001683 if "content-length" in headers:
1684 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001685 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001686 reporthook(blocknum, bs, size)
1687 while 1:
1688 block = fp.read(bs)
1689 if not block:
1690 break
1691 read += len(block)
1692 tfp.write(block)
1693 blocknum += 1
1694 if reporthook:
1695 reporthook(blocknum, bs, size)
1696 finally:
1697 tfp.close()
1698 finally:
1699 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001700
1701 # raise exception if actual size does not match content-length header
1702 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001703 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001704 "retrieval incomplete: got only %i out of %i bytes"
1705 % (read, size), result)
1706
1707 return result
1708
1709 # Each method named open_<type> knows how to open that type of URL
1710
1711 def _open_generic_http(self, connection_factory, url, data):
1712 """Make an HTTP connection using connection_class.
1713
1714 This is an internal method that should be called from
1715 open_http() or open_https().
1716
1717 Arguments:
1718 - connection_factory should take a host name and return an
1719 HTTPConnection instance.
1720 - url is the url to retrieval or a host, relative-path pair.
1721 - data is payload for a POST request or None.
1722 """
1723
1724 user_passwd = None
1725 proxy_passwd= None
1726 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001727 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001728 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001729 user_passwd, host = splituser(host)
1730 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001731 realhost = host
1732 else:
1733 host, selector = url
1734 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001735 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001736 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001737 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001738 url = rest
1739 user_passwd = None
1740 if urltype.lower() != 'http':
1741 realhost = None
1742 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001743 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001744 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001745 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001746 if user_passwd:
1747 selector = "%s://%s%s" % (urltype, realhost, rest)
1748 if proxy_bypass(realhost):
1749 host = realhost
1750
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001751 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001752
1753 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001754 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001755 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001756 else:
1757 proxy_auth = None
1758
1759 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001760 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001761 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001762 else:
1763 auth = None
1764 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001765 headers = {}
1766 if proxy_auth:
1767 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1768 if auth:
1769 headers["Authorization"] = "Basic %s" % auth
1770 if realhost:
1771 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001772
1773 # Add Connection:close as we don't support persistent connections yet.
1774 # This helps in closing the socket and avoiding ResourceWarning
1775
1776 headers["Connection"] = "close"
1777
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001778 for header, value in self.addheaders:
1779 headers[header] = value
1780
1781 if data is not None:
1782 headers["Content-Type"] = "application/x-www-form-urlencoded"
1783 http_conn.request("POST", selector, data, headers)
1784 else:
1785 http_conn.request("GET", selector, headers=headers)
1786
1787 try:
1788 response = http_conn.getresponse()
1789 except http.client.BadStatusLine:
1790 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001791 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001792
1793 # According to RFC 2616, "2xx" code indicates that the client's
1794 # request was successfully received, understood, and accepted.
1795 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001796 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001797 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001798 else:
1799 return self.http_error(
1800 url, response.fp,
1801 response.status, response.reason, response.msg, data)
1802
1803 def open_http(self, url, data=None):
1804 """Use HTTP protocol."""
1805 return self._open_generic_http(http.client.HTTPConnection, url, data)
1806
1807 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1808 """Handle http errors.
1809
1810 Derived class can override this, or provide specific handlers
1811 named http_error_DDD where DDD is the 3-digit error code."""
1812 # First check if there's a specific handler for this error
1813 name = 'http_error_%d' % errcode
1814 if hasattr(self, name):
1815 method = getattr(self, name)
1816 if data is None:
1817 result = method(url, fp, errcode, errmsg, headers)
1818 else:
1819 result = method(url, fp, errcode, errmsg, headers, data)
1820 if result: return result
1821 return self.http_error_default(url, fp, errcode, errmsg, headers)
1822
1823 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001824 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001825 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001826 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001827
1828 if _have_ssl:
1829 def _https_connection(self, host):
1830 return http.client.HTTPSConnection(host,
1831 key_file=self.key_file,
1832 cert_file=self.cert_file)
1833
1834 def open_https(self, url, data=None):
1835 """Use HTTPS protocol."""
1836 return self._open_generic_http(self._https_connection, url, data)
1837
1838 def open_file(self, url):
1839 """Use local file or FTP depending on form of URL."""
1840 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001841 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001842 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001843 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001844 else:
1845 return self.open_local_file(url)
1846
1847 def open_local_file(self, url):
1848 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001849 import email.utils
1850 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001851 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001852 localname = url2pathname(file)
1853 try:
1854 stats = os.stat(localname)
1855 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001856 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001857 size = stats.st_size
1858 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1859 mtype = mimetypes.guess_type(url)[0]
1860 headers = email.message_from_string(
1861 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1862 (mtype or 'text/plain', size, modified))
1863 if not host:
1864 urlfile = file
1865 if file[:1] == '/':
1866 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001867 return addinfourl(open(localname, 'rb'), headers, urlfile)
1868 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001869 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07001870 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001871 urlfile = file
1872 if file[:1] == '/':
1873 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08001874 elif file[:2] == './':
1875 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00001876 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001877 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001878
1879 def open_ftp(self, url):
1880 """Use FTP protocol."""
1881 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001882 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001883 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001884 host, path = splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001885 if not host: raise URLError('ftp error: no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001886 host, port = splitport(host)
1887 user, host = splituser(host)
1888 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001889 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001890 host = unquote(host)
1891 user = unquote(user or '')
1892 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001893 host = socket.gethostbyname(host)
1894 if not port:
1895 import ftplib
1896 port = ftplib.FTP_PORT
1897 else:
1898 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001899 path, attrs = splitattr(path)
1900 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001901 dirs = path.split('/')
1902 dirs, file = dirs[:-1], dirs[-1]
1903 if dirs and not dirs[0]: dirs = dirs[1:]
1904 if dirs and not dirs[0]: dirs[0] = '/'
1905 key = user, host, port, '/'.join(dirs)
1906 # XXX thread unsafe!
1907 if len(self.ftpcache) > MAXFTPCACHE:
1908 # Prune the cache, rather arbitrarily
Benjamin Peterson3c2dca62014-06-07 15:08:04 -07001909 for k in list(self.ftpcache):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001910 if k != key:
1911 v = self.ftpcache[k]
1912 del self.ftpcache[k]
1913 v.close()
1914 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001915 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001916 self.ftpcache[key] = \
1917 ftpwrapper(user, passwd, host, port, dirs)
1918 if not file: type = 'D'
1919 else: type = 'I'
1920 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001921 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001922 if attr.lower() == 'type' and \
1923 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1924 type = value.upper()
1925 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1926 mtype = mimetypes.guess_type("ftp:" + url)[0]
1927 headers = ""
1928 if mtype:
1929 headers += "Content-Type: %s\n" % mtype
1930 if retrlen is not None and retrlen >= 0:
1931 headers += "Content-Length: %d\n" % retrlen
1932 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001933 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001934 except ftperrors() as exp:
1935 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001936
1937 def open_data(self, url, data=None):
1938 """Use "data" URL."""
1939 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001940 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001941 # ignore POSTed data
1942 #
1943 # syntax of data URLs:
1944 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1945 # mediatype := [ type "/" subtype ] *( ";" parameter )
1946 # data := *urlchar
1947 # parameter := attribute "=" value
1948 try:
1949 [type, data] = url.split(',', 1)
1950 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001951 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001952 if not type:
1953 type = 'text/plain;charset=US-ASCII'
1954 semi = type.rfind(';')
1955 if semi >= 0 and '=' not in type[semi:]:
1956 encoding = type[semi+1:]
1957 type = type[:semi]
1958 else:
1959 encoding = ''
1960 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00001961 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001962 time.gmtime(time.time())))
1963 msg.append('Content-type: %s' % type)
1964 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00001965 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001966 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001967 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001968 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001969 msg.append('Content-Length: %d' % len(data))
1970 msg.append('')
1971 msg.append(data)
1972 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001973 headers = email.message_from_string(msg)
1974 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001975 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001976 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001977
1978
1979class FancyURLopener(URLopener):
1980 """Derived class with handlers for errors we can handle (perhaps)."""
1981
1982 def __init__(self, *args, **kwargs):
1983 URLopener.__init__(self, *args, **kwargs)
1984 self.auth_cache = {}
1985 self.tries = 0
1986 self.maxtries = 10
1987
1988 def http_error_default(self, url, fp, errcode, errmsg, headers):
1989 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001990 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001991
1992 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1993 """Error 302 -- relocated (temporarily)."""
1994 self.tries += 1
1995 if self.maxtries and self.tries >= self.maxtries:
1996 if hasattr(self, "http_error_500"):
1997 meth = self.http_error_500
1998 else:
1999 meth = self.http_error_default
2000 self.tries = 0
2001 return meth(url, fp, 500,
2002 "Internal Server Error: Redirect Recursion", headers)
2003 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
2004 data)
2005 self.tries = 0
2006 return result
2007
2008 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
2009 if 'location' in headers:
2010 newurl = headers['location']
2011 elif 'uri' in headers:
2012 newurl = headers['uri']
2013 else:
2014 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002015 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002016
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002017 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002018 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002019
2020 urlparts = urlparse(newurl)
2021
2022 # For security reasons, we don't allow redirection to anything other
2023 # than http, https and ftp.
2024
2025 # We are using newer HTTPError with older redirect_internal method
2026 # This older method will get deprecated in 3.3
2027
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002028 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002029 raise HTTPError(newurl, errcode,
2030 errmsg +
2031 " Redirection to url '%s' is not allowed." % newurl,
2032 headers, fp)
2033
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002034 return self.open(newurl)
2035
2036 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2037 """Error 301 -- also relocated (permanently)."""
2038 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2039
2040 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2041 """Error 303 -- also relocated (essentially identical to 302)."""
2042 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2043
2044 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2045 """Error 307 -- relocated, but turn POST into error."""
2046 if data is None:
2047 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2048 else:
2049 return self.http_error_default(url, fp, errcode, errmsg, headers)
2050
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002051 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2052 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002053 """Error 401 -- authentication required.
2054 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002055 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002056 URLopener.http_error_default(self, url, fp,
2057 errcode, errmsg, headers)
2058 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002059 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2060 if not match:
2061 URLopener.http_error_default(self, url, fp,
2062 errcode, errmsg, headers)
2063 scheme, realm = match.groups()
2064 if scheme.lower() != 'basic':
2065 URLopener.http_error_default(self, url, fp,
2066 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002067 if not retry:
2068 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2069 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002070 name = 'retry_' + self.type + '_basic_auth'
2071 if data is None:
2072 return getattr(self,name)(url, realm)
2073 else:
2074 return getattr(self,name)(url, realm, data)
2075
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002076 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2077 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002078 """Error 407 -- proxy authentication required.
2079 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002080 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002081 URLopener.http_error_default(self, url, fp,
2082 errcode, errmsg, headers)
2083 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002084 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2085 if not match:
2086 URLopener.http_error_default(self, url, fp,
2087 errcode, errmsg, headers)
2088 scheme, realm = match.groups()
2089 if scheme.lower() != 'basic':
2090 URLopener.http_error_default(self, url, fp,
2091 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002092 if not retry:
2093 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2094 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002095 name = 'retry_proxy_' + self.type + '_basic_auth'
2096 if data is None:
2097 return getattr(self,name)(url, realm)
2098 else:
2099 return getattr(self,name)(url, realm, data)
2100
2101 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002102 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002103 newurl = 'http://' + host + selector
2104 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002105 urltype, proxyhost = splittype(proxy)
2106 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002107 i = proxyhost.find('@') + 1
2108 proxyhost = proxyhost[i:]
2109 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2110 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002111 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002112 quote(passwd, safe=''), proxyhost)
2113 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2114 if data is None:
2115 return self.open(newurl)
2116 else:
2117 return self.open(newurl, data)
2118
2119 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002120 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002121 newurl = 'https://' + host + selector
2122 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002123 urltype, proxyhost = splittype(proxy)
2124 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002125 i = proxyhost.find('@') + 1
2126 proxyhost = proxyhost[i:]
2127 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2128 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002129 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002130 quote(passwd, safe=''), proxyhost)
2131 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2132 if data is None:
2133 return self.open(newurl)
2134 else:
2135 return self.open(newurl, data)
2136
2137 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002138 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002139 i = host.find('@') + 1
2140 host = host[i:]
2141 user, passwd = self.get_user_passwd(host, realm, i)
2142 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002143 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002144 quote(passwd, safe=''), host)
2145 newurl = 'http://' + host + selector
2146 if data is None:
2147 return self.open(newurl)
2148 else:
2149 return self.open(newurl, data)
2150
2151 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002152 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002153 i = host.find('@') + 1
2154 host = host[i:]
2155 user, passwd = self.get_user_passwd(host, realm, i)
2156 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002157 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002158 quote(passwd, safe=''), host)
2159 newurl = 'https://' + host + selector
2160 if data is None:
2161 return self.open(newurl)
2162 else:
2163 return self.open(newurl, data)
2164
Florent Xicluna757445b2010-05-17 17:24:07 +00002165 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002166 key = realm + '@' + host.lower()
2167 if key in self.auth_cache:
2168 if clear_cache:
2169 del self.auth_cache[key]
2170 else:
2171 return self.auth_cache[key]
2172 user, passwd = self.prompt_user_passwd(host, realm)
2173 if user or passwd: self.auth_cache[key] = (user, passwd)
2174 return user, passwd
2175
2176 def prompt_user_passwd(self, host, realm):
2177 """Override this in a GUI environment!"""
2178 import getpass
2179 try:
2180 user = input("Enter username for %s at %s: " % (realm, host))
2181 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2182 (user, realm, host))
2183 return user, passwd
2184 except KeyboardInterrupt:
2185 print()
2186 return None, None
2187
2188
2189# Utility functions
2190
2191_localhost = None
2192def localhost():
2193 """Return the IP address of the magic hostname 'localhost'."""
2194 global _localhost
2195 if _localhost is None:
2196 _localhost = socket.gethostbyname('localhost')
2197 return _localhost
2198
2199_thishost = None
2200def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002201 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002202 global _thishost
2203 if _thishost is None:
Senthil Kumarandcdadfe2013-06-01 11:12:17 -07002204 try:
2205 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
2206 except socket.gaierror:
2207 _thishost = tuple(socket.gethostbyname_ex('localhost')[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002208 return _thishost
2209
2210_ftperrors = None
2211def ftperrors():
2212 """Return the set of errors raised by the FTP class."""
2213 global _ftperrors
2214 if _ftperrors is None:
2215 import ftplib
2216 _ftperrors = ftplib.all_errors
2217 return _ftperrors
2218
2219_noheaders = None
2220def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002221 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002222 global _noheaders
2223 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002224 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002225 return _noheaders
2226
2227
2228# Utility classes
2229
2230class ftpwrapper:
2231 """Class used by open_ftp() for cache of open FTP connections."""
2232
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002233 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2234 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002235 self.user = user
2236 self.passwd = passwd
2237 self.host = host
2238 self.port = port
2239 self.dirs = dirs
2240 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002241 self.refcount = 0
2242 self.keepalive = persistent
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002243 self.init()
2244
2245 def init(self):
2246 import ftplib
2247 self.busy = 0
2248 self.ftp = ftplib.FTP()
2249 self.ftp.connect(self.host, self.port, self.timeout)
2250 self.ftp.login(self.user, self.passwd)
Senthil Kumarancaa00fe2013-06-02 11:59:47 -07002251 _target = '/'.join(self.dirs)
2252 self.ftp.cwd(_target)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002253
2254 def retrfile(self, file, type):
2255 import ftplib
2256 self.endtransfer()
2257 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2258 else: cmd = 'TYPE ' + type; isdir = 0
2259 try:
2260 self.ftp.voidcmd(cmd)
2261 except ftplib.all_errors:
2262 self.init()
2263 self.ftp.voidcmd(cmd)
2264 conn = None
2265 if file and not isdir:
2266 # Try to retrieve as a file
2267 try:
2268 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002269 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002270 except ftplib.error_perm as reason:
2271 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002272 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002273 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002274 if not conn:
2275 # Set transfer mode to ASCII!
2276 self.ftp.voidcmd('TYPE A')
2277 # Try a directory listing. Verify that directory exists.
2278 if file:
2279 pwd = self.ftp.pwd()
2280 try:
2281 try:
2282 self.ftp.cwd(file)
2283 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002284 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002285 finally:
2286 self.ftp.cwd(pwd)
2287 cmd = 'LIST ' + file
2288 else:
2289 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002290 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002291 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002292
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002293 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2294 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002295 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002296 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002297 return (ftpobj, retrlen)
2298
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002299 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002300 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002301
2302 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002303 self.keepalive = False
2304 if self.refcount <= 0:
2305 self.real_close()
2306
2307 def file_close(self):
2308 self.endtransfer()
2309 self.refcount -= 1
2310 if self.refcount <= 0 and not self.keepalive:
2311 self.real_close()
2312
2313 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002314 self.endtransfer()
2315 try:
2316 self.ftp.close()
2317 except ftperrors():
2318 pass
2319
2320# Proxy handling
2321def getproxies_environment():
2322 """Return a dictionary of scheme -> proxy server URL mappings.
2323
2324 Scan the environment for variables named <scheme>_proxy;
2325 this seems to be the standard convention. If you need a
2326 different way, you can pass a proxies dictionary to the
2327 [Fancy]URLopener constructor.
2328
2329 """
2330 proxies = {}
2331 for name, value in os.environ.items():
2332 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002333 if value and name[-6:] == '_proxy':
2334 proxies[name[:-6]] = value
2335 return proxies
2336
2337def proxy_bypass_environment(host):
2338 """Test if proxies should not be used for a particular host.
2339
2340 Checks the environment for a variable named no_proxy, which should
2341 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2342 """
2343 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2344 # '*' is special case for always bypass
2345 if no_proxy == '*':
2346 return 1
2347 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002348 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002349 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002350 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2351 for name in no_proxy_list:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002352 if name and (hostonly.endswith(name) or host.endswith(name)):
2353 return 1
2354 # otherwise, don't bypass
2355 return 0
2356
2357
Ronald Oussorene72e1612011-03-14 18:15:25 -04002358# This code tests an OSX specific data structure but is testable on all
2359# platforms
2360def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2361 """
2362 Return True iff this host shouldn't be accessed using a proxy
2363
2364 This function uses the MacOSX framework SystemConfiguration
2365 to fetch the proxy information.
2366
2367 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2368 { 'exclude_simple': bool,
2369 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2370 }
2371 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002372 from fnmatch import fnmatch
2373
2374 hostonly, port = splitport(host)
2375
2376 def ip2num(ipAddr):
2377 parts = ipAddr.split('.')
2378 parts = list(map(int, parts))
2379 if len(parts) != 4:
2380 parts = (parts + [0, 0, 0, 0])[:4]
2381 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2382
2383 # Check for simple host names:
2384 if '.' not in host:
2385 if proxy_settings['exclude_simple']:
2386 return True
2387
2388 hostIP = None
2389
2390 for value in proxy_settings.get('exceptions', ()):
2391 # Items in the list are strings like these: *.local, 169.254/16
2392 if not value: continue
2393
2394 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2395 if m is not None:
2396 if hostIP is None:
2397 try:
2398 hostIP = socket.gethostbyname(hostonly)
2399 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002400 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002401 continue
2402
2403 base = ip2num(m.group(1))
2404 mask = m.group(2)
2405 if mask is None:
2406 mask = 8 * (m.group(1).count('.') + 1)
2407 else:
2408 mask = int(mask[1:])
2409 mask = 32 - mask
2410
2411 if (hostIP >> mask) == (base >> mask):
2412 return True
2413
2414 elif fnmatch(host, value):
2415 return True
2416
2417 return False
2418
2419
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002420if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002421 from _scproxy import _get_proxy_settings, _get_proxies
2422
2423 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002424 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002425 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002426
2427 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002428 """Return a dictionary of scheme -> proxy server URL mappings.
2429
Ronald Oussoren84151202010-04-18 20:46:11 +00002430 This function uses the MacOSX framework SystemConfiguration
2431 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002432 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002433 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002434
Ronald Oussoren84151202010-04-18 20:46:11 +00002435
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002436
2437 def proxy_bypass(host):
2438 if getproxies_environment():
2439 return proxy_bypass_environment(host)
2440 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002441 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002442
2443 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002444 return getproxies_environment() or getproxies_macosx_sysconf()
2445
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002446
2447elif os.name == 'nt':
2448 def getproxies_registry():
2449 """Return a dictionary of scheme -> proxy server URL mappings.
2450
2451 Win32 uses the registry to store proxies.
2452
2453 """
2454 proxies = {}
2455 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002456 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002457 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002458 # Std module, so should be around - but you never know!
2459 return proxies
2460 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002461 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002462 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002463 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002464 'ProxyEnable')[0]
2465 if proxyEnable:
2466 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002467 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002468 'ProxyServer')[0])
2469 if '=' in proxyServer:
2470 # Per-protocol settings
2471 for p in proxyServer.split(';'):
2472 protocol, address = p.split('=', 1)
2473 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002474 if not re.match('^([^/:]+)://', address):
2475 address = '%s://%s' % (protocol, address)
2476 proxies[protocol] = address
2477 else:
2478 # Use one setting for all protocols
2479 if proxyServer[:5] == 'http:':
2480 proxies['http'] = proxyServer
2481 else:
2482 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002483 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002484 proxies['ftp'] = 'ftp://%s' % proxyServer
2485 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002486 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002487 # Either registry key not found etc, or the value in an
2488 # unexpected format.
2489 # proxies already set up to be empty so nothing to do
2490 pass
2491 return proxies
2492
2493 def getproxies():
2494 """Return a dictionary of scheme -> proxy server URL mappings.
2495
2496 Returns settings gathered from the environment, if specified,
2497 or the registry.
2498
2499 """
2500 return getproxies_environment() or getproxies_registry()
2501
2502 def proxy_bypass_registry(host):
2503 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002504 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002505 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002506 # Std modules, so should be around - but you never know!
2507 return 0
2508 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002509 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002510 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002511 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002512 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002513 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002514 'ProxyOverride')[0])
2515 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002516 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002517 return 0
2518 if not proxyEnable or not proxyOverride:
2519 return 0
2520 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002521 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002522 host = [rawHost]
2523 try:
2524 addr = socket.gethostbyname(rawHost)
2525 if addr != rawHost:
2526 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002527 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002528 pass
2529 try:
2530 fqdn = socket.getfqdn(rawHost)
2531 if fqdn != rawHost:
2532 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002533 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002534 pass
2535 # make a check value list from the registry entry: replace the
2536 # '<local>' string by the localhost entry and the corresponding
2537 # canonical entry.
2538 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002539 # now check if we match one of the registry values.
2540 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002541 if test == '<local>':
2542 if '.' not in rawHost:
2543 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002544 test = test.replace(".", r"\.") # mask dots
2545 test = test.replace("*", r".*") # change glob sequence
2546 test = test.replace("?", r".") # change glob char
2547 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002548 if re.match(test, val, re.I):
2549 return 1
2550 return 0
2551
2552 def proxy_bypass(host):
2553 """Return a dictionary of scheme -> proxy server URL mappings.
2554
2555 Returns settings gathered from the environment, if specified,
2556 or the registry.
2557
2558 """
2559 if getproxies_environment():
2560 return proxy_bypass_environment(host)
2561 else:
2562 return proxy_bypass_registry(host)
2563
2564else:
2565 # By default use environment variables
2566 getproxies = getproxies_environment
2567 proxy_bypass = proxy_bypass_environment