asm中导出方法本地变量

在方法出现异常时需要把传入参数值和本地变量值异导出来供分析。之前因为测试的对象都是普通的值，写的loadLocalVarArray能正常工作。

这几天要在方法进入时inject一个long startTime供在方法退出时计算方法方法执行时间，发现如果本地变量是long和double本地变量栈中实际

占用两个位置，而nextLocal和firstLocal是本地变量栈的地址位置，根本无法知道实际的本地变量个数和具体的索引。

当然用for(int i=0;i< nextLocal-firstLocal;i++){

try{ loadLocal(i+firstLocal) }catch(Exception ex){........}

}利用异常来判断是否有某个localIndex是可以牧举出来的，但是这样太猥琐了。

查看GeneratorAdapter发现有一个私有yojgList<Type> localTypes，在visit的时候，本地变量的type会被放在这个数组中。但是存放的方式几乎和栈中差不多。假如有三个double的亦是，栈中是:

firstLocal + 0

firstLocal + 2

firstLocal + 4

这样nextLocal-firstLocal是6，但是loadLocal(6)肯定出异常。而这个list也是把一个type set到对应的位置。这样它就需要把前面没有的地方填null.设置后是这样的：

0 J

1 null

2 J

3 null

4 J

这个list的长度是5，可以正确地 loadLocal(index).但是它是一个私有变量，我只好给GeneratorAdapter增加一个

public int getLocalLength(){
return this.localTypes.size();
}

说明：因为instrumention功能是独立天任何app的，它引用的库在premain中调用，那么如果应用中使用了同样的库而版本不同，则无法再载

入相同的类，因为premain在main之前已经载入了，为了防止版本冲突，我们是把其它 app常用库重新package进来的，这个工作是非常必要的，所以修改源码仍然可以接受。

有了这个getLocalLength()就好办了：

void loadVarArray() {

Type OBJECT_TYPE = Type.getObjectType("java/lang/Object");
int varCnt = getLocalLength();
if (varCnt == 0) { //返回一个空数组
push(0);
newArray(OBJECT_TYPE);
return;
}

/*
* 先检查栈的空位(long和double占两个位，其一个标记为null),如果直接用栈长度指定数组
* 则占位的位置为null，返回出去程序不知道是占位的null还是一个localVar的值为null
*/
int arrLen = 0;
for (int i = 0; i < varCnt; i++) {
Type t = getLocalType(super.firstLocal + i);
if (t != null)
arrLen++;
}
push(arrLen); //实际需要的数组长度
newArray(OBJECT_TYPE);
int arrIndex = 0;
for (int i = 0; i < varCnt; i++) {
int index = super.firstLocal + i;
Type t = getLocalType(index);
if (t == null)
continue;

dup();
push(arrIndex++);
loadLocal(index, t);
box(t);
arrayStore(OBJECT_TYPE);
}
}

这样正确地把所有本地变量加载到数组中(数组现在在操作栈的栈顶，你可以访问它了)。

完整的实现：

package com.alibaba.instrumention.classop;
import com.alibaba.citrus.asm.MethodVisitor;
import com.alibaba.citrus.asm.Type;
import com.alibaba.citrus.asm.commons.AdviceAdapter;
import com.alibaba.instrumention.Util;
/**
 * Comment of InstrumtionAdviceAdapter
 * 
 * @author axman.wang
 */
