graalvm学习（十) 如何通过java代码获取java字节码

本文链接：https://blog.csdn.net/D_I_D/article/details/135318444

ps: demo还是使用之前的HelloWord

分析GraalVm的实现

1、获取java的类
从第五节的-Djava.system.class.loader=com.oracle.svm.hosted.NativeImageSystemClassLoader参数可以知道，java是通过自定义类加载器来获取加载到jvm的类。通过测试可以知道继承ClassLoader也可以实现自定义类加载器。

2、获取类中对应的方法
从第四节可以知道，Graal的实现是通过jvmci接口实现的。不过这里有一个问题，只有热点代码才会触发compileMethod，但在生成可执行文件时根本没有热点代码，这里也就不会触发，无法获取到字节码。怎么解决？
通过研究AnalysisUniverse的lookup方法可以了解到GraalVm也是通过jvmci接口获取字节码，只不过方法有点不一样。
下图为lookup方法，可以看到lookup接收的是一个JavaMethod, 但我们通过自定义类加载器只能获取到一个Method类(该类没有getCode方法，所以无法获取到字节码)。那我们只能再往前查找。
截屏2023-12-23 13.17.08.png

通过定位发现，其实最终的实现是通过graal/compiler/src/jdk.graal.compiler/src/jdk/graal/compiler/hotspot/HotSpotSnippetMetaAccessProvider.java中的lookupJavaMethod实现的：
截屏2023-12-23 13.25.54.png
然后再往前定位delegate，发现它原来还是通过jvmci接口获取的：
graal/compiler/src/jdk.graal.compiler/src/jdk/graal/compiler/hotspot/HotSpotGraalRuntime.java

所以根据自定义类加载器和jvmci，可以写一个小demo实现graal中 (从类-> 方法 -> 字节码) 的小demo。

最终代码

package org.example;


import jdk.vm.ci.hotspot.HotSpotJVMCIRuntime;
import jdk.vm.ci.meta.ResolvedJavaMethod;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.lang.reflect.Method;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.WritableByteChannel;

public class Main {
    public static void main(String[] args) throws ClassNotFoundException {
        MyClassLoader fileReadClassLoader = new MyClassLoader("/Users/xxx/tmp/grall-shell");
        Class<?> account = fileReadClassLoader.loadClass("HelloWorld");
        Method[] methods = account.getMethods();
        for (int i = 0; i < methods.length; i++) {
            Method method = methods[i];
            if (method.getName().equals("main")) {
                ResolvedJavaMethod lookupJavaMethod = HotSpotJVMCIRuntime.runtime().getHostJVMCIBackend().getMetaAccess().lookupJavaMethod(method);
                if (lookupJavaMethod.getCodeSize() > 1) {
                    byte[] code = lookupJavaMethod.getCode();
                    System.out.println(lookupJavaMethod.hasBytecodes());
                    for (int j = 0; j < code.length; j++) {
                        System.out.println(code[j]);
                    }
                }
            }
        }
    }

    // 自定义类加载器
    static class MyClassLoader extends ClassLoader {
        private String dir;

        public MyClassLoader(String dir) {
            this.dir = dir;
        }

        @Override
        protected Class<?> findClass(String name) throws ClassNotFoundException {
            try {
                // 读取class
                byte[] bytes = getClassBytes(name);
                // 将二进制class转换为class对象
                Class<?> c = this.defineClass(null, bytes, 0, bytes.length);
                return c;
            } catch (Exception e) {
                e.printStackTrace();
            }
            return super.findClass(name);
        }

        private byte[] getClassBytes(String name) throws Exception {
            FileInputStream fis = new FileInputStream(new File(this.dir + File.separator + name + ".class"));
            FileChannel fc = fis.getChannel();
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            WritableByteChannel wbc = Channels.newChannel(baos);
            ByteBuffer by = ByteBuffer.allocate(1024);
            while (true) {
                int i = fc.read(by);
                if (i == 0 || i == -1)
                    break;
                by.flip();
                wbc.write(by);
                by.clear();
            }
            fis.close();
            return baos.toByteArray();
        }
    }
}

执行上面的最终代码

java --add-modules jdk.internal.vm.ci \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.runtime=ALL-UNNAMED \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.code=ALL-UNNAMED \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.hotspot=ALL-UNNAMED \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.services=ALL-UNNAMED \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.meta=ALL-UNNAMED \
    --add-exports jdk.internal.vm.ci/jdk.vm.ci.code.site=ALL-UNNAMED \
    -XX:+UnlockExperimentalVMOptions \
    -XX:+EnableJVMCI \
    Main.java