mirror of
https://github.com/xfl03/MCCustomSkinLoader.git
synced 2026-11-12 20:07:37 +00:00
96 lines
4.0 KiB
Java
96 lines
4.0 KiB
Java
package customskinloader.forge;
|
|
|
|
import java.lang.annotation.ElementType;
|
|
import java.lang.annotation.Retention;
|
|
import java.lang.annotation.RetentionPolicy;
|
|
import java.lang.annotation.Target;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import org.objectweb.asm.ClassReader;
|
|
import org.objectweb.asm.ClassWriter;
|
|
import org.objectweb.asm.tree.ClassNode;
|
|
import org.objectweb.asm.tree.MethodNode;
|
|
|
|
import net.minecraft.launchwrapper.IClassTransformer;
|
|
import net.minecraftforge.fml.common.asm.transformers.deobf.FMLDeobfuscatingRemapper;
|
|
import net.minecraftforge.fml.relauncher.FMLRelaunchLog;
|
|
import customskinloader.forge.PlayerTabTransformer.ScoreObjectiveTransformer;
|
|
import customskinloader.forge.SkinManagerTransformer.*;
|
|
|
|
public class TransformerManager implements IClassTransformer {
|
|
@Retention(RetentionPolicy.RUNTIME)
|
|
@Target(ElementType.TYPE)
|
|
public @interface TransformTarget{
|
|
public String className();
|
|
public String[] methodNames();
|
|
public String desc();
|
|
}
|
|
public interface IMethodTransformer{
|
|
public void transform(ClassNode cn,MethodNode mn);
|
|
}
|
|
private static final IMethodTransformer[] TRANFORMERS={
|
|
new InitTransformer(),
|
|
new LoadSkinTransformer(),
|
|
new LoadProfileTexturesTransformer(),
|
|
new LoadSkinFromCacheTransformer(),
|
|
new ScoreObjectiveTransformer()};
|
|
private Map<String, Map<String, IMethodTransformer>> map;
|
|
public TransformerManager(){
|
|
map = new HashMap<String, Map<String, IMethodTransformer>>();
|
|
for(IMethodTransformer t:TRANFORMERS){
|
|
FMLRelaunchLog.info("[CSL DEBUG] REGISTERING TRANSFORMER %s",t.getClass().getName());
|
|
if(!t.getClass().isAnnotationPresent(TransformTarget.class)){
|
|
FMLRelaunchLog.info("[CSL DEBUG] ERROR occurs while parsing Annotation.");
|
|
continue;
|
|
}
|
|
addMethodTransformer(t.getClass().getAnnotation(TransformTarget.class),t);
|
|
}
|
|
}
|
|
private void addMethodTransformer(TransformTarget target, IMethodTransformer transformer) {
|
|
if (!map.containsKey(target.className()))
|
|
map.put(target.className(), new HashMap<String, IMethodTransformer>());
|
|
for(String methodName:target.methodNames()){
|
|
map.get(target.className()).put(methodName + target.desc(), transformer);
|
|
FMLRelaunchLog.info("[CSL DEBUG] REGISTERING METHOD %s(%s)",target.className(),methodName + target.desc());
|
|
}
|
|
}
|
|
|
|
//From: https://github.com/RecursiveG/UniSkinMod/blob/1.9.4/src/main/java/org/devinprogress/uniskinmod/coremod/BaseAsmTransformer.java#L83-L114
|
|
@Override
|
|
public byte[] transform(String obfClassName, String className, byte[] bytes) {
|
|
if (!map.containsKey(className)) return bytes;
|
|
FMLRelaunchLog.info("[CSL DEBUG] CLASS %s will be transformed", className);
|
|
Map<String, IMethodTransformer> transMap = map.get(className);
|
|
|
|
ClassReader cr = new ClassReader(bytes);
|
|
ClassNode cn = new ClassNode();
|
|
cr.accept(cn, 0);
|
|
|
|
// NOTE: `map` = convert obfuscated name to srgName;
|
|
List<MethodNode> ml = new ArrayList<MethodNode>();
|
|
ml.addAll(cn.methods);
|
|
for (MethodNode mn : ml) {
|
|
String methodName = FMLDeobfuscatingRemapper.INSTANCE.mapMethodName(obfClassName, mn.name, mn.desc);
|
|
String methodDesc = FMLDeobfuscatingRemapper.INSTANCE.mapMethodDesc(mn.desc);
|
|
if (transMap.containsKey(methodName + methodDesc)) {
|
|
try {
|
|
FMLRelaunchLog.info("[CSL DEBUG] Transforming method %s in class %s(%s)", methodName + methodDesc, obfClassName, className);
|
|
transMap.get(methodName + methodDesc).transform(cn,mn);
|
|
FMLRelaunchLog.info("[CSL DEBUG] Successfully transformed method %s in class %s(%s)", methodName + methodDesc, obfClassName, className);
|
|
} catch (Exception e) {
|
|
FMLRelaunchLog.warning("[CSL DEBUG] An error happened when transforming method %s in class %s(%s). The whole class was not modified.", methodName + methodDesc, obfClassName, className);
|
|
e.printStackTrace();
|
|
return bytes;
|
|
}
|
|
}
|
|
}
|
|
|
|
ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
|
|
cn.accept(cw);
|
|
return cw.toByteArray();
|
|
}
|
|
}
|