java - 使用自定义 ClassLoader 拦截 ClassLoader.getResource(String) 调用

Question

我们正在尝试调试 WebStart 的一个不可重现的问题，即访问 Jars 内的资源将“随机”失败。也许每 1000 个应用程序运行中就有一个会出现此错误，这种错误可能发生在从 jar 读取资源的任何地方。

在 Google 和 Java Bug 数据库中搜索没有找到类似的结果(或者至少没有任何帮助)。

我们正在尝试通过“检测”应用程序来获取有关客户端上发生的情况的更多信息，因此我们跟踪对 ClassLoader.getResource(String) 的所有调用(包括间接超过 ClassLoader.getResourceAsStream(String) )。在不更改应用程序代码的情况下，我创建了一个“启动器”，它将使用自定义类加载器运行整个应用程序。

不幸的是，我的类加载器似乎被绕过了。我没有看到任何预期的 System.out 输出。这是我尝试过的:

private static final class MyClassLoader extends ClassLoader {
    private MyClassLoader() {
        super(TheClassThatMainIsIn.class.getClassLoader());
    }

    @Override
    public URL getResource(String name) {
        System.out.println("getResource("+name+")");
        // Snip
        return super.getResource(name);
    }

    @Override
    public InputStream getResourceAsStream(String name) {
        System.out.println("getResourceAsStream("+name+")");
        final URL url = getResource(name);
        try {
            return url != null ? url.openStream() : null;
        } catch (final IOException e) {
            return null;
        }
    }
}

public static void main(String[] args) {
    System.out.println("Starting MyRealApp Launcher ...");
    final MyClassLoader loader = new MyClassLoader();
    try {
        Class<?> realAppClasss = loader.loadClass("MyRealAppClass");
        Method main = realAppClasss.getMethod("main", String[].class);
        main.invoke(null, (Object) args);
    } catch (final RuntimeException e) {
        throw e;
    } catch (final Error e) {
        throw e;
    } catch (final InvocationTargetException e) {
        final Throwable cause = e.getCause();
        if (cause instanceof RuntimeException) {
            throw (RuntimeException) cause;
        }
        if (cause instanceof Error) {
            throw (Error) cause;
        }
        throw new UndeclaredThrowableException(cause);
    } catch (final Throwable t) {
        throw new UndeclaredThrowableException(t);
    }
}

我在这里做错了什么？

Answer 1

是的。原则上这是可行的。

但是，您必须考虑资源加载代码如何到达类加载器。由于该类没有出现，看起来他们使用了父类加载器。您必须考虑不同的场景:

使用上下文类加载器的代码，例如:

Thread.currentThread().getContextClassLoader().getResource("via-context");

这很容易实现，通过在调用 main 之前设置它:

Thread.currentThread().setContextClassLoader(loader);
Method main = realAppClasss.getMethod("main", String[].class);
main.invoke(null, (Object) args);

接下来您需要考虑的是从当前类中“获取”类加载器并加载它的代码。当您的类通过父类加载器加载时，它也会使用该类加载器来获取资源。喜欢:

MyRealAppClass.class.getResource("via-class");
MyRealAppClass.class.getClassLoader().getResource("via-class");
objectInfApp.getClass().getClassLoader().getResource("via-class");

为了避免这种情况，您必须确保应用程序类实际上是使用您的类加载器而不是父类加载器加载的。对于简单的 main，您可以从 URL 类加载器扩展，跳过任何父级并使用 URL 的原始类路径。喜欢:

// URL class loader to lookup in jars etc
private static class MyClassLoader extends URLClassLoader
{
    public MyClassLoader(URL[] urls) {
        // Use the given URLs and skip any parent class loader, directly go to the system loader
        super(urls,null);
    }

// ...

// Then setup the class path
    String[] classPath = System.getProperty("java.class.path").split(";");
    URL[] classPathUrls = new URL[classPath.length];
    for (int i = 0; i < classPath.length; i++) {
        classPathUrls[i] = new File(classPath[i]).toURL();

    }
    MyClassLoader loader = new MyClassLoader(classPathUrls);

这应该涵盖最基本的情况。当您的实际应用程序本身有更多的类加载器技巧时，您可能需要设置更多。