1

我正在尝试为拦截来自 java.io 的类的调用（例如 FileInputStream）的代理编写 UnitTest。

我已经按照此处给出的提示查看了 ByteBuddys自己的UnitTests。

private ClassLoader classLoader;

@Before
@AgentAttachmentRule.Enforce
public void setUp() throws Exception
{
    classLoader = new ByteArrayClassLoader.ChildFirst(getClass().getClassLoader(),
                                           ClassFileExtraction.of(FileInputStream.class),
                                           DEFAULT_PROTECTION_DOMAIN,
                                           AccessController.getContext(),
                                           ByteArrayClassLoader.PersistenceHandler.MANIFEST,
                                           PackageDefinitionStrategy.NoOp.INSTANCE);
}

@Test
public void testAgentForFileInputStream() throws Exception
{
    MyAgent.premain("");
    Class<?> type = classLoader.loadClass(FileInputStream.class.getName());
    type.getDeclaredMethod("open").invoke("test");
}

对非 java.* 类的类似测试工作正常，但是在这里我收到：

java.lang.SecurityException: Prohibited package name: java.io

at java.lang.ClassLoader.preDefineClass(ClassLoader.java:659)
at java.lang.ClassLoader.defineClass(ClassLoader.java:758)
at net.bytebuddy.dynamic.loading.ByteArrayClassLoader.findClass(ByteArrayClassLoader.java:197)
at net.bytebuddy.dynamic.loading.ByteArrayClassLoader$ChildFirst.loadClass(ByteArrayClassLoader.java:554)
at java.lang.ClassLoader.loadClass(ClassLoader.java:357)

我尝试将 SecurityManager 设置为 null，但没有成功：

System.setSecurityManager(null);

如果我没记错的话，应该可以拦截这样的类。这可能是由于使用了自定义 ClassLoader。但是我不知道需要改变什么......

4

1 回答 1

2

java.除了引导类加载器之外，不能从任何类加载器加载具有前缀的任何类。此限制不是由 a 强制执行的，SecurityManager而是硬编码到ClassLoader实现中的。

测试这些类的唯一方法是：

  1. 在测试过程中调整名称以允许在另一个类加载器中加载。
  2. 将类加载到引导类加载器中（使用ByteBuddyAgentand ClassInjector.ForInstrumentation）。您仍然应该随机化名称以保证测试的可重复性。

在您的情况下，只有选项 2 似乎可行。由于我不知道您的代理在做什么，我只能假设您尝试操作FileInputStream在任何情况下都需要使用引导加载程序注入的类，因为此类的复制不适用于该类的本机方法适配器.

于 2016-07-07T07:48:38.387 回答