blob: cbcbe1ae373cc0535e430d69601e07bf8207ede6 [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
21IOError); for HTTP errors, raises an HTTPError, which can also be
22treated 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
92import random
93import re
94import socket
95import sys
96import time
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000097import collections
Jeremy Hylton1afc1692008-06-18 20:49:58 +000098
Georg Brandl13e89462008-07-01 19:56:00 +000099from urllib.error import URLError, HTTPError, ContentTooShortError
100from urllib.parse import (
101 urlparse, urlsplit, urljoin, unwrap, quote, unquote,
102 splittype, splithost, splitport, splituser, splitpasswd,
Senthil Kumarand95cc752010-08-08 11:27:53 +0000103 splitattr, splitquery, splitvalue, splittag, to_bytes, urlunparse)
Georg Brandl13e89462008-07-01 19:56:00 +0000104from urllib.response import addinfourl, addclosehook
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000105
106# check for SSL
107try:
108 import ssl
Senthil Kumaranc2958622010-11-22 04:48:26 +0000109except ImportError:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000110 _have_ssl = False
111else:
112 _have_ssl = True
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000113
114# used in User-Agent header sent
115__version__ = sys.version[:3]
116
117_opener = None
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000118def urlopen(url, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
119 *, cafile=None, capath=None):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000120 global _opener
Antoine Pitrou803e6d62010-10-13 10:36:15 +0000121 if cafile or capath:
122 if not _have_ssl:
123 raise ValueError('SSL support not available')
124 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
125 context.options |= ssl.OP_NO_SSLv2
126 if cafile or capath:
127 context.verify_mode = ssl.CERT_REQUIRED
128 context.load_verify_locations(cafile, capath)
129 check_hostname = True
130 else:
131 check_hostname = False
132 https_handler = HTTPSHandler(context=context, check_hostname=check_hostname)
133 opener = build_opener(https_handler)
134 elif _opener is None:
135 _opener = opener = build_opener()
136 else:
137 opener = _opener
138 return opener.open(url, data, timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000139
140def install_opener(opener):
141 global _opener
142 _opener = opener
143
144# TODO(jhylton): Make this work with the same global opener.
145_urlopener = None
146def urlretrieve(url, filename=None, reporthook=None, data=None):
147 global _urlopener
148 if not _urlopener:
149 _urlopener = FancyURLopener()
150 return _urlopener.retrieve(url, filename, reporthook, data)
151
152def urlcleanup():
153 if _urlopener:
154 _urlopener.cleanup()
155 global _opener
156 if _opener:
157 _opener = None
158
159# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000160_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000161def request_host(request):
162 """Return request-host, as defined by RFC 2965.
163
164 Variation from RFC: returned value is lowercased, for convenient
165 comparison.
166
167 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000168 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000169 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000170 if host == "":
171 host = request.get_header("Host", "")
172
173 # remove port, if present
174 host = _cut_port_re.sub("", host, 1)
175 return host.lower()
176
177class Request:
178
179 def __init__(self, url, data=None, headers={},
180 origin_req_host=None, unverifiable=False):
181 # unwrap('<URL:type://host/path>') --> 'type://host/path'
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000182 self.full_url = unwrap(url)
Senthil Kumaran26430412011-04-13 07:01:19 +0800183 self.full_url, self.fragment = splittag(self.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000184 self.data = data
185 self.headers = {}
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000186 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000187 for key, value in headers.items():
188 self.add_header(key, value)
189 self.unredirected_hdrs = {}
190 if origin_req_host is None:
191 origin_req_host = request_host(self)
192 self.origin_req_host = origin_req_host
193 self.unverifiable = unverifiable
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000194 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000195
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000196 def _parse(self):
197 self.type, rest = splittype(self.full_url)
198 if self.type is None:
199 raise ValueError("unknown url type: %s" % self.full_url)
200 self.host, self.selector = splithost(rest)
201 if self.host:
202 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000203
204 def get_method(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000205 if self.data is not None:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000206 return "POST"
207 else:
208 return "GET"
209
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000210 # Begin deprecated methods
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000211
212 def add_data(self, data):
213 self.data = data
214
215 def has_data(self):
216 return self.data is not None
217
218 def get_data(self):
219 return self.data
220
221 def get_full_url(self):
Senthil Kumaran26430412011-04-13 07:01:19 +0800222 if self.fragment:
223 return '%s#%s' % (self.full_url, self.fragment)
224 else:
225 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000226
227 def get_type(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000228 return self.type
229
230 def get_host(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000231 return self.host
232
233 def get_selector(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000234 return self.selector
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000235
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000236 def is_unverifiable(self):
237 return self.unverifiable
Facundo Batista72dc1ea2008-08-16 14:44:32 +0000238
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000239 def get_origin_req_host(self):
240 return self.origin_req_host
241
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000242 # End deprecated methods
243
244 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000245 if self.type == 'https' and not self._tunnel_host:
246 self._tunnel_host = self.host
247 else:
248 self.type= type
249 self.selector = self.full_url
250 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000251
252 def has_proxy(self):
253 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000254
255 def add_header(self, key, val):
256 # useful for something like authentication
257 self.headers[key.capitalize()] = val
258
259 def add_unredirected_header(self, key, val):
260 # will not be added to a redirected request
261 self.unredirected_hdrs[key.capitalize()] = val
262
263 def has_header(self, header_name):
264 return (header_name in self.headers or
265 header_name in self.unredirected_hdrs)
266
267 def get_header(self, header_name, default=None):
268 return self.headers.get(
269 header_name,
270 self.unredirected_hdrs.get(header_name, default))
271
272 def header_items(self):
273 hdrs = self.unredirected_hdrs.copy()
274 hdrs.update(self.headers)
275 return list(hdrs.items())
276
277class OpenerDirector:
278 def __init__(self):
279 client_version = "Python-urllib/%s" % __version__
280 self.addheaders = [('User-agent', client_version)]
R. David Murray25b8cca2010-12-23 19:44:49 +0000281 # self.handlers is retained only for backward compatibility
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000282 self.handlers = []
R. David Murray25b8cca2010-12-23 19:44:49 +0000283 # manage the individual handlers
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000284 self.handle_open = {}
285 self.handle_error = {}
286 self.process_response = {}
287 self.process_request = {}
288
289 def add_handler(self, handler):
290 if not hasattr(handler, "add_parent"):
291 raise TypeError("expected BaseHandler instance, got %r" %
292 type(handler))
293
294 added = False
295 for meth in dir(handler):
296 if meth in ["redirect_request", "do_open", "proxy_open"]:
297 # oops, coincidental match
298 continue
299
300 i = meth.find("_")
301 protocol = meth[:i]
302 condition = meth[i+1:]
303
304 if condition.startswith("error"):
305 j = condition.find("_") + i + 1
306 kind = meth[j+1:]
307 try:
308 kind = int(kind)
309 except ValueError:
310 pass
311 lookup = self.handle_error.get(protocol, {})
312 self.handle_error[protocol] = lookup
313 elif condition == "open":
314 kind = protocol
315 lookup = self.handle_open
316 elif condition == "response":
317 kind = protocol
318 lookup = self.process_response
319 elif condition == "request":
320 kind = protocol
321 lookup = self.process_request
322 else:
323 continue
324
325 handlers = lookup.setdefault(kind, [])
326 if handlers:
327 bisect.insort(handlers, handler)
328 else:
329 handlers.append(handler)
330 added = True
331
332 if added:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000333 bisect.insort(self.handlers, handler)
334 handler.add_parent(self)
335
336 def close(self):
337 # Only exists for backwards compatibility.
338 pass
339
340 def _call_chain(self, chain, kind, meth_name, *args):
341 # Handlers raise an exception if no one else should try to handle
342 # the request, or return None if they can't but another handler
343 # could. Otherwise, they return the response.
344 handlers = chain.get(kind, ())
345 for handler in handlers:
346 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000347 result = func(*args)
348 if result is not None:
349 return result
350
351 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
352 # accept a URL or a Request object
353 if isinstance(fullurl, str):
354 req = Request(fullurl, data)
355 else:
356 req = fullurl
357 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000358 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000359
360 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000361 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000362
363 # pre-process request
364 meth_name = protocol+"_request"
365 for processor in self.process_request.get(protocol, []):
366 meth = getattr(processor, meth_name)
367 req = meth(req)
368
369 response = self._open(req, data)
370
371 # post-process response
372 meth_name = protocol+"_response"
373 for processor in self.process_response.get(protocol, []):
374 meth = getattr(processor, meth_name)
375 response = meth(req, response)
376
377 return response
378
379 def _open(self, req, data=None):
380 result = self._call_chain(self.handle_open, 'default',
381 'default_open', req)
382 if result:
383 return result
384
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000385 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000386 result = self._call_chain(self.handle_open, protocol, protocol +
387 '_open', req)
388 if result:
389 return result
390
391 return self._call_chain(self.handle_open, 'unknown',
392 'unknown_open', req)
393
394 def error(self, proto, *args):
395 if proto in ('http', 'https'):
396 # XXX http[s] protocols are special-cased
397 dict = self.handle_error['http'] # https is not different than http
398 proto = args[2] # YUCK!
399 meth_name = 'http_error_%s' % proto
400 http_err = 1
401 orig_args = args
402 else:
403 dict = self.handle_error
404 meth_name = proto + '_error'
405 http_err = 0
406 args = (dict, proto, meth_name) + args
407 result = self._call_chain(*args)
408 if result:
409 return result
410
411 if http_err:
412 args = (dict, 'default', 'http_error_default') + orig_args
413 return self._call_chain(*args)
414
415# XXX probably also want an abstract factory that knows when it makes
416# sense to skip a superclass in favor of a subclass and when it might
417# make sense to include both
418
419def build_opener(*handlers):
420 """Create an opener object from a list of handlers.
421
422 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000423 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000424
425 If any of the handlers passed as arguments are subclasses of the
426 default handlers, the default handlers will not be used.
427 """
428 def isclass(obj):
429 return isinstance(obj, type) or hasattr(obj, "__bases__")
430
431 opener = OpenerDirector()
432 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
433 HTTPDefaultErrorHandler, HTTPRedirectHandler,
434 FTPHandler, FileHandler, HTTPErrorProcessor]
435 if hasattr(http.client, "HTTPSConnection"):
436 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000437 skip = set()
438 for klass in default_classes:
439 for check in handlers:
440 if isclass(check):
441 if issubclass(check, klass):
442 skip.add(klass)
443 elif isinstance(check, klass):
444 skip.add(klass)
445 for klass in skip:
446 default_classes.remove(klass)
447
448 for klass in default_classes:
449 opener.add_handler(klass())
450
451 for h in handlers:
452 if isclass(h):
453 h = h()
454 opener.add_handler(h)
455 return opener
456
457class BaseHandler:
458 handler_order = 500
459
460 def add_parent(self, parent):
461 self.parent = parent
462
463 def close(self):
464 # Only exists for backwards compatibility
465 pass
466
467 def __lt__(self, other):
468 if not hasattr(other, "handler_order"):
469 # Try to preserve the old behavior of having custom classes
470 # inserted after default ones (works only for custom user
471 # classes which are not aware of handler_order).
472 return True
473 return self.handler_order < other.handler_order
474
475
476class HTTPErrorProcessor(BaseHandler):
477 """Process HTTP error responses."""
478 handler_order = 1000 # after all other processing
479
480 def http_response(self, request, response):
481 code, msg, hdrs = response.code, response.msg, response.info()
482
483 # According to RFC 2616, "2xx" code indicates that the client's
484 # request was successfully received, understood, and accepted.
485 if not (200 <= code < 300):
486 response = self.parent.error(
487 'http', request, response, code, msg, hdrs)
488
489 return response
490
491 https_response = http_response
492
493class HTTPDefaultErrorHandler(BaseHandler):
494 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000495 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000496
497class HTTPRedirectHandler(BaseHandler):
498 # maximum number of redirections to any single URL
499 # this is needed because of the state that cookies introduce
500 max_repeats = 4
501 # maximum total number of redirections (regardless of URL) before
502 # assuming we're in a loop
503 max_redirections = 10
504
505 def redirect_request(self, req, fp, code, msg, headers, newurl):
506 """Return a Request or None in response to a redirect.
507
508 This is called by the http_error_30x methods when a
509 redirection response is received. If a redirection should
510 take place, return a new Request to allow http_error_30x to
511 perform the redirect. Otherwise, raise HTTPError if no-one
512 else should try to handle this url. Return None if you can't
513 but another Handler might.
514 """
515 m = req.get_method()
516 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
517 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000518 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000519
520 # Strictly (according to RFC 2616), 301 or 302 in response to
521 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000522 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000523 # essentially all clients do redirect in this case, so we do
524 # the same.
525 # be conciliant with URIs containing a space
526 newurl = newurl.replace(' ', '%20')
527 CONTENT_HEADERS = ("content-length", "content-type")
528 newheaders = dict((k, v) for k, v in req.headers.items()
529 if k.lower() not in CONTENT_HEADERS)
530 return Request(newurl,
531 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000532 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000533 unverifiable=True)
534
535 # Implementation note: To avoid the server sending us into an
536 # infinite loop, the request object needs to track what URLs we
537 # have already seen. Do this by adding a handler-specific
538 # attribute to the Request object.
539 def http_error_302(self, req, fp, code, msg, headers):
540 # Some servers (incorrectly) return multiple Location headers
541 # (so probably same goes for URI). Use first header.
542 if "location" in headers:
543 newurl = headers["location"]
544 elif "uri" in headers:
545 newurl = headers["uri"]
546 else:
547 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000548
549 # fix a possible malformed URL
550 urlparts = urlparse(newurl)
guido@google.coma119df92011-03-29 11:41:02 -0700551
552 # For security reasons we don't allow redirection to anything other
553 # than http, https or ftp.
554
Senthil Kumaran6497aa32012-01-04 13:46:59 +0800555 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800556 raise HTTPError(
557 newurl, code,
558 "%s - Redirection to url '%s' is not allowed" % (msg, newurl),
559 headers, fp)
guido@google.coma119df92011-03-29 11:41:02 -0700560
Facundo Batistaf24802c2008-08-17 03:36:03 +0000561 if not urlparts.path:
562 urlparts = list(urlparts)
563 urlparts[2] = "/"
564 newurl = urlunparse(urlparts)
565
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000566 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000567
568 # XXX Probably want to forget about the state of the current
569 # request, although that might interact poorly with other
570 # handlers that also use handler-specific request attributes
571 new = self.redirect_request(req, fp, code, msg, headers, newurl)
572 if new is None:
573 return
574
575 # loop detection
576 # .redirect_dict has a key url if url was previously visited.
577 if hasattr(req, 'redirect_dict'):
578 visited = new.redirect_dict = req.redirect_dict
579 if (visited.get(newurl, 0) >= self.max_repeats or
580 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000581 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000582 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000583 else:
584 visited = new.redirect_dict = req.redirect_dict = {}
585 visited[newurl] = visited.get(newurl, 0) + 1
586
587 # Don't close the fp until we are sure that we won't use it
588 # with HTTPError.
589 fp.read()
590 fp.close()
591
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000592 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000593
594 http_error_301 = http_error_303 = http_error_307 = http_error_302
595
596 inf_msg = "The HTTP server returned a redirect error that would " \
597 "lead to an infinite loop.\n" \
598 "The last 30x error message was:\n"
599
600
601def _parse_proxy(proxy):
602 """Return (scheme, user, password, host/port) given a URL or an authority.
603
604 If a URL is supplied, it must have an authority (host:port) component.
605 According to RFC 3986, having an authority component means the URL must
606 have two slashes after the scheme:
607
608 >>> _parse_proxy('file:/ftp.example.com/')
609 Traceback (most recent call last):
610 ValueError: proxy URL with no authority: 'file:/ftp.example.com/'
611
612 The first three items of the returned tuple may be None.
613
614 Examples of authority parsing:
615
616 >>> _parse_proxy('proxy.example.com')
617 (None, None, None, 'proxy.example.com')
618 >>> _parse_proxy('proxy.example.com:3128')
619 (None, None, None, 'proxy.example.com:3128')
620
621 The authority component may optionally include userinfo (assumed to be
622 username:password):
623
624 >>> _parse_proxy('joe:password@proxy.example.com')
625 (None, 'joe', 'password', 'proxy.example.com')
626 >>> _parse_proxy('joe:password@proxy.example.com:3128')
627 (None, 'joe', 'password', 'proxy.example.com:3128')
628
629 Same examples, but with URLs instead:
630
631 >>> _parse_proxy('http://proxy.example.com/')
632 ('http', None, None, 'proxy.example.com')
633 >>> _parse_proxy('http://proxy.example.com:3128/')
634 ('http', None, None, 'proxy.example.com:3128')
635 >>> _parse_proxy('http://joe:password@proxy.example.com/')
636 ('http', 'joe', 'password', 'proxy.example.com')
637 >>> _parse_proxy('http://joe:password@proxy.example.com:3128')
638 ('http', 'joe', 'password', 'proxy.example.com:3128')
639
640 Everything after the authority is ignored:
641
642 >>> _parse_proxy('ftp://joe:password@proxy.example.com/rubbish:3128')
643 ('ftp', 'joe', 'password', 'proxy.example.com')
644
645 Test for no trailing '/' case:
646
647 >>> _parse_proxy('http://joe:password@proxy.example.com')
648 ('http', 'joe', 'password', 'proxy.example.com')
649
650 """
Georg Brandl13e89462008-07-01 19:56:00 +0000651 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000652 if not r_scheme.startswith("/"):
653 # authority
654 scheme = None
655 authority = proxy
656 else:
657 # URL
658 if not r_scheme.startswith("//"):
659 raise ValueError("proxy URL with no authority: %r" % proxy)
660 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
661 # and 3.3.), path is empty or starts with '/'
662 end = r_scheme.find("/", 2)
663 if end == -1:
664 end = None
665 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000666 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000667 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000668 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000669 else:
670 user = password = None
671 return scheme, user, password, hostport
672
673class ProxyHandler(BaseHandler):
674 # Proxies must be in front
675 handler_order = 100
676
677 def __init__(self, proxies=None):
678 if proxies is None:
679 proxies = getproxies()
680 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
681 self.proxies = proxies
682 for type, url in proxies.items():
683 setattr(self, '%s_open' % type,
684 lambda r, proxy=url, type=type, meth=self.proxy_open: \
685 meth(r, proxy, type))
686
687 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000688 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000689 proxy_type, user, password, hostport = _parse_proxy(proxy)
690 if proxy_type is None:
691 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000692
693 if req.host and proxy_bypass(req.host):
694 return None
695
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000696 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000697 user_pass = '%s:%s' % (unquote(user),
698 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000699 creds = base64.b64encode(user_pass.encode()).decode("ascii")
700 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000701 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000702 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000703 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000704 # let other handlers take care of it
705 return None
706 else:
707 # need to start over, because the other handlers don't
708 # grok the proxy's URL type
709 # e.g. if we have a constructor arg proxies like so:
710 # {'http': 'ftp://proxy.example.com'}, we may end up turning
711 # a request for http://acme.example.com/a into one for
712 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000713 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000714
715class HTTPPasswordMgr:
716
717 def __init__(self):
718 self.passwd = {}
719
720 def add_password(self, realm, uri, user, passwd):
721 # uri could be a single URI or a sequence
722 if isinstance(uri, str):
723 uri = [uri]
Senthil Kumaran34d38dc2011-10-20 02:48:01 +0800724 if realm not in self.passwd:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000725 self.passwd[realm] = {}
726 for default_port in True, False:
727 reduced_uri = tuple(
728 [self.reduce_uri(u, default_port) for u in uri])
729 self.passwd[realm][reduced_uri] = (user, passwd)
730
731 def find_user_password(self, realm, authuri):
732 domains = self.passwd.get(realm, {})
733 for default_port in True, False:
734 reduced_authuri = self.reduce_uri(authuri, default_port)
735 for uris, authinfo in domains.items():
736 for uri in uris:
737 if self.is_suburi(uri, reduced_authuri):
738 return authinfo
739 return None, None
740
741 def reduce_uri(self, uri, default_port=True):
742 """Accept authority or URI and extract only the authority and path."""
743 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000744 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000745 if parts[1]:
746 # URI
747 scheme = parts[0]
748 authority = parts[1]
749 path = parts[2] or '/'
750 else:
751 # host or host:port
752 scheme = None
753 authority = uri
754 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000755 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000756 if default_port and port is None and scheme is not None:
757 dport = {"http": 80,
758 "https": 443,
759 }.get(scheme)
760 if dport is not None:
761 authority = "%s:%d" % (host, dport)
762 return authority, path
763
764 def is_suburi(self, base, test):
765 """Check if test is below base in a URI tree
766
767 Both args must be URIs in reduced form.
768 """
769 if base == test:
770 return True
771 if base[0] != test[0]:
772 return False
773 common = posixpath.commonprefix((base[1], test[1]))
774 if len(common) == len(base[1]):
775 return True
776 return False
777
778
779class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
780
781 def find_user_password(self, realm, authuri):
782 user, password = HTTPPasswordMgr.find_user_password(self, realm,
783 authuri)
784 if user is not None:
785 return user, password
786 return HTTPPasswordMgr.find_user_password(self, None, authuri)
787
788
789class AbstractBasicAuthHandler:
790
791 # XXX this allows for multiple auth-schemes, but will stupidly pick
792 # the last one with a realm specified.
793
794 # allow for double- and single-quoted realm values
795 # (single quotes are a violation of the RFC, but appear in the wild)
796 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
Senthil Kumaran34f3fcc2012-05-15 22:30:25 +0800797 'realm=(["\']?)([^"\']*)\\2', re.I)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000798
799 # XXX could pre-emptively send auth info already accepted (RFC 2617,
800 # end of section 2, and section 1.2 immediately after "credentials"
801 # production).
802
803 def __init__(self, password_mgr=None):
804 if password_mgr is None:
805 password_mgr = HTTPPasswordMgr()
806 self.passwd = password_mgr
807 self.add_password = self.passwd.add_password
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000808 self.retried = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000809
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000810 def reset_retry_count(self):
811 self.retried = 0
812
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000813 def http_error_auth_reqed(self, authreq, host, req, headers):
814 # host may be an authority (without userinfo) or a URL with an
815 # authority
816 # XXX could be multiple headers
817 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000818
819 if self.retried > 5:
820 # retry sending the username:password 5 times before failing.
821 raise HTTPError(req.get_full_url(), 401, "basic auth failed",
822 headers, None)
823 else:
824 self.retried += 1
825
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000826 if authreq:
827 mo = AbstractBasicAuthHandler.rx.search(authreq)
828 if mo:
829 scheme, quote, realm = mo.groups()
830 if scheme.lower() == 'basic':
Senthil Kumaran4bb5c272010-08-26 06:16:22 +0000831 response = self.retry_http_basic_auth(host, req, realm)
832 if response and response.code != 401:
833 self.retried = 0
834 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000835
836 def retry_http_basic_auth(self, host, req, realm):
837 user, pw = self.passwd.find_user_password(realm, host)
838 if pw is not None:
839 raw = "%s:%s" % (user, pw)
840 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
841 if req.headers.get(self.auth_header, None) == auth:
842 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000843 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000844 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000845 else:
846 return None
847
848
849class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
850
851 auth_header = 'Authorization'
852
853 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000854 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000855 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000856 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000857 self.reset_retry_count()
858 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000859
860
861class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
862
863 auth_header = 'Proxy-authorization'
864
865 def http_error_407(self, req, fp, code, msg, headers):
866 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000867 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000868 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
869 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000870 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000871 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000872 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000873 self.reset_retry_count()
874 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000875
876
877def randombytes(n):
878 """Return n random bytes."""
879 return os.urandom(n)
880
881class AbstractDigestAuthHandler:
882 # Digest authentication is specified in RFC 2617.
883
884 # XXX The client does not inspect the Authentication-Info header
885 # in a successful response.
886
887 # XXX It should be possible to test this implementation against
888 # a mock server that just generates a static set of challenges.
889
890 # XXX qop="auth-int" supports is shaky
891
892 def __init__(self, passwd=None):
893 if passwd is None:
894 passwd = HTTPPasswordMgr()
895 self.passwd = passwd
896 self.add_password = self.passwd.add_password
897 self.retried = 0
898 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000899 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000900
901 def reset_retry_count(self):
902 self.retried = 0
903
904 def http_error_auth_reqed(self, auth_header, host, req, headers):
905 authreq = headers.get(auth_header, None)
906 if self.retried > 5:
907 # Don't fail endlessly - if we failed once, we'll probably
908 # fail a second time. Hm. Unless the Password Manager is
909 # prompting for the information. Crap. This isn't great
910 # but it's better than the current 'repeat until recursion
911 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000912 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000913 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000914 else:
915 self.retried += 1
916 if authreq:
917 scheme = authreq.split()[0]
918 if scheme.lower() == 'digest':
919 return self.retry_http_digest_auth(req, authreq)
920
921 def retry_http_digest_auth(self, req, auth):
922 token, challenge = auth.split(' ', 1)
923 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
924 auth = self.get_authorization(req, chal)
925 if auth:
926 auth_val = 'Digest %s' % auth
927 if req.headers.get(self.auth_header, None) == auth_val:
928 return None
929 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000930 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000931 return resp
932
933 def get_cnonce(self, nonce):
934 # The cnonce-value is an opaque
935 # quoted string value provided by the client and used by both client
936 # and server to avoid chosen plaintext attacks, to provide mutual
937 # authentication, and to provide some message integrity protection.
938 # This isn't a fabulous effort, but it's probably Good Enough.
939 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
940 b = s.encode("ascii") + randombytes(8)
941 dig = hashlib.sha1(b).hexdigest()
942 return dig[:16]
943
944 def get_authorization(self, req, chal):
945 try:
946 realm = chal['realm']
947 nonce = chal['nonce']
948 qop = chal.get('qop')
949 algorithm = chal.get('algorithm', 'MD5')
950 # mod_digest doesn't send an opaque, even though it isn't
951 # supposed to be optional
952 opaque = chal.get('opaque', None)
953 except KeyError:
954 return None
955
956 H, KD = self.get_algorithm_impls(algorithm)
957 if H is None:
958 return None
959
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000960 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000961 if user is None:
962 return None
963
964 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000965 if req.data is not None:
966 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000967 else:
968 entdig = None
969
970 A1 = "%s:%s:%s" % (user, realm, pw)
971 A2 = "%s:%s" % (req.get_method(),
972 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000973 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000974 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000975 if nonce == self.last_nonce:
976 self.nonce_count += 1
977 else:
978 self.nonce_count = 1
979 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000980 ncvalue = '%08x' % self.nonce_count
981 cnonce = self.get_cnonce(nonce)
982 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
983 respdig = KD(H(A1), noncebit)
984 elif qop is None:
985 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
986 else:
987 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +0000988 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000989
990 # XXX should the partial digests be encoded too?
991
992 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000993 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000994 respdig)
995 if opaque:
996 base += ', opaque="%s"' % opaque
997 if entdig:
998 base += ', digest="%s"' % entdig
999 base += ', algorithm="%s"' % algorithm
1000 if qop:
1001 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1002 return base
1003
1004 def get_algorithm_impls(self, algorithm):
1005 # lambdas assume digest modules are imported at the top level
1006 if algorithm == 'MD5':
1007 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1008 elif algorithm == 'SHA':
1009 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1010 # XXX MD5-sess
1011 KD = lambda s, d: H("%s:%s" % (s, d))
1012 return H, KD
1013
1014 def get_entity_digest(self, data, chal):
1015 # XXX not implemented yet
1016 return None
1017
1018
1019class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1020 """An authentication protocol defined by RFC 2069
1021
1022 Digest authentication improves on basic authentication because it
1023 does not transmit passwords in the clear.
1024 """
1025
1026 auth_header = 'Authorization'
1027 handler_order = 490 # before Basic auth
1028
1029 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001030 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001031 retry = self.http_error_auth_reqed('www-authenticate',
1032 host, req, headers)
1033 self.reset_retry_count()
1034 return retry
1035
1036
1037class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1038
1039 auth_header = 'Proxy-Authorization'
1040 handler_order = 490 # before Basic auth
1041
1042 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001043 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001044 retry = self.http_error_auth_reqed('proxy-authenticate',
1045 host, req, headers)
1046 self.reset_retry_count()
1047 return retry
1048
1049class AbstractHTTPHandler(BaseHandler):
1050
1051 def __init__(self, debuglevel=0):
1052 self._debuglevel = debuglevel
1053
1054 def set_http_debuglevel(self, level):
1055 self._debuglevel = level
1056
1057 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001058 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001059 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001060 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001061
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001062 if request.data is not None: # POST
1063 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001064 if isinstance(data, str):
Senthil Kumaran6b3434a2012-03-15 18:11:16 -07001065 msg = "POST data should be bytes or an iterable of bytes."\
1066 "It cannot be str"
1067 raise TypeError(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001068 if not request.has_header('Content-type'):
1069 request.add_unredirected_header(
1070 'Content-type',
1071 'application/x-www-form-urlencoded')
1072 if not request.has_header('Content-length'):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001073 try:
1074 mv = memoryview(data)
1075 except TypeError:
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001076 if isinstance(data, collections.Iterable):
Georg Brandl61536042011-02-03 07:46:41 +00001077 raise ValueError("Content-Length should be specified "
1078 "for iterable data of type %r %r" % (type(data),
Senthil Kumaran7bc0d872010-12-19 10:49:52 +00001079 data))
1080 else:
1081 request.add_unredirected_header(
Senthil Kumaran1e991f22010-12-24 04:03:59 +00001082 'Content-length', '%d' % (len(mv) * mv.itemsize))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001083
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001084 sel_host = host
1085 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001086 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001087 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001088 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001089 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001090 for name, value in self.parent.addheaders:
1091 name = name.capitalize()
1092 if not request.has_header(name):
1093 request.add_unredirected_header(name, value)
1094
1095 return request
1096
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001097 def do_open(self, http_class, req, **http_conn_args):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001098 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001099
1100 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001101 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001102 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001103 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001104 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001105
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001106 # will parse host:port
1107 h = http_class(host, timeout=req.timeout, **http_conn_args)
Senthil Kumaran42ef4b12010-09-27 01:26:03 +00001108
1109 headers = dict(req.unredirected_hdrs)
1110 headers.update(dict((k, v) for k, v in req.headers.items()
1111 if k not in headers))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001112
1113 # TODO(jhylton): Should this be redesigned to handle
1114 # persistent connections?
1115
1116 # We want to make an HTTP/1.1 request, but the addinfourl
1117 # class isn't prepared to deal with a persistent connection.
1118 # It will try to read all remaining data from the socket,
1119 # which will block while the server waits for the next request.
1120 # So make sure the connection gets closed after the (only)
1121 # request.
1122 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001123 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001124
1125 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001126 tunnel_headers = {}
1127 proxy_auth_hdr = "Proxy-Authorization"
1128 if proxy_auth_hdr in headers:
1129 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1130 # Proxy-Authorization should not be sent to origin
1131 # server.
1132 del headers[proxy_auth_hdr]
1133 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001134
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001135 try:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001136 h.request(req.get_method(), req.selector, req.data, headers)
Senthil Kumaran1299a8f2011-07-27 08:05:58 +08001137 except socket.error as err: # timeout error
Senthil Kumaran45686b42011-07-27 09:31:03 +08001138 h.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001139 raise URLError(err)
Senthil Kumaran45686b42011-07-27 09:31:03 +08001140 else:
1141 r = h.getresponse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001142
Senthil Kumaran26430412011-04-13 07:01:19 +08001143 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001144 # This line replaces the .msg attribute of the HTTPResponse
1145 # with .headers, because urllib clients expect the response to
1146 # have the reason in .msg. It would be good to mark this
1147 # attribute is deprecated and get then to use info() or
1148 # .headers.
1149 r.msg = r.reason
1150 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001151
1152
1153class HTTPHandler(AbstractHTTPHandler):
1154
1155 def http_open(self, req):
1156 return self.do_open(http.client.HTTPConnection, req)
1157
1158 http_request = AbstractHTTPHandler.do_request_
1159
1160if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001161 import ssl
1162
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001163 class HTTPSHandler(AbstractHTTPHandler):
1164
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001165 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1166 AbstractHTTPHandler.__init__(self, debuglevel)
1167 self._context = context
1168 self._check_hostname = check_hostname
1169
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001170 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001171 return self.do_open(http.client.HTTPSConnection, req,
1172 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001173
1174 https_request = AbstractHTTPHandler.do_request_
1175
1176class HTTPCookieProcessor(BaseHandler):
1177 def __init__(self, cookiejar=None):
1178 import http.cookiejar
1179 if cookiejar is None:
1180 cookiejar = http.cookiejar.CookieJar()
1181 self.cookiejar = cookiejar
1182
1183 def http_request(self, request):
1184 self.cookiejar.add_cookie_header(request)
1185 return request
1186
1187 def http_response(self, request, response):
1188 self.cookiejar.extract_cookies(response, request)
1189 return response
1190
1191 https_request = http_request
1192 https_response = http_response
1193
1194class UnknownHandler(BaseHandler):
1195 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001196 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001197 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001198
1199def parse_keqv_list(l):
1200 """Parse list of key=value strings where keys are not duplicated."""
1201 parsed = {}
1202 for elt in l:
1203 k, v = elt.split('=', 1)
1204 if v[0] == '"' and v[-1] == '"':
1205 v = v[1:-1]
1206 parsed[k] = v
1207 return parsed
1208
1209def parse_http_list(s):
1210 """Parse lists as described by RFC 2068 Section 2.
1211
1212 In particular, parse comma-separated lists where the elements of
1213 the list may include quoted-strings. A quoted-string could
1214 contain a comma. A non-quoted string could have quotes in the
1215 middle. Neither commas nor quotes count if they are escaped.
1216 Only double-quotes count, not single-quotes.
1217 """
1218 res = []
1219 part = ''
1220
1221 escape = quote = False
1222 for cur in s:
1223 if escape:
1224 part += cur
1225 escape = False
1226 continue
1227 if quote:
1228 if cur == '\\':
1229 escape = True
1230 continue
1231 elif cur == '"':
1232 quote = False
1233 part += cur
1234 continue
1235
1236 if cur == ',':
1237 res.append(part)
1238 part = ''
1239 continue
1240
1241 if cur == '"':
1242 quote = True
1243
1244 part += cur
1245
1246 # append last part
1247 if part:
1248 res.append(part)
1249
1250 return [part.strip() for part in res]
1251
1252class FileHandler(BaseHandler):
1253 # Use local file or FTP depending on form of URL
1254 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001255 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001256 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1257 req.host != 'localhost'):
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001258 if not req.host is self.get_names():
1259 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001260 else:
1261 return self.open_local_file(req)
1262
1263 # names for the localhost
1264 names = None
1265 def get_names(self):
1266 if FileHandler.names is None:
1267 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001268 FileHandler.names = tuple(
1269 socket.gethostbyname_ex('localhost')[2] +
1270 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001271 except socket.gaierror:
1272 FileHandler.names = (socket.gethostbyname('localhost'),)
1273 return FileHandler.names
1274
1275 # not entirely sure what the rules are here
1276 def open_local_file(self, req):
1277 import email.utils
1278 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001279 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001280 filename = req.selector
1281 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001282 try:
1283 stats = os.stat(localfile)
1284 size = stats.st_size
1285 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001286 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001287 headers = email.message_from_string(
1288 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1289 (mtype or 'text/plain', size, modified))
1290 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001291 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001292 if not host or \
1293 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001294 if host:
1295 origurl = 'file://' + host + filename
1296 else:
1297 origurl = 'file://' + filename
1298 return addinfourl(open(localfile, 'rb'), headers, origurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001299 except OSError as msg:
Georg Brandl029986a2008-06-23 11:44:14 +00001300 # users shouldn't expect OSErrors coming from urlopen()
Georg Brandl13e89462008-07-01 19:56:00 +00001301 raise URLError(msg)
1302 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001303
1304def _safe_gethostbyname(host):
1305 try:
1306 return socket.gethostbyname(host)
1307 except socket.gaierror:
1308 return None
1309
1310class FTPHandler(BaseHandler):
1311 def ftp_open(self, req):
1312 import ftplib
1313 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001314 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001315 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001316 raise URLError('ftp error: no host given')
1317 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001318 if port is None:
1319 port = ftplib.FTP_PORT
1320 else:
1321 port = int(port)
1322
1323 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001324 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001325 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001326 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001327 else:
1328 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001329 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001330 user = user or ''
1331 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001332
1333 try:
1334 host = socket.gethostbyname(host)
1335 except socket.error as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001336 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001337 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001338 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001339 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001340 dirs, file = dirs[:-1], dirs[-1]
1341 if dirs and not dirs[0]:
1342 dirs = dirs[1:]
1343 try:
1344 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1345 type = file and 'I' or 'D'
1346 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001347 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001348 if attr.lower() == 'type' and \
1349 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1350 type = value.upper()
1351 fp, retrlen = fw.retrfile(file, type)
1352 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001353 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001354 if mtype:
1355 headers += "Content-type: %s\n" % mtype
1356 if retrlen is not None and retrlen >= 0:
1357 headers += "Content-length: %d\n" % retrlen
1358 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001359 return addinfourl(fp, headers, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001360 except ftplib.all_errors as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001361 exc = URLError('ftp error: %s' % msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001362 raise exc.with_traceback(sys.exc_info()[2])
1363
1364 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001365 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1366 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001367
1368class CacheFTPHandler(FTPHandler):
1369 # XXX would be nice to have pluggable cache strategies
1370 # XXX this stuff is definitely not thread safe
1371 def __init__(self):
1372 self.cache = {}
1373 self.timeout = {}
1374 self.soonest = 0
1375 self.delay = 60
1376 self.max_conns = 16
1377
1378 def setTimeout(self, t):
1379 self.delay = t
1380
1381 def setMaxConns(self, m):
1382 self.max_conns = m
1383
1384 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1385 key = user, host, port, '/'.join(dirs), timeout
1386 if key in self.cache:
1387 self.timeout[key] = time.time() + self.delay
1388 else:
1389 self.cache[key] = ftpwrapper(user, passwd, host, port,
1390 dirs, timeout)
1391 self.timeout[key] = time.time() + self.delay
1392 self.check_cache()
1393 return self.cache[key]
1394
1395 def check_cache(self):
1396 # first check for old ones
1397 t = time.time()
1398 if self.soonest <= t:
1399 for k, v in list(self.timeout.items()):
1400 if v < t:
1401 self.cache[k].close()
1402 del self.cache[k]
1403 del self.timeout[k]
1404 self.soonest = min(list(self.timeout.values()))
1405
1406 # then check the size
1407 if len(self.cache) == self.max_conns:
1408 for k, v in list(self.timeout.items()):
1409 if v == self.soonest:
1410 del self.cache[k]
1411 del self.timeout[k]
1412 break
1413 self.soonest = min(list(self.timeout.values()))
1414
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001415 def clear_cache(self):
1416 for conn in self.cache.values():
1417 conn.close()
1418 self.cache.clear()
1419 self.timeout.clear()
1420
1421
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001422# Code move from the old urllib module
1423
1424MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1425
1426# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001427if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001428 from nturl2path import url2pathname, pathname2url
1429else:
1430 def url2pathname(pathname):
1431 """OS-specific conversion from a relative URL of the 'file' scheme
1432 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001433 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001434
1435 def pathname2url(pathname):
1436 """OS-specific conversion from a file system path to a relative URL
1437 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001438 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001439
1440# This really consists of two pieces:
1441# (1) a class which handles opening of all sorts of URLs
1442# (plus assorted utilities etc.)
1443# (2) a set of functions for parsing URLs
1444# XXX Should these be separated out into different modules?
1445
1446
1447ftpcache = {}
1448class URLopener:
1449 """Class to open URLs.
1450 This is a class rather than just a subroutine because we may need
1451 more than one set of global protocol-specific options.
1452 Note -- this is a base class for those who don't want the
1453 automatic handling of errors type 302 (relocated) and 401
1454 (authorization needed)."""
1455
1456 __tempfiles = None
1457
1458 version = "Python-urllib/%s" % __version__
1459
1460 # Constructor
1461 def __init__(self, proxies=None, **x509):
1462 if proxies is None:
1463 proxies = getproxies()
1464 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1465 self.proxies = proxies
1466 self.key_file = x509.get('key_file')
1467 self.cert_file = x509.get('cert_file')
1468 self.addheaders = [('User-Agent', self.version)]
1469 self.__tempfiles = []
1470 self.__unlink = os.unlink # See cleanup()
1471 self.tempcache = None
1472 # Undocumented feature: if you assign {} to tempcache,
1473 # it is used to cache files retrieved with
1474 # self.retrieve(). This is not enabled by default
1475 # since it does not work for changing documents (and I
1476 # haven't got the logic to check expiration headers
1477 # yet).
1478 self.ftpcache = ftpcache
1479 # Undocumented feature: you can use a different
1480 # ftp cache by assigning to the .ftpcache member;
1481 # in case you want logically independent URL openers
1482 # XXX This is not threadsafe. Bah.
1483
1484 def __del__(self):
1485 self.close()
1486
1487 def close(self):
1488 self.cleanup()
1489
1490 def cleanup(self):
1491 # This code sometimes runs when the rest of this module
1492 # has already been deleted, so it can't use any globals
1493 # or import anything.
1494 if self.__tempfiles:
1495 for file in self.__tempfiles:
1496 try:
1497 self.__unlink(file)
1498 except OSError:
1499 pass
1500 del self.__tempfiles[:]
1501 if self.tempcache:
1502 self.tempcache.clear()
1503
1504 def addheader(self, *args):
1505 """Add a header to be used by the HTTP interface only
1506 e.g. u.addheader('Accept', 'sound/basic')"""
1507 self.addheaders.append(args)
1508
1509 # External interface
1510 def open(self, fullurl, data=None):
1511 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001512 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001513 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001514 if self.tempcache and fullurl in self.tempcache:
1515 filename, headers = self.tempcache[fullurl]
1516 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001517 return addinfourl(fp, headers, fullurl)
1518 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001519 if not urltype:
1520 urltype = 'file'
1521 if urltype in self.proxies:
1522 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001523 urltype, proxyhost = splittype(proxy)
1524 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001525 url = (host, fullurl) # Signal special case to open_*()
1526 else:
1527 proxy = None
1528 name = 'open_' + urltype
1529 self.type = urltype
1530 name = name.replace('-', '_')
1531 if not hasattr(self, name):
1532 if proxy:
1533 return self.open_unknown_proxy(proxy, fullurl, data)
1534 else:
1535 return self.open_unknown(fullurl, data)
1536 try:
1537 if data is None:
1538 return getattr(self, name)(url)
1539 else:
1540 return getattr(self, name)(url, data)
1541 except socket.error as msg:
1542 raise IOError('socket error', msg).with_traceback(sys.exc_info()[2])
1543
1544 def open_unknown(self, fullurl, data=None):
1545 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001546 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001547 raise IOError('url error', 'unknown url type', type)
1548
1549 def open_unknown_proxy(self, proxy, fullurl, data=None):
1550 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001551 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001552 raise IOError('url error', 'invalid proxy for %s' % type, proxy)
1553
1554 # External interface
1555 def retrieve(self, url, filename=None, reporthook=None, data=None):
1556 """retrieve(url) returns (filename, headers) for a local object
1557 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001558 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001559 if self.tempcache and url in self.tempcache:
1560 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001561 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001562 if filename is None and (not type or type == 'file'):
1563 try:
1564 fp = self.open_local_file(url1)
1565 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001566 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001567 return url2pathname(splithost(url1)[1]), hdrs
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001568 except IOError as msg:
1569 pass
1570 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001571 try:
1572 headers = fp.info()
1573 if filename:
1574 tfp = open(filename, 'wb')
1575 else:
1576 import tempfile
1577 garbage, path = splittype(url)
1578 garbage, path = splithost(path or "")
1579 path, garbage = splitquery(path or "")
1580 path, garbage = splitattr(path or "")
1581 suffix = os.path.splitext(path)[1]
1582 (fd, filename) = tempfile.mkstemp(suffix)
1583 self.__tempfiles.append(filename)
1584 tfp = os.fdopen(fd, 'wb')
1585 try:
1586 result = filename, headers
1587 if self.tempcache is not None:
1588 self.tempcache[url] = result
1589 bs = 1024*8
1590 size = -1
1591 read = 0
1592 blocknum = 0
Senthil Kumarance260142011-11-01 01:35:17 +08001593 if "content-length" in headers:
1594 size = int(headers["Content-Length"])
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001595 if reporthook:
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001596 reporthook(blocknum, bs, size)
1597 while 1:
1598 block = fp.read(bs)
1599 if not block:
1600 break
1601 read += len(block)
1602 tfp.write(block)
1603 blocknum += 1
1604 if reporthook:
1605 reporthook(blocknum, bs, size)
1606 finally:
1607 tfp.close()
1608 finally:
1609 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001610
1611 # raise exception if actual size does not match content-length header
1612 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001613 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001614 "retrieval incomplete: got only %i out of %i bytes"
1615 % (read, size), result)
1616
1617 return result
1618
1619 # Each method named open_<type> knows how to open that type of URL
1620
1621 def _open_generic_http(self, connection_factory, url, data):
1622 """Make an HTTP connection using connection_class.
1623
1624 This is an internal method that should be called from
1625 open_http() or open_https().
1626
1627 Arguments:
1628 - connection_factory should take a host name and return an
1629 HTTPConnection instance.
1630 - url is the url to retrieval or a host, relative-path pair.
1631 - data is payload for a POST request or None.
1632 """
1633
1634 user_passwd = None
1635 proxy_passwd= None
1636 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001637 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001638 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001639 user_passwd, host = splituser(host)
1640 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001641 realhost = host
1642 else:
1643 host, selector = url
1644 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001645 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001646 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001647 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001648 url = rest
1649 user_passwd = None
1650 if urltype.lower() != 'http':
1651 realhost = None
1652 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001653 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001654 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001655 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001656 if user_passwd:
1657 selector = "%s://%s%s" % (urltype, realhost, rest)
1658 if proxy_bypass(realhost):
1659 host = realhost
1660
1661 #print "proxy via http:", host, selector
1662 if not host: raise IOError('http error', 'no host given')
1663
1664 if proxy_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001665 proxy_passwd = unquote(proxy_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001666 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001667 else:
1668 proxy_auth = None
1669
1670 if user_passwd:
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001671 user_passwd = unquote(user_passwd)
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001672 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001673 else:
1674 auth = None
1675 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001676 headers = {}
1677 if proxy_auth:
1678 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1679 if auth:
1680 headers["Authorization"] = "Basic %s" % auth
1681 if realhost:
1682 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001683
1684 # Add Connection:close as we don't support persistent connections yet.
1685 # This helps in closing the socket and avoiding ResourceWarning
1686
1687 headers["Connection"] = "close"
1688
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001689 for header, value in self.addheaders:
1690 headers[header] = value
1691
1692 if data is not None:
1693 headers["Content-Type"] = "application/x-www-form-urlencoded"
1694 http_conn.request("POST", selector, data, headers)
1695 else:
1696 http_conn.request("GET", selector, headers=headers)
1697
1698 try:
1699 response = http_conn.getresponse()
1700 except http.client.BadStatusLine:
1701 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001702 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001703
1704 # According to RFC 2616, "2xx" code indicates that the client's
1705 # request was successfully received, understood, and accepted.
1706 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001707 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001708 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001709 else:
1710 return self.http_error(
1711 url, response.fp,
1712 response.status, response.reason, response.msg, data)
1713
1714 def open_http(self, url, data=None):
1715 """Use HTTP protocol."""
1716 return self._open_generic_http(http.client.HTTPConnection, url, data)
1717
1718 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1719 """Handle http errors.
1720
1721 Derived class can override this, or provide specific handlers
1722 named http_error_DDD where DDD is the 3-digit error code."""
1723 # First check if there's a specific handler for this error
1724 name = 'http_error_%d' % errcode
1725 if hasattr(self, name):
1726 method = getattr(self, name)
1727 if data is None:
1728 result = method(url, fp, errcode, errmsg, headers)
1729 else:
1730 result = method(url, fp, errcode, errmsg, headers, data)
1731 if result: return result
1732 return self.http_error_default(url, fp, errcode, errmsg, headers)
1733
1734 def http_error_default(self, url, fp, errcode, errmsg, headers):
1735 """Default error handler: close the connection and raise IOError."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001736 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001737 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001738
1739 if _have_ssl:
1740 def _https_connection(self, host):
1741 return http.client.HTTPSConnection(host,
1742 key_file=self.key_file,
1743 cert_file=self.cert_file)
1744
1745 def open_https(self, url, data=None):
1746 """Use HTTPS protocol."""
1747 return self._open_generic_http(self._https_connection, url, data)
1748
1749 def open_file(self, url):
1750 """Use local file or FTP depending on form of URL."""
1751 if not isinstance(url, str):
1752 raise URLError('file error', 'proxy support for file protocol currently not implemented')
1753 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001754 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001755 else:
1756 return self.open_local_file(url)
1757
1758 def open_local_file(self, url):
1759 """Use local file."""
1760 import mimetypes, email.utils
1761 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001762 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001763 localname = url2pathname(file)
1764 try:
1765 stats = os.stat(localname)
1766 except OSError as e:
1767 raise URLError(e.errno, e.strerror, e.filename)
1768 size = stats.st_size
1769 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1770 mtype = mimetypes.guess_type(url)[0]
1771 headers = email.message_from_string(
1772 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1773 (mtype or 'text/plain', size, modified))
1774 if not host:
1775 urlfile = file
1776 if file[:1] == '/':
1777 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001778 return addinfourl(open(localname, 'rb'), headers, urlfile)
1779 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001780 if (not port
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001781 and socket.gethostbyname(host) in (localhost() + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001782 urlfile = file
1783 if file[:1] == '/':
1784 urlfile = 'file://' + file
Senthil Kumaran3800ea92012-01-21 11:52:48 +08001785 elif file[:2] == './':
1786 raise ValueError("local file url may start with / or file:. Unknown url of type: %s" % url)
Georg Brandl13e89462008-07-01 19:56:00 +00001787 return addinfourl(open(localname, 'rb'), headers, urlfile)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001788 raise URLError('local file error', 'not on local host')
1789
1790 def open_ftp(self, url):
1791 """Use FTP protocol."""
1792 if not isinstance(url, str):
1793 raise URLError('ftp error', 'proxy support for ftp protocol currently not implemented')
1794 import mimetypes
1795 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001796 host, path = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001797 if not host: raise URLError('ftp error', 'no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001798 host, port = splitport(host)
1799 user, host = splituser(host)
1800 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001801 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001802 host = unquote(host)
1803 user = unquote(user or '')
1804 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001805 host = socket.gethostbyname(host)
1806 if not port:
1807 import ftplib
1808 port = ftplib.FTP_PORT
1809 else:
1810 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001811 path, attrs = splitattr(path)
1812 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001813 dirs = path.split('/')
1814 dirs, file = dirs[:-1], dirs[-1]
1815 if dirs and not dirs[0]: dirs = dirs[1:]
1816 if dirs and not dirs[0]: dirs[0] = '/'
1817 key = user, host, port, '/'.join(dirs)
1818 # XXX thread unsafe!
1819 if len(self.ftpcache) > MAXFTPCACHE:
1820 # Prune the cache, rather arbitrarily
1821 for k in self.ftpcache.keys():
1822 if k != key:
1823 v = self.ftpcache[k]
1824 del self.ftpcache[k]
1825 v.close()
1826 try:
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001827 if key not in self.ftpcache:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001828 self.ftpcache[key] = \
1829 ftpwrapper(user, passwd, host, port, dirs)
1830 if not file: type = 'D'
1831 else: type = 'I'
1832 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001833 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001834 if attr.lower() == 'type' and \
1835 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1836 type = value.upper()
1837 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1838 mtype = mimetypes.guess_type("ftp:" + url)[0]
1839 headers = ""
1840 if mtype:
1841 headers += "Content-Type: %s\n" % mtype
1842 if retrlen is not None and retrlen >= 0:
1843 headers += "Content-Length: %d\n" % retrlen
1844 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001845 return addinfourl(fp, headers, "ftp:" + url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001846 except ftperrors() as msg:
1847 raise URLError('ftp error', msg).with_traceback(sys.exc_info()[2])
1848
1849 def open_data(self, url, data=None):
1850 """Use "data" URL."""
1851 if not isinstance(url, str):
1852 raise URLError('data error', 'proxy support for data protocol currently not implemented')
1853 # ignore POSTed data
1854 #
1855 # syntax of data URLs:
1856 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1857 # mediatype := [ type "/" subtype ] *( ";" parameter )
1858 # data := *urlchar
1859 # parameter := attribute "=" value
1860 try:
1861 [type, data] = url.split(',', 1)
1862 except ValueError:
1863 raise IOError('data error', 'bad data URL')
1864 if not type:
1865 type = 'text/plain;charset=US-ASCII'
1866 semi = type.rfind(';')
1867 if semi >= 0 and '=' not in type[semi:]:
1868 encoding = type[semi+1:]
1869 type = type[:semi]
1870 else:
1871 encoding = ''
1872 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00001873 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001874 time.gmtime(time.time())))
1875 msg.append('Content-type: %s' % type)
1876 if encoding == 'base64':
Georg Brandl706824f2009-06-04 09:42:55 +00001877 # XXX is this encoding/decoding ok?
1878 data = base64.decodebytes(data.encode('ascii')).decode('latin1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001879 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001880 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001881 msg.append('Content-Length: %d' % len(data))
1882 msg.append('')
1883 msg.append(data)
1884 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001885 headers = email.message_from_string(msg)
1886 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001887 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001888 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001889
1890
1891class FancyURLopener(URLopener):
1892 """Derived class with handlers for errors we can handle (perhaps)."""
1893
1894 def __init__(self, *args, **kwargs):
1895 URLopener.__init__(self, *args, **kwargs)
1896 self.auth_cache = {}
1897 self.tries = 0
1898 self.maxtries = 10
1899
1900 def http_error_default(self, url, fp, errcode, errmsg, headers):
1901 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001902 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001903
1904 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1905 """Error 302 -- relocated (temporarily)."""
1906 self.tries += 1
1907 if self.maxtries and self.tries >= self.maxtries:
1908 if hasattr(self, "http_error_500"):
1909 meth = self.http_error_500
1910 else:
1911 meth = self.http_error_default
1912 self.tries = 0
1913 return meth(url, fp, 500,
1914 "Internal Server Error: Redirect Recursion", headers)
1915 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
1916 data)
1917 self.tries = 0
1918 return result
1919
1920 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
1921 if 'location' in headers:
1922 newurl = headers['location']
1923 elif 'uri' in headers:
1924 newurl = headers['uri']
1925 else:
1926 return
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001927 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07001928
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001929 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00001930 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07001931
1932 urlparts = urlparse(newurl)
1933
1934 # For security reasons, we don't allow redirection to anything other
1935 # than http, https and ftp.
1936
1937 # We are using newer HTTPError with older redirect_internal method
1938 # This older method will get deprecated in 3.3
1939
Senthil Kumaran6497aa32012-01-04 13:46:59 +08001940 if urlparts.scheme not in ('http', 'https', 'ftp', ''):
guido@google.coma119df92011-03-29 11:41:02 -07001941 raise HTTPError(newurl, errcode,
1942 errmsg +
1943 " Redirection to url '%s' is not allowed." % newurl,
1944 headers, fp)
1945
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001946 return self.open(newurl)
1947
1948 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
1949 """Error 301 -- also relocated (permanently)."""
1950 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1951
1952 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
1953 """Error 303 -- also relocated (essentially identical to 302)."""
1954 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1955
1956 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
1957 """Error 307 -- relocated, but turn POST into error."""
1958 if data is None:
1959 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1960 else:
1961 return self.http_error_default(url, fp, errcode, errmsg, headers)
1962
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001963 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
1964 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001965 """Error 401 -- authentication required.
1966 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001967 if 'www-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001968 URLopener.http_error_default(self, url, fp,
1969 errcode, errmsg, headers)
1970 stuff = headers['www-authenticate']
1971 import re
1972 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1973 if not match:
1974 URLopener.http_error_default(self, url, fp,
1975 errcode, errmsg, headers)
1976 scheme, realm = match.groups()
1977 if scheme.lower() != 'basic':
1978 URLopener.http_error_default(self, url, fp,
1979 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001980 if not retry:
1981 URLopener.http_error_default(self, url, fp, errcode, errmsg,
1982 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001983 name = 'retry_' + self.type + '_basic_auth'
1984 if data is None:
1985 return getattr(self,name)(url, realm)
1986 else:
1987 return getattr(self,name)(url, realm, data)
1988
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001989 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
1990 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001991 """Error 407 -- proxy authentication required.
1992 This function supports Basic authentication only."""
Senthil Kumaran34d38dc2011-10-20 02:48:01 +08001993 if 'proxy-authenticate' not in headers:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001994 URLopener.http_error_default(self, url, fp,
1995 errcode, errmsg, headers)
1996 stuff = headers['proxy-authenticate']
1997 import re
1998 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1999 if not match:
2000 URLopener.http_error_default(self, url, fp,
2001 errcode, errmsg, headers)
2002 scheme, realm = match.groups()
2003 if scheme.lower() != 'basic':
2004 URLopener.http_error_default(self, url, fp,
2005 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002006 if not retry:
2007 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2008 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002009 name = 'retry_proxy_' + self.type + '_basic_auth'
2010 if data is None:
2011 return getattr(self,name)(url, realm)
2012 else:
2013 return getattr(self,name)(url, realm, data)
2014
2015 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002016 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002017 newurl = 'http://' + host + selector
2018 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002019 urltype, proxyhost = splittype(proxy)
2020 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002021 i = proxyhost.find('@') + 1
2022 proxyhost = proxyhost[i:]
2023 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2024 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002025 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002026 quote(passwd, safe=''), proxyhost)
2027 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2028 if data is None:
2029 return self.open(newurl)
2030 else:
2031 return self.open(newurl, data)
2032
2033 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002034 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002035 newurl = 'https://' + host + selector
2036 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002037 urltype, proxyhost = splittype(proxy)
2038 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002039 i = proxyhost.find('@') + 1
2040 proxyhost = proxyhost[i:]
2041 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2042 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002043 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002044 quote(passwd, safe=''), proxyhost)
2045 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2046 if data is None:
2047 return self.open(newurl)
2048 else:
2049 return self.open(newurl, data)
2050
2051 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002052 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002053 i = host.find('@') + 1
2054 host = host[i:]
2055 user, passwd = self.get_user_passwd(host, realm, i)
2056 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002057 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002058 quote(passwd, safe=''), host)
2059 newurl = 'http://' + host + selector
2060 if data is None:
2061 return self.open(newurl)
2062 else:
2063 return self.open(newurl, data)
2064
2065 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002066 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002067 i = host.find('@') + 1
2068 host = host[i:]
2069 user, passwd = self.get_user_passwd(host, realm, i)
2070 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002071 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002072 quote(passwd, safe=''), host)
2073 newurl = 'https://' + host + selector
2074 if data is None:
2075 return self.open(newurl)
2076 else:
2077 return self.open(newurl, data)
2078
Florent Xicluna757445b2010-05-17 17:24:07 +00002079 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002080 key = realm + '@' + host.lower()
2081 if key in self.auth_cache:
2082 if clear_cache:
2083 del self.auth_cache[key]
2084 else:
2085 return self.auth_cache[key]
2086 user, passwd = self.prompt_user_passwd(host, realm)
2087 if user or passwd: self.auth_cache[key] = (user, passwd)
2088 return user, passwd
2089
2090 def prompt_user_passwd(self, host, realm):
2091 """Override this in a GUI environment!"""
2092 import getpass
2093 try:
2094 user = input("Enter username for %s at %s: " % (realm, host))
2095 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2096 (user, realm, host))
2097 return user, passwd
2098 except KeyboardInterrupt:
2099 print()
2100 return None, None
2101
2102
2103# Utility functions
2104
2105_localhost = None
2106def localhost():
2107 """Return the IP address of the magic hostname 'localhost'."""
2108 global _localhost
2109 if _localhost is None:
2110 _localhost = socket.gethostbyname('localhost')
2111 return _localhost
2112
2113_thishost = None
2114def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002115 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002116 global _thishost
2117 if _thishost is None:
Senthil Kumaran1b7da512011-10-06 00:32:02 +08002118 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002119 return _thishost
2120
2121_ftperrors = None
2122def ftperrors():
2123 """Return the set of errors raised by the FTP class."""
2124 global _ftperrors
2125 if _ftperrors is None:
2126 import ftplib
2127 _ftperrors = ftplib.all_errors
2128 return _ftperrors
2129
2130_noheaders = None
2131def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002132 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002133 global _noheaders
2134 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002135 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002136 return _noheaders
2137
2138
2139# Utility classes
2140
2141class ftpwrapper:
2142 """Class used by open_ftp() for cache of open FTP connections."""
2143
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002144 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2145 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002146 self.user = user
2147 self.passwd = passwd
2148 self.host = host
2149 self.port = port
2150 self.dirs = dirs
2151 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002152 self.refcount = 0
2153 self.keepalive = persistent
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002154 self.init()
2155
2156 def init(self):
2157 import ftplib
2158 self.busy = 0
2159 self.ftp = ftplib.FTP()
2160 self.ftp.connect(self.host, self.port, self.timeout)
2161 self.ftp.login(self.user, self.passwd)
2162 for dir in self.dirs:
2163 self.ftp.cwd(dir)
2164
2165 def retrfile(self, file, type):
2166 import ftplib
2167 self.endtransfer()
2168 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2169 else: cmd = 'TYPE ' + type; isdir = 0
2170 try:
2171 self.ftp.voidcmd(cmd)
2172 except ftplib.all_errors:
2173 self.init()
2174 self.ftp.voidcmd(cmd)
2175 conn = None
2176 if file and not isdir:
2177 # Try to retrieve as a file
2178 try:
2179 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002180 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002181 except ftplib.error_perm as reason:
2182 if str(reason)[:3] != '550':
Georg Brandl13e89462008-07-01 19:56:00 +00002183 raise URLError('ftp error', reason).with_traceback(
2184 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002185 if not conn:
2186 # Set transfer mode to ASCII!
2187 self.ftp.voidcmd('TYPE A')
2188 # Try a directory listing. Verify that directory exists.
2189 if file:
2190 pwd = self.ftp.pwd()
2191 try:
2192 try:
2193 self.ftp.cwd(file)
2194 except ftplib.error_perm as reason:
Georg Brandl13e89462008-07-01 19:56:00 +00002195 raise URLError('ftp error', reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002196 finally:
2197 self.ftp.cwd(pwd)
2198 cmd = 'LIST ' + file
2199 else:
2200 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002201 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002202 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002203
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002204 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2205 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002206 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002207 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002208 return (ftpobj, retrlen)
2209
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002210 def endtransfer(self):
2211 if not self.busy:
2212 return
2213 self.busy = 0
2214 try:
2215 self.ftp.voidresp()
2216 except ftperrors():
2217 pass
2218
2219 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002220 self.keepalive = False
2221 if self.refcount <= 0:
2222 self.real_close()
2223
2224 def file_close(self):
2225 self.endtransfer()
2226 self.refcount -= 1
2227 if self.refcount <= 0 and not self.keepalive:
2228 self.real_close()
2229
2230 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002231 self.endtransfer()
2232 try:
2233 self.ftp.close()
2234 except ftperrors():
2235 pass
2236
2237# Proxy handling
2238def getproxies_environment():
2239 """Return a dictionary of scheme -> proxy server URL mappings.
2240
2241 Scan the environment for variables named <scheme>_proxy;
2242 this seems to be the standard convention. If you need a
2243 different way, you can pass a proxies dictionary to the
2244 [Fancy]URLopener constructor.
2245
2246 """
2247 proxies = {}
2248 for name, value in os.environ.items():
2249 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002250 if value and name[-6:] == '_proxy':
2251 proxies[name[:-6]] = value
2252 return proxies
2253
2254def proxy_bypass_environment(host):
2255 """Test if proxies should not be used for a particular host.
2256
2257 Checks the environment for a variable named no_proxy, which should
2258 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2259 """
2260 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2261 # '*' is special case for always bypass
2262 if no_proxy == '*':
2263 return 1
2264 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002265 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002266 # check if the host ends with any of the DNS suffixes
Senthil Kumaran89976f12011-08-06 12:27:40 +08002267 no_proxy_list = [proxy.strip() for proxy in no_proxy.split(',')]
2268 for name in no_proxy_list:
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002269 if name and (hostonly.endswith(name) or host.endswith(name)):
2270 return 1
2271 # otherwise, don't bypass
2272 return 0
2273
2274
Ronald Oussorene72e1612011-03-14 18:15:25 -04002275# This code tests an OSX specific data structure but is testable on all
2276# platforms
2277def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2278 """
2279 Return True iff this host shouldn't be accessed using a proxy
2280
2281 This function uses the MacOSX framework SystemConfiguration
2282 to fetch the proxy information.
2283
2284 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2285 { 'exclude_simple': bool,
2286 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2287 }
2288 """
2289 import re
2290 import socket
2291 from fnmatch import fnmatch
2292
2293 hostonly, port = splitport(host)
2294
2295 def ip2num(ipAddr):
2296 parts = ipAddr.split('.')
2297 parts = list(map(int, parts))
2298 if len(parts) != 4:
2299 parts = (parts + [0, 0, 0, 0])[:4]
2300 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2301
2302 # Check for simple host names:
2303 if '.' not in host:
2304 if proxy_settings['exclude_simple']:
2305 return True
2306
2307 hostIP = None
2308
2309 for value in proxy_settings.get('exceptions', ()):
2310 # Items in the list are strings like these: *.local, 169.254/16
2311 if not value: continue
2312
2313 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2314 if m is not None:
2315 if hostIP is None:
2316 try:
2317 hostIP = socket.gethostbyname(hostonly)
2318 hostIP = ip2num(hostIP)
2319 except socket.error:
2320 continue
2321
2322 base = ip2num(m.group(1))
2323 mask = m.group(2)
2324 if mask is None:
2325 mask = 8 * (m.group(1).count('.') + 1)
2326 else:
2327 mask = int(mask[1:])
2328 mask = 32 - mask
2329
2330 if (hostIP >> mask) == (base >> mask):
2331 return True
2332
2333 elif fnmatch(host, value):
2334 return True
2335
2336 return False
2337
2338
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002339if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002340 from _scproxy import _get_proxy_settings, _get_proxies
2341
2342 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002343 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002344 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002345
2346 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002347 """Return a dictionary of scheme -> proxy server URL mappings.
2348
Ronald Oussoren84151202010-04-18 20:46:11 +00002349 This function uses the MacOSX framework SystemConfiguration
2350 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002351 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002352 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002353
Ronald Oussoren84151202010-04-18 20:46:11 +00002354
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002355
2356 def proxy_bypass(host):
2357 if getproxies_environment():
2358 return proxy_bypass_environment(host)
2359 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002360 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002361
2362 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002363 return getproxies_environment() or getproxies_macosx_sysconf()
2364
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002365
2366elif os.name == 'nt':
2367 def getproxies_registry():
2368 """Return a dictionary of scheme -> proxy server URL mappings.
2369
2370 Win32 uses the registry to store proxies.
2371
2372 """
2373 proxies = {}
2374 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002375 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002376 except ImportError:
2377 # Std module, so should be around - but you never know!
2378 return proxies
2379 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002380 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002381 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002382 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002383 'ProxyEnable')[0]
2384 if proxyEnable:
2385 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002386 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002387 'ProxyServer')[0])
2388 if '=' in proxyServer:
2389 # Per-protocol settings
2390 for p in proxyServer.split(';'):
2391 protocol, address = p.split('=', 1)
2392 # See if address has a type:// prefix
2393 import re
2394 if not re.match('^([^/:]+)://', address):
2395 address = '%s://%s' % (protocol, address)
2396 proxies[protocol] = address
2397 else:
2398 # Use one setting for all protocols
2399 if proxyServer[:5] == 'http:':
2400 proxies['http'] = proxyServer
2401 else:
2402 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002403 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002404 proxies['ftp'] = 'ftp://%s' % proxyServer
2405 internetSettings.Close()
2406 except (WindowsError, ValueError, TypeError):
2407 # Either registry key not found etc, or the value in an
2408 # unexpected format.
2409 # proxies already set up to be empty so nothing to do
2410 pass
2411 return proxies
2412
2413 def getproxies():
2414 """Return a dictionary of scheme -> proxy server URL mappings.
2415
2416 Returns settings gathered from the environment, if specified,
2417 or the registry.
2418
2419 """
2420 return getproxies_environment() or getproxies_registry()
2421
2422 def proxy_bypass_registry(host):
2423 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002424 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002425 import re
2426 except ImportError:
2427 # Std modules, so should be around - but you never know!
2428 return 0
2429 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002430 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002431 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002432 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002433 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002434 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002435 'ProxyOverride')[0])
2436 # ^^^^ Returned as Unicode but problems if not converted to ASCII
2437 except WindowsError:
2438 return 0
2439 if not proxyEnable or not proxyOverride:
2440 return 0
2441 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002442 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002443 host = [rawHost]
2444 try:
2445 addr = socket.gethostbyname(rawHost)
2446 if addr != rawHost:
2447 host.append(addr)
2448 except socket.error:
2449 pass
2450 try:
2451 fqdn = socket.getfqdn(rawHost)
2452 if fqdn != rawHost:
2453 host.append(fqdn)
2454 except socket.error:
2455 pass
2456 # make a check value list from the registry entry: replace the
2457 # '<local>' string by the localhost entry and the corresponding
2458 # canonical entry.
2459 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002460 # now check if we match one of the registry values.
2461 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002462 if test == '<local>':
2463 if '.' not in rawHost:
2464 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002465 test = test.replace(".", r"\.") # mask dots
2466 test = test.replace("*", r".*") # change glob sequence
2467 test = test.replace("?", r".") # change glob char
2468 for val in host:
2469 # print "%s <--> %s" %( test, val )
2470 if re.match(test, val, re.I):
2471 return 1
2472 return 0
2473
2474 def proxy_bypass(host):
2475 """Return a dictionary of scheme -> proxy server URL mappings.
2476
2477 Returns settings gathered from the environment, if specified,
2478 or the registry.
2479
2480 """
2481 if getproxies_environment():
2482 return proxy_bypass_environment(host)
2483 else:
2484 return proxy_bypass_registry(host)
2485
2486else:
2487 # By default use environment variables
2488 getproxies = getproxies_environment
2489 proxy_bypass = proxy_bypass_environment