Added checking for main-classes

im-frizzy · im-frizzy · commit a00d48ee7b50 · 2015-02-28T22:19:51.000-06:00
diff --git a/src/the/bytecode/club/bytecodeviewer/BytecodeViewer.java b/src/the/bytecode/club/bytecodeviewer/BytecodeViewer.java
@@ -14,6 +14,7 @@
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map.Entry;
 
 import javax.swing.JDialog;
@@ -120,6 +121,7 @@ public class BytecodeViewer {
 	
 	/* ASM Re-mapping Constants */
 	public static Refactorer refactorer = new Refactorer();
+	public static List<String> main_classes = new ArrayList<>();
 	/* ASM Re-mapping Constants */
 	
 	/**
diff --git a/src/the/bytecode/club/bytecodeviewer/obfuscators/rename/RenameClasses.java b/src/the/bytecode/club/bytecodeviewer/obfuscators/rename/RenameClasses.java
@@ -27,6 +27,12 @@ public void obfuscate() {
 		System.out.println("Obfuscating class names...");
 		classLoop: for (ClassNode c : BytecodeViewer.getLoadedClasses()) {
 			
+			/** As we dont want to rename classes the user specified as main-classes */
+			for (String name : BytecodeViewer.main_classes) {
+				if (name.equals(c.name))
+					continue classLoop;
+			}
+			
 			/** As we dont want to rename classes that contain native dll methods */
 			for (Object o : c.methods) {
 				MethodNode m = (MethodNode) o;
