blob: c6767d0b26601bde5a1a6c11eacd07630fca1c3d [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
Jeremy Hylton1afc1692008-06-18 20:49:58 +000097
Georg Brandl13e89462008-07-01 19:56:00 +000098from urllib.error import URLError, HTTPError, ContentTooShortError
99from urllib.parse import (
100 urlparse, urlsplit, urljoin, unwrap, quote, unquote,
101 splittype, splithost, splitport, splituser, splitpasswd,
Facundo Batistaf24802c2008-08-17 03:36:03 +0000102 splitattr, splitquery, splitvalue, to_bytes, urlunparse)
Georg Brandl13e89462008-07-01 19:56:00 +0000103from urllib.response import addinfourl, addclosehook
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000104
105# check for SSL
106try:
107 import ssl
108except:
109 _have_ssl = False
110else:
111 _have_ssl = True
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000112
113# used in User-Agent header sent
114__version__ = sys.version[:3]
115
116_opener = None
117def urlopen(url, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
118 global _opener
119 if _opener is None:
120 _opener = build_opener()
121 return _opener.open(url, data, timeout)
122
123def install_opener(opener):
124 global _opener
125 _opener = opener
126
127# TODO(jhylton): Make this work with the same global opener.
128_urlopener = None
129def urlretrieve(url, filename=None, reporthook=None, data=None):
130 global _urlopener
131 if not _urlopener:
132 _urlopener = FancyURLopener()
133 return _urlopener.retrieve(url, filename, reporthook, data)
134
135def urlcleanup():
136 if _urlopener:
137 _urlopener.cleanup()
138 global _opener
139 if _opener:
140 _opener = None
141
142# copied from cookielib.py
Antoine Pitroufd036452008-08-19 17:56:33 +0000143_cut_port_re = re.compile(r":\d+$", re.ASCII)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000144def request_host(request):
145 """Return request-host, as defined by RFC 2965.
146
147 Variation from RFC: returned value is lowercased, for convenient
148 comparison.
149
150 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000151 url = request.full_url
Georg Brandl13e89462008-07-01 19:56:00 +0000152 host = urlparse(url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000153 if host == "":
154 host = request.get_header("Host", "")
155
156 # remove port, if present
157 host = _cut_port_re.sub("", host, 1)
158 return host.lower()
159
160class Request:
161
162 def __init__(self, url, data=None, headers={},
163 origin_req_host=None, unverifiable=False):
164 # unwrap('<URL:type://host/path>') --> 'type://host/path'
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000165 self.full_url = unwrap(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000166 self.data = data
167 self.headers = {}
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000168 self._tunnel_host = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000169 for key, value in headers.items():
170 self.add_header(key, value)
171 self.unredirected_hdrs = {}
172 if origin_req_host is None:
173 origin_req_host = request_host(self)
174 self.origin_req_host = origin_req_host
175 self.unverifiable = unverifiable
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000176 self._parse()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000177
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000178 def _parse(self):
179 self.type, rest = splittype(self.full_url)
180 if self.type is None:
181 raise ValueError("unknown url type: %s" % self.full_url)
182 self.host, self.selector = splithost(rest)
183 if self.host:
184 self.host = unquote(self.host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000185
186 def get_method(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000187 if self.data is not None:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000188 return "POST"
189 else:
190 return "GET"
191
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000192 # Begin deprecated methods
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000193
194 def add_data(self, data):
195 self.data = data
196
197 def has_data(self):
198 return self.data is not None
199
200 def get_data(self):
201 return self.data
202
203 def get_full_url(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000204 return self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000205
206 def get_type(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000207 return self.type
208
209 def get_host(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000210 return self.host
211
212 def get_selector(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000213 return self.selector
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000214
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000215 def is_unverifiable(self):
216 return self.unverifiable
Facundo Batista72dc1ea2008-08-16 14:44:32 +0000217
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000218 def get_origin_req_host(self):
219 return self.origin_req_host
220
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000221 # End deprecated methods
222
223 def set_proxy(self, host, type):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000224 if self.type == 'https' and not self._tunnel_host:
225 self._tunnel_host = self.host
226 else:
227 self.type= type
228 self.selector = self.full_url
229 self.host = host
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000230
231 def has_proxy(self):
232 return self.selector == self.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000233
234 def add_header(self, key, val):
235 # useful for something like authentication
236 self.headers[key.capitalize()] = val
237
238 def add_unredirected_header(self, key, val):
239 # will not be added to a redirected request
240 self.unredirected_hdrs[key.capitalize()] = val
241
242 def has_header(self, header_name):
243 return (header_name in self.headers or
244 header_name in self.unredirected_hdrs)
245
246 def get_header(self, header_name, default=None):
247 return self.headers.get(
248 header_name,
249 self.unredirected_hdrs.get(header_name, default))
250
251 def header_items(self):
252 hdrs = self.unredirected_hdrs.copy()
253 hdrs.update(self.headers)
254 return list(hdrs.items())
255
256class OpenerDirector:
257 def __init__(self):
258 client_version = "Python-urllib/%s" % __version__
259 self.addheaders = [('User-agent', client_version)]
260 # manage the individual handlers
261 self.handlers = []
262 self.handle_open = {}
263 self.handle_error = {}
264 self.process_response = {}
265 self.process_request = {}
266
267 def add_handler(self, handler):
268 if not hasattr(handler, "add_parent"):
269 raise TypeError("expected BaseHandler instance, got %r" %
270 type(handler))
271
272 added = False
273 for meth in dir(handler):
274 if meth in ["redirect_request", "do_open", "proxy_open"]:
275 # oops, coincidental match
276 continue
277
278 i = meth.find("_")
279 protocol = meth[:i]
280 condition = meth[i+1:]
281
282 if condition.startswith("error"):
283 j = condition.find("_") + i + 1
284 kind = meth[j+1:]
285 try:
286 kind = int(kind)
287 except ValueError:
288 pass
289 lookup = self.handle_error.get(protocol, {})
290 self.handle_error[protocol] = lookup
291 elif condition == "open":
292 kind = protocol
293 lookup = self.handle_open
294 elif condition == "response":
295 kind = protocol
296 lookup = self.process_response
297 elif condition == "request":
298 kind = protocol
299 lookup = self.process_request
300 else:
301 continue
302
303 handlers = lookup.setdefault(kind, [])
304 if handlers:
305 bisect.insort(handlers, handler)
306 else:
307 handlers.append(handler)
308 added = True
309
310 if added:
311 # the handlers must work in an specific order, the order
312 # is specified in a Handler attribute
313 bisect.insort(self.handlers, handler)
314 handler.add_parent(self)
315
316 def close(self):
317 # Only exists for backwards compatibility.
318 pass
319
320 def _call_chain(self, chain, kind, meth_name, *args):
321 # Handlers raise an exception if no one else should try to handle
322 # the request, or return None if they can't but another handler
323 # could. Otherwise, they return the response.
324 handlers = chain.get(kind, ())
325 for handler in handlers:
326 func = getattr(handler, meth_name)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000327 result = func(*args)
328 if result is not None:
329 return result
330
331 def open(self, fullurl, data=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
332 # accept a URL or a Request object
333 if isinstance(fullurl, str):
334 req = Request(fullurl, data)
335 else:
336 req = fullurl
337 if data is not None:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000338 req.data = data
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000339
340 req.timeout = timeout
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000341 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000342
343 # pre-process request
344 meth_name = protocol+"_request"
345 for processor in self.process_request.get(protocol, []):
346 meth = getattr(processor, meth_name)
347 req = meth(req)
348
349 response = self._open(req, data)
350
351 # post-process response
352 meth_name = protocol+"_response"
353 for processor in self.process_response.get(protocol, []):
354 meth = getattr(processor, meth_name)
355 response = meth(req, response)
356
357 return response
358
359 def _open(self, req, data=None):
360 result = self._call_chain(self.handle_open, 'default',
361 'default_open', req)
362 if result:
363 return result
364
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000365 protocol = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000366 result = self._call_chain(self.handle_open, protocol, protocol +
367 '_open', req)
368 if result:
369 return result
370
371 return self._call_chain(self.handle_open, 'unknown',
372 'unknown_open', req)
373
374 def error(self, proto, *args):
375 if proto in ('http', 'https'):
376 # XXX http[s] protocols are special-cased
377 dict = self.handle_error['http'] # https is not different than http
378 proto = args[2] # YUCK!
379 meth_name = 'http_error_%s' % proto
380 http_err = 1
381 orig_args = args
382 else:
383 dict = self.handle_error
384 meth_name = proto + '_error'
385 http_err = 0
386 args = (dict, proto, meth_name) + args
387 result = self._call_chain(*args)
388 if result:
389 return result
390
391 if http_err:
392 args = (dict, 'default', 'http_error_default') + orig_args
393 return self._call_chain(*args)
394
395# XXX probably also want an abstract factory that knows when it makes
396# sense to skip a superclass in favor of a subclass and when it might
397# make sense to include both
398
399def build_opener(*handlers):
400 """Create an opener object from a list of handlers.
401
402 The opener will use several default handlers, including support
Senthil Kumaran1107c5d2009-11-15 06:20:55 +0000403 for HTTP, FTP and when applicable HTTPS.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000404
405 If any of the handlers passed as arguments are subclasses of the
406 default handlers, the default handlers will not be used.
407 """
408 def isclass(obj):
409 return isinstance(obj, type) or hasattr(obj, "__bases__")
410
411 opener = OpenerDirector()
412 default_classes = [ProxyHandler, UnknownHandler, HTTPHandler,
413 HTTPDefaultErrorHandler, HTTPRedirectHandler,
414 FTPHandler, FileHandler, HTTPErrorProcessor]
415 if hasattr(http.client, "HTTPSConnection"):
416 default_classes.append(HTTPSHandler)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000417 skip = set()
418 for klass in default_classes:
419 for check in handlers:
420 if isclass(check):
421 if issubclass(check, klass):
422 skip.add(klass)
423 elif isinstance(check, klass):
424 skip.add(klass)
425 for klass in skip:
426 default_classes.remove(klass)
427
428 for klass in default_classes:
429 opener.add_handler(klass())
430
431 for h in handlers:
432 if isclass(h):
433 h = h()
434 opener.add_handler(h)
435 return opener
436
437class BaseHandler:
438 handler_order = 500
439
440 def add_parent(self, parent):
441 self.parent = parent
442
443 def close(self):
444 # Only exists for backwards compatibility
445 pass
446
447 def __lt__(self, other):
448 if not hasattr(other, "handler_order"):
449 # Try to preserve the old behavior of having custom classes
450 # inserted after default ones (works only for custom user
451 # classes which are not aware of handler_order).
452 return True
453 return self.handler_order < other.handler_order
454
455
456class HTTPErrorProcessor(BaseHandler):
457 """Process HTTP error responses."""
458 handler_order = 1000 # after all other processing
459
460 def http_response(self, request, response):
461 code, msg, hdrs = response.code, response.msg, response.info()
462
463 # According to RFC 2616, "2xx" code indicates that the client's
464 # request was successfully received, understood, and accepted.
465 if not (200 <= code < 300):
466 response = self.parent.error(
467 'http', request, response, code, msg, hdrs)
468
469 return response
470
471 https_response = http_response
472
473class HTTPDefaultErrorHandler(BaseHandler):
474 def http_error_default(self, req, fp, code, msg, hdrs):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000475 raise HTTPError(req.full_url, code, msg, hdrs, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000476
477class HTTPRedirectHandler(BaseHandler):
478 # maximum number of redirections to any single URL
479 # this is needed because of the state that cookies introduce
480 max_repeats = 4
481 # maximum total number of redirections (regardless of URL) before
482 # assuming we're in a loop
483 max_redirections = 10
484
485 def redirect_request(self, req, fp, code, msg, headers, newurl):
486 """Return a Request or None in response to a redirect.
487
488 This is called by the http_error_30x methods when a
489 redirection response is received. If a redirection should
490 take place, return a new Request to allow http_error_30x to
491 perform the redirect. Otherwise, raise HTTPError if no-one
492 else should try to handle this url. Return None if you can't
493 but another Handler might.
494 """
495 m = req.get_method()
496 if (not (code in (301, 302, 303, 307) and m in ("GET", "HEAD")
497 or code in (301, 302, 303) and m == "POST")):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000498 raise HTTPError(req.full_url, code, msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000499
500 # Strictly (according to RFC 2616), 301 or 302 in response to
501 # a POST MUST NOT cause a redirection without confirmation
Georg Brandl029986a2008-06-23 11:44:14 +0000502 # from the user (of urllib.request, in this case). In practice,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000503 # essentially all clients do redirect in this case, so we do
504 # the same.
505 # be conciliant with URIs containing a space
506 newurl = newurl.replace(' ', '%20')
507 CONTENT_HEADERS = ("content-length", "content-type")
508 newheaders = dict((k, v) for k, v in req.headers.items()
509 if k.lower() not in CONTENT_HEADERS)
510 return Request(newurl,
511 headers=newheaders,
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000512 origin_req_host=req.origin_req_host,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000513 unverifiable=True)
514
515 # Implementation note: To avoid the server sending us into an
516 # infinite loop, the request object needs to track what URLs we
517 # have already seen. Do this by adding a handler-specific
518 # attribute to the Request object.
519 def http_error_302(self, req, fp, code, msg, headers):
520 # Some servers (incorrectly) return multiple Location headers
521 # (so probably same goes for URI). Use first header.
522 if "location" in headers:
523 newurl = headers["location"]
524 elif "uri" in headers:
525 newurl = headers["uri"]
526 else:
527 return
Facundo Batistaf24802c2008-08-17 03:36:03 +0000528
529 # fix a possible malformed URL
530 urlparts = urlparse(newurl)
531 if not urlparts.path:
532 urlparts = list(urlparts)
533 urlparts[2] = "/"
534 newurl = urlunparse(urlparts)
535
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000536 newurl = urljoin(req.full_url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000537
538 # XXX Probably want to forget about the state of the current
539 # request, although that might interact poorly with other
540 # handlers that also use handler-specific request attributes
541 new = self.redirect_request(req, fp, code, msg, headers, newurl)
542 if new is None:
543 return
544
545 # loop detection
546 # .redirect_dict has a key url if url was previously visited.
547 if hasattr(req, 'redirect_dict'):
548 visited = new.redirect_dict = req.redirect_dict
549 if (visited.get(newurl, 0) >= self.max_repeats or
550 len(visited) >= self.max_redirections):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000551 raise HTTPError(req.full_url, code,
Georg Brandl13e89462008-07-01 19:56:00 +0000552 self.inf_msg + msg, headers, fp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000553 else:
554 visited = new.redirect_dict = req.redirect_dict = {}
555 visited[newurl] = visited.get(newurl, 0) + 1
556
557 # Don't close the fp until we are sure that we won't use it
558 # with HTTPError.
559 fp.read()
560 fp.close()
561
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000562 return self.parent.open(new, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000563
564 http_error_301 = http_error_303 = http_error_307 = http_error_302
565
566 inf_msg = "The HTTP server returned a redirect error that would " \
567 "lead to an infinite loop.\n" \
568 "The last 30x error message was:\n"
569
570
571def _parse_proxy(proxy):
572 """Return (scheme, user, password, host/port) given a URL or an authority.
573
574 If a URL is supplied, it must have an authority (host:port) component.
575 According to RFC 3986, having an authority component means the URL must
576 have two slashes after the scheme:
577
578 >>> _parse_proxy('file:/ftp.example.com/')
579 Traceback (most recent call last):
580 ValueError: proxy URL with no authority: 'file:/ftp.example.com/'
581
582 The first three items of the returned tuple may be None.
583
584 Examples of authority parsing:
585
586 >>> _parse_proxy('proxy.example.com')
587 (None, None, None, 'proxy.example.com')
588 >>> _parse_proxy('proxy.example.com:3128')
589 (None, None, None, 'proxy.example.com:3128')
590
591 The authority component may optionally include userinfo (assumed to be
592 username:password):
593
594 >>> _parse_proxy('joe:password@proxy.example.com')
595 (None, 'joe', 'password', 'proxy.example.com')
596 >>> _parse_proxy('joe:password@proxy.example.com:3128')
597 (None, 'joe', 'password', 'proxy.example.com:3128')
598
599 Same examples, but with URLs instead:
600
601 >>> _parse_proxy('http://proxy.example.com/')
602 ('http', None, None, 'proxy.example.com')
603 >>> _parse_proxy('http://proxy.example.com:3128/')
604 ('http', None, None, 'proxy.example.com:3128')
605 >>> _parse_proxy('http://joe:password@proxy.example.com/')
606 ('http', 'joe', 'password', 'proxy.example.com')
607 >>> _parse_proxy('http://joe:password@proxy.example.com:3128')
608 ('http', 'joe', 'password', 'proxy.example.com:3128')
609
610 Everything after the authority is ignored:
611
612 >>> _parse_proxy('ftp://joe:password@proxy.example.com/rubbish:3128')
613 ('ftp', 'joe', 'password', 'proxy.example.com')
614
615 Test for no trailing '/' case:
616
617 >>> _parse_proxy('http://joe:password@proxy.example.com')
618 ('http', 'joe', 'password', 'proxy.example.com')
619
620 """
Georg Brandl13e89462008-07-01 19:56:00 +0000621 scheme, r_scheme = splittype(proxy)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000622 if not r_scheme.startswith("/"):
623 # authority
624 scheme = None
625 authority = proxy
626 else:
627 # URL
628 if not r_scheme.startswith("//"):
629 raise ValueError("proxy URL with no authority: %r" % proxy)
630 # We have an authority, so for RFC 3986-compliant URLs (by ss 3.
631 # and 3.3.), path is empty or starts with '/'
632 end = r_scheme.find("/", 2)
633 if end == -1:
634 end = None
635 authority = r_scheme[2:end]
Georg Brandl13e89462008-07-01 19:56:00 +0000636 userinfo, hostport = splituser(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000637 if userinfo is not None:
Georg Brandl13e89462008-07-01 19:56:00 +0000638 user, password = splitpasswd(userinfo)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000639 else:
640 user = password = None
641 return scheme, user, password, hostport
642
643class ProxyHandler(BaseHandler):
644 # Proxies must be in front
645 handler_order = 100
646
647 def __init__(self, proxies=None):
648 if proxies is None:
649 proxies = getproxies()
650 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
651 self.proxies = proxies
652 for type, url in proxies.items():
653 setattr(self, '%s_open' % type,
654 lambda r, proxy=url, type=type, meth=self.proxy_open: \
655 meth(r, proxy, type))
656
657 def proxy_open(self, req, proxy, type):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000658 orig_type = req.type
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000659 proxy_type, user, password, hostport = _parse_proxy(proxy)
660 if proxy_type is None:
661 proxy_type = orig_type
Senthil Kumaran7bb04972009-10-11 04:58:55 +0000662
663 if req.host and proxy_bypass(req.host):
664 return None
665
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000666 if user and password:
Georg Brandl13e89462008-07-01 19:56:00 +0000667 user_pass = '%s:%s' % (unquote(user),
668 unquote(password))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000669 creds = base64.b64encode(user_pass.encode()).decode("ascii")
670 req.add_header('Proxy-authorization', 'Basic ' + creds)
Georg Brandl13e89462008-07-01 19:56:00 +0000671 hostport = unquote(hostport)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000672 req.set_proxy(hostport, proxy_type)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000673 if orig_type == proxy_type or orig_type == 'https':
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000674 # let other handlers take care of it
675 return None
676 else:
677 # need to start over, because the other handlers don't
678 # grok the proxy's URL type
679 # e.g. if we have a constructor arg proxies like so:
680 # {'http': 'ftp://proxy.example.com'}, we may end up turning
681 # a request for http://acme.example.com/a into one for
682 # ftp://proxy.example.com/a
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000683 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000684
685class HTTPPasswordMgr:
686
687 def __init__(self):
688 self.passwd = {}
689
690 def add_password(self, realm, uri, user, passwd):
691 # uri could be a single URI or a sequence
692 if isinstance(uri, str):
693 uri = [uri]
694 if not realm in self.passwd:
695 self.passwd[realm] = {}
696 for default_port in True, False:
697 reduced_uri = tuple(
698 [self.reduce_uri(u, default_port) for u in uri])
699 self.passwd[realm][reduced_uri] = (user, passwd)
700
701 def find_user_password(self, realm, authuri):
702 domains = self.passwd.get(realm, {})
703 for default_port in True, False:
704 reduced_authuri = self.reduce_uri(authuri, default_port)
705 for uris, authinfo in domains.items():
706 for uri in uris:
707 if self.is_suburi(uri, reduced_authuri):
708 return authinfo
709 return None, None
710
711 def reduce_uri(self, uri, default_port=True):
712 """Accept authority or URI and extract only the authority and path."""
713 # note HTTP URLs do not have a userinfo component
Georg Brandl13e89462008-07-01 19:56:00 +0000714 parts = urlsplit(uri)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000715 if parts[1]:
716 # URI
717 scheme = parts[0]
718 authority = parts[1]
719 path = parts[2] or '/'
720 else:
721 # host or host:port
722 scheme = None
723 authority = uri
724 path = '/'
Georg Brandl13e89462008-07-01 19:56:00 +0000725 host, port = splitport(authority)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000726 if default_port and port is None and scheme is not None:
727 dport = {"http": 80,
728 "https": 443,
729 }.get(scheme)
730 if dport is not None:
731 authority = "%s:%d" % (host, dport)
732 return authority, path
733
734 def is_suburi(self, base, test):
735 """Check if test is below base in a URI tree
736
737 Both args must be URIs in reduced form.
738 """
739 if base == test:
740 return True
741 if base[0] != test[0]:
742 return False
743 common = posixpath.commonprefix((base[1], test[1]))
744 if len(common) == len(base[1]):
745 return True
746 return False
747
748
749class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
750
751 def find_user_password(self, realm, authuri):
752 user, password = HTTPPasswordMgr.find_user_password(self, realm,
753 authuri)
754 if user is not None:
755 return user, password
756 return HTTPPasswordMgr.find_user_password(self, None, authuri)
757
758
759class AbstractBasicAuthHandler:
760
761 # XXX this allows for multiple auth-schemes, but will stupidly pick
762 # the last one with a realm specified.
763
764 # allow for double- and single-quoted realm values
765 # (single quotes are a violation of the RFC, but appear in the wild)
766 rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+'
767 'realm=(["\'])(.*?)\\2', re.I)
768
769 # XXX could pre-emptively send auth info already accepted (RFC 2617,
770 # end of section 2, and section 1.2 immediately after "credentials"
771 # production).
772
773 def __init__(self, password_mgr=None):
774 if password_mgr is None:
775 password_mgr = HTTPPasswordMgr()
776 self.passwd = password_mgr
777 self.add_password = self.passwd.add_password
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000778 self.retried = 0
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000779
780 def http_error_auth_reqed(self, authreq, host, req, headers):
781 # host may be an authority (without userinfo) or a URL with an
782 # authority
783 # XXX could be multiple headers
784 authreq = headers.get(authreq, None)
Senthil Kumaranf4998ac2010-06-01 12:53:48 +0000785
786 if self.retried > 5:
787 # retry sending the username:password 5 times before failing.
788 raise HTTPError(req.get_full_url(), 401, "basic auth failed",
789 headers, None)
790 else:
791 self.retried += 1
792
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000793 if authreq:
794 mo = AbstractBasicAuthHandler.rx.search(authreq)
795 if mo:
796 scheme, quote, realm = mo.groups()
797 if scheme.lower() == 'basic':
798 return self.retry_http_basic_auth(host, req, realm)
799
800 def retry_http_basic_auth(self, host, req, realm):
801 user, pw = self.passwd.find_user_password(realm, host)
802 if pw is not None:
803 raw = "%s:%s" % (user, pw)
804 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
805 if req.headers.get(self.auth_header, None) == auth:
806 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000807 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000808 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000809 else:
810 return None
811
812
813class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
814
815 auth_header = 'Authorization'
816
817 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000818 url = req.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000819 return self.http_error_auth_reqed('www-authenticate',
820 url, req, headers)
821
822
823class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
824
825 auth_header = 'Proxy-authorization'
826
827 def http_error_407(self, req, fp, code, msg, headers):
828 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000829 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000830 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
831 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000832 authority = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000833 return self.http_error_auth_reqed('proxy-authenticate',
834 authority, req, headers)
835
836
837def randombytes(n):
838 """Return n random bytes."""
839 return os.urandom(n)
840
841class AbstractDigestAuthHandler:
842 # Digest authentication is specified in RFC 2617.
843
844 # XXX The client does not inspect the Authentication-Info header
845 # in a successful response.
846
847 # XXX It should be possible to test this implementation against
848 # a mock server that just generates a static set of challenges.
849
850 # XXX qop="auth-int" supports is shaky
851
852 def __init__(self, passwd=None):
853 if passwd is None:
854 passwd = HTTPPasswordMgr()
855 self.passwd = passwd
856 self.add_password = self.passwd.add_password
857 self.retried = 0
858 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000859 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000860
861 def reset_retry_count(self):
862 self.retried = 0
863
864 def http_error_auth_reqed(self, auth_header, host, req, headers):
865 authreq = headers.get(auth_header, None)
866 if self.retried > 5:
867 # Don't fail endlessly - if we failed once, we'll probably
868 # fail a second time. Hm. Unless the Password Manager is
869 # prompting for the information. Crap. This isn't great
870 # but it's better than the current 'repeat until recursion
871 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000872 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000873 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000874 else:
875 self.retried += 1
876 if authreq:
877 scheme = authreq.split()[0]
878 if scheme.lower() == 'digest':
879 return self.retry_http_digest_auth(req, authreq)
880
881 def retry_http_digest_auth(self, req, auth):
882 token, challenge = auth.split(' ', 1)
883 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
884 auth = self.get_authorization(req, chal)
885 if auth:
886 auth_val = 'Digest %s' % auth
887 if req.headers.get(self.auth_header, None) == auth_val:
888 return None
889 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000890 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000891 return resp
892
893 def get_cnonce(self, nonce):
894 # The cnonce-value is an opaque
895 # quoted string value provided by the client and used by both client
896 # and server to avoid chosen plaintext attacks, to provide mutual
897 # authentication, and to provide some message integrity protection.
898 # This isn't a fabulous effort, but it's probably Good Enough.
899 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
900 b = s.encode("ascii") + randombytes(8)
901 dig = hashlib.sha1(b).hexdigest()
902 return dig[:16]
903
904 def get_authorization(self, req, chal):
905 try:
906 realm = chal['realm']
907 nonce = chal['nonce']
908 qop = chal.get('qop')
909 algorithm = chal.get('algorithm', 'MD5')
910 # mod_digest doesn't send an opaque, even though it isn't
911 # supposed to be optional
912 opaque = chal.get('opaque', None)
913 except KeyError:
914 return None
915
916 H, KD = self.get_algorithm_impls(algorithm)
917 if H is None:
918 return None
919
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000920 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000921 if user is None:
922 return None
923
924 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000925 if req.data is not None:
926 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000927 else:
928 entdig = None
929
930 A1 = "%s:%s:%s" % (user, realm, pw)
931 A2 = "%s:%s" % (req.get_method(),
932 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000933 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000934 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000935 if nonce == self.last_nonce:
936 self.nonce_count += 1
937 else:
938 self.nonce_count = 1
939 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000940 ncvalue = '%08x' % self.nonce_count
941 cnonce = self.get_cnonce(nonce)
942 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
943 respdig = KD(H(A1), noncebit)
944 elif qop is None:
945 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
946 else:
947 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +0000948 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000949
950 # XXX should the partial digests be encoded too?
951
952 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000953 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000954 respdig)
955 if opaque:
956 base += ', opaque="%s"' % opaque
957 if entdig:
958 base += ', digest="%s"' % entdig
959 base += ', algorithm="%s"' % algorithm
960 if qop:
961 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
962 return base
963
964 def get_algorithm_impls(self, algorithm):
965 # lambdas assume digest modules are imported at the top level
966 if algorithm == 'MD5':
967 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
968 elif algorithm == 'SHA':
969 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
970 # XXX MD5-sess
971 KD = lambda s, d: H("%s:%s" % (s, d))
972 return H, KD
973
974 def get_entity_digest(self, data, chal):
975 # XXX not implemented yet
976 return None
977
978
979class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
980 """An authentication protocol defined by RFC 2069
981
982 Digest authentication improves on basic authentication because it
983 does not transmit passwords in the clear.
984 """
985
986 auth_header = 'Authorization'
987 handler_order = 490 # before Basic auth
988
989 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000990 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000991 retry = self.http_error_auth_reqed('www-authenticate',
992 host, req, headers)
993 self.reset_retry_count()
994 return retry
995
996
997class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
998
999 auth_header = 'Proxy-Authorization'
1000 handler_order = 490 # before Basic auth
1001
1002 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001003 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001004 retry = self.http_error_auth_reqed('proxy-authenticate',
1005 host, req, headers)
1006 self.reset_retry_count()
1007 return retry
1008
1009class AbstractHTTPHandler(BaseHandler):
1010
1011 def __init__(self, debuglevel=0):
1012 self._debuglevel = debuglevel
1013
1014 def set_http_debuglevel(self, level):
1015 self._debuglevel = level
1016
1017 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001018 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001019 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001020 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001021
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001022 if request.data is not None: # POST
1023 data = request.data
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001024 if not request.has_header('Content-type'):
1025 request.add_unredirected_header(
1026 'Content-type',
1027 'application/x-www-form-urlencoded')
1028 if not request.has_header('Content-length'):
1029 request.add_unredirected_header(
1030 'Content-length', '%d' % len(data))
1031
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001032 sel_host = host
1033 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001034 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001035 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001036 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001037 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001038 for name, value in self.parent.addheaders:
1039 name = name.capitalize()
1040 if not request.has_header(name):
1041 request.add_unredirected_header(name, value)
1042
1043 return request
1044
1045 def do_open(self, http_class, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001046 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001047
1048 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001049 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001050 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001051 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001052 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001053
1054 h = http_class(host, timeout=req.timeout) # will parse host:port
1055 headers = dict(req.headers)
1056 headers.update(req.unredirected_hdrs)
1057
1058 # TODO(jhylton): Should this be redesigned to handle
1059 # persistent connections?
1060
1061 # We want to make an HTTP/1.1 request, but the addinfourl
1062 # class isn't prepared to deal with a persistent connection.
1063 # It will try to read all remaining data from the socket,
1064 # which will block while the server waits for the next request.
1065 # So make sure the connection gets closed after the (only)
1066 # request.
1067 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001068 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001069
1070 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001071 tunnel_headers = {}
1072 proxy_auth_hdr = "Proxy-Authorization"
1073 if proxy_auth_hdr in headers:
1074 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1075 # Proxy-Authorization should not be sent to origin
1076 # server.
1077 del headers[proxy_auth_hdr]
1078 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001079
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001080 try:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001081 h.request(req.get_method(), req.selector, req.data, headers)
1082 r = h.getresponse() # an HTTPResponse instance
1083 except socket.error as err:
Georg Brandl13e89462008-07-01 19:56:00 +00001084 raise URLError(err)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001085
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001086 r.url = req.full_url
1087 # This line replaces the .msg attribute of the HTTPResponse
1088 # with .headers, because urllib clients expect the response to
1089 # have the reason in .msg. It would be good to mark this
1090 # attribute is deprecated and get then to use info() or
1091 # .headers.
1092 r.msg = r.reason
1093 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001094
1095
1096class HTTPHandler(AbstractHTTPHandler):
1097
1098 def http_open(self, req):
1099 return self.do_open(http.client.HTTPConnection, req)
1100
1101 http_request = AbstractHTTPHandler.do_request_
1102
1103if hasattr(http.client, 'HTTPSConnection'):
1104 class HTTPSHandler(AbstractHTTPHandler):
1105
1106 def https_open(self, req):
1107 return self.do_open(http.client.HTTPSConnection, req)
1108
1109 https_request = AbstractHTTPHandler.do_request_
1110
1111class HTTPCookieProcessor(BaseHandler):
1112 def __init__(self, cookiejar=None):
1113 import http.cookiejar
1114 if cookiejar is None:
1115 cookiejar = http.cookiejar.CookieJar()
1116 self.cookiejar = cookiejar
1117
1118 def http_request(self, request):
1119 self.cookiejar.add_cookie_header(request)
1120 return request
1121
1122 def http_response(self, request, response):
1123 self.cookiejar.extract_cookies(response, request)
1124 return response
1125
1126 https_request = http_request
1127 https_response = http_response
1128
1129class UnknownHandler(BaseHandler):
1130 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001131 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001132 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001133
1134def parse_keqv_list(l):
1135 """Parse list of key=value strings where keys are not duplicated."""
1136 parsed = {}
1137 for elt in l:
1138 k, v = elt.split('=', 1)
1139 if v[0] == '"' and v[-1] == '"':
1140 v = v[1:-1]
1141 parsed[k] = v
1142 return parsed
1143
1144def parse_http_list(s):
1145 """Parse lists as described by RFC 2068 Section 2.
1146
1147 In particular, parse comma-separated lists where the elements of
1148 the list may include quoted-strings. A quoted-string could
1149 contain a comma. A non-quoted string could have quotes in the
1150 middle. Neither commas nor quotes count if they are escaped.
1151 Only double-quotes count, not single-quotes.
1152 """
1153 res = []
1154 part = ''
1155
1156 escape = quote = False
1157 for cur in s:
1158 if escape:
1159 part += cur
1160 escape = False
1161 continue
1162 if quote:
1163 if cur == '\\':
1164 escape = True
1165 continue
1166 elif cur == '"':
1167 quote = False
1168 part += cur
1169 continue
1170
1171 if cur == ',':
1172 res.append(part)
1173 part = ''
1174 continue
1175
1176 if cur == '"':
1177 quote = True
1178
1179 part += cur
1180
1181 # append last part
1182 if part:
1183 res.append(part)
1184
1185 return [part.strip() for part in res]
1186
1187class FileHandler(BaseHandler):
1188 # Use local file or FTP depending on form of URL
1189 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001190 url = req.selector
Senthil Kumaran2ef16322010-07-11 03:12:43 +00001191 if url[:2] == '//' and url[2:3] != '/' and (req.host and
1192 req.host != 'localhost'):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001193 req.type = 'ftp'
1194 return self.parent.open(req)
1195 else:
1196 return self.open_local_file(req)
1197
1198 # names for the localhost
1199 names = None
1200 def get_names(self):
1201 if FileHandler.names is None:
1202 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001203 FileHandler.names = tuple(
1204 socket.gethostbyname_ex('localhost')[2] +
1205 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001206 except socket.gaierror:
1207 FileHandler.names = (socket.gethostbyname('localhost'),)
1208 return FileHandler.names
1209
1210 # not entirely sure what the rules are here
1211 def open_local_file(self, req):
1212 import email.utils
1213 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001214 host = req.host
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001215 filename = req.selector
1216 localfile = url2pathname(filename)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001217 try:
1218 stats = os.stat(localfile)
1219 size = stats.st_size
1220 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001221 mtype = mimetypes.guess_type(filename)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001222 headers = email.message_from_string(
1223 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1224 (mtype or 'text/plain', size, modified))
1225 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001226 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001227 if not host or \
1228 (not port and _safe_gethostbyname(host) in self.get_names()):
Senthil Kumaran06f5a532010-05-08 05:12:05 +00001229 if host:
1230 origurl = 'file://' + host + filename
1231 else:
1232 origurl = 'file://' + filename
1233 return addinfourl(open(localfile, 'rb'), headers, origurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001234 except OSError as msg:
Georg Brandl029986a2008-06-23 11:44:14 +00001235 # users shouldn't expect OSErrors coming from urlopen()
Georg Brandl13e89462008-07-01 19:56:00 +00001236 raise URLError(msg)
1237 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001238
1239def _safe_gethostbyname(host):
1240 try:
1241 return socket.gethostbyname(host)
1242 except socket.gaierror:
1243 return None
1244
1245class FTPHandler(BaseHandler):
1246 def ftp_open(self, req):
1247 import ftplib
1248 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001249 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001250 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001251 raise URLError('ftp error: no host given')
1252 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001253 if port is None:
1254 port = ftplib.FTP_PORT
1255 else:
1256 port = int(port)
1257
1258 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001259 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001260 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001261 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001262 else:
1263 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001264 host = unquote(host)
1265 user = unquote(user or '')
1266 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001267
1268 try:
1269 host = socket.gethostbyname(host)
1270 except socket.error as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001271 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001272 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001273 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001274 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001275 dirs, file = dirs[:-1], dirs[-1]
1276 if dirs and not dirs[0]:
1277 dirs = dirs[1:]
1278 try:
1279 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1280 type = file and 'I' or 'D'
1281 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001282 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001283 if attr.lower() == 'type' and \
1284 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1285 type = value.upper()
1286 fp, retrlen = fw.retrfile(file, type)
1287 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001288 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001289 if mtype:
1290 headers += "Content-type: %s\n" % mtype
1291 if retrlen is not None and retrlen >= 0:
1292 headers += "Content-length: %d\n" % retrlen
1293 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001294 return addinfourl(fp, headers, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001295 except ftplib.all_errors as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001296 exc = URLError('ftp error: %s' % msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001297 raise exc.with_traceback(sys.exc_info()[2])
1298
1299 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1300 fw = ftpwrapper(user, passwd, host, port, dirs, timeout)
1301 return fw
1302
1303class CacheFTPHandler(FTPHandler):
1304 # XXX would be nice to have pluggable cache strategies
1305 # XXX this stuff is definitely not thread safe
1306 def __init__(self):
1307 self.cache = {}
1308 self.timeout = {}
1309 self.soonest = 0
1310 self.delay = 60
1311 self.max_conns = 16
1312
1313 def setTimeout(self, t):
1314 self.delay = t
1315
1316 def setMaxConns(self, m):
1317 self.max_conns = m
1318
1319 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1320 key = user, host, port, '/'.join(dirs), timeout
1321 if key in self.cache:
1322 self.timeout[key] = time.time() + self.delay
1323 else:
1324 self.cache[key] = ftpwrapper(user, passwd, host, port,
1325 dirs, timeout)
1326 self.timeout[key] = time.time() + self.delay
1327 self.check_cache()
1328 return self.cache[key]
1329
1330 def check_cache(self):
1331 # first check for old ones
1332 t = time.time()
1333 if self.soonest <= t:
1334 for k, v in list(self.timeout.items()):
1335 if v < t:
1336 self.cache[k].close()
1337 del self.cache[k]
1338 del self.timeout[k]
1339 self.soonest = min(list(self.timeout.values()))
1340
1341 # then check the size
1342 if len(self.cache) == self.max_conns:
1343 for k, v in list(self.timeout.items()):
1344 if v == self.soonest:
1345 del self.cache[k]
1346 del self.timeout[k]
1347 break
1348 self.soonest = min(list(self.timeout.values()))
1349
1350# Code move from the old urllib module
1351
1352MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1353
1354# Helper for non-unix systems
Ronald Oussoren94f25282010-05-05 19:11:21 +00001355if os.name == 'nt':
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001356 from nturl2path import url2pathname, pathname2url
1357else:
1358 def url2pathname(pathname):
1359 """OS-specific conversion from a relative URL of the 'file' scheme
1360 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001361 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001362
1363 def pathname2url(pathname):
1364 """OS-specific conversion from a file system path to a relative URL
1365 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001366 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001367
1368# This really consists of two pieces:
1369# (1) a class which handles opening of all sorts of URLs
1370# (plus assorted utilities etc.)
1371# (2) a set of functions for parsing URLs
1372# XXX Should these be separated out into different modules?
1373
1374
1375ftpcache = {}
1376class URLopener:
1377 """Class to open URLs.
1378 This is a class rather than just a subroutine because we may need
1379 more than one set of global protocol-specific options.
1380 Note -- this is a base class for those who don't want the
1381 automatic handling of errors type 302 (relocated) and 401
1382 (authorization needed)."""
1383
1384 __tempfiles = None
1385
1386 version = "Python-urllib/%s" % __version__
1387
1388 # Constructor
1389 def __init__(self, proxies=None, **x509):
1390 if proxies is None:
1391 proxies = getproxies()
1392 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1393 self.proxies = proxies
1394 self.key_file = x509.get('key_file')
1395 self.cert_file = x509.get('cert_file')
1396 self.addheaders = [('User-Agent', self.version)]
1397 self.__tempfiles = []
1398 self.__unlink = os.unlink # See cleanup()
1399 self.tempcache = None
1400 # Undocumented feature: if you assign {} to tempcache,
1401 # it is used to cache files retrieved with
1402 # self.retrieve(). This is not enabled by default
1403 # since it does not work for changing documents (and I
1404 # haven't got the logic to check expiration headers
1405 # yet).
1406 self.ftpcache = ftpcache
1407 # Undocumented feature: you can use a different
1408 # ftp cache by assigning to the .ftpcache member;
1409 # in case you want logically independent URL openers
1410 # XXX This is not threadsafe. Bah.
1411
1412 def __del__(self):
1413 self.close()
1414
1415 def close(self):
1416 self.cleanup()
1417
1418 def cleanup(self):
1419 # This code sometimes runs when the rest of this module
1420 # has already been deleted, so it can't use any globals
1421 # or import anything.
1422 if self.__tempfiles:
1423 for file in self.__tempfiles:
1424 try:
1425 self.__unlink(file)
1426 except OSError:
1427 pass
1428 del self.__tempfiles[:]
1429 if self.tempcache:
1430 self.tempcache.clear()
1431
1432 def addheader(self, *args):
1433 """Add a header to be used by the HTTP interface only
1434 e.g. u.addheader('Accept', 'sound/basic')"""
1435 self.addheaders.append(args)
1436
1437 # External interface
1438 def open(self, fullurl, data=None):
1439 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001440 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001441 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001442 if self.tempcache and fullurl in self.tempcache:
1443 filename, headers = self.tempcache[fullurl]
1444 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001445 return addinfourl(fp, headers, fullurl)
1446 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001447 if not urltype:
1448 urltype = 'file'
1449 if urltype in self.proxies:
1450 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001451 urltype, proxyhost = splittype(proxy)
1452 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001453 url = (host, fullurl) # Signal special case to open_*()
1454 else:
1455 proxy = None
1456 name = 'open_' + urltype
1457 self.type = urltype
1458 name = name.replace('-', '_')
1459 if not hasattr(self, name):
1460 if proxy:
1461 return self.open_unknown_proxy(proxy, fullurl, data)
1462 else:
1463 return self.open_unknown(fullurl, data)
1464 try:
1465 if data is None:
1466 return getattr(self, name)(url)
1467 else:
1468 return getattr(self, name)(url, data)
1469 except socket.error as msg:
1470 raise IOError('socket error', msg).with_traceback(sys.exc_info()[2])
1471
1472 def open_unknown(self, fullurl, data=None):
1473 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001474 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001475 raise IOError('url error', 'unknown url type', type)
1476
1477 def open_unknown_proxy(self, proxy, fullurl, data=None):
1478 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001479 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001480 raise IOError('url error', 'invalid proxy for %s' % type, proxy)
1481
1482 # External interface
1483 def retrieve(self, url, filename=None, reporthook=None, data=None):
1484 """retrieve(url) returns (filename, headers) for a local object
1485 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001486 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001487 if self.tempcache and url in self.tempcache:
1488 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001489 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001490 if filename is None and (not type or type == 'file'):
1491 try:
1492 fp = self.open_local_file(url1)
1493 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001494 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001495 return url2pathname(splithost(url1)[1]), hdrs
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001496 except IOError as msg:
1497 pass
1498 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001499 try:
1500 headers = fp.info()
1501 if filename:
1502 tfp = open(filename, 'wb')
1503 else:
1504 import tempfile
1505 garbage, path = splittype(url)
1506 garbage, path = splithost(path or "")
1507 path, garbage = splitquery(path or "")
1508 path, garbage = splitattr(path or "")
1509 suffix = os.path.splitext(path)[1]
1510 (fd, filename) = tempfile.mkstemp(suffix)
1511 self.__tempfiles.append(filename)
1512 tfp = os.fdopen(fd, 'wb')
1513 try:
1514 result = filename, headers
1515 if self.tempcache is not None:
1516 self.tempcache[url] = result
1517 bs = 1024*8
1518 size = -1
1519 read = 0
1520 blocknum = 0
1521 if reporthook:
1522 if "content-length" in headers:
1523 size = int(headers["Content-Length"])
1524 reporthook(blocknum, bs, size)
1525 while 1:
1526 block = fp.read(bs)
1527 if not block:
1528 break
1529 read += len(block)
1530 tfp.write(block)
1531 blocknum += 1
1532 if reporthook:
1533 reporthook(blocknum, bs, size)
1534 finally:
1535 tfp.close()
1536 finally:
1537 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001538
1539 # raise exception if actual size does not match content-length header
1540 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001541 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001542 "retrieval incomplete: got only %i out of %i bytes"
1543 % (read, size), result)
1544
1545 return result
1546
1547 # Each method named open_<type> knows how to open that type of URL
1548
1549 def _open_generic_http(self, connection_factory, url, data):
1550 """Make an HTTP connection using connection_class.
1551
1552 This is an internal method that should be called from
1553 open_http() or open_https().
1554
1555 Arguments:
1556 - connection_factory should take a host name and return an
1557 HTTPConnection instance.
1558 - url is the url to retrieval or a host, relative-path pair.
1559 - data is payload for a POST request or None.
1560 """
1561
1562 user_passwd = None
1563 proxy_passwd= None
1564 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001565 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001566 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001567 user_passwd, host = splituser(host)
1568 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001569 realhost = host
1570 else:
1571 host, selector = url
1572 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001573 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001574 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001575 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001576 url = rest
1577 user_passwd = None
1578 if urltype.lower() != 'http':
1579 realhost = None
1580 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001581 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001582 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001583 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001584 if user_passwd:
1585 selector = "%s://%s%s" % (urltype, realhost, rest)
1586 if proxy_bypass(realhost):
1587 host = realhost
1588
1589 #print "proxy via http:", host, selector
1590 if not host: raise IOError('http error', 'no host given')
1591
1592 if proxy_passwd:
1593 import base64
1594 proxy_auth = base64.b64encode(proxy_passwd).strip()
1595 else:
1596 proxy_auth = None
1597
1598 if user_passwd:
1599 import base64
1600 auth = base64.b64encode(user_passwd).strip()
1601 else:
1602 auth = None
1603 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001604 headers = {}
1605 if proxy_auth:
1606 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1607 if auth:
1608 headers["Authorization"] = "Basic %s" % auth
1609 if realhost:
1610 headers["Host"] = realhost
1611 for header, value in self.addheaders:
1612 headers[header] = value
1613
1614 if data is not None:
1615 headers["Content-Type"] = "application/x-www-form-urlencoded"
1616 http_conn.request("POST", selector, data, headers)
1617 else:
1618 http_conn.request("GET", selector, headers=headers)
1619
1620 try:
1621 response = http_conn.getresponse()
1622 except http.client.BadStatusLine:
1623 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001624 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001625
1626 # According to RFC 2616, "2xx" code indicates that the client's
1627 # request was successfully received, understood, and accepted.
1628 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001629 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001630 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001631 else:
1632 return self.http_error(
1633 url, response.fp,
1634 response.status, response.reason, response.msg, data)
1635
1636 def open_http(self, url, data=None):
1637 """Use HTTP protocol."""
1638 return self._open_generic_http(http.client.HTTPConnection, url, data)
1639
1640 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1641 """Handle http errors.
1642
1643 Derived class can override this, or provide specific handlers
1644 named http_error_DDD where DDD is the 3-digit error code."""
1645 # First check if there's a specific handler for this error
1646 name = 'http_error_%d' % errcode
1647 if hasattr(self, name):
1648 method = getattr(self, name)
1649 if data is None:
1650 result = method(url, fp, errcode, errmsg, headers)
1651 else:
1652 result = method(url, fp, errcode, errmsg, headers, data)
1653 if result: return result
1654 return self.http_error_default(url, fp, errcode, errmsg, headers)
1655
1656 def http_error_default(self, url, fp, errcode, errmsg, headers):
1657 """Default error handler: close the connection and raise IOError."""
1658 void = fp.read()
1659 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001660 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001661
1662 if _have_ssl:
1663 def _https_connection(self, host):
1664 return http.client.HTTPSConnection(host,
1665 key_file=self.key_file,
1666 cert_file=self.cert_file)
1667
1668 def open_https(self, url, data=None):
1669 """Use HTTPS protocol."""
1670 return self._open_generic_http(self._https_connection, url, data)
1671
1672 def open_file(self, url):
1673 """Use local file or FTP depending on form of URL."""
1674 if not isinstance(url, str):
1675 raise URLError('file error', 'proxy support for file protocol currently not implemented')
1676 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
1677 return self.open_ftp(url)
1678 else:
1679 return self.open_local_file(url)
1680
1681 def open_local_file(self, url):
1682 """Use local file."""
1683 import mimetypes, email.utils
1684 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001685 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001686 localname = url2pathname(file)
1687 try:
1688 stats = os.stat(localname)
1689 except OSError as e:
1690 raise URLError(e.errno, e.strerror, e.filename)
1691 size = stats.st_size
1692 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1693 mtype = mimetypes.guess_type(url)[0]
1694 headers = email.message_from_string(
1695 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1696 (mtype or 'text/plain', size, modified))
1697 if not host:
1698 urlfile = file
1699 if file[:1] == '/':
1700 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001701 return addinfourl(open(localname, 'rb'), headers, urlfile)
1702 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001703 if (not port
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001704 and socket.gethostbyname(host) in (localhost() + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001705 urlfile = file
1706 if file[:1] == '/':
1707 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001708 return addinfourl(open(localname, 'rb'), headers, urlfile)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001709 raise URLError('local file error', 'not on local host')
1710
1711 def open_ftp(self, url):
1712 """Use FTP protocol."""
1713 if not isinstance(url, str):
1714 raise URLError('ftp error', 'proxy support for ftp protocol currently not implemented')
1715 import mimetypes
1716 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001717 host, path = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001718 if not host: raise URLError('ftp error', 'no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001719 host, port = splitport(host)
1720 user, host = splituser(host)
1721 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001722 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001723 host = unquote(host)
1724 user = unquote(user or '')
1725 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001726 host = socket.gethostbyname(host)
1727 if not port:
1728 import ftplib
1729 port = ftplib.FTP_PORT
1730 else:
1731 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001732 path, attrs = splitattr(path)
1733 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001734 dirs = path.split('/')
1735 dirs, file = dirs[:-1], dirs[-1]
1736 if dirs and not dirs[0]: dirs = dirs[1:]
1737 if dirs and not dirs[0]: dirs[0] = '/'
1738 key = user, host, port, '/'.join(dirs)
1739 # XXX thread unsafe!
1740 if len(self.ftpcache) > MAXFTPCACHE:
1741 # Prune the cache, rather arbitrarily
1742 for k in self.ftpcache.keys():
1743 if k != key:
1744 v = self.ftpcache[k]
1745 del self.ftpcache[k]
1746 v.close()
1747 try:
1748 if not key in self.ftpcache:
1749 self.ftpcache[key] = \
1750 ftpwrapper(user, passwd, host, port, dirs)
1751 if not file: type = 'D'
1752 else: type = 'I'
1753 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001754 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001755 if attr.lower() == 'type' and \
1756 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1757 type = value.upper()
1758 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1759 mtype = mimetypes.guess_type("ftp:" + url)[0]
1760 headers = ""
1761 if mtype:
1762 headers += "Content-Type: %s\n" % mtype
1763 if retrlen is not None and retrlen >= 0:
1764 headers += "Content-Length: %d\n" % retrlen
1765 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001766 return addinfourl(fp, headers, "ftp:" + url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001767 except ftperrors() as msg:
1768 raise URLError('ftp error', msg).with_traceback(sys.exc_info()[2])
1769
1770 def open_data(self, url, data=None):
1771 """Use "data" URL."""
1772 if not isinstance(url, str):
1773 raise URLError('data error', 'proxy support for data protocol currently not implemented')
1774 # ignore POSTed data
1775 #
1776 # syntax of data URLs:
1777 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1778 # mediatype := [ type "/" subtype ] *( ";" parameter )
1779 # data := *urlchar
1780 # parameter := attribute "=" value
1781 try:
1782 [type, data] = url.split(',', 1)
1783 except ValueError:
1784 raise IOError('data error', 'bad data URL')
1785 if not type:
1786 type = 'text/plain;charset=US-ASCII'
1787 semi = type.rfind(';')
1788 if semi >= 0 and '=' not in type[semi:]:
1789 encoding = type[semi+1:]
1790 type = type[:semi]
1791 else:
1792 encoding = ''
1793 msg = []
Senthil Kumaranf6c456d2010-05-01 08:29:18 +00001794 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001795 time.gmtime(time.time())))
1796 msg.append('Content-type: %s' % type)
1797 if encoding == 'base64':
1798 import base64
Georg Brandl706824f2009-06-04 09:42:55 +00001799 # XXX is this encoding/decoding ok?
1800 data = base64.decodebytes(data.encode('ascii')).decode('latin1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001801 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001802 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001803 msg.append('Content-Length: %d' % len(data))
1804 msg.append('')
1805 msg.append(data)
1806 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001807 headers = email.message_from_string(msg)
1808 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001809 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001810 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001811
1812
1813class FancyURLopener(URLopener):
1814 """Derived class with handlers for errors we can handle (perhaps)."""
1815
1816 def __init__(self, *args, **kwargs):
1817 URLopener.__init__(self, *args, **kwargs)
1818 self.auth_cache = {}
1819 self.tries = 0
1820 self.maxtries = 10
1821
1822 def http_error_default(self, url, fp, errcode, errmsg, headers):
1823 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001824 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001825
1826 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1827 """Error 302 -- relocated (temporarily)."""
1828 self.tries += 1
1829 if self.maxtries and self.tries >= self.maxtries:
1830 if hasattr(self, "http_error_500"):
1831 meth = self.http_error_500
1832 else:
1833 meth = self.http_error_default
1834 self.tries = 0
1835 return meth(url, fp, 500,
1836 "Internal Server Error: Redirect Recursion", headers)
1837 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
1838 data)
1839 self.tries = 0
1840 return result
1841
1842 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
1843 if 'location' in headers:
1844 newurl = headers['location']
1845 elif 'uri' in headers:
1846 newurl = headers['uri']
1847 else:
1848 return
1849 void = fp.read()
1850 fp.close()
1851 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00001852 newurl = urljoin(self.type + ":" + url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001853 return self.open(newurl)
1854
1855 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
1856 """Error 301 -- also relocated (permanently)."""
1857 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1858
1859 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
1860 """Error 303 -- also relocated (essentially identical to 302)."""
1861 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1862
1863 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
1864 """Error 307 -- relocated, but turn POST into error."""
1865 if data is None:
1866 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1867 else:
1868 return self.http_error_default(url, fp, errcode, errmsg, headers)
1869
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001870 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None,
1871 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001872 """Error 401 -- authentication required.
1873 This function supports Basic authentication only."""
1874 if not 'www-authenticate' in headers:
1875 URLopener.http_error_default(self, url, fp,
1876 errcode, errmsg, headers)
1877 stuff = headers['www-authenticate']
1878 import re
1879 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1880 if not match:
1881 URLopener.http_error_default(self, url, fp,
1882 errcode, errmsg, headers)
1883 scheme, realm = match.groups()
1884 if scheme.lower() != 'basic':
1885 URLopener.http_error_default(self, url, fp,
1886 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001887 if not retry:
1888 URLopener.http_error_default(self, url, fp, errcode, errmsg,
1889 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001890 name = 'retry_' + self.type + '_basic_auth'
1891 if data is None:
1892 return getattr(self,name)(url, realm)
1893 else:
1894 return getattr(self,name)(url, realm, data)
1895
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001896 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None,
1897 retry=False):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001898 """Error 407 -- proxy authentication required.
1899 This function supports Basic authentication only."""
1900 if not 'proxy-authenticate' in headers:
1901 URLopener.http_error_default(self, url, fp,
1902 errcode, errmsg, headers)
1903 stuff = headers['proxy-authenticate']
1904 import re
1905 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1906 if not match:
1907 URLopener.http_error_default(self, url, fp,
1908 errcode, errmsg, headers)
1909 scheme, realm = match.groups()
1910 if scheme.lower() != 'basic':
1911 URLopener.http_error_default(self, url, fp,
1912 errcode, errmsg, headers)
Senthil Kumaran80f1b052010-06-18 15:08:18 +00001913 if not retry:
1914 URLopener.http_error_default(self, url, fp, errcode, errmsg,
1915 headers)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001916 name = 'retry_proxy_' + self.type + '_basic_auth'
1917 if data is None:
1918 return getattr(self,name)(url, realm)
1919 else:
1920 return getattr(self,name)(url, realm, data)
1921
1922 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001923 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001924 newurl = 'http://' + host + selector
1925 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00001926 urltype, proxyhost = splittype(proxy)
1927 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001928 i = proxyhost.find('@') + 1
1929 proxyhost = proxyhost[i:]
1930 user, passwd = self.get_user_passwd(proxyhost, realm, i)
1931 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001932 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001933 quote(passwd, safe=''), proxyhost)
1934 self.proxies['http'] = 'http://' + proxyhost + proxyselector
1935 if data is None:
1936 return self.open(newurl)
1937 else:
1938 return self.open(newurl, data)
1939
1940 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001941 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001942 newurl = 'https://' + host + selector
1943 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00001944 urltype, proxyhost = splittype(proxy)
1945 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001946 i = proxyhost.find('@') + 1
1947 proxyhost = proxyhost[i:]
1948 user, passwd = self.get_user_passwd(proxyhost, realm, i)
1949 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001950 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001951 quote(passwd, safe=''), proxyhost)
1952 self.proxies['https'] = 'https://' + proxyhost + proxyselector
1953 if data is None:
1954 return self.open(newurl)
1955 else:
1956 return self.open(newurl, data)
1957
1958 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001959 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001960 i = host.find('@') + 1
1961 host = host[i:]
1962 user, passwd = self.get_user_passwd(host, realm, i)
1963 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001964 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001965 quote(passwd, safe=''), host)
1966 newurl = 'http://' + host + selector
1967 if data is None:
1968 return self.open(newurl)
1969 else:
1970 return self.open(newurl, data)
1971
1972 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001973 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001974 i = host.find('@') + 1
1975 host = host[i:]
1976 user, passwd = self.get_user_passwd(host, realm, i)
1977 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001978 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001979 quote(passwd, safe=''), host)
1980 newurl = 'https://' + host + selector
1981 if data is None:
1982 return self.open(newurl)
1983 else:
1984 return self.open(newurl, data)
1985
Florent Xicluna757445b2010-05-17 17:24:07 +00001986 def get_user_passwd(self, host, realm, clear_cache=0):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001987 key = realm + '@' + host.lower()
1988 if key in self.auth_cache:
1989 if clear_cache:
1990 del self.auth_cache[key]
1991 else:
1992 return self.auth_cache[key]
1993 user, passwd = self.prompt_user_passwd(host, realm)
1994 if user or passwd: self.auth_cache[key] = (user, passwd)
1995 return user, passwd
1996
1997 def prompt_user_passwd(self, host, realm):
1998 """Override this in a GUI environment!"""
1999 import getpass
2000 try:
2001 user = input("Enter username for %s at %s: " % (realm, host))
2002 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
2003 (user, realm, host))
2004 return user, passwd
2005 except KeyboardInterrupt:
2006 print()
2007 return None, None
2008
2009
2010# Utility functions
2011
2012_localhost = None
2013def localhost():
2014 """Return the IP address of the magic hostname 'localhost'."""
2015 global _localhost
2016 if _localhost is None:
2017 _localhost = socket.gethostbyname('localhost')
2018 return _localhost
2019
2020_thishost = None
2021def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002022 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002023 global _thishost
2024 if _thishost is None:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002025 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname()[2]))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002026 return _thishost
2027
2028_ftperrors = None
2029def ftperrors():
2030 """Return the set of errors raised by the FTP class."""
2031 global _ftperrors
2032 if _ftperrors is None:
2033 import ftplib
2034 _ftperrors = ftplib.all_errors
2035 return _ftperrors
2036
2037_noheaders = None
2038def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002039 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002040 global _noheaders
2041 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002042 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002043 return _noheaders
2044
2045
2046# Utility classes
2047
2048class ftpwrapper:
2049 """Class used by open_ftp() for cache of open FTP connections."""
2050
2051 def __init__(self, user, passwd, host, port, dirs, timeout=None):
2052 self.user = user
2053 self.passwd = passwd
2054 self.host = host
2055 self.port = port
2056 self.dirs = dirs
2057 self.timeout = timeout
2058 self.init()
2059
2060 def init(self):
2061 import ftplib
2062 self.busy = 0
2063 self.ftp = ftplib.FTP()
2064 self.ftp.connect(self.host, self.port, self.timeout)
2065 self.ftp.login(self.user, self.passwd)
2066 for dir in self.dirs:
2067 self.ftp.cwd(dir)
2068
2069 def retrfile(self, file, type):
2070 import ftplib
2071 self.endtransfer()
2072 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2073 else: cmd = 'TYPE ' + type; isdir = 0
2074 try:
2075 self.ftp.voidcmd(cmd)
2076 except ftplib.all_errors:
2077 self.init()
2078 self.ftp.voidcmd(cmd)
2079 conn = None
2080 if file and not isdir:
2081 # Try to retrieve as a file
2082 try:
2083 cmd = 'RETR ' + file
2084 conn = self.ftp.ntransfercmd(cmd)
2085 except ftplib.error_perm as reason:
2086 if str(reason)[:3] != '550':
Georg Brandl13e89462008-07-01 19:56:00 +00002087 raise URLError('ftp error', reason).with_traceback(
2088 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002089 if not conn:
2090 # Set transfer mode to ASCII!
2091 self.ftp.voidcmd('TYPE A')
2092 # Try a directory listing. Verify that directory exists.
2093 if file:
2094 pwd = self.ftp.pwd()
2095 try:
2096 try:
2097 self.ftp.cwd(file)
2098 except ftplib.error_perm as reason:
Georg Brandl13e89462008-07-01 19:56:00 +00002099 raise URLError('ftp error', reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002100 finally:
2101 self.ftp.cwd(pwd)
2102 cmd = 'LIST ' + file
2103 else:
2104 cmd = 'LIST'
2105 conn = self.ftp.ntransfercmd(cmd)
2106 self.busy = 1
2107 # Pass back both a suitably decorated object and a retrieval length
Georg Brandl13e89462008-07-01 19:56:00 +00002108 return (addclosehook(conn[0].makefile('rb'), self.endtransfer), conn[1])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002109 def endtransfer(self):
2110 if not self.busy:
2111 return
2112 self.busy = 0
2113 try:
2114 self.ftp.voidresp()
2115 except ftperrors():
2116 pass
2117
2118 def close(self):
2119 self.endtransfer()
2120 try:
2121 self.ftp.close()
2122 except ftperrors():
2123 pass
2124
2125# Proxy handling
2126def getproxies_environment():
2127 """Return a dictionary of scheme -> proxy server URL mappings.
2128
2129 Scan the environment for variables named <scheme>_proxy;
2130 this seems to be the standard convention. If you need a
2131 different way, you can pass a proxies dictionary to the
2132 [Fancy]URLopener constructor.
2133
2134 """
2135 proxies = {}
2136 for name, value in os.environ.items():
2137 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002138 if value and name[-6:] == '_proxy':
2139 proxies[name[:-6]] = value
2140 return proxies
2141
2142def proxy_bypass_environment(host):
2143 """Test if proxies should not be used for a particular host.
2144
2145 Checks the environment for a variable named no_proxy, which should
2146 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2147 """
2148 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2149 # '*' is special case for always bypass
2150 if no_proxy == '*':
2151 return 1
2152 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002153 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002154 # check if the host ends with any of the DNS suffixes
2155 for name in no_proxy.split(','):
2156 if name and (hostonly.endswith(name) or host.endswith(name)):
2157 return 1
2158 # otherwise, don't bypass
2159 return 0
2160
2161
2162if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002163 from _scproxy import _get_proxy_settings, _get_proxies
2164
2165 def proxy_bypass_macosx_sysconf(host):
2166 """
2167 Return True iff this host shouldn't be accessed using a proxy
2168
2169 This function uses the MacOSX framework SystemConfiguration
2170 to fetch the proxy information.
2171 """
2172 import re
2173 import socket
2174 from fnmatch import fnmatch
2175
2176 hostonly, port = splitport(host)
2177
2178 def ip2num(ipAddr):
2179 parts = ipAddr.split('.')
Mark Dickinsonc3f45c22010-05-09 12:16:29 +00002180 parts = list(map(int, parts))
Ronald Oussoren84151202010-04-18 20:46:11 +00002181 if len(parts) != 4:
2182 parts = (parts + [0, 0, 0, 0])[:4]
2183 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2184
2185 proxy_settings = _get_proxy_settings()
2186
2187 # Check for simple host names:
2188 if '.' not in host:
2189 if proxy_settings['exclude_simple']:
2190 return True
2191
2192 hostIP = None
2193
2194 for value in proxy_settings.get('exceptions', ()):
2195 # Items in the list are strings like these: *.local, 169.254/16
2196 if not value: continue
2197
2198 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2199 if m is not None:
2200 if hostIP is None:
2201 try:
2202 hostIP = socket.gethostbyname(hostonly)
2203 hostIP = ip2num(hostIP)
2204 except socket.error:
2205 continue
2206
2207 base = ip2num(m.group(1))
Ronald Oussorenab90f8e2010-06-27 14:26:30 +00002208 mask = m.group(2)
2209 if mask is None:
2210 mask = 8 * (m.group(1).count('.') + 1)
2211
2212 else:
2213 mask = int(mask[1:])
2214 mask = 32 - mask
Ronald Oussoren84151202010-04-18 20:46:11 +00002215
2216 if (hostIP >> mask) == (base >> mask):
2217 return True
2218
2219 elif fnmatch(host, value):
2220 return True
2221
2222 return False
2223
2224
2225 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002226 """Return a dictionary of scheme -> proxy server URL mappings.
2227
Ronald Oussoren84151202010-04-18 20:46:11 +00002228 This function uses the MacOSX framework SystemConfiguration
2229 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002230 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002231 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002232
Ronald Oussoren84151202010-04-18 20:46:11 +00002233
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002234
2235 def proxy_bypass(host):
2236 if getproxies_environment():
2237 return proxy_bypass_environment(host)
2238 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002239 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002240
2241 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002242 return getproxies_environment() or getproxies_macosx_sysconf()
2243
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002244
2245elif os.name == 'nt':
2246 def getproxies_registry():
2247 """Return a dictionary of scheme -> proxy server URL mappings.
2248
2249 Win32 uses the registry to store proxies.
2250
2251 """
2252 proxies = {}
2253 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002254 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002255 except ImportError:
2256 # Std module, so should be around - but you never know!
2257 return proxies
2258 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002259 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002260 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002261 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002262 'ProxyEnable')[0]
2263 if proxyEnable:
2264 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002265 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002266 'ProxyServer')[0])
2267 if '=' in proxyServer:
2268 # Per-protocol settings
2269 for p in proxyServer.split(';'):
2270 protocol, address = p.split('=', 1)
2271 # See if address has a type:// prefix
2272 import re
2273 if not re.match('^([^/:]+)://', address):
2274 address = '%s://%s' % (protocol, address)
2275 proxies[protocol] = address
2276 else:
2277 # Use one setting for all protocols
2278 if proxyServer[:5] == 'http:':
2279 proxies['http'] = proxyServer
2280 else:
2281 proxies['http'] = 'http://%s' % proxyServer
2282 proxies['ftp'] = 'ftp://%s' % proxyServer
2283 internetSettings.Close()
2284 except (WindowsError, ValueError, TypeError):
2285 # Either registry key not found etc, or the value in an
2286 # unexpected format.
2287 # proxies already set up to be empty so nothing to do
2288 pass
2289 return proxies
2290
2291 def getproxies():
2292 """Return a dictionary of scheme -> proxy server URL mappings.
2293
2294 Returns settings gathered from the environment, if specified,
2295 or the registry.
2296
2297 """
2298 return getproxies_environment() or getproxies_registry()
2299
2300 def proxy_bypass_registry(host):
2301 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002302 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002303 import re
2304 except ImportError:
2305 # Std modules, so should be around - but you never know!
2306 return 0
2307 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002308 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002309 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002310 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002311 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002312 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002313 'ProxyOverride')[0])
2314 # ^^^^ Returned as Unicode but problems if not converted to ASCII
2315 except WindowsError:
2316 return 0
2317 if not proxyEnable or not proxyOverride:
2318 return 0
2319 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002320 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002321 host = [rawHost]
2322 try:
2323 addr = socket.gethostbyname(rawHost)
2324 if addr != rawHost:
2325 host.append(addr)
2326 except socket.error:
2327 pass
2328 try:
2329 fqdn = socket.getfqdn(rawHost)
2330 if fqdn != rawHost:
2331 host.append(fqdn)
2332 except socket.error:
2333 pass
2334 # make a check value list from the registry entry: replace the
2335 # '<local>' string by the localhost entry and the corresponding
2336 # canonical entry.
2337 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002338 # now check if we match one of the registry values.
2339 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002340 if test == '<local>':
2341 if '.' not in rawHost:
2342 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002343 test = test.replace(".", r"\.") # mask dots
2344 test = test.replace("*", r".*") # change glob sequence
2345 test = test.replace("?", r".") # change glob char
2346 for val in host:
2347 # print "%s <--> %s" %( test, val )
2348 if re.match(test, val, re.I):
2349 return 1
2350 return 0
2351
2352 def proxy_bypass(host):
2353 """Return a dictionary of scheme -> proxy server URL mappings.
2354
2355 Returns settings gathered from the environment, if specified,
2356 or the registry.
2357
2358 """
2359 if getproxies_environment():
2360 return proxy_bypass_environment(host)
2361 else:
2362 return proxy_bypass_registry(host)
2363
2364else:
2365 # By default use environment variables
2366 getproxies = getproxies_environment
2367 proxy_bypass = proxy_bypass_environment