使用ASM读取方法出口处的异常

Question

我正在编写一个小型java工具，该工具使用ASM 5.2记录所有方法入口和出口。下面是运行良好的程序。这个程序基本上使用了try-and-finally(无catch(块。onMethodEnter((，它记录方法条目。当整个方法被放入try/finally时，在finally块中记录方法退出。

import org.objectweb.asm.Label;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.commons.AdviceAdapter;
public class SGMethodAdapter extends AdviceAdapter {
private String entry;
private String exit;
private Label startFinally = new Label();
public SGMethodAdapter(int api, MethodVisitor mv, int access, String name, String desc, String className) {
super(api, mv, access, name, desc);
entry = ">" + className + ";" + name;
exit = "<" + className + ";" + name;
}
public void visitCode() {
super.visitCode();
mv.visitLabel(startFinally);
}
public void visitMaxs(int maxStack, int maxLocals) {
Label endFinally = new Label();
mv.visitTryCatchBlock(startFinally, endFinally, endFinally, null);
mv.visitLabel(endFinally);
onFinally(ATHROW);
mv.visitInsn(ATHROW);
mv.visitMaxs(maxStack, maxLocals);
}
private void onFinally(int opcode) {
try {
mv.visitLdcInsn(exit);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, "com/test/MethodEntryExitRecorder", "methodExit",
"(Ljava/lang/String;)V", false);
} catch (Throwable t) {
t.printStackTrace();
}
}
protected void onMethodEnter() {
try {
super.visitLdcInsn(entry);
super.visitMethodInsn(Opcodes.INVOKESTATIC, "com/test/MethodEntryExitRecorder", "methodEntry",
"(Ljava/lang/String;)V", false);
} catch (Throwable t) {
t.printStackTrace();
}
}
protected void onMethodExit(int opcode) {
if (opcode != ATHROW) {
onFinally(opcode);
}
}
}

现在我希望这个程序在退出之前也能读取任何方法抛出的异常(只是读取，而不是捕获(。如果没有抛出异常，我只想记录正常的方法退出。如果某个方法抛出异常，JVM保证在方法退出时，异常对象在堆栈顶部可用。下面是我为添加此行为而调整的相同程序。问题是，这只是读取少数例外，而不是全部。无法理解哪些异常已读取，哪些未读取。不确定代码中的错误在哪里。。。非常感谢您的帮助。。。

private void onFinally(int opcode) {
if(opcode == ATHROW) {
try {
mv.visitInsn(DUP);
mv.visitLdcInsn(exit);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, "com/test/MethodEntryExitRecorder", "methodExitWithException",
"(Ljava/lang/Throwable;Ljava/lang/String;)V", false);
} catch (Throwable t) {
t.printStackTrace();
}
} else {
try {
mv.visitLdcInsn(exit);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, "com/test/MethodEntryExitRecorder", "methodExit",
"(Ljava/lang/String;)V", false);
} catch (Throwable t) {
t.printStackTrace();
}
}
}
protected void onMethodEnter() {
try {
super.visitLdcInsn(entry);
super.visitMethodInsn(Opcodes.INVOKESTATIC, "com/test/MethodEntryExitRecorder", "methodEntry",
"(Ljava/lang/String;)V", false);
} catch (Throwable t) {
t.printStackTrace();
}
}
protected void onMethodExit(int opcode) {
onFinally(opcode);
}

更新：更好地解释问题

看来我对这个问题的解释不清楚。

给你。

1( 我正在开发一个记录所有java方法入口和出口的java工具(基本上在应用程序运行时生成一个动态调用图(

2( 为此，我不想更改应用程序代码。该工具应在不改变应用程序行为的情况下工作

3( 为了实现这一点，我正在使用ASM进行字节码检测

4( 该工具需要在方法体开始和结束时将字节码注入java应用程序的每个方法

5( 因为一个方法中可能有多个退出/返回点，所以在方法主体结尾注入记录代码是不够的，因为如果方法在主体的中途返回，方法退出将不会被记录

6( 为了克服这个问题，我把整个应用程序方法体放在try/finaly(使用仪器(中，没有捕获(因为应用程序需要捕获异常，而不是这个工具(

7( 上面的第一个程序是ASM的AdviceAdapter，它将在类加载时向应用程序的所有方法注入字节码

8( onMethodEnter((将注入记录代码，记录在运行时调用时输入的应用程序方法

9( onMethodExit((将在新注入的finally块中注入记录代码，以将应用程序的方法记录为在运行时退出的方法(在finally中确保始终执行此代码块(

到目前为止，这一切都很顺利。它正确地生成动态调用图。

现在，对于上面的第一个程序，我想注入额外的代码，这些代码也将读取应用程序/方法抛出的异常(类型和堆栈跟踪(。再说一遍，只读书不抓人。为此，我在上面的第二个程序中添加了这个附加代码。问题是，第二个程序只读取少数例外，而不是全部。不确定代码中的错误在哪里。非常感谢您的帮助。

谢谢，

Srinivas

Answer 1

您的需求反映了我的需求&我使用的是下面的代码，它记录了特定java方法抛出的异常-它只记录了使用try。。。接住块

package com.abc.agent.adapter;
import java.util.Map;
import com.abc.tm.Constants;
import com.abc.org.objectweb.asm.Label;
import com.abc.org.objectweb.asm.MethodVisitor;
import com.abc.org.objectweb.asm.Opcodes;
import com.abc.org.objectweb.asm.Type;
import com.abc.org.objectweb.asm.commons.AdviceAdapter;
public class PojoMethodAdviceAdapter extends AdviceAdapter  {
private String methodName;
private String className;
private String description;
private int okFlag =  newLocal(Type.BOOLEAN_TYPE); 
Label startFinally = new Label();
public PojoMethodAdviceAdapter(int access , MethodVisitor mv , String methodName, String description, String className , int classFileVersion, boolean learning, int lineNumber, Map pendingMethod){
super(Opcodes.ASM5 , mv, access, methodName, description);
this.className = className;
this.methodName = methodName;
this.description = description;
}
public void visitCode() {
super.visitCode(); 
mv.visitLabel(startFinally);
}
protected void onMethodEnter(){
mv.visitInsn(Opcodes.ICONST_0);
mv.visitVarInsn(ISTORE, okFlag);
mv.visitLdcInsn(className);
mv.visitLdcInsn(methodName);
mv.visitLdcInsn(description);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, Constants.RootTracer, Constants.pojoMethodBegin, Constants.pojoMethodBeginDesc, false);
mv.visitVarInsn(ISTORE, okFlag);
}
protected void onMethodExit(int opcode){
if(opcode!=ATHROW) {
onFinally(opcode);
}
}
public void visitMaxs(int maxStack, int maxLocals){
Label endFinally = new Label();
mv.visitTryCatchBlock(startFinally, endFinally, endFinally, null);
mv.visitLabel(endFinally);
onFinally(ATHROW);
mv.visitInsn(ATHROW);
mv.visitMaxs(maxStack+4, maxLocals);
}
private void onFinally(int opcode){
if(opcode == ATHROW){
mv.visitInsn(Opcodes.DUP);
mv.visitLdcInsn(className);
mv.visitLdcInsn(methodName);
mv.visitLdcInsn(description);
mv.visitVarInsn(ILOAD, okFlag);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, Constants.RootTracer, Constants.recordPOJOException, Constants.recordPOJOExceptionDesc, false);
}
mv.visitLdcInsn(className);
mv.visitLdcInsn(methodName);
mv.visitLdcInsn(description);
mv.visitVarInsn(ILOAD, okFlag);
mv.visitLdcInsn(opcode);
mv.visitMethodInsn(Opcodes.INVOKESTATIC, Constants.RootTracer, Constants.pojoMethodEnd, Constants.pojoMethodEndDesc, false);
}

}