Improve CMS session.
[lgpl/argeo-commons.git] / org.argeo.cms / src / org / argeo / cms / auth / UserAdminLoginModule.java
index d4c41022f62dff2dd07e5d138f720e7a422b8070..7e95c951a3f8eb67f80f1970feda3712b5fcdbfb 100644 (file)
@@ -1,9 +1,15 @@
 package org.argeo.cms.auth;
 
 import java.io.IOException;
+import java.security.PrivilegedAction;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 import java.util.Locale;
 import java.util.Map;
+import java.util.Set;
 
+import javax.naming.ldap.LdapName;
 import javax.security.auth.Subject;
 import javax.security.auth.callback.Callback;
 import javax.security.auth.callback.CallbackHandler;
@@ -11,44 +17,48 @@ import javax.security.auth.callback.LanguageCallback;
 import javax.security.auth.callback.NameCallback;
 import javax.security.auth.callback.PasswordCallback;
 import javax.security.auth.callback.UnsupportedCallbackException;
+import javax.security.auth.kerberos.KerberosPrincipal;
 import javax.security.auth.login.CredentialNotFoundException;
 import javax.security.auth.login.FailedLoginException;
 import javax.security.auth.login.LoginException;
 import javax.security.auth.spi.LoginModule;
+import javax.servlet.http.HttpServletRequest;
 
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
 import org.argeo.cms.CmsException;
 import org.argeo.eclipse.ui.specific.UiContext;
+import org.argeo.naming.LdapAttrs;
+import org.argeo.osgi.useradmin.IpaUtils;
 import org.osgi.framework.BundleContext;
 import org.osgi.framework.FrameworkUtil;
 import org.osgi.service.useradmin.Authorization;
 import org.osgi.service.useradmin.User;
 import org.osgi.service.useradmin.UserAdmin;
 
