001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common Development
008: * and Distribution License("CDDL") (collectively, the "License"). You
009: * may not use this file except in compliance with the License. You can obtain
010: * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
011: * or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
012: * language governing permissions and limitations under the License.
013: *
014: * When distributing the software, include this License Header Notice in each
015: * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
016: * Sun designates this particular file as subject to the "Classpath" exception
017: * as provided by Sun in the GPL Version 2 section of the License file that
018: * accompanied this code. If applicable, add the following below the License
019: * Header, with the fields enclosed by brackets [] replaced by your own
020: * identifying information: "Portions Copyrighted [year]
021: * [name of copyright owner]"
022: *
023: * Contributor(s):
024: *
025: * If you wish your version of this file to be governed by only the CDDL or
026: * only the GPL Version 2, indicate your decision by adding "[Contributor]
027: * elects to include this software in this distribution under the [CDDL or GPL
028: * Version 2] license." If you don't indicate a single choice of license, a
029: * recipient has the option to distribute your version of this file under
030: * either the CDDL, the GPL Version 2 or to extend the choice of license to
031: * its licensees as provided above. However, if you add GPL Version 2 code
032: * and therefore, elected the GPL Version 2 license, then the option applies
033: * only if the new code is made subject to such option by the copyright
034: * holder.
035: */
036:
037: package com.sun.tools.xjc.api.util;
038:
039: import java.io.ByteArrayOutputStream;
040: import java.io.IOException;
041: import java.io.InputStream;
042: import java.io.File;
043: import java.net.URL;
044: import java.net.URLClassLoader;
045: import java.net.MalformedURLException;
046:
047: import com.sun.istack.Nullable;
048:
049: /**
050: * {@link ClassLoader} that loads APT and specified classes
051: * both into the same classloader, so that they can reference each other.
052: *
053: * @author Bhakti Mehta
054: * @since 2.0 beta
055: */
056: public final class APTClassLoader extends URLClassLoader {
057: /**
058: * List of package prefixes we want to mask the
059: * parent classLoader from loading
060: */
061: private final String[] packagePrefixes;
062:
063: /**
064: *
065: * @param packagePrefixes
066: * The package prefixes that are forced to resolve within this class loader.
067: * @param parent
068: * The parent class loader to delegate to. Null to indicate bootstrap classloader.
069: */
070: public APTClassLoader(@Nullable
071: ClassLoader parent, String[] packagePrefixes)
072: throws ToolsJarNotFoundException {
073: super (getToolsJar(parent), parent);
074: if (getURLs().length == 0)
075: // if tools.jar was found in our classloader, no need to create
076: // a parallel classes
077: this .packagePrefixes = new String[0];
078: else
079: this .packagePrefixes = packagePrefixes;
080: }
081:
082: public Class loadClass(String className)
083: throws ClassNotFoundException {
084: for (String prefix : packagePrefixes) {
085: if (className.startsWith(prefix)) {
086: // we need to load those classes in this class loader
087: // without delegation.
088: return findClass(className);
089: }
090: }
091:
092: return super .loadClass(className);
093:
094: }
095:
096: protected Class findClass(String name)
097: throws ClassNotFoundException {
098:
099: StringBuilder sb = new StringBuilder(name.length() + 6);
100: sb.append(name.replace('.', '/')).append(".class");
101:
102: InputStream is = getResourceAsStream(sb.toString());
103: if (is == null)
104: throw new ClassNotFoundException("Class not found" + sb);
105:
106: try {
107: ByteArrayOutputStream baos = new ByteArrayOutputStream();
108: byte[] buf = new byte[1024];
109: int len;
110: while ((len = is.read(buf)) >= 0)
111: baos.write(buf, 0, len);
112:
113: buf = baos.toByteArray();
114:
115: // define package if not defined yet
116: int i = name.lastIndexOf('.');
117: if (i != -1) {
118: String pkgname = name.substring(0, i);
119: Package pkg = getPackage(pkgname);
120: if (pkg == null)
121: definePackage(pkgname, null, null, null, null,
122: null, null, null);
123: }
124:
125: return defineClass(name, buf, 0, buf.length);
126: } catch (IOException e) {
127: throw new ClassNotFoundException(name, e);
128: }
129: }
130:
131: /**
132: * Returns a class loader that can load classes from JDK tools.jar.
133: * @param parent
134: */
135: private static URL[] getToolsJar(@Nullable
136: ClassLoader parent) throws ToolsJarNotFoundException {
137:
138: try {
139: Class.forName("com.sun.tools.javac.Main", false, parent);
140: Class.forName("com.sun.tools.apt.Main", false, parent);
141: return new URL[0];
142: // we can already load them in the parent class loader.
143: // so no need to look for tools.jar.
144: // this happens when we are run inside IDE/Ant, or
145: // in Mac OS.
146: } catch (ClassNotFoundException e) {
147: // otherwise try to find tools.jar
148: }
149:
150: File jreHome = new File(System.getProperty("java.home"));
151: File toolsJar = new File(jreHome.getParent(), "lib/tools.jar");
152:
153: if (!toolsJar.exists()) {
154: throw new ToolsJarNotFoundException(toolsJar);
155: }
156:
157: try {
158: return new URL[] { toolsJar.toURL() };
159: } catch (MalformedURLException e) {
160: // impossible
161: throw new AssertionError(e);
162: }
163: }
164: }
|