Rename classes last so we can do it in 1 pass
This commit is contained in:
@@ -143,9 +143,8 @@ public class RenameUnique implements Deobfuscator
|
|||||||
return list;
|
return list;
|
||||||
}
|
}
|
||||||
|
|
||||||
private NameMappings generateClassNames(ClassGroup group)
|
private void generateClassNames(NameMappings map, ClassGroup group)
|
||||||
{
|
{
|
||||||
NameMappings map = new NameMappings();
|
|
||||||
int i = 0;
|
int i = 0;
|
||||||
|
|
||||||
for (ClassFile cf : group.getClasses())
|
for (ClassFile cf : group.getClasses())
|
||||||
@@ -155,13 +154,10 @@ public class RenameUnique implements Deobfuscator
|
|||||||
|
|
||||||
map.map(cf.getPoolClass(), "class" + i++);
|
map.map(cf.getPoolClass(), "class" + i++);
|
||||||
}
|
}
|
||||||
|
|
||||||
return map;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private NameMappings generatFieldNames(ClassGroup group)
|
private void generatFieldNames(NameMappings map, ClassGroup group)
|
||||||
{
|
{
|
||||||
NameMappings map = new NameMappings();
|
|
||||||
int i = 0;
|
int i = 0;
|
||||||
|
|
||||||
for (ClassFile cf : group.getClasses())
|
for (ClassFile cf : group.getClasses())
|
||||||
@@ -172,13 +168,10 @@ public class RenameUnique implements Deobfuscator
|
|||||||
|
|
||||||
map.map(field.getPoolField(), "field" + i++);
|
map.map(field.getPoolField(), "field" + i++);
|
||||||
}
|
}
|
||||||
|
|
||||||
return map;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private NameMappings generateMethodNames(ClassGroup group)
|
private void generateMethodNames(NameMappings map, ClassGroup group)
|
||||||
{
|
{
|
||||||
NameMappings map = new NameMappings();
|
|
||||||
int i = 0;
|
int i = 0;
|
||||||
|
|
||||||
for (ClassFile cf : group.getClasses())
|
for (ClassFile cf : group.getClasses())
|
||||||
@@ -199,8 +192,6 @@ public class RenameUnique implements Deobfuscator
|
|||||||
for (Method m : virtualMethods)
|
for (Method m : virtualMethods)
|
||||||
map.map(m.getPoolMethod(), name);
|
map.map(m.getPoolMethod(), name);
|
||||||
}
|
}
|
||||||
|
|
||||||
return map;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private void regeneratePool(ClassGroup group)
|
private void regeneratePool(ClassGroup group)
|
||||||
@@ -222,22 +213,14 @@ public class RenameUnique implements Deobfuscator
|
|||||||
group.buildClassGraph();
|
group.buildClassGraph();
|
||||||
group.lookup();
|
group.lookup();
|
||||||
|
|
||||||
NameMappings mappings = this.generateClassNames(group);
|
NameMappings mappings = new NameMappings();
|
||||||
|
|
||||||
|
this.generateClassNames(mappings, group);
|
||||||
|
this.generatFieldNames(mappings, group);
|
||||||
|
this.generateMethodNames(mappings, group);
|
||||||
|
|
||||||
int classes = 0, fields = 0, methods = 0;
|
int classes = 0, fields = 0, methods = 0;
|
||||||
|
|
||||||
for (ClassFile cf : group.getClasses())
|
|
||||||
{
|
|
||||||
String newName = mappings.get(cf.getPoolClass());
|
|
||||||
if (newName == null)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
renameClass(group, cf, newName);
|
|
||||||
++classes;
|
|
||||||
}
|
|
||||||
|
|
||||||
mappings = this.generatFieldNames(group);
|
|
||||||
|
|
||||||
// rename fields
|
// rename fields
|
||||||
for (ClassFile cf : group.getClasses())
|
for (ClassFile cf : group.getClasses())
|
||||||
for (Field field : cf.getFields().getFields())
|
for (Field field : cf.getFields().getFields())
|
||||||
@@ -250,8 +233,6 @@ public class RenameUnique implements Deobfuscator
|
|||||||
++fields;
|
++fields;
|
||||||
}
|
}
|
||||||
|
|
||||||
mappings = this.generateMethodNames(group);
|
|
||||||
|
|
||||||
// rename methods
|
// rename methods
|
||||||
for (ClassFile cf : group.getClasses())
|
for (ClassFile cf : group.getClasses())
|
||||||
for (Method method : cf.getMethods().getMethods())
|
for (Method method : cf.getMethods().getMethods())
|
||||||
@@ -268,6 +249,16 @@ public class RenameUnique implements Deobfuscator
|
|||||||
methods += virtualMethods.size();
|
methods += virtualMethods.size();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for (ClassFile cf : group.getClasses())
|
||||||
|
{
|
||||||
|
String newName = mappings.get(cf.getPoolClass());
|
||||||
|
if (newName == null)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
renameClass(group, cf, newName);
|
||||||
|
++classes;
|
||||||
|
}
|
||||||
|
|
||||||
this.regeneratePool(group);
|
this.regeneratePool(group);
|
||||||
|
|
||||||
System.out.println("Uniquely renamed " + classes + " classes, " + fields + " fields, and " + methods + " methods");
|
System.out.println("Uniquely renamed " + classes + " classes, " + fields + " fields, and " + methods + " methods");
|
||||||
|
|||||||
Reference in New Issue
Block a user