88 lines
1.5 KiB
Java
88 lines
1.5 KiB
Java
package info.sigterm.deob;
|
|
|
|
import info.sigterm.deob.pool.NameAndType;
|
|
|
|
import java.io.DataInputStream;
|
|
import java.io.DataOutputStream;
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class Methods
|
|
{
|
|
ClassFile classFile;
|
|
|
|
private List<Method> methods = new ArrayList<>();
|
|
|
|
Methods(ClassFile cf) throws IOException
|
|
{
|
|
classFile = cf;
|
|
|
|
DataInputStream is = cf.getStream();
|
|
|
|
int count = is.readUnsignedShort();
|
|
|
|
for (int i = 0; i < count; ++i)
|
|
methods.add(new Method(this));
|
|
}
|
|
|
|
public void write(DataOutputStream out) throws IOException
|
|
{
|
|
out.writeShort(methods.size());
|
|
for (Method m : methods)
|
|
m.write(out);
|
|
}
|
|
|
|
public void removeMethod(Method m)
|
|
{
|
|
m.remove();
|
|
methods.remove(m);
|
|
}
|
|
|
|
public ClassFile getClassFile()
|
|
{
|
|
return classFile;
|
|
}
|
|
|
|
public List<Method> getMethods()
|
|
{
|
|
return methods;
|
|
}
|
|
|
|
public Method findMethod(NameAndType nat)
|
|
{
|
|
for (Method m : methods)
|
|
if (m.getName().equals(nat.getName()) && m.getDescriptor().equals(nat.getDescriptor()))
|
|
return m;
|
|
return null;
|
|
}
|
|
|
|
public Method findMethod(String name, String type)
|
|
{
|
|
for (Method m : methods)
|
|
if (m.getName().equals(name) && m.getDescriptor().equals(type))
|
|
return m;
|
|
return null;
|
|
}
|
|
|
|
public Method findMethod(String name)
|
|
{
|
|
for (Method m : methods)
|
|
if (m.getName().equals(name))
|
|
return m;
|
|
return null;
|
|
}
|
|
|
|
public void buildInstructionGraph()
|
|
{
|
|
for (Method m : methods)
|
|
m.buildInstructionGraph();
|
|
}
|
|
|
|
public void buildCallGraph()
|
|
{
|
|
for (Method m : methods)
|
|
m.buildCallGraph();
|
|
}
|
|
}
|