-public class UserAdminLoginModule implements LoginModule, AuthConstants {
-       // private final static Log log =
-       // LogFactory.getLog(UserAdminLoginModule.class);
-       //
-       // private Subject subject;
+public class UserAdminLoginModule implements LoginModule {
+       private final static Log log = LogFactory.getLog(UserAdminLoginModule.class);
+
+       private Subject subject;
        private CallbackHandler callbackHandler;
        private Map<String, Object> sharedState = null;
 
-       private boolean isAnonymous = false;
+       private List<String> indexedUserProperties = Arrays
+                       .asList(new String[] { LdapAttrs.uid.name(), LdapAttrs.mail.name(), LdapAttrs.cn.name() });
 
-       // private HttpServletRequest request = null;
+       // private state
        private BundleContext bc;
+       private User authenticatedUser = null;
 
        @SuppressWarnings("unchecked")
        @Override
        public void initialize(Subject subject, CallbackHandler callbackHandler, Map<String, ?> sharedState,
                        Map<String, ?> options) {
+               this.subject = subject;
                try {
                        bc = FrameworkUtil.getBundle(UserAdminLoginModule.class).getBundleContext();
-                       assert bc != null;
-                       // this.subject = subject;
                        this.callbackHandler = callbackHandler;
                        this.sharedState = (Map<String, Object>) sharedState;
-                       if (options.containsKey("anonymous"))
-                               isAnonymous = Boolean.parseBoolean(options.get("anonymous").toString());
                } catch (Exception e) {
                        throw new CmsException("Cannot initialize login module", e);
                }
@@ -56,12 +66,34 @@ public class UserAdminLoginModule implements LoginModule, AuthConstants {
 
        @Override
        public boolean login() throws LoginException {
+               // Authorization sharedAuth = (Authorization)
+               // sharedState.get(CmsAuthUtils.SHARED_STATE_AUTHORIZATION);
+               // if (sharedAuth != null) {
+               // if (callbackHandler == null && sharedAuth.getName() != null)
+               // throw new LoginException("Shared authorization should be anonymous");
+               // return false;
+               // }
                UserAdmin userAdmin = bc.getService(bc.getServiceReference(UserAdmin.class));
-               Authorization authorization = null;
-               if (isAnonymous) {
-                       authorization = userAdmin.getAuthorization(null);
+               if (callbackHandler == null) {// anonymous
+                       // authorization = userAdmin.getAuthorization(null);
+                       // sharedState.put(CmsAuthUtils.SHARED_STATE_AUTHORIZATION,
+                       // authorization);
+                       return true;
+               }
+
+               final String username;
+               final char[] password;
+               if (sharedState.containsKey(CmsAuthUtils.SHARED_STATE_NAME)
+                               && sharedState.containsKey(CmsAuthUtils.SHARED_STATE_PWD)) {
+                       // NB: required by Basic http auth
+                       username = (String) sharedState.get(CmsAuthUtils.SHARED_STATE_NAME);
+                       password = (char[]) sharedState.get(CmsAuthUtils.SHARED_STATE_PWD);
+                       // // TODO locale?
+                       // // NB: raw user name is used
+                       // AuthenticatingUser authenticatingUser = new
+                       // AuthenticatingUser(username, password);
+                       // authorization = userAdmin.getAuthorization(authenticatingUser);
                } else {
-                       // HttpRequestCallback httpCallback = new HttpRequestCallback();
                        // ask for username and password
                        NameCallback nameCallback = new NameCallback("User");
                        PasswordCallback passwordCallback = new PasswordCallback("Password", false);
@@ -70,163 +102,128 @@ public class UserAdminLoginModule implements LoginModule, AuthConstants {
                                callbackHandler.handle(new Callback[] { nameCallback, passwordCallback, langCallback });
                        } catch (IOException e) {
                                throw new LoginException("Cannot handle callback: " + e.getMessage());
-                       } catch (ThreadDeath e) {
-                               throw new ThreadDeathLoginException("Callbackhandler thread died", e);
                        } catch (UnsupportedCallbackException e) {
                                return false;
                        }
 
-                       // check http
-                       // request = httpCallback.getRequest();
-                       // authorization = checkHttp();
-
                        // i18n
                        Locale locale = langCallback.getLocale();
                        if (locale == null)
                                locale = Locale.getDefault();
                        UiContext.setLocale(locale);
 
-                       authorization = (Authorization) sharedState.get(SHARED_STATE_AUTHORIZATION);
-
-                       if (authorization == null) {
-                               // create credentials
-                               final String username = nameCallback.getName();
-                               if (username == null || username.trim().equals("")) {
-                                       // authorization = userAdmin.getAuthorization(null);
-                                       throw new CredentialNotFoundException("No credentials provided");
-                               } else {
-                                       char[] password = {};
-                                       if (passwordCallback.getPassword() != null)
-                                               password = passwordCallback.getPassword();
-                                       else
-                                               throw new CredentialNotFoundException("No credentials provided");
-
-                                       User user = userAdmin.getUser(null, username);
-                                       if (user == null)
-                                               throw new FailedLoginException("Invalid credentials");
-                                       if (!user.hasCredential(null, password))
-                                               throw new FailedLoginException("Invalid credentials");
-                                       // return false;
-
-                                       // Log and monitor new login
-                                       // if (log.isDebugEnabled())
-                                       // log.debug("Logged in to CMS with username [" + username +
-                                       // "]");
-
-                                       authorization = userAdmin.getAuthorization(user);
-                               }
+                       username = nameCallback.getName();
+                       if (username == null || username.trim().equals("")) {
+                               // authorization = userAdmin.getAuthorization(null);
+                               throw new CredentialNotFoundException("No credentials provided");
                        }
+                       if (passwordCallback.getPassword() != null)
+                               password = passwordCallback.getPassword();
+                       else
+                               throw new CredentialNotFoundException("No credentials provided");
                }
-               if (!sharedState.containsKey(SHARED_STATE_AUTHORIZATION))
-                       sharedState.put(SHARED_STATE_AUTHORIZATION, authorization);
-               // subject.getPrivateCredentials().add(authorization);
+
+               // User user = userAdmin.getUser(null, username);
+               User user = searchForUser(userAdmin, username);
+               if (user == null)
+                       return true;// expect Kerberos
+               // throw new FailedLoginException("Invalid credentials");
+               if (!user.hasCredential(null, password))
+                       throw new FailedLoginException("Invalid credentials");
+               authenticatedUser = user;
                return true;
        }
 
-       // private Authorization checkHttp() {
-       // Authorization authorization = null;
-       // if (request != null) {
-       // authorization = (Authorization)
-       // request.getAttribute(HttpContext.AUTHORIZATION);
-       // if (authorization == null) {
-       // String sessionId = request.getSession().getId();
-       // authorization = (Authorization)
-       // request.getSession().getAttribute(HttpContext.AUTHORIZATION);
-       // if (authorization == null) {
-       // Collection<ServiceReference<CmsSession>> sr;
-       // try {
-       // sr = bc.getServiceReferences(CmsSession.class,
-       // "(" + CmsSession.CMS_SESSION_ID + "=" + sessionId + ")");
-       // } catch (InvalidSyntaxException e) {
-       // throw new CmsException("Cannot get CMS session for id " + sessionId, e);
-       // }
-       // if (sr.size() == 1) {
-       // CmsSession cmsSession = bc.getService(sr.iterator().next());
-       // authorization = cmsSession.getAuthorization();
-       // if (log.isTraceEnabled())
-       // log.trace("Retrieved authorization from " + cmsSession);
-       // }
-       // }
-       // }
-       // }
-       // return authorization;
-       // }
-
        @Override
        public boolean commit() throws LoginException {
-               // Authorization authorization =
-               // subject.getPrivateCredentials(Authorization.class).iterator().next();
-               // if (request != null && authorization.getName() != null) {
-               // request.setAttribute(HttpContext.REMOTE_USER,
-               // authorization.getName());
-               // request.setAttribute(HttpContext.AUTHORIZATION, authorization);
-               //
-               // HttpSession httpSession = request.getSession();
-               // if (httpSession.getAttribute(HttpContext.AUTHORIZATION) == null) {
-               //
-               // String sessionId = request.getSession().getId();
-               // Collection<ServiceReference<CmsSession>> sr;
-               // try {
-               // sr = bc.getServiceReferences(CmsSession.class,
-               // "(" + CmsSession.CMS_SESSION_ID + "=" + sessionId + ")");
-               // } catch (InvalidSyntaxException e) {
-               // throw new CmsException("Cannot get CMS session for id " + sessionId,
-               // e);
-               // }
-               // CmsSession cmsSession;
-               // if (sr.size() == 1) {
-               // cmsSession = bc.getService(sr.iterator().next());
-               // } else if (sr.size() == 0) {
-               // Hashtable<String, String> props = new Hashtable<>();
-               // props.put(CmsSession.CMS_DN, authorization.getName());
-               // props.put(CmsSession.CMS_SESSION_ID, sessionId);
-               // cmsSession = new CmsSessionImpl(sessionId, authorization);
-               // bc.registerService(CmsSession.class, cmsSession, props);
-               // if (log.isDebugEnabled())
-               // log.debug("Initialized " + cmsSession + " for " +
-               // authorization.getName());
-               // } else
-               // throw new CmsException(sr.size() + " CMS sessions registered for " +
-               // sessionId);
-               // cmsSession.addHttpSession(request);
-               // if (log.isTraceEnabled())
-               // log.trace("Added " + request.getServletPath() + " to " + cmsSession +
-               // " (" + request.getRequestURI()
-               // + ")");
-               // httpSession.setAttribute(HttpContext.AUTHORIZATION, authorization);
-               // }
-               // subject.getPrivateCredentials().add(request.getSession());
-               // }
+               UserAdmin userAdmin = bc.getService(bc.getServiceReference(UserAdmin.class));
+               Authorization authorization;
+               if (callbackHandler == null) {// anonymous
+                       authorization = userAdmin.getAuthorization(null);
+               } else {
+                       User authenticatingUser;
+                       Set<KerberosPrincipal> kerberosPrincipals = subject.getPrincipals(KerberosPrincipal.class);
+                       if (kerberosPrincipals.isEmpty()) {
+                               if (authenticatedUser == null) {
+                                       if (log.isTraceEnabled())
+                                               log.trace("Neither kerberos nor user admin login succeeded. Login failed.");
+                                       return false;
+                               } else {
+                                       authenticatingUser = authenticatedUser;
+                               }
+                       } else {
+                               KerberosPrincipal kerberosPrincipal = kerberosPrincipals.iterator().next();
+                               LdapName dn = IpaUtils.kerberosToDn(kerberosPrincipal.getName());
+                               authenticatingUser = new AuthenticatingUser(dn);
+                               if (authenticatedUser != null && !authenticatingUser.getName().equals(authenticatedUser.getName()))
+                                       throw new LoginException("Kerberos login " + authenticatingUser.getName()
+                                                       + " is inconsistent with user admin login " + authenticatedUser.getName());
+                       }
+                       authorization = Subject.doAs(subject, new PrivilegedAction<Authorization>() {
+
+                               @Override
+                               public Authorization run() {
+                                       Authorization authorization = userAdmin.getAuthorization(authenticatingUser);
+                                       return authorization;
+                               }
+
+                       });
+                       if (authorization == null)
+                               throw new LoginException(
+                                               "User admin found no authorization for authenticated user " + authenticatingUser.getName());
+               }
+               // Log and monitor new login
+               CmsAuthUtils.addAuthorization(subject, authorization,
+                               (HttpServletRequest) sharedState.get(CmsAuthUtils.SHARED_STATE_HTTP_REQUEST));
+               if (log.isDebugEnabled())
+                       log.debug("Logged in to CMS: " + subject);
                return true;
        }
 
        @Override
        public boolean abort() throws LoginException {
-               // cleanUp();
                return true;
        }
 
        @Override
        public boolean logout() throws LoginException {
-               // Set<HttpSession> httpSession =
-               // subject.getPrivateCredentials(HttpSession.class);
-               // Iterator<HttpSession> it = httpSession.iterator();
-               // while (it.hasNext()) {
-               // HttpSession sess = it.next();
-               // sess.setAttribute(HttpContext.AUTHORIZATION, null);
-               // // sess.setMaxInactiveInterval(1);// invalidate session
-               //
-               // // TODO log out CMS session
-               // }
-               // subject.getPrivateCredentials().removeAll(httpSession);
-               //
-               // cleanUp();
+               if (log.isTraceEnabled())
+                       log.trace("Logging out from CMS... " + subject);
+               // boolean httpSessionLogoutOk = CmsAuthUtils.logoutSession(bc,
+               // subject);
+               CmsAuthUtils.cleanUp(subject);
                return true;
        }
 
-       // private void cleanUp() {
-       // subject.getPrivateCredentials().removeAll(subject.getPrivateCredentials(Authorization.class));
-       // subject = null;
-       // }
+       protected User searchForUser(UserAdmin userAdmin, String providedUsername) {
+               try {
+                       // TODO check value null or empty
+                       List<User> collectedUsers = new ArrayList<User>();
+                       // try dn
+                       User user = null;
+                       try {
+                               user = (User) userAdmin.getRole(providedUsername);
+                               if (user != null)
+                                       collectedUsers.add(user);
+                       } catch (Exception e) {
+                               // silent
+                       }
+                       // try all indexes
+                       for (String attr : indexedUserProperties) {
+                               user = userAdmin.getUser(attr, providedUsername);
+                               if (user != null)
+                                       collectedUsers.add(user);
+                       }
+                       if (collectedUsers.size() == 1)
+                               return collectedUsers.get(0);
+                       else if (collectedUsers.size() > 1)
+                               log.warn(collectedUsers.size() + " users for provided username" + providedUsername);
+                       return null;
+               } catch (Exception e) {
+                       if (log.isTraceEnabled())
+                               log.warn("Cannot search for user " + providedUsername, e);
+                       return null;
+               }
 
+       }
 }