使用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;}@Overridepublic 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);}@Overridepublic 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.Studentcr = 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);}@Overridepublic 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);}@Overridepublic 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);}@Overridepublic 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);}}

结果：