使用ASM为一个类增加属性工具类

简介: 使用ASM为一个类增加属性工具类

使用ASM可以方便的访问和修改一个类,或者在JVM载入前动态更新一个类也是可以的,如下,

为一个类动态增加一个属性,类Student如下:

public class Student {
  private int age;
  private String name;
 
  public int getAge() {
    return age;
  }
 
  public void setAge(int age) {
    this.age = age;
  }
 
  public String getName() {
    return name;
  }
 
  public void setName(String name) {
    this.name = name;
  }
}

使用ASM实现一个类写入属性AddFieldAdapter,如下:

public class AddFieldAdapter extends ClassAdapter {
    private int accessModifier;
    private String name;
    private String desc;
    private boolean isFieldPresent;
 
    public AddFieldAdapter(ClassVisitor cv, int accessModifier, String name, String desc) {
        super(cv);
        this.accessModifier = accessModifier;
        this.name = name;
        this.desc = desc;
    }
 
    @Override
    public FieldVisitor visitField(int access, String name, String desc,
            String signature, Object value) {
        if (name.equals(this.name)) {
            isFieldPresent = true;
        }
        return cv.visitField(access, name, desc, signature, value);
    }
 
    @Override
    public void visitEnd() {
        if (!isFieldPresent) {
          // 若属性不存在则写入
            FieldVisitor fv = cv.visitField(accessModifier, name, desc, null, null);
            if (fv != null) {
                fv.visitEnd();
            }
        }
        cv.visitEnd();
    }
} 

实现一个类AddField调用上面的类去添加属性,如下:

public class AddField {
  private Class<?> clazz;
    private ClassReader cr;
    private ClassWriter cw;
    private ClassAdapter ca;
    private File classFile;
 
    private final static String CLASS_FILE_SUFFIX = ".class";
 
    public AddField(Class<?> clazz) {
        this.clazz = clazz;
    }
 
    public void addPublicField(String fieldName, String fieldDesc) {
        if(cr == null) {
            try {
              // com.demo2.Student
                cr = new ClassReader(clazz.getCanonicalName());
            } catch (IOException e) {
                e.printStackTrace();
            }
            cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS);
        }
        if(ca == null) {
            ca = new AddFieldAdapter(cw, Opcodes.ACC_PUBLIC, fieldName, fieldDesc);
        } else {
            ca = new AddFieldAdapter(ca, Opcodes.ACC_PUBLIC, fieldName, fieldDesc);
        }
    }
 
    public void writeByteCode() {
        cr.accept(ca, ClassReader.SKIP_DEBUG);
        byte[] bys = cw.toByteArray();
        OutputStream os = null;
        try {
            os = new FileOutputStream(getFile());
            os.write(bys);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                os.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
 
    private File getFile() {
        if(classFile == null) {
            StringBuffer sb = new StringBuffer();
            sb.append(clazz.getResource("/"))
                .append(clazz.getCanonicalName().replace(".", File.separator))
                .append(CLASS_FILE_SUFFIX);
            classFile = new File(sb.substring(6));
        }
        return classFile;
    } 
 
 
    //测试方法
    public static void main(String[] args) {
        // 为 Student 添加字段
        AddField add = new AddField(Student.class);
        // 添加一个名为 address,类型为 java.lang.String 的 public 字段
        add.addPublicField("address", "Ljava/lang/String;");
        // 再增加一个名为 tel,类型为 int 的 public 方法
        add.addPublicField("tel", "I");
        // 重新生成 .class 文件
        add.writeByteCode();
    }
}

使用反编译工具查看生成的CLASS文件,即可看到动态添加的属性了。


ASM操作类属性

我们看到了ASM中的Core API中使用的是XXXVisitor操作类中的对应部分。本文将展示如何使用ASM中的Core API对类的属性的操作。

首先,我们定义一个类Student,如下:

public class Student {
  public String name = "Jimmy";
  public String address = "China";
}

这里,我们将属性定义为public类型,目的是为了我们使用反射去调用这个属性,接下来我们要为这个类添加一个int类型的属性,名字叫age。

   第一个问题,ASM的Core API允许我们在那些方法中来添加属性?

   在ASM的Core API中你要为类添加属性就必须要自己去实现ClassVisitor这个接口,这个接口中的visitInnerClass、visitField、visitMethod和visitEnd方法允许我们进行添加一个类属性操作,其余的方法是不允许的。这里我们依然使用Core API中的ClassAdapter类,我们继承这个类,定义一个去添加属性的类,ClassAdapter实现了ClassVisitor。

   第二个问题,我们要在这些方法中写什么样的代码才能添加一个属性?

   在使用ASM的Core API添加一个属性时只需要调用一句语句就可以,如下

 
classVisitor.visitField(Opcodes.ACC_PUBLIC, "age", Type.getDescriptor(int.class),  
                null, null);  

第一个参数指定的是这个属性的操作权限,第二个参数是属性名,第三个参数是类型描述,第四个参数是泛型类型,第五个参数是初始化的值,这里需要注意一下的是第五个参数,这个参数只有属性为static时才有效,也就是说只有为static时,这个值才真正会赋值给我们添加的属性上,对于非static属性,它将被忽略。

我们现在用两种方式来添加一个名字为age的属性:

第一种方式:

用ClassWriter直接创建

ClassReader cr = new ClassReader(Student.class.getName());
    ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS);
    ClassVisitor cv = new AddFieldClassAdapter(cw);
    cr.accept(cv, Opcodes.ASM4);
    byte[] code = cw.toByteArray();
    AddField loader = new AddField();
    Class<?> exampleClass = loader.defineClass(Student.class.getName(), code, 0, code.length);
    Object obj = exampleClass.newInstance();
    System.out.println(exampleClass.getField("name").get(obj));
    System.out.println(exampleClass.getField("age").get(obj));

