de.theit.jenkins.crowd.CrowdServletFilter.java Source code

Java tutorial

Introduction

Here is the source code for de.theit.jenkins.crowd.CrowdServletFilter.java

Source

/*
 * @(#)CrowdServletFilter.java
 * 
 * The MIT License
 * 
 * Copyright (C)2011 Thorsten Heit.
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package de.theit.jenkins.crowd;

import static de.theit.jenkins.crowd.ErrorMessages.operationFailed;

import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import com.atlassian.crowd.exception.OperationFailedException;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.web.authentication.RememberMeServices;
import static org.springframework.security.web.authentication.rememberme.AbstractRememberMeServices.SPRING_SECURITY_REMEMBER_ME_COOKIE_KEY;

/**
 * This class realizes a servlet filter that checks on each request the status
 * of the SSO session. If the session isn't valid anymore, the user is logged
 * out automatically, and vice-versa: If there's a SSO session but the user
 * isn't logged in, (s)he is automatically logged in.
 * 
 * @author <a href="mailto:theit@gmx.de">Thorsten Heit (theit@gmx.de)</a>
 * @since 09.09.2011
 * @version $Id$
 */
public class CrowdServletFilter implements Filter {
    /** Used for logging purposes. */
    private static final Logger LOG = Logger.getLogger(CrowdServletFilter.class.getName());

    /**
     * The configuration data necessary for accessing the services on the remote
     * Crowd server.
     */
    private CrowdConfigurationService configuration;

    /** The default servlet filter. */
    private Filter defaultFilter;

    /**
     * The Crowd security realm. Used for logging out users when the SSO session
     * isn't valid anymore.
     */
    private CrowdSecurityRealm securityRealm;

    /** Holds the {@link RememberMeServices} that is used for auto-login. */
    private CrowdRememberMeServices rememberMe;

    /**
     * Creates a new instance of this class.
     * 
     * @param pSecurityRealm
     *            The Crowd security realm. Necessary for logging out users when
     *            the SSO session isn't valid anymore. May not be
     *            <code>null</code>.
     * @param pConfiguration
     *            The configuration to access the services on the remote Crowd
     *            server. May not be <code>null</code>.
     * @param pDefaultFilter
     *            The default filter to use when the Crowd security filter is
     *            not used during runtime. May not be <code>null</code>.
     */
    public CrowdServletFilter(CrowdSecurityRealm pSecurityRealm, CrowdConfigurationService pConfiguration,
            Filter pDefaultFilter) {
        this.securityRealm = pSecurityRealm;
        this.configuration = pConfiguration;
        this.defaultFilter = pDefaultFilter;

        if (this.securityRealm.getSecurityComponents().rememberMe instanceof CrowdRememberMeServices) {
            this.rememberMe = (CrowdRememberMeServices) this.securityRealm.getSecurityComponents().rememberMe;
        }
    }

    /**
     * {@inheritDoc}
     * 
     * @see javax.servlet.Filter#init(javax.servlet.FilterConfig)
     */
    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        this.defaultFilter.init(filterConfig);
    }

    /**
     * {@inheritDoc}
     * 
     * @see javax.servlet.Filter#doFilter(javax.servlet.ServletRequest,
     *      javax.servlet.ServletResponse, javax.servlet.FilterChain)
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {
        if (request instanceof HttpServletRequest && response instanceof HttpServletResponse) {
            HttpServletRequest req = (HttpServletRequest) request;
            HttpServletResponse res = (HttpServletResponse) response;

            // check if we have a token
            // if it is not present, we are not / no longer authenticated
            boolean isValidated = false;
            try {
                isValidated = this.configuration.crowdHttpAuthenticator.isAuthenticated(req, res);
            } catch (OperationFailedException ex) {
                LOG.log(Level.SEVERE, operationFailed(), ex);
            }

            if (!isValidated) {
                if (LOG.isLoggable(Level.FINE)) {
                    LOG.fine("User is not logged in (anymore) via Crowd => logout user");
                }
                SecurityContext sc = SecurityContextHolder.getContext();
                sc.setAuthentication(null);
                // close the SSO session
                if (null != this.rememberMe) {
                    this.rememberMe.logout(req, res);
                }

                // invalidate the current session
                // (see SecurityRealm#doLogout())
                HttpSession session = req.getSession(false);
                if (session != null) {
                    session.invalidate();
                }
                SecurityContextHolder.clearContext();

                // reset remember-me cookie
                Cookie cookie = new Cookie(SPRING_SECURITY_REMEMBER_ME_COOKIE_KEY, "");
                cookie.setPath(req.getContextPath().length() > 0 ? req.getContextPath() : "/");
                res.addCookie(cookie);
            } else {
                SecurityContext sc = SecurityContextHolder.getContext();

                if (!(sc.getAuthentication() instanceof CrowdAuthenticationToken)) {
                    // user logged in via Crowd, but no Crowd-specific
                    // authentication token available
                    // => try to auto-login the user
                    if (null != this.rememberMe) {
                        if (LOG.isLoggable(Level.FINE)) {
                            LOG.fine(
                                    "User is logged in via Crowd, but no authentication token available; trying auto-login...");
                        }
                        Authentication auth = this.rememberMe.autoLogin(req, res);
                        if (null != auth) {
                            if (LOG.isLoggable(Level.FINE)) {
                                LOG.fine("User sucessfully logged in");
                            }
                            sc.setAuthentication(auth);
                        }
                    }
                }
            }
        }

        this.defaultFilter.doFilter(request, response, chain);
    }

    /**
     * {@inheritDoc}
     * 
     * @see javax.servlet.Filter#destroy()
     */
    @Override
    public void destroy() {
        this.defaultFilter.destroy();
    }
}