| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| |
| package org.apache.catalina.session; |
| |
| import java.io.BufferedInputStream; |
| import java.io.BufferedOutputStream; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileNotFoundException; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.io.ObjectInputStream; |
| import java.io.ObjectOutputStream; |
| import java.security.AccessController; |
| import java.security.PrivilegedActionException; |
| import java.security.PrivilegedExceptionAction; |
| import java.util.ArrayList; |
| import java.util.Iterator; |
| |
| import javax.servlet.ServletContext; |
| |
| import org.apache.catalina.Context; |
| import org.apache.catalina.LifecycleException; |
| import org.apache.catalina.LifecycleState; |
| import org.apache.catalina.Loader; |
| import org.apache.catalina.Session; |
| import org.apache.catalina.security.SecurityUtil; |
| import org.apache.catalina.util.CustomObjectInputStream; |
| import org.apache.juli.logging.Log; |
| import org.apache.juli.logging.LogFactory; |
| import org.apache.tomcat.util.ExceptionUtils; |
| /** |
| * Standard implementation of the <b>Manager</b> interface that provides |
| * simple session persistence across restarts of this component (such as |
| * when the entire server is shut down and restarted, or when a particular |
| * web application is reloaded. |
| * <p> |
| * <b>IMPLEMENTATION NOTE</b>: Correct behavior of session storing and |
| * reloading depends upon external calls to the <code>start()</code> and |
| * <code>stop()</code> methods of this class at the correct times. |
| * |
| * @author Craig R. McClanahan |
| * @author Jean-Francois Arcand |
| * @version $Id: StandardManager.java,v 1.1 2011/06/28 21:08:19 rherrmann Exp $ |
| */ |
| |
| public class StandardManager extends ManagerBase { |
| |
| private final Log log = LogFactory.getLog(StandardManager.class); // must not be static |
| |
| // ---------------------------------------------------- Security Classes |
| private class PrivilegedDoLoad |
| implements PrivilegedExceptionAction<Void> { |
| |
| PrivilegedDoLoad() { |
| // NOOP |
| } |
| |
| @Override |
| public Void run() throws Exception{ |
| doLoad(); |
| return null; |
| } |
| } |
| |
| private class PrivilegedDoUnload |
| implements PrivilegedExceptionAction<Void> { |
| |
| PrivilegedDoUnload() { |
| // NOOP |
| } |
| |
| @Override |
| public Void run() throws Exception{ |
| doUnload(); |
| return null; |
| } |
| |
| } |
| |
| |
| // ----------------------------------------------------- Instance Variables |
| |
| |
| /** |
| * The descriptive information about this implementation. |
| */ |
| protected static final String info = "StandardManager/1.0"; |
| |
| |
| /** |
| * The descriptive name of this Manager implementation (for logging). |
| */ |
| protected static String name = "StandardManager"; |
| |
| |
| /** |
| * Path name of the disk file in which active sessions are saved |
| * when we stop, and from which these sessions are loaded when we start. |
| * A <code>null</code> value indicates that no persistence is desired. |
| * If this pathname is relative, it will be resolved against the |
| * temporary working directory provided by our context, available via |
| * the <code>javax.servlet.context.tempdir</code> context attribute. |
| */ |
| protected String pathname = "SESSIONS.ser"; |
| |
| |
| // ------------------------------------------------------------- Properties |
| |
| |
| /** |
| * Return descriptive information about this Manager implementation and |
| * the corresponding version number, in the format |
| * <code><description>/<version></code>. |
| */ |
| @Override |
| public String getInfo() { |
| |
| return (info); |
| |
| } |
| |
| |
| /** |
| * Return the descriptive short name of this Manager implementation. |
| */ |
| @Override |
| public String getName() { |
| |
| return (name); |
| |
| } |
| |
| |
| /** |
| * Return the session persistence pathname, if any. |
| */ |
| public String getPathname() { |
| |
| return (this.pathname); |
| |
| } |
| |
| |
| /** |
| * Set the session persistence pathname to the specified value. If no |
| * persistence support is desired, set the pathname to <code>null</code>. |
| * |
| * @param pathname New session persistence pathname |
| */ |
| public void setPathname(String pathname) { |
| |
| String oldPathname = this.pathname; |
| this.pathname = pathname; |
| support.firePropertyChange("pathname", oldPathname, this.pathname); |
| |
| } |
| |
| |
| // --------------------------------------------------------- Public Methods |
| |
| /** |
| * Load any currently active sessions that were previously unloaded |
| * to the appropriate persistence mechanism, if any. If persistence is not |
| * supported, this method returns without doing anything. |
| * |
| * @exception ClassNotFoundException if a serialized class cannot be |
| * found during the reload |
| * @exception IOException if an input/output error occurs |
| */ |
| @Override |
| public void load() throws ClassNotFoundException, IOException { |
| if (SecurityUtil.isPackageProtectionEnabled()){ |
| try{ |
| AccessController.doPrivileged( new PrivilegedDoLoad() ); |
| } catch (PrivilegedActionException ex){ |
| Exception exception = ex.getException(); |
| if (exception instanceof ClassNotFoundException){ |
| throw (ClassNotFoundException)exception; |
| } else if (exception instanceof IOException){ |
| throw (IOException)exception; |
| } |
| if (log.isDebugEnabled()) |
| log.debug("Unreported exception in load() " |
| + exception); |
| } |
| } else { |
| doLoad(); |
| } |
| } |
| |
| |
| /** |
| * Load any currently active sessions that were previously unloaded |
| * to the appropriate persistence mechanism, if any. If persistence is not |
| * supported, this method returns without doing anything. |
| * |
| * @exception ClassNotFoundException if a serialized class cannot be |
| * found during the reload |
| * @exception IOException if an input/output error occurs |
| */ |
| protected void doLoad() throws ClassNotFoundException, IOException { |
| if (log.isDebugEnabled()) |
| log.debug("Start: Loading persisted sessions"); |
| |
| // Initialize our internal data structures |
| sessions.clear(); |
| |
| // Open an input stream to the specified pathname, if any |
| File file = file(); |
| if (file == null) |
| return; |
| if (log.isDebugEnabled()) |
| log.debug(sm.getString("standardManager.loading", pathname)); |
| FileInputStream fis = null; |
| BufferedInputStream bis = null; |
| ObjectInputStream ois = null; |
| Loader loader = null; |
| ClassLoader classLoader = null; |
| try { |
| fis = new FileInputStream(file.getAbsolutePath()); |
| bis = new BufferedInputStream(fis); |
| if (container != null) |
| loader = container.getLoader(); |
| if (loader != null) |
| classLoader = loader.getClassLoader(); |
| if (classLoader != null) { |
| if (log.isDebugEnabled()) |
| log.debug("Creating custom object input stream for class loader "); |
| ois = new CustomObjectInputStream(bis, classLoader); |
| } else { |
| if (log.isDebugEnabled()) |
| log.debug("Creating standard object input stream"); |
| ois = new ObjectInputStream(bis); |
| } |
| } catch (FileNotFoundException e) { |
| if (log.isDebugEnabled()) |
| log.debug("No persisted data file found"); |
| return; |
| } catch (IOException e) { |
| log.error(sm.getString("standardManager.loading.ioe", e), e); |
| if (fis != null) { |
| try { |
| fis.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| } |
| if (bis != null) { |
| try { |
| bis.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| } |
| throw e; |
| } |
| |
| // Load the previously unloaded active sessions |
| synchronized (sessions) { |
| try { |
| Integer count = (Integer) ois.readObject(); |
| int n = count.intValue(); |
| if (log.isDebugEnabled()) |
| log.debug("Loading " + n + " persisted sessions"); |
| for (int i = 0; i < n; i++) { |
| StandardSession session = getNewSession(); |
| session.readObjectData(ois); |
| session.setManager(this); |
| sessions.put(session.getIdInternal(), session); |
| session.activate(); |
| if (!session.isValidInternal()) { |
| // If session is already invalid, |
| // expire session to prevent memory leak. |
| session.setValid(true); |
| session.expire(); |
| } |
| sessionCounter++; |
| } |
| } catch (ClassNotFoundException e) { |
| log.error(sm.getString("standardManager.loading.cnfe", e), e); |
| try { |
| ois.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| throw e; |
| } catch (IOException e) { |
| log.error(sm.getString("standardManager.loading.ioe", e), e); |
| try { |
| ois.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| throw e; |
| } finally { |
| // Close the input stream |
| try { |
| ois.close(); |
| } catch (IOException f) { |
| // ignored |
| } |
| |
| // Delete the persistent storage file |
| if (file.exists() ) |
| file.delete(); |
| } |
| } |
| |
| if (log.isDebugEnabled()) |
| log.debug("Finish: Loading persisted sessions"); |
| } |
| |
| |
| /** |
| * Save any currently active sessions in the appropriate persistence |
| * mechanism, if any. If persistence is not supported, this method |
| * returns without doing anything. |
| * |
| * @exception IOException if an input/output error occurs |
| */ |
| @Override |
| public void unload() throws IOException { |
| if (SecurityUtil.isPackageProtectionEnabled()){ |
| try{ |
| AccessController.doPrivileged( new PrivilegedDoUnload() ); |
| } catch (PrivilegedActionException ex){ |
| Exception exception = ex.getException(); |
| if (exception instanceof IOException){ |
| throw (IOException)exception; |
| } |
| if (log.isDebugEnabled()) |
| log.debug("Unreported exception in unLoad() " |
| + exception); |
| } |
| } else { |
| doUnload(); |
| } |
| } |
| |
| |
| /** |
| * Save any currently active sessions in the appropriate persistence |
| * mechanism, if any. If persistence is not supported, this method |
| * returns without doing anything. |
| * |
| * @exception IOException if an input/output error occurs |
| */ |
| protected void doUnload() throws IOException { |
| |
| if (log.isDebugEnabled()) |
| log.debug(sm.getString("standardManager.unloading.debug")); |
| |
| if (sessions.isEmpty()) { |
| log.debug(sm.getString("standardManager.unloading.nosessions")); |
| return; // nothing to do |
| } |
| |
| // Open an output stream to the specified pathname, if any |
| File file = file(); |
| if (file == null) |
| return; |
| if (log.isDebugEnabled()) |
| log.debug(sm.getString("standardManager.unloading", pathname)); |
| FileOutputStream fos = null; |
| ObjectOutputStream oos = null; |
| try { |
| fos = new FileOutputStream(file.getAbsolutePath()); |
| oos = new ObjectOutputStream(new BufferedOutputStream(fos)); |
| } catch (IOException e) { |
| log.error(sm.getString("standardManager.unloading.ioe", e), e); |
| if (fos != null) { |
| try { |
| fos.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| } |
| throw e; |
| } |
| |
| // Write the number of active sessions, followed by the details |
| ArrayList<StandardSession> list = new ArrayList<StandardSession>(); |
| synchronized (sessions) { |
| if (log.isDebugEnabled()) |
| log.debug("Unloading " + sessions.size() + " sessions"); |
| try { |
| oos.writeObject(new Integer(sessions.size())); |
| Iterator<Session> elements = sessions.values().iterator(); |
| while (elements.hasNext()) { |
| StandardSession session = |
| (StandardSession) elements.next(); |
| list.add(session); |
| session.passivate(); |
| session.writeObjectData(oos); |
| } |
| } catch (IOException e) { |
| log.error(sm.getString("standardManager.unloading.ioe", e), e); |
| try { |
| oos.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| throw e; |
| } |
| } |
| |
| // Flush and close the output stream |
| try { |
| oos.flush(); |
| } finally { |
| try { |
| oos.close(); |
| } catch (IOException f) { |
| // Ignore |
| } |
| } |
| |
| // Expire all the sessions we just wrote |
| if (log.isDebugEnabled()) |
| log.debug("Expiring " + list.size() + " persisted sessions"); |
| Iterator<StandardSession> expires = list.iterator(); |
| while (expires.hasNext()) { |
| StandardSession session = expires.next(); |
| try { |
| session.expire(false); |
| } catch (Throwable t) { |
| ExceptionUtils.handleThrowable(t); |
| } finally { |
| session.recycle(); |
| } |
| } |
| |
| if (log.isDebugEnabled()) |
| log.debug("Unloading complete"); |
| |
| } |
| |
| |
| /** |
| * Start this component and implement the requirements |
| * of {@link org.apache.catalina.util.LifecycleBase#startInternal()}. |
| * |
| * @exception LifecycleException if this component detects a fatal error |
| * that prevents this component from being used |
| */ |
| @Override |
| protected synchronized void startInternal() throws LifecycleException { |
| |
| super.startInternal(); |
| |
| // Load unloaded sessions, if any |
| try { |
| load(); |
| } catch (Throwable t) { |
| ExceptionUtils.handleThrowable(t); |
| log.error(sm.getString("standardManager.managerLoad"), t); |
| } |
| |
| setState(LifecycleState.STARTING); |
| } |
| |
| |
| /** |
| * Stop this component and implement the requirements |
| * of {@link org.apache.catalina.util.LifecycleBase#stopInternal()}. |
| * |
| * @exception LifecycleException if this component detects a fatal error |
| * that prevents this component from being used |
| */ |
| @Override |
| protected synchronized void stopInternal() throws LifecycleException { |
| |
| if (log.isDebugEnabled()) |
| log.debug("Stopping"); |
| |
| setState(LifecycleState.STOPPING); |
| |
| // Write out sessions |
| try { |
| unload(); |
| } catch (Throwable t) { |
| ExceptionUtils.handleThrowable(t); |
| log.error(sm.getString("standardManager.managerUnload"), t); |
| } |
| |
| // Expire all active sessions |
| Session sessions[] = findSessions(); |
| for (int i = 0; i < sessions.length; i++) { |
| Session session = sessions[i]; |
| try { |
| if (session.isValid()) { |
| session.expire(); |
| } |
| } catch (Throwable t) { |
| ExceptionUtils.handleThrowable(t); |
| } finally { |
| // Measure against memory leaking if references to the session |
| // object are kept in a shared field somewhere |
| session.recycle(); |
| } |
| } |
| |
| // Require a new random number generator if we are restarted |
| super.stopInternal(); |
| } |
| |
| |
| // ------------------------------------------------------ Protected Methods |
| |
| |
| /** |
| * Return a File object representing the pathname to our |
| * persistence file, if any. |
| */ |
| protected File file() { |
| |
| if ((pathname == null) || (pathname.length() == 0)) |
| return (null); |
| File file = new File(pathname); |
| if (!file.isAbsolute()) { |
| if (container instanceof Context) { |
| ServletContext servletContext = |
| ((Context) container).getServletContext(); |
| File tempdir = (File) |
| servletContext.getAttribute(ServletContext.TEMPDIR); |
| if (tempdir != null) |
| file = new File(tempdir, pathname); |
| } |
| } |
| // if (!file.isAbsolute()) |
| // return (null); |
| return (file); |
| |
| } |
| } |