利用asm对jar进行修改

一.导读

有些时候，需要对jar进行修改，并加入一些“自定义”的代码，利用asm工具就可以到。

二.流程图

三.操作

步骤1，2 直接跳过了，这里从步骤3开始讲。首先需要Idea中安装一个插件Asm byteCode outline，这个工具可以通过.class文件生成我们需要的asm代码。

• 目前有个Source.class文件,这个函数比较简单，就是输出一个hello,如下所示（这个代码是根据idea反编译得到的），我现在想把它的输出语句进行删除掉，现在就开始。

package source;

public class SourceClass {
    public SourceClass() {
    }

    public static void test() {
        System.out.println("hello");
    }
}

• 首先，通过asm工具可以看到其asm操作代码，如下图所示
  ，如果右边区域为空，则在Source.class编辑区域，右键-》show ByteCode outline

• 然后，拷贝右边的asm代码中dump方法，并且新建一个GenerateClass类,这个类的主要功能是对dump方法进行修改，并输出二进制到Source.class中，这个Source.class文件就是我们修改后的文件，如下所示

package source;

import demo.FileUtils;
import org.objectweb.asm.*;

import static org.objectweb.asm.Opcodes.*;

public class GenerateClass {

    public static void main(String[] args) throws Exception {

        FileUtils.writeFile(GenerateClass.dump(),"SourceChange.class");
    }


    public static byte[] dump() throws Exception {

        ClassWriter cw = new ClassWriter(0);
        FieldVisitor fv;
        MethodVisitor mv;
        AnnotationVisitor av0;

        cw.visit(52, ACC_PUBLIC + ACC_SUPER, "source/SourceClass", null, "java/lang/Object", null);

        cw.visitSource("SourceClass.java", null);

        {
            mv = cw.visitMethod(ACC_PUBLIC, "<init>", "()V", null, null);
            mv.visitCode();
            Label l0 = new Label();
            mv.visitLabel(l0);
            mv.visitLineNumber(3, l0);
            mv.visitVarInsn(ALOAD, 0);
            mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "<init>", "()V", false);
            mv.visitInsn(RETURN);
            Label l1 = new Label();
            mv.visitLabel(l1);
            mv.visitLocalVariable("this", "Lsource/SourceClass;", null, l0, l1, 0);
            mv.visitMaxs(1, 1);
            mv.visitEnd();
        }
        {
            mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "test", "()V", null, null);
            mv.visitCode();
            Label l0 = new Label();
            mv.visitLabel(l0);
            mv.visitLineNumber(7, l0);
            mv.visitFieldInsn(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
            mv.visitLdcInsn("hello");
            mv.visitMethodInsn(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false);
            Label l1 = new Label();
            mv.visitLabel(l1);
            mv.visitLineNumber(8, l1);
            mv.visitInsn(RETURN);
            mv.visitMaxs(2, 0);
            mv.visitEnd();
        }
        cw.visitEnd();

        return cw.toByteArray();
    }
}

• 接下去是对找到输出方法的执行语句，通过方法名字，可以定位到是这几句

		    mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "test", "()V", null, null);
            mv.visitCode();
            Label l0 = new Label();
            mv.visitLabel(l0);
            mv.visitLineNumber(7, l0);
            mv.visitFieldInsn(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
            mv.visitLdcInsn("hello");
            mv.visitMethodInsn(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false);
            Label l1 = new Label();
            mv.visitLabel(l1);
            mv.visitLineNumber(8, l1);
            mv.visitInsn(RETURN);
            mv.visitMaxs(2, 0);
            mv.visitEnd();

直接去掉打印的语句

            Label l0 = new Label();
            mv.visitLabel(l0);
            mv.visitLineNumber(7, l0);
            mv.visitFieldInsn(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
            mv.visitLdcInsn("hello");
            mv.visitMethodInsn(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false);

• 最后可以得到，新的Source.class，如图所示
  

• 接下去就可以用这个class对原有的jar进行替换了，通过执行指令 jar -uf A.jar Source.class (这里注意下，如果A中的Source.class的路径为a/Source.class，那么这个指令的路径也应为 jar -uf A.jar a/Source.class)

结束语

• 此文只是做了一个最简单的举例。比起直接使用Asm的api对.class文件进行修改，我觉得这个方法还是比较偷懒的，效率也比较高。适合对asm不是很熟的人使用。