在这里,如果我们的age没有被添加进去那么(2)这个地方将会报错,看一下结果:

Jimmy

0

第二种方式,写一个adapter过滤器,重写visitEnd方法

public class AddFieldClassAdapter extends ClassVisitor implements Opcodes{
 
  public AddFieldClassAdapter(ClassVisitor cv) {
    super(Opcodes.ASM4, cv);
  }
  
  
  @Override
  public void visitEnd() {
    cv.visitField(Opcodes.ACC_PUBLIC, "age", Type.getDescriptor(int.class),  
                null, null);
    super.visitEnd();
  }
 
}
public class AddField extends ClassLoader implements Opcodes{
  public static void main(String args[]) throws IOException, SecurityException, NoSuchFieldException, InstantiationException, IllegalAccessException{
    ClassReader cr = new ClassReader(Student.class.getName());
    ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS);
    ClassVisitor cv = new AddFieldClassAdapter(cw);
    cr.accept(cv, Opcodes.ASM4);
 
    byte[] code = cw.toByteArray();
    AddField loader = new AddField();
    Class<?> exampleClass = loader.defineClass(Student.class.getName(), code, 0, code.length);
    Object obj = exampleClass.newInstance();
    System.out.println(exampleClass.getField("name").get(obj));
    System.out.println(exampleClass.getField("age").get(obj));
  }
}

结果如下:

Jimmy

0

 和第一种方式返回的结果一一致。

接下来,我们在visitField方法中再次添加age属性,如下:

public class AddFieldClassAdapter extends ClassVisitor implements Opcodes{
 
  public AddFieldClassAdapter(ClassVisitor cv) {
    super(Opcodes.ASM4, cv);
  }
  
  @Override
  public FieldVisitor visitField(int access, String name, String desc,
            String signature, Object value) {
    cv.visitField(Opcodes.ACC_PUBLIC, "age", Type.getDescriptor(int.class),  
              null, null);
    return super.visitField(access, name, desc, signature, value);
    }
  
 
 
}

这时,我们再次运行测试类,结果如下:

Exception in thread "main" java.lang.ClassFormatError: Duplicate field name&signature in class file field/Student

   at java.lang.ClassLoader.defineClass1(Native Method)

   at java.lang.ClassLoader.defineClassCond(ClassLoader.java:631)

   at java.lang.ClassLoader.defineClass(ClassLoader.java:615)

   at java.lang.ClassLoader.defineClass(ClassLoader.java:465)

   at field.AddField.main(AddField.java:21)

 没有重复的名字,而我们添加的是age也不重复,为什么会报重复属性名错误呢?

原因是,在我们的Transform类中的visitField方法,这个方法会在每次属性被访问时调用,而ASM在对这个类操作时会遍历到每个属性,也就是说有一个属性就会调用一次visitField方法,有两个属性就会调用两次visitField方法,所以当我们原类中有两个属性时visitField方法被调用了两次,因此创建了两个同名的age属性。

为了验证visitField是被多次调用, 我那么可以做一个实验,在visitField()方法中加入System.out.println("name:" + name);把每次调用的属性名称都打印出来。

public class AddFieldClassAdapter extends ClassVisitor implements Opcodes{
 
  public AddFieldClassAdapter(ClassVisitor cv) {
    super(Opcodes.ASM4, cv);
  }
  
  @Override
  public FieldVisitor visitField(int access, String name, String desc,
            String signature, Object value) {
    System.out.println("name:" + name);
 
    return super.visitField(access, name, desc, signature, value);
    }
  
 
 
}

结果:

name:name

name:address

把Student所有的属性都打印出来,这证明了visitField()是被多次调用的!

相关文章
|
4月前
|
SQL Java
访问者模式问题之在ASM中,实现一个访问者来删除指定的类属性,如何解决
访问者模式问题之在ASM中,实现一个访问者来删除指定的类属性,如何解决
|
6月前
|
Java
使用ASM动态生成类的拦截代理类
使用ASM动态生成类的拦截代理类
75 0
|
6月前
|
Java 数据库连接 mybatis
使用ASM动态创建接口实现类
使用ASM动态创建接口实现类
48 0
|
6月前
|
Java API Android开发
ASM 框架:字节码操作的常见用法(生成类,修改类,方法插桩,方法注入)
ASM 框架:字节码操作的常见用法(生成类,修改类,方法插桩,方法注入)
107 0
|
6月前
|
Java Android开发
编写ASM代码工具类
编写ASM代码工具类
51 0
|
Java
[ASM教程]#1分析类
ASM是一种通用Java字节码操作和分析框架,它可以用于修改现有的class文件或动态生成class文件。
140 0
[ASM教程]#3增加移除类成员
可以委托一个ClassWriter。
79 0
[ASM教程]#2生成类
使用classWriter生成类
73 0
|
7月前
|
Oracle 关系型数据库
oracle asm 磁盘显示offline
oracle asm 磁盘显示offline
349 2
|
2月前
|
存储 Oracle 关系型数据库
数据库数据恢复—Oracle ASM磁盘组故障数据恢复案例
Oracle数据库数据恢复环境&故障: Oracle ASM磁盘组由4块磁盘组成。Oracle ASM磁盘组掉线 ,ASM实例不能mount。 Oracle数据库故障分析&恢复方案: 数据库数据恢复工程师对组成ASM磁盘组的磁盘进行分析。对ASM元数据进行分析发现ASM存储元数据损坏,导致磁盘组无法挂载。