138 lines
3.3 KiB
Java
138 lines
3.3 KiB
Java
package info.sigterm.deob;
|
|
|
|
import info.sigterm.deob.execution.Execution;
|
|
import info.sigterm.deob.pool.NameAndType;
|
|
import info.sigterm.deob.attributes.code.Instruction;
|
|
import info.sigterm.deob.attributes.code.instruction.types.LVTInstruction;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.DataInputStream;
|
|
import java.io.DataOutputStream;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.Enumeration;
|
|
import java.util.List;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarFile;
|
|
import java.util.jar.JarOutputStream;
|
|
import java.util.jar.Manifest;
|
|
|
|
public class Deob
|
|
{
|
|
public static void main(String[] args) throws IOException
|
|
{
|
|
ClassGroup group = new ClassGroup();
|
|
|
|
JarFile jar = new JarFile(args[0]);
|
|
for (Enumeration<JarEntry> it = jar.entries(); it.hasMoreElements();)
|
|
{
|
|
JarEntry entry = it.nextElement();
|
|
|
|
if (!entry.getName().endsWith(".class"))
|
|
continue;
|
|
|
|
InputStream is = jar.getInputStream(entry);
|
|
group.addClass(entry.getName(), new DataInputStream(is));
|
|
}
|
|
jar.close();
|
|
|
|
group.buildClassGraph();
|
|
group.buildInstructionGraph();
|
|
group.buildCallGraph();
|
|
|
|
checkCallGraph(group);
|
|
checkParameters(group);
|
|
|
|
//execute(group);
|
|
|
|
JarOutputStream jout = new JarOutputStream(new FileOutputStream(args[1]), new Manifest());
|
|
|
|
for (ClassFile cf : group.getClasses())
|
|
{
|
|
JarEntry entry = new JarEntry(cf.getName() + ".class");
|
|
jout.putNextEntry(entry);
|
|
|
|
ByteArrayOutputStream bout = new ByteArrayOutputStream();
|
|
cf.write(new DataOutputStream(bout));
|
|
jout.write(bout.toByteArray());
|
|
|
|
jout.closeEntry();
|
|
}
|
|
|
|
jout.close();
|
|
}
|
|
|
|
private static void execute(ClassGroup group) throws IOException
|
|
{
|
|
ClassFile cf = group.findClass("client");
|
|
Method method = cf.findMethod("init");
|
|
|
|
Execution e = new Execution(group);
|
|
e.run(cf, method);
|
|
}
|
|
|
|
private static void checkCallGraph(ClassGroup group)
|
|
{
|
|
int i = 0;
|
|
for (ClassFile cf : group.getClasses())
|
|
{
|
|
for (Method m : new ArrayList<>(cf.getMethods().getMethods()))
|
|
{
|
|
/* assume obfuscated names are <= 2 chars */
|
|
if (m.getName().length() > 2)
|
|
continue;
|
|
|
|
if (!m.isUsed())
|
|
{
|
|
System.out.println(cf.getName() + " " + m.getName());
|
|
cf.getMethods().removeMethod(m);
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
System.out.println("Removed " + i + " methods");
|
|
}
|
|
|
|
private static boolean parameterUsed(int num, List lv)
|
|
{
|
|
if (lv.isEmpty())
|
|
return false;
|
|
|
|
// these instructions aren't in order so we can't do this
|
|
//LVTInstruction ins = (LVTInstruction) lv.get(0);
|
|
//return !ins.store();
|
|
return true;
|
|
}
|
|
|
|
private static void checkParameters(ClassGroup group)
|
|
{
|
|
int count = 0;
|
|
for (ClassFile cf : group.getClasses())
|
|
{
|
|
for (Method m : new ArrayList<>(cf.getMethods().getMethods()))
|
|
{
|
|
int start = m.isStatic() ? 0 : 1;
|
|
NameAndType nat = m.getNameAndType();
|
|
int numParams = start + nat.getNumberOfArgs();
|
|
|
|
for (int i = start; i < numParams; ++i)
|
|
{
|
|
List lv = m.findLVTInstructionsForVariable(i);
|
|
|
|
if (lv == null)
|
|
continue;
|
|
|
|
if (!parameterUsed(i, lv))
|
|
{
|
|
System.out.println("Not used param " + i + " of " + cf.getName() + " " + m.getName() + " static: " + m.isStatic());
|
|
++count;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
System.out.println("Detected " + count + " unused parameters");
|
|
}
|
|
}
|