github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2025-03-23 16:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix a classloader filter issue for java 9+

Browse Source
This commit is contained in:
金戟 2020-12-10 19:21:33 +08:00
parent dfb1e2c094
commit 260fe94a49
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
View File

@ -19,7 +19,6 @@ import org.objectweb.asm.tree.MethodNode;
import java.io.IOException; import java.io.IOException;
import java.lang.instrument.ClassFileTransformer; import java.lang.instrument.ClassFileTransformer;
import java.net.URLClassLoader;
import java.security.ProtectionDomain; import java.security.ProtectionDomain;
import java.util.*; import java.util.*;
@ -34,13 +33,22 @@ public class TestableClassTransformer implements ClassFileTransformer {
private final Map<ComparableWeakRef<String>, CachedMockParameter> loadedClass = private final Map<ComparableWeakRef<String>, CachedMockParameter> loadedClass =
new WeakHashMap<ComparableWeakRef<String>, CachedMockParameter>(); new WeakHashMap<ComparableWeakRef<String>, CachedMockParameter>();
/**
* Just avoid spend time to scan those surely non-user classes
* Should keep this list as tiny as possible
*/
private final String[] SYSTEM_PREFIXES = new String[] { "jdk/", "java/", "javax/", "com/sun/",
"org/apache/maven/", "com/alibaba/testable/", "junit/", "org/junit/", "org/testng/" };
@Override @Override
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,
ProtectionDomain protectionDomain, byte[] classFileBuffer) { ProtectionDomain protectionDomain, byte[] classFileBuffer) {
if (isSystemClass(loader, className) || loadedClass.containsKey(new ComparableWeakRef<String>(className))) { if (isSystemClass(className) || loadedClass.containsKey(new ComparableWeakRef<String>(className))) {
// Ignore system class and reloaded class // Ignore system class and reloaded class
LogUtil.verbose("Ignore class: " + (className == null ? "<lambda>" : className));
return null; return null;
} }
LogUtil.verbose("Handle class: " + className);
byte[] bytes = null; byte[] bytes = null;
try { try {
if (shouldTransformAsSourceClass(className)) { if (shouldTransformAsSourceClass(className)) {
@ -69,9 +77,17 @@ public class TestableClassTransformer implements ClassFileTransformer {
return className.endsWith(ConstPool.TEST_POSTFIX) && hasMockAnnotation(className); return className.endsWith(ConstPool.TEST_POSTFIX) && hasMockAnnotation(className);
} }
private boolean isSystemClass(ClassLoader loader, String className) { private boolean isSystemClass(String className) {
// className can be null for Java 8 lambdas // className can be null for Java 8 lambdas
return !(loader instanceof URLClassLoader) || null == className || className.startsWith("jdk/"); if (null == className) {
return true;
}
for (String prefix : SYSTEM_PREFIXES) {
if (className.startsWith(prefix)) {
return true;
}
}
return false;
} }
private List<MethodInfo> getTestableMockMethods(String className) { private List<MethodInfo> getTestableMockMethods(String className) {