Skip to content

Instantly share code, notes, and snippets.

@phase
Forked from VijayKrishna/ReturnAdapter.java
Last active September 20, 2015 21:05
Show Gist options
  • Save phase/894af8f66b133eb27c10 to your computer and use it in GitHub Desktop.
Save phase/894af8f66b133eb27c10 to your computer and use it in GitHub Desktop.
Example code showing how the AdviceAdapter in ASM(.ow2.org) can be used/extended.
package self.vpalepu.stackoverflow;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.commons.AdviceAdapter;
/**
* Class visitor that adapts a java class to insert code before
* the return instructions in the methods of the class.
* @author vijay
*
*/
public class ReturnAdapter extends ClassVisitor {
private String className;
public ReturnAdapter(ClassVisitor cv, String className) {
super(Opcodes.ASM4, cv);
}
@Override
public MethodVisitor visitMethod(
int access,
String name,
String desc,
String signature,
String[] exceptions) {
MethodVisitor mv;
mv = cv.visitMethod(access, name, desc, signature, exceptions);
mv = new MethodReturnAdapter(Opcodes.ASM4, className, access, name, desc, mv);
return mv;
}
public static void main(String[] args) throws IOException {
String classFile = args[0];//path of the class file
String className = args[1];//name of the class
File inFile = new File(classFile);
FileInputStream in = new FileInputStream(inFile);
// adapting the class.
ClassReader cr = new ClassReader(in);
ClassWriter cw = new ClassWriter(ClassReader.EXPAND_FRAMES);
ReturnAdapter returnAdapter = new ReturnAdapter(cw, className);
cr.accept(returnAdapter, 0);
}
}
/**
* Method Visitor that inserts code right before its return instruction(s),
* using the onMethodExit(int opcode) method of the AdviceAdapter class,
* from ASM(.ow2.org).
* @author vijay
*
*/
class MethodReturnAdapter extends AdviceAdapter {
public MethodReturnAdapter(
int api,
String owner,
int access,
String name,
String desc,
MethodVisitor mv) {
super(Opcodes.ASM4, mv, access, name, desc);
}
public MethodReturnAdapter(
MethodVisitor mv,
int access,
String name,
String desc) {
super(Opcodes.ASM4, mv, access, name, desc);
}
@Override
protected void onMethodExit(int opcode) {
if(opcode != Opcodes.ATHROW) {
mv.visitVarInsn(Opcodes.ALOAD, 42);
// and/or any other visit instructions.
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment