+
−# httpconnection.py - urllib2 handler for new http support
+
−#
+
−# 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>
+
−# Copyright 2011 Google, Inc.
+
−#
+
−# This software may be used and distributed according to the terms of the
+
−# GNU General Public License version 2 or any later version.
+
−
+
−from __future__ import absolute_import
+
−
+
−import os
+
−
+
−from .i18n import _
+
−from .pycompat import open
+
−from . import (
+
−    pycompat,
+
−    util,
+
−)
+
−
+
−urlerr = util.urlerr
+
−urlreq = util.urlreq
+
−
+
−# moved here from url.py to avoid a cycle
+
−class httpsendfile(object):
+
−    """This is a wrapper around the objects returned by python's "open".
+
−
+
−    Its purpose is to send file-like objects via HTTP.
+
−    It do however not define a __len__ attribute because the length
+
−    might be more than Py_ssize_t can handle.
+
−    """
+
−
+
−    def __init__(self, ui, *args, **kwargs):
+
−        self.ui = ui
+
−        self._data = open(*args, **kwargs)
+
−        self.seek = self._data.seek
+
−        self.close = self._data.close
+
−        self.write = self._data.write
+
−        self.length = os.fstat(self._data.fileno()).st_size
+
−        self._pos = 0
+
−        # We pass double the max for total because we currently have
+
−        # to send the bundle twice in the case of a server that
+
−        # requires authentication. Since we can't know until we try
+
−        # once whether authentication will be required, just lie to
+
−        # the user and maybe the push succeeds suddenly at 50%.
+
−        self._progress = ui.makeprogress(
+
−            _(b'sending'), unit=_(b'kb'), total=(self.length // 1024 * 2)
+
−        )
+
−
+
−    def read(self, *args, **kwargs):
+
−        ret = self._data.read(*args, **kwargs)
+
−        if not ret:
+
−            self._progress.complete()
+
−            return ret
+
−        self._pos += len(ret)
+
−        self._progress.update(self._pos // 1024)
+
−        return ret
+
−
+
−    def __enter__(self):
+
−        return self
+
−
+
−    def __exit__(self, exc_type, exc_val, exc_tb):
+
−        self.close()
+
−
+
−
+
−# moved here from url.py to avoid a cycle
+
−def readauthforuri(ui, uri, user):
+
−    uri = pycompat.bytesurl(uri)
+
−    # Read configuration
+
−    groups = {}
+
−    for key, val in ui.configitems(b'auth'):
+
−        if key in (b'cookiefile',):
+
−            continue
+
−
+
−        if b'.' not in key:
+
−            ui.warn(_(b"ignoring invalid [auth] key '%s'\n") % key)
+
−            continue
+
−        group, setting = key.rsplit(b'.', 1)
+
−        gdict = groups.setdefault(group, {})
+
−        if setting in (b'username', b'cert', b'key'):
+
−            val = util.expandpath(val)
+
−        gdict[setting] = val
+
−
+
−    # Find the best match
+
−    scheme, hostpath = uri.split(b'://', 1)
+
−    bestuser = None
+
−    bestlen = 0
+
−    bestauth = None
+
−    for group, auth in pycompat.iteritems(groups):
+
−        if user and user != auth.get(b'username', user):
+
−            # If a username was set in the URI, the entry username
+
−            # must either match it or be unset
+
−            continue
+
−        prefix = auth.get(b'prefix')
+
−        if not prefix:
+
−            continue
+
−
+
−        prefixurl = util.url(prefix)
+
−        if prefixurl.user and prefixurl.user != user:
+
−            # If a username was set in the prefix, it must match the username in
+
−            # the URI.
+
−            continue
+
−
+
−        # The URI passed in has been stripped of credentials, so erase the user
+
−        # here to allow simpler matching.
+
−        prefixurl.user = None
+
−        prefix = bytes(prefixurl)
+
−
+
−        p = prefix.split(b'://', 1)
+
−        if len(p) > 1:
+
−            schemes, prefix = [p[0]], p[1]
+
−        else:
+
−            schemes = (auth.get(b'schemes') or b'https').split()
+
−        if (
+
−            (prefix == b'*' or hostpath.startswith(prefix))
+
−            and (
+
−                len(prefix) > bestlen
+
−                or (
+
−                    len(prefix) == bestlen
+
−                    and not bestuser
+
−                    and b'username' in auth
+
−                )
+
−            )
+
−            and scheme in schemes
+
−        ):
+
−            bestlen = len(prefix)
+
−            bestauth = group, auth
+
−            bestuser = auth.get(b'username')
+
−            if user and not bestuser:
+
−                auth[b'username'] = user
+
−    return bestauth