1+ /*
2+ * Copyright 2006-2009 the original author or authors.
3+ *
4+ * Licensed under the Apache License, Version 2.0 (the "License");
5+ * you may not use this file except in compliance with the License.
6+ * You may obtain a copy of the License at
7+ *
8+ * http://www.apache.org/licenses/LICENSE-2.0
9+ *
10+ * Unless required by applicable law or agreed to in writing, software
11+ * distributed under the License is distributed on an "AS IS" BASIS,
12+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+ * See the License for the specific language governing permissions and
14+ * limitations under the License.
15+ */
16+ package org .springframework .instrument .classloading .glassfish ;
17+
18+ import java .lang .instrument .ClassFileTransformer ;
19+ import java .lang .reflect .InvocationTargetException ;
20+ import java .lang .reflect .Method ;
21+
22+ /**
23+ * Reflective wrapper around the GlassFish class loader. Used to
24+ * encapsulate the classloader-specific methods (discovered and
25+ * called through reflection) from the load-time weaver.
26+ *
27+ * <p/> Supports GlassFish V1, V2 and V3 (currently in beta).
28+ *
29+ * @author Costin Leau
30+ * @since 3.0.0
31+ */
32+ class GlassFishClassLoaderAdapter {
33+
34+ static final String INSTRUMENTABLE_CLASSLOADER_GLASSFISH_V2 = "com.sun.enterprise.loader.InstrumentableClassLoader" ;
35+ static final String INSTRUMENTABLE_CLASSLOADER_GLASSFISH_V3 = "org.glassfish.api.deployment.InstrumentableClassLoader" ;
36+ private static final String CLASS_TRANSFORMER = "javax.persistence.spi.ClassTransformer" ;
37+
38+ private final ClassLoader classLoader ;
39+ private final Method addTransformer ;
40+ private final Method copy ;
41+ private final boolean glassFishV3 ;
42+
43+ public GlassFishClassLoaderAdapter (ClassLoader classLoader ) {
44+ Class <?> instrumentableLoaderClass = null ;
45+ boolean glassV3 = false ;
46+ try {
47+ // try the V1/V2 API first
48+ instrumentableLoaderClass = classLoader .loadClass (INSTRUMENTABLE_CLASSLOADER_GLASSFISH_V2 );
49+ } catch (ClassNotFoundException ex ) {
50+ // fall back to V3
51+ try {
52+ instrumentableLoaderClass = classLoader .loadClass (INSTRUMENTABLE_CLASSLOADER_GLASSFISH_V3 );
53+ glassV3 = true ;
54+ } catch (ClassNotFoundException cnfe ) {
55+ throw new IllegalStateException (
56+ "Could not initialize GlassFish LoadTimeWeaver because GlassFish (V1, V2 or V3) API classes are not available" ,
57+ ex );
58+ }
59+ }
60+ try {
61+ Class <?> classTransformerClass = (glassV3 ? ClassFileTransformer .class : classLoader
62+ .loadClass (CLASS_TRANSFORMER ));
63+
64+ addTransformer = instrumentableLoaderClass .getMethod ("addTransformer" , classTransformerClass );
65+ copy = instrumentableLoaderClass .getMethod ("copy" );
66+
67+ } catch (Exception ex ) {
68+ throw new IllegalStateException (
69+ "Could not initialize GlassFish LoadTimeWeaver because GlassFish API classes are not available" , ex );
70+ }
71+
72+ ClassLoader clazzLoader = null ;
73+ // Detect transformation-aware ClassLoader by traversing the hierarchy
74+ // (as in GlassFish, Spring can be loaded by the WebappClassLoader).
75+ for (ClassLoader cl = classLoader ; cl != null && clazzLoader == null ; cl = cl .getParent ()) {
76+ if (instrumentableLoaderClass .isInstance (cl )) {
77+ clazzLoader = cl ;
78+ }
79+ }
80+
81+ if (clazzLoader == null ) {
82+ throw new IllegalArgumentException (classLoader + " and its parents are not suitable ClassLoaders: " + "A ["
83+ + instrumentableLoaderClass .getName () + "] implementation is required." );
84+ }
85+
86+ this .classLoader = clazzLoader ;
87+ this .glassFishV3 = glassV3 ;
88+ }
89+
90+ public void addTransformer (ClassFileTransformer transformer ) {
91+ try {
92+ addTransformer .invoke (classLoader , (glassFishV3 ? transformer : new ClassTransformerAdapter (transformer )));
93+ } catch (InvocationTargetException ex ) {
94+ throw new IllegalStateException ("GlassFish addTransformer method threw exception " , ex .getCause ());
95+ } catch (Exception ex ) {
96+ throw new IllegalStateException ("Could not invoke GlassFish addTransformer method" , ex );
97+ }
98+ }
99+
100+ public ClassLoader getClassLoader () {
101+ return this .classLoader ;
102+ }
103+
104+ public ClassLoader getThrowawayClassLoader () {
105+ try {
106+ return (ClassLoader ) copy .invoke (classLoader , (Object []) null );
107+ } catch (InvocationTargetException ex ) {
108+ throw new IllegalStateException ("GlassFish copy method threw exception " , ex .getCause ());
109+ } catch (Exception ex ) {
110+ throw new IllegalStateException ("Could not invoke GlassFish copy method" , ex );
111+ }
112+ }
113+ }
0 commit comments