package org.webswing.classloader;

import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Field;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.net.URLStreamHandler;
import java.security.CodeSource;
import java.security.ProtectionDomain;
import java.security.cert.Certificate;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;

import org.apache.bcel.Constants;
import org.apache.bcel.classfile.ClassParser;
import org.apache.bcel.classfile.Constant;
import org.apache.bcel.classfile.ConstantClass;
import org.apache.bcel.classfile.JavaClass;
import org.apache.bcel.classfile.Method;
import org.apache.bcel.generic.ALOAD;
import org.apache.bcel.generic.CHECKCAST;
import org.apache.bcel.generic.CPInstruction;
import org.apache.bcel.generic.ClassGen;
import org.apache.bcel.generic.ConstantPoolGen;
import org.apache.bcel.generic.INSTANCEOF;
import org.apache.bcel.generic.INVOKESTATIC;
import org.apache.bcel.generic.Instruction;
import org.apache.bcel.generic.InstructionConstants;
import org.apache.bcel.generic.InstructionFactory;
import org.apache.bcel.generic.InstructionHandle;
import org.apache.bcel.generic.InstructionList;
import org.apache.bcel.generic.MethodGen;
import org.apache.bcel.generic.ObjectType;
import org.apache.bcel.generic.Type;
import org.apache.bcel.util.ClassLoaderRepository;
import org.apache.commons.io.IOUtils;
import org.webswing.util.AppLogger;
import org.webswing.util.ClassLoaderUtil;

import com.google.common.collect.BiMap;
import com.google.common.collect.ImmutableBiMap;
import com.google.common.collect.ImmutableBiMap.Builder;

import sun.security.util.SecurityConstants;

@SuppressWarnings("restriction")
public class SwingClassloader extends URLClassLoader {

	private static BiMap<String, String> classReplacementMapping;
	private static BiMap<String, String> methodReplacementMapping;
	private static BiMap<String, String> methodOverrideMapping;
	private static URLStreamHandler originalJarHandler;

