blob: a17c86859aa986248bb9f0211a6f1b6c3a8e43cb [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,
Antoine Pitroude9ac6c2012-05-16 21:40:01 +0200139 *, cafile=None, capath=None, cadefault=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000140 global _opener
Antoine Pitroude9ac6c2012-05-16 21:40:01 +0200141 if cafile or capath or cadefault:
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000142 if not _have_ssl:
143 raise ValueError('SSL support not available')
Christian Heimes67986f92013-11-23 22:43:47 +0100144 context = ssl._create_stdlib_context(cert_reqs=ssl.CERT_REQUIRED,
145 cafile=cafile,
146 capath=capath)
Antoine Pitrou9a8d6932013-04-01 18:55:35 +0200147 https_handler = HTTPSHandler(context=context, check_hostname=True)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000148 opener = build_opener(https_handler)
149 elif _opener is None:
150 _opener = opener = build_opener()
151 else:
152 opener = _opener
153 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000154
155def install_opener(opener):
156 global _opener
157 _opener = opener
158
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700159_url_tempfiles = []
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000160def urlretrieve(url, filename=None, reporthook=None, data=None):
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700161 """
162 Retrieve a URL into a temporary location on disk.
163
164 Requires a URL argument. If a filename is passed, it is used as
165 the temporary file location. The reporthook argument should be
166 a callable that accepts a block number, a read size, and the
167 total file size of the URL target. The data argument should be
168 valid URL encoded data.
169
170 If a filename is passed and the URL points to a local resource,
171 the result is a copy from local file to new file.
172
173 Returns a tuple containing the path to the newly created
174 data file as well as the resulting HTTPMessage object.
175 """
176 url_type, path = splittype(url)
177
178 with contextlib.closing(urlopen(url, data)) as fp:
179 headers = fp.info()
180
181 # Just return the local path and the "headers" for file://
182 # URLs. No sense in performing a copy unless requested.
183 if url_type == "file" and not filename:
184 return os.path.normpath(path), headers
185
186 # Handle temporary file setup.
187 if filename:
188 tfp = open(filename, 'wb')
189 else:
190 tfp = tempfile.NamedTemporaryFile(delete=False)
191 filename = tfp.name
192 _url_tempfiles.append(filename)
193
194 with tfp:
195 result = filename, headers
196 bs = 1024*8
197 size = -1
198 read = 0
199 blocknum = 0
200 if "content-length" in headers:
201 size = int(headers["Content-Length"])
202
203 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800204 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700205
206 while True:
207 block = fp.read(bs)
208 if not block:
209 break
210 read += len(block)
211 tfp.write(block)
212 blocknum += 1
213 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800214 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700215
216 if size >= 0 and read < size:
217 raise ContentTooShortError(
218 "retrieval incomplete: got only %i out of %i bytes"
219 % (read, size), result)
220
221 return result
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000222
223def urlcleanup():
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700224 for temp_file in _url_tempfiles:
225 try:
226 os.unlink(temp_file)
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200227 except OSError:
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700228 pass
229
230 del _url_tempfiles[:]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000231 global _opener
232 if _opener:
233 _opener = None
234
235# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000236_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000237def request_host(request):
238 """Return request-host, as defined by RFC 2965.
239
240 Variation from RFC: returned value is lowercased, for convenient
241 comparison.
242
243 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000244 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000245 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000246 if host == "":
247 host = request.get_header("Host", "")
248
249 # remove port, if present
250 host = _cut_port_re.sub("", host, 1)
251 return host.lower()
252
253class Request:
254
255 def __init__(self, url, data=None, headers={},
Senthil Kumarande49d642011-10-16 23:54:44 +0800256 origin_req_host=None, unverifiable=False,
257 method=None):
Senthil Kumaran52380922013-04-25 05:45:48 -0700258 self.full_url = url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000259 self.headers = {}
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200260 self.unredirected_hdrs = {}
261 self._data = None
262 self.data = data
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000263 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000264 for key, value in headers.items():
265 self.add_header(key, value)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000266 if origin_req_host is None:
267 origin_req_host = request_host(self)
268 self.origin_req_host = origin_req_host
269 self.unverifiable = unverifiable
Jason R. Coombs7dc4f4b2013-09-08 12:47:07 -0400270 if method:
271 self.method = method
Senthil Kumaran52380922013-04-25 05:45:48 -0700272
273 @property
274 def full_url(self):
Senthil Kumaran83070752013-05-24 09:14:12 -0700275 if self.fragment:
276 return '{}#{}'.format(self._full_url, self.fragment)
Senthil Kumaran52380922013-04-25 05:45:48 -0700277 return self._full_url
278
279 @full_url.setter
280 def full_url(self, url):
281 # unwrap('<URL:type://host/path>') --> 'type://host/path'
282 self._full_url = unwrap(url)
283 self._full_url, self.fragment = splittag(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000284 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000285
Senthil Kumaran52380922013-04-25 05:45:48 -0700286 @full_url.deleter
287 def full_url(self):
288 self._full_url = None
289 self.fragment = None
290 self.selector = ''
291
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200292 @property
293 def data(self):
294 return self._data
295
296 @data.setter
297 def data(self, data):
298 if data != self._data:
299 self._data = data
300 # issue 16464
301 # if we change data we need to remove content-length header
302 # (cause it's most probably calculated for previous value)
303 if self.has_header("Content-length"):
304 self.remove_header("Content-length")
305
306 @data.deleter
307 def data(self):
R David Murray9cc7d452013-03-20 00:10:51 -0400308 self.data = None
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200309
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000310 def _parse(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700311 self.type, rest = splittype(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000312 if self.type is None:
R David Murrayd8a46962013-04-03 06:58:34 -0400313 raise ValueError("unknown url type: %r" % self.full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000314 self.host, self.selector = splithost(rest)
315 if self.host:
316 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000317
318 def get_method(self):
Senthil Kumarande49d642011-10-16 23:54:44 +0800319 """Return a string indicating the HTTP request method."""
Jason R. Coombsaae6a1d2013-09-08 12:54:33 -0400320 default_method = "POST" if self.data is not None else "GET"
321 return getattr(self, 'method', default_method)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000322
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000323 def get_full_url(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700324 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000325
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000326 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000327 if self.type == 'https' and not self._tunnel_host:
328 self._tunnel_host = self.host
329 else:
330 self.type= type
331 self.selector = self.full_url
332 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000333
334 def has_proxy(self):
335 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000336
337 def add_header(self, key, val):
338 # useful for something like authentication
339 self.headers[key.capitalize()] = val
340
341 def add_unredirected_header(self, key, val):
342 # will not be added to a redirected request
343 self.unredirected_hdrs[key.capitalize()] = val
344
345 def has_header(self, header_name):
346 return (header_name in self.headers or
347 header_name in self.unredirected_hdrs)
348
349 def get_header(self, header_name, default=None):
350 return self.headers.get(
351 header_name,
352 self.unredirected_hdrs.get(header_name, default))
353
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200354 def remove_header(self, header_name):
355 self.headers.pop(header_name, None)
356 self.unredirected_hdrs.pop(header_name, None)
357
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000358 def header_items(self):
359 hdrs = self.unredirected_hdrs.copy()
360 hdrs.update(self.headers)
361 return list(hdrs.items())
362
363class OpenerDirector:
364 def __init__(self):
365 client_version = "Python-urllib/%s" % __version__
366 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000367 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000368 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000369 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000370 self.handle_open = {}
371 self.handle_error = {}
372 self.process_response = {}
373 self.process_request = {}
374
375 def add_handler(self, handler):
376 if not hasattr(handler, "add_parent"):
377 raise TypeError("expected BaseHandler instance, got %r" %
378 type(handler))
379
380 added = False
381 for meth in dir(handler):
382 if meth in ["redirect_request", "do_open", "proxy_open"]:
383 # oops, coincidental match
384 continue
385
386 i = meth.find("_")
387 protocol = meth[:i]
388 condition = meth[i+1:]
389
390 if condition.startswith("error"):
391 j = condition.find("_") + i + 1
392 kind = meth[j+1:]
393 try:
394 kind = int(kind)
395 except ValueError:
396 pass
397 lookup = self.handle_error.get(protocol, {})
398 self.handle_error[protocol] = lookup
399 elif condition == "open":
400 kind = protocol
401 lookup = self.handle_open
402 elif condition == "response":
403 kind = protocol
404 lookup = self.process_response
405 elif condition == "request":
406 kind = protocol
407 lookup = self.process_request
408 else:
409 continue
410
411 handlers = lookup.setdefault(kind, [])
412 if handlers:
413 bisect.insort(handlers, handler)
414 else:
415 handlers.append(handler)
416 added = True
417
418 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000419 bisect.insort(self.handlers, handler)
420 handler.add_parent(self)
421
422 def close(self):
423 # Only exists for backwards compatibility.
424 pass
425
426 def _call_chain(self, chain, kind, meth_name, *args):
427 # Handlers raise an exception if no one else should try to handle
428 # the request, or return None if they can't but another handler
429 # could. Otherwise, they return the response.
430 handlers = chain.get(kind, ())
431 for handler in handlers:
432 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000433 result = func(*args)
434 if result is not None:
435 return result
436
437 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
438 # accept a URL or a Request object
439 if isinstance(fullurl, str):
440 req = Request(fullurl, data)
441 else:
442 req = fullurl
443 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000444 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000445
446 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000447 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000448
449 # pre-process request
450 meth_name = protocol+"_request"
451 for processor in self.process_request.get(protocol, []):
452 meth = getattr(processor, meth_name)
453 req = meth(req)
454
455 response = self._open(req, data)
456
457 # post-process response
458 meth_name = protocol+"_response"
459 for processor in self.process_response.get(protocol, []):
460 meth = getattr(processor, meth_name)
461 response = meth(req, response)
462
463 return response
464
465 def _open(self, req, data=None):
466 result = self._call_chain(self.handle_open, 'default',
467 'default_open', req)
468 if result:
469 return result
470
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000471 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000472 result = self._call_chain(self.handle_open, protocol, protocol +
473 '_open', req)
474 if result:
475 return result
476
477 return self._call_chain(self.handle_open, 'unknown',
478 'unknown_open', req)
479
480 def error(self, proto, *args):
481 if proto in ('http', 'https'):
482 # XXX http[s] protocols are special-cased
483 dict = self.handle_error['http'] # https is not different than http
484 proto = args[2] # YUCK!
485 meth_name = 'http_error_%s' % proto
486 http_err = 1
487 orig_args = args
488 else:
489 dict = self.handle_error
490 meth_name = proto + '_error'
491 http_err = 0
492 args = (dict, proto, meth_name) + args
493 result = self._call_chain(*args)
494 if result:
495 return result
496
497 if http_err:
498 args = (dict, 'default', 'http_error_default') + orig_args
499 return self._call_chain(*args)
500
501# XXX probably also want an abstract factory that knows when it makes
502# sense to skip a superclass in favor of a subclass and when it might
503# make sense to include both
504
505def build_opener(*handlers):
506 """Create an opener object from a list of handlers.
507
508 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000509 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000510
511 If any of the handlers passed as arguments are subclasses of the
512 default handlers, the default handlers will not be used.
513 """
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000514 opener = OpenerDirector()
515 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
516 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100517 FTPHandler, FileHandler, HTTPErrorProcessor,
518 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000519 if hasattr(http.client, "HTTPSConnection"):
520 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000521 skip = set()
522 for klass in default_classes:
523 for check in handlers:
Benjamin Peterson78c85382014-04-01 16:27:30 -0400524 if isinstance(check, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000525 if issubclass(check, klass):
526 skip.add(klass)
527 elif isinstance(check, klass):
528 skip.add(klass)
529 for klass in skip:
530 default_classes.remove(klass)
531
532 for klass in default_classes:
533 opener.add_handler(klass())
534
535 for h in handlers:
Benjamin Peterson5dd3cae2014-04-01 14:20:56 -0400536 if isinstance(h, type):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000537 h = h()
538 opener.add_handler(h)
539 return opener
540
541class BaseHandler:
542 handler_order = 500
543
544 def add_parent(self, parent):
545 self.parent = parent
546
547 def close(self):
548 # Only exists for backwards compatibility
549 pass
550
551 def __lt__(self, other):
552 if not hasattr(other, "handler_order"):
553 # Try to preserve the old behavior of having custom classes
554 # inserted after default ones (works only for custom user
555 # classes which are not aware of handler_order).
556 return True
557 return self.handler_order < other.handler_order
558
559
560class HTTPErrorProcessor(BaseHandler):
561 """Process HTTP error responses."""
562 handler_order = 1000 # after all other processing
563
564 def http_response(self, request, response):
565 code, msg, hdrs = response.code, response.msg, response.info()
566
567 # According to RFC 2616, "2xx" code indicates that the client's
568 # request was successfully received, understood, and accepted.
569 if not (200 <= code < 300):
570 response = self.parent.error(
571 'http', request, response, code, msg, hdrs)
572
573 return response
574
575 https_response = http_response
576
577class HTTPDefaultErrorHandler(BaseHandler):
578 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000579 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000580
581class HTTPRedirectHandler(BaseHandler):
582 # maximum number of redirections to any single URL
583 # this is needed because of the state that cookies introduce
584 max_repeats = 4
585 # maximum total number of redirections (regardless of URL) before
586 # assuming we're in a loop
587 max_redirections = 10
588
589 def redirect_request(self, req, fp, code, msg, headers, newurl):
590 """Return a Request or None in response to a redirect.
591
592 This is called by the http_error_30x methods when a
593 redirection response is received. If a redirection should
594 take place, return a new Request to allow http_error_30x to
595 perform the redirect. Otherwise, raise HTTPError if no-one
596 else should try to handle this url. Return None if you can't
597 but another Handler might.
598 """
599 m = req.get_method()
600 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
601 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000602 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000603
604 # Strictly (according to RFC 2616), 301 or 302 in response to
605 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000606 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000607 # essentially all clients do redirect in this case, so we do
608 # the same.
609 # be conciliant with URIs containing a space
610 newurl = newurl.replace(' ', '%20')
611 CONTENT_HEADERS = ("content-length", "content-type")
612 newheaders = dict((k, v) for k, v in req.headers.items()
613 if k.lower() not in CONTENT_HEADERS)
614 return Request(newurl,
615 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000616 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000617 unverifiable=True)
618
619 # Implementation note: To avoid the server sending us into an
620 # infinite loop, the request object needs to track what URLs we
621 # have already seen. Do this by adding a handler-specific
622 # attribute to the Request object.
623 def http_error_302(self, req, fp, code, msg, headers):
624 # Some servers (incorrectly) return multiple Location headers
625 # (so probably same goes for URI). Use first header.
626 if "location" in headers:
627 newurl = headers["location"]
628 elif "uri" in headers:
629 newurl = headers["uri"]
630 else:
631 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000632
633 # fix a possible malformed URL
634 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700635
636 # For security reasons we don't allow redirection to anything other
637 # than http, https or ftp.
638
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800639 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800640 raise HTTPError(
641 newurl, code,
642 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
643 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700644
Facundo Batistaf24802c2008-08-17 03:36:03 +0000645 if not urlparts.path:
646 urlparts = list(urlparts)
647 urlparts[2] = "/"
648 newurl = urlunparse(urlparts)
649
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000650 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000651
652 # XXX Probably want to forget about the state of the current
653 # request, although that might interact poorly with other
654 # handlers that also use handler-specific request attributes
655 new = self.redirect_request(req, fp, code, msg, headers, newurl)
656 if new is None:
657 return
658
659 # loop detection
660 # .redirect_dict has a key url if url was previously visited.
661 if hasattr(req, 'redirect_dict'):
662 visited = new.redirect_dict = req.redirect_dict
663 if (visited.get(newurl, 0) >= self.max_repeats or
664 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000665 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000666 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000667 else:
668 visited = new.redirect_dict = req.redirect_dict = {}
669 visited[newurl] = visited.get(newurl, 0) + 1
670
671 # Don't close the fp until we are sure that we won't use it
672 # with HTTPError.
673 fp.read()
674 fp.close()
675
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000676 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000677
678 http_error_301 = http_error_303 = http_error_307 = http_error_302
679
680 inf_msg = "The HTTP server returned a redirect error that would " \
681 "lead to an infinite loop.\n" \
682 "The last 30x error message was:\n"
683
684
685def _parse_proxy(proxy):
686 """Return (scheme, user, password, host/port) given a URL or an authority.
687
688 If a URL is supplied, it must have an authority (host:port) component.
689 According to RFC 3986, having an authority component means the URL must
Senthil Kumarand8e24f12014-04-14 16:32:20 -0400690 have two slashes after the scheme.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000691 """
Georg Brandl13e89462008-07-01 19:56:00 +0000692 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000693 if not r_scheme.startswith("/"):
694 # authority
695 scheme = None
696 authority = proxy
697 else:
698 # URL
699 if not r_scheme.startswith("//"):
700 raise ValueError("proxy URL with no authority: %r" % proxy)
701 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
702 # and 3.3.), path is empty or starts with '/'
703 end = r_scheme.find("/", 2)
704 if end == -1:
705 end = None
706 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000707 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000708 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000709 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000710 else:
711 user = password = None
712 return scheme, user, password, hostport
713
714class ProxyHandler(BaseHandler):
715 # Proxies must be in front
716 handler_order = 100
717
718 def __init__(self, proxies=None):
719 if proxies is None:
720 proxies = getproxies()
721 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
722 self.proxies = proxies
723 for type, url in proxies.items():
724 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200725 lambda r, proxy=url, type=type, meth=self.proxy_open:
726 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000727
728 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000729 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000730 proxy_type, user, password, hostport = _parse_proxy(proxy)
731 if proxy_type is None:
732 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000733
734 if req.host and proxy_bypass(req.host):
735 return None
736
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000737 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000738 user_pass = '%s:%s' % (unquote(user),
739 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000740 creds = base64.b64encode(user_pass.encode()).decode("ascii")
741 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000742 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000743 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000744 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000745 # let other handlers take care of it
746 return None
747 else:
748 # need to start over, because the other handlers don't
749 # grok the proxy's URL type
750 # e.g. if we have a constructor arg proxies like so:
751 # {'http': 'ftp://proxy.example.com'}, we may end up turning
752 # a request for http://acme.example.com/a into one for
753 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000754 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000755
756class HTTPPasswordMgr:
757
758 def __init__(self):
759 self.passwd = {}
760
761 def add_password(self, realm, uri, user, passwd):
762 # uri could be a single URI or a sequence
763 if isinstance(uri, str):
764 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800765 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000766 self.passwd[realm] = {}
767 for default_port in True, False:
768 reduced_uri = tuple(
769 [self.reduce_uri(u, default_port) for u in uri])
770 self.passwd[realm][reduced_uri] = (user, passwd)
771
772 def find_user_password(self, realm, authuri):
773 domains = self.passwd.get(realm, {})
774 for default_port in True, False:
775 reduced_authuri = self.reduce_uri(authuri, default_port)
776 for uris, authinfo in domains.items():
777 for uri in uris:
778 if self.is_suburi(uri, reduced_authuri):
779 return authinfo
780 return None, None
781
782 def reduce_uri(self, uri, default_port=True):
783 """Accept authority or URI and extract only the authority and path."""
784 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000785 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000786 if parts[1]:
787 # URI
788 scheme = parts[0]
789 authority = parts[1]
790 path = parts[2] or '/'
791 else:
792 # host or host:port
793 scheme = None
794 authority = uri
795 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000796 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000797 if default_port and port is None and scheme is not None:
798 dport = {"http": 80,
799 "https": 443,
800 }.get(scheme)
801 if dport is not None:
802 authority = "%s:%d" % (host, dport)
803 return authority, path
804
805 def is_suburi(self, base, test):
806 """Check if test is below base in a URI tree
807
808 Both args must be URIs in reduced form.
809 """
810 if base == test:
811 return True
812 if base[0] != test[0]:
813 return False
814 common = posixpath.commonprefix((base[1], test[1]))
815 if len(common) == len(base[1]):
816 return True
817 return False
818
819
820class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
821
822 def find_user_password(self, realm, authuri):
823 user, password = HTTPPasswordMgr.find_user_password(self, realm,
824 authuri)
825 if user is not None:
826 return user, password
827 return HTTPPasswordMgr.find_user_password(self, None, authuri)
828
829
830class AbstractBasicAuthHandler:
831
832 # XXX this allows for multiple auth-schemes, but will stupidly pick
833 # the last one with a realm specified.
834
835 # allow for double- and single-quoted realm values
836 # (single quotes are a violation of the RFC, but appear in the wild)
837 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800838 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000839
840 # XXX could pre-emptively send auth info already accepted (RFC 2617,
841 # end of section 2, and section 1.2 immediately after "credentials"
842 # production).
843
844 def __init__(self, password_mgr=None):
845 if password_mgr is None:
846 password_mgr = HTTPPasswordMgr()
847 self.passwd = password_mgr
848 self.add_password = self.passwd.add_password
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000849
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000850 def http_error_auth_reqed(self, authreq, host, req, headers):
851 # host may be an authority (without userinfo) or a URL with an
852 # authority
853 # XXX could be multiple headers
854 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000855
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000856 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800857 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800858 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800859 raise ValueError("AbstractBasicAuthHandler does not"
860 " support the following scheme: '%s'" %
861 scheme)
862 else:
863 mo = AbstractBasicAuthHandler.rx.search(authreq)
864 if mo:
865 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800866 if quote not in ['"',"'"]:
867 warnings.warn("Basic Auth Realm was unquoted",
868 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800869 if scheme.lower() == 'basic':
Senthil Kumaran78373762014-08-20 07:53:58 +0530870 return self.retry_http_basic_auth(host, req, realm)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000871
872 def retry_http_basic_auth(self, host, req, realm):
873 user, pw = self.passwd.find_user_password(realm, host)
874 if pw is not None:
875 raw = "%s:%s" % (user, pw)
876 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
Senthil Kumaran78373762014-08-20 07:53:58 +0530877 if req.get_header(self.auth_header, None) == auth:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000878 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000879 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000880 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000881 else:
882 return None
883
884
885class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
886
887 auth_header = 'Authorization'
888
889 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000890 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000891 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000892 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000893 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000894
895
896class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
897
898 auth_header = 'Proxy-authorization'
899
900 def http_error_407(self, req, fp, code, msg, headers):
901 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000902 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000903 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
904 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000905 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000906 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000907 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000908 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000909
910
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800911# Return n random bytes.
912_randombytes = os.urandom
913
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000914
915class AbstractDigestAuthHandler:
916 # Digest authentication is specified in RFC 2617.
917
918 # XXX The client does not inspect the Authentication-Info header
919 # in a successful response.
920
921 # XXX It should be possible to test this implementation against
922 # a mock server that just generates a static set of challenges.
923
924 # XXX qop="auth-int" supports is shaky
925
926 def __init__(self, passwd=None):
927 if passwd is None:
928 passwd = HTTPPasswordMgr()
929 self.passwd = passwd
930 self.add_password = self.passwd.add_password
931 self.retried = 0
932 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000933 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000934
935 def reset_retry_count(self):
936 self.retried = 0
937
938 def http_error_auth_reqed(self, auth_header, host, req, headers):
939 authreq = headers.get(auth_header, None)
940 if self.retried > 5:
941 # Don't fail endlessly - if we failed once, we'll probably
942 # fail a second time. Hm. Unless the Password Manager is
943 # prompting for the information. Crap. This isn't great
944 # but it's better than the current 'repeat until recursion
945 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000946 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000947 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000948 else:
949 self.retried += 1
950 if authreq:
951 scheme = authreq.split()[0]
952 if scheme.lower() == 'digest':
953 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800954 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800955 raise ValueError("AbstractDigestAuthHandler does not support"
956 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000957
958 def retry_http_digest_auth(self, req, auth):
959 token, challenge = auth.split(' ', 1)
960 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
961 auth = self.get_authorization(req, chal)
962 if auth:
963 auth_val = 'Digest %s' % auth
964 if req.headers.get(self.auth_header, None) == auth_val:
965 return None
966 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000967 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000968 return resp
969
970 def get_cnonce(self, nonce):
971 # The cnonce-value is an opaque
972 # quoted string value provided by the client and used by both client
973 # and server to avoid chosen plaintext attacks, to provide mutual
974 # authentication, and to provide some message integrity protection.
975 # This isn't a fabulous effort, but it's probably Good Enough.
976 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800977 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000978 dig = hashlib.sha1(b).hexdigest()
979 return dig[:16]
980
981 def get_authorization(self, req, chal):
982 try:
983 realm = chal['realm']
984 nonce = chal['nonce']
985 qop = chal.get('qop')
986 algorithm = chal.get('algorithm', 'MD5')
987 # mod_digest doesn't send an opaque, even though it isn't
988 # supposed to be optional
989 opaque = chal.get('opaque', None)
990 except KeyError:
991 return None
992
993 H, KD = self.get_algorithm_impls(algorithm)
994 if H is None:
995 return None
996
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000997 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000998 if user is None:
999 return None
1000
1001 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001002 if req.data is not None:
1003 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001004 else:
1005 entdig = None
1006
1007 A1 = "%s:%s:%s" % (user, realm, pw)
1008 A2 = "%s:%s" % (req.get_method(),
1009 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001010 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001011 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001012 if nonce == self.last_nonce:
1013 self.nonce_count += 1
1014 else:
1015 self.nonce_count = 1
1016 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001017 ncvalue = '%08x' % self.nonce_count
1018 cnonce = self.get_cnonce(nonce)
1019 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1020 respdig = KD(H(A1), noncebit)
1021 elif qop is None:
1022 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1023 else:
1024 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001025 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001026
1027 # XXX should the partial digests be encoded too?
1028
1029 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001030 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001031 respdig)
1032 if opaque:
1033 base += ', opaque="%s"' % opaque
1034 if entdig:
1035 base += ', digest="%s"' % entdig
1036 base += ', algorithm="%s"' % algorithm
1037 if qop:
1038 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1039 return base
1040
1041 def get_algorithm_impls(self, algorithm):
1042 # lambdas assume digest modules are imported at the top level
1043 if algorithm == 'MD5':
1044 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1045 elif algorithm == 'SHA':
1046 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1047 # XXX MD5-sess
1048 KD = lambda s, d: H("%s:%s" % (s, d))
1049 return H, KD
1050
1051 def get_entity_digest(self, data, chal):
1052 # XXX not implemented yet
1053 return None
1054
1055
1056class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1057 """An authentication protocol defined by RFC 2069
1058
1059 Digest authentication improves on basic authentication because it
1060 does not transmit passwords in the clear.
1061 """
1062
1063 auth_header = 'Authorization'
1064 handler_order = 490 # before Basic auth
1065
1066 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001067 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001068 retry = self.http_error_auth_reqed('www-authenticate',
1069 host, req, headers)
1070 self.reset_retry_count()
1071 return retry
1072
1073
1074class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1075
1076 auth_header = 'Proxy-Authorization'
1077 handler_order = 490 # before Basic auth
1078
1079 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001080 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001081 retry = self.http_error_auth_reqed('proxy-authenticate',
1082 host, req, headers)
1083 self.reset_retry_count()
1084 return retry
1085
1086class AbstractHTTPHandler(BaseHandler):
1087
1088 def __init__(self, debuglevel=0):
1089 self._debuglevel = debuglevel
1090
1091 def set_http_debuglevel(self, level):
1092 self._debuglevel = level
1093
1094 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001095 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001096 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001097 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001098
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001099 if request.data is not None: # POST
1100 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001101 if isinstance(data, str):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001102 msg = "POST data should be bytes or an iterable of bytes. " \
1103 "It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001104 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001105 if not request.has_header('Content-type'):
1106 request.add_unredirected_header(
1107 'Content-type',
1108 'application/x-www-form-urlencoded')
1109 if not request.has_header('Content-length'):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001110 try:
1111 mv = memoryview(data)
1112 except TypeError:
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001113 if isinstance(data, collections.Iterable):
Georg Brandl61536042011-02-03 07:46:41 +00001114 raise ValueError("Content-Length should be specified "
1115 "for iterable data of type %r %r" % (type(data),
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001116 data))
1117 else:
1118 request.add_unredirected_header(
Senthil Kumaran1e991f22010-12-24 04:03:59 +00001119 'Content-length', '%d' % (len(mv) * mv.itemsize))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001120
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001121 sel_host = host
1122 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001123 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001124 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001125 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001126 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001127 for name, value in self.parent.addheaders:
1128 name = name.capitalize()
1129 if not request.has_header(name):
1130 request.add_unredirected_header(name, value)
1131
1132 return request
1133
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001134 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001135 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001136
1137 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001138 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001139 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001140 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001141 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001142
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001143 # will parse host:port
1144 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001145
1146 headers = dict(req.unredirected_hdrs)
1147 headers.update(dict((k, v) for k, v in req.headers.items()
1148 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001149
1150 # TODO(jhylton): Should this be redesigned to handle
1151 # persistent connections?
1152
1153 # We want to make an HTTP/1.1 request, but the addinfourl
1154 # class isn't prepared to deal with a persistent connection.
1155 # It will try to read all remaining data from the socket,
1156 # which will block while the server waits for the next request.
1157 # So make sure the connection gets closed after the (only)
1158 # request.
1159 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001160 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001161
1162 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001163 tunnel_headers = {}
1164 proxy_auth_hdr = "Proxy-Authorization"
1165 if proxy_auth_hdr in headers:
1166 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1167 # Proxy-Authorization should not be sent to origin
1168 # server.
1169 del headers[proxy_auth_hdr]
1170 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001171
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001172 try:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001173 h.request(req.get_method(), req.selector, req.data, headers)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001174 except OSError as err: # timeout error
Senthil Kumaran45686b42011-07-27 09:31:03 +08001175 h.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001176 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001177 else:
1178 r = h.getresponse()
Nadeem Vawdabd26b542012-10-21 17:37:43 +02001179 # If the server does not send us a 'Connection: close' header,
1180 # HTTPConnection assumes the socket should be left open. Manually
1181 # mark the socket to be closed when this response object goes away.
1182 if h.sock:
1183 h.sock.close()
1184 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001185
Senthil Kumaran26430412011-04-13 07:01:19 +08001186 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001187 # This line replaces the .msg attribute of the HTTPResponse
1188 # with .headers, because urllib clients expect the response to
1189 # have the reason in .msg. It would be good to mark this
1190 # attribute is deprecated and get then to use info() or
1191 # .headers.
1192 r.msg = r.reason
1193 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001194
1195
1196class HTTPHandler(AbstractHTTPHandler):
1197
1198 def http_open(self, req):
1199 return self.do_open(http.client.HTTPConnection, req)
1200
1201 http_request = AbstractHTTPHandler.do_request_
1202
1203if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001204
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205 class HTTPSHandler(AbstractHTTPHandler):
1206
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001207 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1208 AbstractHTTPHandler.__init__(self, debuglevel)
1209 self._context = context
1210 self._check_hostname = check_hostname
1211
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001212 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001213 return self.do_open(http.client.HTTPSConnection, req,
1214 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001215
1216 https_request = AbstractHTTPHandler.do_request_
1217
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001218 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001219
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001220class HTTPCookieProcessor(BaseHandler):
1221 def __init__(self, cookiejar=None):
1222 import http.cookiejar
1223 if cookiejar is None:
1224 cookiejar = http.cookiejar.CookieJar()
1225 self.cookiejar = cookiejar
1226
1227 def http_request(self, request):
1228 self.cookiejar.add_cookie_header(request)
1229 return request
1230
1231 def http_response(self, request, response):
1232 self.cookiejar.extract_cookies(response, request)
1233 return response
1234
1235 https_request = http_request
1236 https_response = http_response
1237
1238class UnknownHandler(BaseHandler):
1239 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001240 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001241 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001242
1243def parse_keqv_list(l):
1244 """Parse list of key=value strings where keys are not duplicated."""
1245 parsed = {}
1246 for elt in l:
1247 k, v = elt.split('=', 1)
1248 if v[0] == '"' and v[-1] == '"':
1249 v = v[1:-1]
1250 parsed[k] = v
1251 return parsed
1252
1253def parse_http_list(s):
1254 """Parse lists as described by RFC 2068 Section 2.
1255
1256 In particular, parse comma-separated lists where the elements of
1257 the list may include quoted-strings. A quoted-string could
1258 contain a comma. A non-quoted string could have quotes in the
1259 middle. Neither commas nor quotes count if they are escaped.
1260 Only double-quotes count, not single-quotes.
1261 """
1262 res = []
1263 part = ''
1264
1265 escape = quote = False
1266 for cur in s:
1267 if escape:
1268 part += cur
1269 escape = False
1270 continue
1271 if quote:
1272 if cur == '\\':
1273 escape = True
1274 continue
1275 elif cur == '"':
1276 quote = False
1277 part += cur
1278 continue
1279
1280 if cur == ',':
1281 res.append(part)
1282 part = ''
1283 continue
1284
1285 if cur == '"':
1286 quote = True
1287
1288 part += cur
1289
1290 # append last part
1291 if part:
1292 res.append(part)
1293
1294 return [part.strip() for part in res]
1295
1296class FileHandler(BaseHandler):
1297 # Use local file or FTP depending on form of URL
1298 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001299 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001300 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1301 req.host != 'localhost'):
Senthil Kumaranbc07ac52014-07-22 00:15:20 -07001302 if not req.host in self.get_names():
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001303 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001304 else:
1305 return self.open_local_file(req)
1306
1307 # names for the localhost
1308 names = None
1309 def get_names(self):
1310 if FileHandler.names is None:
1311 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001312 FileHandler.names = tuple(
1313 socket.gethostbyname_ex('localhost')[2] +
1314 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001315 except socket.gaierror:
1316 FileHandler.names = (socket.gethostbyname('localhost'),)
1317 return FileHandler.names
1318
1319 # not entirely sure what the rules are here
1320 def open_local_file(self, req):
1321 import email.utils
1322 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001323 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001324 filename = req.selector
1325 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001326 try:
1327 stats = os.stat(localfile)
1328 size = stats.st_size
1329 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001330 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001331 headers = email.message_from_string(
1332 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1333 (mtype or 'text/plain', size, modified))
1334 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001335 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001336 if not host or \
1337 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001338 if host:
1339 origurl = 'file://' + host + filename
1340 else:
1341 origurl = 'file://' + filename
1342 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001343 except OSError as exp:
Georg Brandl029986a2008-06-23 11:44:14 +00001344 # users shouldn't expect OSErrors coming from urlopen()
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001345 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001346 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001347
1348def _safe_gethostbyname(host):
1349 try:
1350 return socket.gethostbyname(host)
1351 except socket.gaierror:
1352 return None
1353
1354class FTPHandler(BaseHandler):
1355 def ftp_open(self, req):
1356 import ftplib
1357 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001358 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001359 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001360 raise URLError('ftp error: no host given')
1361 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001362 if port is None:
1363 port = ftplib.FTP_PORT
1364 else:
1365 port = int(port)
1366
1367 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001368 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001369 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001370 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001371 else:
1372 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001373 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001374 user = user or ''
1375 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001376
1377 try:
1378 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001379 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001380 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001381 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001382 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001383 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001384 dirs, file = dirs[:-1], dirs[-1]
1385 if dirs and not dirs[0]:
1386 dirs = dirs[1:]
1387 try:
1388 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1389 type = file and 'I' or 'D'
1390 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001391 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001392 if attr.lower() == 'type' and \
1393 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1394 type = value.upper()
1395 fp, retrlen = fw.retrfile(file, type)
1396 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001397 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001398 if mtype:
1399 headers += "Content-type: %s\n" % mtype
1400 if retrlen is not None and retrlen >= 0:
1401 headers += "Content-length: %d\n" % retrlen
1402 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001403 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001404 except ftplib.all_errors as exp:
1405 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001406 raise exc.with_traceback(sys.exc_info()[2])
1407
1408 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001409 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1410 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001411
1412class CacheFTPHandler(FTPHandler):
1413 # XXX would be nice to have pluggable cache strategies
1414 # XXX this stuff is definitely not thread safe
1415 def __init__(self):
1416 self.cache = {}
1417 self.timeout = {}
1418 self.soonest = 0
1419 self.delay = 60
1420 self.max_conns = 16
1421
1422 def setTimeout(self, t):
1423 self.delay = t
1424
1425 def setMaxConns(self, m):
1426 self.max_conns = m
1427
1428 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1429 key = user, host, port, '/'.join(dirs), timeout
1430 if key in self.cache:
1431 self.timeout[key] = time.time() + self.delay
1432 else:
1433 self.cache[key] = ftpwrapper(user, passwd, host, port,
1434 dirs, timeout)
1435 self.timeout[key] = time.time() + self.delay
1436 self.check_cache()
1437 return self.cache[key]
1438
1439 def check_cache(self):
1440 # first check for old ones
1441 t = time.time()
1442 if self.soonest <= t:
1443 for k, v in list(self.timeout.items()):
1444 if v < t:
1445 self.cache[k].close()
1446 del self.cache[k]
1447 del self.timeout[k]
1448 self.soonest = min(list(self.timeout.values()))
1449
1450 # then check the size
1451 if len(self.cache) == self.max_conns:
1452 for k, v in list(self.timeout.items()):
1453 if v == self.soonest:
1454 del self.cache[k]
1455 del self.timeout[k]
1456 break
1457 self.soonest = min(list(self.timeout.values()))
1458
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001459 def clear_cache(self):
1460 for conn in self.cache.values():
1461 conn.close()
1462 self.cache.clear()
1463 self.timeout.clear()
1464
Antoine Pitroudf204be2012-11-24 17:59:08 +01001465class DataHandler(BaseHandler):
1466 def data_open(self, req):
1467 # data URLs as specified in RFC 2397.
1468 #
1469 # ignores POSTed data
1470 #
1471 # syntax:
1472 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1473 # mediatype := [ type "/" subtype ] *( ";" parameter )
1474 # data := *urlchar
1475 # parameter := attribute "=" value
1476 url = req.full_url
1477
1478 scheme, data = url.split(":",1)
1479 mediatype, data = data.split(",",1)
1480
1481 # even base64 encoded data URLs might be quoted so unquote in any case:
1482 data = unquote_to_bytes(data)
1483 if mediatype.endswith(";base64"):
1484 data = base64.decodebytes(data)
1485 mediatype = mediatype[:-7]
1486
1487 if not mediatype:
1488 mediatype = "text/plain;charset=US-ASCII"
1489
1490 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1491 (mediatype, len(data)))
1492
1493 return addinfourl(io.BytesIO(data), headers, url)
1494
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001495
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001496# Code move from the old urllib module
1497
1498MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1499
1500# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001501if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001502 from nturl2path import url2pathname, pathname2url
1503else:
1504 def url2pathname(pathname):
1505 """OS-specific conversion from a relative URL of the 'file' scheme
1506 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001507 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001508
1509 def pathname2url(pathname):
1510 """OS-specific conversion from a file system path to a relative URL
1511 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001512 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001513
1514# This really consists of two pieces:
1515# (1) a class which handles opening of all sorts of URLs
1516# (plus assorted utilities etc.)
1517# (2) a set of functions for parsing URLs
1518# XXX Should these be separated out into different modules?
1519
1520
1521ftpcache = {}
1522class URLopener:
1523 """Class to open URLs.
1524 This is a class rather than just a subroutine because we may need
1525 more than one set of global protocol-specific options.
1526 Note -- this is a base class for those who don't want the
1527 automatic handling of errors type 302 (relocated) and 401
1528 (authorization needed)."""
1529
1530 __tempfiles = None
1531
1532 version = "Python-urllib/%s" % __version__
1533
1534 # Constructor
1535 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001536 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001537 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1538 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001539 if proxies is None:
1540 proxies = getproxies()
1541 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1542 self.proxies = proxies
1543 self.key_file = x509.get('key_file')
1544 self.cert_file = x509.get('cert_file')
1545 self.addheaders = [('User-Agent', self.version)]
1546 self.__tempfiles = []
1547 self.__unlink = os.unlink # See cleanup()
1548 self.tempcache = None
1549 # Undocumented feature: if you assign {} to tempcache,
1550 # it is used to cache files retrieved with
1551 # self.retrieve(). This is not enabled by default
1552 # since it does not work for changing documents (and I
1553 # haven't got the logic to check expiration headers
1554 # yet).
1555 self.ftpcache = ftpcache
1556 # Undocumented feature: you can use a different
1557 # ftp cache by assigning to the .ftpcache member;
1558 # in case you want logically independent URL openers
1559 # XXX This is not threadsafe. Bah.
1560
1561 def __del__(self):
1562 self.close()
1563
1564 def close(self):
1565 self.cleanup()
1566
1567 def cleanup(self):
1568 # This code sometimes runs when the rest of this module
1569 # has already been deleted, so it can't use any globals
1570 # or import anything.
1571 if self.__tempfiles:
1572 for file in self.__tempfiles:
1573 try:
1574 self.__unlink(file)
1575 except OSError:
1576 pass
1577 del self.__tempfiles[:]
1578 if self.tempcache:
1579 self.tempcache.clear()
1580
1581 def addheader(self, *args):
1582 """Add a header to be used by the HTTP interface only
1583 e.g. u.addheader('Accept', 'sound/basic')"""
1584 self.addheaders.append(args)
1585
1586 # External interface
1587 def open(self, fullurl, data=None):
1588 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001589 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001590 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001591 if self.tempcache and fullurl in self.tempcache:
1592 filename, headers = self.tempcache[fullurl]
1593 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001594 return addinfourl(fp, headers, fullurl)
1595 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001596 if not urltype:
1597 urltype = 'file'
1598 if urltype in self.proxies:
1599 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001600 urltype, proxyhost = splittype(proxy)
1601 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001602 url = (host, fullurl) # Signal special case to open_*()
1603 else:
1604 proxy = None
1605 name = 'open_' + urltype
1606 self.type = urltype
1607 name = name.replace('-', '_')
1608 if not hasattr(self, name):
1609 if proxy:
1610 return self.open_unknown_proxy(proxy, fullurl, data)
1611 else:
1612 return self.open_unknown(fullurl, data)
1613 try:
1614 if data is None:
1615 return getattr(self, name)(url)
1616 else:
1617 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001618 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001619 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001620 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001621 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001622
1623 def open_unknown(self, fullurl, data=None):
1624 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001625 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001626 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001627
1628 def open_unknown_proxy(self, proxy, fullurl, data=None):
1629 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001630 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001631 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001632
1633 # External interface
1634 def retrieve(self, url, filename=None, reporthook=None, data=None):
1635 """retrieve(url) returns (filename, headers) for a local object
1636 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001637 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001638 if self.tempcache and url in self.tempcache:
1639 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001640 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001641 if filename is None and (not type or type == 'file'):
1642 try:
1643 fp = self.open_local_file(url1)
1644 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001645 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001646 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001647 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001648 pass
1649 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001650 try:
1651 headers = fp.info()
1652 if filename:
1653 tfp = open(filename, 'wb')
1654 else:
1655 import tempfile
1656 garbage, path = splittype(url)
1657 garbage, path = splithost(path or "")
1658 path, garbage = splitquery(path or "")
1659 path, garbage = splitattr(path or "")
1660 suffix = os.path.splitext(path)[1]
1661 (fd, filename) = tempfile.mkstemp(suffix)
1662 self.__tempfiles.append(filename)
1663 tfp = os.fdopen(fd, 'wb')
1664 try:
1665 result = filename, headers
1666 if self.tempcache is not None:
1667 self.tempcache[url] = result
1668 bs = 1024*8
1669 size = -1
1670 read = 0
1671 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001672 if "content-length" in headers:
1673 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001674 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001675 reporthook(blocknum, bs, size)
1676 while 1:
1677 block = fp.read(bs)
1678 if not block:
1679 break
1680 read += len(block)
1681 tfp.write(block)
1682 blocknum += 1
1683 if reporthook:
1684 reporthook(blocknum, bs, size)
1685 finally:
1686 tfp.close()
1687 finally:
1688 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001689
1690 # raise exception if actual size does not match content-length header
1691 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001692 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001693 "retrieval incomplete: got only %i out of %i bytes"
1694 % (read, size), result)
1695
1696 return result
1697
1698 # Each method named open_<type> knows how to open that type of URL
1699
1700 def _open_generic_http(self, connection_factory, url, data):
1701 """Make an HTTP connection using connection_class.
1702
1703 This is an internal method that should be called from
1704 open_http() or open_https().
1705
1706 Arguments:
1707 - connection_factory should take a host name and return an
1708 HTTPConnection instance.
1709 - url is the url to retrieval or a host, relative-path pair.
1710 - data is payload for a POST request or None.
1711 """
1712
1713 user_passwd = None
1714 proxy_passwd= None
1715 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001716 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001717 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001718 user_passwd, host = splituser(host)
1719 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001720 realhost = host
1721 else:
1722 host, selector = url
1723 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001724 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001725 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001726 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001727 url = rest
1728 user_passwd = None
1729 if urltype.lower() != 'http':
1730 realhost = None
1731 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001732 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001733 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001734 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001735 if user_passwd:
1736 selector = "%s://%s%s" % (urltype, realhost, rest)
1737 if proxy_bypass(realhost):
1738 host = realhost
1739
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001740 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001741
1742 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001743 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001744 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001745 else:
1746 proxy_auth = None
1747
1748 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001749 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001750 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001751 else:
1752 auth = None
1753 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001754 headers = {}
1755 if proxy_auth:
1756 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1757 if auth:
1758 headers["Authorization"] = "Basic %s" % auth
1759 if realhost:
1760 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001761
1762 # Add Connection:close as we don't support persistent connections yet.
1763 # This helps in closing the socket and avoiding ResourceWarning
1764
1765 headers["Connection"] = "close"
1766
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001767 for header, value in self.addheaders:
1768 headers[header] = value
1769
1770 if data is not None:
1771 headers["Content-Type"] = "application/x-www-form-urlencoded"
1772 http_conn.request("POST", selector, data, headers)
1773 else:
1774 http_conn.request("GET", selector, headers=headers)
1775
1776 try:
1777 response = http_conn.getresponse()
1778 except http.client.BadStatusLine:
1779 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001780 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001781
1782 # According to RFC 2616, "2xx" code indicates that the client's
1783 # request was successfully received, understood, and accepted.
1784 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001785 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001786 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001787 else:
1788 return self.http_error(
1789 url, response.fp,
1790 response.status, response.reason, response.msg, data)
1791
1792 def open_http(self, url, data=None):
1793 """Use HTTP protocol."""
1794 return self._open_generic_http(http.client.HTTPConnection, url, data)
1795
1796 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1797 """Handle http errors.
1798
1799 Derived class can override this, or provide specific handlers
1800 named http_error_DDD where DDD is the 3-digit error code."""
1801 # First check if there's a specific handler for this error
1802 name = 'http_error_%d' % errcode
1803 if hasattr(self, name):
1804 method = getattr(self, name)
1805 if data is None:
1806 result = method(url, fp, errcode, errmsg, headers)
1807 else:
1808 result = method(url, fp, errcode, errmsg, headers, data)
1809 if result: return result
1810 return self.http_error_default(url, fp, errcode, errmsg, headers)
1811
1812 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001813 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001814 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001815 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001816
1817 if _have_ssl:
1818 def _https_connection(self, host):
1819 return http.client.HTTPSConnection(host,
1820 key_file=self.key_file,
1821 cert_file=self.cert_file)
1822
1823 def open_https(self, url, data=None):
1824 """Use HTTPS protocol."""
1825 return self._open_generic_http(self._https_connection, url, data)
1826
1827 def open_file(self, url):
1828 """Use local file or FTP depending on form of URL."""
1829 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001830 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001831 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001832 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001833 else:
1834 return self.open_local_file(url)
1835
1836 def open_local_file(self, url):
1837 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001838 import email.utils
1839 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001840 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001841 localname = url2pathname(file)
1842 try:
1843 stats = os.stat(localname)
1844 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001845 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001846 size = stats.st_size
1847 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1848 mtype = mimetypes.guess_type(url)[0]
1849 headers = email.message_from_string(
1850 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1851 (mtype or 'text/plain', size, modified))
1852 if not host:
1853 urlfile = file
1854 if file[:1] == '/':
1855 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001856 return addinfourl(open(localname, 'rb'), headers, urlfile)
1857 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001858 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07001859 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001860 urlfile = file
1861 if file[:1] == '/':
1862 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08001863 elif file[:2] == './':
1864 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00001865 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001866 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001867
1868 def open_ftp(self, url):
1869 """Use FTP protocol."""
1870 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001871 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001872 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001873 host, path = splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001874 if not host: raise URLError('ftp error: no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001875 host, port = splitport(host)
1876 user, host = splituser(host)
1877 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001878 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001879 host = unquote(host)
1880 user = unquote(user or '')
1881 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001882 host = socket.gethostbyname(host)
1883 if not port:
1884 import ftplib
1885 port = ftplib.FTP_PORT
1886 else:
1887 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001888 path, attrs = splitattr(path)
1889 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001890 dirs = path.split('/')
1891 dirs, file = dirs[:-1], dirs[-1]
1892 if dirs and not dirs[0]: dirs = dirs[1:]
1893 if dirs and not dirs[0]: dirs[0] = '/'
1894 key = user, host, port, '/'.join(dirs)
1895 # XXX thread unsafe!
1896 if len(self.ftpcache) > MAXFTPCACHE:
1897 # Prune the cache, rather arbitrarily
Benjamin Peterson3c2dca62014-06-07 15:08:04 -07001898 for k in list(self.ftpcache):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001899 if k != key:
1900 v = self.ftpcache[k]
1901 del self.ftpcache[k]
1902 v.close()
1903 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001904 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001905 self.ftpcache[key] = \
1906 ftpwrapper(user, passwd, host, port, dirs)
1907 if not file: type = 'D'
1908 else: type = 'I'
1909 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001910 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001911 if attr.lower() == 'type' and \
1912 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1913 type = value.upper()
1914 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1915 mtype = mimetypes.guess_type("ftp:" + url)[0]
1916 headers = ""
1917 if mtype:
1918 headers += "Content-Type: %s\n" % mtype
1919 if retrlen is not None and retrlen >= 0:
1920 headers += "Content-Length: %d\n" % retrlen
1921 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001922 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001923 except ftperrors() as exp:
1924 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001925
1926 def open_data(self, url, data=None):
1927 """Use "data" URL."""
1928 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001929 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001930 # ignore POSTed data
1931 #
1932 # syntax of data URLs:
1933 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1934 # mediatype := [ type "/" subtype ] *( ";" parameter )
1935 # data := *urlchar
1936 # parameter := attribute "=" value
1937 try:
1938 [type, data] = url.split(',', 1)
1939 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001940 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001941 if not type:
1942 type = 'text/plain;charset=US-ASCII'
1943 semi = type.rfind(';')
1944 if semi >= 0 and '=' not in type[semi:]:
1945 encoding = type[semi+1:]
1946 type = type[:semi]
1947 else:
1948 encoding = ''
1949 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00001950 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001951 time.gmtime(time.time())))
1952 msg.append('Content-type: %s' % type)
1953 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00001954 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001955 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001956 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001957 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001958 msg.append('Content-Length: %d' % len(data))
1959 msg.append('')
1960 msg.append(data)
1961 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001962 headers = email.message_from_string(msg)
1963 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001964 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001965 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001966
1967
1968class FancyURLopener(URLopener):
1969 """Derived class with handlers for errors we can handle (perhaps)."""
1970
1971 def __init__(self, *args, **kwargs):
1972 URLopener.__init__(self, *args, **kwargs)
1973 self.auth_cache = {}
1974 self.tries = 0
1975 self.maxtries = 10
1976
1977 def http_error_default(self, url, fp, errcode, errmsg, headers):
1978 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001979 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001980
1981 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1982 """Error 302 -- relocated (temporarily)."""
1983 self.tries += 1
1984 if self.maxtries and self.tries >= self.maxtries:
1985 if hasattr(self, "http_error_500"):
1986 meth = self.http_error_500
1987 else:
1988 meth = self.http_error_default
1989 self.tries = 0
1990 return meth(url, fp, 500,
1991 "Internal Server Error: Redirect Recursion", headers)
1992 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
1993 data)
1994 self.tries = 0
1995 return result
1996
1997 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
1998 if 'location' in headers:
1999 newurl = headers['location']
2000 elif 'uri' in headers:
2001 newurl = headers['uri']
2002 else:
2003 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002004 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002005
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002006 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002007 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002008
2009 urlparts = urlparse(newurl)
2010
2011 # For security reasons, we don't allow redirection to anything other
2012 # than http, https and ftp.
2013
2014 # We are using newer HTTPError with older redirect_internal method
2015 # This older method will get deprecated in 3.3
2016
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002017 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002018 raise HTTPError(newurl, errcode,
2019 errmsg +
2020 " Redirection to url '%s' is not allowed." % newurl,
2021 headers, fp)
2022
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002023 return self.open(newurl)
2024
2025 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2026 """Error 301 -- also relocated (permanently)."""
2027 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2028
2029 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2030 """Error 303 -- also relocated (essentially identical to 302)."""
2031 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2032
2033 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2034 """Error 307 -- relocated, but turn POST into error."""
2035 if data is None:
2036 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2037 else:
2038 return self.http_error_default(url, fp, errcode, errmsg, headers)
2039
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002040 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2041 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002042 """Error 401 -- authentication required.
2043 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002044 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002045 URLopener.http_error_default(self, url, fp,
2046 errcode, errmsg, headers)
2047 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002048 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2049 if not match:
2050 URLopener.http_error_default(self, url, fp,
2051 errcode, errmsg, headers)
2052 scheme, realm = match.groups()
2053 if scheme.lower() != 'basic':
2054 URLopener.http_error_default(self, url, fp,
2055 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002056 if not retry:
2057 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2058 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002059 name = 'retry_' + self.type + '_basic_auth'
2060 if data is None:
2061 return getattr(self,name)(url, realm)
2062 else:
2063 return getattr(self,name)(url, realm, data)
2064
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002065 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2066 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002067 """Error 407 -- proxy authentication required.
2068 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002069 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002070 URLopener.http_error_default(self, url, fp,
2071 errcode, errmsg, headers)
2072 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002073 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2074 if not match:
2075 URLopener.http_error_default(self, url, fp,
2076 errcode, errmsg, headers)
2077 scheme, realm = match.groups()
2078 if scheme.lower() != 'basic':
2079 URLopener.http_error_default(self, url, fp,
2080 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002081 if not retry:
2082 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2083 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002084 name = 'retry_proxy_' + self.type + '_basic_auth'
2085 if data is None:
2086 return getattr(self,name)(url, realm)
2087 else:
2088 return getattr(self,name)(url, realm, data)
2089
2090 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002091 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002092 newurl = 'http://' + host + selector
2093 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002094 urltype, proxyhost = splittype(proxy)
2095 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002096 i = proxyhost.find('@') + 1
2097 proxyhost = proxyhost[i:]
2098 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2099 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002100 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002101 quote(passwd, safe=''), proxyhost)
2102 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2103 if data is None:
2104 return self.open(newurl)
2105 else:
2106 return self.open(newurl, data)
2107
2108 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002109 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002110 newurl = 'https://' + host + selector
2111 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002112 urltype, proxyhost = splittype(proxy)
2113 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002114 i = proxyhost.find('@') + 1
2115 proxyhost = proxyhost[i:]
2116 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2117 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002118 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002119 quote(passwd, safe=''), proxyhost)
2120 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2121 if data is None:
2122 return self.open(newurl)
2123 else:
2124 return self.open(newurl, data)
2125
2126 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002127 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002128 i = host.find('@') + 1
2129 host = host[i:]
2130 user, passwd = self.get_user_passwd(host, realm, i)
2131 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002132 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002133 quote(passwd, safe=''), host)
2134 newurl = 'http://' + host + selector
2135 if data is None:
2136 return self.open(newurl)
2137 else:
2138 return self.open(newurl, data)
2139
2140 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002141 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002142 i = host.find('@') + 1
2143 host = host[i:]
2144 user, passwd = self.get_user_passwd(host, realm, i)
2145 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002146 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002147 quote(passwd, safe=''), host)
2148 newurl = 'https://' + host + selector
2149 if data is None:
2150 return self.open(newurl)
2151 else:
2152 return self.open(newurl, data)
2153
Florent Xicluna757445b2010-05-17 17:24:07 +00002154 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002155 key = realm + '@' + host.lower()
2156 if key in self.auth_cache:
2157 if clear_cache:
2158 del self.auth_cache[key]
2159 else:
2160 return self.auth_cache[key]
2161 user, passwd = self.prompt_user_passwd(host, realm)
2162 if user or passwd: self.auth_cache[key] = (user, passwd)
2163 return user, passwd
2164
2165 def prompt_user_passwd(self, host, realm):
2166 """Override this in a GUI environment!"""
2167 import getpass
2168 try:
2169 user = input("Enter username for %s at %s: " % (realm, host))
2170 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2171 (user, realm, host))
2172 return user, passwd
2173 except KeyboardInterrupt:
2174 print()
2175 return None, None
2176
2177
2178# Utility functions
2179
2180_localhost = None
2181def localhost():
2182 """Return the IP address of the magic hostname 'localhost'."""
2183 global _localhost
2184 if _localhost is None:
2185 _localhost = socket.gethostbyname('localhost')
2186 return _localhost
2187
2188_thishost = None
2189def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002190 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002191 global _thishost
2192 if _thishost is None:
Senthil Kumarandcdadfe2013-06-01 11:12:17 -07002193 try:
2194 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
2195 except socket.gaierror:
2196 _thishost = tuple(socket.gethostbyname_ex('localhost')[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002197 return _thishost
2198
2199_ftperrors = None
2200def ftperrors():
2201 """Return the set of errors raised by the FTP class."""
2202 global _ftperrors
2203 if _ftperrors is None:
2204 import ftplib
2205 _ftperrors = ftplib.all_errors
2206 return _ftperrors
2207
2208_noheaders = None
2209def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002210 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002211 global _noheaders
2212 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002213 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002214 return _noheaders
2215
2216
2217# Utility classes
2218
2219class ftpwrapper:
2220 """Class used by open_ftp() for cache of open FTP connections."""
2221
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002222 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2223 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002224 self.user = user
2225 self.passwd = passwd
2226 self.host = host
2227 self.port = port
2228 self.dirs = dirs
2229 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002230 self.refcount = 0
2231 self.keepalive = persistent
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002232 self.init()
2233
2234 def init(self):
2235 import ftplib
2236 self.busy = 0
2237 self.ftp = ftplib.FTP()
2238 self.ftp.connect(self.host, self.port, self.timeout)
2239 self.ftp.login(self.user, self.passwd)
Senthil Kumarancaa00fe2013-06-02 11:59:47 -07002240 _target = '/'.join(self.dirs)
2241 self.ftp.cwd(_target)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002242
2243 def retrfile(self, file, type):
2244 import ftplib
2245 self.endtransfer()
2246 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2247 else: cmd = 'TYPE ' + type; isdir = 0
2248 try:
2249 self.ftp.voidcmd(cmd)
2250 except ftplib.all_errors:
2251 self.init()
2252 self.ftp.voidcmd(cmd)
2253 conn = None
2254 if file and not isdir:
2255 # Try to retrieve as a file
2256 try:
2257 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002258 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002259 except ftplib.error_perm as reason:
2260 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002261 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002262 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002263 if not conn:
2264 # Set transfer mode to ASCII!
2265 self.ftp.voidcmd('TYPE A')
2266 # Try a directory listing. Verify that directory exists.
2267 if file:
2268 pwd = self.ftp.pwd()
2269 try:
2270 try:
2271 self.ftp.cwd(file)
2272 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002273 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002274 finally:
2275 self.ftp.cwd(pwd)
2276 cmd = 'LIST ' + file
2277 else:
2278 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002279 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002280 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002281
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002282 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2283 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002284 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002285 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002286 return (ftpobj, retrlen)
2287
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002288 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002289 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002290
2291 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002292 self.keepalive = False
2293 if self.refcount <= 0:
2294 self.real_close()
2295
2296 def file_close(self):
2297 self.endtransfer()
2298 self.refcount -= 1
2299 if self.refcount <= 0 and not self.keepalive:
2300 self.real_close()
2301
2302 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002303 self.endtransfer()
2304 try:
2305 self.ftp.close()
2306 except ftperrors():
2307 pass
2308
2309# Proxy handling
2310def getproxies_environment():
2311 """Return a dictionary of scheme -> proxy server URL mappings.
2312
2313 Scan the environment for variables named <scheme>_proxy;
2314 this seems to be the standard convention. If you need a
2315 different way, you can pass a proxies dictionary to the
2316 [Fancy]URLopener constructor.
2317
2318 """
2319 proxies = {}
2320 for name, value in os.environ.items():
2321 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002322 if value and name[-6:] == '_proxy':
2323 proxies[name[:-6]] = value
2324 return proxies
2325
2326def proxy_bypass_environment(host):
2327 """Test if proxies should not be used for a particular host.
2328
2329 Checks the environment for a variable named no_proxy, which should
2330 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2331 """
2332 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2333 # '*' is special case for always bypass
2334 if no_proxy == '*':
2335 return 1
2336 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002337 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002338 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002339 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2340 for name in no_proxy_list:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002341 if name and (hostonly.endswith(name) or host.endswith(name)):
2342 return 1
2343 # otherwise, don't bypass
2344 return 0
2345
2346
Ronald Oussorene72e1612011-03-14 18:15:25 -04002347# This code tests an OSX specific data structure but is testable on all
2348# platforms
2349def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2350 """
2351 Return True iff this host shouldn't be accessed using a proxy
2352
2353 This function uses the MacOSX framework SystemConfiguration
2354 to fetch the proxy information.
2355
2356 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2357 { 'exclude_simple': bool,
2358 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2359 }
2360 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002361 from fnmatch import fnmatch
2362
2363 hostonly, port = splitport(host)
2364
2365 def ip2num(ipAddr):
2366 parts = ipAddr.split('.')
2367 parts = list(map(int, parts))
2368 if len(parts) != 4:
2369 parts = (parts + [0, 0, 0, 0])[:4]
2370 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2371
2372 # Check for simple host names:
2373 if '.' not in host:
2374 if proxy_settings['exclude_simple']:
2375 return True
2376
2377 hostIP = None
2378
2379 for value in proxy_settings.get('exceptions', ()):
2380 # Items in the list are strings like these: *.local, 169.254/16
2381 if not value: continue
2382
2383 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2384 if m is not None:
2385 if hostIP is None:
2386 try:
2387 hostIP = socket.gethostbyname(hostonly)
2388 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002389 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002390 continue
2391
2392 base = ip2num(m.group(1))
2393 mask = m.group(2)
2394 if mask is None:
2395 mask = 8 * (m.group(1).count('.') + 1)
2396 else:
2397 mask = int(mask[1:])
2398 mask = 32 - mask
2399
2400 if (hostIP >> mask) == (base >> mask):
2401 return True
2402
2403 elif fnmatch(host, value):
2404 return True
2405
2406 return False
2407
2408
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002409if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002410 from _scproxy import _get_proxy_settings, _get_proxies
2411
2412 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002413 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002414 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002415
2416 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002417 """Return a dictionary of scheme -> proxy server URL mappings.
2418
Ronald Oussoren84151202010-04-18 20:46:11 +00002419 This function uses the MacOSX framework SystemConfiguration
2420 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002421 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002422 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002423
Ronald Oussoren84151202010-04-18 20:46:11 +00002424
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002425
2426 def proxy_bypass(host):
2427 if getproxies_environment():
2428 return proxy_bypass_environment(host)
2429 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002430 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002431
2432 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002433 return getproxies_environment() or getproxies_macosx_sysconf()
2434
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002435
2436elif os.name == 'nt':
2437 def getproxies_registry():
2438 """Return a dictionary of scheme -> proxy server URL mappings.
2439
2440 Win32 uses the registry to store proxies.
2441
2442 """
2443 proxies = {}
2444 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002445 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002446 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002447 # Std module, so should be around - but you never know!
2448 return proxies
2449 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002450 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002451 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002452 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002453 'ProxyEnable')[0]
2454 if proxyEnable:
2455 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002456 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002457 'ProxyServer')[0])
2458 if '=' in proxyServer:
2459 # Per-protocol settings
2460 for p in proxyServer.split(';'):
2461 protocol, address = p.split('=', 1)
2462 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002463 if not re.match('^([^/:]+)://', address):
2464 address = '%s://%s' % (protocol, address)
2465 proxies[protocol] = address
2466 else:
2467 # Use one setting for all protocols
2468 if proxyServer[:5] == 'http:':
2469 proxies['http'] = proxyServer
2470 else:
2471 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002472 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002473 proxies['ftp'] = 'ftp://%s' % proxyServer
2474 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002475 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002476 # Either registry key not found etc, or the value in an
2477 # unexpected format.
2478 # proxies already set up to be empty so nothing to do
2479 pass
2480 return proxies
2481
2482 def getproxies():
2483 """Return a dictionary of scheme -> proxy server URL mappings.
2484
2485 Returns settings gathered from the environment, if specified,
2486 or the registry.
2487
2488 """
2489 return getproxies_environment() or getproxies_registry()
2490
2491 def proxy_bypass_registry(host):
2492 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002493 import winreg
Brett Cannoncd171c82013-07-04 17:43:24 -04002494 except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002495 # Std modules, so should be around - but you never know!
2496 return 0
2497 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002498 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002499 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002500 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002501 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002502 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002503 'ProxyOverride')[0])
2504 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002505 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002506 return 0
2507 if not proxyEnable or not proxyOverride:
2508 return 0
2509 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002510 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002511 host = [rawHost]
2512 try:
2513 addr = socket.gethostbyname(rawHost)
2514 if addr != rawHost:
2515 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002516 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002517 pass
2518 try:
2519 fqdn = socket.getfqdn(rawHost)
2520 if fqdn != rawHost:
2521 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002522 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002523 pass
2524 # make a check value list from the registry entry: replace the
2525 # '<local>' string by the localhost entry and the corresponding
2526 # canonical entry.
2527 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002528 # now check if we match one of the registry values.
2529 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002530 if test == '<local>':
2531 if '.' not in rawHost:
2532 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002533 test = test.replace(".", r"\.") # mask dots
2534 test = test.replace("*", r".*") # change glob sequence
2535 test = test.replace("?", r".") # change glob char
2536 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002537 if re.match(test, val, re.I):
2538 return 1
2539 return 0
2540
2541 def proxy_bypass(host):
2542 """Return a dictionary of scheme -> proxy server URL mappings.
2543
2544 Returns settings gathered from the environment, if specified,
2545 or the registry.
2546
2547 """
2548 if getproxies_environment():
2549 return proxy_bypass_environment(host)
2550 else:
2551 return proxy_bypass_registry(host)
2552
2553else:
2554 # By default use environment variables
2555 getproxies = getproxies_environment
2556 proxy_bypass = proxy_bypass_environment