/* 
 * Mutability Detector
 *
 * Copyright 2009 Graham Allan
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * 		http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package org.mutabilitydetector.asmoverride;

import org.mutabilitydetector.cli.URLFallbackClassLoader;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.analysis.SimpleVerifier;

public class CustomClassLoadingSimpleVerifier extends SimpleVerifier {

	@Override
	protected Class<?> getClass(Type t) {

		URLFallbackClassLoader classLoader = new URLFallbackClassLoader();
		String className;
		
		try {
			if (t.getSort() == Type.ARRAY) {
				className = t.getDescriptor().replace('/', '.');
			} else {
				className = t.getClassName();
			}
			return classLoader.getClass(className);
		} catch (ClassNotFoundException e) {
			throw new RuntimeException(e);
		}
	}
}

/* 
 * Mutability Detector
 *
 * Copyright 2009 Graham Allan
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * 		http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package org.mutabilitydetector.asmoverride;

import org.mutabilitydetector.cli.URLFallbackClassLoader;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.analysis.SimpleVerifier;

public class CustomClassLoadingSimpleVerifier extends SimpleVerifier {

	
	private URLFallbackClassLoader classLoader;
	
	public CustomClassLoadingSimpleVerifier() {
		classLoader = new URLFallbackClassLoader();
	}

	@Override
	protected Class<?> getClass(Type t) {
		String className;
		
		try {
			if (t.getSort() == Type.ARRAY) {
				className = t.getDescriptor().replace('/', '.');
			} else {
				className = t.getClassName();
			}
			return classLoader.getClass(className);
		} catch (ClassNotFoundException e) {
			throw new RuntimeException(e);
		}
	}
}

Improve performance of the URLFallbackClassLoader. Non-scientific stopwatch tests showed a >10% speed increase, translating to a whopping seven seconds off when analysing rt.jar.