Re: Classloading issues...

/*
 * Copyright (c) 2004, Brian Hawkins
 * brianhks-BYHubErIwDGZvNlLnfxc/[email protected]
 * 
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free 
 * Software Foundation; either version 2 of the License, or (at your option)
 * any later version.
 *
 * This program 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 General Public License 
 * for more details.
 *
 * You should have received a copy of the GNU General Public License along 
 * with this program; if not, write to the 
 * Free Software Foundation, Inc., 
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */
 
package bhutils;
 
import java.net.*;
import java.util.*;
import java.io.*;

public class ClassLoaderWrapper extends ClassLoader
	{
	private static ClassLoaderWrapper s_loader = null;
	ClassLoader m_parentLoader;
	
	private ClassLoaderWrapper()
		{
		//super(new URL[0], null);
		super(null);
		m_parentLoader = getClass().getClassLoader();
		}
		
	
	public static ClassLoaderWrapper getClassLoader()
		{
		if (s_loader == null)
			s_loader = new ClassLoaderWrapper();
			
		return (s_loader);
		}
		
	public static ClassLoader getSystemClassLoader()
		{
		System.out.println("Returning system class loader");
		return (getClassLoader());
		}
		
	public static InputStream getSystemResourceAsStream(String name)
		{
		System.out.println("System resource");
		return (ClassLoader.getSystemResourceAsStream(name));
		}
		
	public static Enumeration getSystemResources(String name)
			throws IOException
		{
		System.out.println("System resource");
		return (ClassLoader.getSystemResources(name));
		}
		
	public URL getResource(String name)
		{
		//System.out.println("Getting resource "+name);
		return (m_parentLoader.getResource(name));
		}
		
	public InputStream getResourceAsStream(String name)
		{
		//System.out.println("Getting resource "+name);
		return (m_parentLoader.getResourceAsStream(name));
		}
		
	//Special case for ClassLoader and ClassLoaderWrapper
	public Class loadClass(String name)
			throws ClassNotFoundException
		{
		Class ret;
		System.out.println("Loading "+name);
		
		ret = findLoadedClass(name);
		
		/*if (name.equals("java.lang.ClassLoader"))
			{
			System.out.println("Returning ClassLoader");
			Thread.dumpStack();
			ret = this.getClass();
			}
		else*/ 
		if (name.startsWith("java."))
			ret = m_parentLoader.loadClass(name);
		
		if (ret == null)
			{
			InputStream classIS = m_parentLoader.getResourceAsStream(name.replace('.', '/')+".class");
		
			if (classIS == null)
				ret = super.loadClass(name);
			else
				{
				int data;
				ByteArrayOutputStream classOS = new ByteArrayOutputStream();
				
				try
					{
					while ((data = classIS.read()) != -1)
						classOS.write(data);
					}
				catch(IOException ioe)
					{
					throw new ClassNotFoundException(ioe.getMessage());
					}
					
				ret = defineClass(name, classOS.toByteArray(), 0, classOS.size());
				resolveClass(ret);
				}
			
			}
		
		return (ret);
		}
		
	}