如何检测自定义类加载器加载的类？

Question

我试图修改几个类的字节代码，这些类的打包jar文件不在类路径中-它们是在给定URL的运行时由自定义的classloader加载的。我尝试使用Java代理和ClassFileTransformer来拦截那些类，但失败了。类加载器是遗留项目的一部分，因此我无法直接对其进行更改。

代理可以很好地处理AppClassLoader“本地”加载的类，但只会忽略那些由自定义类加载器加载的类。

CustomClassLoader：

    public class CustomClassLoader extends URLClassLoader {

    public CustomClassLoader(URL[] urls) {
        super(urls, CustomClassLoader.class.getClassLoader());
    }

    // violates parent-delegation pattern
    @Override
    protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {
        synchronized (getClassLoadingLock(name)) {
            Class<?> clazz = findLoadedClass(name);
            if (clazz == null) {
                try {
                    clazz = findClass(name);
                } catch (ClassNotFoundException e) {
                }

                if (clazz == null) {
                    clazz = getParent().loadClass(name);
                }
            }
            if (resolve) {
                resolveClass(clazz);
            }
            return clazz;
        }
    }
}

我的代理中使用的ClassFileTransformer（带有javassist）：

public class MyTransformer implements ClassFileTransformer
{
    public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,
            ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException
    {
        byte[] byteCode = null;

        if (className.replace("/", ".").equals("com.example.services.TargetService"))
        {
            ClassPool cp = ClassPool.getDefault();
            CtClass cc;
            try
            {
                cc = cp.get("com.example.services.TargetService");
                CtMethod verifyMethod = cc.getDeclaredMethod("verify");
                //invalidate the verification process of method : verify
                verifyMethod.insertBefore("{return true;}");
                byteCode = cc.toBytecode();
                cc.detach();
                return byteCode;
            }
            catch (Exception e)
            {
                e.printStackTrace();
            }
        }

        return byteCode;
    }
}

代理：

public class Agent
{
    public static void premain(String agentArgs, Instrumentation inst)
    {
        inst.addTransformer(new MyTransformer());
    }

}

我通过检测CustomClassLoader本身，调用instrumentation.RedifineClasses()，找到了一个解决方法，但不知道如何将检测实例传递到CustomClassLoader实例中；我是一个新的仪器/类加载，仍然不是很清楚他们的机制。有人帮忙吗？多谢了。

为了简单起见：

1. 在应用程序内部创建一个自定义URLClassLoader，它在运行时将一些jar文件加载到文件系统的其他地方。
2. 实现一个java代理，转换由类加载器加载的类，替换它的方法主体或其他内容。
3. 在应用程序内部，将类的实例分配给它的接口，并调用它的检测方法。
4. 使用-javaagent运行应用程序进行检查。

Answer 1

我假设您的类没有正确检测，因为您正在调用ClassPool.getDefault()，它不包括对自定义类加载器可见的类文件，而只对系统类加载器可见。您永远不会注册classfilebuffer类文件。

作为另一种选择,您可以试用Byte Buddy,它提供了对检测API的更容易的访问：

new AgentBuilder.Default()
  .type(named("com.example.services.TargetService"))
  .transform((builder, type, loader) -> {
    builder.method(named("verify")).intercept(FixedValue.of(true));
  }).installOn(instrumentation);

可以从agentmain或premain方法调用上述代理。您还可以禁用类文件格式更改，并重新定义现有类，以防类已经（可能）在附件期间加载。