AstralTransRocketries/authlib-injector
1
0
Fork 0
mirror of https://github.com/unmojang/authlib-injector.git synced 2025-10-02 15:51:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add ClassLoadingListener interface

Browse Source
This commit is contained in:
yushijinhun 2018-11-24 14:32:22 +08:00
parent 5010ea67b7
commit 1f5605c973
No known key found for this signature in database
GPG Key ID: 5BC167F73EA558E4
2 changed files with 39 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/moe/yushi/authlibinjector/transform/ClassLoadingListener.java Normal file
View File

@ -0,0 +1,9 @@
package moe.yushi.authlibinjector.transform;
import java.util.List;
public interface ClassLoadingListener {
void onClassLoading(ClassLoader classLoader, String className, byte[] bytecode, List<TransformUnit> appliedTransformers);
}

40
src/main/java/moe/yushi/authlibinjector/transform/ClassTransformer.java
View File

@ -1,5 +1,7 @@
package moe.yushi.authlibinjector.transform; package moe.yushi.authlibinjector.transform;
import static java.util.Collections.emptyList;
import java.io.IOException; import java.io.IOException;
import java.lang.instrument.ClassFileTransformer; import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException; import java.lang.instrument.IllegalClassFormatException;
@ -9,6 +11,7 @@ import java.nio.file.Paths;
import java.nio.file.StandardOpenOption; import java.nio.file.StandardOpenOption;
import java.security.ProtectionDomain; import java.security.ProtectionDomain;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet; import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Optional; import java.util.Optional;
@ -22,12 +25,13 @@ import moe.yushi.authlibinjector.util.Logging;
public class ClassTransformer implements ClassFileTransformer { public class ClassTransformer implements ClassFileTransformer {
public List<TransformUnit> units = new ArrayList<>(); public List<TransformUnit> units = new ArrayList<>();
public List<ClassLoadingListener> listeners = new ArrayList<>();
public Set<String> ignores = new HashSet<>(); public Set<String> ignores = new HashSet<>();
public boolean debugSaveClass; public boolean debugSaveClass;
private static class TransformHandle { private static class TransformHandle {
private boolean modified = false; private List<TransformUnit> appliedTransformers;
private boolean currentModified; private boolean currentModified;
private String className; private String className;
private byte[] classBuffer; private byte[] classBuffer;
@ -54,7 +58,10 @@ public class ClassTransformer implements ClassFileTransformer {
reader.accept(optionalVisitor.get(), 0); reader.accept(optionalVisitor.get(), 0);
if (currentModified) { if (currentModified) {
Logging.TRANSFORM.info("Transformed [" + className + "] with [" + unit + "]"); Logging.TRANSFORM.info("Transformed [" + className + "] with [" + unit + "]");
modified = true; if (appliedTransformers == null) {
appliedTransformers = new ArrayList<>();
}
appliedTransformers.add(unit);
classBuffer = writer.toByteArray(); classBuffer = writer.toByteArray();
} }
} else { } else {
@ -62,14 +69,21 @@ public class ClassTransformer implements ClassFileTransformer {
} }
} }
public Optional<byte[]> getResult() { public Optional<byte[]> getTransformResult() {
if (modified) { if (appliedTransformers == null || appliedTransformers.isEmpty()) {
return Optional.of(classBuffer);
} else {
return Optional.empty(); return Optional.empty();
} else {
return Optional.of(classBuffer);
} }
} }
public List<TransformUnit> getAppliedTransformers() {
return appliedTransformers == null ? emptyList() : appliedTransformers;
}
public byte[] getFinalResult() {
return classBuffer;
}
} }
@Override @Override
@ -77,13 +91,19 @@ public class ClassTransformer implements ClassFileTransformer {
if (internalClassName != null && classfileBuffer != null) { if (internalClassName != null && classfileBuffer != null) {
try { try {
String className = internalClassName.replace('/', '.'); String className = internalClassName.replace('/', '.');
for (String prefix : ignores) for (String prefix : ignores) {
if (className.startsWith(prefix)) return null; if (className.startsWith(prefix)) {
listeners.forEach(it -> it.onClassLoading(loader, className, classfileBuffer, Collections.emptyList()));
return null;
}
}
TransformHandle handle = new TransformHandle(className, classfileBuffer); TransformHandle handle = new TransformHandle(className, classfileBuffer);
units.forEach(handle::accept); units.forEach(handle::accept);
if (handle.getResult().isPresent()) { listeners.forEach(it -> it.onClassLoading(loader, className, handle.getFinalResult(), handle.getAppliedTransformers()));
byte[] classBuffer = handle.getResult().get();
if (handle.getTransformResult().isPresent()) {
byte[] classBuffer = handle.getTransformResult().get();
if (debugSaveClass) { if (debugSaveClass) {
saveClassFile(className, classBuffer); saveClassFile(className, classBuffer);
} }