SimpleClassMatcher support read classname from bytes.

alibaba · Nov 24, 2020 · 6c8bb25 · 6c8bb25
1 parent 641360a
commit 6c8bb25
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/bytekit-core/src/main/java/com/alibaba/bytekit/asm/matcher/SimpleClassMatcher.java b/bytekit-core/src/main/java/com/alibaba/bytekit/asm/matcher/SimpleClassMatcher.java
@@ -5,6 +5,8 @@
 import java.util.HashSet;
 import java.util.Set;
 
+import com.alibaba.bytekit.utils.AsmUtils;
+
 /**
  * 
  * @author hengyunabc 2020-11-12
@@ -28,7 +30,7 @@ public SimpleClassMatcher(Collection<String> names) {
     public boolean match(ClassLoader loader, String className, Class<?> classBeingRedefined,
             ProtectionDomain protectionDomain, byte[] classfileBuffer) {
         if (className == null) {
-            return false;
+            className = AsmUtils.className(classfileBuffer);
         }
         className = className.replace('/', '.');
         if (classNames != null && classNames.contains(className)) {

diff --git a/bytekit-core/src/main/java/com/alibaba/bytekit/utils/AsmUtils.java b/bytekit-core/src/main/java/com/alibaba/bytekit/utils/AsmUtils.java
@@ -586,4 +586,8 @@ public static int setMajorVersion(int version, int majorVersion) {
         return (version & 0xFFFF0000) | majorVersion;
     }
 
+    public static String className(byte[] classBytes) {
+        return new ClassReader(classBytes).getClassName();
+    }
+
 }