使用ASM生成Java字節碼（可以通過代碼動態生成java字節碼）

使用ASM生成Java字節碼，下面通過java程序模擬實現spring aop 動態織入

 

Account.java

package com.market.search.test;

/**
 * Created by xiaotian on 2018/5/8.
 */
public class Account {
    public void operation() {
        //通過字節碼生成技術 動態添加check功能  模擬aop 動態織入
        System.out.println("operation....");
    }

}

SecurityChecker.java

package com.market.search.test;

/**
 * Created by xiaotian on 2018/5/8.
 * 將checkSecurity 通過ASE方式動態 增強Account.operation 中
 */
public class SecurityChecker {

    public static boolean checkSecurity() {
        System.out.println("SecurityChecker.checkSecurity ...");
        return true;
    }

}

AddSecurityCheckClassAdapter.java  類適配

package com.market.search.test;

import jdk.internal.org.objectweb.asm.ClassVisitor;
import jdk.internal.org.objectweb.asm.MethodVisitor;
import jdk.internal.org.objectweb.asm.Opcodes;

/**
 * Created by xiaotian on 2018/5/8.
 */
public class AddSecurityCheckClassAdapter extends ClassVisitor {
    public AddSecurityCheckClassAdapter( ClassVisitor cv) {
        super(Opcodes.ASM5, cv);
    }
    // 重寫 visitMethod，訪問到 "operation" 方法時，
    // 給出自定義 MethodVisitor，實際改寫方法內容
    public MethodVisitor visitMethod(final int access, final String name,
                                     final String desc, final String signature, final String[] exceptions) {
        MethodVisitor mv = cv.visitMethod(access, name, desc, signature,exceptions);
        MethodVisitor wrappedMv = mv;
        if (mv != null) {
            // 對於 "operation" 方法
            if (name.equals("operation")) {
                // 使用自定義 MethodVisitor，實際改寫方法內容
                wrappedMv = new AddSecurityCheckMethodAdapter(mv);
            }
        }
        return wrappedMv;
    }

}

AddSecurityCheckMethodAdapter.java 方法適配

package com.market.search.test;

import jdk.internal.org.objectweb.asm.MethodVisitor;
import jdk.internal.org.objectweb.asm.Opcodes;

/**
 * Created by xiaotian on 2018/5/8.
 */
public class AddSecurityCheckMethodAdapter extends MethodVisitor {
    public AddSecurityCheckMethodAdapter(MethodVisitor mv) {
        super(Opcodes.ASM5,mv);
    }
    public void visitCode() {
        visitMethodInsn(Opcodes.INVOKESTATIC, "com/market/search/test/SecurityChecker",
                "checkSecurity", "()Z");
        super.visitCode();
    }

}

Generator.java  測試生成字節碼

package com.market.search.test;

import jdk.internal.org.objectweb.asm.ClassReader;
import jdk.internal.org.objectweb.asm.ClassWriter;

import java.io.File;
import java.io.FileOutputStream;
import java.net.URL;
import java.net.URLClassLoader;

/**
 * Created by xiaotian on 2018/5/8.
 */
public class Generator {
    private static AccountGeneratorClassLoader classLoader = new AccountGeneratorClassLoader();
    public static void main(String args[]) throws Exception {
        ClassReader cr = new ClassReader("com.market.search.test.Account");
        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS|ClassWriter.COMPUTE_FRAMES);
        AddSecurityCheckClassAdapter classAdapter = new AddSecurityCheckClassAdapter(cw);
        cr.accept(classAdapter, ClassReader.SKIP_DEBUG);

        byte[] data = cw.toByteArray();
        //File file = new File("C:\\Users\\xiaotian\\Desktop\\jvm原理\\Account.class");
        File file = new File("D:\\devlop\\workspace.market.new\\market-search-main-BL\\search-service_1.0.2_BR\\target\\classes\\com\\market\\search\\test\\Account.class");
        FileOutputStream fout = new FileOutputStream(file);
        fout.write(data);
        fout.close();

//        Class account = classLoader.defineClassFromClassFile("com.market.search.test.Account", data);
//        Account acc =(Account) account.newInstance();
//        acc.operation();
        Account account = new Account();
        account.operation();


    }
    private static class AccountGeneratorClassLoader extends ClassLoader {
        public Class defineClassFromClassFile(String className, byte[] classFile) throws ClassFormatError {
            return defineClass(className, classFile, 0,classFile.length);
        }
    }
}