- Merge JAAS configs
[lgpl/argeo-commons.git] / org.argeo.cms / src / org / argeo / cms / auth / UserAdminLoginModule.java
index b6ab04b213896db4a908112e7e7a2659016c366b..c77a5929f868dd3d4a45788a51a40a59b938caed 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,13 +17,19 @@ 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;
@@ -25,15 +37,20 @@ import org.osgi.service.useradmin.User;
 import org.osgi.service.useradmin.UserAdmin;
 
 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 state
        private BundleContext bc;
-       private Authorization authorization;
+       // private Authorization authorization;
+       private User authenticatedUser = null;
 
        @SuppressWarnings("unchecked")
        @Override
@@ -64,23 +81,23 @@ public class UserAdminLoginModule implements LoginModule {
                }
                UserAdmin userAdmin = bc.getService(bc.getServiceReference(UserAdmin.class));
                if (callbackHandler == null) {// anonymous
-                       authorization = userAdmin.getAuthorization(null);
-                       sharedState.put(CmsAuthUtils.SHARED_STATE_AUTHORIZATION, authorization);
+//                     authorization = userAdmin.getAuthorization(null);
+//                     sharedState.put(CmsAuthUtils.SHARED_STATE_AUTHORIZATION, authorization);
                        return true;
                }
 
                final String username;
                final char[] password;
-               if (callbackHandler == null && sharedState.containsKey(CmsAuthUtils.SHARED_STATE_NAME)
+               if (sharedState.containsKey(CmsAuthUtils.SHARED_STATE_NAME)
                                && sharedState.containsKey(CmsAuthUtils.SHARED_STATE_PWD)) {
                        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);
+                       // // TODO locale?
+                       // // NB: raw user name is used
+                       // AuthenticatingUser authenticatingUser = new
+                       // AuthenticatingUser(username, password);
+                       // authorization = userAdmin.getAuthorization(authenticatingUser);
                } else {
-
                        // ask for username and password
                        NameCallback nameCallback = new NameCallback("User");
                        PasswordCallback passwordCallback = new PasswordCallback("Password", false);
@@ -89,9 +106,6 @@ public class UserAdminLoginModule implements LoginModule {
                                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;
                        }
@@ -102,71 +116,91 @@ public class UserAdminLoginModule implements LoginModule {
                                locale = Locale.getDefault();
                        UiContext.setLocale(locale);
 
-                       // authorization = (Authorization)
-                       // sharedState.get(CmsAuthUtils.SHARED_STATE_AUTHORIZATION);
-                       //
-                       // if (authorization == null) {
-                       // create credentials
                        username = nameCallback.getName();
                        if (username == null || username.trim().equals("")) {
                                // authorization = userAdmin.getAuthorization(null);
                                throw new CredentialNotFoundException("No credentials provided");
                        }
-                       // char[] password = {};
                        if (passwordCallback.getPassword() != null)
                                password = passwordCallback.getPassword();
                        else
                                throw new CredentialNotFoundException("No credentials provided");
                        // FIXME move Argeo specific convention from user admin to here
-                       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);
-                       assert authorization != null;
                }
 
-               // }
-               // if
-               // (!sharedState.containsKey(CmsAuthUtils.SHARED_STATE_AUTHORIZATION))
-               sharedState.put(CmsAuthUtils.SHARED_STATE_AUTHORIZATION, authorization);
-               return authorization != null;
+               // 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 false;
+
+               // Log and monitor new login
+               // if (log.isDebugEnabled())
+               // log.debug("Logged in to CMS with username [" + username +
+               // "]");
+
+               // authorization = userAdmin.getAuthorization(user);
+               // assert authorization != null;
+               //
+               // sharedState.put(CmsAuthUtils.SHARED_STATE_AUTHORIZATION,
+               // authorization);
+               return true;
        }
 
        @Override
        public boolean commit() throws LoginException {
-               // Set<KerberosPrincipal> kerberosPrincipals =
-               // subject.getPrincipals(KerberosPrincipal.class);
-               // if (kerberosPrincipals.size() != 0) {
-               // KerberosPrincipal kerberosPrincipal =
-               // kerberosPrincipals.iterator().next();
-               // System.out.println(kerberosPrincipal);
-               // UserAdmin userAdmin =
-               // bc.getService(bc.getServiceReference(UserAdmin.class));
-               // User user = userAdmin.getUser(null, kerberosPrincipal.getName());
-               // Authorization authorization = userAdmin.getAuthorization(user);
-               // sharedState.put(SHARED_STATE_AUTHORIZATION, authorization);
+               // if (authorization == null) {
+               // return false;
+               // // throw new LoginException("Authorization should not be null");
+               // } else {
+               // CmsAuthUtils.addAuthentication(subject, authorization);
+               // return true;
                // }
-               if (authorization == null) {
-                       return false;
-                       // throw new LoginException("Authorization should not be null");
+               UserAdmin userAdmin = bc.getService(bc.getServiceReference(UserAdmin.class));
+               Authorization authorization = null;
+               User authenticatingUser;
+               Set<KerberosPrincipal> kerberosPrincipals = subject.getPrincipals(KerberosPrincipal.class);
+               if (kerberosPrincipals.isEmpty()) {
+                       if (callbackHandler == null) {
+                               authorization = userAdmin.getAuthorization(null);
+                       }
+                       if (authenticatedUser == null) {
+                               return false;
+                       } else {
+                               authenticatingUser = authenticatedUser;
+                       }
                } else {
-                       CmsAuthUtils.addAuthentication(subject, authorization);
-                       return true;
+                       KerberosPrincipal kerberosPrincipal = kerberosPrincipals.iterator().next();
+                       LdapName dn = IpaUtils.kerberosToDn(kerberosPrincipal.getName());
+                       authenticatingUser = new AuthenticatingUser(dn);
+               }
+               if (authorization == null)
+                       authorization = Subject.doAs(subject, new PrivilegedAction<Authorization>() {
+
+                               @Override
+                               public Authorization run() {
+                                       Authorization authorization = userAdmin.getAuthorization(authenticatingUser);
+                                       return authorization;
+                               }
+
+                       });
+               if (authorization == null)
+                       return false;
+               CmsAuthUtils.addAuthentication(subject, authorization);
+               HttpServletRequest request = (HttpServletRequest) sharedState.get(CmsAuthUtils.SHARED_STATE_HTTP_REQUEST);
+               if (request != null) {
+                       CmsAuthUtils.registerSessionAuthorization(bc, request, subject, authorization);
                }
+               return true;
        }
 
        @Override
        public boolean abort() throws LoginException {
-               authorization = null;
+//             authorization = null;
                return true;
        }
 
@@ -175,4 +209,36 @@ public class UserAdminLoginModule implements LoginModule {
                CmsAuthUtils.cleanUp(subject);
                return true;
        }
+
+       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;
+               }
+
+       }
 }