blob: ee819c3336bb233f0335172d0190db51e8a5b3fb [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
778
779 def http_error_auth_reqed(self, authreq, host, req, headers):
780 # host may be an authority (without userinfo) or a URL with an
781 # authority
782 # XXX could be multiple headers
783 authreq = headers.get(authreq, None)
784 if authreq:
785 mo = AbstractBasicAuthHandler.rx.search(authreq)
786 if mo:
787 scheme, quote, realm = mo.groups()
788 if scheme.lower() == 'basic':
789 return self.retry_http_basic_auth(host, req, realm)
790
791 def retry_http_basic_auth(self, host, req, realm):
792 user, pw = self.passwd.find_user_password(realm, host)
793 if pw is not None:
794 raw = "%s:%s" % (user, pw)
795 auth = "Basic " + base64.b64encode(raw.encode()).decode("ascii")
796 if req.headers.get(self.auth_header, None) == auth:
797 return None
Senthil Kumaranca2fc9e2010-02-24 16:53:16 +0000798 req.add_unredirected_header(self.auth_header, auth)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000799 return self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000800 else:
801 return None
802
803
804class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
805
806 auth_header = 'Authorization'
807
808 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000809 url = req.full_url
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000810 return self.http_error_auth_reqed('www-authenticate',
811 url, req, headers)
812
813
814class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
815
816 auth_header = 'Proxy-authorization'
817
818 def http_error_407(self, req, fp, code, msg, headers):
819 # http_error_auth_reqed requires that there is no userinfo component in
Georg Brandl029986a2008-06-23 11:44:14 +0000820 # authority. Assume there isn't one, since urllib.request does not (and
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000821 # should not, RFC 3986 s. 3.2.1) support requests for URLs containing
822 # userinfo.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000823 authority = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000824 return self.http_error_auth_reqed('proxy-authenticate',
825 authority, req, headers)
826
827
828def randombytes(n):
829 """Return n random bytes."""
830 return os.urandom(n)
831
832class AbstractDigestAuthHandler:
833 # Digest authentication is specified in RFC 2617.
834
835 # XXX The client does not inspect the Authentication-Info header
836 # in a successful response.
837
838 # XXX It should be possible to test this implementation against
839 # a mock server that just generates a static set of challenges.
840
841 # XXX qop="auth-int" supports is shaky
842
843 def __init__(self, passwd=None):
844 if passwd is None:
845 passwd = HTTPPasswordMgr()
846 self.passwd = passwd
847 self.add_password = self.passwd.add_password
848 self.retried = 0
849 self.nonce_count = 0
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000850 self.last_nonce = None
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000851
852 def reset_retry_count(self):
853 self.retried = 0
854
855 def http_error_auth_reqed(self, auth_header, host, req, headers):
856 authreq = headers.get(auth_header, None)
857 if self.retried > 5:
858 # Don't fail endlessly - if we failed once, we'll probably
859 # fail a second time. Hm. Unless the Password Manager is
860 # prompting for the information. Crap. This isn't great
861 # but it's better than the current 'repeat until recursion
862 # depth exceeded' approach <wink>
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000863 raise HTTPError(req.full_url, 401, "digest auth failed",
Georg Brandl13e89462008-07-01 19:56:00 +0000864 headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000865 else:
866 self.retried += 1
867 if authreq:
868 scheme = authreq.split()[0]
869 if scheme.lower() == 'digest':
870 return self.retry_http_digest_auth(req, authreq)
871
872 def retry_http_digest_auth(self, req, auth):
873 token, challenge = auth.split(' ', 1)
874 chal = parse_keqv_list(filter(None, parse_http_list(challenge)))
875 auth = self.get_authorization(req, chal)
876 if auth:
877 auth_val = 'Digest %s' % auth
878 if req.headers.get(self.auth_header, None) == auth_val:
879 return None
880 req.add_unredirected_header(self.auth_header, auth_val)
Senthil Kumaranfb8cc2f2009-07-19 02:44:19 +0000881 resp = self.parent.open(req, timeout=req.timeout)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000882 return resp
883
884 def get_cnonce(self, nonce):
885 # The cnonce-value is an opaque
886 # quoted string value provided by the client and used by both client
887 # and server to avoid chosen plaintext attacks, to provide mutual
888 # authentication, and to provide some message integrity protection.
889 # This isn't a fabulous effort, but it's probably Good Enough.
890 s = "%s:%s:%s:" % (self.nonce_count, nonce, time.ctime())
891 b = s.encode("ascii") + randombytes(8)
892 dig = hashlib.sha1(b).hexdigest()
893 return dig[:16]
894
895 def get_authorization(self, req, chal):
896 try:
897 realm = chal['realm']
898 nonce = chal['nonce']
899 qop = chal.get('qop')
900 algorithm = chal.get('algorithm', 'MD5')
901 # mod_digest doesn't send an opaque, even though it isn't
902 # supposed to be optional
903 opaque = chal.get('opaque', None)
904 except KeyError:
905 return None
906
907 H, KD = self.get_algorithm_impls(algorithm)
908 if H is None:
909 return None
910
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000911 user, pw = self.passwd.find_user_password(realm, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000912 if user is None:
913 return None
914
915 # XXX not implemented yet
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000916 if req.data is not None:
917 entdig = self.get_entity_digest(req.data, chal)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000918 else:
919 entdig = None
920
921 A1 = "%s:%s:%s" % (user, realm, pw)
922 A2 = "%s:%s" % (req.get_method(),
923 # XXX selector: what about proxies and full urls
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000924 req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000925 if qop == 'auth':
Senthil Kumaran4c7eaee2009-11-15 08:43:45 +0000926 if nonce == self.last_nonce:
927 self.nonce_count += 1
928 else:
929 self.nonce_count = 1
930 self.last_nonce = nonce
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000931 ncvalue = '%08x' % self.nonce_count
932 cnonce = self.get_cnonce(nonce)
933 noncebit = "%s:%s:%s:%s:%s" % (nonce, ncvalue, cnonce, qop, H(A2))
934 respdig = KD(H(A1), noncebit)
935 elif qop is None:
936 respdig = KD(H(A1), "%s:%s" % (nonce, H(A2)))
937 else:
938 # XXX handle auth-int.
Georg Brandl13e89462008-07-01 19:56:00 +0000939 raise URLError("qop '%s' is not supported." % qop)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000940
941 # XXX should the partial digests be encoded too?
942
943 base = 'username="%s", realm="%s", nonce="%s", uri="%s", ' \
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000944 'response="%s"' % (user, realm, nonce, req.selector,
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000945 respdig)
946 if opaque:
947 base += ', opaque="%s"' % opaque
948 if entdig:
949 base += ', digest="%s"' % entdig
950 base += ', algorithm="%s"' % algorithm
951 if qop:
952 base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
953 return base
954
955 def get_algorithm_impls(self, algorithm):
956 # lambdas assume digest modules are imported at the top level
957 if algorithm == 'MD5':
958 H = lambda x: hashlib.md5(x.encode("ascii")).hexdigest()
959 elif algorithm == 'SHA':
960 H = lambda x: hashlib.sha1(x.encode("ascii")).hexdigest()
961 # XXX MD5-sess
962 KD = lambda s, d: H("%s:%s" % (s, d))
963 return H, KD
964
965 def get_entity_digest(self, data, chal):
966 # XXX not implemented yet
967 return None
968
969
970class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
971 """An authentication protocol defined by RFC 2069
972
973 Digest authentication improves on basic authentication because it
974 does not transmit passwords in the clear.
975 """
976
977 auth_header = 'Authorization'
978 handler_order = 490 # before Basic auth
979
980 def http_error_401(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000981 host = urlparse(req.full_url)[1]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000982 retry = self.http_error_auth_reqed('www-authenticate',
983 host, req, headers)
984 self.reset_retry_count()
985 return retry
986
987
988class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
989
990 auth_header = 'Proxy-Authorization'
991 handler_order = 490 # before Basic auth
992
993 def http_error_407(self, req, fp, code, msg, headers):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000994 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000995 retry = self.http_error_auth_reqed('proxy-authenticate',
996 host, req, headers)
997 self.reset_retry_count()
998 return retry
999
1000class AbstractHTTPHandler(BaseHandler):
1001
1002 def __init__(self, debuglevel=0):
1003 self._debuglevel = debuglevel
1004
1005 def set_http_debuglevel(self, level):
1006 self._debuglevel = level
1007
1008 def do_request_(self, request):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001009 host = request.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001010 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001011 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001012
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001013 if request.data is not None: # POST
1014 data = request.data
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001015 if not request.has_header('Content-type'):
1016 request.add_unredirected_header(
1017 'Content-type',
1018 'application/x-www-form-urlencoded')
1019 if not request.has_header('Content-length'):
1020 request.add_unredirected_header(
1021 'Content-length', '%d' % len(data))
1022
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001023 sel_host = host
1024 if request.has_proxy():
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001025 scheme, sel = splittype(request.selector)
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001026 sel_host, sel_path = splithost(sel)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001027 if not request.has_header('Host'):
Facundo Batista72dc1ea2008-08-16 14:44:32 +00001028 request.add_unredirected_header('Host', sel_host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001029 for name, value in self.parent.addheaders:
1030 name = name.capitalize()
1031 if not request.has_header(name):
1032 request.add_unredirected_header(name, value)
1033
1034 return request
1035
1036 def do_open(self, http_class, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001037 """Return an HTTPResponse object for the request, using http_class.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001038
1039 http_class must implement the HTTPConnection API from http.client.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001040 """
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001041 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001042 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001043 raise URLError('no host given')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001044
1045 h = http_class(host, timeout=req.timeout) # will parse host:port
1046 headers = dict(req.headers)
1047 headers.update(req.unredirected_hdrs)
1048
1049 # TODO(jhylton): Should this be redesigned to handle
1050 # persistent connections?
1051
1052 # We want to make an HTTP/1.1 request, but the addinfourl
1053 # class isn't prepared to deal with a persistent connection.
1054 # It will try to read all remaining data from the socket,
1055 # which will block while the server waits for the next request.
1056 # So make sure the connection gets closed after the (only)
1057 # request.
1058 headers["Connection"] = "close"
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001059 headers = dict((name.title(), val) for name, val in headers.items())
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001060
1061 if req._tunnel_host:
Senthil Kumaran47fff872009-12-20 07:10:31 +00001062 tunnel_headers = {}
1063 proxy_auth_hdr = "Proxy-Authorization"
1064 if proxy_auth_hdr in headers:
1065 tunnel_headers[proxy_auth_hdr] = headers[proxy_auth_hdr]
1066 # Proxy-Authorization should not be sent to origin
1067 # server.
1068 del headers[proxy_auth_hdr]
1069 h.set_tunnel(req._tunnel_host, headers=tunnel_headers)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001070
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001071 try:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001072 h.request(req.get_method(), req.selector, req.data, headers)
1073 r = h.getresponse() # an HTTPResponse instance
1074 except socket.error as err:
Georg Brandl13e89462008-07-01 19:56:00 +00001075 raise URLError(err)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001076
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001077 r.url = req.full_url
1078 # This line replaces the .msg attribute of the HTTPResponse
1079 # with .headers, because urllib clients expect the response to
1080 # have the reason in .msg. It would be good to mark this
1081 # attribute is deprecated and get then to use info() or
1082 # .headers.
1083 r.msg = r.reason
1084 return r
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001085
1086
1087class HTTPHandler(AbstractHTTPHandler):
1088
1089 def http_open(self, req):
1090 return self.do_open(http.client.HTTPConnection, req)
1091
1092 http_request = AbstractHTTPHandler.do_request_
1093
1094if hasattr(http.client, 'HTTPSConnection'):
1095 class HTTPSHandler(AbstractHTTPHandler):
1096
1097 def https_open(self, req):
1098 return self.do_open(http.client.HTTPSConnection, req)
1099
1100 https_request = AbstractHTTPHandler.do_request_
1101
1102class HTTPCookieProcessor(BaseHandler):
1103 def __init__(self, cookiejar=None):
1104 import http.cookiejar
1105 if cookiejar is None:
1106 cookiejar = http.cookiejar.CookieJar()
1107 self.cookiejar = cookiejar
1108
1109 def http_request(self, request):
1110 self.cookiejar.add_cookie_header(request)
1111 return request
1112
1113 def http_response(self, request, response):
1114 self.cookiejar.extract_cookies(response, request)
1115 return response
1116
1117 https_request = http_request
1118 https_response = http_response
1119
1120class UnknownHandler(BaseHandler):
1121 def unknown_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001122 type = req.type
Georg Brandl13e89462008-07-01 19:56:00 +00001123 raise URLError('unknown url type: %s' % type)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001124
1125def parse_keqv_list(l):
1126 """Parse list of key=value strings where keys are not duplicated."""
1127 parsed = {}
1128 for elt in l:
1129 k, v = elt.split('=', 1)
1130 if v[0] == '"' and v[-1] == '"':
1131 v = v[1:-1]
1132 parsed[k] = v
1133 return parsed
1134
1135def parse_http_list(s):
1136 """Parse lists as described by RFC 2068 Section 2.
1137
1138 In particular, parse comma-separated lists where the elements of
1139 the list may include quoted-strings. A quoted-string could
1140 contain a comma. A non-quoted string could have quotes in the
1141 middle. Neither commas nor quotes count if they are escaped.
1142 Only double-quotes count, not single-quotes.
1143 """
1144 res = []
1145 part = ''
1146
1147 escape = quote = False
1148 for cur in s:
1149 if escape:
1150 part += cur
1151 escape = False
1152 continue
1153 if quote:
1154 if cur == '\\':
1155 escape = True
1156 continue
1157 elif cur == '"':
1158 quote = False
1159 part += cur
1160 continue
1161
1162 if cur == ',':
1163 res.append(part)
1164 part = ''
1165 continue
1166
1167 if cur == '"':
1168 quote = True
1169
1170 part += cur
1171
1172 # append last part
1173 if part:
1174 res.append(part)
1175
1176 return [part.strip() for part in res]
1177
1178class FileHandler(BaseHandler):
1179 # Use local file or FTP depending on form of URL
1180 def file_open(self, req):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001181 url = req.selector
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001182 if url[:2] == '//' and url[2:3] != '/':
1183 req.type = 'ftp'
1184 return self.parent.open(req)
1185 else:
1186 return self.open_local_file(req)
1187
1188 # names for the localhost
1189 names = None
1190 def get_names(self):
1191 if FileHandler.names is None:
1192 try:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001193 FileHandler.names = tuple(
1194 socket.gethostbyname_ex('localhost')[2] +
1195 socket.gethostbyname_ex(socket.gethostname())[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001196 except socket.gaierror:
1197 FileHandler.names = (socket.gethostbyname('localhost'),)
1198 return FileHandler.names
1199
1200 # not entirely sure what the rules are here
1201 def open_local_file(self, req):
1202 import email.utils
1203 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001204 host = req.host
1205 file = req.selector
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001206 localfile = url2pathname(file)
1207 try:
1208 stats = os.stat(localfile)
1209 size = stats.st_size
1210 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1211 mtype = mimetypes.guess_type(file)[0]
1212 headers = email.message_from_string(
1213 'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
1214 (mtype or 'text/plain', size, modified))
1215 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001216 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001217 if not host or \
1218 (not port and _safe_gethostbyname(host) in self.get_names()):
Georg Brandl13e89462008-07-01 19:56:00 +00001219 return addinfourl(open(localfile, 'rb'), headers, 'file:'+file)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001220 except OSError as msg:
Georg Brandl029986a2008-06-23 11:44:14 +00001221 # users shouldn't expect OSErrors coming from urlopen()
Georg Brandl13e89462008-07-01 19:56:00 +00001222 raise URLError(msg)
1223 raise URLError('file not on local host')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001224
1225def _safe_gethostbyname(host):
1226 try:
1227 return socket.gethostbyname(host)
1228 except socket.gaierror:
1229 return None
1230
1231class FTPHandler(BaseHandler):
1232 def ftp_open(self, req):
1233 import ftplib
1234 import mimetypes
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001235 host = req.host
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001236 if not host:
Georg Brandl13e89462008-07-01 19:56:00 +00001237 raise URLError('ftp error: no host given')
1238 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001239 if port is None:
1240 port = ftplib.FTP_PORT
1241 else:
1242 port = int(port)
1243
1244 # username/password handling
Georg Brandl13e89462008-07-01 19:56:00 +00001245 user, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001246 if user:
Georg Brandl13e89462008-07-01 19:56:00 +00001247 user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001248 else:
1249 passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001250 host = unquote(host)
1251 user = unquote(user or '')
1252 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001253
1254 try:
1255 host = socket.gethostbyname(host)
1256 except socket.error as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001257 raise URLError(msg)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001258 path, attrs = splitattr(req.selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001259 dirs = path.split('/')
Georg Brandl13e89462008-07-01 19:56:00 +00001260 dirs = list(map(unquote, dirs))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001261 dirs, file = dirs[:-1], dirs[-1]
1262 if dirs and not dirs[0]:
1263 dirs = dirs[1:]
1264 try:
1265 fw = self.connect_ftp(user, passwd, host, port, dirs, req.timeout)
1266 type = file and 'I' or 'D'
1267 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001268 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001269 if attr.lower() == 'type' and \
1270 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1271 type = value.upper()
1272 fp, retrlen = fw.retrfile(file, type)
1273 headers = ""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001274 mtype = mimetypes.guess_type(req.full_url)[0]
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001275 if mtype:
1276 headers += "Content-type: %s\n" % mtype
1277 if retrlen is not None and retrlen >= 0:
1278 headers += "Content-length: %d\n" % retrlen
1279 headers = email.message_from_string(headers)
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001280 return addinfourl(fp, headers, req.full_url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001281 except ftplib.all_errors as msg:
Georg Brandl13e89462008-07-01 19:56:00 +00001282 exc = URLError('ftp error: %s' % msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001283 raise exc.with_traceback(sys.exc_info()[2])
1284
1285 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1286 fw = ftpwrapper(user, passwd, host, port, dirs, timeout)
1287 return fw
1288
1289class CacheFTPHandler(FTPHandler):
1290 # XXX would be nice to have pluggable cache strategies
1291 # XXX this stuff is definitely not thread safe
1292 def __init__(self):
1293 self.cache = {}
1294 self.timeout = {}
1295 self.soonest = 0
1296 self.delay = 60
1297 self.max_conns = 16
1298
1299 def setTimeout(self, t):
1300 self.delay = t
1301
1302 def setMaxConns(self, m):
1303 self.max_conns = m
1304
1305 def connect_ftp(self, user, passwd, host, port, dirs, timeout):
1306 key = user, host, port, '/'.join(dirs), timeout
1307 if key in self.cache:
1308 self.timeout[key] = time.time() + self.delay
1309 else:
1310 self.cache[key] = ftpwrapper(user, passwd, host, port,
1311 dirs, timeout)
1312 self.timeout[key] = time.time() + self.delay
1313 self.check_cache()
1314 return self.cache[key]
1315
1316 def check_cache(self):
1317 # first check for old ones
1318 t = time.time()
1319 if self.soonest <= t:
1320 for k, v in list(self.timeout.items()):
1321 if v < t:
1322 self.cache[k].close()
1323 del self.cache[k]
1324 del self.timeout[k]
1325 self.soonest = min(list(self.timeout.values()))
1326
1327 # then check the size
1328 if len(self.cache) == self.max_conns:
1329 for k, v in list(self.timeout.items()):
1330 if v == self.soonest:
1331 del self.cache[k]
1332 del self.timeout[k]
1333 break
1334 self.soonest = min(list(self.timeout.values()))
1335
1336# Code move from the old urllib module
1337
1338MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
1339
1340# Helper for non-unix systems
1341if os.name == 'mac':
1342 from macurl2path import url2pathname, pathname2url
1343elif os.name == 'nt':
1344 from nturl2path import url2pathname, pathname2url
1345else:
1346 def url2pathname(pathname):
1347 """OS-specific conversion from a relative URL of the 'file' scheme
1348 to a file system path; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001349 return unquote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001350
1351 def pathname2url(pathname):
1352 """OS-specific conversion from a file system path to a relative URL
1353 of the 'file' scheme; not recommended for general use."""
Georg Brandl13e89462008-07-01 19:56:00 +00001354 return quote(pathname)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001355
1356# This really consists of two pieces:
1357# (1) a class which handles opening of all sorts of URLs
1358# (plus assorted utilities etc.)
1359# (2) a set of functions for parsing URLs
1360# XXX Should these be separated out into different modules?
1361
1362
1363ftpcache = {}
1364class URLopener:
1365 """Class to open URLs.
1366 This is a class rather than just a subroutine because we may need
1367 more than one set of global protocol-specific options.
1368 Note -- this is a base class for those who don't want the
1369 automatic handling of errors type 302 (relocated) and 401
1370 (authorization needed)."""
1371
1372 __tempfiles = None
1373
1374 version = "Python-urllib/%s" % __version__
1375
1376 # Constructor
1377 def __init__(self, proxies=None, **x509):
1378 if proxies is None:
1379 proxies = getproxies()
1380 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
1381 self.proxies = proxies
1382 self.key_file = x509.get('key_file')
1383 self.cert_file = x509.get('cert_file')
1384 self.addheaders = [('User-Agent', self.version)]
1385 self.__tempfiles = []
1386 self.__unlink = os.unlink # See cleanup()
1387 self.tempcache = None
1388 # Undocumented feature: if you assign {} to tempcache,
1389 # it is used to cache files retrieved with
1390 # self.retrieve(). This is not enabled by default
1391 # since it does not work for changing documents (and I
1392 # haven't got the logic to check expiration headers
1393 # yet).
1394 self.ftpcache = ftpcache
1395 # Undocumented feature: you can use a different
1396 # ftp cache by assigning to the .ftpcache member;
1397 # in case you want logically independent URL openers
1398 # XXX This is not threadsafe. Bah.
1399
1400 def __del__(self):
1401 self.close()
1402
1403 def close(self):
1404 self.cleanup()
1405
1406 def cleanup(self):
1407 # This code sometimes runs when the rest of this module
1408 # has already been deleted, so it can't use any globals
1409 # or import anything.
1410 if self.__tempfiles:
1411 for file in self.__tempfiles:
1412 try:
1413 self.__unlink(file)
1414 except OSError:
1415 pass
1416 del self.__tempfiles[:]
1417 if self.tempcache:
1418 self.tempcache.clear()
1419
1420 def addheader(self, *args):
1421 """Add a header to be used by the HTTP interface only
1422 e.g. u.addheader('Accept', 'sound/basic')"""
1423 self.addheaders.append(args)
1424
1425 # External interface
1426 def open(self, fullurl, data=None):
1427 """Use URLopener().open(file) instead of open(file, 'r')."""
Georg Brandl13e89462008-07-01 19:56:00 +00001428 fullurl = unwrap(to_bytes(fullurl))
Senthil Kumaran734f0592010-02-20 22:19:04 +00001429 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001430 if self.tempcache and fullurl in self.tempcache:
1431 filename, headers = self.tempcache[fullurl]
1432 fp = open(filename, 'rb')
Georg Brandl13e89462008-07-01 19:56:00 +00001433 return addinfourl(fp, headers, fullurl)
1434 urltype, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001435 if not urltype:
1436 urltype = 'file'
1437 if urltype in self.proxies:
1438 proxy = self.proxies[urltype]
Georg Brandl13e89462008-07-01 19:56:00 +00001439 urltype, proxyhost = splittype(proxy)
1440 host, selector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001441 url = (host, fullurl) # Signal special case to open_*()
1442 else:
1443 proxy = None
1444 name = 'open_' + urltype
1445 self.type = urltype
1446 name = name.replace('-', '_')
1447 if not hasattr(self, name):
1448 if proxy:
1449 return self.open_unknown_proxy(proxy, fullurl, data)
1450 else:
1451 return self.open_unknown(fullurl, data)
1452 try:
1453 if data is None:
1454 return getattr(self, name)(url)
1455 else:
1456 return getattr(self, name)(url, data)
1457 except socket.error as msg:
1458 raise IOError('socket error', msg).with_traceback(sys.exc_info()[2])
1459
1460 def open_unknown(self, fullurl, data=None):
1461 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001462 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001463 raise IOError('url error', 'unknown url type', type)
1464
1465 def open_unknown_proxy(self, proxy, fullurl, data=None):
1466 """Overridable interface to open unknown URL type."""
Georg Brandl13e89462008-07-01 19:56:00 +00001467 type, url = splittype(fullurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001468 raise IOError('url error', 'invalid proxy for %s' % type, proxy)
1469
1470 # External interface
1471 def retrieve(self, url, filename=None, reporthook=None, data=None):
1472 """retrieve(url) returns (filename, headers) for a local object
1473 or (tempfilename, headers) for a remote object."""
Georg Brandl13e89462008-07-01 19:56:00 +00001474 url = unwrap(to_bytes(url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001475 if self.tempcache and url in self.tempcache:
1476 return self.tempcache[url]
Georg Brandl13e89462008-07-01 19:56:00 +00001477 type, url1 = splittype(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001478 if filename is None and (not type or type == 'file'):
1479 try:
1480 fp = self.open_local_file(url1)
1481 hdrs = fp.info()
Philip Jenveycb134d72009-12-03 02:45:01 +00001482 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001483 return url2pathname(splithost(url1)[1]), hdrs
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001484 except IOError as msg:
1485 pass
1486 fp = self.open(url, data)
Benjamin Peterson5f28b7b2009-03-26 21:49:58 +00001487 try:
1488 headers = fp.info()
1489 if filename:
1490 tfp = open(filename, 'wb')
1491 else:
1492 import tempfile
1493 garbage, path = splittype(url)
1494 garbage, path = splithost(path or "")
1495 path, garbage = splitquery(path or "")
1496 path, garbage = splitattr(path or "")
1497 suffix = os.path.splitext(path)[1]
1498 (fd, filename) = tempfile.mkstemp(suffix)
1499 self.__tempfiles.append(filename)
1500 tfp = os.fdopen(fd, 'wb')
1501 try:
1502 result = filename, headers
1503 if self.tempcache is not None:
1504 self.tempcache[url] = result
1505 bs = 1024*8
1506 size = -1
1507 read = 0
1508 blocknum = 0
1509 if reporthook:
1510 if "content-length" in headers:
1511 size = int(headers["Content-Length"])
1512 reporthook(blocknum, bs, size)
1513 while 1:
1514 block = fp.read(bs)
1515 if not block:
1516 break
1517 read += len(block)
1518 tfp.write(block)
1519 blocknum += 1
1520 if reporthook:
1521 reporthook(blocknum, bs, size)
1522 finally:
1523 tfp.close()
1524 finally:
1525 fp.close()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001526
1527 # raise exception if actual size does not match content-length header
1528 if size >= 0 and read < size:
Georg Brandl13e89462008-07-01 19:56:00 +00001529 raise ContentTooShortError(
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001530 "retrieval incomplete: got only %i out of %i bytes"
1531 % (read, size), result)
1532
1533 return result
1534
1535 # Each method named open_<type> knows how to open that type of URL
1536
1537 def _open_generic_http(self, connection_factory, url, data):
1538 """Make an HTTP connection using connection_class.
1539
1540 This is an internal method that should be called from
1541 open_http() or open_https().
1542
1543 Arguments:
1544 - connection_factory should take a host name and return an
1545 HTTPConnection instance.
1546 - url is the url to retrieval or a host, relative-path pair.
1547 - data is payload for a POST request or None.
1548 """
1549
1550 user_passwd = None
1551 proxy_passwd= None
1552 if isinstance(url, str):
Georg Brandl13e89462008-07-01 19:56:00 +00001553 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001554 if host:
Georg Brandl13e89462008-07-01 19:56:00 +00001555 user_passwd, host = splituser(host)
1556 host = unquote(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001557 realhost = host
1558 else:
1559 host, selector = url
1560 # check whether the proxy contains authorization information
Georg Brandl13e89462008-07-01 19:56:00 +00001561 proxy_passwd, host = splituser(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001562 # now we proceed with the url we want to obtain
Georg Brandl13e89462008-07-01 19:56:00 +00001563 urltype, rest = splittype(selector)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001564 url = rest
1565 user_passwd = None
1566 if urltype.lower() != 'http':
1567 realhost = None
1568 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001569 realhost, rest = splithost(rest)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001570 if realhost:
Georg Brandl13e89462008-07-01 19:56:00 +00001571 user_passwd, realhost = splituser(realhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001572 if user_passwd:
1573 selector = "%s://%s%s" % (urltype, realhost, rest)
1574 if proxy_bypass(realhost):
1575 host = realhost
1576
1577 #print "proxy via http:", host, selector
1578 if not host: raise IOError('http error', 'no host given')
1579
1580 if proxy_passwd:
1581 import base64
1582 proxy_auth = base64.b64encode(proxy_passwd).strip()
1583 else:
1584 proxy_auth = None
1585
1586 if user_passwd:
1587 import base64
1588 auth = base64.b64encode(user_passwd).strip()
1589 else:
1590 auth = None
1591 http_conn = connection_factory(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001592 headers = {}
1593 if proxy_auth:
1594 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
1595 if auth:
1596 headers["Authorization"] = "Basic %s" % auth
1597 if realhost:
1598 headers["Host"] = realhost
1599 for header, value in self.addheaders:
1600 headers[header] = value
1601
1602 if data is not None:
1603 headers["Content-Type"] = "application/x-www-form-urlencoded"
1604 http_conn.request("POST", selector, data, headers)
1605 else:
1606 http_conn.request("GET", selector, headers=headers)
1607
1608 try:
1609 response = http_conn.getresponse()
1610 except http.client.BadStatusLine:
1611 # something went wrong with the HTTP status line
Georg Brandl13e89462008-07-01 19:56:00 +00001612 raise URLError("http protocol error: bad status line")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001613
1614 # According to RFC 2616, "2xx" code indicates that the client's
1615 # request was successfully received, understood, and accepted.
1616 if 200 <= response.status < 300:
Antoine Pitroub353c122009-02-11 00:39:14 +00001617 return addinfourl(response, response.msg, "http:" + url,
Georg Brandl13e89462008-07-01 19:56:00 +00001618 response.status)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001619 else:
1620 return self.http_error(
1621 url, response.fp,
1622 response.status, response.reason, response.msg, data)
1623
1624 def open_http(self, url, data=None):
1625 """Use HTTP protocol."""
1626 return self._open_generic_http(http.client.HTTPConnection, url, data)
1627
1628 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
1629 """Handle http errors.
1630
1631 Derived class can override this, or provide specific handlers
1632 named http_error_DDD where DDD is the 3-digit error code."""
1633 # First check if there's a specific handler for this error
1634 name = 'http_error_%d' % errcode
1635 if hasattr(self, name):
1636 method = getattr(self, name)
1637 if data is None:
1638 result = method(url, fp, errcode, errmsg, headers)
1639 else:
1640 result = method(url, fp, errcode, errmsg, headers, data)
1641 if result: return result
1642 return self.http_error_default(url, fp, errcode, errmsg, headers)
1643
1644 def http_error_default(self, url, fp, errcode, errmsg, headers):
1645 """Default error handler: close the connection and raise IOError."""
1646 void = fp.read()
1647 fp.close()
Georg Brandl13e89462008-07-01 19:56:00 +00001648 raise HTTPError(url, errcode, errmsg, headers, None)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001649
1650 if _have_ssl:
1651 def _https_connection(self, host):
1652 return http.client.HTTPSConnection(host,
1653 key_file=self.key_file,
1654 cert_file=self.cert_file)
1655
1656 def open_https(self, url, data=None):
1657 """Use HTTPS protocol."""
1658 return self._open_generic_http(self._https_connection, url, data)
1659
1660 def open_file(self, url):
1661 """Use local file or FTP depending on form of URL."""
1662 if not isinstance(url, str):
1663 raise URLError('file error', 'proxy support for file protocol currently not implemented')
1664 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
1665 return self.open_ftp(url)
1666 else:
1667 return self.open_local_file(url)
1668
1669 def open_local_file(self, url):
1670 """Use local file."""
1671 import mimetypes, email.utils
1672 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001673 host, file = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001674 localname = url2pathname(file)
1675 try:
1676 stats = os.stat(localname)
1677 except OSError as e:
1678 raise URLError(e.errno, e.strerror, e.filename)
1679 size = stats.st_size
1680 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
1681 mtype = mimetypes.guess_type(url)[0]
1682 headers = email.message_from_string(
1683 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
1684 (mtype or 'text/plain', size, modified))
1685 if not host:
1686 urlfile = file
1687 if file[:1] == '/':
1688 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001689 return addinfourl(open(localname, 'rb'), headers, urlfile)
1690 host, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001691 if (not port
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00001692 and socket.gethostbyname(host) in (localhost() + thishost())):
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001693 urlfile = file
1694 if file[:1] == '/':
1695 urlfile = 'file://' + file
Georg Brandl13e89462008-07-01 19:56:00 +00001696 return addinfourl(open(localname, 'rb'), headers, urlfile)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001697 raise URLError('local file error', 'not on local host')
1698
1699 def open_ftp(self, url):
1700 """Use FTP protocol."""
1701 if not isinstance(url, str):
1702 raise URLError('ftp error', 'proxy support for ftp protocol currently not implemented')
1703 import mimetypes
1704 from io import StringIO
Georg Brandl13e89462008-07-01 19:56:00 +00001705 host, path = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001706 if not host: raise URLError('ftp error', 'no host given')
Georg Brandl13e89462008-07-01 19:56:00 +00001707 host, port = splitport(host)
1708 user, host = splituser(host)
1709 if user: user, passwd = splitpasswd(user)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001710 else: passwd = None
Georg Brandl13e89462008-07-01 19:56:00 +00001711 host = unquote(host)
1712 user = unquote(user or '')
1713 passwd = unquote(passwd or '')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001714 host = socket.gethostbyname(host)
1715 if not port:
1716 import ftplib
1717 port = ftplib.FTP_PORT
1718 else:
1719 port = int(port)
Georg Brandl13e89462008-07-01 19:56:00 +00001720 path, attrs = splitattr(path)
1721 path = unquote(path)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001722 dirs = path.split('/')
1723 dirs, file = dirs[:-1], dirs[-1]
1724 if dirs and not dirs[0]: dirs = dirs[1:]
1725 if dirs and not dirs[0]: dirs[0] = '/'
1726 key = user, host, port, '/'.join(dirs)
1727 # XXX thread unsafe!
1728 if len(self.ftpcache) > MAXFTPCACHE:
1729 # Prune the cache, rather arbitrarily
1730 for k in self.ftpcache.keys():
1731 if k != key:
1732 v = self.ftpcache[k]
1733 del self.ftpcache[k]
1734 v.close()
1735 try:
1736 if not key in self.ftpcache:
1737 self.ftpcache[key] = \
1738 ftpwrapper(user, passwd, host, port, dirs)
1739 if not file: type = 'D'
1740 else: type = 'I'
1741 for attr in attrs:
Georg Brandl13e89462008-07-01 19:56:00 +00001742 attr, value = splitvalue(attr)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001743 if attr.lower() == 'type' and \
1744 value in ('a', 'A', 'i', 'I', 'd', 'D'):
1745 type = value.upper()
1746 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
1747 mtype = mimetypes.guess_type("ftp:" + url)[0]
1748 headers = ""
1749 if mtype:
1750 headers += "Content-Type: %s\n" % mtype
1751 if retrlen is not None and retrlen >= 0:
1752 headers += "Content-Length: %d\n" % retrlen
1753 headers = email.message_from_string(headers)
Georg Brandl13e89462008-07-01 19:56:00 +00001754 return addinfourl(fp, headers, "ftp:" + url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001755 except ftperrors() as msg:
1756 raise URLError('ftp error', msg).with_traceback(sys.exc_info()[2])
1757
1758 def open_data(self, url, data=None):
1759 """Use "data" URL."""
1760 if not isinstance(url, str):
1761 raise URLError('data error', 'proxy support for data protocol currently not implemented')
1762 # ignore POSTed data
1763 #
1764 # syntax of data URLs:
1765 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
1766 # mediatype := [ type "/" subtype ] *( ";" parameter )
1767 # data := *urlchar
1768 # parameter := attribute "=" value
1769 try:
1770 [type, data] = url.split(',', 1)
1771 except ValueError:
1772 raise IOError('data error', 'bad data URL')
1773 if not type:
1774 type = 'text/plain;charset=US-ASCII'
1775 semi = type.rfind(';')
1776 if semi >= 0 and '=' not in type[semi:]:
1777 encoding = type[semi+1:]
1778 type = type[:semi]
1779 else:
1780 encoding = ''
1781 msg = []
1782 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %T GMT',
1783 time.gmtime(time.time())))
1784 msg.append('Content-type: %s' % type)
1785 if encoding == 'base64':
1786 import base64
Georg Brandl706824f2009-06-04 09:42:55 +00001787 # XXX is this encoding/decoding ok?
1788 data = base64.decodebytes(data.encode('ascii')).decode('latin1')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001789 else:
Georg Brandl13e89462008-07-01 19:56:00 +00001790 data = unquote(data)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001791 msg.append('Content-Length: %d' % len(data))
1792 msg.append('')
1793 msg.append(data)
1794 msg = '\n'.join(msg)
Georg Brandl13e89462008-07-01 19:56:00 +00001795 headers = email.message_from_string(msg)
1796 f = io.StringIO(msg)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001797 #f.fileno = None # needed for addinfourl
Georg Brandl13e89462008-07-01 19:56:00 +00001798 return addinfourl(f, headers, url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001799
1800
1801class FancyURLopener(URLopener):
1802 """Derived class with handlers for errors we can handle (perhaps)."""
1803
1804 def __init__(self, *args, **kwargs):
1805 URLopener.__init__(self, *args, **kwargs)
1806 self.auth_cache = {}
1807 self.tries = 0
1808 self.maxtries = 10
1809
1810 def http_error_default(self, url, fp, errcode, errmsg, headers):
1811 """Default error handling -- don't raise an exception."""
Georg Brandl13e89462008-07-01 19:56:00 +00001812 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001813
1814 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
1815 """Error 302 -- relocated (temporarily)."""
1816 self.tries += 1
1817 if self.maxtries and self.tries >= self.maxtries:
1818 if hasattr(self, "http_error_500"):
1819 meth = self.http_error_500
1820 else:
1821 meth = self.http_error_default
1822 self.tries = 0
1823 return meth(url, fp, 500,
1824 "Internal Server Error: Redirect Recursion", headers)
1825 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
1826 data)
1827 self.tries = 0
1828 return result
1829
1830 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
1831 if 'location' in headers:
1832 newurl = headers['location']
1833 elif 'uri' in headers:
1834 newurl = headers['uri']
1835 else:
1836 return
1837 void = fp.read()
1838 fp.close()
1839 # In case the server sent a relative URL, join with original:
Georg Brandl13e89462008-07-01 19:56:00 +00001840 newurl = urljoin(self.type + ":" + url, newurl)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001841 return self.open(newurl)
1842
1843 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
1844 """Error 301 -- also relocated (permanently)."""
1845 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1846
1847 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
1848 """Error 303 -- also relocated (essentially identical to 302)."""
1849 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1850
1851 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
1852 """Error 307 -- relocated, but turn POST into error."""
1853 if data is None:
1854 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
1855 else:
1856 return self.http_error_default(url, fp, errcode, errmsg, headers)
1857
1858 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None):
1859 """Error 401 -- authentication required.
1860 This function supports Basic authentication only."""
1861 if not 'www-authenticate' in headers:
1862 URLopener.http_error_default(self, url, fp,
1863 errcode, errmsg, headers)
1864 stuff = headers['www-authenticate']
1865 import re
1866 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1867 if not match:
1868 URLopener.http_error_default(self, url, fp,
1869 errcode, errmsg, headers)
1870 scheme, realm = match.groups()
1871 if scheme.lower() != 'basic':
1872 URLopener.http_error_default(self, url, fp,
1873 errcode, errmsg, headers)
1874 name = 'retry_' + self.type + '_basic_auth'
1875 if data is None:
1876 return getattr(self,name)(url, realm)
1877 else:
1878 return getattr(self,name)(url, realm, data)
1879
1880 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None):
1881 """Error 407 -- proxy authentication required.
1882 This function supports Basic authentication only."""
1883 if not 'proxy-authenticate' in headers:
1884 URLopener.http_error_default(self, url, fp,
1885 errcode, errmsg, headers)
1886 stuff = headers['proxy-authenticate']
1887 import re
1888 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
1889 if not match:
1890 URLopener.http_error_default(self, url, fp,
1891 errcode, errmsg, headers)
1892 scheme, realm = match.groups()
1893 if scheme.lower() != 'basic':
1894 URLopener.http_error_default(self, url, fp,
1895 errcode, errmsg, headers)
1896 name = 'retry_proxy_' + self.type + '_basic_auth'
1897 if data is None:
1898 return getattr(self,name)(url, realm)
1899 else:
1900 return getattr(self,name)(url, realm, data)
1901
1902 def retry_proxy_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001903 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001904 newurl = 'http://' + host + selector
1905 proxy = self.proxies['http']
Georg Brandl13e89462008-07-01 19:56:00 +00001906 urltype, proxyhost = splittype(proxy)
1907 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001908 i = proxyhost.find('@') + 1
1909 proxyhost = proxyhost[i:]
1910 user, passwd = self.get_user_passwd(proxyhost, realm, i)
1911 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001912 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001913 quote(passwd, safe=''), proxyhost)
1914 self.proxies['http'] = 'http://' + proxyhost + proxyselector
1915 if data is None:
1916 return self.open(newurl)
1917 else:
1918 return self.open(newurl, data)
1919
1920 def retry_proxy_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001921 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001922 newurl = 'https://' + host + selector
1923 proxy = self.proxies['https']
Georg Brandl13e89462008-07-01 19:56:00 +00001924 urltype, proxyhost = splittype(proxy)
1925 proxyhost, proxyselector = splithost(proxyhost)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001926 i = proxyhost.find('@') + 1
1927 proxyhost = proxyhost[i:]
1928 user, passwd = self.get_user_passwd(proxyhost, realm, i)
1929 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001930 proxyhost = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001931 quote(passwd, safe=''), proxyhost)
1932 self.proxies['https'] = 'https://' + proxyhost + proxyselector
1933 if data is None:
1934 return self.open(newurl)
1935 else:
1936 return self.open(newurl, data)
1937
1938 def retry_http_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001939 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001940 i = host.find('@') + 1
1941 host = host[i:]
1942 user, passwd = self.get_user_passwd(host, realm, i)
1943 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001944 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001945 quote(passwd, safe=''), host)
1946 newurl = 'http://' + host + selector
1947 if data is None:
1948 return self.open(newurl)
1949 else:
1950 return self.open(newurl, data)
1951
1952 def retry_https_basic_auth(self, url, realm, data=None):
Georg Brandl13e89462008-07-01 19:56:00 +00001953 host, selector = splithost(url)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001954 i = host.find('@') + 1
1955 host = host[i:]
1956 user, passwd = self.get_user_passwd(host, realm, i)
1957 if not (user or passwd): return None
Georg Brandl13e89462008-07-01 19:56:00 +00001958 host = "%s:%s@%s" % (quote(user, safe=''),
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001959 quote(passwd, safe=''), host)
1960 newurl = 'https://' + host + selector
1961 if data is None:
1962 return self.open(newurl)
1963 else:
1964 return self.open(newurl, data)
1965
1966 def get_user_passwd(self, host, realm, clear_cache = 0):
1967 key = realm + '@' + host.lower()
1968 if key in self.auth_cache:
1969 if clear_cache:
1970 del self.auth_cache[key]
1971 else:
1972 return self.auth_cache[key]
1973 user, passwd = self.prompt_user_passwd(host, realm)
1974 if user or passwd: self.auth_cache[key] = (user, passwd)
1975 return user, passwd
1976
1977 def prompt_user_passwd(self, host, realm):
1978 """Override this in a GUI environment!"""
1979 import getpass
1980 try:
1981 user = input("Enter username for %s at %s: " % (realm, host))
1982 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
1983 (user, realm, host))
1984 return user, passwd
1985 except KeyboardInterrupt:
1986 print()
1987 return None, None
1988
1989
1990# Utility functions
1991
1992_localhost = None
1993def localhost():
1994 """Return the IP address of the magic hostname 'localhost'."""
1995 global _localhost
1996 if _localhost is None:
1997 _localhost = socket.gethostbyname('localhost')
1998 return _localhost
1999
2000_thishost = None
2001def thishost():
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002002 """Return the IP addresses of the current host."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002003 global _thishost
2004 if _thishost is None:
Senthil Kumaran99b2c8f2009-12-27 10:13:39 +00002005 _thishost = tuple(socket.gethostbyname_ex(socket.gethostname()[2]))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002006 return _thishost
2007
2008_ftperrors = None
2009def ftperrors():
2010 """Return the set of errors raised by the FTP class."""
2011 global _ftperrors
2012 if _ftperrors is None:
2013 import ftplib
2014 _ftperrors = ftplib.all_errors
2015 return _ftperrors
2016
2017_noheaders = None
2018def noheaders():
Georg Brandl13e89462008-07-01 19:56:00 +00002019 """Return an empty email Message object."""
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002020 global _noheaders
2021 if _noheaders is None:
Georg Brandl13e89462008-07-01 19:56:00 +00002022 _noheaders = email.message_from_string("")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002023 return _noheaders
2024
2025
2026# Utility classes
2027
2028class ftpwrapper:
2029 """Class used by open_ftp() for cache of open FTP connections."""
2030
2031 def __init__(self, user, passwd, host, port, dirs, timeout=None):
2032 self.user = user
2033 self.passwd = passwd
2034 self.host = host
2035 self.port = port
2036 self.dirs = dirs
2037 self.timeout = timeout
2038 self.init()
2039
2040 def init(self):
2041 import ftplib
2042 self.busy = 0
2043 self.ftp = ftplib.FTP()
2044 self.ftp.connect(self.host, self.port, self.timeout)
2045 self.ftp.login(self.user, self.passwd)
2046 for dir in self.dirs:
2047 self.ftp.cwd(dir)
2048
2049 def retrfile(self, file, type):
2050 import ftplib
2051 self.endtransfer()
2052 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
2053 else: cmd = 'TYPE ' + type; isdir = 0
2054 try:
2055 self.ftp.voidcmd(cmd)
2056 except ftplib.all_errors:
2057 self.init()
2058 self.ftp.voidcmd(cmd)
2059 conn = None
2060 if file and not isdir:
2061 # Try to retrieve as a file
2062 try:
2063 cmd = 'RETR ' + file
2064 conn = self.ftp.ntransfercmd(cmd)
2065 except ftplib.error_perm as reason:
2066 if str(reason)[:3] != '550':
Georg Brandl13e89462008-07-01 19:56:00 +00002067 raise URLError('ftp error', reason).with_traceback(
2068 sys.exc_info()[2])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002069 if not conn:
2070 # Set transfer mode to ASCII!
2071 self.ftp.voidcmd('TYPE A')
2072 # Try a directory listing. Verify that directory exists.
2073 if file:
2074 pwd = self.ftp.pwd()
2075 try:
2076 try:
2077 self.ftp.cwd(file)
2078 except ftplib.error_perm as reason:
Georg Brandl13e89462008-07-01 19:56:00 +00002079 raise URLError('ftp error', reason) from reason
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002080 finally:
2081 self.ftp.cwd(pwd)
2082 cmd = 'LIST ' + file
2083 else:
2084 cmd = 'LIST'
2085 conn = self.ftp.ntransfercmd(cmd)
2086 self.busy = 1
2087 # Pass back both a suitably decorated object and a retrieval length
Georg Brandl13e89462008-07-01 19:56:00 +00002088 return (addclosehook(conn[0].makefile('rb'), self.endtransfer), conn[1])
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002089 def endtransfer(self):
2090 if not self.busy:
2091 return
2092 self.busy = 0
2093 try:
2094 self.ftp.voidresp()
2095 except ftperrors():
2096 pass
2097
2098 def close(self):
2099 self.endtransfer()
2100 try:
2101 self.ftp.close()
2102 except ftperrors():
2103 pass
2104
2105# Proxy handling
2106def getproxies_environment():
2107 """Return a dictionary of scheme -> proxy server URL mappings.
2108
2109 Scan the environment for variables named <scheme>_proxy;
2110 this seems to be the standard convention. If you need a
2111 different way, you can pass a proxies dictionary to the
2112 [Fancy]URLopener constructor.
2113
2114 """
2115 proxies = {}
2116 for name, value in os.environ.items():
2117 name = name.lower()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002118 if value and name[-6:] == '_proxy':
2119 proxies[name[:-6]] = value
2120 return proxies
2121
2122def proxy_bypass_environment(host):
2123 """Test if proxies should not be used for a particular host.
2124
2125 Checks the environment for a variable named no_proxy, which should
2126 be a list of DNS suffixes separated by commas, or '*' for all hosts.
2127 """
2128 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
2129 # '*' is special case for always bypass
2130 if no_proxy == '*':
2131 return 1
2132 # strip port off host
Georg Brandl13e89462008-07-01 19:56:00 +00002133 hostonly, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002134 # check if the host ends with any of the DNS suffixes
2135 for name in no_proxy.split(','):
2136 if name and (hostonly.endswith(name) or host.endswith(name)):
2137 return 1
2138 # otherwise, don't bypass
2139 return 0
2140
2141
2142if sys.platform == 'darwin':
Ronald Oussoren84151202010-04-18 20:46:11 +00002143 from _scproxy import _get_proxy_settings, _get_proxies
2144
2145 def proxy_bypass_macosx_sysconf(host):
2146 """
2147 Return True iff this host shouldn't be accessed using a proxy
2148
2149 This function uses the MacOSX framework SystemConfiguration
2150 to fetch the proxy information.
2151 """
2152 import re
2153 import socket
2154 from fnmatch import fnmatch
2155
2156 hostonly, port = splitport(host)
2157
2158 def ip2num(ipAddr):
2159 parts = ipAddr.split('.')
2160 parts = map(int, parts)
2161 if len(parts) != 4:
2162 parts = (parts + [0, 0, 0, 0])[:4]
2163 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
2164
2165 proxy_settings = _get_proxy_settings()
2166
2167 # Check for simple host names:
2168 if '.' not in host:
2169 if proxy_settings['exclude_simple']:
2170 return True
2171
2172 hostIP = None
2173
2174 for value in proxy_settings.get('exceptions', ()):
2175 # Items in the list are strings like these: *.local, 169.254/16
2176 if not value: continue
2177
2178 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
2179 if m is not None:
2180 if hostIP is None:
2181 try:
2182 hostIP = socket.gethostbyname(hostonly)
2183 hostIP = ip2num(hostIP)
2184 except socket.error:
2185 continue
2186
2187 base = ip2num(m.group(1))
2188 mask = int(m.group(2)[1:])
2189 mask = 32 - mask
2190
2191 if (hostIP >> mask) == (base >> mask):
2192 return True
2193
2194 elif fnmatch(host, value):
2195 return True
2196
2197 return False
2198
2199
2200 def getproxies_macosx_sysconf():
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002201 """Return a dictionary of scheme -> proxy server URL mappings.
2202
Ronald Oussoren84151202010-04-18 20:46:11 +00002203 This function uses the MacOSX framework SystemConfiguration
2204 to fetch the proxy information.
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002205 """
Ronald Oussoren84151202010-04-18 20:46:11 +00002206 return _get_proxies()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002207
Ronald Oussoren84151202010-04-18 20:46:11 +00002208
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002209
2210 def proxy_bypass(host):
2211 if getproxies_environment():
2212 return proxy_bypass_environment(host)
2213 else:
Ronald Oussoren84151202010-04-18 20:46:11 +00002214 return proxy_bypass_macosx_sysconf(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002215
2216 def getproxies():
Ronald Oussoren84151202010-04-18 20:46:11 +00002217 return getproxies_environment() or getproxies_macosx_sysconf()
2218
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002219
2220elif os.name == 'nt':
2221 def getproxies_registry():
2222 """Return a dictionary of scheme -> proxy server URL mappings.
2223
2224 Win32 uses the registry to store proxies.
2225
2226 """
2227 proxies = {}
2228 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002229 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002230 except ImportError:
2231 # Std module, so should be around - but you never know!
2232 return proxies
2233 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002234 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002235 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002236 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002237 'ProxyEnable')[0]
2238 if proxyEnable:
2239 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002240 proxyServer = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002241 'ProxyServer')[0])
2242 if '=' in proxyServer:
2243 # Per-protocol settings
2244 for p in proxyServer.split(';'):
2245 protocol, address = p.split('=', 1)
2246 # See if address has a type:// prefix
2247 import re
2248 if not re.match('^([^/:]+)://', address):
2249 address = '%s://%s' % (protocol, address)
2250 proxies[protocol] = address
2251 else:
2252 # Use one setting for all protocols
2253 if proxyServer[:5] == 'http:':
2254 proxies['http'] = proxyServer
2255 else:
2256 proxies['http'] = 'http://%s' % proxyServer
2257 proxies['ftp'] = 'ftp://%s' % proxyServer
2258 internetSettings.Close()
2259 except (WindowsError, ValueError, TypeError):
2260 # Either registry key not found etc, or the value in an
2261 # unexpected format.
2262 # proxies already set up to be empty so nothing to do
2263 pass
2264 return proxies
2265
2266 def getproxies():
2267 """Return a dictionary of scheme -> proxy server URL mappings.
2268
2269 Returns settings gathered from the environment, if specified,
2270 or the registry.
2271
2272 """
2273 return getproxies_environment() or getproxies_registry()
2274
2275 def proxy_bypass_registry(host):
2276 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002277 import winreg
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002278 import re
2279 except ImportError:
2280 # Std modules, so should be around - but you never know!
2281 return 0
2282 try:
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002283 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002284 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002285 proxyEnable = winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002286 'ProxyEnable')[0]
Georg Brandl4ed72ac2009-04-01 04:28:33 +00002287 proxyOverride = str(winreg.QueryValueEx(internetSettings,
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002288 'ProxyOverride')[0])
2289 # ^^^^ Returned as Unicode but problems if not converted to ASCII
2290 except WindowsError:
2291 return 0
2292 if not proxyEnable or not proxyOverride:
2293 return 0
2294 # try to make a host list from name and IP address.
Georg Brandl13e89462008-07-01 19:56:00 +00002295 rawHost, port = splitport(host)
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002296 host = [rawHost]
2297 try:
2298 addr = socket.gethostbyname(rawHost)
2299 if addr != rawHost:
2300 host.append(addr)
2301 except socket.error:
2302 pass
2303 try:
2304 fqdn = socket.getfqdn(rawHost)
2305 if fqdn != rawHost:
2306 host.append(fqdn)
2307 except socket.error:
2308 pass
2309 # make a check value list from the registry entry: replace the
2310 # '<local>' string by the localhost entry and the corresponding
2311 # canonical entry.
2312 proxyOverride = proxyOverride.split(';')
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002313 # now check if we match one of the registry values.
2314 for test in proxyOverride:
Senthil Kumaran49476062009-05-01 06:00:23 +00002315 if test == '<local>':
2316 if '.' not in rawHost:
2317 return 1
Jeremy Hylton1afc1692008-06-18 20:49:58 +00002318 test = test.replace(".", r"\.") # mask dots
2319 test = test.replace("*", r".*") # change glob sequence
2320 test = test.replace("?", r".") # change glob char
2321 for val in host:
2322 # print "%s <--> %s" %( test, val )
2323 if re.match(test, val, re.I):
2324 return 1
2325 return 0
2326
2327 def proxy_bypass(host):
2328 """Return a dictionary of scheme -> proxy server URL mappings.
2329
2330 Returns settings gathered from the environment, if specified,
2331 or the registry.
2332
2333 """
2334 if getproxies_environment():
2335 return proxy_bypass_environment(host)
2336 else:
2337 return proxy_bypass_registry(host)
2338
2339else:
2340 # By default use environment variables
2341 getproxies = getproxies_environment
2342 proxy_bypass = proxy_bypass_environment