public class InstrumentionAdviceAdapter extends AdviceAdapter {
    private String methodName;
    private String desc;
    private String className;
    private int    startLocal;
    protected InstrumentionAdviceAdapter(MethodVisitor mv, int access, String methodName,
                                         String desc, String signature, String[] exceptions,
                                         String className) {
        super(mv, access, methodName, desc);
        this.methodName = methodName;
        this.desc = desc;
        this.className = className;
    }
    protected void onMethodEnter() {
        /* invoke the statbaen */
        this.mv.visitTypeInsn(NEW, Util.METHODINFO_CLASSNAME);
        this.mv.visitInsn(DUP);
        this.mv.visitLdcInsn(this.methodName);
        this.mv.visitLdcInsn(this.desc);
        this.mv.visitLdcInsn(this.className);
        this.mv.visitMethodInsn(INVOKESPECIAL, Util.METHODINFO_CLASSNAME, Util.CSTOR_NAME,
                Util.METHODINFO_CSTOR_DESC);
        this.mv.visitMethodInsn(INVOKESTATIC, Util.STAT_CLASSNAME, Util.STAT_BEFORE_NAME,
                Util.STAT_BEFORE_DESC);
        /* make a timer start variable */
        this.startLocal = this.newLocal(Type.LONG_TYPE);
        this.mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "nanoTime", "()J");
        this.mv.visitVarInsn(LSTORE, startLocal);
    }
    protected void onMethodExit(int opcode) {
        if (opcode >= IRETURN && opcode <= RETURN) {
            this.mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "nanoTime", "()J");
            this.mv.visitVarInsn(LLOAD, startLocal);
            this.mv.visitInsn(LSUB);
            this.mv.visitTypeInsn(NEW, Util.METHODINFO_CLASSNAME);
            this.mv.visitInsn(DUP);
            this.mv.visitLdcInsn(this.methodName);
            this.mv.visitLdcInsn(this.desc);
            this.mv.visitLdcInsn(this.className);
            this.mv.visitMethodInsn(INVOKESPECIAL, Util.METHODINFO_CLASSNAME, Util.CSTOR_NAME,
                    Util.METHODINFO_CSTOR_DESC);
            this.mv.visitMethodInsn(INVOKESTATIC, Util.STAT_CLASSNAME, Util.STAT_AFTER_NAME,
                    Util.STAT_AFTER_DESC);
        } else if (opcode == ATHROW) {
            this.mv.visitInsn(DUP); //copy Exception instance of stack top
            this.mv.visitTypeInsn(NEW, Util.METHODINFO_CLASSNAME);
            this.mv.visitInsn(DUP);
            this.mv.visitLdcInsn(this.methodName);
            this.mv.visitLdcInsn(this.desc);
            this.mv.visitLdcInsn(this.className);
            this.mv.visitMethodInsn(INVOKESPECIAL, Util.METHODINFO_CLASSNAME, Util.CSTOR_NAME,
                    Util.METHODINFO_CSTOR_DESC);
            loadVarArray(true); //加载本地变量数组            
            loadArgArray(); //加载参数数组
            this.mv.visitMethodInsn(INVOKESTATIC, Util.STAT_CLASSNAME, Util.STAT_ERROR_NAME,
                    Util.STAT_ERROR_DESC);
        }
    }
    private void loadVarArray(boolean removeInjection) {
        Type OBJECT_TYPE = Type.getObjectType("java/lang/Object");
        int varCnt = getLocalLength();
        if (varCnt == 0) { //返回一个空数组
            push(0);
            newArray(OBJECT_TYPE);
            return;
        }
        /*
         * 先检查栈的空位(long和double占两个位，其一个标记为null),如果直接用栈长度指定数组
         * 则占位的位置为null，返回出去程序不知道是占位的null还是一个localVar的值为null
         */
        int arrLen = 0;
        for (int i = 0; i < varCnt; i++) {
            Type t = getLocalType(super.firstLocal + i);
            if (t != null)
                arrLen++;
        }
        if (removeInjection) {
            arrLen--; //除去注入的startLocal
        }
        push(arrLen); //实际需要的数组长度
        newArray(OBJECT_TYPE);
        int arrIndex = 0;
        boolean isfirst = true;
        for (int i = 0; i < varCnt; i++) {
            int index = super.firstLocal + i;
            Type t = getLocalType(index);
            if (t == null)
                continue;
            if (removeInjection && isfirst){
                isfirst = false;
                continue;
            }
            dup();
            push(arrIndex++);
            loadLocal(index, t);
            box(t);
            arrayStore(OBJECT_TYPE);
        }
    }
}