1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
package ftbsc.lll.loader;
import cpw.mods.modlauncher.serviceapi.ILaunchPluginService;
import ftbsc.lll.IInjector;
import ftbsc.lll.exception.InjectionException;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.MethodNode;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Map;
import java.util.ServiceLoader;
import java.util.stream.Collectors;
public class LilleroLoader implements ILaunchPluginService {
public static final Logger LOGGER = LogManager.getLogger(LilleroLoader.class.getCanonicalName());
public static final String NAME = "lll-loader";
private List<IInjector> injectors = new ArrayList<>();
public LilleroLoader() {
LOGGER.info("INIT", "Patch Loader initialized");
}
@Override
public String name() {
return NAME;
}
// Load mods requesting patches from resources
@Override
public void offerResource(Path resource, String name) {
LOGGER.error("RESOURCE", String.format("Resource offered to us (%s@%s) but no action was taken", name, resource.toString()));
}
@Override
public void addResources(List<Map.Entry<String, Path>> resources) {
LOGGER.debug("RESOURCE", "Resources being added:");
for (Map.Entry<String, Path> row : resources) {
LOGGER.debug("RESOURCE", String.format("> %s (%s)", row.getKey(), row.getValue().toString()));
try {
URL jarUrl = new URL("file:" + row.getValue().toString());
URLClassLoader loader = new URLClassLoader(new URL[] { jarUrl });
for (IInjector inj : ServiceLoader.load(IInjector.class, loader)) {
LOGGER.info("RESOURCE", String.format("Registering injector %s", inj.name()));
this.injectors.add(inj);
}
} catch (MalformedURLException e) {
LOGGER.error("RESOURCE", String.format("Malformed URL for resource %s - 'file:%s'", row.getKey(), row.getValue().toString()));
}
}
}
// Filter only classes we need to patch
@Override
public EnumSet<Phase> handlesClass(Type classType, final boolean isEmpty) {
throw new IllegalStateException("Outdated ModLauncher"); //mixin does it
}
private static final EnumSet<Phase> YAY = EnumSet.of(Phase.BEFORE);
private static final EnumSet<Phase> NAY = EnumSet.noneOf(Phase.class);
@Override
public EnumSet<Phase> handlesClass(Type classType, final boolean isEmpty, final String reason) {
if (isEmpty) return NAY;
// TODO can I make a set of target classes to make this faster
for (IInjector inj : this.injectors) {
if (inj.targetClass().equals(classType.getClassName())) {
LOGGER.debug("HANDLER", String.format("Marked class %s as handled by us", classType.getClassName()));
return YAY;
}
}
return NAY;
}
// Process classes and inject methods
@Override
public int processClassWithFlags(Phase phase, ClassNode classNode, Type classType, String reason) {
LOGGER.debug("PATCHER", "Processing class {} in phase {} of {}", classType.getClassName(), phase.name(), reason);
List<IInjector> relevantInjectors = this.injectors.stream()
.filter(i -> i.targetClass().equals(classType.getClassName()))
.collect(Collectors.toList());
boolean modified = false;
for (MethodNode method : classNode.methods) {
for (IInjector inj : relevantInjectors) {
if (
inj.methodName().equals(method.name) &&
inj.methodDesc().equals(method.desc)
) {
LOGGER.info("PATCHER", String.format("Patching %s.%s with %s (%s)", classType.getClassName(), method.name, inj.name(), inj.reason()));
try {
inj.inject(classNode, method);
modified = true;
} catch (InjectionException e) {
LOGGER.error("PATCHER", String.format("Error applying patch '%s' : %s", inj.name(), e.toString()));
}
}
}
}
return modified ? ComputeFlags.COMPUTE_FRAMES : ComputeFlags.NO_REWRITE;
}
}
|
zaaarf