blob: 83a21719df6dc8b2b9a94662a8ebbb7abdf22dd7 (
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
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
|
/*******************************************************************************
* Copyright (c) 2014 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Public License v3.0
* which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/gpl.html
*
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma;
import java.io.File;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import com.strobel.assembler.metadata.JarTypeLoader;
import com.strobel.decompiler.Decompiler;
import com.strobel.decompiler.DecompilerSettings;
import com.strobel.decompiler.PlainTextOutput;
public class Deobfuscator
{
private File m_file;
private JarFile m_jar;
private DecompilerSettings m_settings;
private static Comparator<ClassFile> m_obfuscatedClassSorter;
static
{
m_obfuscatedClassSorter = new Comparator<ClassFile>( )
{
@Override
public int compare( ClassFile a, ClassFile b )
{
if( a.getName().length() != b.getName().length() )
{
return a.getName().length() - b.getName().length();
}
return a.getName().compareTo( b.getName() );
}
};
}
public Deobfuscator( File file )
throws IOException
{
m_file = file;
m_jar = new JarFile( m_file );
m_settings = DecompilerSettings.javaDefaults();
m_settings.setTypeLoader( new JarTypeLoader( m_jar ) );
m_settings.setForceExplicitImports( true );
m_settings.setShowSyntheticMembers( true );
}
public List<ClassFile> getObfuscatedClasses( )
{
List<ClassFile> classes = new ArrayList<ClassFile>();
Enumeration<JarEntry> entries = m_jar.entries();
while( entries.hasMoreElements() )
{
JarEntry entry = entries.nextElement();
// get the class name
String className = toClassName( entry.getName() );
if( className == null )
{
continue;
}
ClassFile classFile = new ClassFile( className );
if( classFile.isObfuscated() )
{
classes.add( classFile );
}
}
Collections.sort( classes, m_obfuscatedClassSorter );
return classes;
}
// TODO: could go somewhere more generic
private static String toClassName( String fileName )
{
final String suffix = ".class";
if( !fileName.endsWith( suffix ) )
{
return null;
}
return fileName.substring( 0, fileName.length() - suffix.length() ).replace( "/", "." );
}
public String getSource( final ClassFile classFile )
{
StringWriter buf = new StringWriter();
Decompiler.decompile( classFile.getName(), new PlainTextOutput( buf ), m_settings );
return buf.toString();
}
}
|
