+
−# copyright 2003-2012 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+
−# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+
−#
+
−# This file is part of CubicWeb.
+
−#
+
−# CubicWeb is free software: you can redistribute it and/or modify it under the
+
−# terms of the GNU Lesser General Public License as published by the Free
+
−# Software Foundation, either version 2.1 of the License, or (at your option)
+
−# any later version.
+
−#
+
−# CubicWeb is distributed in the hope that it will be useful, but WITHOUT
+
−# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+
−# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
+
−# details.
+
−#
+
−# You should have received a copy of the GNU Lesser General Public License along
+
−# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
+
−"""user authentication component"""
+
−
+
−__docformat__ = "restructuredtext en"
+
−
+
−from threading import Lock
+
−
+
−from logilab.common.decorators import clear_cache
+
−
+
−from cubicweb import AuthenticationError, BadConnectionId
+
−from cubicweb.view import Component
+
−from cubicweb.dbapi import _repo_connect, ConnectionProperties
+
−from cubicweb.web import InvalidSession
+
−from cubicweb.web.application import AbstractAuthenticationManager
+
−
+
−class NoAuthInfo(Exception): pass
+
−
+
−
+
−class WebAuthInfoRetreiver(Component):
+
−    __registry__ = 'webauth'
+
−    order = None
+
−    __abstract__ = True
+
−
+
−    def authentication_information(self, req):
+
−        """retreive authentication information from the given request, raise
+
−        NoAuthInfo if expected information is not found.
+
−        """
+
−        raise NotImplementedError()
+
−
+
−    def authenticated(self, retreiver, req, cnx, login, authinfo):
+
−        """callback when return authentication information have opened a
+
−        repository connection successfully. Take care req has no session
+
−        attached yet, hence req.execute isn't available.
+
−        """
+
−        pass
+
−
+
−    def request_has_auth_info(self, req):
+
−        """tells from the request if it has enough information
+
−        to proceed to authentication, would the current session
+
−        be invalidated
+
−        """
+
−        raise NotImplementedError()
+
−
+
−    def revalidate_login(self, req):
+
−        """returns a login string or None, for repository session validation
+
−        purposes
+
−        """
+
−        raise NotImplementedError()
+
−
+
−    def cleanup_authentication_information(self, req):
+
−        """called when the retriever has returned some authentication
+
−        information but we get an authentication error when using them, so it
+
−        get a chance to cleanup things (e.g. remove cookie)
+
−        """
+
−        pass
+
−
+
−
+
−class LoginPasswordRetreiver(WebAuthInfoRetreiver):
+
−    __regid__ = 'loginpwdauth'
+
−    order = 10
+
−
+
−    def authentication_information(self, req):
+
−        """retreive authentication information from the given request, raise
+
−        NoAuthInfo if expected information is not found.
+
−        """
+
−        login, password = req.get_authorization()
+
−        if not login:
+
−            raise NoAuthInfo()
+
−        return login, {'password': password}
+
−
+
−    def request_has_auth_info(self, req):
+
−        return req.get_authorization()[0] is not None
+
−
+
−    def revalidate_login(self, req):
+
−        return req.get_authorization()[0]
+
−
+
−class RepositoryAuthenticationManager(AbstractAuthenticationManager):
+
−    """authenticate user associated to a request and check session validity"""
+
−
+
−    def __init__(self, repo):
+
−        super(RepositoryAuthenticationManager, self).__init__(repo)
+
−        self.repo = repo
+
−        vreg = repo.vreg
+
−        self.log_queries = vreg.config['query-log-file']
+
−        self.authinforetrievers = sorted(vreg['webauth'].possible_objects(vreg),
+
−                                         key=lambda x: x.order)
+
−        # 2-uple login / password, login is None when no anonymous access
+
−        # configured
+
−        self.anoninfo = vreg.config.anonymous_user()
+
−        if self.anoninfo[0]:
+
−            self.anoninfo = (self.anoninfo[0], {'password': self.anoninfo[1]})
+
−
+
−    def validate_session(self, req, session):
+
−        """check session validity and return the connected user on success.
+
−
+
−        raise :exc:`InvalidSession` if session is corrupted for a reason or
+
−        another and should be closed
+
−
+
−        also invoked while going from anonymous to logged in
+
−        """
+
−        for retriever in self.authinforetrievers:
+
−            if retriever.request_has_auth_info(req):
+
−                login = retriever.revalidate_login(req)
+
−                return self._validate_session(req, session, login)
+
−        # let's try with the current session
+
−        return self._validate_session(req, session, None)
+
−
+
−    def _validate_session(self, req, session, login):
+
−        # check session.login and not user.login, since in case of login by
+
−        # email, login and cnx.login are the email while user.login is the
+
−        # actual user login
+
−        if login and session.login != login:
+
−            raise InvalidSession('login mismatch')
+
−        try:
+
−            # calling cnx.user() check connection validity, raise
+
−            # BadConnectionId on failure
+
−            user = session.cnx.user(req)
+
−        except BadConnectionId:
+
−            raise InvalidSession('bad connection id')
+
−        return user
+
−
+
−    def authenticate(self, req):
+
−        """authenticate user using connection information found in the request,
+
−        and return corresponding a :class:`~cubicweb.dbapi.Connection` instance,
+
−        as well as login used to open the connection.
+
−
+
−        raise :exc:`cubicweb.AuthenticationError` if authentication failed
+
−        (no authentication info found or wrong user/password)
+
−        """
+
−        for retriever in self.authinforetrievers:
+
−            try:
+
−                login, authinfo = retriever.authentication_information(req)
+
−            except NoAuthInfo:
+
−                continue
+
−            try:
+
−                cnx = self._authenticate(login, authinfo)
+
−            except AuthenticationError:
+
−                retriever.cleanup_authentication_information(req)
+
−                continue # the next one may succeed
+
−            for retriever_ in self.authinforetrievers:
+
−                retriever_.authenticated(retriever, req, cnx, login, authinfo)
+
−            return cnx, login
+
−        # false if no authentication info found, eg this is not an
+
−        # authentication failure
+
−        if 'login' in locals():
+
−            req.set_message(req._('authentication failure'))
+
−        login, authinfo = self.anoninfo
+
−        if login:
+
−            cnx = self._authenticate(login, authinfo)
+
−            return cnx, login
+
−        raise AuthenticationError()
+
−
+
−    def _authenticate(self, login, authinfo):
+
−        cnxprops = ConnectionProperties(close=False, log=self.log_queries)
+
−        cnx = _repo_connect(self.repo, login, cnxprops=cnxprops, **authinfo)
+
−        # decorate connection
+
−        cnx.vreg = self.vreg
+
−        return cnx
+
−