创建一个ClassLoader以从字节数组加载JAR文件

Z时代
2024-01-10
分类：问答

我正在寻找编写一个自定义类加载器，它将JAR通过自定义网络加载文件。最后，我要做的只是JAR文件的字节数组。

我无法将字节数组转储到文件系统上并使用URLClassLoader。

我的第一个计划是JarFile从流或字节数组创建对象，但它仅支持File对象。

我已经写了一些使用的东西JarInputStream：

public class RemoteClassLoader extends ClassLoader {
    private final byte[] jarBytes;
    public RemoteClassLoader(byte[] jarBytes) {
        this.jarBytes = jarBytes;
    }
    @Override
    public Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {
        Class<?> clazz = findLoadedClass(name);
        if (clazz == null) {
            try {
                InputStream in = getResourceAsStream(name.replace('.', '/') + ".class");
                ByteArrayOutputStream out = new ByteArrayOutputStream();
                StreamUtils.writeTo(in, out);
                byte[] bytes = out.toByteArray();
                clazz = defineClass(name, bytes, 0, bytes.length);
                if (resolve) {
                    resolveClass(clazz);
                }
            } catch (Exception e) {
                clazz = super.loadClass(name, resolve);
            }
        }
        return clazz;
    }
    @Override
    public URL getResource(String name) {
        return null;
    }
    @Override
    public InputStream getResourceAsStream(String name) {
        try (JarInputStream jis = new JarInputStream(new ByteArrayInputStream(jarBytes))) {
            JarEntry entry;
            while ((entry = jis.getNextJarEntry()) != null) {
                if (entry.getName().equals(name)) {
                    return jis;
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
}