chizui/Il2CppInspectorRedux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Output: Tidy up Il2CppIDAScriptDumper

Browse Source
This commit is contained in:
Katy Coe
2020-01-26 22:08:07 +01:00
parent e75f3931fa
commit 90f563f8d2
1 changed files with 27 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
Il2CppDumper/Il2CppIDAScriptDumper.cs
View File

@@ -13,28 +13,27 @@ namespace Il2CppInspector
{ {
public class Il2CppIDAScriptDumper public class Il2CppIDAScriptDumper
{ {
private Il2CppModel model; private readonly Il2CppModel model;
private StreamWriter writer;
public Il2CppIDAScriptDumper(Il2CppModel model) => this.model = model; public Il2CppIDAScriptDumper(Il2CppModel model) => this.model = model;
#region Writing
public void WriteScriptToFile(string outputFile) { public void WriteScriptToFile(string outputFile) {
using (var fs = new FileStream(outputFile, FileMode.Create)) using var fs = new FileStream(outputFile, FileMode.Create);
using (var sw = new StreamWriter(fs, Encoding.UTF8)) { writer = new StreamWriter(fs, Encoding.UTF8);
writeSectionHeader(sw, "Preamble");
writePreamble(sw);
writeSectionHeader(sw, "Methods"); writeSectionHeader("Preamble");
writeMethods(sw, this.model.Types); writePreamble();
writeSectionHeader(sw, "Usages"); writeSectionHeader("Methods");
writeUsages(sw, this.model); writeMethods();
}
writeSectionHeader( "Usages");
writeUsages();
} }
private static void writePreamble(StreamWriter writer) { private void writePreamble() {
writeLines(writer, writeLines(
@"#encoding: utf-8 @"#encoding: utf-8
import idaapi import idaapi
@@ -55,44 +54,36 @@ index = 1
); );
} }
private static void writeMethods(StreamWriter writer, IEnumerable<TypeInfo> types) { private void writeMethods() {
foreach (var type in types.Where(t => t != null)) { foreach (var type in model.Types.Where(t => t != null)) {
writeMethods(writer, type.Name, type.DeclaredConstructors); writeMethods(type.Name, type.DeclaredConstructors);
writeMethods(writer, type.Name, type.DeclaredMethods); writeMethods(type.Name, type.DeclaredMethods);
} }
} }
private static void writeMethods(StreamWriter writer, string typeName, IEnumerable<MethodBase> methods) { private void writeMethods(string typeName, IEnumerable<MethodBase> methods) {
foreach (var method in methods.Where(m => m.VirtualAddress.HasValue)) { foreach (var method in methods.Where(m => m.VirtualAddress.HasValue)) {
writeLines(writer, writeLines($"SetName({method.VirtualAddress.Value.Start.ToAddressString()}, '{typeName}$${method.Name}')");
$"SetName({method.VirtualAddress.Value.Start.ToAddressString()}, '{typeName}$${method.Name}')"
);
} }
} }
private static void writeUsages(StreamWriter writer, Il2CppModel model) { private void writeUsages() {
foreach (var usage in model.Package.MetadataUsages) { foreach (var usage in model.Package.MetadataUsages) {
switch (usage.Type) { switch (usage.Type) {
case MetadataUsageType.TypeInfo: case MetadataUsageType.TypeInfo:
case MetadataUsageType.Type: case MetadataUsageType.Type:
var type = model.GetTypeFromUsage(usage.SourceIndex); var type = model.GetTypeFromUsage(usage.SourceIndex);
writeLines(writer, writeLines($"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Class${type.Name}')");
$"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Class${type.Name}')"
);
break; break;
case MetadataUsageType.MethodDef: case MetadataUsageType.MethodDef:
var method = model.MethodsByDefinitionIndex[usage.SourceIndex]; var method = model.MethodsByDefinitionIndex[usage.SourceIndex];
writeLines(writer, writeLines($"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Method${method.DeclaringType.Name}.{method.Name}')");
$"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Method${method.DeclaringType.Name}.{method.Name}')"
);
break; break;
case MetadataUsageType.FieldInfo: case MetadataUsageType.FieldInfo:
var field = model.Package.Fields[usage.SourceIndex]; var field = model.Package.Fields[usage.SourceIndex];
type = model.GetTypeFromUsage(field.typeIndex); type = model.GetTypeFromUsage(field.typeIndex);
var fieldName = model.Package.Strings[field.nameIndex]; var fieldName = model.Package.Strings[field.nameIndex];
writeLines(writer, writeLines($"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Field${type.Name}.{fieldName}')");
$"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Field${type.Name}.{fieldName}')"
);
break; break;
case MetadataUsageType.StringLiteral: case MetadataUsageType.StringLiteral:
// TODO: String literals // TODO: String literals
@@ -101,29 +92,23 @@ index = 1
var methodSpec = model.Package.MethodSpecs[usage.SourceIndex]; var methodSpec = model.Package.MethodSpecs[usage.SourceIndex];
method = model.MethodsByDefinitionIndex[methodSpec.methodDefinitionIndex]; method = model.MethodsByDefinitionIndex[methodSpec.methodDefinitionIndex];
type = method.DeclaringType; type = method.DeclaringType;
writeLines(writer, writeLines($"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Method${type.Name}.{method.Name}')");
$"SetName({model.Package.BinaryMetadataUsages[usage.DestinationIndex].ToAddressString()}, 'Method${type.Name}.{method.Name}')"
);
break;
default:
break; break;
} }
} }
} }
private static void writeSectionHeader(StreamWriter writer, string sectionName) { private void writeSectionHeader(string sectionName) {
writeLines(writer, writeLines(
$"# SECTION: {sectionName}", $"# SECTION: {sectionName}",
$"# -----------------------------" $"# -----------------------------"
); );
} }
private static void writeLines(StreamWriter writer, params string[] lines) { private void writeLines(params string[] lines) {
foreach (var line in lines) { foreach (var line in lines) {
writer.WriteLine(line); writer.WriteLine(line);
} }
} }
#endregion
} }
} }