盘古BPM体验地址    盘古BPM交流群盘古BPM交流群号:963222735

使用asm移除类中的方法

分享牛 1923℃

本文重点学习一下如何使用asm修改类中的字段或者方法,比如移除类中的方法,或者动态的添加字段,同其他文章一样,我们还是使用asm,这点一定要了解。

1、移除类中的方法。

我们可以通过ClassVisitor类中的visit方法解析并访问类的成员。我们首先看一下这个类中的核心方法,如下所示:

visitOuterClass

visitAnnotation

visitInnerClass

visitField

visitMethod

visitEnd

ClassVisitor类提供了一系列访问类方法、字段、属性、内部类、静态内部类的方法。我们只需要按照自己的业务进行访问处理即可,比如我现在在访问a方法,正常情况下,访问完毕之后,会返回一个MethodVisitor实例对象,如果我们修改改代码的实现逻辑,比如访问a方法的时候,我们让他返回null,那这样类似我们就屏蔽了a方法的内部实现细节了。我们不妨写一个小demo试下。

2.首先看一下Person类,实例代码如下:

public class Person {
    private int id;
    private String name;
    public int getId() {
        return id;
    }
    public void setId(int id) {
        this.id = id;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
}

我们现在期望做的事情,就是以Person类为模板生成一个新的类,新的类没有setId方法。需求有了那就开始写代码吧。

3.定义一个访问类并继承ClassVisitor类,实例代码如下:

public class ShareniuRemovingClassesVisitor extends ClassVisitor {
     public ShareniuRemovingClassesVisitor(int api) {
            super(Opcodes.ASM5);
     }
    public ShareniuRemovingClassesVisitor( ClassVisitor cv) {
        super(Opcodes.ASM5, cv);
    }
    @Override
     public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
            System.out.println("visit方法:");
            super.visit(version, access, name, signature, superName, interfaces);
    }
    @Override
    public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
        System.out.println("visitMethod:"+name);
        if (name.equals("setId")) {
            //如果是setId 直接返回null
                return null;
        }
        return super.visitMethod(access, name, desc, signature, exceptions);
    }
    @Override
    public FieldVisitor visitField(int access, String name, String desc, String signature, Object value) {
        return super.visitField(access, name, desc, signature, value);
    }
}


4.开始自定义一个类访问Person类,并生成新的类,然后我们观察输出结果,实例代码如下:

        ClassReader cr = new ClassReader("com.shareniu.asm.helloword.Person");  
            ClassWriter cw = new ClassWriter(0);  
           ClassVisitor cv = new ShareniuRemovingClassesVisitor(cw);
          ClassVisitor cv = new ShareniuMultiClassAdapter(cw);
        cr.accept(cv, 0);  
        byte[] toByte = cw.toByteArray();// byt 和toByte其实是相同的数组  
        // 输出到class文件  
        FileOutputStream fout = new FileOutputStream("Shareniu.class");
        fout.write(toByte);  
        fout.close();  
}

5.新生成的类结构如下:

    public class Person {

private int id;

private String name;

public int getId() {

return id;

}

public String getName() {

return name;

}

public void setName(String name) {

this.name = name;

}

至此,移除类中的方法已经ok。


 


转载请注明:分享牛 » 使用asm移除类中的方法