1 files changed, 361 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/ConvertMain.java b/src/main/java/cuchaz/enigma/ConvertMain.java
new file mode 100644
index 0000000..4c6be7f
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/ConvertMain.java
@@ -0,0 +1,361 @@
+/*******************************************************************************
+ * Copyright (c) 2015 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Lesser General Public
+ * License v3.0 which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/lgpl.html
+ * <p>
+ * Contributors:
+ * Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.jar.JarFile;
+
+import cuchaz.enigma.convert.*;
+import cuchaz.enigma.gui.ClassMatchingGui;
+import cuchaz.enigma.gui.MemberMatchingGui;
+import cuchaz.enigma.mapping.*;
+import cuchaz.enigma.throwables.MappingConflict;
+import cuchaz.enigma.throwables.MappingParseException;
+
+
+public class ConvertMain {
+
+    public static void main(String[] args)
+            throws IOException, MappingParseException {
+        try {
+            //Get all are args
+            String JarOld = getArg(args, 1, "Path to Old Jar", true);
+            String JarNew = getArg(args, 2, "Path to New Jar", true);
+            String OldMappings = getArg(args, 3, "Path to old .mappings file", true);
+            String NewMappings = getArg(args, 4, "Path to new .mappings file", true);
+            String ClassMatches = getArg(args, 5, "Path to Class .matches file", true);
+            String FieldMatches = getArg(args, 6, "Path to Field .matches file", true);
+            String MethodMatches = getArg(args, 7, "Path to Method .matches file", true);
+            //OldJar
+            JarFile sourceJar = new JarFile(new File(JarOld));
+            //NewJar
+            JarFile destJar = new JarFile(new File(JarNew));
+            //Get the mapping files
+            File inMappingsFile = new File(OldMappings);
+            File outMappingsFile = new File(NewMappings);
+            Mappings mappings = new MappingsEnigmaReader().read(inMappingsFile);
+            //Make the Match Files..
+            File classMatchesFile = new File(ClassMatches);
+            File fieldMatchesFile = new File(FieldMatches);
+            File methodMatchesFile = new File(MethodMatches);
+
+            String command = getArg(args, 0, "command", true);
+
+            if (command.equalsIgnoreCase("computeClassMatches")) {
+                computeClassMatches(classMatchesFile, sourceJar, destJar, mappings);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile);
+            } else if (command.equalsIgnoreCase("editClassMatches")) {
+                editClasssMatches(classMatchesFile, sourceJar, destJar, mappings);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile);
+            } else if (command.equalsIgnoreCase("computeFieldMatches")) {
+                computeFieldMatches(fieldMatchesFile, destJar, outMappingsFile, classMatchesFile);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile);
+            } else if (command.equalsIgnoreCase("editFieldMatches")) {
+                editFieldMatches(sourceJar, destJar, outMappingsFile, mappings, classMatchesFile, fieldMatchesFile);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile);
+            } else if (command.equalsIgnoreCase("computeMethodMatches")) {
+                computeMethodMatches(methodMatchesFile, destJar, outMappingsFile, classMatchesFile);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
+            } else if (command.equalsIgnoreCase("editMethodMatches")) {
+                editMethodMatches(sourceJar, destJar, outMappingsFile, mappings, classMatchesFile, methodMatchesFile);
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
+            } else if (command.equalsIgnoreCase("convertMappings")) {
+                convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
+            }
+        } catch (MappingConflict ex) {
+            System.out.println(ex.getMessage());
+            ex.printStackTrace();
+        } catch (IllegalArgumentException ex) {
+            System.out.println(ex.getMessage());
+            printHelp();
+        }
+    }
+
+    private static void printHelp() {
+        System.out.println(String.format("%s - %s", Constants.NAME, Constants.VERSION));
+        System.out.println("Usage:");
+        System.out.println("\tjava -cp enigma.jar cuchaz.enigma.ConvertMain <command> <old-jar> <new-jar> <old-mappings> <new-mappings> <class-matches> <field-matches> <method-matches>");
+        System.out.println("\tWhere <command> is one of:");
+        System.out.println("\t\tcomputeClassMatches");
+        System.out.println("\t\teditClassMatches");
+        System.out.println("\t\tcomputeFieldMatches");
+        System.out.println("\t\teditFieldMatches");
+        System.out.println("\t\teditMethodMatches");
+        System.out.println("\t\tconvertMappings");
+        System.out.println("\tWhere <old-jar> is the already mapped jar.");
+        System.out.println("\tWhere <new-jar> is the unmapped jar.");
+        System.out.println("\tWhere <old-mappings> is the path to the mappings for the old jar.");
+        System.out.println("\tWhere <new-mappings> is the new mappings. (Where you want to save them and there name)");
+        System.out.println("\tWhere <class-matches> is the class matches file.");
+        System.out.println("\tWhere <field-matches> is the field matches file.");
+        System.out.println("\tWhere <method-matches> is the method matches file.");
+    }
+
+    //Copy of getArg from CommandMain.... Should make a utils class.
+    private static String getArg(String[] args, int i, String name, boolean required) {
+        if (i >= args.length) {
+            if (required) {
+                throw new IllegalArgumentException(name + " is required");
+            } else {
+                return null;
+            }
+        }
+        return args[i];
+    }
+
+    private static void computeClassMatches(File classMatchesFile, JarFile sourceJar, JarFile destJar, Mappings mappings)
+            throws IOException {
+        ClassMatches classMatches = MappingsConverter.computeClassMatches(sourceJar, destJar, mappings);
+        MatchesWriter.writeClasses(classMatches, classMatchesFile);
+        System.out.println("Wrote:\n\t" + classMatchesFile.getAbsolutePath());
+    }
+
+    private static void editClasssMatches(final File classMatchesFile, JarFile sourceJar, JarFile destJar, Mappings mappings)
+            throws IOException {
+        System.out.println("Reading class matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(mappings);
+        System.out.println("Starting GUI...");
+        new ClassMatchingGui(classMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(new ClassMatchingGui.SaveListener() {
+            @Override
+            public void save(ClassMatches matches) {
+                try {
+                    MatchesWriter.writeClasses(matches, classMatchesFile);
+                } catch (IOException ex) {
+                    throw new Error(ex);
+                }
+            }
+        });
+    }
+
+    @SuppressWarnings("unused")
+    private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile)
+            throws IOException, MappingConflict {
+        System.out.println("Reading class matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(mappings);
+
+        Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
+        new MappingsEnigmaWriter().write(outMappingsFile, newMappings, true);
+        System.out.println("Write converted mappings to: " + outMappingsFile.getAbsolutePath());
+    }
+
+    private static void computeFieldMatches(File memberMatchesFile, JarFile destJar, File destMappingsFile, File classMatchesFile)
+            throws IOException, MappingParseException {
+
+        System.out.println("Reading class matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        System.out.println("Reading mappings...");
+        Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
+        System.out.println("Indexing dest jar...");
+        Deobfuscator destDeobfuscator = new Deobfuscator(destJar);
+
+        System.out.println("Writing matches...");
+
+        // get the matched and unmatched mappings
+        MemberMatches<FieldEntry> fieldMatches = MappingsConverter.computeMemberMatches(
+                destDeobfuscator,
+                destMappings,
+                classMatches,
+                MappingsConverter.getFieldDoer()
+        );
+
+        MatchesWriter.writeMembers(fieldMatches, memberMatchesFile);
+        System.out.println("Wrote:\n\t" + memberMatchesFile.getAbsolutePath());
+    }
+
+    private static void editFieldMatches(JarFile sourceJar, JarFile destJar, File destMappingsFile, Mappings sourceMappings, File classMatchesFile, final File fieldMatchesFile)
+            throws IOException, MappingParseException {
+
+        System.out.println("Reading matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);
+
+        // prep deobfuscators
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(sourceMappings);
+        Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
+        MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
+        checker.dropBrokenMappings(destMappings);
+        deobfuscators.dest.setMappings(destMappings);
+
+        new MemberMatchingGui<>(classMatches, fieldMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(new MemberMatchingGui.SaveListener<FieldEntry>() {
+            @Override
+            public void save(MemberMatches<FieldEntry> matches) {
+                try {
+                    MatchesWriter.writeMembers(matches, fieldMatchesFile);
+                } catch (IOException ex) {
+                    throw new Error(ex);
+                }
+            }
+        });
+    }
+
+    @SuppressWarnings("unused")
+    private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile, File fieldMatchesFile)
+            throws IOException, MappingConflict {
+
+        System.out.println("Reading matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);
+
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(mappings);
+
+        // apply matches
+        Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
+        MappingsConverter.applyMemberMatches(newMappings, classMatches, fieldMatches, MappingsConverter.getFieldDoer());
+
+        // write out the converted mappings
+
+        new MappingsEnigmaWriter().write(outMappingsFile, newMappings, true);
+        System.out.println("Wrote converted mappings to:\n\t" + outMappingsFile.getAbsolutePath());
+    }
+
+
+    private static void computeMethodMatches(File methodMatchesFile, JarFile destJar, File destMappingsFile, File classMatchesFile)
+            throws IOException, MappingParseException {
+
+        System.out.println("Reading class matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        System.out.println("Reading mappings...");
+        Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
+        System.out.println("Indexing dest jar...");
+        Deobfuscator destDeobfuscator = new Deobfuscator(destJar);
+
+        System.out.println("Writing method matches...");
+
+        // get the matched and unmatched mappings
+        MemberMatches<BehaviorEntry> methodMatches = MappingsConverter.computeMemberMatches(
+                destDeobfuscator,
+                destMappings,
+                classMatches,
+                MappingsConverter.getMethodDoer()
+        );
+
+        MatchesWriter.writeMembers(methodMatches, methodMatchesFile);
+        System.out.println("Wrote:\n\t" + methodMatchesFile.getAbsolutePath());
+    }
+
+    private static void editMethodMatches(JarFile sourceJar, JarFile destJar, File destMappingsFile, Mappings sourceMappings, File classMatchesFile, final File methodMatchesFile)
+            throws IOException, MappingParseException {
+
+        System.out.println("Reading matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        MemberMatches<BehaviorEntry> methodMatches = MatchesReader.readMembers(methodMatchesFile);
+
+        // prep deobfuscators
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(sourceMappings);
+        Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
+        MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
+        checker.dropBrokenMappings(destMappings);
+        deobfuscators.dest.setMappings(destMappings);
+
+        new MemberMatchingGui<>(classMatches, methodMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(new MemberMatchingGui.SaveListener<BehaviorEntry>() {
+            @Override
+            public void save(MemberMatches<BehaviorEntry> matches) {
+                try {
+                    MatchesWriter.writeMembers(matches, methodMatchesFile);
+                } catch (IOException ex) {
+                    throw new Error(ex);
+                }
+            }
+        });
+    }
+
+    private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile, File fieldMatchesFile, File methodMatchesFile)
+            throws IOException, MappingConflict {
+
+        System.out.println("Reading matches...");
+        ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
+        MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);
+        MemberMatches<BehaviorEntry> methodMatches = MatchesReader.readMembers(methodMatchesFile);
+
+        Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
+        deobfuscators.source.setMappings(mappings);
+
+        // apply matches
+        Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
+        MappingsConverter.applyMemberMatches(newMappings, classMatches, fieldMatches, MappingsConverter.getFieldDoer());
+        MappingsConverter.applyMemberMatches(newMappings, classMatches, methodMatches, MappingsConverter.getMethodDoer());
+
+        // check the final mappings
+        MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
+        checker.dropBrokenMappings(newMappings);
+
+        for (java.util.Map.Entry<ClassEntry, ClassMapping> mapping : checker.getDroppedClassMappings().entrySet()) {
+            System.out.println("WARNING: Broken class entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
+        }
+        for (java.util.Map.Entry<ClassEntry, ClassMapping> mapping : checker.getDroppedInnerClassMappings().entrySet()) {
+            System.out.println("WARNING: Broken inner class entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
+        }
+        for (java.util.Map.Entry<FieldEntry, FieldMapping> mapping : checker.getDroppedFieldMappings().entrySet()) {
+            System.out.println("WARNING: Broken field entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
+        }
+        for (java.util.Map.Entry<BehaviorEntry, MethodMapping> mapping : checker.getDroppedMethodMappings().entrySet()) {
+            System.out.println("WARNING: Broken behavior entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
+        }
+
+        //TODO Fix
+        // write out the converted mappings
+//        try (FileWriter out = new FileWriter(outMappingsFile)) {
+//            new MappingsWriter().write(out, newMappings);
+//        }
+        System.out.println("Wrote converted mappings to:\n\t" + outMappingsFile.getAbsolutePath());
+    }
+
+    private static class Deobfuscators {
+
+        public Deobfuscator source;
+        public Deobfuscator dest;
+
+        public Deobfuscators(JarFile sourceJar, JarFile destJar) {
+            System.out.println("Indexing source jar...");
+            IndexerThread sourceIndexer = new IndexerThread(sourceJar);
+            sourceIndexer.start();
+            System.out.println("Indexing dest jar...");
+            IndexerThread destIndexer = new IndexerThread(destJar);
+            destIndexer.start();
+            sourceIndexer.joinOrBail();
+            destIndexer.joinOrBail();
+            source = sourceIndexer.deobfuscator;
+            dest = destIndexer.deobfuscator;
+        }
+    }
+
+    private static class IndexerThread extends Thread {
+
+        private JarFile m_jarFile;
+        public Deobfuscator deobfuscator;
+
+        public IndexerThread(JarFile jarFile) {
+            m_jarFile = jarFile;
+            deobfuscator = null;
+        }
+
+        public void joinOrBail() {
+            try {
+                join();
+            } catch (InterruptedException ex) {
+                throw new Error(ex);
+            }
+        }
+
+        @Override
+        public void run() {
+            deobfuscator = new Deobfuscator(m_jarFile);
+        }
+    }
+}
\ No newline at end of file