Only dump relevant classes when enabled

MinimallyCorrect · Jun 22, 2016 · 3ac9396 · 3ac9396
1 parent ae7e834
commit 3ac9396
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/main/java/me/nallar/modpatcher/LaunchClassLoaderUtil.java b/src/main/java/me/nallar/modpatcher/LaunchClassLoaderUtil.java
@@ -50,7 +50,9 @@ public static void addTransformer(IClassTransformer transformer) {
 		PatcherLog.warn("Didn't find deobfuscation transformer " + LaunchClassLoaderUtil.DEOBF_TRANSFORMER_NAME + " in transformers list.\n" +
 			"Did you forget to set the SortingIndex for your coremod >= 1001? This message is expected in a deobf environment.");
 		transformers.add(transformer);
+	}
 
+	public static void dumpTransformersIfEnabled() {
 		if (System.getProperty(DUMP_TRANSFORMERS_PROPERTY_NAME) != null)
 			PatcherLog.info("Transformers: " + transformers.toString(), new Throwable());
 	}

diff --git a/src/main/java/me/nallar/modpatcher/ModPatcherTransformer.java b/src/main/java/me/nallar/modpatcher/ModPatcherTransformer.java
@@ -107,6 +107,7 @@ public static void initialiseClassLoader(LaunchClassLoader classLoader) {
 
 		LaunchClassLoaderUtil.instance = classLoader;
 		LaunchClassLoaderUtil.addTransformer(ModPatcherTransformer.getInstance());
+		LaunchClassLoaderUtil.dumpTransformersIfEnabled();
 		LaunchClassLoaderUtil.removeRedundantExclusions();
 	}
 
@@ -131,7 +132,7 @@ private static class ClassTransformer implements IClassTransformer {
 		private boolean init;
 
 		private static void dumpIfEnabled(String name, byte[] data) {
-			if (!DUMP)
+			if (!DUMP || !name.contains("net.minecraft"))
 				return;
 
 			Path path = Paths.get("./DUMP/" + name + ".class");
@@ -150,11 +151,12 @@ public byte[] transform(String name, String transformedName, byte[] bytes) {
 				patcher.logDebugInfo();
 			}
 
-			dumpIfEnabled(name + "_unpatched", bytes);
+			dumpIfEnabled(transformedName + "_unpatched", bytes);
 
 			byte[] transformed = transformationHook(transformedName, bytes);
 
-			dumpIfEnabled(name, transformed);
+			if (bytes != transformed)
+				dumpIfEnabled(transformedName, transformed);
 
 			return transformed;
 		}