62 lines
2.2 KiB
Java
62 lines
2.2 KiB
Java
package ch.usi.inf.sp.dbi.agent;
|
|
|
|
import org.objectweb.asm.ClassReader;
|
|
import org.objectweb.asm.ClassWriter;
|
|
import org.objectweb.asm.Opcodes;
|
|
import org.objectweb.asm.tree.*;
|
|
|
|
import java.lang.instrument.ClassFileTransformer;
|
|
import java.lang.instrument.IllegalClassFormatException;
|
|
import java.security.ProtectionDomain;
|
|
|
|
public class ClassTransformer implements ClassFileTransformer {
|
|
public static byte[] instrument(final byte[] bytecode) {
|
|
final ClassReader cr = new ClassReader(bytecode);
|
|
final ClassNode cn = new ClassNode();
|
|
cr.accept(cn, ClassReader.SKIP_FRAMES);
|
|
|
|
instrumentClassNode(cn);
|
|
|
|
final ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
|
|
cn.accept(cw);
|
|
return cw.toByteArray();
|
|
}
|
|
|
|
private static void instrumentClassNode(final ClassNode cn) {
|
|
for (final MethodNode mn : cn.methods) {
|
|
InsnList patch = new InsnList(); // instructions to be added
|
|
patch.add(new LdcInsnNode("Method " + mn.name + mn.desc + " called"));
|
|
patch.add(new MethodInsnNode(Opcodes.INVOKESTATIC,
|
|
"ch/usi/inf/sp/dbi/profiler/Profiler",
|
|
"log",
|
|
"(Ljava/lang/String;)V"));
|
|
|
|
mn.instructions.insert(patch);
|
|
}
|
|
|
|
}
|
|
|
|
@Override
|
|
public byte[] transform(ClassLoader loader,
|
|
String className,
|
|
Class<?> classBeingRedefined,
|
|
ProtectionDomain protectionDomain,
|
|
byte[] classfileBuffer) throws IllegalClassFormatException {
|
|
if (className.startsWith("java/") ||
|
|
className.startsWith("sun/") ||
|
|
className.startsWith("ch/usi/inf/sp/dbi/agent/") ||
|
|
className.startsWith("ch/usi/inf/sp/dbi/profiler/")) {
|
|
System.out.println("Skipping class <" + loader + ", " + className + ">");
|
|
return classfileBuffer;
|
|
}
|
|
|
|
System.out.println("About to transform class <" + loader + ", " + className + ">");
|
|
try {
|
|
return instrument(classfileBuffer); // no changes
|
|
} catch (Throwable e) {
|
|
e.printStackTrace(System.err);
|
|
return classfileBuffer;
|
|
}
|
|
}
|
|
}
|