summaryrefslogtreecommitdiff
path: root/src/main/java/ftbsc
diff options
context:
space:
mode:
author dev@ftbsc <dev@fantabos.co>2023-02-07 01:42:13 +0100
committer dev@ftbsc <dev@fantabos.co>2023-02-07 01:42:13 +0100
commitbe0f5c58ea4bec1db7f16fbfa65eb644b236bcbc (patch)
tree1ac6c3cabe1fe556046e670cd994de919a99fb14 /src/main/java/ftbsc
parent7e0923f8151e6a64e6793dfe8d6f57d1d477e857 (diff)
build: added loader and service into main lib
Diffstat (limited to 'src/main/java/ftbsc')
-rw-r--r--src/main/java/ftbsc/lll/loader/LilleroLoader.java113
1 files changed, 113 insertions, 0 deletions
diff --git a/src/main/java/ftbsc/lll/loader/LilleroLoader.java b/src/main/java/ftbsc/lll/loader/LilleroLoader.java
new file mode 100644
index 0000000..d7b0453
--- /dev/null
+++ b/src/main/java/ftbsc/lll/loader/LilleroLoader.java
@@ -0,0 +1,113 @@
+package ftbsc.lll.loader;
+
+import cpw.mods.modlauncher.serviceapi.ILaunchPluginService;
+
+import ftbsc.lll.IInjector;
+
+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");
+ public static final String NAME = "lillero-loader";
+
+ private List<IInjector> injectors = new ArrayList<>();
+
+ public LilleroLoader() {
+ LOGGER.info("Lillero Patch Loader initialized");
+ }
+
+ @Override
+ public String name() {
+ return NAME;
+ }
+
+
+ // Load mods requesting patches from resources
+
+ @Override
+ public void offerResource(Path resource, String name) {
+ LOGGER.warn(String.format("Resource offered to us: %s @ '%s'", name, resource.toString()));
+ }
+
+ @Override
+ public void addResources(List<Map.Entry<String, Path>> resources) {
+ LOGGER.info("Resources being added:");
+ for (Map.Entry<String, Path> row : resources) {
+ LOGGER.info(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(String.format("Registering injector %s", inj.name()));
+ this.injectors.add(inj);
+ }
+ } catch (MalformedURLException e) {
+ LOGGER.error(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()))
+ return YAY;
+ }
+ return NAY;
+ }
+
+
+ // Process classes and inject methods
+
+ @Override
+ public int processClassWithFlags(Phase phase, ClassNode classNode, Type classType, String reason) {
+ LOGGER.debug("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(String.format("Patching %s.%s with %s", classType.getClassName(), method.name, inj.name()));
+ inj.inject(classNode, method); // TODO catch patching exceptions
+ modified = true;
+ }
+ }
+ }
+
+ return modified ? ComputeFlags.COMPUTE_FRAMES : ComputeFlags.NO_REWRITE;
+ }
+}