	static {
		Builder<String, String> classBuilder = new ImmutableBiMap.Builder<String, String>();
		// classBuilder.put("javax.swing.JComponent", "sk.web.swing.JComponent$$BCEL$$");
		// classBuilder.put("javax.swing.JButton", "sk.web.swing.JButton$$BCEL$$");
		// classBuilder.put("javax.swing.JCheckBoxMenuItem", "sk.web.swing.JCheckBoxMenuItem$$BCEL$$");
		// classBuilder.put("javax.swing.JMenu", "sk.web.swing.JMenu$$BCEL$$");
		// classBuilder.put("javax.swing.JMenuBar", "sk.web.swing.JMenuBar$$BCEL$$");
		// classBuilder.put("javax.swing.JMenuItem", "sk.web.swing.JMenuItem$$BCEL$$");
		// classBuilder.put("javax.swing.JToolBar", "sk.web.swing.JToolBar$$BCEL$$");
		// classBuilder.put("javax.swing.JPopupMenu", "sk.web.swing.JPopupMenu$$BCEL$$");
		// classBuilder.put("javax.swing.JRadioButtonMenuItem", "sk.web.swing.JRadioButtonMenuItem$$BCEL$$");
		// classBuilder.put("javax.swing.JToggleButton", "sk.web.swing.JToggleButton$$BCEL$$");
		// classBuilder.put("javax.swing.JCheckBox", "sk.web.swing.JCheckBox$$BCEL$$");
		// classBuilder.put("javax.swing.JRadioButton", "sk.web.swing.JRadioButton$$BCEL$$");
		// classBuilder.put("javax.swing.Box", "sk.web.swing.Box$$BCEL$$");
		// classBuilder.put("javax.swing.JColorChooser", "sk.web.swing.JColorChooser$$BCEL$$");
		// classBuilder.put("javax.swing.JFileChooser", "sk.web.swing.JFileChooser$$BCEL$$");
		// classBuilder.put("javax.swing.JComboBox", "sk.web.swing.JComboBox$$BCEL$$");
		// classBuilder.put("javax.swing.JLabel", "sk.web.swing.JLabel$$BCEL$$");
		// classBuilder.put("javax.swing.JTextField", "sk.web.swing.JTextField$$BCEL$$");
		// classBuilder.put("javax.swing.JTextArea", "sk.web.swing.JTextArea$$BCEL$$");
		// classBuilder.put("javax.swing.JTextPane", "sk.web.swing.JTextPane$$BCEL$$");
		// classBuilder.put("javax.swing.DefaultListCellRenderer", "sk.web.swing.DefaultListCellRenderer$$BCEL$$");
		// classBuilder.put("javax.swing.table.DefaultTableCellRenderer", "sk.web.swing.table.DefaultTableCellRenderer$$BCEL$$");
		// classBuilder.put("javax.swing.tree.DefaultTreeCellRenderer", "sk.web.swing.tree.DefaultTreeCellRenderer$$BCEL$$");
		// classBuilder.put("javax.swing.JList", "sk.web.swing.JList$$BCEL$$");
		// classBuilder.put("javax.swing.JLayeredPane", "sk.web.swing.JLayeredPane$$BCEL$$");
		// classBuilder.put("javax.swing.JDesktopPane", "sk.web.swing.JDesktopPane$$BCEL$$");
		// classBuilder.put("javax.swing.JRootPane", "sk.web.swing.JRootPane$$BCEL$$");
		// classBuilder.put("javax.swing.JSplitPane", "sk.web.swing.JSplitPane$$BCEL$$");
		// classBuilder.put("javax.swing.JPanel", "sk.web.swing.JPanel$$BCEL$$");
		// classBuilder.put("javax.swing.JScrollPane", "sk.web.swing.JScrollPane$$BCEL$$");
		// classBuilder.put("javax.swing.JScrollBar", "sk.web.swing.JScrollBar$$BCEL$$");
		// classBuilder.put("javax.swing.JViewPort", "sk.web.swing.JViewPort$$BCEL$$");
		// classBuilder.put("javax.swing.JOptionPane", "sk.web.swing.JOptionPane$$BCEL$$");
		// classBuilder.put("javax.swing.JFrame", "org.webswing.containers.WebJFrame");
		// classBuilder.put("javax.swing.JDialog", "org.webswing.containers.WebJDialog");
		// classBuilder.put("javax.swing.JWindow", "org.webswing.containers.WebJWindow");
		classReplacementMapping = classBuilder.build();

		Builder<String, String> methodReplacementBuilder = new ImmutableBiMap.Builder<String, String>();
		methodReplacementBuilder.put("java.lang.ClassLoader getSystemClassLoader ()Ljava/lang/ClassLoader;", "org.webswing.special.RedirectedMethods getSystemClassLoader ()Ljava/lang/ClassLoader;");
		methodReplacementBuilder.put("java.lang.ClassLoader getPlatformClassLoader ()Ljava/lang/ClassLoader;", "org.webswing.special.RedirectedMethods getPlatformClassLoader ()Ljava/lang/ClassLoader;");
		methodReplacementBuilder.put("com.intellij.openapi.util.ClassLoaderUtil getPlatformLoaderParentIfOnJdk9 ()Ljava/lang/ClassLoader;", "org.webswing.special.RedirectedMethods getPlatformLoaderParentIfOnJdk9 ()Ljava/lang/ClassLoader;");
		methodReplacementBuilder.put("java.lang.ClassLoader getSystemResource (Ljava/lang/String;)Ljava/net/URL;", "org.webswing.special.RedirectedMethods getSystemResource (Ljava/lang/String;)Ljava/net/URL;");
		methodReplacementBuilder.put("java.lang.ClassLoader getSystemResourceAsStream (Ljava/lang/String;)Ljava/io/InputStream;", "org.webswing.special.RedirectedMethods getSystemResourceAsStream (Ljava/lang/String;)Ljava/io/InputStream;");
		methodReplacementBuilder.put("java.lang.ClassLoader getSystemResources (Ljava/lang/String;)Ljava/util/Enumeration;", "org.webswing.special.RedirectedMethods getSystemResources (Ljava/lang/String;)Ljava/util/Enumeration;");
		if(!Boolean.getBoolean(org.webswing.Constants.SWING_START_SYS_ALLOW_REDIRECT_STD_OUT)) {
			methodReplacementBuilder.put("java.lang.System setErr (Ljava/io/PrintStream;)V", "org.webswing.special.RedirectedMethods setErr (Ljava/io/PrintStream;)V");
			methodReplacementBuilder.put("java.lang.System setOut (Ljava/io/PrintStream;)V", "org.webswing.special.RedirectedMethods setOut (Ljava/io/PrintStream;)V");
		}
		//methodReplacementBuilder.put("java.beans.XMLEncoder writeObject (Ljava/lang/Object;)V", "org.webswing.special.RedirectedMethods writeObject (Ljava/lang/Object;)V");
		methodReplacementBuilder.put("netscape.javascript.JSObject getWindow (Ljava/applet/Applet;)Lnetscape/javascript/JSObject;", "org.webswing.toolkit.jslink.WebJSObject getWindow (Ljava/applet/Applet;)Lnetscape/javascript/JSObject;");
		// ignore outline drag mode for JDesktopPane
		methodReplacementBuilder.put("javax.swing.JDesktopPane putClientProperty (Ljava/lang/Object;Ljava/lang/Object;)V", "org.webswing.special.RedirectedMethods putClientProperty (Ljavax/swing/JDesktopPane;Ljava/lang/Object;Ljava/lang/Object;)V");

		if (System.getProperty(org.webswing.Constants.SWING_START_SYS_PROP_ISOLATED_FS, "").equalsIgnoreCase("true")) {
			methodReplacementBuilder.put("java.io.File listRoots ()[Ljava/io/File;", "org.webswing.special.RedirectedMethods listRoots ()[Ljava/io/File;");
		}
		//printing methods
		//PrintService[] javax.print.PrintServiceLookup.lookupPrintServices(DocFlavor flavor, AttributeSet attributes)
		methodReplacementBuilder.put("javax.print.PrintServiceLookup lookupPrintServices (Ljavax/print/DocFlavor;Ljavax/print/attribute/AttributeSet;)[Ljavax/print/PrintService;", "org.webswing.special.RedirectedMethods lookupPrintServices (Ljavax/print/DocFlavor;Ljavax/print/attribute/AttributeSet;)[Ljavax/print/PrintService;");

		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/lang/Object;)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/lang/Object;)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", "org.webswing.special.RedirectedJOptionPane showInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;)V", "org.webswing.special.RedirectedJOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)V", "org.webswing.special.RedirectedJOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;)V", "org.webswing.special.RedirectedJOptionPane showMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;)I", "org.webswing.special.RedirectedJOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)I", "org.webswing.special.RedirectedJOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;II)I", "org.webswing.special.RedirectedJOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;II)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;)I", "org.webswing.special.RedirectedJOptionPane showConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showOptionDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)I", "org.webswing.special.RedirectedJOptionPane showOptionDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;)V", "org.webswing.special.RedirectedJOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)V", "org.webswing.special.RedirectedJOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;)V", "org.webswing.special.RedirectedJOptionPane showInternalMessageDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;)V");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;)I", "org.webswing.special.RedirectedJOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)I", "org.webswing.special.RedirectedJOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;II)I", "org.webswing.special.RedirectedJOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;II)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;)I", "org.webswing.special.RedirectedJOptionPane showInternalConfirmDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalOptionDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)I", "org.webswing.special.RedirectedJOptionPane showInternalOptionDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;IILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)I");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)Ljava/lang/String;", "org.webswing.special.RedirectedJOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;I)Ljava/lang/String;");
		// methodReplacementBuilder.put("javax.swing.JOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;", "org.webswing.special.RedirectedJOptionPane showInternalInputDialog (Ljava/awt/Component;Ljava/lang/Object;Ljava/lang/String;ILjavax/swing/Icon;[Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;");
		// methodReplacementBuilder.put("javax.swing.JColorChooser showDialog (Ljava/awt/Component;Ljava/lang/String;Ljava/awt/Color;)Ljava/awt/Color;", "org.webswing.special.RedirectedJColorChooser showDialog (Ljava/awt/Component;Ljava/lang/String;Ljava/awt/Color;)Ljava/awt/Color;");
		// methodReplacementBuilder.put("javax.swing.JColorChooser createDialog (Ljava/awt/Component;Ljava/lang/String;ZLjavax/swing/JColorChooser;Ljava/awt/event/ActionListener;Ljava/awt/event/ActionListener;)Ljavax/swing/JDialog;", "org.webswing.special.RedirectedJColorChooser createDialog (Ljava/awt/Component;Ljava/lang/String;ZLjavax/swing/JColorChooser;Ljava/awt/event/ActionListener;Ljava/awt/event/ActionListener;)Ljavax/swing/JDialog;");
		// methodReplacementBuilder.put("java.awt.Desktop isDesktopSupported ()Z", "org.webswing.special.RedirectedDesktop isDesktopSupported ()Z");
		// methodReplacementBuilder.put("java.awt.Desktop isSupported (Ljava/awt/Desktop/Action;)Z", "org.webswing.special.RedirectedDesktop isSupported (Ljava/awt/Desktop/Action;)Z");
		// methodReplacementBuilder.put("java.awt.Desktop browse (Ljava/net/URI;)V", "org.webswing.special.RedirectedDesktop browse (Ljava/net/URI;)V");
		// methodReplacementBuilder.put("java.awt.Desktop mail (Ljava/net/URI;)V", "org.webswing.special.RedirectedDesktop mail (Ljava/net/URI;)V");
		// methodReplacementBuilder.put("java.awt.Desktop mail ()V", "org.webswing.special.RedirectedDesktop mail ()V");
		methodReplacementMapping = methodReplacementBuilder.build();

		Builder<String, String> methodOverrideBuilder = new ImmutableBiMap.Builder<String, String>();
		// methodOverrideBuilder.put("sk.web.swing.JFileChooser$$BCEL$$ createDialog (Ljava/awt/Component;)Ljavax/swing/JDialog;","org.webswing.special.OverridenMethods createDialog");
		methodOverrideMapping = methodOverrideBuilder.build();

		try {
			List<Field> candidates = new ArrayList<Field>();
			for (Field f : URL.class.getDeclaredFields()) {
				if (f.getType() == URLStreamHandler.class) {
					candidates.add(f);
				}
			}
			Field f = candidates.get(0);
			f.setAccessible(true);
			originalJarHandler = (URLStreamHandler) f.get(new URL("jar:file:/sample.jar!/"));
		} catch (Throwable t) {
			AppLogger.error("Unable to resolve the original URL stream handler:", t);
		}
	}

	private Hashtable<String, Class<?>> classes = new Hashtable<String, Class<?>>(); // Hashtable is synchronized thus thread-safe
	private String[] ignored_packages;
	private ClassModificationRegister modRegister= new ClassModificationRegister();
	private ClassLoaderRepository repository;
	private final FallbackURLClassLoader repoClassLoader;

	public SwingClassloader(URL[] classpath, ClassLoader parent) {
		super(new URL[] {}, parent);
		this.ignored_packages = new String[] { "org.webswing.special.", "org.webswing.model.", "org.webswing.toolkit.", "netscape.javascript." };
		this.repoClassLoader = new FallbackURLClassLoader(classpath);

		this.repository = new ClassLoaderRepository(repoClassLoader);
	}

	protected synchronized Class<?> loadClass(String class_name, boolean resolve) throws ClassNotFoundException {
		Class<?> cl = null;
		/*
		 * First try: lookup hash table.
		 */
		if ((cl = classes.get(class_name)) == null) {
			/*
			 * Second try: Load system class using system class loader. You better don't mess around with them.
			 */
			try {
				cl = getSystemClassLoader().loadClass(class_name);
			} catch (ClassNotFoundException e) {
				// not found in system classloader
			}
			/*
			 * Third try: Load ignored packages outside SwingClassloader
			 */
			if (cl == null) {
				for (int i = 0; i < ignored_packages.length; i++) {
					if (class_name.startsWith(ignored_packages[i])) {
						cl = getParent().loadClass(class_name);
						break;
					}
				}
			}
			if (cl == null) {
				/*
				 * Fourth try: Special request?
				 */

				cl=findClass(class_name);

			}
			if (resolve) {
				resolveClass(cl);
			}
		}
		classes.put(class_name, cl);
		return cl;
	}

	@Override
	protected void addURL(URL url) {
		try {
			java.lang.reflect.Method m = repoClassLoader.getClass().getSuperclass().getDeclaredMethod("addURL", URL.class);
			m.setAccessible(true);
			m.invoke(repoClassLoader, url);
		} catch (Exception e) {
			AppLogger.error("Failed to addURL to SwingClassloader", e);
		}
	}

	protected void addAppURL(URL url) {
		addURL(url);
	}

	private JavaClass createClass(String class_name) {
		throw new RuntimeException("not creating classes");
	}

	protected JavaClass modifyClass(JavaClass clazz) {

		// do not modify classes placed in this list
		if (ClassLoaderUtil.isInPackage(clazz.getPackageName(), new String[] { "org.webswing" })) {
			return clazz;
		}

		definePackage(clazz.getPackageName());

		if (!classReplacementMapping.values().contains(clazz.getClassName())) {
			ClassGen cg = new ClassGen(clazz);
			ConstantPoolGen cp = cg.getConstantPool(); // cg creates constant pool
			InstructionFactory f = new InstructionFactory(cg);

			// +++++++++ Rerouting swing component classes to generated subclasses with intercepted paint methods +++++++
			// rerouteSwingClasses(clazz, cg, cp);

			// +++++++++ Intercept paint method if current class is subclass of JComponent +++++++
			// interceptPaintMethod(clazz, cg, cp, f);

			// +++++++++ Reroute (static) methods that require special handling +++++++++++++
			boolean rerouted = rerouteMehods(clazz, cg, cp, f);

			// +++++++++ Override methods that needs to be modified +++++++++++++
			//overrideMehods(cg, cp, f);

			// dump
			// try {
			// if (cg.getClassName().equals("org.netbeans.core.startup.TopLogging"))
			// cg.getJavaClass().dump(cg.getClassName() + ".class");
			// } catch (Exception ex) {
			// ex.printStackTrace();
			// }
			// end dump

			modRegister.setModificationState(clazz.getClassName(),rerouted);
			return cg.getJavaClass();
		} else {
			return clazz;
		}
	}

	private void definePackage(String pkgname) {
		Package s = getPackage(pkgname);
		if (s == null) {
			super.definePackage(pkgname, null, null, null, null, null, null, null);
		}
	}

	private void overrideMehods(ClassGen cg, ConstantPoolGen cp, InstructionFactory f) {
		for (String method : methodOverrideMapping.keySet()) {
			String[] md = method.split(" ");
			if (md[0].equals(cg.getClassName())) {
				String methodName = md[1];
				String signature = md[2];
				Type returnType = Type.getReturnType(signature);
				Type[] signatureTypes = Type.getArgumentTypes(signature);
				Method m = cg.containsMethod(methodName, signature);
				InstructionList il = new InstructionList();
				MethodGen mg;

				if (m == null) {
					mg = new MethodGen(Constants.ACC_PUBLIC, // access flags
							returnType, // return type
							signatureTypes, ClassLoaderUtil.createArgNames(signatureTypes.length), // arg types
							methodName, cg.getClassName(), // method, class
							il, cp);
				} else {
					mg = new MethodGen(m, cg.getClassName(), cp);
				}

				// call method defined in map
				String[] overridenMd = methodOverrideMapping.get(method).split(" ");
				String overridenClassName = overridenMd[0];
				String overridenMethodName = overridenMd[1];
				Type[] overridenArgs = new Type[signatureTypes.length + 1];
				for (int i = 0; i <= signatureTypes.length; i++) {
					overridenArgs[i] = i == 0 ? new ObjectType(classReplacementMapping.containsValue(cg.getClassName()) ? classReplacementMapping.inverse().get(cg.getClassName()) : cg.getClassName()) : signatureTypes[i - 1];
					il.append(new ALOAD(i));
				}
				il.append(f.createInvoke(overridenClassName, overridenMethodName, returnType, overridenArgs, Constants.INVOKESTATIC));
				if (!returnType.equals(Type.VOID)) {
					il.append(InstructionConstants.RETURN);
				}

				mg.setMaxStack();
				cg.addMethod(mg.getMethod());
				il.dispose();
			}
		}

	}

	private static boolean rerouteMehods(JavaClass clazz, ClassGen cg, ConstantPoolGen cp, InstructionFactory f) {
		Map<Integer, Integer> indexReplacementMap = new HashMap<Integer, Integer>();
		// find methods to replace in current class
		for (String methodDef : methodReplacementMapping.keySet()) {
			String[] md = methodDef.split(" ");
			int methodIndex = cp.lookupMethodref(md[0], md[1], md[2]);
			if (methodIndex != -1) {
				// method found in current class, create a replacement MethodRefConstant in cp
				String[] replace = methodReplacementMapping.get(methodDef).split(" ");
				int replacementIndex = cp.addMethodref(replace[0], replace[1], replace[2]);
				indexReplacementMap.put(methodIndex, replacementIndex);
			}
		}
		// replace methodRef indexes in instructions
		if (indexReplacementMap.size() > 0) {
			for (Method m : cg.getMethods()) {
				MethodGen mg = new MethodGen(m, clazz.getClassName(), cp);
				InstructionList il = mg.getInstructionList();
				if (il != null) {
					boolean dirtyFlag = false;
					for (Instruction instruction : il.getInstructions()) {
						if (instruction instanceof CPInstruction) {
							CPInstruction i = (CPInstruction) instruction;
							if (indexReplacementMap.containsKey(i.getIndex())) {
								InstructionHandle handle = ClassLoaderUtil.findInstructionHandle(il, i);
								INVOKESTATIC replacedInstruction = new INVOKESTATIC(indexReplacementMap.get(i.getIndex()));
								handle.setInstruction(replacedInstruction);
								dirtyFlag = true;
							}
						}
					}
					if (dirtyFlag) {
						il.setPositions();
						mg.setInstructionList(il);
						mg.setMaxStack();
						mg.setMaxLocals();
						mg.removeLineNumbers();
						cg.replaceMethod(m, mg.getMethod());
						il.dispose();
					}
				}
			}
			return true;
		}else {
			return false;
		}
	}

	@SuppressWarnings("unused")
	private void rerouteSwingClasses(JavaClass clazz, ClassGen cg, ConstantPoolGen cp) {
		// Reroute all swing components to our own proxy component
		// 1.add class constant to constant pool
		for (String jComponentName : classReplacementMapping.keySet()) {
			int jComponentClassConstantPosition = cp.lookupClass(jComponentName);
			if (jComponentClassConstantPosition != -1) {
				cp.addClass(classReplacementMapping.get(jComponentName));
			}
		}
		// 2.replace superclass type
		if (classReplacementMapping.containsKey(cg.getSuperclassName())) {
			cg.setSuperclassName(classReplacementMapping.get(cg.getSuperclassName()));
		}

		// 3.reroute instructions that refer to changed classes, except for all "instanceof"/"checkcast" instructions.
		for (Method m : cg.getMethods()) {
			MethodGen mg = new MethodGen(m, clazz.getClassName(), cp);
			InstructionList il = mg.getInstructionList();
			if (il != null) {
				boolean dirtyFlag = false;
				for (Instruction instruction : il.getInstructions()) {
					if (instruction instanceof CPInstruction) {
						CPInstruction i = (CPInstruction) instruction;
						Constant instConstant = cp.getConstant(i.getIndex());
						if (instConstant instanceof ConstantClass) {
							ConstantClass classConstant = (ConstantClass) instConstant;
							String referencedClassName = ((String) classConstant.getConstantValue(cp.getConstantPool())).replace("/", ".");
							if (classReplacementMapping.containsKey(referencedClassName)) {
								if (i instanceof INSTANCEOF || i instanceof CHECKCAST) {
									// dont change;
									continue;
								}
								String myProxyClassName = classReplacementMapping.get(referencedClassName);
								int myProxyConstantClassPosition = cp.lookupClass(myProxyClassName);
								InstructionHandle handle = ClassLoaderUtil.findInstructionHandle(il, i);
								i.setIndex(myProxyConstantClassPosition);
								handle.setInstruction(i);
								dirtyFlag = true;
							}
						}

					}
				}
				if (dirtyFlag) {
					il.setPositions();
					mg.setInstructionList(il);
					mg.setMaxStack();
					mg.setMaxLocals();
					mg.removeLineNumbers();
					cg.replaceMethod(m, mg.getMethod());
					il.dispose();
				}
			}
		}
	}

	@Override
	public InputStream getResourceAsStream(String name) {
		InputStream result = repoClassLoader.getResourceAsStream(name);
		if (result == null) {
			result = getParent().getResourceAsStream(name);
		}
		return result;
	}

	@Override
	public URL getResource(String name) {
		return repoClassLoader.getResource(name);
	}

	@Override
	public Enumeration<URL> getResources(String name) throws IOException {
		return repoClassLoader.getResources(name);
	}

	@Override
	protected Class<?> findClass(String class_name) throws ClassNotFoundException {
		Class<?> cl;
		byte[] bytes=null;

		if (class_name.indexOf("$$BCEL$$") >= 0) {
			bytes = createClass(class_name).getBytes();
		} else { // Fifth try: Load classes via repository
			try {
				if(modRegister.canSkipModification(class_name)) {
					if(class_name.lastIndexOf(".")>0) {
						definePackage(class_name.substring(0, class_name.lastIndexOf(".")));
					}
					bytes=loadClassBytes(class_name);
				}else {
					JavaClass clazz = repository.loadClass(class_name);
					clazz = modifyClass(clazz);
					repository.removeClass(clazz);
					bytes = clazz.getBytes();
				}
				modRegister.notifyClassLoaded(class_name);
			} catch (ClassNotFoundException e) {
				//in case the class was loaded from external source using defineClass (ie. remote EJB stub)
				cl = findLoadedClass(class_name);
				if (cl != null) {
					return cl;
				} else {
					throw new ClassNotFoundException(class_name);
				}
			}
		}
		if (bytes != null) {
			java.security.Permissions perms = new java.security.Permissions();
			perms.add(SecurityConstants.ALL_PERMISSION);
			String classFilePath = repoClassLoader.getResource(class_name.replace('.', '/') + ".class").toExternalForm();
			int jarSeparatorIndex = classFilePath.lastIndexOf('!');
			boolean inJar = classFilePath.startsWith("jar:");
			classFilePath = jarSeparatorIndex > 0 && inJar ? classFilePath.substring(4, jarSeparatorIndex) : classFilePath;
			CodeSource source = null;
			try {
				source = new CodeSource(new URL(classFilePath), new Certificate[] {});
			} catch (MalformedURLException e) {
				AppLogger.fatal("Exception resolving code source:", e);
				// should not happen
			}
			ProtectionDomain allPermDomain = new java.security.ProtectionDomain(source, perms);
			cl = defineClass(class_name, bytes, 0, bytes.length, allPermDomain);
		} else {
			cl = Class.forName(class_name);
		}
		return cl;
	}

	public byte[] loadClassBytes(String className) throws ClassNotFoundException {
		String classFile = className.replace('.', '/');
		try {
			InputStream is = repoClassLoader.getResourceAsStream(classFile + ".class");
			Throwable error = null;
			byte[] result=null;
			try {
				if (is == null) {
					throw new ClassNotFoundException(className + " not found.");
				}
				result = IOUtils.toByteArray(is);;
			} catch (Throwable e) {
				error = e;
				throw e;
			} finally {
				if (is != null) {
					if (error != null) {
						try {
							is.close();
						} catch (Throwable e) {
							error.addSuppressed(e);
						}
					} else {
						is.close();
					}
				}

			}

			return result;
		} catch (IOException e) {
			throw new ClassNotFoundException(className + " not found: " + e, e);
		}
	}

	public static boolean isModified(Class<?> c) throws Exception {
		String classFile = c.getName().replace('.', '/');
		InputStream is = c.getClassLoader().getResourceAsStream(classFile + ".class");
		ClassParser parser = new ClassParser(is, c.getName());
		JavaClass javaClass = parser.parse();
		ClassGen cg = new ClassGen(javaClass);
		ConstantPoolGen cp = cg.getConstantPool(); // cg creates constant pool
		InstructionFactory f = new InstructionFactory(cg);
		boolean result = rerouteMehods(javaClass, cg, cp, f);
		return result;
	}

	private static class FallbackURLClassLoader extends URLClassLoader{

		public FallbackURLClassLoader(URL[] urls) {
			super(urls);
		}

		@Override
		public URL findResource(String name) {
			// This is to prevent ClassCircularityError caused by custom jar URLStreamHandler (ie. in netbeans platform apps)
			URL unsafe = super.findResource(name);
			if (originalJarHandler != null && unsafe != null && unsafe.getProtocol().startsWith("jar")) {
				try {
					URL safeurl = new URL(null, unsafe.toString(), originalJarHandler);
					return safeurl;
				} catch (MalformedURLException e) {
					AppLogger.error("Converting " + unsafe.toString() + " to safe url failed", e);
				}
			}
			return unsafe;
		}



	}
}