Remove me.nallar. classes from invalidClasses set after adding to LCL

MinimallyCorrect · Feb 7, 2016 · f4e382a · f4e382a
1 parent de36529
commit f4e382a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/me/nallar/modpatcher/ModPatcher.java b/src/main/java/me/nallar/modpatcher/ModPatcher.java
@@ -4,6 +4,7 @@
 import com.google.common.io.Resources;
 import me.nallar.javapatcher.patcher.Patcher;
 import net.minecraft.launchwrapper.LaunchClassLoader;
+import net.minecraftforge.fml.relauncher.ReflectionHelper;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
@@ -12,6 +13,7 @@
 import java.net.*;
 import java.nio.file.FileSystem;
 import java.nio.file.*;
+import java.util.*;
 import java.util.concurrent.*;
 
 /**
@@ -203,6 +205,8 @@ private static void addToCurrentClassLoader() {
 				LaunchClassLoader lcl = (LaunchClassLoader) cl;
 				lcl.addTransformerExclusion("me.nallar.modpatcher");
 				method = LaunchClassLoader.class.getDeclaredMethod("addURL", URL.class);
+				Set<String> invalidClasses = ReflectionHelper.<Set<String>, LaunchClassLoader>getPrivateValue(LaunchClassLoader.class, lcl, "invalidClasses");
+				invalidClasses.removeIf(it -> it.startsWith("me.nallar."));
 			}
 			method.setAccessible(true);
 			method.invoke(cl, modPatcherPath.toUri().toURL());