1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
package cuchaz.enigma;
import com.strobel.assembler.metadata.ITypeLoader;
import com.strobel.assembler.metadata.MetadataSystem;
import com.strobel.assembler.metadata.TypeDefinition;
import com.strobel.assembler.metadata.TypeReference;
import com.strobel.decompiler.DecompilerContext;
import com.strobel.decompiler.DecompilerSettings;
import com.strobel.decompiler.PlainTextOutput;
import com.strobel.decompiler.languages.java.BraceStyle;
import com.strobel.decompiler.languages.java.JavaFormattingOptions;
import com.strobel.decompiler.languages.java.JavaOutputVisitor;
import com.strobel.decompiler.languages.java.ast.AstBuilder;
import com.strobel.decompiler.languages.java.ast.CompilationUnit;
import com.strobel.decompiler.languages.java.ast.InsertParenthesesVisitor;
import com.strobel.decompiler.languages.java.ast.transforms.IAstTransform;
import cuchaz.enigma.utils.Utils;
import oml.ast.transformers.*;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Arrays;
import java.util.List;
public class SourceProvider {
private final DecompilerSettings settings;
private final ITypeLoader typeLoader;
private final MetadataSystem metadataSystem;
public SourceProvider(DecompilerSettings settings, ITypeLoader typeLoader, MetadataSystem metadataSystem) {
this.settings = settings;
this.typeLoader = typeLoader;
this.metadataSystem = metadataSystem;
}
public SourceProvider(DecompilerSettings settings, ITypeLoader typeLoader) {
this(settings, typeLoader, new NoRetryMetadataSystem(typeLoader));
}
public static DecompilerSettings createSettings() {
DecompilerSettings settings = DecompilerSettings.javaDefaults();
settings.setMergeVariables(Utils.getSystemPropertyAsBoolean("enigma.mergeVariables", true));
settings.setForceExplicitImports(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitImports", true));
settings.setForceExplicitTypeArguments(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitTypeArguments", true));
settings.setShowDebugLineNumbers(Utils.getSystemPropertyAsBoolean("enigma.showDebugLineNumbers", false));
settings.setShowSyntheticMembers(Utils.getSystemPropertyAsBoolean("enigma.showSyntheticMembers", false));
JavaFormattingOptions formattingOptions = settings.getJavaFormattingOptions();
formattingOptions.ClassBraceStyle = BraceStyle.EndOfLine;
formattingOptions.InterfaceBraceStyle = BraceStyle.EndOfLine;
formattingOptions.EnumBraceStyle = BraceStyle.EndOfLine;
return settings;
}
public CompilationUnit getSources(String name) {
TypeReference type = metadataSystem.lookupType(name);
if (type == null) {
throw new Error(String.format("Unable to find desc: %s", name));
}
TypeDefinition resolvedType = type.resolve();
settings.setTypeLoader(typeLoader);
// decompile it!
DecompilerContext context = new DecompilerContext();
context.setCurrentType(resolvedType);
context.setSettings(settings);
AstBuilder builder = new AstBuilder(context);
builder.addType(resolvedType);
builder.runTransformations(null);
runCustomTransforms(builder, context);
return builder.getCompilationUnit();
}
public void writeSource(Writer writer, CompilationUnit sourceTree) {
// render the AST into source
sourceTree.acceptVisitor(new InsertParenthesesVisitor(), null);
sourceTree.acceptVisitor(new JavaOutputVisitor(new PlainTextOutput(writer), settings), null);
}
public String writeSourceToString(CompilationUnit sourceTree) {
StringWriter writer = new StringWriter();
writeSource(writer, sourceTree);
return writer.toString();
}
private static void runCustomTransforms(AstBuilder builder, DecompilerContext context) {
List<IAstTransform> transformers = Arrays.asList(
new ObfuscatedEnumSwitchRewriterTransform(context),
new VarargsFixer(context),
new RemoveObjectCasts(context),
new Java8Generics(),
new InvalidIdentifierFix()
);
for (IAstTransform transform : transformers) {
transform.run(builder.getCompilationUnit());
}
}
}
|
