blob: 37bc0c86022d7e1a8a42aca051592b260ac7c777 (
plain) (
blame)
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
|
package cuchaz.enigma.source.procyon;
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.languages.java.BraceStyle;
import com.strobel.decompiler.languages.java.JavaFormattingOptions;
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 cuchaz.enigma.ClassProvider;
import cuchaz.enigma.api.EnigmaPluginContext;
import cuchaz.enigma.source.Source;
import cuchaz.enigma.source.Decompiler;
import cuchaz.enigma.source.SourceSettings;
import cuchaz.enigma.source.procyon.transformers.*;
import cuchaz.enigma.source.procyon.typeloader.CompiledSourceTypeLoader;
import cuchaz.enigma.source.procyon.typeloader.NoRetryMetadataSystem;
import cuchaz.enigma.source.procyon.typeloader.SynchronizedTypeLoader;
import cuchaz.enigma.utils.Utils;
public class ProcyonDecompiler implements Decompiler {
private final SourceSettings settings;
private final DecompilerSettings decompilerSettings;
private final MetadataSystem metadataSystem;
public ProcyonDecompiler(ClassProvider classProvider, SourceSettings settings) {
ITypeLoader typeLoader = new SynchronizedTypeLoader(new CompiledSourceTypeLoader(classProvider));
metadataSystem = new NoRetryMetadataSystem(typeLoader);
metadataSystem.setEagerMethodLoadingEnabled(true);
decompilerSettings = DecompilerSettings.javaDefaults();
decompilerSettings.setMergeVariables(Utils.getSystemPropertyAsBoolean("enigma.mergeVariables", true));
decompilerSettings.setForceExplicitImports(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitImports", true));
decompilerSettings.setForceExplicitTypeArguments(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitTypeArguments", true));
decompilerSettings.setShowDebugLineNumbers(Utils.getSystemPropertyAsBoolean("enigma.showDebugLineNumbers", false));
decompilerSettings.setShowSyntheticMembers(Utils.getSystemPropertyAsBoolean("enigma.showSyntheticMembers", false));
decompilerSettings.setTypeLoader(typeLoader);
JavaFormattingOptions formattingOptions = decompilerSettings.getJavaFormattingOptions();
formattingOptions.ClassBraceStyle = BraceStyle.EndOfLine;
formattingOptions.InterfaceBraceStyle = BraceStyle.EndOfLine;
formattingOptions.EnumBraceStyle = BraceStyle.EndOfLine;
this.settings = settings;
}
@Override
public Source getSource(String className) {
TypeReference type = metadataSystem.lookupType(className);
if (type == null) {
throw new Error(String.format("Unable to find desc: %s", className));
}
TypeDefinition resolvedType = type.resolve();
DecompilerContext context = new DecompilerContext();
context.setCurrentType(resolvedType);
context.setSettings(decompilerSettings);
AstBuilder builder = new AstBuilder(context);
builder.addType(resolvedType);
builder.runTransformations(null);
CompilationUnit source = builder.getCompilationUnit();
new ObfuscatedEnumSwitchRewriterTransform(context).run(source);
new VarargsFixer(context).run(source);
new RemoveObjectCasts(context).run(source);
new Java8Generics().run(source);
new InvalidIdentifierFix().run(source);
if (settings.removeImports) DropImportAstTransform.INSTANCE.run(source);
if (settings.removeVariableFinal) DropVarModifiersAstTransform.INSTANCE.run(source);
source.acceptVisitor(new InsertParenthesesVisitor(), null);
return new ProcyonSource(source, decompilerSettings);
}
}
|
