ra/runelite
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Attempt to map non static methods using PME too

Browse Source
This commit is contained in:
Adam
2016-02-12 15:34:52 -05:00
parent 466da0bcb3
commit 8efd637a87
2 changed files with 51 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/main/java/net/runelite/deob/deobfuscators/rename/MethodSignatureMapper.java
View File

@@ -1,44 +1,42 @@
package net.runelite.deob.deobfuscators.rename;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Multimap;
import net.runelite.deob.ClassFile;
import net.runelite.deob.Method;
import net.runelite.deob.Methods;
import net.runelite.deob.signature.Signature;
public class MethodSignatureMapper
{
private Map<Method, Method> map = new HashMap<>();
private long count(Methods methods, Signature sig)
{
return methods.getMethods().stream().filter(m -> m.getDescriptor().equals(sig)).count();
}
private Method get(Methods methods, Signature sig)
{
Optional<Method> o = methods.getMethods().stream().filter(m -> m.getDescriptor().equals(sig)).findFirst();
return o.isPresent() ? o.get() : null;
}
private Multimap<Method, Method> map = ArrayListMultimap.create();
public void map(ClassFile c1, ClassFile c2)
{
for (Method m : c1.getMethods().getMethods())
{
if (m.isStatic() || count(c1.getMethods(), m.getDescriptor()) > 1)
if (m.isStatic() || m.getCode() == null)
continue;
Method other = get(c2.getMethods(), m.getDescriptor());
if (other == null)
continue;
boolean isConstructor = m.getName().equals("<init>");
map.put(m, other);
for (Method m2 : c2.getMethods().getMethods())
{
if (m2.getCode() == null)
continue;
if (!m.getDescriptor().equals(m2.getDescriptor()))
continue;
boolean isConstructor2 = m2.getName().equals("<init>");
if (isConstructor != isConstructor2)
continue;
map.put(m, m2);
}
}
}
public Map<Method, Method> getMap()
public Multimap<Method, Method> getMap()
{
return map;
}

65
src/test/java/net/runelite/deob/deobfuscators/rename/MapStaticTest.java
View File

@@ -1,5 +1,6 @@
package net.runelite.deob.deobfuscators.rename;
import com.google.common.collect.Multimap;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
@@ -116,8 +117,8 @@ public class MapStaticTest
ClassGroup group1 = JarUtil.loadJar(new File(JAR1));
ClassGroup group2 = JarUtil.loadJar(new File(JAR2));
Method m1 = group1.findClass("class92").findMethod("method2176");
Method m2 = group2.findClass("client").findMethod("method540");
Method m1 = group1.findClass("client").findMethod("method273");
Method m2 = group2.findClass("client").findMethod("method235");
HashMap<Object, Object> all = new HashMap();
List<ParallelExecutorMapping> pmes = new ArrayList<>();
@@ -209,25 +210,6 @@ public class MapStaticTest
map(all, pmes, m1, m2);
}
for (int i = 0; i < 250; ++i)
{
ClassFile c1 = group1.findClass("class" + i);
ClassFile c2 = group2.findClass("class" + i);
if (c1 == null || c2 == null)
continue;
MethodSignatureMapper msm = new MethodSignatureMapper();
msm.map(c1, c2);
Map<Method, Method> map = msm.getMap();
for (Entry<Method, Method> e : map.entrySet())
{
HashMap<Object, Object> all = new HashMap();
map(all, pmes, e.getKey(), e.getValue());
}
}
ParallelExecutorMapping finalm = new ParallelExecutorMapping(group1, group2);
for (ParallelExecutorMapping pme : pmes)
finalm.merge(pme);
@@ -245,16 +227,16 @@ public class MapStaticTest
System.out.println("GROUP 2 " + sg2);
// for (Method m : group1.findClass("client").getMethods().getMethods())
// {
// if (!finalm.getMap().containsKey(m))
// System.out.println("missing " + m);
// }
// for (Field m : group1.findClass("client").getFields().getFields())
// {
// if (!finalm.getMap().containsKey(m))
// System.out.println("missing " + m);
// }
for (Method m : group1.findClass("client").getMethods().getMethods())
{
if (!finalm.getMap().containsKey(m))
System.out.println("missing " + m);
}
for (Field m : group1.findClass("client").getFields().getFields())
{
if (!finalm.getMap().containsKey(m))
System.out.println("missing " + m);
}
}
//@Test
@@ -275,11 +257,24 @@ public class MapStaticTest
MethodSignatureMapper msm = new MethodSignatureMapper();
msm.map(c1, c2);
Map<Method, Method> map = msm.getMap();
for (Entry<Method, Method> e : map.entrySet())
Multimap<Method, Method> map = msm.getMap();
for (Method m : msm.getMap().keySet())
{
HashMap<Object, Object> all = new HashMap();
map(all, pmes, e.getKey(), e.getValue());
Collection<Method> methods = msm.getMap().get(m);
for (Method other : methods)
{
ParallelExecutorMapping pme = MappingExecutorUtil.map(m, other);
if (pme.getMap().isEmpty())
continue;
pme.map(m, other);
pmes.add(pme);
}
//HashMap<Object, Object> all = new HashMap();
//map(all, pmes, e.getKey(), e.getValue());
}
}
ParallelExecutorMapping finalm = new ParallelExecutorMapping(one, two);