url.py
484 lines
| 17.5 KiB
| text/x-python
|
PythonLexer
/ mercurial / url.py
|
r7270 | # url.py - HTTP handling for mercurial | ||
# | ||||
# Copyright 2005, 2006, 2007, 2008 Matt Mackall <mpm@selenic.com> | ||||
# Copyright 2006, 2007 Alexis S. L. Carvalho <alexis@cecm.usp.br> | ||||
# Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com> | ||||
# | ||||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r7270 | |||
|
r14076 | import urllib, urllib2, httplib, os, socket, cStringIO | ||
|
r7270 | from i18n import _ | ||
|
r14204 | import keepalive, util, sslutil | ||
|
r14244 | import httpconnection as httpconnectionmod | ||
|
r13371 | |||
|
r7270 | class passwordmgr(urllib2.HTTPPasswordMgrWithDefaultRealm): | ||
def __init__(self, ui): | ||||
urllib2.HTTPPasswordMgrWithDefaultRealm.__init__(self) | ||||
self.ui = ui | ||||
def find_user_password(self, realm, authuri): | ||||
authinfo = urllib2.HTTPPasswordMgrWithDefaultRealm.find_user_password( | ||||
self, realm, authuri) | ||||
user, passwd = authinfo | ||||
if user and passwd: | ||||
|
r8333 | self._writedebug(user, passwd) | ||
|
r7270 | return (user, passwd) | ||
|
r15005 | if not user or not passwd: | ||
|
r15025 | res = httpconnectionmod.readauthforuri(self.ui, authuri, user) | ||
|
r13372 | if res: | ||
group, auth = res | ||||
|
r8847 | user, passwd = auth.get('username'), auth.get('password') | ||
|
r13372 | self.ui.debug("using auth.%s.* for authentication\n" % group) | ||
|
r8333 | if not user or not passwd: | ||
|
r20291 | u = util.url(authuri) | ||
u.query = None | ||||
|
r8333 | if not self.ui.interactive(): | ||
|
r20291 | raise util.Abort(_('http authorization required for %s') % | ||
util.hidepassword(str(u))) | ||||
|
r7270 | |||
|
r20291 | self.ui.write(_("http authorization required for %s\n") % | ||
util.hidepassword(str(u))) | ||||
|
r12862 | self.ui.write(_("realm: %s\n") % realm) | ||
|
r8333 | if user: | ||
|
r12862 | self.ui.write(_("user: %s\n") % user) | ||
|
r8333 | else: | ||
user = self.ui.prompt(_("user:"), default=None) | ||||
|
r7270 | |||
|
r8333 | if not passwd: | ||
passwd = self.ui.getpass() | ||||
|
r7270 | |||
self.add_password(realm, authuri, user, passwd) | ||||
|
r8333 | self._writedebug(user, passwd) | ||
|
r7270 | return (user, passwd) | ||
|
r8333 | def _writedebug(self, user, passwd): | ||
msg = _('http auth: user %s, password %s\n') | ||||
self.ui.debug(msg % (user, passwd and '*' * len(passwd) or 'not set')) | ||||
|
r15025 | def find_stored_password(self, authuri): | ||
return urllib2.HTTPPasswordMgrWithDefaultRealm.find_user_password( | ||||
self, None, authuri) | ||||
|
r7270 | class proxyhandler(urllib2.ProxyHandler): | ||
def __init__(self, ui): | ||||
proxyurl = ui.config("http_proxy", "host") or os.getenv('http_proxy') | ||||
# XXX proxyauthinfo = None | ||||
if proxyurl: | ||||
# proxy can be proper url or host[:port] | ||||
if not (proxyurl.startswith('http:') or | ||||
proxyurl.startswith('https:')): | ||||
proxyurl = 'http://' + proxyurl + '/' | ||||
|
r14076 | proxy = util.url(proxyurl) | ||
|
r13820 | if not proxy.user: | ||
proxy.user = ui.config("http_proxy", "user") | ||||
proxy.passwd = ui.config("http_proxy", "passwd") | ||||
|
r7270 | |||
# see if we should use a proxy for this url | ||||
|
r10282 | no_list = ["localhost", "127.0.0.1"] | ||
|
r7270 | no_list.extend([p.lower() for | ||
p in ui.configlist("http_proxy", "no")]) | ||||
no_list.extend([p.strip().lower() for | ||||
p in os.getenv("no_proxy", '').split(',') | ||||
if p.strip()]) | ||||
# "http_proxy.always" config is for running tests on localhost | ||||
if ui.configbool("http_proxy", "always"): | ||||
self.no_list = [] | ||||
else: | ||||
self.no_list = no_list | ||||
|
r13820 | proxyurl = str(proxy) | ||
|
r7270 | proxies = {'http': proxyurl, 'https': proxyurl} | ||
|
r9467 | ui.debug('proxying through http://%s:%s\n' % | ||
|
r13820 | (proxy.host, proxy.port)) | ||
|
r7270 | else: | ||
proxies = {} | ||||
# urllib2 takes proxy values from the environment and those | ||||
|
r15077 | # will take precedence if found. So, if there's a config entry | ||
# defining a proxy, drop the environment ones | ||||
if ui.config("http_proxy", "host"): | ||||
for env in ["HTTP_PROXY", "http_proxy", "no_proxy"]: | ||||
try: | ||||
if env in os.environ: | ||||
del os.environ[env] | ||||
except OSError: | ||||
pass | ||||
|
r7270 | |||
urllib2.ProxyHandler.__init__(self, proxies) | ||||
self.ui = ui | ||||
def proxy_open(self, req, proxy, type_): | ||||
host = req.get_host().split(':')[0] | ||||
|
r19535 | for e in self.no_list: | ||
if host == e: | ||||
return None | ||||
if e.startswith('*.') and host.endswith(e[2:]): | ||||
return None | ||||
if e.startswith('.') and host.endswith(e[1:]): | ||||
return None | ||||
|
r7270 | |||
# work around a bug in Python < 2.4.2 | ||||
# (it leaves a "\n" at the end of Proxy-authorization headers) | ||||
baseclass = req.__class__ | ||||
class _request(baseclass): | ||||
def add_header(self, key, val): | ||||
if key.lower() == 'proxy-authorization': | ||||
val = val.strip() | ||||
return baseclass.add_header(self, key, val) | ||||
req.__class__ = _request | ||||
return urllib2.ProxyHandler.proxy_open(self, req, proxy, type_) | ||||
|
r13420 | def _gen_sendfile(orgsend): | ||
|
r7270 | def _sendfile(self, data): | ||
# send a file | ||||
|
r14244 | if isinstance(data, httpconnectionmod.httpsendfile): | ||
|
r7270 | # if auth required, some data sent twice, so rewind here | ||
data.seek(0) | ||||
for chunk in util.filechunkiter(data): | ||||
|
r13420 | orgsend(self, chunk) | ||
|
r7270 | else: | ||
|
r13420 | orgsend(self, data) | ||
|
r7270 | return _sendfile | ||
|
r14964 | has_https = util.safehasattr(urllib2, 'HTTPSHandler') | ||
|
r8590 | if has_https: | ||
try: | ||||
|
r10409 | _create_connection = socket.create_connection | ||
|
r10411 | except AttributeError: | ||
|
r10482 | _GLOBAL_DEFAULT_TIMEOUT = object() | ||
|
r10409 | def _create_connection(address, timeout=_GLOBAL_DEFAULT_TIMEOUT, | ||
source_address=None): | ||||
# lifted from Python 2.6 | ||||
msg = "getaddrinfo returns an empty list" | ||||
host, port = address | ||||
for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM): | ||||
af, socktype, proto, canonname, sa = res | ||||
sock = None | ||||
try: | ||||
sock = socket.socket(af, socktype, proto) | ||||
if timeout is not _GLOBAL_DEFAULT_TIMEOUT: | ||||
sock.settimeout(timeout) | ||||
if source_address: | ||||
sock.bind(source_address) | ||||
sock.connect(sa) | ||||
return sock | ||||
except socket.error, msg: | ||||
if sock is not None: | ||||
sock.close() | ||||
|
r18176 | raise socket.error(msg) | ||
|
r10409 | |||
|
r7270 | class httpconnection(keepalive.HTTPConnection): | ||
# must be able to send big bundle as stream. | ||||
|
r13420 | send = _gen_sendfile(keepalive.HTTPConnection.send) | ||
|
r7270 | |||
|
r8590 | def connect(self): | ||
|
r10415 | if has_https and self.realhostport: # use CONNECT proxy | ||
|
r8590 | self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | ||
self.sock.connect((self.host, self.port)) | ||||
|
r9852 | if _generic_proxytunnel(self): | ||
|
r17424 | # we do not support client X.509 certificates | ||
|
r14204 | self.sock = sslutil.ssl_wrap_socket(self.sock, None, None) | ||
|
r8590 | else: | ||
keepalive.HTTPConnection.connect(self) | ||||
def getresponse(self): | ||||
proxyres = getattr(self, 'proxyres', None) | ||||
if proxyres: | ||||
if proxyres.will_close: | ||||
self.close() | ||||
self.proxyres = None | ||||
return proxyres | ||||
return keepalive.HTTPConnection.getresponse(self) | ||||
|
r9852 | # general transaction handler to support different ways to handle | ||
# HTTPS proxying before and after Python 2.6.3. | ||||
def _generic_start_transaction(handler, h, req): | ||||
|
r14964 | tunnel_host = getattr(req, '_tunnel_host', None) | ||
if tunnel_host: | ||||
|
r9852 | if tunnel_host[:7] not in ['http://', 'https:/']: | ||
tunnel_host = 'https://' + tunnel_host | ||||
new_tunnel = True | ||||
else: | ||||
tunnel_host = req.get_selector() | ||||
new_tunnel = False | ||||
if new_tunnel or tunnel_host == req.get_full_url(): # has proxy | ||||
|
r14076 | u = util.url(tunnel_host) | ||
|
r13820 | if new_tunnel or u.scheme == 'https': # only use CONNECT for HTTPS | ||
h.realhostport = ':'.join([u.host, (u.port or '443')]) | ||||
|
r9852 | h.headers = req.headers.copy() | ||
h.headers.update(handler.parent.addheaders) | ||||
return | ||||
|
r10415 | h.realhostport = None | ||
|
r9852 | h.headers = None | ||
def _generic_proxytunnel(self): | ||||
proxyheaders = dict( | ||||
[(x, self.headers[x]) for x in self.headers | ||||
if x.lower().startswith('proxy-')]) | ||||
self._set_hostport(self.host, self.port) | ||||
|
r10415 | self.send('CONNECT %s HTTP/1.0\r\n' % self.realhostport) | ||
|
r9852 | for header in proxyheaders.iteritems(): | ||
self.send('%s: %s\r\n' % header) | ||||
self.send('\r\n') | ||||
# majority of the following code is duplicated from | ||||
# httplib.HTTPConnection as there are no adequate places to | ||||
# override functions to provide the needed functionality | ||||
res = self.response_class(self.sock, | ||||
strict=self.strict, | ||||
method=self._method) | ||||
while True: | ||||
version, status, reason = res._read_status() | ||||
if status != httplib.CONTINUE: | ||||
break | ||||
while True: | ||||
skip = res.fp.readline().strip() | ||||
if not skip: | ||||
break | ||||
res.status = status | ||||
res.reason = reason.strip() | ||||
if res.status == 200: | ||||
while True: | ||||
line = res.fp.readline() | ||||
if line == '\r\n': | ||||
break | ||||
return True | ||||
if version == 'HTTP/1.0': | ||||
res.version = 10 | ||||
elif version.startswith('HTTP/1.'): | ||||
res.version = 11 | ||||
elif version == 'HTTP/0.9': | ||||
res.version = 9 | ||||
else: | ||||
raise httplib.UnknownProtocol(version) | ||||
if res.version == 9: | ||||
res.length = None | ||||
res.chunked = 0 | ||||
res.will_close = 1 | ||||
res.msg = httplib.HTTPMessage(cStringIO.StringIO()) | ||||
return False | ||||
res.msg = httplib.HTTPMessage(res.fp) | ||||
res.msg.fp = None | ||||
# are we using the chunked-style of transfer encoding? | ||||
trenc = res.msg.getheader('transfer-encoding') | ||||
if trenc and trenc.lower() == "chunked": | ||||
res.chunked = 1 | ||||
res.chunk_left = None | ||||
else: | ||||
res.chunked = 0 | ||||
# will the connection close at the end of the response? | ||||
res.will_close = res._check_close() | ||||
# do we have a Content-Length? | ||||
|
r17428 | # NOTE: RFC 2616, section 4.4, #3 says we ignore this if | ||
# transfer-encoding is "chunked" | ||||
|
r9852 | length = res.msg.getheader('content-length') | ||
if length and not res.chunked: | ||||
try: | ||||
res.length = int(length) | ||||
except ValueError: | ||||
res.length = None | ||||
else: | ||||
if res.length < 0: # ignore nonsensical negative lengths | ||||
res.length = None | ||||
else: | ||||
res.length = None | ||||
# does the body have a fixed length? (of zero) | ||||
if (status == httplib.NO_CONTENT or status == httplib.NOT_MODIFIED or | ||||
100 <= status < 200 or # 1xx codes | ||||
res._method == 'HEAD'): | ||||
res.length = 0 | ||||
# if the connection remains open, and we aren't using chunked, and | ||||
# a content-length was not provided, then assume that the connection | ||||
# WILL close. | ||||
if (not res.will_close and | ||||
not res.chunked and | ||||
res.length is None): | ||||
res.will_close = 1 | ||||
self.proxyres = res | ||||
return False | ||||
|
r7270 | class httphandler(keepalive.HTTPHandler): | ||
def http_open(self, req): | ||||
return self.do_open(httpconnection, req) | ||||
|
r8590 | def _start_transaction(self, h, req): | ||
|
r9852 | _generic_start_transaction(self, h, req) | ||
|
r8590 | return keepalive.HTTPHandler._start_transaction(self, h, req) | ||
|
r7270 | if has_https: | ||
|
r13424 | class httpsconnection(httplib.HTTPSConnection): | ||
response_class = keepalive.HTTPResponse | ||||
# must be able to send big bundle as stream. | ||||
send = _gen_sendfile(keepalive.safesend) | ||||
getresponse = keepalive.wrapgetresponse(httplib.HTTPSConnection) | ||||
|
r9726 | |||
|
r10409 | def connect(self): | ||
|
r13422 | self.sock = _create_connection((self.host, self.port)) | ||
|
r13421 | host = self.host | ||
|
r13424 | if self.realhostport: # use CONNECT proxy | ||
|
r14064 | _generic_proxytunnel(self) | ||
|
r13424 | host = self.realhostport.rsplit(':', 1)[0] | ||
|
r14204 | self.sock = sslutil.ssl_wrap_socket( | ||
self.sock, self.key_file, self.cert_file, | ||||
**sslutil.sslkwargs(self.ui, host)) | ||||
sslutil.validator(self.ui, host)(self.sock) | ||||
|
r10409 | |||
|
r7270 | class httpshandler(keepalive.KeepAliveHandler, urllib2.HTTPSHandler): | ||
|
r8847 | def __init__(self, ui): | ||
keepalive.KeepAliveHandler.__init__(self) | ||||
urllib2.HTTPSHandler.__init__(self) | ||||
self.ui = ui | ||||
self.pwmgr = passwordmgr(self.ui) | ||||
|
r9852 | def _start_transaction(self, h, req): | ||
_generic_start_transaction(self, h, req) | ||||
return keepalive.KeepAliveHandler._start_transaction(self, h, req) | ||||
|
r7270 | def https_open(self, req): | ||
|
r15025 | # req.get_full_url() does not contain credentials and we may | ||
# need them to match the certificates. | ||||
url = req.get_full_url() | ||||
user, password = self.pwmgr.find_stored_password(url) | ||||
res = httpconnectionmod.readauthforuri(self.ui, url, user) | ||||
|
r13372 | if res: | ||
group, auth = res | ||||
self.auth = auth | ||||
self.ui.debug("using auth.%s.* for authentication\n" % group) | ||||
else: | ||||
self.auth = None | ||||
|
r8847 | return self.do_open(self._makeconnection, req) | ||
|
r10408 | def _makeconnection(self, host, port=None, *args, **kwargs): | ||
|
r8847 | keyfile = None | ||
certfile = None | ||||
|
r10511 | if len(args) >= 1: # key_file | ||
keyfile = args[0] | ||||
if len(args) >= 2: # cert_file | ||||
certfile = args[1] | ||||
args = args[2:] | ||||
|
r8847 | |||
# if the user has specified different key/cert files in | ||||
# hgrc, we prefer these | ||||
if self.auth and 'key' in self.auth and 'cert' in self.auth: | ||||
keyfile = self.auth['key'] | ||||
certfile = self.auth['cert'] | ||||
|
r16683 | conn = httpsconnection(host, port, keyfile, certfile, *args, | ||
**kwargs) | ||||
|
r10409 | conn.ui = self.ui | ||
return conn | ||||
|
r7270 | |||
class httpdigestauthhandler(urllib2.HTTPDigestAuthHandler): | ||||
|
r11457 | def __init__(self, *args, **kwargs): | ||
urllib2.HTTPDigestAuthHandler.__init__(self, *args, **kwargs) | ||||
self.retried_req = None | ||||
def reset_retry_count(self): | ||||
# Python 2.6.5 will call this on 401 or 407 errors and thus loop | ||||
# forever. We disable reset_retry_count completely and reset in | ||||
# http_error_auth_reqed instead. | ||||
pass | ||||
|
r7270 | def http_error_auth_reqed(self, auth_header, host, req, headers): | ||
|
r11457 | # Reset the retry counter once for each request. | ||
if req is not self.retried_req: | ||||
self.retried_req = req | ||||
self.retried = 0 | ||||
# In python < 2.5 AbstractDigestAuthHandler raises a ValueError if | ||||
# it doesn't know about the auth type requested. This can happen if | ||||
# somebody is using BasicAuth and types a bad password. | ||||
|
r7270 | try: | ||
return urllib2.HTTPDigestAuthHandler.http_error_auth_reqed( | ||||
self, auth_header, host, req, headers) | ||||
except ValueError, inst: | ||||
arg = inst.args[0] | ||||
if arg.startswith("AbstractDigestAuthHandler doesn't know "): | ||||
return | ||||
raise | ||||
|
r11844 | class httpbasicauthhandler(urllib2.HTTPBasicAuthHandler): | ||
def __init__(self, *args, **kwargs): | ||||
urllib2.HTTPBasicAuthHandler.__init__(self, *args, **kwargs) | ||||
self.retried_req = None | ||||
def reset_retry_count(self): | ||||
# Python 2.6.5 will call this on 401 or 407 errors and thus loop | ||||
# forever. We disable reset_retry_count completely and reset in | ||||
# http_error_auth_reqed instead. | ||||
pass | ||||
def http_error_auth_reqed(self, auth_header, host, req, headers): | ||||
# Reset the retry counter once for each request. | ||||
if req is not self.retried_req: | ||||
self.retried_req = req | ||||
self.retried = 0 | ||||
return urllib2.HTTPBasicAuthHandler.http_error_auth_reqed( | ||||
self, auth_header, host, req, headers) | ||||
|
r9347 | handlerfuncs = [] | ||
|
r7270 | def opener(ui, authinfo=None): | ||
''' | ||||
construct an opener suitable for urllib2 | ||||
authinfo will be added to the password manager | ||||
''' | ||||
|
r14244 | if ui.configbool('ui', 'usehttp2', False): | ||
handlers = [httpconnectionmod.http2handler(ui, passwordmgr(ui))] | ||||
else: | ||||
handlers = [httphandler()] | ||||
if has_https: | ||||
handlers.append(httpshandler(ui)) | ||||
|
r7270 | |||
handlers.append(proxyhandler(ui)) | ||||
passmgr = passwordmgr(ui) | ||||
if authinfo is not None: | ||||
passmgr.add_password(*authinfo) | ||||
user, passwd = authinfo[2:4] | ||||
|
r9467 | ui.debug('http auth: user %s, password %s\n' % | ||
|
r7270 | (user, passwd and '*' * len(passwd) or 'not set')) | ||
|
r11844 | handlers.extend((httpbasicauthhandler(passmgr), | ||
|
r7270 | httpdigestauthhandler(passmgr))) | ||
|
r9347 | handlers.extend([h(ui, passmgr) for h in handlerfuncs]) | ||
|
r7270 | opener = urllib2.build_opener(*handlers) | ||
# 1.0 here is the _protocol_ version | ||||
opener.addheaders = [('User-agent', 'mercurial/proto-1.0')] | ||||
opener.addheaders.append(('Accept', 'application/mercurial-0.1')) | ||||
return opener | ||||
|
r13818 | def open(ui, url_, data=None): | ||
|
r14076 | u = util.url(url_) | ||
|
r13818 | if u.scheme: | ||
u.scheme = u.scheme.lower() | ||||
url_, authinfo = u.authinfo() | ||||
else: | ||||
path = util.normpath(os.path.abspath(url_)) | ||||
url_ = 'file://' + urllib.pathname2url(path) | ||||
|
r7284 | authinfo = None | ||
|
r13818 | return opener(ui, authinfo).open(url_, data) | ||