blob: fdb1ec87b051e3dc7c03f1bdd339573fc71ffa57 [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
Senthil Kumaranc2958622010-11-22 04:48:26 +0000113except 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')
144 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
145 context.options |= ssl.OP_NO_SSLv2
Antoine Pitrou9a8d6932013-04-01 18:55:35 +0200146 context.verify_mode = ssl.CERT_REQUIRED
147 if cafile or capath:
148 context.load_verify_locations(cafile, capath)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000149 else:
Antoine Pitrou9a8d6932013-04-01 18:55:35 +0200150 context.set_default_verify_paths()
151 https_handler = HTTPSHandler(context=context, check_hostname=True)
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000152 opener = build_opener(https_handler)
153 elif _opener is None:
154 _opener = opener = build_opener()
155 else:
156 opener = _opener
157 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000158
159def install_opener(opener):
160 global _opener
161 _opener = opener
162
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700163_url_tempfiles = []
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000164def urlretrieve(url, filename=None, reporthook=None, data=None):
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700165 """
166 Retrieve a URL into a temporary location on disk.
167
168 Requires a URL argument. If a filename is passed, it is used as
169 the temporary file location. The reporthook argument should be
170 a callable that accepts a block number, a read size, and the
171 total file size of the URL target. The data argument should be
172 valid URL encoded data.
173
174 If a filename is passed and the URL points to a local resource,
175 the result is a copy from local file to new file.
176
177 Returns a tuple containing the path to the newly created
178 data file as well as the resulting HTTPMessage object.
179 """
180 url_type, path = splittype(url)
181
182 with contextlib.closing(urlopen(url, data)) as fp:
183 headers = fp.info()
184
185 # Just return the local path and the "headers" for file://
186 # URLs. No sense in performing a copy unless requested.
187 if url_type == "file" and not filename:
188 return os.path.normpath(path), headers
189
190 # Handle temporary file setup.
191 if filename:
192 tfp = open(filename, 'wb')
193 else:
194 tfp = tempfile.NamedTemporaryFile(delete=False)
195 filename = tfp.name
196 _url_tempfiles.append(filename)
197
198 with tfp:
199 result = filename, headers
200 bs = 1024*8
201 size = -1
202 read = 0
203 blocknum = 0
204 if "content-length" in headers:
205 size = int(headers["Content-Length"])
206
207 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800208 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700209
210 while True:
211 block = fp.read(bs)
212 if not block:
213 break
214 read += len(block)
215 tfp.write(block)
216 blocknum += 1
217 if reporthook:
Gregory P. Smith6b0bdab2012-11-10 13:43:44 -0800218 reporthook(blocknum, bs, size)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700219
220 if size >= 0 and read < size:
221 raise ContentTooShortError(
222 "retrieval incomplete: got only %i out of %i bytes"
223 % (read, size), result)
224
225 return result
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000226
227def urlcleanup():
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700228 for temp_file in _url_tempfiles:
229 try:
230 os.unlink(temp_file)
Andrew Svetlov3438fa42012-12-17 23:35:18 +0200231 except OSError:
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700232 pass
233
234 del _url_tempfiles[:]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000235 global _opener
236 if _opener:
237 _opener = None
238
239# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000240_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000241def request_host(request):
242 """Return request-host, as defined by RFC 2965.
243
244 Variation from RFC: returned value is lowercased, for convenient
245 comparison.
246
247 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000248 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000249 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000250 if host == "":
251 host = request.get_header("Host", "")
252
253 # remove port, if present
254 host = _cut_port_re.sub("", host, 1)
255 return host.lower()
256
257class Request:
258
259 def __init__(self, url, data=None, headers={},
Senthil Kumarande49d642011-10-16 23:54:44 +0800260 origin_req_host=None, unverifiable=False,
261 method=None):
Senthil Kumaran52380922013-04-25 05:45:48 -0700262 self.full_url = url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000263 self.headers = {}
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200264 self.unredirected_hdrs = {}
265 self._data = None
266 self.data = data
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000267 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000268 for key, value in headers.items():
269 self.add_header(key, value)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000270 if origin_req_host is None:
271 origin_req_host = request_host(self)
272 self.origin_req_host = origin_req_host
273 self.unverifiable = unverifiable
Senthil Kumarande49d642011-10-16 23:54:44 +0800274 self.method = method
Senthil Kumaran52380922013-04-25 05:45:48 -0700275
276 @property
277 def full_url(self):
Senthil Kumaran83070752013-05-24 09:14:12 -0700278 if self.fragment:
279 return '{}#{}'.format(self._full_url, self.fragment)
Senthil Kumaran52380922013-04-25 05:45:48 -0700280 return self._full_url
281
282 @full_url.setter
283 def full_url(self, url):
284 # unwrap('<URL:type://host/path>') --> 'type://host/path'
285 self._full_url = unwrap(url)
286 self._full_url, self.fragment = splittag(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000287 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000288
Senthil Kumaran52380922013-04-25 05:45:48 -0700289 @full_url.deleter
290 def full_url(self):
291 self._full_url = None
292 self.fragment = None
293 self.selector = ''
294
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200295 @property
296 def data(self):
297 return self._data
298
299 @data.setter
300 def data(self, data):
301 if data != self._data:
302 self._data = data
303 # issue 16464
304 # if we change data we need to remove content-length header
305 # (cause it's most probably calculated for previous value)
306 if self.has_header("Content-length"):
307 self.remove_header("Content-length")
308
309 @data.deleter
310 def data(self):
R David Murray9cc7d452013-03-20 00:10:51 -0400311 self.data = None
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200312
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000313 def _parse(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700314 self.type, rest = splittype(self._full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000315 if self.type is None:
R David Murrayd8a46962013-04-03 06:58:34 -0400316 raise ValueError("unknown url type: %r" % self.full_url)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000317 self.host, self.selector = splithost(rest)
318 if self.host:
319 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000320
321 def get_method(self):
Senthil Kumarande49d642011-10-16 23:54:44 +0800322 """Return a string indicating the HTTP request method."""
323 if self.method is not None:
324 return self.method
325 elif self.data is not None:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000326 return "POST"
327 else:
328 return "GET"
329
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000330 def get_full_url(self):
Senthil Kumaran52380922013-04-25 05:45:48 -0700331 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000332
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000333 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000334 if self.type == 'https' and not self._tunnel_host:
335 self._tunnel_host = self.host
336 else:
337 self.type= type
338 self.selector = self.full_url
339 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000340
341 def has_proxy(self):
342 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000343
344 def add_header(self, key, val):
345 # useful for something like authentication
346 self.headers[key.capitalize()] = val
347
348 def add_unredirected_header(self, key, val):
349 # will not be added to a redirected request
350 self.unredirected_hdrs[key.capitalize()] = val
351
352 def has_header(self, header_name):
353 return (header_name in self.headers or
354 header_name in self.unredirected_hdrs)
355
356 def get_header(self, header_name, default=None):
357 return self.headers.get(
358 header_name,
359 self.unredirected_hdrs.get(header_name, default))
360
Andrew Svetlovbff98fe2012-11-27 23:06:19 +0200361 def remove_header(self, header_name):
362 self.headers.pop(header_name, None)
363 self.unredirected_hdrs.pop(header_name, None)
364
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000365 def header_items(self):
366 hdrs = self.unredirected_hdrs.copy()
367 hdrs.update(self.headers)
368 return list(hdrs.items())
369
370class OpenerDirector:
371 def __init__(self):
372 client_version = "Python-urllib/%s" % __version__
373 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000374 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000375 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000376 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000377 self.handle_open = {}
378 self.handle_error = {}
379 self.process_response = {}
380 self.process_request = {}
381
382 def add_handler(self, handler):
383 if not hasattr(handler, "add_parent"):
384 raise TypeError("expected BaseHandler instance, got %r" %
385 type(handler))
386
387 added = False
388 for meth in dir(handler):
389 if meth in ["redirect_request", "do_open", "proxy_open"]:
390 # oops, coincidental match
391 continue
392
393 i = meth.find("_")
394 protocol = meth[:i]
395 condition = meth[i+1:]
396
397 if condition.startswith("error"):
398 j = condition.find("_") + i + 1
399 kind = meth[j+1:]
400 try:
401 kind = int(kind)
402 except ValueError:
403 pass
404 lookup = self.handle_error.get(protocol, {})
405 self.handle_error[protocol] = lookup
406 elif condition == "open":
407 kind = protocol
408 lookup = self.handle_open
409 elif condition == "response":
410 kind = protocol
411 lookup = self.process_response
412 elif condition == "request":
413 kind = protocol
414 lookup = self.process_request
415 else:
416 continue
417
418 handlers = lookup.setdefault(kind, [])
419 if handlers:
420 bisect.insort(handlers, handler)
421 else:
422 handlers.append(handler)
423 added = True
424
425 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000426 bisect.insort(self.handlers, handler)
427 handler.add_parent(self)
428
429 def close(self):
430 # Only exists for backwards compatibility.
431 pass
432
433 def _call_chain(self, chain, kind, meth_name, *args):
434 # Handlers raise an exception if no one else should try to handle
435 # the request, or return None if they can't but another handler
436 # could. Otherwise, they return the response.
437 handlers = chain.get(kind, ())
438 for handler in handlers:
439 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000440 result = func(*args)
441 if result is not None:
442 return result
443
444 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
445 # accept a URL or a Request object
446 if isinstance(fullurl, str):
447 req = Request(fullurl, data)
448 else:
449 req = fullurl
450 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000451 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000452
453 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000454 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000455
456 # pre-process request
457 meth_name = protocol+"_request"
458 for processor in self.process_request.get(protocol, []):
459 meth = getattr(processor, meth_name)
460 req = meth(req)
461
462 response = self._open(req, data)
463
464 # post-process response
465 meth_name = protocol+"_response"
466 for processor in self.process_response.get(protocol, []):
467 meth = getattr(processor, meth_name)
468 response = meth(req, response)
469
470 return response
471
472 def _open(self, req, data=None):
473 result = self._call_chain(self.handle_open, 'default',
474 'default_open', req)
475 if result:
476 return result
477
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000478 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000479 result = self._call_chain(self.handle_open, protocol, protocol +
480 '_open', req)
481 if result:
482 return result
483
484 return self._call_chain(self.handle_open, 'unknown',
485 'unknown_open', req)
486
487 def error(self, proto, *args):
488 if proto in ('http', 'https'):
489 # XXX http[s] protocols are special-cased
490 dict = self.handle_error['http'] # https is not different than http
491 proto = args[2] # YUCK!
492 meth_name = 'http_error_%s' % proto
493 http_err = 1
494 orig_args = args
495 else:
496 dict = self.handle_error
497 meth_name = proto + '_error'
498 http_err = 0
499 args = (dict, proto, meth_name) + args
500 result = self._call_chain(*args)
501 if result:
502 return result
503
504 if http_err:
505 args = (dict, 'default', 'http_error_default') + orig_args
506 return self._call_chain(*args)
507
508# XXX probably also want an abstract factory that knows when it makes
509# sense to skip a superclass in favor of a subclass and when it might
510# make sense to include both
511
512def build_opener(*handlers):
513 """Create an opener object from a list of handlers.
514
515 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000516 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000517
518 If any of the handlers passed as arguments are subclasses of the
519 default handlers, the default handlers will not be used.
520 """
521 def isclass(obj):
522 return isinstance(obj, type) or hasattr(obj, "__bases__")
523
524 opener = OpenerDirector()
525 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
526 HTTPDefaultErrorHandler, HTTPRedirectHandler,
Antoine Pitroudf204be2012-11-24 17:59:08 +0100527 FTPHandler, FileHandler, HTTPErrorProcessor,
528 DataHandler]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000529 if hasattr(http.client, "HTTPSConnection"):
530 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000531 skip = set()
532 for klass in default_classes:
533 for check in handlers:
534 if isclass(check):
535 if issubclass(check, klass):
536 skip.add(klass)
537 elif isinstance(check, klass):
538 skip.add(klass)
539 for klass in skip:
540 default_classes.remove(klass)
541
542 for klass in default_classes:
543 opener.add_handler(klass())
544
545 for h in handlers:
546 if isclass(h):
547 h = h()
548 opener.add_handler(h)
549 return opener
550
551class BaseHandler:
552 handler_order = 500
553
554 def add_parent(self, parent):
555 self.parent = parent
556
557 def close(self):
558 # Only exists for backwards compatibility
559 pass
560
561 def __lt__(self, other):
562 if not hasattr(other, "handler_order"):
563 # Try to preserve the old behavior of having custom classes
564 # inserted after default ones (works only for custom user
565 # classes which are not aware of handler_order).
566 return True
567 return self.handler_order < other.handler_order
568
569
570class HTTPErrorProcessor(BaseHandler):
571 """Process HTTP error responses."""
572 handler_order = 1000 # after all other processing
573
574 def http_response(self, request, response):
575 code, msg, hdrs = response.code, response.msg, response.info()
576
577 # According to RFC 2616, "2xx" code indicates that the client's
578 # request was successfully received, understood, and accepted.
579 if not (200 <= code < 300):
580 response = self.parent.error(
581 'http', request, response, code, msg, hdrs)
582
583 return response
584
585 https_response = http_response
586
587class HTTPDefaultErrorHandler(BaseHandler):
588 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000589 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000590
591class HTTPRedirectHandler(BaseHandler):
592 # maximum number of redirections to any single URL
593 # this is needed because of the state that cookies introduce
594 max_repeats = 4
595 # maximum total number of redirections (regardless of URL) before
596 # assuming we're in a loop
597 max_redirections = 10
598
599 def redirect_request(self, req, fp, code, msg, headers, newurl):
600 """Return a Request or None in response to a redirect.
601
602 This is called by the http_error_30x methods when a
603 redirection response is received. If a redirection should
604 take place, return a new Request to allow http_error_30x to
605 perform the redirect. Otherwise, raise HTTPError if no-one
606 else should try to handle this url. Return None if you can't
607 but another Handler might.
608 """
609 m = req.get_method()
610 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
611 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000612 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000613
614 # Strictly (according to RFC 2616), 301 or 302 in response to
615 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000616 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000617 # essentially all clients do redirect in this case, so we do
618 # the same.
619 # be conciliant with URIs containing a space
620 newurl = newurl.replace(' ', '%20')
621 CONTENT_HEADERS = ("content-length", "content-type")
622 newheaders = dict((k, v) for k, v in req.headers.items()
623 if k.lower() not in CONTENT_HEADERS)
624 return Request(newurl,
625 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000626 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000627 unverifiable=True)
628
629 # Implementation note: To avoid the server sending us into an
630 # infinite loop, the request object needs to track what URLs we
631 # have already seen. Do this by adding a handler-specific
632 # attribute to the Request object.
633 def http_error_302(self, req, fp, code, msg, headers):
634 # Some servers (incorrectly) return multiple Location headers
635 # (so probably same goes for URI). Use first header.
636 if "location" in headers:
637 newurl = headers["location"]
638 elif "uri" in headers:
639 newurl = headers["uri"]
640 else:
641 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000642
643 # fix a possible malformed URL
644 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700645
646 # For security reasons we don't allow redirection to anything other
647 # than http, https or ftp.
648
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800649 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800650 raise HTTPError(
651 newurl, code,
652 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
653 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700654
Facundo Batistaf24802c2008-08-17 03:36:03 +0000655 if not urlparts.path:
656 urlparts = list(urlparts)
657 urlparts[2] = "/"
658 newurl = urlunparse(urlparts)
659
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000660 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000661
662 # XXX Probably want to forget about the state of the current
663 # request, although that might interact poorly with other
664 # handlers that also use handler-specific request attributes
665 new = self.redirect_request(req, fp, code, msg, headers, newurl)
666 if new is None:
667 return
668
669 # loop detection
670 # .redirect_dict has a key url if url was previously visited.
671 if hasattr(req, 'redirect_dict'):
672 visited = new.redirect_dict = req.redirect_dict
673 if (visited.get(newurl, 0) >= self.max_repeats or
674 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000675 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000676 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000677 else:
678 visited = new.redirect_dict = req.redirect_dict = {}
679 visited[newurl] = visited.get(newurl, 0) + 1
680
681 # Don't close the fp until we are sure that we won't use it
682 # with HTTPError.
683 fp.read()
684 fp.close()
685
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000686 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000687
688 http_error_301 = http_error_303 = http_error_307 = http_error_302
689
690 inf_msg = "The HTTP server returned a redirect error that would " \
691 "lead to an infinite loop.\n" \
692 "The last 30x error message was:\n"
693
694
695def _parse_proxy(proxy):
696 """Return (scheme, user, password, host/port) given a URL or an authority.
697
698 If a URL is supplied, it must have an authority (host:port) component.
699 According to RFC 3986, having an authority component means the URL must
700 have two slashes after the scheme:
701
702 >>> _parse_proxy('file:/ftp.example.com/')
703 Traceback (most recent call last):
704 ValueError: proxy URL with no authority: 'file:/ftp.example.com/'
705
706 The first three items of the returned tuple may be None.
707
708 Examples of authority parsing:
709
710 >>> _parse_proxy('proxy.example.com')
711 (None, None, None, 'proxy.example.com')
712 >>> _parse_proxy('proxy.example.com:3128')
713 (None, None, None, 'proxy.example.com:3128')
714
715 The authority component may optionally include userinfo (assumed to be
716 username:password):
717
718 >>> _parse_proxy('joe:password@proxy.example.com')
719 (None, 'joe', 'password', 'proxy.example.com')
720 >>> _parse_proxy('joe:password@proxy.example.com:3128')
721 (None, 'joe', 'password', 'proxy.example.com:3128')
722
723 Same examples, but with URLs instead:
724
725 >>> _parse_proxy('http://proxy.example.com/')
726 ('http', None, None, 'proxy.example.com')
727 >>> _parse_proxy('http://proxy.example.com:3128/')
728 ('http', None, None, 'proxy.example.com:3128')
729 >>> _parse_proxy('http://joe:password@proxy.example.com/')
730 ('http', 'joe', 'password', 'proxy.example.com')
731 >>> _parse_proxy('http://joe:password@proxy.example.com:3128')
732 ('http', 'joe', 'password', 'proxy.example.com:3128')
733
734 Everything after the authority is ignored:
735
736 >>> _parse_proxy('ftp://joe:password@proxy.example.com/rubbish:3128')
737 ('ftp', 'joe', 'password', 'proxy.example.com')
738
739 Test for no trailing '/' case:
740
741 >>> _parse_proxy('http://joe:password@proxy.example.com')
742 ('http', 'joe', 'password', 'proxy.example.com')
743
744 """
Georg Brandl13e89462008-07-01 19:56:00 +0000745 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000746 if not r_scheme.startswith("/"):
747 # authority
748 scheme = None
749 authority = proxy
750 else:
751 # URL
752 if not r_scheme.startswith("//"):
753 raise ValueError("proxy URL with no authority: %r" % proxy)
754 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
755 # and 3.3.), path is empty or starts with '/'
756 end = r_scheme.find("/", 2)
757 if end == -1:
758 end = None
759 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000760 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000761 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000762 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000763 else:
764 user = password = None
765 return scheme, user, password, hostport
766
767class ProxyHandler(BaseHandler):
768 # Proxies must be in front
769 handler_order = 100
770
771 def __init__(self, proxies=None):
772 if proxies is None:
773 proxies = getproxies()
774 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
775 self.proxies = proxies
776 for type, url in proxies.items():
777 setattr(self, '%s_open' % type,
Georg Brandlfcbdbf22012-06-24 19:56:31 +0200778 lambda r, proxy=url, type=type, meth=self.proxy_open:
779 meth(r, proxy, type))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000780
781 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000782 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000783 proxy_type, user, password, hostport = _parse_proxy(proxy)
784 if proxy_type is None:
785 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000786
787 if req.host and proxy_bypass(req.host):
788 return None
789
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000790 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000791 user_pass = '%s:%s' % (unquote(user),
792 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000793 creds = base64.b64encode(user_pass.encode()).decode("ascii")
794 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000795 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000796 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000797 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000798 # let other handlers take care of it
799 return None
800 else:
801 # need to start over, because the other handlers don't
802 # grok the proxy's URL type
803 # e.g. if we have a constructor arg proxies like so:
804 # {'http': 'ftp://proxy.example.com'}, we may end up turning
805 # a request for http://acme.example.com/a into one for
806 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000807 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000808
809class HTTPPasswordMgr:
810
811 def __init__(self):
812 self.passwd = {}
813
814 def add_password(self, realm, uri, user, passwd):
815 # uri could be a single URI or a sequence
816 if isinstance(uri, str):
817 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800818 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000819 self.passwd[realm] = {}
820 for default_port in True, False:
821 reduced_uri = tuple(
822 [self.reduce_uri(u, default_port) for u in uri])
823 self.passwd[realm][reduced_uri] = (user, passwd)
824
825 def find_user_password(self, realm, authuri):
826 domains = self.passwd.get(realm, {})
827 for default_port in True, False:
828 reduced_authuri = self.reduce_uri(authuri, default_port)
829 for uris, authinfo in domains.items():
830 for uri in uris:
831 if self.is_suburi(uri, reduced_authuri):
832 return authinfo
833 return None, None
834
835 def reduce_uri(self, uri, default_port=True):
836 """Accept authority or URI and extract only the authority and path."""
837 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000838 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000839 if parts[1]:
840 # URI
841 scheme = parts[0]
842 authority = parts[1]
843 path = parts[2] or '/'
844 else:
845 # host or host:port
846 scheme = None
847 authority = uri
848 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000849 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000850 if default_port and port is None and scheme is not None:
851 dport = {"http": 80,
852 "https": 443,
853 }.get(scheme)
854 if dport is not None:
855 authority = "%s:%d" % (host, dport)
856 return authority, path
857
858 def is_suburi(self, base, test):
859 """Check if test is below base in a URI tree
860
861 Both args must be URIs in reduced form.
862 """
863 if base == test:
864 return True
865 if base[0] != test[0]:
866 return False
867 common = posixpath.commonprefix((base[1], test[1]))
868 if len(common) == len(base[1]):
869 return True
870 return False
871
872
873class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
874
875 def find_user_password(self, realm, authuri):
876 user, password = HTTPPasswordMgr.find_user_password(self, realm,
877 authuri)
878 if user is not None:
879 return user, password
880 return HTTPPasswordMgr.find_user_password(self, None, authuri)
881
882
883class AbstractBasicAuthHandler:
884
885 # XXX this allows for multiple auth-schemes, but will stupidly pick
886 # the last one with a realm specified.
887
888 # allow for double- and single-quoted realm values
889 # (single quotes are a violation of the RFC, but appear in the wild)
890 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800891 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000892
893 # XXX could pre-emptively send auth info already accepted (RFC 2617,
894 # end of section 2, and section 1.2 immediately after "credentials"
895 # production).
896
897 def __init__(self, password_mgr=None):
898 if password_mgr is None:
899 password_mgr = HTTPPasswordMgr()
900 self.passwd = password_mgr
901 self.add_password = self.passwd.add_password
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000902 self.retried = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000903
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000904 def reset_retry_count(self):
905 self.retried = 0
906
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000907 def http_error_auth_reqed(self, authreq, host, req, headers):
908 # host may be an authority (without userinfo) or a URL with an
909 # authority
910 # XXX could be multiple headers
911 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000912
913 if self.retried > 5:
914 # retry sending the username:password 5 times before failing.
915 raise HTTPError(req.get_full_url(), 401, "basic auth failed",
916 headers, None)
917 else:
918 self.retried += 1
919
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000920 if authreq:
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800921 scheme = authreq.split()[0]
Senthil Kumaran1a129c82011-10-20 02:50:13 +0800922 if scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800923 raise ValueError("AbstractBasicAuthHandler does not"
924 " support the following scheme: '%s'" %
925 scheme)
926 else:
927 mo = AbstractBasicAuthHandler.rx.search(authreq)
928 if mo:
929 scheme, quote, realm = mo.groups()
Senthil Kumaran92a5bf02012-05-16 00:03:29 +0800930 if quote not in ['"',"'"]:
931 warnings.warn("Basic Auth Realm was unquoted",
932 UserWarning, 2)
Senthil Kumaran4de00a22011-05-11 21:17:57 +0800933 if scheme.lower() == 'basic':
934 response = self.retry_http_basic_auth(host, req, realm)
935 if response and response.code != 401:
936 self.retried = 0
937 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000938
939 def retry_http_basic_auth(self, host, req, realm):
940 user, pw = self.passwd.find_user_password(realm, host)
941 if pw is not None:
942 raw = "%s:%s" % (user, pw)
943 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
944 if req.headers.get(self.auth_header, None) == auth:
945 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000946 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000947 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000948 else:
949 return None
950
951
952class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
953
954 auth_header = 'Authorization'
955
956 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000957 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000958 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000959 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000960 self.reset_retry_count()
961 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000962
963
964class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
965
966 auth_header = 'Proxy-authorization'
967
968 def http_error_407(self, req, fp, code, msg, headers):
969 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000970 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000971 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
972 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000973 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000974 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000975 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000976 self.reset_retry_count()
977 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000978
979
Senthil Kumaran6c5bd402011-11-01 23:20:31 +0800980# Return n random bytes.
981_randombytes = os.urandom
982
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000983
984class AbstractDigestAuthHandler:
985 # Digest authentication is specified in RFC 2617.
986
987 # XXX The client does not inspect the Authentication-Info header
988 # in a successful response.
989
990 # XXX It should be possible to test this implementation against
991 # a mock server that just generates a static set of challenges.
992
993 # XXX qop="auth-int" supports is shaky
994
995 def __init__(self, passwd=None):
996 if passwd is None:
997 passwd = HTTPPasswordMgr()
998 self.passwd = passwd
999 self.add_password = self.passwd.add_password
1000 self.retried = 0
1001 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001002 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001003
1004 def reset_retry_count(self):
1005 self.retried = 0
1006
1007 def http_error_auth_reqed(self, auth_header, host, req, headers):
1008 authreq = headers.get(auth_header, None)
1009 if self.retried > 5:
1010 # Don't fail endlessly - if we failed once, we'll probably
1011 # fail a second time. Hm. Unless the Password Manager is
1012 # prompting for the information. Crap. This isn't great
1013 # but it's better than the current 'repeat until recursion
1014 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001015 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +00001016 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001017 else:
1018 self.retried += 1
1019 if authreq:
1020 scheme = authreq.split()[0]
1021 if scheme.lower() == 'digest':
1022 return self.retry_http_digest_auth(req, authreq)
Senthil Kumaran1a129c82011-10-20 02:50:13 +08001023 elif scheme.lower() != 'basic':
Senthil Kumaran4de00a22011-05-11 21:17:57 +08001024 raise ValueError("AbstractDigestAuthHandler does not support"
1025 " the following scheme: '%s'" % scheme)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001026
1027 def retry_http_digest_auth(self, req, auth):
1028 token, challenge = auth.split(' ', 1)
1029 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
1030 auth = self.get_authorization(req, chal)
1031 if auth:
1032 auth_val = 'Digest %s' % auth
1033 if req.headers.get(self.auth_header, None) == auth_val:
1034 return None
1035 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +00001036 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001037 return resp
1038
1039 def get_cnonce(self, nonce):
1040 # The cnonce-value is an opaque
1041 # quoted string value provided by the client and used by both client
1042 # and server to avoid chosen plaintext attacks, to provide mutual
1043 # authentication, and to provide some message integrity protection.
1044 # This isn't a fabulous effort, but it's probably Good Enough.
1045 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001046 b = s.encode("ascii") + _randombytes(8)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001047 dig = hashlib.sha1(b).hexdigest()
1048 return dig[:16]
1049
1050 def get_authorization(self, req, chal):
1051 try:
1052 realm = chal['realm']
1053 nonce = chal['nonce']
1054 qop = chal.get('qop')
1055 algorithm = chal.get('algorithm', 'MD5')
1056 # mod_digest doesn't send an opaque, even though it isn't
1057 # supposed to be optional
1058 opaque = chal.get('opaque', None)
1059 except KeyError:
1060 return None
1061
1062 H, KD = self.get_algorithm_impls(algorithm)
1063 if H is None:
1064 return None
1065
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001066 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001067 if user is None:
1068 return None
1069
1070 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001071 if req.data is not None:
1072 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001073 else:
1074 entdig = None
1075
1076 A1 = "%s:%s:%s" % (user, realm, pw)
1077 A2 = "%s:%s" % (req.get_method(),
1078 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001079 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001080 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +00001081 if nonce == self.last_nonce:
1082 self.nonce_count += 1
1083 else:
1084 self.nonce_count = 1
1085 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001086 ncvalue = '%08x' % self.nonce_count
1087 cnonce = self.get_cnonce(nonce)
1088 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
1089 respdig = KD(H(A1), noncebit)
1090 elif qop is None:
1091 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
1092 else:
1093 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +00001094 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001095
1096 # XXX should the partial digests be encoded too?
1097
1098 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001099 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001100 respdig)
1101 if opaque:
1102 base += ', opaque="%s"' % opaque
1103 if entdig:
1104 base += ', digest="%s"' % entdig
1105 base += ', algorithm="%s"' % algorithm
1106 if qop:
1107 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1108 return base
1109
1110 def get_algorithm_impls(self, algorithm):
1111 # lambdas assume digest modules are imported at the top level
1112 if algorithm == 'MD5':
1113 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1114 elif algorithm == 'SHA':
1115 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1116 # XXX MD5-sess
1117 KD = lambda s, d: H("%s:%s" % (s, d))
1118 return H, KD
1119
1120 def get_entity_digest(self, data, chal):
1121 # XXX not implemented yet
1122 return None
1123
1124
1125class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1126 """An authentication protocol defined by RFC 2069
1127
1128 Digest authentication improves on basic authentication because it
1129 does not transmit passwords in the clear.
1130 """
1131
1132 auth_header = 'Authorization'
1133 handler_order = 490 # before Basic auth
1134
1135 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001136 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001137 retry = self.http_error_auth_reqed('www-authenticate',
1138 host, req, headers)
1139 self.reset_retry_count()
1140 return retry
1141
1142
1143class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1144
1145 auth_header = 'Proxy-Authorization'
1146 handler_order = 490 # before Basic auth
1147
1148 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001149 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001150 retry = self.http_error_auth_reqed('proxy-authenticate',
1151 host, req, headers)
1152 self.reset_retry_count()
1153 return retry
1154
1155class AbstractHTTPHandler(BaseHandler):
1156
1157 def __init__(self, debuglevel=0):
1158 self._debuglevel = debuglevel
1159
1160 def set_http_debuglevel(self, level):
1161 self._debuglevel = level
1162
1163 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001164 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001165 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001166 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001167
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001168 if request.data is not None: # POST
1169 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001170 if isinstance(data, str):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001171 msg = "POST data should be bytes or an iterable of bytes. " \
1172 "It cannot be of type str."
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001173 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001174 if not request.has_header('Content-type'):
1175 request.add_unredirected_header(
1176 'Content-type',
1177 'application/x-www-form-urlencoded')
1178 if not request.has_header('Content-length'):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001179 try:
1180 mv = memoryview(data)
1181 except TypeError:
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001182 if isinstance(data, collections.Iterable):
Georg Brandl61536042011-02-03 07:46:41 +00001183 raise ValueError("Content-Length should be specified "
1184 "for iterable data of type %r %r" % (type(data),
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001185 data))
1186 else:
1187 request.add_unredirected_header(
Senthil Kumaran1e991f22010-12-24 04:03:59 +00001188 'Content-length', '%d' % (len(mv) * mv.itemsize))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001189
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001190 sel_host = host
1191 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001192 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001193 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001194 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001195 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001196 for name, value in self.parent.addheaders:
1197 name = name.capitalize()
1198 if not request.has_header(name):
1199 request.add_unredirected_header(name, value)
1200
1201 return request
1202
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001203 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001204 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205
1206 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001207 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001208 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001209 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001210 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001211
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001212 # will parse host:port
1213 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001214
1215 headers = dict(req.unredirected_hdrs)
1216 headers.update(dict((k, v) for k, v in req.headers.items()
1217 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001218
1219 # TODO(jhylton): Should this be redesigned to handle
1220 # persistent connections?
1221
1222 # We want to make an HTTP/1.1 request, but the addinfourl
1223 # class isn't prepared to deal with a persistent connection.
1224 # It will try to read all remaining data from the socket,
1225 # which will block while the server waits for the next request.
1226 # So make sure the connection gets closed after the (only)
1227 # request.
1228 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001229 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001230
1231 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001232 tunnel_headers = {}
1233 proxy_auth_hdr = "Proxy-Authorization"
1234 if proxy_auth_hdr in headers:
1235 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1236 # Proxy-Authorization should not be sent to origin
1237 # server.
1238 del headers[proxy_auth_hdr]
1239 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001240
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001241 try:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001242 h.request(req.get_method(), req.selector, req.data, headers)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001243 except OSError as err: # timeout error
Senthil Kumaran45686b42011-07-27 09:31:03 +08001244 h.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001245 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001246 else:
1247 r = h.getresponse()
Nadeem Vawdabd26b542012-10-21 17:37:43 +02001248 # If the server does not send us a 'Connection: close' header,
1249 # HTTPConnection assumes the socket should be left open. Manually
1250 # mark the socket to be closed when this response object goes away.
1251 if h.sock:
1252 h.sock.close()
1253 h.sock = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001254
Senthil Kumaran26430412011-04-13 07:01:19 +08001255 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001256 # This line replaces the .msg attribute of the HTTPResponse
1257 # with .headers, because urllib clients expect the response to
1258 # have the reason in .msg. It would be good to mark this
1259 # attribute is deprecated and get then to use info() or
1260 # .headers.
1261 r.msg = r.reason
1262 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001263
1264
1265class HTTPHandler(AbstractHTTPHandler):
1266
1267 def http_open(self, req):
1268 return self.do_open(http.client.HTTPConnection, req)
1269
1270 http_request = AbstractHTTPHandler.do_request_
1271
1272if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001273
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001274 class HTTPSHandler(AbstractHTTPHandler):
1275
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001276 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1277 AbstractHTTPHandler.__init__(self, debuglevel)
1278 self._context = context
1279 self._check_hostname = check_hostname
1280
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001281 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001282 return self.do_open(http.client.HTTPSConnection, req,
1283 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001284
1285 https_request = AbstractHTTPHandler.do_request_
1286
Senthil Kumaran4c875a92011-11-01 23:57:57 +08001287 __all__.append('HTTPSHandler')
Senthil Kumaran0d54eb92011-11-01 23:49:46 +08001288
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001289class HTTPCookieProcessor(BaseHandler):
1290 def __init__(self, cookiejar=None):
1291 import http.cookiejar
1292 if cookiejar is None:
1293 cookiejar = http.cookiejar.CookieJar()
1294 self.cookiejar = cookiejar
1295
1296 def http_request(self, request):
1297 self.cookiejar.add_cookie_header(request)
1298 return request
1299
1300 def http_response(self, request, response):
1301 self.cookiejar.extract_cookies(response, request)
1302 return response
1303
1304 https_request = http_request
1305 https_response = http_response
1306
1307class UnknownHandler(BaseHandler):
1308 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001309 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001310 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001311
1312def parse_keqv_list(l):
1313 """Parse list of key=value strings where keys are not duplicated."""
1314 parsed = {}
1315 for elt in l:
1316 k, v = elt.split('=', 1)
1317 if v[0] == '"' and v[-1] == '"':
1318 v = v[1:-1]
1319 parsed[k] = v
1320 return parsed
1321
1322def parse_http_list(s):
1323 """Parse lists as described by RFC 2068 Section 2.
1324
1325 In particular, parse comma-separated lists where the elements of
1326 the list may include quoted-strings. A quoted-string could
1327 contain a comma. A non-quoted string could have quotes in the
1328 middle. Neither commas nor quotes count if they are escaped.
1329 Only double-quotes count, not single-quotes.
1330 """
1331 res = []
1332 part = ''
1333
1334 escape = quote = False
1335 for cur in s:
1336 if escape:
1337 part += cur
1338 escape = False
1339 continue
1340 if quote:
1341 if cur == '\\':
1342 escape = True
1343 continue
1344 elif cur == '"':
1345 quote = False
1346 part += cur
1347 continue
1348
1349 if cur == ',':
1350 res.append(part)
1351 part = ''
1352 continue
1353
1354 if cur == '"':
1355 quote = True
1356
1357 part += cur
1358
1359 # append last part
1360 if part:
1361 res.append(part)
1362
1363 return [part.strip() for part in res]
1364
1365class FileHandler(BaseHandler):
1366 # Use local file or FTP depending on form of URL
1367 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001368 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001369 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1370 req.host != 'localhost'):
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001371 if not req.host is self.get_names():
1372 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001373 else:
1374 return self.open_local_file(req)
1375
1376 # names for the localhost
1377 names = None
1378 def get_names(self):
1379 if FileHandler.names is None:
1380 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001381 FileHandler.names = tuple(
1382 socket.gethostbyname_ex('localhost')[2] +
1383 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001384 except socket.gaierror:
1385 FileHandler.names = (socket.gethostbyname('localhost'),)
1386 return FileHandler.names
1387
1388 # not entirely sure what the rules are here
1389 def open_local_file(self, req):
1390 import email.utils
1391 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001392 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001393 filename = req.selector
1394 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001395 try:
1396 stats = os.stat(localfile)
1397 size = stats.st_size
1398 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001399 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001400 headers = email.message_from_string(
1401 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1402 (mtype or 'text/plain', size, modified))
1403 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001404 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001405 if not host or \
1406 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001407 if host:
1408 origurl = 'file://' + host + filename
1409 else:
1410 origurl = 'file://' + filename
1411 return addinfourl(open(localfile, 'rb'), headers, origurl)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001412 except OSError as exp:
Georg Brandl029986a2008-06-23 11:44:14 +00001413 # users shouldn't expect OSErrors coming from urlopen()
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001414 raise URLError(exp)
Georg Brandl13e89462008-07-01 19:56:00 +00001415 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001416
1417def _safe_gethostbyname(host):
1418 try:
1419 return socket.gethostbyname(host)
1420 except socket.gaierror:
1421 return None
1422
1423class FTPHandler(BaseHandler):
1424 def ftp_open(self, req):
1425 import ftplib
1426 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001427 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001428 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001429 raise URLError('ftp error: no host given')
1430 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001431 if port is None:
1432 port = ftplib.FTP_PORT
1433 else:
1434 port = int(port)
1435
1436 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001437 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001438 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001439 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001440 else:
1441 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001442 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001443 user = user or ''
1444 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001445
1446 try:
1447 host = socket.gethostbyname(host)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001448 except OSError as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001449 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001450 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001451 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001452 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001453 dirs, file = dirs[:-1], dirs[-1]
1454 if dirs and not dirs[0]:
1455 dirs = dirs[1:]
1456 try:
1457 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1458 type = file and 'I' or 'D'
1459 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001460 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001461 if attr.lower() == 'type' and \
1462 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1463 type = value.upper()
1464 fp, retrlen = fw.retrfile(file, type)
1465 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001466 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001467 if mtype:
1468 headers += "Content-type: %s\n" % mtype
1469 if retrlen is not None and retrlen >= 0:
1470 headers += "Content-length: %d\n" % retrlen
1471 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001472 return addinfourl(fp, headers, req.full_url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001473 except ftplib.all_errors as exp:
1474 exc = URLError('ftp error: %r' % exp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001475 raise exc.with_traceback(sys.exc_info()[2])
1476
1477 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001478 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1479 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001480
1481class CacheFTPHandler(FTPHandler):
1482 # XXX would be nice to have pluggable cache strategies
1483 # XXX this stuff is definitely not thread safe
1484 def __init__(self):
1485 self.cache = {}
1486 self.timeout = {}
1487 self.soonest = 0
1488 self.delay = 60
1489 self.max_conns = 16
1490
1491 def setTimeout(self, t):
1492 self.delay = t
1493
1494 def setMaxConns(self, m):
1495 self.max_conns = m
1496
1497 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1498 key = user, host, port, '/'.join(dirs), timeout
1499 if key in self.cache:
1500 self.timeout[key] = time.time() + self.delay
1501 else:
1502 self.cache[key] = ftpwrapper(user, passwd, host, port,
1503 dirs, timeout)
1504 self.timeout[key] = time.time() + self.delay
1505 self.check_cache()
1506 return self.cache[key]
1507
1508 def check_cache(self):
1509 # first check for old ones
1510 t = time.time()
1511 if self.soonest <= t:
1512 for k, v in list(self.timeout.items()):
1513 if v < t:
1514 self.cache[k].close()
1515 del self.cache[k]
1516 del self.timeout[k]
1517 self.soonest = min(list(self.timeout.values()))
1518
1519 # then check the size
1520 if len(self.cache) == self.max_conns:
1521 for k, v in list(self.timeout.items()):
1522 if v == self.soonest:
1523 del self.cache[k]
1524 del self.timeout[k]
1525 break
1526 self.soonest = min(list(self.timeout.values()))
1527
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001528 def clear_cache(self):
1529 for conn in self.cache.values():
1530 conn.close()
1531 self.cache.clear()
1532 self.timeout.clear()
1533
Antoine Pitroudf204be2012-11-24 17:59:08 +01001534class DataHandler(BaseHandler):
1535 def data_open(self, req):
1536 # data URLs as specified in RFC 2397.
1537 #
1538 # ignores POSTed data
1539 #
1540 # syntax:
1541 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1542 # mediatype := [ type "/" subtype ] *( ";" parameter )
1543 # data := *urlchar
1544 # parameter := attribute "=" value
1545 url = req.full_url
1546
1547 scheme, data = url.split(":",1)
1548 mediatype, data = data.split(",",1)
1549
1550 # even base64 encoded data URLs might be quoted so unquote in any case:
1551 data = unquote_to_bytes(data)
1552 if mediatype.endswith(";base64"):
1553 data = base64.decodebytes(data)
1554 mediatype = mediatype[:-7]
1555
1556 if not mediatype:
1557 mediatype = "text/plain;charset=US-ASCII"
1558
1559 headers = email.message_from_string("Content-type: %s\nContent-length: %d\n" %
1560 (mediatype, len(data)))
1561
1562 return addinfourl(io.BytesIO(data), headers, url)
1563
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001564
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001565# Code move from the old urllib module
1566
1567MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1568
1569# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001570if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001571 from nturl2path import url2pathname, pathname2url
1572else:
1573 def url2pathname(pathname):
1574 """OS-specific conversion from a relative URL of the 'file' scheme
1575 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001576 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001577
1578 def pathname2url(pathname):
1579 """OS-specific conversion from a file system path to a relative URL
1580 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001581 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001582
1583# This really consists of two pieces:
1584# (1) a class which handles opening of all sorts of URLs
1585# (plus assorted utilities etc.)
1586# (2) a set of functions for parsing URLs
1587# XXX Should these be separated out into different modules?
1588
1589
1590ftpcache = {}
1591class URLopener:
1592 """Class to open URLs.
1593 This is a class rather than just a subroutine because we may need
1594 more than one set of global protocol-specific options.
1595 Note -- this is a base class for those who don't want the
1596 automatic handling of errors type 302 (relocated) and 401
1597 (authorization needed)."""
1598
1599 __tempfiles = None
1600
1601 version = "Python-urllib/%s" % __version__
1602
1603 # Constructor
1604 def __init__(self, proxies=None, **x509):
Georg Brandlfcbdbf22012-06-24 19:56:31 +02001605 msg = "%(class)s style of invoking requests is deprecated. " \
Senthil Kumaran38b968b92012-03-14 13:43:53 -07001606 "Use newer urlopen functions/methods" % {'class': self.__class__.__name__}
1607 warnings.warn(msg, DeprecationWarning, stacklevel=3)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001608 if proxies is None:
1609 proxies = getproxies()
1610 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1611 self.proxies = proxies
1612 self.key_file = x509.get('key_file')
1613 self.cert_file = x509.get('cert_file')
1614 self.addheaders = [('User-Agent', self.version)]
1615 self.__tempfiles = []
1616 self.__unlink = os.unlink # See cleanup()
1617 self.tempcache = None
1618 # Undocumented feature: if you assign {} to tempcache,
1619 # it is used to cache files retrieved with
1620 # self.retrieve(). This is not enabled by default
1621 # since it does not work for changing documents (and I
1622 # haven't got the logic to check expiration headers
1623 # yet).
1624 self.ftpcache = ftpcache
1625 # Undocumented feature: you can use a different
1626 # ftp cache by assigning to the .ftpcache member;
1627 # in case you want logically independent URL openers
1628 # XXX This is not threadsafe. Bah.
1629
1630 def __del__(self):
1631 self.close()
1632
1633 def close(self):
1634 self.cleanup()
1635
1636 def cleanup(self):
1637 # This code sometimes runs when the rest of this module
1638 # has already been deleted, so it can't use any globals
1639 # or import anything.
1640 if self.__tempfiles:
1641 for file in self.__tempfiles:
1642 try:
1643 self.__unlink(file)
1644 except OSError:
1645 pass
1646 del self.__tempfiles[:]
1647 if self.tempcache:
1648 self.tempcache.clear()
1649
1650 def addheader(self, *args):
1651 """Add a header to be used by the HTTP interface only
1652 e.g. u.addheader('Accept', 'sound/basic')"""
1653 self.addheaders.append(args)
1654
1655 # External interface
1656 def open(self, fullurl, data=None):
1657 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001658 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001659 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001660 if self.tempcache and fullurl in self.tempcache:
1661 filename, headers = self.tempcache[fullurl]
1662 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001663 return addinfourl(fp, headers, fullurl)
1664 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001665 if not urltype:
1666 urltype = 'file'
1667 if urltype in self.proxies:
1668 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001669 urltype, proxyhost = splittype(proxy)
1670 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001671 url = (host, fullurl) # Signal special case to open_*()
1672 else:
1673 proxy = None
1674 name = 'open_' + urltype
1675 self.type = urltype
1676 name = name.replace('-', '_')
1677 if not hasattr(self, name):
1678 if proxy:
1679 return self.open_unknown_proxy(proxy, fullurl, data)
1680 else:
1681 return self.open_unknown(fullurl, data)
1682 try:
1683 if data is None:
1684 return getattr(self, name)(url)
1685 else:
1686 return getattr(self, name)(url, data)
Senthil Kumaranf5776862012-10-21 13:30:02 -07001687 except (HTTPError, URLError):
Antoine Pitrou6b4883d2011-10-12 02:54:14 +02001688 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001689 except OSError as msg:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001690 raise OSError('socket error', msg).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001691
1692 def open_unknown(self, fullurl, data=None):
1693 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001694 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001695 raise OSError('url error', 'unknown url type', type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001696
1697 def open_unknown_proxy(self, proxy, fullurl, data=None):
1698 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001699 type, url = splittype(fullurl)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001700 raise OSError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001701
1702 # External interface
1703 def retrieve(self, url, filename=None, reporthook=None, data=None):
1704 """retrieve(url) returns (filename, headers) for a local object
1705 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001706 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001707 if self.tempcache and url in self.tempcache:
1708 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001709 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001710 if filename is None and (not type or type == 'file'):
1711 try:
1712 fp = self.open_local_file(url1)
1713 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001714 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001715 return url2pathname(splithost(url1)[1]), hdrs
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001716 except OSError as msg:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001717 pass
1718 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001719 try:
1720 headers = fp.info()
1721 if filename:
1722 tfp = open(filename, 'wb')
1723 else:
1724 import tempfile
1725 garbage, path = splittype(url)
1726 garbage, path = splithost(path or "")
1727 path, garbage = splitquery(path or "")
1728 path, garbage = splitattr(path or "")
1729 suffix = os.path.splitext(path)[1]
1730 (fd, filename) = tempfile.mkstemp(suffix)
1731 self.__tempfiles.append(filename)
1732 tfp = os.fdopen(fd, 'wb')
1733 try:
1734 result = filename, headers
1735 if self.tempcache is not None:
1736 self.tempcache[url] = result
1737 bs = 1024*8
1738 size = -1
1739 read = 0
1740 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001741 if "content-length" in headers:
1742 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001743 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001744 reporthook(blocknum, bs, size)
1745 while 1:
1746 block = fp.read(bs)
1747 if not block:
1748 break
1749 read += len(block)
1750 tfp.write(block)
1751 blocknum += 1
1752 if reporthook:
1753 reporthook(blocknum, bs, size)
1754 finally:
1755 tfp.close()
1756 finally:
1757 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001758
1759 # raise exception if actual size does not match content-length header
1760 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001761 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001762 "retrieval incomplete: got only %i out of %i bytes"
1763 % (read, size), result)
1764
1765 return result
1766
1767 # Each method named open_<type> knows how to open that type of URL
1768
1769 def _open_generic_http(self, connection_factory, url, data):
1770 """Make an HTTP connection using connection_class.
1771
1772 This is an internal method that should be called from
1773 open_http() or open_https().
1774
1775 Arguments:
1776 - connection_factory should take a host name and return an
1777 HTTPConnection instance.
1778 - url is the url to retrieval or a host, relative-path pair.
1779 - data is payload for a POST request or None.
1780 """
1781
1782 user_passwd = None
1783 proxy_passwd= None
1784 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001785 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001786 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001787 user_passwd, host = splituser(host)
1788 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001789 realhost = host
1790 else:
1791 host, selector = url
1792 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001793 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001794 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001795 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001796 url = rest
1797 user_passwd = None
1798 if urltype.lower() != 'http':
1799 realhost = None
1800 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001801 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001802 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001803 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001804 if user_passwd:
1805 selector = "%s://%s%s" % (urltype, realhost, rest)
1806 if proxy_bypass(realhost):
1807 host = realhost
1808
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001809 if not host: raise OSError('http error', 'no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001810
1811 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001812 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001813 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001814 else:
1815 proxy_auth = None
1816
1817 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001818 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001819 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001820 else:
1821 auth = None
1822 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001823 headers = {}
1824 if proxy_auth:
1825 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1826 if auth:
1827 headers["Authorization"] = "Basic %s" % auth
1828 if realhost:
1829 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001830
1831 # Add Connection:close as we don't support persistent connections yet.
1832 # This helps in closing the socket and avoiding ResourceWarning
1833
1834 headers["Connection"] = "close"
1835
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001836 for header, value in self.addheaders:
1837 headers[header] = value
1838
1839 if data is not None:
1840 headers["Content-Type"] = "application/x-www-form-urlencoded"
1841 http_conn.request("POST", selector, data, headers)
1842 else:
1843 http_conn.request("GET", selector, headers=headers)
1844
1845 try:
1846 response = http_conn.getresponse()
1847 except http.client.BadStatusLine:
1848 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001849 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001850
1851 # According to RFC 2616, "2xx" code indicates that the client's
1852 # request was successfully received, understood, and accepted.
1853 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001854 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001855 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001856 else:
1857 return self.http_error(
1858 url, response.fp,
1859 response.status, response.reason, response.msg, data)
1860
1861 def open_http(self, url, data=None):
1862 """Use HTTP protocol."""
1863 return self._open_generic_http(http.client.HTTPConnection, url, data)
1864
1865 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1866 """Handle http errors.
1867
1868 Derived class can override this, or provide specific handlers
1869 named http_error_DDD where DDD is the 3-digit error code."""
1870 # First check if there's a specific handler for this error
1871 name = 'http_error_%d' % errcode
1872 if hasattr(self, name):
1873 method = getattr(self, name)
1874 if data is None:
1875 result = method(url, fp, errcode, errmsg, headers)
1876 else:
1877 result = method(url, fp, errcode, errmsg, headers, data)
1878 if result: return result
1879 return self.http_error_default(url, fp, errcode, errmsg, headers)
1880
1881 def http_error_default(self, url, fp, errcode, errmsg, headers):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001882 """Default error handler: close the connection and raise OSError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001883 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001884 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001885
1886 if _have_ssl:
1887 def _https_connection(self, host):
1888 return http.client.HTTPSConnection(host,
1889 key_file=self.key_file,
1890 cert_file=self.cert_file)
1891
1892 def open_https(self, url, data=None):
1893 """Use HTTPS protocol."""
1894 return self._open_generic_http(self._https_connection, url, data)
1895
1896 def open_file(self, url):
1897 """Use local file or FTP depending on form of URL."""
1898 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001899 raise URLError('file error: proxy support for file protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001900 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001901 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001902 else:
1903 return self.open_local_file(url)
1904
1905 def open_local_file(self, url):
1906 """Use local file."""
Senthil Kumaran6c5bd402011-11-01 23:20:31 +08001907 import email.utils
1908 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001909 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001910 localname = url2pathname(file)
1911 try:
1912 stats = os.stat(localname)
1913 except OSError as e:
Senthil Kumaranf5776862012-10-21 13:30:02 -07001914 raise URLError(e.strerror, e.filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001915 size = stats.st_size
1916 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1917 mtype = mimetypes.guess_type(url)[0]
1918 headers = email.message_from_string(
1919 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1920 (mtype or 'text/plain', size, modified))
1921 if not host:
1922 urlfile = file
1923 if file[:1] == '/':
1924 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001925 return addinfourl(open(localname, 'rb'), headers, urlfile)
1926 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001927 if (not port
Senthil Kumaran40d80782012-10-22 09:43:04 -07001928 and socket.gethostbyname(host) in ((localhost(),) + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001929 urlfile = file
1930 if file[:1] == '/':
1931 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08001932 elif file[:2] == './':
1933 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00001934 return addinfourl(open(localname, 'rb'), headers, urlfile)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001935 raise URLError('local file error: not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001936
1937 def open_ftp(self, url):
1938 """Use FTP protocol."""
1939 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001940 raise URLError('ftp error: proxy support for ftp protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001941 import mimetypes
Georg Brandl13e89462008-07-01 19:56:00 +00001942 host, path = splithost(url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001943 if not host: raise URLError('ftp error: no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001944 host, port = splitport(host)
1945 user, host = splituser(host)
1946 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001947 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001948 host = unquote(host)
1949 user = unquote(user or '')
1950 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001951 host = socket.gethostbyname(host)
1952 if not port:
1953 import ftplib
1954 port = ftplib.FTP_PORT
1955 else:
1956 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001957 path, attrs = splitattr(path)
1958 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001959 dirs = path.split('/')
1960 dirs, file = dirs[:-1], dirs[-1]
1961 if dirs and not dirs[0]: dirs = dirs[1:]
1962 if dirs and not dirs[0]: dirs[0] = '/'
1963 key = user, host, port, '/'.join(dirs)
1964 # XXX thread unsafe!
1965 if len(self.ftpcache) > MAXFTPCACHE:
1966 # Prune the cache, rather arbitrarily
1967 for k in self.ftpcache.keys():
1968 if k != key:
1969 v = self.ftpcache[k]
1970 del self.ftpcache[k]
1971 v.close()
1972 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001973 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001974 self.ftpcache[key] = \
1975 ftpwrapper(user, passwd, host, port, dirs)
1976 if not file: type = 'D'
1977 else: type = 'I'
1978 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001979 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001980 if attr.lower() == 'type' and \
1981 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1982 type = value.upper()
1983 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1984 mtype = mimetypes.guess_type("ftp:" + url)[0]
1985 headers = ""
1986 if mtype:
1987 headers += "Content-Type: %s\n" % mtype
1988 if retrlen is not None and retrlen >= 0:
1989 headers += "Content-Length: %d\n" % retrlen
1990 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001991 return addinfourl(fp, headers, "ftp:" + url)
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001992 except ftperrors() as exp:
1993 raise URLError('ftp error %r' % exp).with_traceback(sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001994
1995 def open_data(self, url, data=None):
1996 """Use "data" URL."""
1997 if not isinstance(url, str):
Senthil Kumaran3ebef362012-10-21 18:31:25 -07001998 raise URLError('data error: proxy support for data protocol currently not implemented')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001999 # ignore POSTed data
2000 #
2001 # syntax of data URLs:
2002 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
2003 # mediatype := [ type "/" subtype ] *( ";" parameter )
2004 # data := *urlchar
2005 # parameter := attribute "=" value
2006 try:
2007 [type, data] = url.split(',', 1)
2008 except ValueError:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002009 raise OSError('data error', 'bad data URL')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002010 if not type:
2011 type = 'text/plain;charset=US-ASCII'
2012 semi = type.rfind(';')
2013 if semi >= 0 and '=' not in type[semi:]:
2014 encoding = type[semi+1:]
2015 type = type[:semi]
2016 else:
2017 encoding = ''
2018 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00002019 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002020 time.gmtime(time.time())))
2021 msg.append('Content-type: %s' % type)
2022 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00002023 # XXX is this encoding/decoding ok?
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00002024 data = base64.decodebytes(data.encode('ascii')).decode('latin-1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002025 else:
Georg Brandl13e89462008-07-01 19:56:00 +00002026 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002027 msg.append('Content-Length: %d' % len(data))
2028 msg.append('')
2029 msg.append(data)
2030 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00002031 headers = email.message_from_string(msg)
2032 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002033 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00002034 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002035
2036
2037class FancyURLopener(URLopener):
2038 """Derived class with handlers for errors we can handle (perhaps)."""
2039
2040 def __init__(self, *args, **kwargs):
2041 URLopener.__init__(self, *args, **kwargs)
2042 self.auth_cache = {}
2043 self.tries = 0
2044 self.maxtries = 10
2045
2046 def http_error_default(self, url, fp, errcode, errmsg, headers):
2047 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00002048 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002049
2050 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
2051 """Error 302 -- relocated (temporarily)."""
2052 self.tries += 1
2053 if self.maxtries and self.tries >= self.maxtries:
2054 if hasattr(self, "http_error_500"):
2055 meth = self.http_error_500
2056 else:
2057 meth = self.http_error_default
2058 self.tries = 0
2059 return meth(url, fp, 500,
2060 "Internal Server Error: Redirect Recursion", headers)
2061 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
2062 data)
2063 self.tries = 0
2064 return result
2065
2066 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
2067 if 'location' in headers:
2068 newurl = headers['location']
2069 elif 'uri' in headers:
2070 newurl = headers['uri']
2071 else:
2072 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002073 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07002074
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002075 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00002076 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07002077
2078 urlparts = urlparse(newurl)
2079
2080 # For security reasons, we don't allow redirection to anything other
2081 # than http, https and ftp.
2082
2083 # We are using newer HTTPError with older redirect_internal method
2084 # This older method will get deprecated in 3.3
2085
Senthil Kumaran6497aa32012-01-04 13:46:59 +08002086 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07002087 raise HTTPError(newurl, errcode,
2088 errmsg +
2089 " Redirection to url '%s' is not allowed." % newurl,
2090 headers, fp)
2091
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002092 return self.open(newurl)
2093
2094 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
2095 """Error 301 -- also relocated (permanently)."""
2096 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2097
2098 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
2099 """Error 303 -- also relocated (essentially identical to 302)."""
2100 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2101
2102 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
2103 """Error 307 -- relocated, but turn POST into error."""
2104 if data is None:
2105 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
2106 else:
2107 return self.http_error_default(url, fp, errcode, errmsg, headers)
2108
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002109 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
2110 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002111 """Error 401 -- authentication required.
2112 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002113 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002114 URLopener.http_error_default(self, url, fp,
2115 errcode, errmsg, headers)
2116 stuff = headers['www-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002117 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2118 if not match:
2119 URLopener.http_error_default(self, url, fp,
2120 errcode, errmsg, headers)
2121 scheme, realm = match.groups()
2122 if scheme.lower() != 'basic':
2123 URLopener.http_error_default(self, url, fp,
2124 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002125 if not retry:
2126 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2127 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002128 name = 'retry_' + self.type + '_basic_auth'
2129 if data is None:
2130 return getattr(self,name)(url, realm)
2131 else:
2132 return getattr(self,name)(url, realm, data)
2133
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002134 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
2135 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002136 """Error 407 -- proxy authentication required.
2137 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08002138 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002139 URLopener.http_error_default(self, url, fp,
2140 errcode, errmsg, headers)
2141 stuff = headers['proxy-authenticate']
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002142 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2143 if not match:
2144 URLopener.http_error_default(self, url, fp,
2145 errcode, errmsg, headers)
2146 scheme, realm = match.groups()
2147 if scheme.lower() != 'basic':
2148 URLopener.http_error_default(self, url, fp,
2149 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002150 if not retry:
2151 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2152 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002153 name = 'retry_proxy_' + self.type + '_basic_auth'
2154 if data is None:
2155 return getattr(self,name)(url, realm)
2156 else:
2157 return getattr(self,name)(url, realm, data)
2158
2159 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002160 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002161 newurl = 'http://' + host + selector
2162 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002163 urltype, proxyhost = splittype(proxy)
2164 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002165 i = proxyhost.find('@') + 1
2166 proxyhost = proxyhost[i:]
2167 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2168 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002169 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002170 quote(passwd, safe=''), proxyhost)
2171 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2172 if data is None:
2173 return self.open(newurl)
2174 else:
2175 return self.open(newurl, data)
2176
2177 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002178 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002179 newurl = 'https://' + host + selector
2180 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002181 urltype, proxyhost = splittype(proxy)
2182 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002183 i = proxyhost.find('@') + 1
2184 proxyhost = proxyhost[i:]
2185 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2186 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002187 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002188 quote(passwd, safe=''), proxyhost)
2189 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2190 if data is None:
2191 return self.open(newurl)
2192 else:
2193 return self.open(newurl, data)
2194
2195 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002196 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002197 i = host.find('@') + 1
2198 host = host[i:]
2199 user, passwd = self.get_user_passwd(host, realm, i)
2200 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002201 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002202 quote(passwd, safe=''), host)
2203 newurl = 'http://' + host + selector
2204 if data is None:
2205 return self.open(newurl)
2206 else:
2207 return self.open(newurl, data)
2208
2209 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002210 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002211 i = host.find('@') + 1
2212 host = host[i:]
2213 user, passwd = self.get_user_passwd(host, realm, i)
2214 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002215 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002216 quote(passwd, safe=''), host)
2217 newurl = 'https://' + host + selector
2218 if data is None:
2219 return self.open(newurl)
2220 else:
2221 return self.open(newurl, data)
2222
Florent Xicluna757445b2010-05-17 17:24:07 +00002223 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002224 key = realm + '@' + host.lower()
2225 if key in self.auth_cache:
2226 if clear_cache:
2227 del self.auth_cache[key]
2228 else:
2229 return self.auth_cache[key]
2230 user, passwd = self.prompt_user_passwd(host, realm)
2231 if user or passwd: self.auth_cache[key] = (user, passwd)
2232 return user, passwd
2233
2234 def prompt_user_passwd(self, host, realm):
2235 """Override this in a GUI environment!"""
2236 import getpass
2237 try:
2238 user = input("Enter username for %s at %s: " % (realm, host))
2239 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2240 (user, realm, host))
2241 return user, passwd
2242 except KeyboardInterrupt:
2243 print()
2244 return None, None
2245
2246
2247# Utility functions
2248
2249_localhost = None
2250def localhost():
2251 """Return the IP address of the magic hostname 'localhost'."""
2252 global _localhost
2253 if _localhost is None:
2254 _localhost = socket.gethostbyname('localhost')
2255 return _localhost
2256
2257_thishost = None
2258def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002259 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002260 global _thishost
2261 if _thishost is None:
Senthil Kumaran1b7da512011-10-06 00:32:02 +08002262 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002263 return _thishost
2264
2265_ftperrors = None
2266def ftperrors():
2267 """Return the set of errors raised by the FTP class."""
2268 global _ftperrors
2269 if _ftperrors is None:
2270 import ftplib
2271 _ftperrors = ftplib.all_errors
2272 return _ftperrors
2273
2274_noheaders = None
2275def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002276 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002277 global _noheaders
2278 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002279 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002280 return _noheaders
2281
2282
2283# Utility classes
2284
2285class ftpwrapper:
2286 """Class used by open_ftp() for cache of open FTP connections."""
2287
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002288 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2289 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002290 self.user = user
2291 self.passwd = passwd
2292 self.host = host
2293 self.port = port
2294 self.dirs = dirs
2295 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002296 self.refcount = 0
2297 self.keepalive = persistent
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002298 self.init()
2299
2300 def init(self):
2301 import ftplib
2302 self.busy = 0
2303 self.ftp = ftplib.FTP()
2304 self.ftp.connect(self.host, self.port, self.timeout)
2305 self.ftp.login(self.user, self.passwd)
2306 for dir in self.dirs:
2307 self.ftp.cwd(dir)
2308
2309 def retrfile(self, file, type):
2310 import ftplib
2311 self.endtransfer()
2312 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2313 else: cmd = 'TYPE ' + type; isdir = 0
2314 try:
2315 self.ftp.voidcmd(cmd)
2316 except ftplib.all_errors:
2317 self.init()
2318 self.ftp.voidcmd(cmd)
2319 conn = None
2320 if file and not isdir:
2321 # Try to retrieve as a file
2322 try:
2323 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002324 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002325 except ftplib.error_perm as reason:
2326 if str(reason)[:3] != '550':
Benjamin Peterson901a2782013-05-12 19:01:52 -05002327 raise URLError('ftp error: %r' % reason).with_traceback(
Georg Brandl13e89462008-07-01 19:56:00 +00002328 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002329 if not conn:
2330 # Set transfer mode to ASCII!
2331 self.ftp.voidcmd('TYPE A')
2332 # Try a directory listing. Verify that directory exists.
2333 if file:
2334 pwd = self.ftp.pwd()
2335 try:
2336 try:
2337 self.ftp.cwd(file)
2338 except ftplib.error_perm as reason:
Benjamin Peterson901a2782013-05-12 19:01:52 -05002339 raise URLError('ftp error: %r' % reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002340 finally:
2341 self.ftp.cwd(pwd)
2342 cmd = 'LIST ' + file
2343 else:
2344 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002345 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002346 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002347
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002348 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2349 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002350 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002351 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002352 return (ftpobj, retrlen)
2353
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002354 def endtransfer(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002355 self.busy = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002356
2357 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002358 self.keepalive = False
2359 if self.refcount <= 0:
2360 self.real_close()
2361
2362 def file_close(self):
2363 self.endtransfer()
2364 self.refcount -= 1
2365 if self.refcount <= 0 and not self.keepalive:
2366 self.real_close()
2367
2368 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002369 self.endtransfer()
2370 try:
2371 self.ftp.close()
2372 except ftperrors():
2373 pass
2374
2375# Proxy handling
2376def getproxies_environment():
2377 """Return a dictionary of scheme -> proxy server URL mappings.
2378
2379 Scan the environment for variables named <scheme>_proxy;
2380 this seems to be the standard convention. If you need a
2381 different way, you can pass a proxies dictionary to the
2382 [Fancy]URLopener constructor.
2383
2384 """
2385 proxies = {}
2386 for name, value in os.environ.items():
2387 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002388 if value and name[-6:] == '_proxy':
2389 proxies[name[:-6]] = value
2390 return proxies
2391
2392def proxy_bypass_environment(host):
2393 """Test if proxies should not be used for a particular host.
2394
2395 Checks the environment for a variable named no_proxy, which should
2396 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2397 """
2398 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2399 # '*' is special case for always bypass
2400 if no_proxy == '*':
2401 return 1
2402 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002403 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002404 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002405 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2406 for name in no_proxy_list:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002407 if name and (hostonly.endswith(name) or host.endswith(name)):
2408 return 1
2409 # otherwise, don't bypass
2410 return 0
2411
2412
Ronald Oussorene72e1612011-03-14 18:15:25 -04002413# This code tests an OSX specific data structure but is testable on all
2414# platforms
2415def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2416 """
2417 Return True iff this host shouldn't be accessed using a proxy
2418
2419 This function uses the MacOSX framework SystemConfiguration
2420 to fetch the proxy information.
2421
2422 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2423 { 'exclude_simple': bool,
2424 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2425 }
2426 """
Ronald Oussorene72e1612011-03-14 18:15:25 -04002427 from fnmatch import fnmatch
2428
2429 hostonly, port = splitport(host)
2430
2431 def ip2num(ipAddr):
2432 parts = ipAddr.split('.')
2433 parts = list(map(int, parts))
2434 if len(parts) != 4:
2435 parts = (parts + [0, 0, 0, 0])[:4]
2436 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2437
2438 # Check for simple host names:
2439 if '.' not in host:
2440 if proxy_settings['exclude_simple']:
2441 return True
2442
2443 hostIP = None
2444
2445 for value in proxy_settings.get('exceptions', ()):
2446 # Items in the list are strings like these: *.local, 169.254/16
2447 if not value: continue
2448
2449 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2450 if m is not None:
2451 if hostIP is None:
2452 try:
2453 hostIP = socket.gethostbyname(hostonly)
2454 hostIP = ip2num(hostIP)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002455 except OSError:
Ronald Oussorene72e1612011-03-14 18:15:25 -04002456 continue
2457
2458 base = ip2num(m.group(1))
2459 mask = m.group(2)
2460 if mask is None:
2461 mask = 8 * (m.group(1).count('.') + 1)
2462 else:
2463 mask = int(mask[1:])
2464 mask = 32 - mask
2465
2466 if (hostIP >> mask) == (base >> mask):
2467 return True
2468
2469 elif fnmatch(host, value):
2470 return True
2471
2472 return False
2473
2474
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002475if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002476 from _scproxy import _get_proxy_settings, _get_proxies
2477
2478 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002479 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002480 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002481
2482 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002483 """Return a dictionary of scheme -> proxy server URL mappings.
2484
Ronald Oussoren84151202010-04-18 20:46:11 +00002485 This function uses the MacOSX framework SystemConfiguration
2486 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002487 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002488 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002489
Ronald Oussoren84151202010-04-18 20:46:11 +00002490
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002491
2492 def proxy_bypass(host):
2493 if getproxies_environment():
2494 return proxy_bypass_environment(host)
2495 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002496 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002497
2498 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002499 return getproxies_environment() or getproxies_macosx_sysconf()
2500
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002501
2502elif os.name == 'nt':
2503 def getproxies_registry():
2504 """Return a dictionary of scheme -> proxy server URL mappings.
2505
2506 Win32 uses the registry to store proxies.
2507
2508 """
2509 proxies = {}
2510 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002511 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002512 except ImportError:
2513 # Std module, so should be around - but you never know!
2514 return proxies
2515 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002516 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002517 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002518 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002519 'ProxyEnable')[0]
2520 if proxyEnable:
2521 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002522 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002523 'ProxyServer')[0])
2524 if '=' in proxyServer:
2525 # Per-protocol settings
2526 for p in proxyServer.split(';'):
2527 protocol, address = p.split('=', 1)
2528 # See if address has a type:// prefix
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002529 if not re.match('^([^/:]+)://', address):
2530 address = '%s://%s' % (protocol, address)
2531 proxies[protocol] = address
2532 else:
2533 # Use one setting for all protocols
2534 if proxyServer[:5] == 'http:':
2535 proxies['http'] = proxyServer
2536 else:
2537 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002538 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002539 proxies['ftp'] = 'ftp://%s' % proxyServer
2540 internetSettings.Close()
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002541 except (OSError, ValueError, TypeError):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002542 # Either registry key not found etc, or the value in an
2543 # unexpected format.
2544 # proxies already set up to be empty so nothing to do
2545 pass
2546 return proxies
2547
2548 def getproxies():
2549 """Return a dictionary of scheme -> proxy server URL mappings.
2550
2551 Returns settings gathered from the environment, if specified,
2552 or the registry.
2553
2554 """
2555 return getproxies_environment() or getproxies_registry()
2556
2557 def proxy_bypass_registry(host):
2558 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002559 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002560 except ImportError:
2561 # Std modules, so should be around - but you never know!
2562 return 0
2563 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002564 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002565 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002566 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002567 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002568 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002569 'ProxyOverride')[0])
2570 # ^^^^ Returned as Unicode but problems if not converted to ASCII
Andrew Svetlov2606a6f2012-12-19 14:33:35 +02002571 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002572 return 0
2573 if not proxyEnable or not proxyOverride:
2574 return 0
2575 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002576 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002577 host = [rawHost]
2578 try:
2579 addr = socket.gethostbyname(rawHost)
2580 if addr != rawHost:
2581 host.append(addr)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002582 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002583 pass
2584 try:
2585 fqdn = socket.getfqdn(rawHost)
2586 if fqdn != rawHost:
2587 host.append(fqdn)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002588 except OSError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002589 pass
2590 # make a check value list from the registry entry: replace the
2591 # '<local>' string by the localhost entry and the corresponding
2592 # canonical entry.
2593 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002594 # now check if we match one of the registry values.
2595 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002596 if test == '<local>':
2597 if '.' not in rawHost:
2598 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002599 test = test.replace(".", r"\.") # mask dots
2600 test = test.replace("*", r".*") # change glob sequence
2601 test = test.replace("?", r".") # change glob char
2602 for val in host:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002603 if re.match(test, val, re.I):
2604 return 1
2605 return 0
2606
2607 def proxy_bypass(host):
2608 """Return a dictionary of scheme -> proxy server URL mappings.
2609
2610 Returns settings gathered from the environment, if specified,
2611 or the registry.
2612
2613 """
2614 if getproxies_environment():
2615 return proxy_bypass_environment(host)
2616 else:
2617 return proxy_bypass_registry(host)
2618
2619else:
2620 # By default use environment variables
2621 getproxies = getproxies_environment
2622 proxy_bypass = proxy_bypass_environment