blob: 534408d815c5bd9b5d0a18640bbc944be98bab1f [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
555 if not urlparts.scheme in ('http', 'https', 'ftp'):
556 raise HTTPError(newurl, code,
557 msg +
558 " - Redirection to url '%s' is not allowed" %
559 newurl,
560 headers, fp)
561
Facundo Batistaf24802c2008-08-17 03:36:03 +0000562 if not urlparts.path:
563 urlparts = list(urlparts)
564 urlparts[2] = "/"
565 newurl = urlunparse(urlparts)
566
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000567 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000568
569 # XXX Probably want to forget about the state of the current
570 # request, although that might interact poorly with other
571 # handlers that also use handler-specific request attributes
572 new = self.redirect_request(req, fp, code, msg, headers, newurl)
573 if new is None:
574 return
575
576 # loop detection
577 # .redirect_dict has a key url if url was previously visited.
578 if hasattr(req, 'redirect_dict'):
579 visited = new.redirect_dict = req.redirect_dict
580 if (visited.get(newurl, 0) >= self.max_repeats or
581 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000582 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000583 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000584 else:
585 visited = new.redirect_dict = req.redirect_dict = {}
586 visited[newurl] = visited.get(newurl, 0) + 1
587
588 # Don't close the fp until we are sure that we won't use it
589 # with HTTPError.
590 fp.read()
591 fp.close()
592
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000593 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000594
595 http_error_301 = http_error_303 = http_error_307 = http_error_302
596
597 inf_msg = "The HTTP server returned a redirect error that would " \
598 "lead to an infinite loop.\n" \
599 "The last 30x error message was:\n"
600
601
602def _parse_proxy(proxy):
603 """Return (scheme, user, password, host/port) given a URL or an authority.
604
605 If a URL is supplied, it must have an authority (host:port) component.
606 According to RFC 3986, having an authority component means the URL must
607 have two slashes after the scheme:
608
609 >>> _parse_proxy('file:/ftp.example.com/')
610 Traceback (most recent call last):
611 ValueError: proxy URL with no authority: 'file:/ftp.example.com/'
612
613 The first three items of the returned tuple may be None.
614
615 Examples of authority parsing:
616
617 >>> _parse_proxy('proxy.example.com')
618 (None, None, None, 'proxy.example.com')
619 >>> _parse_proxy('proxy.example.com:3128')
620 (None, None, None, 'proxy.example.com:3128')
621
622 The authority component may optionally include userinfo (assumed to be
623 username:password):
624
625 >>> _parse_proxy('joe:password@proxy.example.com')
626 (None, 'joe', 'password', 'proxy.example.com')
627 >>> _parse_proxy('joe:password@proxy.example.com:3128')
628 (None, 'joe', 'password', 'proxy.example.com:3128')
629
630 Same examples, but with URLs instead:
631
632 >>> _parse_proxy('http://proxy.example.com/')
633 ('http', None, None, 'proxy.example.com')
634 >>> _parse_proxy('http://proxy.example.com:3128/')
635 ('http', None, None, 'proxy.example.com:3128')
636 >>> _parse_proxy('http://joe:password@proxy.example.com/')
637 ('http', 'joe', 'password', 'proxy.example.com')
638 >>> _parse_proxy('http://joe:password@proxy.example.com:3128')
639 ('http', 'joe', 'password', 'proxy.example.com:3128')
640
641 Everything after the authority is ignored:
642
643 >>> _parse_proxy('ftp://joe:password@proxy.example.com/rubbish:3128')
644 ('ftp', 'joe', 'password', 'proxy.example.com')
645
646 Test for no trailing '/' case:
647
648 >>> _parse_proxy('http://joe:password@proxy.example.com')
649 ('http', 'joe', 'password', 'proxy.example.com')
650
651 """
Georg Brandl13e89462008-07-01 19:56:00 +0000652 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000653 if not r_scheme.startswith("/"):
654 # authority
655 scheme = None
656 authority = proxy
657 else:
658 # URL
659 if not r_scheme.startswith("//"):
660 raise ValueError("proxy URL with no authority: %r" % proxy)
661 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
662 # and 3.3.), path is empty or starts with '/'
663 end = r_scheme.find("/", 2)
664 if end == -1:
665 end = None
666 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000667 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000668 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000669 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000670 else:
671 user = password = None
672 return scheme, user, password, hostport
673
674class ProxyHandler(BaseHandler):
675 # Proxies must be in front
676 handler_order = 100
677
678 def __init__(self, proxies=None):
679 if proxies is None:
680 proxies = getproxies()
681 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
682 self.proxies = proxies
683 for type, url in proxies.items():
684 setattr(self, '%s_open' % type,
685 lambda r, proxy=url, type=type, meth=self.proxy_open: \
686 meth(r, proxy, type))
687
688 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000689 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000690 proxy_type, user, password, hostport = _parse_proxy(proxy)
691 if proxy_type is None:
692 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000693
694 if req.host and proxy_bypass(req.host):
695 return None
696
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000697 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000698 user_pass = '%s:%s' % (unquote(user),
699 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000700 creds = base64.b64encode(user_pass.encode()).decode("ascii")
701 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000702 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000703 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000704 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000705 # let other handlers take care of it
706 return None
707 else:
708 # need to start over, because the other handlers don't
709 # grok the proxy's URL type
710 # e.g. if we have a constructor arg proxies like so:
711 # {'http': 'ftp://proxy.example.com'}, we may end up turning
712 # a request for http://acme.example.com/a into one for
713 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000714 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000715
716class HTTPPasswordMgr:
717
718 def __init__(self):
719 self.passwd = {}
720
721 def add_password(self, realm, uri, user, passwd):
722 # uri could be a single URI or a sequence
723 if isinstance(uri, str):
724 uri = [uri]
725 if not realm in self.passwd:
726 self.passwd[realm] = {}
727 for default_port in True, False:
728 reduced_uri = tuple(
729 [self.reduce_uri(u, default_port) for u in uri])
730 self.passwd[realm][reduced_uri] = (user, passwd)
731
732 def find_user_password(self, realm, authuri):
733 domains = self.passwd.get(realm, {})
734 for default_port in True, False:
735 reduced_authuri = self.reduce_uri(authuri, default_port)
736 for uris, authinfo in domains.items():
737 for uri in uris:
738 if self.is_suburi(uri, reduced_authuri):
739 return authinfo
740 return None, None
741
742 def reduce_uri(self, uri, default_port=True):
743 """Accept authority or URI and extract only the authority and path."""
744 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000745 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000746 if parts[1]:
747 # URI
748 scheme = parts[0]
749 authority = parts[1]
750 path = parts[2] or '/'
751 else:
752 # host or host:port
753 scheme = None
754 authority = uri
755 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000756 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000757 if default_port and port is None and scheme is not None:
758 dport = {"http": 80,
759 "https": 443,
760 }.get(scheme)
761 if dport is not None:
762 authority = "%s:%d" % (host, dport)
763 return authority, path
764
765 def is_suburi(self, base, test):
766 """Check if test is below base in a URI tree
767
768 Both args must be URIs in reduced form.
769 """
770 if base == test:
771 return True
772 if base[0] != test[0]:
773 return False
774 common = posixpath.commonprefix((base[1], test[1]))
775 if len(common) == len(base[1]):
776 return True
777 return False
778
779
780class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
781
782 def find_user_password(self, realm, authuri):
783 user, password = HTTPPasswordMgr.find_user_password(self, realm,
784 authuri)
785 if user is not None:
786 return user, password
787 return HTTPPasswordMgr.find_user_password(self, None, authuri)
788
789
790class AbstractBasicAuthHandler:
791
792 # XXX this allows for multiple auth-schemes, but will stupidly pick
793 # the last one with a realm specified.
794
795 # allow for double- and single-quoted realm values
796 # (single quotes are a violation of the RFC, but appear in the wild)
797 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
798 'realm=(["\'])(.*?)\\2', re.I)
799
800 # XXX could pre-emptively send auth info already accepted (RFC 2617,
801 # end of section 2, and section 1.2 immediately after "credentials"
802 # production).
803
804 def __init__(self, password_mgr=None):
805 if password_mgr is None:
806 password_mgr = HTTPPasswordMgr()
807 self.passwd = password_mgr
808 self.add_password = self.passwd.add_password
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000809 self.retried = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000810
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000811 def reset_retry_count(self):
812 self.retried = 0
813
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000814 def http_error_auth_reqed(self, authreq, host, req, headers):
815 # host may be an authority (without userinfo) or a URL with an
816 # authority
817 # XXX could be multiple headers
818 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000819
820 if self.retried > 5:
821 # retry sending the username:password 5 times before failing.
822 raise HTTPError(req.get_full_url(), 401, "basic auth failed",
823 headers, None)
824 else:
825 self.retried += 1
826
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000827 if authreq:
828 mo = AbstractBasicAuthHandler.rx.search(authreq)
829 if mo:
830 scheme, quote, realm = mo.groups()
831 if scheme.lower() == 'basic':
Senthil Kumaran4bb5c272010-08-26 06:16:22 +0000832 response = self.retry_http_basic_auth(host, req, realm)
833 if response and response.code != 401:
834 self.retried = 0
835 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000836
837 def retry_http_basic_auth(self, host, req, realm):
838 user, pw = self.passwd.find_user_password(realm, host)
839 if pw is not None:
840 raw = "%s:%s" % (user, pw)
841 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
842 if req.headers.get(self.auth_header, None) == auth:
843 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000844 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000845 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000846 else:
847 return None
848
849
850class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
851
852 auth_header = 'Authorization'
853
854 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000855 url = req.full_url
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000856 response = self.http_error_auth_reqed('www-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000857 url, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000858 self.reset_retry_count()
859 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000860
861
862class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
863
864 auth_header = 'Proxy-authorization'
865
866 def http_error_407(self, req, fp, code, msg, headers):
867 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000868 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000869 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
870 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000871 authority = req.host
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000872 response = self.http_error_auth_reqed('proxy-authenticate',
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000873 authority, req, headers)
Senthil Kumaran67a62a42010-08-19 17:50:31 +0000874 self.reset_retry_count()
875 return response
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000876
877
878def randombytes(n):
879 """Return n random bytes."""
880 return os.urandom(n)
881
882class AbstractDigestAuthHandler:
883 # Digest authentication is specified in RFC 2617.
884
885 # XXX The client does not inspect the Authentication-Info header
886 # in a successful response.
887
888 # XXX It should be possible to test this implementation against
889 # a mock server that just generates a static set of challenges.
890
891 # XXX qop="auth-int" supports is shaky
892
893 def __init__(self, passwd=None):
894 if passwd is None:
895 passwd = HTTPPasswordMgr()
896 self.passwd = passwd
897 self.add_password = self.passwd.add_password
898 self.retried = 0
899 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000900 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000901
902 def reset_retry_count(self):
903 self.retried = 0
904
905 def http_error_auth_reqed(self, auth_header, host, req, headers):
906 authreq = headers.get(auth_header, None)
907 if self.retried > 5:
908 # Don't fail endlessly - if we failed once, we'll probably
909 # fail a second time. Hm. Unless the Password Manager is
910 # prompting for the information. Crap. This isn't great
911 # but it's better than the current 'repeat until recursion
912 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000913 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000914 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000915 else:
916 self.retried += 1
917 if authreq:
918 scheme = authreq.split()[0]
919 if scheme.lower() == 'digest':
920 return self.retry_http_digest_auth(req, authreq)
921
922 def retry_http_digest_auth(self, req, auth):
923 token, challenge = auth.split(' ', 1)
924 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
925 auth = self.get_authorization(req, chal)
926 if auth:
927 auth_val = 'Digest %s' % auth
928 if req.headers.get(self.auth_header, None) == auth_val:
929 return None
930 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000931 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000932 return resp
933
934 def get_cnonce(self, nonce):
935 # The cnonce-value is an opaque
936 # quoted string value provided by the client and used by both client
937 # and server to avoid chosen plaintext attacks, to provide mutual
938 # authentication, and to provide some message integrity protection.
939 # This isn't a fabulous effort, but it's probably Good Enough.
940 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
941 b = s.encode("ascii") + randombytes(8)
942 dig = hashlib.sha1(b).hexdigest()
943 return dig[:16]
944
945 def get_authorization(self, req, chal):
946 try:
947 realm = chal['realm']
948 nonce = chal['nonce']
949 qop = chal.get('qop')
950 algorithm = chal.get('algorithm', 'MD5')
951 # mod_digest doesn't send an opaque, even though it isn't
952 # supposed to be optional
953 opaque = chal.get('opaque', None)
954 except KeyError:
955 return None
956
957 H, KD = self.get_algorithm_impls(algorithm)
958 if H is None:
959 return None
960
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000961 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000962 if user is None:
963 return None
964
965 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000966 if req.data is not None:
967 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000968 else:
969 entdig = None
970
971 A1 = "%s:%s:%s" % (user, realm, pw)
972 A2 = "%s:%s" % (req.get_method(),
973 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000974 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000975 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000976 if nonce == self.last_nonce:
977 self.nonce_count += 1
978 else:
979 self.nonce_count = 1
980 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000981 ncvalue = '%08x' % self.nonce_count
982 cnonce = self.get_cnonce(nonce)
983 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
984 respdig = KD(H(A1), noncebit)
985 elif qop is None:
986 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
987 else:
988 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +0000989 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000990
991 # XXX should the partial digests be encoded too?
992
993 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000994 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000995 respdig)
996 if opaque:
997 base += ', opaque="%s"' % opaque
998 if entdig:
999 base += ', digest="%s"' % entdig
1000 base += ', algorithm="%s"' % algorithm
1001 if qop:
1002 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
1003 return base
1004
1005 def get_algorithm_impls(self, algorithm):
1006 # lambdas assume digest modules are imported at the top level
1007 if algorithm == 'MD5':
1008 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
1009 elif algorithm == 'SHA':
1010 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
1011 # XXX MD5-sess
1012 KD = lambda s, d: H("%s:%s" % (s, d))
1013 return H, KD
1014
1015 def get_entity_digest(self, data, chal):
1016 # XXX not implemented yet
1017 return None
1018
1019
1020class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1021 """An authentication protocol defined by RFC 2069
1022
1023 Digest authentication improves on basic authentication because it
1024 does not transmit passwords in the clear.
1025 """
1026
1027 auth_header = 'Authorization'
1028 handler_order = 490 # before Basic auth
1029
1030 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001031 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001032 retry = self.http_error_auth_reqed('www-authenticate',
1033 host, req, headers)
1034 self.reset_retry_count()
1035 return retry
1036
1037
1038class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
1039
1040 auth_header = 'Proxy-Authorization'
1041 handler_order = 490 # before Basic auth
1042
1043 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001044 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001045 retry = self.http_error_auth_reqed('proxy-authenticate',
1046 host, req, headers)
1047 self.reset_retry_count()
1048 return retry
1049
1050class AbstractHTTPHandler(BaseHandler):
1051
1052 def __init__(self, debuglevel=0):
1053 self._debuglevel = debuglevel
1054
1055 def set_http_debuglevel(self, level):
1056 self._debuglevel = level
1057
1058 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001059 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001060 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001061 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001062
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001063 if request.data is not None: # POST
1064 data = request.data
Senthil Kumaran29333122011-02-11 11:25:47 +00001065 if isinstance(data, str):
1066 raise TypeError("POST data should be bytes"
1067 " or an iterable of bytes. It cannot be str.")
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
Georg Brandl13e89462008-07-01 19:56:00 +00001138 raise URLError(err)
Victor Stinnera4c45d72011-06-17 14:01:18 +02001139 finally:
Senthil Kumaran1299a8f2011-07-27 08:05:58 +08001140 try:
1141 r = h.getresponse() # an HTTPResponse instance
1142 except Exception as exp:
1143 h.close()
1144 raise exp
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001145
Senthil Kumaran26430412011-04-13 07:01:19 +08001146 r.url = req.get_full_url()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001147 # This line replaces the .msg attribute of the HTTPResponse
1148 # with .headers, because urllib clients expect the response to
1149 # have the reason in .msg. It would be good to mark this
1150 # attribute is deprecated and get then to use info() or
1151 # .headers.
1152 r.msg = r.reason
1153 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001154
1155
1156class HTTPHandler(AbstractHTTPHandler):
1157
1158 def http_open(self, req):
1159 return self.do_open(http.client.HTTPConnection, req)
1160
1161 http_request = AbstractHTTPHandler.do_request_
1162
1163if hasattr(http.client, 'HTTPSConnection'):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001164 import ssl
1165
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001166 class HTTPSHandler(AbstractHTTPHandler):
1167
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001168 def __init__(self, debuglevel=0, context=None, check_hostname=None):
1169 AbstractHTTPHandler.__init__(self, debuglevel)
1170 self._context = context
1171 self._check_hostname = check_hostname
1172
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001173 def https_open(self, req):
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001174 return self.do_open(http.client.HTTPSConnection, req,
1175 context=self._context, check_hostname=self._check_hostname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001176
1177 https_request = AbstractHTTPHandler.do_request_
1178
1179class HTTPCookieProcessor(BaseHandler):
1180 def __init__(self, cookiejar=None):
1181 import http.cookiejar
1182 if cookiejar is None:
1183 cookiejar = http.cookiejar.CookieJar()
1184 self.cookiejar = cookiejar
1185
1186 def http_request(self, request):
1187 self.cookiejar.add_cookie_header(request)
1188 return request
1189
1190 def http_response(self, request, response):
1191 self.cookiejar.extract_cookies(response, request)
1192 return response
1193
1194 https_request = http_request
1195 https_response = http_response
1196
1197class UnknownHandler(BaseHandler):
1198 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001199 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001200 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001201
1202def parse_keqv_list(l):
1203 """Parse list of key=value strings where keys are not duplicated."""
1204 parsed = {}
1205 for elt in l:
1206 k, v = elt.split('=', 1)
1207 if v[0] == '"' and v[-1] == '"':
1208 v = v[1:-1]
1209 parsed[k] = v
1210 return parsed
1211
1212def parse_http_list(s):
1213 """Parse lists as described by RFC 2068 Section 2.
1214
1215 In particular, parse comma-separated lists where the elements of
1216 the list may include quoted-strings. A quoted-string could
1217 contain a comma. A non-quoted string could have quotes in the
1218 middle. Neither commas nor quotes count if they are escaped.
1219 Only double-quotes count, not single-quotes.
1220 """
1221 res = []
1222 part = ''
1223
1224 escape = quote = False
1225 for cur in s:
1226 if escape:
1227 part += cur
1228 escape = False
1229 continue
1230 if quote:
1231 if cur == '\\':
1232 escape = True
1233 continue
1234 elif cur == '"':
1235 quote = False
1236 part += cur
1237 continue
1238
1239 if cur == ',':
1240 res.append(part)
1241 part = ''
1242 continue
1243
1244 if cur == '"':
1245 quote = True
1246
1247 part += cur
1248
1249 # append last part
1250 if part:
1251 res.append(part)
1252
1253 return [part.strip() for part in res]
1254
1255class FileHandler(BaseHandler):
1256 # Use local file or FTP depending on form of URL
1257 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001258 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001259 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1260 req.host != 'localhost'):
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001261 if not req.host is self.get_names():
1262 raise URLError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001263 else:
1264 return self.open_local_file(req)
1265
1266 # names for the localhost
1267 names = None
1268 def get_names(self):
1269 if FileHandler.names is None:
1270 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001271 FileHandler.names = tuple(
1272 socket.gethostbyname_ex('localhost')[2] +
1273 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001274 except socket.gaierror:
1275 FileHandler.names = (socket.gethostbyname('localhost'),)
1276 return FileHandler.names
1277
1278 # not entirely sure what the rules are here
1279 def open_local_file(self, req):
1280 import email.utils
1281 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001282 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001283 filename = req.selector
1284 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001285 try:
1286 stats = os.stat(localfile)
1287 size = stats.st_size
1288 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001289 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001290 headers = email.message_from_string(
1291 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1292 (mtype or 'text/plain', size, modified))
1293 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001294 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001295 if not host or \
1296 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001297 if host:
1298 origurl = 'file://' + host + filename
1299 else:
1300 origurl = 'file://' + filename
1301 return addinfourl(open(localfile, 'rb'), headers, origurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001302 except OSError as msg:
Georg Brandl029986a2008-06-23 11:44:14 +00001303 # users shouldn't expect OSErrors coming from urlopen()
Georg Brandl13e89462008-07-01 19:56:00 +00001304 raise URLError(msg)
1305 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001306
1307def _safe_gethostbyname(host):
1308 try:
1309 return socket.gethostbyname(host)
1310 except socket.gaierror:
1311 return None
1312
1313class FTPHandler(BaseHandler):
1314 def ftp_open(self, req):
1315 import ftplib
1316 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001317 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001318 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001319 raise URLError('ftp error: no host given')
1320 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001321 if port is None:
1322 port = ftplib.FTP_PORT
1323 else:
1324 port = int(port)
1325
1326 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001327 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001328 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001329 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001330 else:
1331 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001332 host = unquote(host)
Senthil Kumarandaa29d02010-11-18 15:36:41 +00001333 user = user or ''
1334 passwd = passwd or ''
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001335
1336 try:
1337 host = socket.gethostbyname(host)
1338 except socket.error as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001339 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001340 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001341 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001342 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001343 dirs, file = dirs[:-1], dirs[-1]
1344 if dirs and not dirs[0]:
1345 dirs = dirs[1:]
1346 try:
1347 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1348 type = file and 'I' or 'D'
1349 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001350 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001351 if attr.lower() == 'type' and \
1352 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1353 type = value.upper()
1354 fp, retrlen = fw.retrfile(file, type)
1355 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001356 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001357 if mtype:
1358 headers += "Content-type: %s\n" % mtype
1359 if retrlen is not None and retrlen >= 0:
1360 headers += "Content-length: %d\n" % retrlen
1361 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001362 return addinfourl(fp, headers, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001363 except ftplib.all_errors as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001364 exc = URLError('ftp error: %s' % msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001365 raise exc.with_traceback(sys.exc_info()[2])
1366
1367 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001368 return ftpwrapper(user, passwd, host, port, dirs, timeout,
1369 persistent=False)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001370
1371class CacheFTPHandler(FTPHandler):
1372 # XXX would be nice to have pluggable cache strategies
1373 # XXX this stuff is definitely not thread safe
1374 def __init__(self):
1375 self.cache = {}
1376 self.timeout = {}
1377 self.soonest = 0
1378 self.delay = 60
1379 self.max_conns = 16
1380
1381 def setTimeout(self, t):
1382 self.delay = t
1383
1384 def setMaxConns(self, m):
1385 self.max_conns = m
1386
1387 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1388 key = user, host, port, '/'.join(dirs), timeout
1389 if key in self.cache:
1390 self.timeout[key] = time.time() + self.delay
1391 else:
1392 self.cache[key] = ftpwrapper(user, passwd, host, port,
1393 dirs, timeout)
1394 self.timeout[key] = time.time() + self.delay
1395 self.check_cache()
1396 return self.cache[key]
1397
1398 def check_cache(self):
1399 # first check for old ones
1400 t = time.time()
1401 if self.soonest <= t:
1402 for k, v in list(self.timeout.items()):
1403 if v < t:
1404 self.cache[k].close()
1405 del self.cache[k]
1406 del self.timeout[k]
1407 self.soonest = min(list(self.timeout.values()))
1408
1409 # then check the size
1410 if len(self.cache) == self.max_conns:
1411 for k, v in list(self.timeout.items()):
1412 if v == self.soonest:
1413 del self.cache[k]
1414 del self.timeout[k]
1415 break
1416 self.soonest = min(list(self.timeout.values()))
1417
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02001418 def clear_cache(self):
1419 for conn in self.cache.values():
1420 conn.close()
1421 self.cache.clear()
1422 self.timeout.clear()
1423
1424
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001425# Code move from the old urllib module
1426
1427MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1428
1429# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001430if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001431 from nturl2path import url2pathname, pathname2url
1432else:
1433 def url2pathname(pathname):
1434 """OS-specific conversion from a relative URL of the 'file' scheme
1435 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001436 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001437
1438 def pathname2url(pathname):
1439 """OS-specific conversion from a file system path to a relative URL
1440 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001441 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001442
1443# This really consists of two pieces:
1444# (1) a class which handles opening of all sorts of URLs
1445# (plus assorted utilities etc.)
1446# (2) a set of functions for parsing URLs
1447# XXX Should these be separated out into different modules?
1448
1449
1450ftpcache = {}
1451class URLopener:
1452 """Class to open URLs.
1453 This is a class rather than just a subroutine because we may need
1454 more than one set of global protocol-specific options.
1455 Note -- this is a base class for those who don't want the
1456 automatic handling of errors type 302 (relocated) and 401
1457 (authorization needed)."""
1458
1459 __tempfiles = None
1460
1461 version = "Python-urllib/%s" % __version__
1462
1463 # Constructor
1464 def __init__(self, proxies=None, **x509):
1465 if proxies is None:
1466 proxies = getproxies()
1467 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1468 self.proxies = proxies
1469 self.key_file = x509.get('key_file')
1470 self.cert_file = x509.get('cert_file')
1471 self.addheaders = [('User-Agent', self.version)]
1472 self.__tempfiles = []
1473 self.__unlink = os.unlink # See cleanup()
1474 self.tempcache = None
1475 # Undocumented feature: if you assign {} to tempcache,
1476 # it is used to cache files retrieved with
1477 # self.retrieve(). This is not enabled by default
1478 # since it does not work for changing documents (and I
1479 # haven't got the logic to check expiration headers
1480 # yet).
1481 self.ftpcache = ftpcache
1482 # Undocumented feature: you can use a different
1483 # ftp cache by assigning to the .ftpcache member;
1484 # in case you want logically independent URL openers
1485 # XXX This is not threadsafe. Bah.
1486
1487 def __del__(self):
1488 self.close()
1489
1490 def close(self):
1491 self.cleanup()
1492
1493 def cleanup(self):
1494 # This code sometimes runs when the rest of this module
1495 # has already been deleted, so it can't use any globals
1496 # or import anything.
1497 if self.__tempfiles:
1498 for file in self.__tempfiles:
1499 try:
1500 self.__unlink(file)
1501 except OSError:
1502 pass
1503 del self.__tempfiles[:]
1504 if self.tempcache:
1505 self.tempcache.clear()
1506
1507 def addheader(self, *args):
1508 """Add a header to be used by the HTTP interface only
1509 e.g. u.addheader('Accept', 'sound/basic')"""
1510 self.addheaders.append(args)
1511
1512 # External interface
1513 def open(self, fullurl, data=None):
1514 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001515 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001516 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001517 if self.tempcache and fullurl in self.tempcache:
1518 filename, headers = self.tempcache[fullurl]
1519 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001520 return addinfourl(fp, headers, fullurl)
1521 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001522 if not urltype:
1523 urltype = 'file'
1524 if urltype in self.proxies:
1525 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001526 urltype, proxyhost = splittype(proxy)
1527 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001528 url = (host, fullurl) # Signal special case to open_*()
1529 else:
1530 proxy = None
1531 name = 'open_' + urltype
1532 self.type = urltype
1533 name = name.replace('-', '_')
1534 if not hasattr(self, name):
1535 if proxy:
1536 return self.open_unknown_proxy(proxy, fullurl, data)
1537 else:
1538 return self.open_unknown(fullurl, data)
1539 try:
1540 if data is None:
1541 return getattr(self, name)(url)
1542 else:
1543 return getattr(self, name)(url, data)
1544 except socket.error as msg:
1545 raise IOError('socket error', msg).with_traceback(sys.exc_info()[2])
1546
1547 def open_unknown(self, fullurl, data=None):
1548 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001549 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001550 raise IOError('url error', 'unknown url type', type)
1551
1552 def open_unknown_proxy(self, proxy, fullurl, data=None):
1553 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001554 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001555 raise IOError('url error', 'invalid proxy for %s' % type, proxy)
1556
1557 # External interface
1558 def retrieve(self, url, filename=None, reporthook=None, data=None):
1559 """retrieve(url) returns (filename, headers) for a local object
1560 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001561 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001562 if self.tempcache and url in self.tempcache:
1563 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001564 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001565 if filename is None and (not type or type == 'file'):
1566 try:
1567 fp = self.open_local_file(url1)
1568 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001569 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001570 return url2pathname(splithost(url1)[1]), hdrs
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001571 except IOError as msg:
1572 pass
1573 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001574 try:
1575 headers = fp.info()
1576 if filename:
1577 tfp = open(filename, 'wb')
1578 else:
1579 import tempfile
1580 garbage, path = splittype(url)
1581 garbage, path = splithost(path or "")
1582 path, garbage = splitquery(path or "")
1583 path, garbage = splitattr(path or "")
1584 suffix = os.path.splitext(path)[1]
1585 (fd, filename) = tempfile.mkstemp(suffix)
1586 self.__tempfiles.append(filename)
1587 tfp = os.fdopen(fd, 'wb')
1588 try:
1589 result = filename, headers
1590 if self.tempcache is not None:
1591 self.tempcache[url] = result
1592 bs = 1024*8
1593 size = -1
1594 read = 0
1595 blocknum = 0
1596 if reporthook:
1597 if "content-length" in headers:
1598 size = int(headers["Content-Length"])
1599 reporthook(blocknum, bs, size)
1600 while 1:
1601 block = fp.read(bs)
1602 if not block:
1603 break
1604 read += len(block)
1605 tfp.write(block)
1606 blocknum += 1
1607 if reporthook:
1608 reporthook(blocknum, bs, size)
1609 finally:
1610 tfp.close()
1611 finally:
1612 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001613
1614 # raise exception if actual size does not match content-length header
1615 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001616 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001617 "retrieval incomplete: got only %i out of %i bytes"
1618 % (read, size), result)
1619
1620 return result
1621
1622 # Each method named open_<type> knows how to open that type of URL
1623
1624 def _open_generic_http(self, connection_factory, url, data):
1625 """Make an HTTP connection using connection_class.
1626
1627 This is an internal method that should be called from
1628 open_http() or open_https().
1629
1630 Arguments:
1631 - connection_factory should take a host name and return an
1632 HTTPConnection instance.
1633 - url is the url to retrieval or a host, relative-path pair.
1634 - data is payload for a POST request or None.
1635 """
1636
1637 user_passwd = None
1638 proxy_passwd= None
1639 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001640 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001641 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001642 user_passwd, host = splituser(host)
1643 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001644 realhost = host
1645 else:
1646 host, selector = url
1647 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001648 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001649 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001650 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001651 url = rest
1652 user_passwd = None
1653 if urltype.lower() != 'http':
1654 realhost = None
1655 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001656 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001657 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001658 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001659 if user_passwd:
1660 selector = "%s://%s%s" % (urltype, realhost, rest)
1661 if proxy_bypass(realhost):
1662 host = realhost
1663
1664 #print "proxy via http:", host, selector
1665 if not host: raise IOError('http error', 'no host given')
1666
1667 if proxy_passwd:
1668 import base64
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001669 proxy_auth = base64.b64encode(proxy_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001670 else:
1671 proxy_auth = None
1672
1673 if user_passwd:
1674 import base64
Senthil Kumaran5626eec2010-08-04 17:46:23 +00001675 auth = base64.b64encode(user_passwd.encode()).decode('ascii')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001676 else:
1677 auth = None
1678 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001679 headers = {}
1680 if proxy_auth:
1681 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1682 if auth:
1683 headers["Authorization"] = "Basic %s" % auth
1684 if realhost:
1685 headers["Host"] = realhost
Senthil Kumarand91ffca2011-03-19 17:25:27 +08001686
1687 # Add Connection:close as we don't support persistent connections yet.
1688 # This helps in closing the socket and avoiding ResourceWarning
1689
1690 headers["Connection"] = "close"
1691
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001692 for header, value in self.addheaders:
1693 headers[header] = value
1694
1695 if data is not None:
1696 headers["Content-Type"] = "application/x-www-form-urlencoded"
1697 http_conn.request("POST", selector, data, headers)
1698 else:
1699 http_conn.request("GET", selector, headers=headers)
1700
1701 try:
1702 response = http_conn.getresponse()
1703 except http.client.BadStatusLine:
1704 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001705 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001706
1707 # According to RFC 2616, "2xx" code indicates that the client's
1708 # request was successfully received, understood, and accepted.
1709 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001710 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001711 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001712 else:
1713 return self.http_error(
1714 url, response.fp,
1715 response.status, response.reason, response.msg, data)
1716
1717 def open_http(self, url, data=None):
1718 """Use HTTP protocol."""
1719 return self._open_generic_http(http.client.HTTPConnection, url, data)
1720
1721 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1722 """Handle http errors.
1723
1724 Derived class can override this, or provide specific handlers
1725 named http_error_DDD where DDD is the 3-digit error code."""
1726 # First check if there's a specific handler for this error
1727 name = 'http_error_%d' % errcode
1728 if hasattr(self, name):
1729 method = getattr(self, name)
1730 if data is None:
1731 result = method(url, fp, errcode, errmsg, headers)
1732 else:
1733 result = method(url, fp, errcode, errmsg, headers, data)
1734 if result: return result
1735 return self.http_error_default(url, fp, errcode, errmsg, headers)
1736
1737 def http_error_default(self, url, fp, errcode, errmsg, headers):
1738 """Default error handler: close the connection and raise IOError."""
1739 void = fp.read()
1740 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001741 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001742
1743 if _have_ssl:
1744 def _https_connection(self, host):
1745 return http.client.HTTPSConnection(host,
1746 key_file=self.key_file,
1747 cert_file=self.cert_file)
1748
1749 def open_https(self, url, data=None):
1750 """Use HTTPS protocol."""
1751 return self._open_generic_http(self._https_connection, url, data)
1752
1753 def open_file(self, url):
1754 """Use local file or FTP depending on form of URL."""
1755 if not isinstance(url, str):
1756 raise URLError('file error', 'proxy support for file protocol currently not implemented')
1757 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Senthil Kumaran383c32d2010-10-14 11:57:35 +00001758 raise ValueError("file:// scheme is supported only on localhost")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001759 else:
1760 return self.open_local_file(url)
1761
1762 def open_local_file(self, url):
1763 """Use local file."""
1764 import mimetypes, email.utils
1765 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001766 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001767 localname = url2pathname(file)
1768 try:
1769 stats = os.stat(localname)
1770 except OSError as e:
1771 raise URLError(e.errno, e.strerror, e.filename)
1772 size = stats.st_size
1773 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1774 mtype = mimetypes.guess_type(url)[0]
1775 headers = email.message_from_string(
1776 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1777 (mtype or 'text/plain', size, modified))
1778 if not host:
1779 urlfile = file
1780 if file[:1] == '/':
1781 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001782 return addinfourl(open(localname, 'rb'), headers, urlfile)
1783 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001784 if (not port
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001785 and socket.gethostbyname(host) in (localhost() + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001786 urlfile = file
1787 if file[:1] == '/':
1788 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001789 return addinfourl(open(localname, 'rb'), headers, urlfile)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001790 raise URLError('local file error', 'not on local host')
1791
1792 def open_ftp(self, url):
1793 """Use FTP protocol."""
1794 if not isinstance(url, str):
1795 raise URLError('ftp error', 'proxy support for ftp protocol currently not implemented')
1796 import mimetypes
1797 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001798 host, path = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001799 if not host: raise URLError('ftp error', 'no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001800 host, port = splitport(host)
1801 user, host = splituser(host)
1802 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001803 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001804 host = unquote(host)
1805 user = unquote(user or '')
1806 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001807 host = socket.gethostbyname(host)
1808 if not port:
1809 import ftplib
1810 port = ftplib.FTP_PORT
1811 else:
1812 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001813 path, attrs = splitattr(path)
1814 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001815 dirs = path.split('/')
1816 dirs, file = dirs[:-1], dirs[-1]
1817 if dirs and not dirs[0]: dirs = dirs[1:]
1818 if dirs and not dirs[0]: dirs[0] = '/'
1819 key = user, host, port, '/'.join(dirs)
1820 # XXX thread unsafe!
1821 if len(self.ftpcache) > MAXFTPCACHE:
1822 # Prune the cache, rather arbitrarily
1823 for k in self.ftpcache.keys():
1824 if k != key:
1825 v = self.ftpcache[k]
1826 del self.ftpcache[k]
1827 v.close()
1828 try:
1829 if not key in self.ftpcache:
1830 self.ftpcache[key] = \
1831 ftpwrapper(user, passwd, host, port, dirs)
1832 if not file: type = 'D'
1833 else: type = 'I'
1834 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001835 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001836 if attr.lower() == 'type' and \
1837 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1838 type = value.upper()
1839 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1840 mtype = mimetypes.guess_type("ftp:" + url)[0]
1841 headers = ""
1842 if mtype:
1843 headers += "Content-Type: %s\n" % mtype
1844 if retrlen is not None and retrlen >= 0:
1845 headers += "Content-Length: %d\n" % retrlen
1846 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001847 return addinfourl(fp, headers, "ftp:" + url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001848 except ftperrors() as msg:
1849 raise URLError('ftp error', msg).with_traceback(sys.exc_info()[2])
1850
1851 def open_data(self, url, data=None):
1852 """Use "data" URL."""
1853 if not isinstance(url, str):
1854 raise URLError('data error', 'proxy support for data protocol currently not implemented')
1855 # ignore POSTed data
1856 #
1857 # syntax of data URLs:
1858 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1859 # mediatype := [ type "/" subtype ] *( ";" parameter )
1860 # data := *urlchar
1861 # parameter := attribute "=" value
1862 try:
1863 [type, data] = url.split(',', 1)
1864 except ValueError:
1865 raise IOError('data error', 'bad data URL')
1866 if not type:
1867 type = 'text/plain;charset=US-ASCII'
1868 semi = type.rfind(';')
1869 if semi >= 0 and '=' not in type[semi:]:
1870 encoding = type[semi+1:]
1871 type = type[:semi]
1872 else:
1873 encoding = ''
1874 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00001875 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001876 time.gmtime(time.time())))
1877 msg.append('Content-type: %s' % type)
1878 if encoding == 'base64':
1879 import base64
Georg Brandl706824f2009-06-04 09:42:55 +00001880 # XXX is this encoding/decoding ok?
1881 data = base64.decodebytes(data.encode('ascii')).decode('latin1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001882 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001883 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001884 msg.append('Content-Length: %d' % len(data))
1885 msg.append('')
1886 msg.append(data)
1887 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001888 headers = email.message_from_string(msg)
1889 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001890 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001891 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001892
1893
1894class FancyURLopener(URLopener):
1895 """Derived class with handlers for errors we can handle (perhaps)."""
1896
1897 def __init__(self, *args, **kwargs):
1898 URLopener.__init__(self, *args, **kwargs)
1899 self.auth_cache = {}
1900 self.tries = 0
1901 self.maxtries = 10
1902
1903 def http_error_default(self, url, fp, errcode, errmsg, headers):
1904 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001905 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001906
1907 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1908 """Error 302 -- relocated (temporarily)."""
1909 self.tries += 1
1910 if self.maxtries and self.tries >= self.maxtries:
1911 if hasattr(self, "http_error_500"):
1912 meth = self.http_error_500
1913 else:
1914 meth = self.http_error_default
1915 self.tries = 0
1916 return meth(url, fp, 500,
1917 "Internal Server Error: Redirect Recursion", headers)
1918 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
1919 data)
1920 self.tries = 0
1921 return result
1922
1923 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
1924 if 'location' in headers:
1925 newurl = headers['location']
1926 elif 'uri' in headers:
1927 newurl = headers['uri']
1928 else:
1929 return
1930 void = fp.read()
1931 fp.close()
guido@google.coma119df92011-03-29 11:41:02 -07001932
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001933 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00001934 newurl = urljoin(self.type + ":" + url, newurl)
guido@google.coma119df92011-03-29 11:41:02 -07001935
1936 urlparts = urlparse(newurl)
1937
1938 # For security reasons, we don't allow redirection to anything other
1939 # than http, https and ftp.
1940
1941 # We are using newer HTTPError with older redirect_internal method
1942 # This older method will get deprecated in 3.3
1943
1944 if not urlparts.scheme in ('http', 'https', 'ftp'):
1945 raise HTTPError(newurl, errcode,
1946 errmsg +
1947 " Redirection to url '%s' is not allowed." % newurl,
1948 headers, fp)
1949
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001950 return self.open(newurl)
1951
1952 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
1953 """Error 301 -- also relocated (permanently)."""
1954 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1955
1956 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
1957 """Error 303 -- also relocated (essentially identical to 302)."""
1958 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1959
1960 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
1961 """Error 307 -- relocated, but turn POST into error."""
1962 if data is None:
1963 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1964 else:
1965 return self.http_error_default(url, fp, errcode, errmsg, headers)
1966
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001967 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
1968 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001969 """Error 401 -- authentication required.
1970 This function supports Basic authentication only."""
1971 if not 'www-authenticate' in headers:
1972 URLopener.http_error_default(self, url, fp,
1973 errcode, errmsg, headers)
1974 stuff = headers['www-authenticate']
1975 import re
1976 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1977 if not match:
1978 URLopener.http_error_default(self, url, fp,
1979 errcode, errmsg, headers)
1980 scheme, realm = match.groups()
1981 if scheme.lower() != 'basic':
1982 URLopener.http_error_default(self, url, fp,
1983 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001984 if not retry:
1985 URLopener.http_error_default(self, url, fp, errcode, errmsg,
1986 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001987 name = 'retry_' + self.type + '_basic_auth'
1988 if data is None:
1989 return getattr(self,name)(url, realm)
1990 else:
1991 return getattr(self,name)(url, realm, data)
1992
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001993 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
1994 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001995 """Error 407 -- proxy authentication required.
1996 This function supports Basic authentication only."""
1997 if not 'proxy-authenticate' in headers:
1998 URLopener.http_error_default(self, url, fp,
1999 errcode, errmsg, headers)
2000 stuff = headers['proxy-authenticate']
2001 import re
2002 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
2003 if not match:
2004 URLopener.http_error_default(self, url, fp,
2005 errcode, errmsg, headers)
2006 scheme, realm = match.groups()
2007 if scheme.lower() != 'basic':
2008 URLopener.http_error_default(self, url, fp,
2009 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00002010 if not retry:
2011 URLopener.http_error_default(self, url, fp, errcode, errmsg,
2012 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002013 name = 'retry_proxy_' + self.type + '_basic_auth'
2014 if data is None:
2015 return getattr(self,name)(url, realm)
2016 else:
2017 return getattr(self,name)(url, realm, data)
2018
2019 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002020 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002021 newurl = 'http://' + host + selector
2022 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00002023 urltype, proxyhost = splittype(proxy)
2024 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002025 i = proxyhost.find('@') + 1
2026 proxyhost = proxyhost[i:]
2027 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2028 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002029 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002030 quote(passwd, safe=''), proxyhost)
2031 self.proxies['http'] = 'http://' + proxyhost + proxyselector
2032 if data is None:
2033 return self.open(newurl)
2034 else:
2035 return self.open(newurl, data)
2036
2037 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002038 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002039 newurl = 'https://' + host + selector
2040 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00002041 urltype, proxyhost = splittype(proxy)
2042 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002043 i = proxyhost.find('@') + 1
2044 proxyhost = proxyhost[i:]
2045 user, passwd = self.get_user_passwd(proxyhost, realm, i)
2046 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002047 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002048 quote(passwd, safe=''), proxyhost)
2049 self.proxies['https'] = 'https://' + proxyhost + proxyselector
2050 if data is None:
2051 return self.open(newurl)
2052 else:
2053 return self.open(newurl, data)
2054
2055 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002056 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002057 i = host.find('@') + 1
2058 host = host[i:]
2059 user, passwd = self.get_user_passwd(host, realm, i)
2060 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002061 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002062 quote(passwd, safe=''), host)
2063 newurl = 'http://' + host + selector
2064 if data is None:
2065 return self.open(newurl)
2066 else:
2067 return self.open(newurl, data)
2068
2069 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00002070 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002071 i = host.find('@') + 1
2072 host = host[i:]
2073 user, passwd = self.get_user_passwd(host, realm, i)
2074 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00002075 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002076 quote(passwd, safe=''), host)
2077 newurl = 'https://' + host + selector
2078 if data is None:
2079 return self.open(newurl)
2080 else:
2081 return self.open(newurl, data)
2082
Florent Xicluna757445b2010-05-17 17:24:07 +00002083 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002084 key = realm + '@' + host.lower()
2085 if key in self.auth_cache:
2086 if clear_cache:
2087 del self.auth_cache[key]
2088 else:
2089 return self.auth_cache[key]
2090 user, passwd = self.prompt_user_passwd(host, realm)
2091 if user or passwd: self.auth_cache[key] = (user, passwd)
2092 return user, passwd
2093
2094 def prompt_user_passwd(self, host, realm):
2095 """Override this in a GUI environment!"""
2096 import getpass
2097 try:
2098 user = input("Enter username for %s at %s: " % (realm, host))
2099 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2100 (user, realm, host))
2101 return user, passwd
2102 except KeyboardInterrupt:
2103 print()
2104 return None, None
2105
2106
2107# Utility functions
2108
2109_localhost = None
2110def localhost():
2111 """Return the IP address of the magic hostname 'localhost'."""
2112 global _localhost
2113 if _localhost is None:
2114 _localhost = socket.gethostbyname('localhost')
2115 return _localhost
2116
2117_thishost = None
2118def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002119 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002120 global _thishost
2121 if _thishost is None:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002122 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname()[2]))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002123 return _thishost
2124
2125_ftperrors = None
2126def ftperrors():
2127 """Return the set of errors raised by the FTP class."""
2128 global _ftperrors
2129 if _ftperrors is None:
2130 import ftplib
2131 _ftperrors = ftplib.all_errors
2132 return _ftperrors
2133
2134_noheaders = None
2135def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002136 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002137 global _noheaders
2138 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002139 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002140 return _noheaders
2141
2142
2143# Utility classes
2144
2145class ftpwrapper:
2146 """Class used by open_ftp() for cache of open FTP connections."""
2147
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002148 def __init__(self, user, passwd, host, port, dirs, timeout=None,
2149 persistent=True):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002150 self.user = user
2151 self.passwd = passwd
2152 self.host = host
2153 self.port = port
2154 self.dirs = dirs
2155 self.timeout = timeout
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002156 self.refcount = 0
2157 self.keepalive = persistent
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002158 self.init()
2159
2160 def init(self):
2161 import ftplib
2162 self.busy = 0
2163 self.ftp = ftplib.FTP()
2164 self.ftp.connect(self.host, self.port, self.timeout)
2165 self.ftp.login(self.user, self.passwd)
2166 for dir in self.dirs:
2167 self.ftp.cwd(dir)
2168
2169 def retrfile(self, file, type):
2170 import ftplib
2171 self.endtransfer()
2172 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2173 else: cmd = 'TYPE ' + type; isdir = 0
2174 try:
2175 self.ftp.voidcmd(cmd)
2176 except ftplib.all_errors:
2177 self.init()
2178 self.ftp.voidcmd(cmd)
2179 conn = None
2180 if file and not isdir:
2181 # Try to retrieve as a file
2182 try:
2183 cmd = 'RETR ' + file
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002184 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002185 except ftplib.error_perm as reason:
2186 if str(reason)[:3] != '550':
Georg Brandl13e89462008-07-01 19:56:00 +00002187 raise URLError('ftp error', reason).with_traceback(
2188 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002189 if not conn:
2190 # Set transfer mode to ASCII!
2191 self.ftp.voidcmd('TYPE A')
2192 # Try a directory listing. Verify that directory exists.
2193 if file:
2194 pwd = self.ftp.pwd()
2195 try:
2196 try:
2197 self.ftp.cwd(file)
2198 except ftplib.error_perm as reason:
Georg Brandl13e89462008-07-01 19:56:00 +00002199 raise URLError('ftp error', reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002200 finally:
2201 self.ftp.cwd(pwd)
2202 cmd = 'LIST ' + file
2203 else:
2204 cmd = 'LIST'
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002205 conn, retrlen = self.ftp.ntransfercmd(cmd)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002206 self.busy = 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002207
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002208 ftpobj = addclosehook(conn.makefile('rb'), self.file_close)
2209 self.refcount += 1
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002210 conn.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002211 # Pass back both a suitably decorated object and a retrieval length
Senthil Kumaran2024acd2011-03-24 11:46:19 +08002212 return (ftpobj, retrlen)
2213
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002214 def endtransfer(self):
2215 if not self.busy:
2216 return
2217 self.busy = 0
2218 try:
2219 self.ftp.voidresp()
2220 except ftperrors():
2221 pass
2222
2223 def close(self):
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +02002224 self.keepalive = False
2225 if self.refcount <= 0:
2226 self.real_close()
2227
2228 def file_close(self):
2229 self.endtransfer()
2230 self.refcount -= 1
2231 if self.refcount <= 0 and not self.keepalive:
2232 self.real_close()
2233
2234 def real_close(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002235 self.endtransfer()
2236 try:
2237 self.ftp.close()
2238 except ftperrors():
2239 pass
2240
2241# Proxy handling
2242def getproxies_environment():
2243 """Return a dictionary of scheme -> proxy server URL mappings.
2244
2245 Scan the environment for variables named <scheme>_proxy;
2246 this seems to be the standard convention. If you need a
2247 different way, you can pass a proxies dictionary to the
2248 [Fancy]URLopener constructor.
2249
2250 """
2251 proxies = {}
2252 for name, value in os.environ.items():
2253 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002254 if value and name[-6:] == '_proxy':
2255 proxies[name[:-6]] = value
2256 return proxies
2257
2258def proxy_bypass_environment(host):
2259 """Test if proxies should not be used for a particular host.
2260
2261 Checks the environment for a variable named no_proxy, which should
2262 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2263 """
2264 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2265 # '*' is special case for always bypass
2266 if no_proxy == '*':
2267 return 1
2268 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002269 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002270 # check if the host ends with any of the DNS suffixes
2271 for name in no_proxy.split(','):
2272 if name and (hostonly.endswith(name) or host.endswith(name)):
2273 return 1
2274 # otherwise, don't bypass
2275 return 0
2276
2277
Ronald Oussorene72e1612011-03-14 18:15:25 -04002278# This code tests an OSX specific data structure but is testable on all
2279# platforms
2280def _proxy_bypass_macosx_sysconf(host, proxy_settings):
2281 """
2282 Return True iff this host shouldn't be accessed using a proxy
2283
2284 This function uses the MacOSX framework SystemConfiguration
2285 to fetch the proxy information.
2286
2287 proxy_settings come from _scproxy._get_proxy_settings or get mocked ie:
2288 { 'exclude_simple': bool,
2289 'exceptions': ['foo.bar', '*.bar.com', '127.0.0.1', '10.1', '10.0/16']
2290 }
2291 """
2292 import re
2293 import socket
2294 from fnmatch import fnmatch
2295
2296 hostonly, port = splitport(host)
2297
2298 def ip2num(ipAddr):
2299 parts = ipAddr.split('.')
2300 parts = list(map(int, parts))
2301 if len(parts) != 4:
2302 parts = (parts + [0, 0, 0, 0])[:4]
2303 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2304
2305 # Check for simple host names:
2306 if '.' not in host:
2307 if proxy_settings['exclude_simple']:
2308 return True
2309
2310 hostIP = None
2311
2312 for value in proxy_settings.get('exceptions', ()):
2313 # Items in the list are strings like these: *.local, 169.254/16
2314 if not value: continue
2315
2316 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2317 if m is not None:
2318 if hostIP is None:
2319 try:
2320 hostIP = socket.gethostbyname(hostonly)
2321 hostIP = ip2num(hostIP)
2322 except socket.error:
2323 continue
2324
2325 base = ip2num(m.group(1))
2326 mask = m.group(2)
2327 if mask is None:
2328 mask = 8 * (m.group(1).count('.') + 1)
2329 else:
2330 mask = int(mask[1:])
2331 mask = 32 - mask
2332
2333 if (hostIP >> mask) == (base >> mask):
2334 return True
2335
2336 elif fnmatch(host, value):
2337 return True
2338
2339 return False
2340
2341
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002342if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002343 from _scproxy import _get_proxy_settings, _get_proxies
2344
2345 def proxy_bypass_macosx_sysconf(host):
Ronald Oussoren84151202010-04-18 20:46:11 +00002346 proxy_settings = _get_proxy_settings()
Ronald Oussorene72e1612011-03-14 18:15:25 -04002347 return _proxy_bypass_macosx_sysconf(host, proxy_settings)
Ronald Oussoren84151202010-04-18 20:46:11 +00002348
2349 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002350 """Return a dictionary of scheme -> proxy server URL mappings.
2351
Ronald Oussoren84151202010-04-18 20:46:11 +00002352 This function uses the MacOSX framework SystemConfiguration
2353 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002354 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002355 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002356
Ronald Oussoren84151202010-04-18 20:46:11 +00002357
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002358
2359 def proxy_bypass(host):
2360 if getproxies_environment():
2361 return proxy_bypass_environment(host)
2362 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002363 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002364
2365 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002366 return getproxies_environment() or getproxies_macosx_sysconf()
2367
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002368
2369elif os.name == 'nt':
2370 def getproxies_registry():
2371 """Return a dictionary of scheme -> proxy server URL mappings.
2372
2373 Win32 uses the registry to store proxies.
2374
2375 """
2376 proxies = {}
2377 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002378 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002379 except ImportError:
2380 # Std module, so should be around - but you never know!
2381 return proxies
2382 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002383 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002384 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002385 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002386 'ProxyEnable')[0]
2387 if proxyEnable:
2388 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002389 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002390 'ProxyServer')[0])
2391 if '=' in proxyServer:
2392 # Per-protocol settings
2393 for p in proxyServer.split(';'):
2394 protocol, address = p.split('=', 1)
2395 # See if address has a type:// prefix
2396 import re
2397 if not re.match('^([^/:]+)://', address):
2398 address = '%s://%s' % (protocol, address)
2399 proxies[protocol] = address
2400 else:
2401 # Use one setting for all protocols
2402 if proxyServer[:5] == 'http:':
2403 proxies['http'] = proxyServer
2404 else:
2405 proxies['http'] = 'http://%s' % proxyServer
Senthil Kumaran04f31b82010-07-14 20:10:52 +00002406 proxies['https'] = 'https://%s' % proxyServer
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002407 proxies['ftp'] = 'ftp://%s' % proxyServer
2408 internetSettings.Close()
2409 except (WindowsError, ValueError, TypeError):
2410 # Either registry key not found etc, or the value in an
2411 # unexpected format.
2412 # proxies already set up to be empty so nothing to do
2413 pass
2414 return proxies
2415
2416 def getproxies():
2417 """Return a dictionary of scheme -> proxy server URL mappings.
2418
2419 Returns settings gathered from the environment, if specified,
2420 or the registry.
2421
2422 """
2423 return getproxies_environment() or getproxies_registry()
2424
2425 def proxy_bypass_registry(host):
2426 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002427 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002428 import re
2429 except ImportError:
2430 # Std modules, so should be around - but you never know!
2431 return 0
2432 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002433 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002434 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002435 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002436 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002437 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002438 'ProxyOverride')[0])
2439 # ^^^^ Returned as Unicode but problems if not converted to ASCII
2440 except WindowsError:
2441 return 0
2442 if not proxyEnable or not proxyOverride:
2443 return 0
2444 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002445 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002446 host = [rawHost]
2447 try:
2448 addr = socket.gethostbyname(rawHost)
2449 if addr != rawHost:
2450 host.append(addr)
2451 except socket.error:
2452 pass
2453 try:
2454 fqdn = socket.getfqdn(rawHost)
2455 if fqdn != rawHost:
2456 host.append(fqdn)
2457 except socket.error:
2458 pass
2459 # make a check value list from the registry entry: replace the
2460 # '<local>' string by the localhost entry and the corresponding
2461 # canonical entry.
2462 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002463 # now check if we match one of the registry values.
2464 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002465 if test == '<local>':
2466 if '.' not in rawHost:
2467 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002468 test = test.replace(".", r"\.") # mask dots
2469 test = test.replace("*", r".*") # change glob sequence
2470 test = test.replace("?", r".") # change glob char
2471 for val in host:
2472 # print "%s <--> %s" %( test, val )
2473 if re.match(test, val, re.I):
2474 return 1
2475 return 0
2476
2477 def proxy_bypass(host):
2478 """Return a dictionary of scheme -> proxy server URL mappings.
2479
2480 Returns settings gathered from the environment, if specified,
2481 or the registry.
2482
2483 """
2484 if getproxies_environment():
2485 return proxy_bypass_environment(host)
2486 else:
2487 return proxy_bypass_registry(host)
2488
2489else:
2490 # By default use environment variables
2491 getproxies = getproxies_environment
2492 proxy_bypass = proxy_bypass_environment