[jboss-cvs] JBossAS SVN: r69328 - projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins.

[jboss-cvs-commits at lists.jboss.org]

Author: adrian at jboss.org
Date: 2008-01-24 19:13:06 -0500 (Thu, 24 Jan 2008)
New Revision: 69328

Added:
   projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/SecurityActions.java
Modified:
   projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/AbstractController.java
Log:
[JBMICROCONT-230] - Use any specifc classloader as the context classloader for the bean when performing callbacks

Modified: projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/AbstractController.java
===================================================================
--- projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/AbstractController.java	2008-01-25 00:02:59 UTC (rev 69327)
+++ projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/AbstractController.java	2008-01-25 00:13:06 UTC (rev 69328)
@@ -1168,8 +1168,10 @@
     */
    protected void resolveCallbacks(ControllerContext context, ControllerState state, boolean isInstallPhase)
    {
+      ClassLoader previous = null;
       try
       {
+         previous = SecurityActions.setContextClassLoader(context);
          // existing owner callbacks
          Set<CallbackItem<?>> installs = getDependencyCallbacks(context, true);
          resolveCallbacks(installs, state, isInstallPhase, isInstallPhase, true);
@@ -1217,6 +1219,11 @@
       {
          log.warn("Cannot resolve callbacks.", t);
       }
+      finally
+      {
+         if (previous != null)
+            SecurityActions.resetContextClassLoader(previous);
+      }
    }
 
    protected void handleInstallLifecycleCallbacks(ControllerContext context, ControllerState state) throws Throwable

Added: projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/SecurityActions.java
===================================================================
--- projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/SecurityActions.java	                        (rev 0)
+++ projects/microcontainer/trunk/dependency/src/main/org/jboss/dependency/plugins/SecurityActions.java	2008-01-25 00:13:06 UTC (rev 69328)
@@ -0,0 +1,104 @@
+/*
+* JBoss, Home of Professional Open Source
+* Copyright 2007, JBoss Inc., and individual contributors as indicated
+* by the @authors tag. See the copyright.txt in the distribution for a
+* full listing of individual contributors.
+*
+* This 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.
+*
+* This software 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 this software; if not, write to the Free
+* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+*/
+package org.jboss.dependency.plugins;
+
+import java.security.AccessController;
+import java.security.PrivilegedAction;
+import java.security.PrivilegedActionException;
+import java.security.PrivilegedExceptionAction;
+
+import org.jboss.dependency.spi.ControllerContext;
+import org.jboss.dependency.spi.dispatch.InvokeDispatchContext;
+
+/**
+ * SecurityActions.
+ * 
+ * @author <a href="adrian at jboss.com">Adrian Brock</a>
+ * @version $Revision: 1.1 $
+ */
+class SecurityActions
+{
+   static ClassLoader setContextClassLoader(final ControllerContext context) throws Throwable
+   {
+      if (System.getSecurityManager() == null)
+      {
+         ClassLoader result = Thread.currentThread().getContextClassLoader();
+         if (context instanceof InvokeDispatchContext)
+            Thread.currentThread().setContextClassLoader(((InvokeDispatchContext) context).getClassLoader());
+         return result;
+      }
+      else
+      {
+         try
+         {
+            return AccessController.doPrivileged(new PrivilegedExceptionAction<ClassLoader>()
+            {
+                public ClassLoader run() throws Exception
+                {
+                   try
+                   {
+                      ClassLoader result = Thread.currentThread().getContextClassLoader();
+                      if (context instanceof InvokeDispatchContext)
+                         Thread.currentThread().setContextClassLoader(((InvokeDispatchContext) context).getClassLoader());
+                      return result;
+                   }
+                   catch (Exception e)
+                   {
+                      throw e;
+                   }
+                   catch (Error e)
+                   {
+                      throw e;
+                   }
+                   catch (Throwable e)
+                   {
+                      throw new RuntimeException("Error setting context classloader", e);
+                   }
+                }
+            });
+         }
+         catch (PrivilegedActionException e)
+         {
+            throw e.getCause();
+         }
+      }
+   }
+
+   static void resetContextClassLoader(final ClassLoader classLoader)
+   {
+      if (System.getSecurityManager() == null)
+      {
+         Thread.currentThread().setContextClassLoader(classLoader);
+      }
+      else
+      {
+         AccessController.doPrivileged(new PrivilegedAction<Object>()
+         {
+             public Object run()
+             {
+                Thread.currentThread().setContextClassLoader(classLoader);
+                return null;
+             }
+         });
+      }
+   }
+}