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 --- .../cuchaz/enigma/mapping/MappingsOldWriter.java | 88 ++++++++++++++++++++++ .../cuchaz/enigma/mapping/MappingsSRGWriter.java | 82 ++++++++++++++++++++ 2 files changed, 170 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/java/cuchaz/enigma/mapping') 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