1. 程式人生 > >利用asm對jar進行修改

利用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不是很熟的人使用。