funkemunky/KauriV3-mirror
1
0
Fork 0
mirror of https://github.com/funkemunky/KauriV3.git synced 2026-06-02 22:42:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improvements

Browse Source
This commit is contained in:
Dawson
2022-08-30 09:37:50 -04:00
parent ec18c17720
commit abb2b8be2a
3 changed files with 57 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/dev/brighten/ac/utils/ClassScanner.java
+44 -7
View File
@@ -4,6 +4,7 @@ import dev.brighten.ac.Anticheat;
import dev.brighten.ac.utils.annotation.Init;
import dev.brighten.ac.utils.reflections.Reflections;
import dev.brighten.ac.utils.reflections.types.WrappedClass;
import jdk.tools.jlink.resources.plugins;
import org.bukkit.Bukkit;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.tree.AnnotationNode;
@@ -30,19 +31,31 @@ public class ClassScanner {
//TODO Get check classes too
public static Set<WrappedClass> getClasses(Class<? extends Annotation> annotationClass,
String packageName) {
return getNames().stream().filter(pkg -> pkg.startsWith(packageName) && findClass(map.get(n), Init.class) != null)
Map<String, byte[]> map = (Map<String, byte[]>) Anticheat.INSTANCE.getStuffs();
return getNames().stream().filter(pkg -> pkg.startsWith(packageName) && findClass(map.get(pkg), annotationClass) != null)
.map(Reflections::getClass).collect(Collectors.toSet());
}
public static Set<String> getNames() {
Map<String, byte[]> map = (Map<String, byte[]>) Anticheat.INSTANCE.getStuffs();
return map.keySet().stream().filter(n -> {
return !n.endsWith(".yml") && !n.endsWith(".xml") && !n.endsWith(".") && !n.endsWith(".properties")
&& !n.contains("dev.brighten.ac.packet")
&& Character.isLetterOrDigit(n.charAt(n.length() - 1))
&& findClass(map.get(n), Init.class) != null;
}).collect(Collectors.toSet());
Set<String> nameSet = new HashSet<>();
for (String loadedClass : Anticheat.INSTANCE.getLoadedClasses()) {
InputStream stream = Anticheat.INSTANCE.getClassLoader2().getResourceAsStream(loadedClass);
if(findClass(stream, Init.class) != null) {
nameSet.add(loadedClass);
}
}
map.keySet().stream().filter(n -> !n.endsWith(".yml")
&& !n.endsWith(".xml") && !n.endsWith(".") && !n.endsWith(".properties")
&& !n.contains("dev.brighten.ac.packet")
&& Character.isLetterOrDigit(n.charAt(n.length() - 1))
&& findClass(map.get(n), Init.class) != null).forEach(nameSet::add);
return nameSet;
}
public static String findClass(byte[] array, Class<? extends Annotation> annotationClass) {
@@ -69,6 +82,30 @@ public class ClassScanner {
return null;
}
public static String findClass(InputStream stream, Class<? extends Annotation> annotationClass) {
try {
ClassReader reader = new ClassReader(stream);
ClassNode classNode = new ClassNode();
reader.accept(classNode, ClassReader.SKIP_CODE | ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES);
String className = classNode.name.replace('/', '.');
final String anName = annotationClass.getName().replace(".", "/");
if (classNode.visibleAnnotations != null) {
for (Object node : classNode.visibleAnnotations) {
AnnotationNode annotation = (AnnotationNode) node;
if (annotation.desc
.equals("L" + anName + ";")) {
return className;
}
}
}
} catch (Exception e) {
//Bukkit.getLogger().info("Failed to scan");
//e.printStackTrace();
}
return null;
}
private static <E> HashSet<E> newHashSet(E... elements) {
HashSet<E> set = new HashSet<>();
Collections.addAll(set, elements);