From 6c90b5f41ec9ea8c512981e4dbe69d5d9769651f Mon Sep 17 00:00:00 2001 From: modmuss50 Date: Thu, 11 Aug 2016 14:18:43 +0100 Subject: Allow exporting mappings as SRG or Enigma --- src/main/java/cuchaz/enigma/gui/Gui.java | 4 + src/main/java/cuchaz/enigma/gui/GuiController.java | 12 +++ .../java/cuchaz/enigma/gui/elements/MenuBar.java | 34 +++++++++ .../cuchaz/enigma/mapping/MappingsOldWriter.java | 88 ++++++++++++++++++++++ .../cuchaz/enigma/mapping/MappingsSRGWriter.java | 82 ++++++++++++++++++++ 5 files changed, 220 insertions(+) create mode 100644 src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java create mode 100644 src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java (limited to 'src/main') diff --git a/src/main/java/cuchaz/enigma/gui/Gui.java b/src/main/java/cuchaz/enigma/gui/Gui.java index d93aa9f..5b0b599 100644 --- a/src/main/java/cuchaz/enigma/gui/Gui.java +++ b/src/main/java/cuchaz/enigma/gui/Gui.java @@ -319,6 +319,8 @@ public class Gui { this.menuBar.openMappingsMenu.setEnabled(true); this.menuBar.saveMappingsMenu.setEnabled(false); this.menuBar.saveMappingsAsMenu.setEnabled(true); + this.menuBar.saveMappingsOldMenu.setEnabled(true); + this.menuBar.saveMappingsSrgMenu.setEnabled(true); this.menuBar.closeMappingsMenu.setEnabled(true); this.menuBar.exportSourceMenu.setEnabled(true); this.menuBar.exportJarMenu.setEnabled(true); @@ -340,6 +342,8 @@ public class Gui { this.menuBar.openMappingsMenu.setEnabled(false); this.menuBar.saveMappingsMenu.setEnabled(false); this.menuBar.saveMappingsAsMenu.setEnabled(false); + this.menuBar.saveMappingsOldMenu.setEnabled(false); + this.menuBar.saveMappingsSrgMenu.setEnabled(false); this.menuBar.closeMappingsMenu.setEnabled(false); this.menuBar.exportSourceMenu.setEnabled(false); this.menuBar.exportJarMenu.setEnabled(false); diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java index c301594..fe7d097 100644 --- a/src/main/java/cuchaz/enigma/gui/GuiController.java +++ b/src/main/java/cuchaz/enigma/gui/GuiController.java @@ -17,6 +17,7 @@ import com.strobel.decompiler.languages.java.ast.CompilationUnit; import java.io.File; import java.io.FileReader; +import java.io.FileWriter; import java.io.IOException; import java.util.Collection; import java.util.Deque; @@ -87,6 +88,17 @@ public class GuiController { this.isDirty = false; } + public void saveOldMappings(File file) throws IOException { + FileWriter out = new FileWriter(file); + new MappingsOldWriter().write(out, this.deobfuscator.getMappings()); + this.isDirty = false; + } + + public void saveSRGMappings(File file) throws IOException { + new MappingsSRGWriter().write(file, this.deobfuscator); + this.isDirty = false; + } + public void closeMappings() { this.deobfuscator.setMappings(null); this.gui.setMappingsFile(null); diff --git a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java index e870334..e79476b 100644 --- a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java +++ b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java @@ -22,6 +22,8 @@ public class MenuBar extends JMenuBar { public final JMenuItem saveMappingsMenu; public final JMenuItem saveMappingsAsMenu; + public final JMenuItem saveMappingsOldMenu; + public final JMenuItem saveMappingsSrgMenu; public final JMenuItem closeMappingsMenu; @@ -120,6 +122,38 @@ public class MenuBar extends JMenuBar { item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_S, InputEvent.CTRL_DOWN_MASK | InputEvent.SHIFT_DOWN_MASK)); this.saveMappingsAsMenu = item; } + { + JMenuItem item = new JMenuItem("Save Mappings as Enigma"); + menu.add(item); + item.addActionListener(event -> { + if (this.gui.mappingsFileChooser.showSaveDialog(this.gui.getFrame()) == JFileChooser.APPROVE_OPTION) { + try { + this.gui.getController().saveOldMappings(this.gui.mappingsFileChooser.getSelectedFile()); + this.saveMappingsMenu.setEnabled(true); + } catch (IOException ex) { + throw new Error(ex); + } + } + }); + item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_S, InputEvent.CTRL_DOWN_MASK | InputEvent.SHIFT_DOWN_MASK)); + this.saveMappingsOldMenu = item; + } + { + JMenuItem item = new JMenuItem("Save Mappings as SRG"); + menu.add(item); + item.addActionListener(event -> { + if (this.gui.mappingsFileChooser.showSaveDialog(this.gui.getFrame()) == JFileChooser.APPROVE_OPTION) { + try { + this.gui.getController().saveSRGMappings(this.gui.mappingsFileChooser.getSelectedFile()); + this.saveMappingsMenu.setEnabled(true); + } catch (IOException ex) { + throw new Error(ex); + } + } + }); + item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_S, InputEvent.CTRL_DOWN_MASK | InputEvent.SHIFT_DOWN_MASK)); + this.saveMappingsSrgMenu = item; + } { JMenuItem item = new JMenuItem("Close Mappings"); menu.add(item); diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java new file mode 100644 index 0000000..8b766a3 --- /dev/null +++ b/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java @@ -0,0 +1,88 @@ +/******************************************************************************* + * 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 + * + * Contributors: + * Jeff Martin - initial API and implementation + ******************************************************************************/ +package cuchaz.enigma.mapping; + +import java.io.IOException; +import java.io.PrintWriter; +import java.io.Writer; +import java.util.ArrayList; +import java.util.Collections; +import java.util.List; + +public class MappingsOldWriter { + + public void write(Writer out, Mappings mappings) throws IOException { + write(new PrintWriter(out), mappings); + } + + public void write(PrintWriter out, Mappings mappings) throws IOException { + for (ClassMapping classMapping : sorted(mappings.classes())) { + write(out, classMapping, 0); + } + } + + private void write(PrintWriter out, ClassMapping classMapping, int depth) throws IOException { + if (classMapping.getDeobfName() == null) { + out.format("%sCLASS %s\n", getIndent(depth), classMapping.getObfFullName()); + } else { + out.format("%sCLASS %s %s\n", getIndent(depth), classMapping.getObfFullName(), classMapping.getDeobfName()); + } + + for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) { + write(out, innerClassMapping, depth + 1); + } + + for (FieldMapping fieldMapping : sorted(classMapping.fields())) { + write(out, fieldMapping, depth + 1); + } + + for (MethodMapping methodMapping : sorted(classMapping.methods())) { + write(out, methodMapping, depth + 1); + } + } + + private void write(PrintWriter out, FieldMapping fieldMapping, int depth) throws IOException { + out.format("%sFIELD %s %s %s\n", getIndent(depth), fieldMapping.getObfName(), fieldMapping.getDeobfName(), fieldMapping.getObfType().toString()); + } + + private void write(PrintWriter out, MethodMapping methodMapping, int depth) throws IOException { + if (methodMapping.getDeobfName() == null) { + out.format("%sMETHOD %s %s\n", getIndent(depth), methodMapping.getObfName(), methodMapping.getObfSignature()); + } else { + out.format("%sMETHOD %s %s %s\n", getIndent(depth), methodMapping.getObfName(), methodMapping.getDeobfName(), methodMapping.getObfSignature()); + } + + for (ArgumentMapping argumentMapping : sorted(methodMapping.arguments())) { + write(out, argumentMapping, depth + 1); + } + } + + private void write(PrintWriter out, ArgumentMapping argumentMapping, int depth) throws IOException { + out.format("%sARG %d %s\n", getIndent(depth), argumentMapping.getIndex(), argumentMapping.getName()); + } + + private > List sorted(Iterable classes) { + List out = new ArrayList(); + for (T t : classes) { + out.add(t); + } + Collections.sort(out); + return out; + } + + private String getIndent(int depth) { + StringBuilder buf = new StringBuilder(); + for (int i = 0; i < depth; i++) { + buf.append("\t"); + } + return buf.toString(); + } +} diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java new file mode 100644 index 0000000..64da709 --- /dev/null +++ b/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java @@ -0,0 +1,82 @@ +package cuchaz.enigma.mapping; + +import cuchaz.enigma.Deobfuscator; + +import java.io.File; +import java.io.FileWriter; +import java.io.IOException; +import java.util.ArrayList; +import java.util.Collections; +import java.util.List; + +/** + * Created by Mark on 11/08/2016. + */ +public class MappingsSRGWriter { + + public void write(File file, Deobfuscator deobfuscator) throws IOException { + if(file.exists()){ + file.delete(); + } + file.createNewFile(); + + Mappings mappings = deobfuscator.getMappings(); + + FileWriter writer = new FileWriter(file); + List fieldMappings = new ArrayList<>(); + List methodMappings = new ArrayList<>(); + for (ClassMapping classMapping : sorted(mappings.classes())) { + if(classMapping.getDeobfName() == null || classMapping.getObfSimpleName() == null || classMapping.getDeobfName() == null){ + continue; + } + writer.write("CL: " + classMapping.getObfSimpleName() + " " + classMapping.getDeobfName()); + writer.write(System.lineSeparator()); + for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) { + if(innerClassMapping.getDeobfName() == null || innerClassMapping.getObfSimpleName() == null || innerClassMapping.getDeobfName() == null){ + continue; + } + String innerClassName = classMapping.getObfSimpleName() + "$" + innerClassMapping.getObfSimpleName().replace("none/", ""); + String innerDebofClassName = classMapping.getDeobfName() + "$" + innerClassMapping.getDeobfName().replace("none/", ""); + writer.write("CL: " + innerClassName + " " + classMapping.getDeobfName() + "$" + innerClassMapping.getDeobfName()); + writer.write(System.lineSeparator()); + for (FieldMapping fieldMapping : sorted(innerClassMapping.fields())) { + fieldMappings.add("FD: " + innerClassName + "/" + fieldMapping.getObfName() + " " + innerDebofClassName + "/" + fieldMapping.getDeobfName()); + } + + for (MethodMapping methodMapping : sorted(innerClassMapping.methods())) { + methodMappings.add("MD: " + innerClassName + "/" + methodMapping.getObfName() + " " + methodMapping.getObfSignature().toString().replace("none/", "") + " " + innerDebofClassName + "/" + methodMapping.getDeobfName() + " " + deobfuscator.getTranslator(TranslationDirection.Deobfuscating).translateSignature(methodMapping.getObfSignature())); + } + } + + for (FieldMapping fieldMapping : sorted(classMapping.fields())) { + fieldMappings.add("FD: " + classMapping.getObfFullName().replace("none/", "") + "/" + fieldMapping.getObfName() + " " + classMapping.getDeobfName() + "/" + fieldMapping.getDeobfName()); + } + + for (MethodMapping methodMapping : sorted(classMapping.methods())) { + methodMappings.add("MD: " + classMapping.getObfFullName().replace("none/", "") + "/" + methodMapping.getObfName() + " " + methodMapping.getObfSignature().toString().replace("none/", "") + " " + classMapping.getDeobfName() + "/" + methodMapping.getDeobfName() + " " + deobfuscator.getTranslator(TranslationDirection.Deobfuscating).translateSignature(methodMapping.getObfSignature())); + } + } + for(String fd : fieldMappings){ + writer.write(fd); + writer.write(System.lineSeparator()); + } + + for(String md : methodMappings){ + writer.write(md); + writer.write(System.lineSeparator()); + } + + + writer.close(); + } + + + private > List sorted(Iterable classes) { + List out = new ArrayList(); + for (T t : classes) { + out.add(t); + } + Collections.sort(out); + return out; + } +} -- cgit v1.2.3