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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
|
/*******************************************************************************
* 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.gui;
import com.strobel.decompiler.languages.java.ast.CompilationUnit;
import java.awt.Rectangle;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.JEditorPane;
import javax.swing.SwingUtilities;
import javax.swing.Timer;
import javax.swing.text.BadLocationException;
import javax.swing.text.Highlighter.HighlightPainter;
import cuchaz.enigma.Deobfuscator;
import cuchaz.enigma.analysis.EntryReference;
import cuchaz.enigma.analysis.SourceIndex;
import cuchaz.enigma.analysis.Token;
import cuchaz.enigma.mapping.ClassEntry;
import cuchaz.enigma.mapping.Entry;
import de.sciss.syntaxpane.DefaultSyntaxKit;
public class CodeReader extends JEditorPane {
private static final long serialVersionUID = 3673180950485748810L;
private static final Object lock = new Object();
public interface SelectionListener {
void onSelect(EntryReference<Entry, Entry> reference);
}
private SelectionHighlightPainter selectionHighlightPainter;
private SourceIndex sourceIndex;
private SelectionListener selectionListener;
public CodeReader() {
setEditable(false);
setContentType("text/java");
// turn off token highlighting (it's wrong most of the time anyway...)
DefaultSyntaxKit kit = (DefaultSyntaxKit) getEditorKit();
kit.toggleComponent(this, "de.sciss.syntaxpane.components.TokenMarker");
// hook events
addCaretListener(event -> {
if (this.selectionListener != null && this.sourceIndex != null) {
Token token = this.sourceIndex.getReferenceToken(event.getDot());
if (token != null) {
this.selectionListener.onSelect(this.sourceIndex.getDeobfReference(token));
} else {
this.selectionListener.onSelect(null);
}
}
});
this.selectionHighlightPainter = new SelectionHighlightPainter();
this.sourceIndex = null;
this.selectionListener = null;
}
public void setSelectionListener(SelectionListener val) {
this.selectionListener = val;
}
public void setCode(String code) {
// sadly, the java lexer is not thread safe, so we have to serialize all these calls
synchronized (lock) {
setText(code);
}
}
public SourceIndex getSourceIndex() {
return this.sourceIndex;
}
public void decompileClass(ClassEntry classEntry, Deobfuscator deobfuscator, Runnable callback) {
decompileClass(classEntry, deobfuscator, null, callback);
}
public void decompileClass(final ClassEntry classEntry, final Deobfuscator deobfuscator, final Boolean ignoreBadTokens, final Runnable callback) {
if (classEntry == null) {
setCode(null);
return;
}
setCode("(decompiling...)");
// run decompilation in a separate thread to keep ui responsive
new Thread() {
@Override
public void run() {
// decompile it
CompilationUnit sourceTree = deobfuscator.getSourceTree(classEntry.getOutermostClassName());
String source = deobfuscator.getSource(sourceTree);
setCode(source);
sourceIndex = deobfuscator.getSourceIndex(sourceTree, source, ignoreBadTokens);
if (callback != null) {
callback.run();
}
}
}.start();
}
public void navigateToClassDeclaration(ClassEntry classEntry) {
// navigate to the class declaration
Token token = this.sourceIndex.getDeclarationToken(classEntry);
if (token == null) {
// couldn't find the class declaration token, might be an anonymous class
// look for any declaration in that class instead
for (Entry entry : this.sourceIndex.declarations()) {
if (entry.getClassEntry().equals(classEntry)) {
token = this.sourceIndex.getDeclarationToken(entry);
break;
}
}
}
if (token != null) {
navigateToToken(token);
} else {
// couldn't find anything =(
System.out.println("Unable to find declaration in source for " + classEntry);
}
}
public void navigateToToken(final Token token) {
navigateToToken(this, token, this.selectionHighlightPainter);
}
// HACKHACK: someday we can update the main GUI to use this code reader
public static void navigateToToken(final JEditorPane editor, final Token token, final HighlightPainter highlightPainter) {
// set the caret position to the token
editor.setCaretPosition(token.start);
editor.grabFocus();
try {
// make sure the token is visible in the scroll window
Rectangle start = editor.modelToView(token.start);
Rectangle end = editor.modelToView(token.end);
final Rectangle show = start.union(end);
show.grow(start.width * 10, start.height * 6);
SwingUtilities.invokeLater(() -> editor.scrollRectToVisible(show));
} catch (BadLocationException ex) {
throw new Error(ex);
}
// highlight the token momentarily
final Timer timer = new Timer(200, new ActionListener() {
private int m_counter = 0;
private Object m_highlight = null;
@Override
public void actionPerformed(ActionEvent event) {
if (m_counter % 2 == 0) {
try {
m_highlight = editor.getHighlighter().addHighlight(token.start, token.end, highlightPainter);
} catch (BadLocationException ex) {
// don't care
}
} else if (m_highlight != null) {
editor.getHighlighter().removeHighlight(m_highlight);
}
if (m_counter++ > 6) {
Timer timer = (Timer) event.getSource();
timer.stop();
}
}
});
timer.start();
}
public void setHighlightedToken(Token token, HighlightPainter painter) {
try {
getHighlighter().addHighlight(token.start, token.end, painter);
} catch (BadLocationException ex) {
throw new IllegalArgumentException(ex);
}
}
public void clearHighlights() {
getHighlighter().removeAllHighlights();
}
}
|
