edu.columbia.cs.psl.phosphor.PartialInstrumentationInferencerCV Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of embedded-server Show documentation
The newest version!
package edu.columbia.cs.psl.phosphor;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;

/**
 * Infers additional methods to instrument apart from the list fed to phosphor
 * 
 * @author manojthakur
 *
 */
public class PartialInstrumentationInferencerCV extends ClassVisitor{
	public static Map> classToSuperClass = new HashMap>();
	private static String superClassesToConsider = "java/io/.+|java/nio/.+|java/util/.+";
	
	List methodCallingAsStream = new ArrayList();
	String className;
	boolean isInterface = false;
	Map> map = new HashMap>();
	String[] interfaces = null;
	String superClass;
	List superClasses = new ArrayList();
	
	public PartialInstrumentationInferencerCV()  {
		super(Configuration.ASM_VERSION);
	}
	
	public PartialInstrumentationInferencerCV(final ClassVisitor cv) {
		super(Configuration.ASM_VERSION, cv);
	}
		
	@Override
	public void visit(int version, int access, String name, String signature,
			String superName, String[] interfaces) {
		this.className = name;
		this.interfaces = interfaces;
		this.superClass = superName;
		for(String inface : interfaces) {
			superClasses.add(inface);
		}
		
		if((access & Opcodes.ACC_INTERFACE) != 0 || (access & Opcodes.ACC_ABSTRACT) != 0)
			isInterface = true;
		
		super.visit(version, access, name, signature, superName, interfaces);
	}
	
	@Override
	public MethodVisitor visitMethod(int access, String name, String desc,
			String signature, String[] exceptions) {
			
		MethodDescriptor mdesc = new MethodDescriptor(name, className, desc);
		
		if(this.className.contains("NioEndpoint") && superClass.equals("java/util/concurrent/ConcurrentLinkedQueue") && name.equals("offer"))
			SelectiveInstrumentationManager.methodsToInstrument.add(mdesc);
		if(this.className.equals("org/apache/catalina/startup/Catalina"))
			SelectiveInstrumentationManager.methodsToInstrument.add(mdesc);
		
		
//		if(!this.className.startsWith("java/") && !this.className.startsWith("sun/") && !this.className.startsWith("javax/")) {
//			Set supers = ClassHierarchyCreator.allSupers(this.className);
//			
//			for(String inter : supers) {
//				MethodDescriptor supr = new MethodDescriptor(name, inter, desc);
//				if(SelectiveInstrumentationManager.methodsToInstrument.contains(supr)) {
//					SelectiveInstrumentationManager.methodsToInstrument.add(mdesc);
//					break;
//				}	
//			}
//		}
		
//		for(String inter : interfaces) {
//			MethodDescriptor supr = new MethodDescriptor(name, inter, desc);
//			if(SelectiveInstrumentationManager.methodsToInstrument.contains(supr)) {
//				SelectiveInstrumentationManager.methodsToInstrument.add(mdesc);
//				break;
//			}	
//		}
//		MethodDescriptor supr = new MethodDescriptor(name, superClass, desc);
//		
//		if(SelectiveInstrumentationManager.methodsToInstrument.contains(supr))
//			SelectiveInstrumentationManager.methodsToInstrument.add(mdesc);
		
		MethodVisitor next = super.visitMethod(access, name, desc, signature, exceptions);
		map.put(mdesc, new ArrayList());
		return next;
//		return new PartialInstrumentationInferencerMV(Configuration.ASM_VERSION, mdesc, next, map, this.superClasses);
	}
}