Output: Organize nested types correctly

This commit is contained in:
Katy Coe
2019-10-31 16:11:49 +01:00
parent 70a541c36c
commit edbf9dc2fa

View File

@@ -19,15 +19,18 @@ namespace Il2CppInspector
public Il2CppCSharpDumper(Il2CppModel model) => this.model = model;
private StreamWriter writer;
private string formatAddress(ulong address) => model.Package.BinaryImage.Bits == 32
? string.Format($"0x{(uint) address:X8}")
: string.Format($"0x{address:X16}");
public void WriteFile(string outFile) {
using (var writer = new StreamWriter(new FileStream(outFile, FileMode.Create), Encoding.UTF8)) {
using (writer = new StreamWriter(new FileStream(outFile, FileMode.Create), Encoding.UTF8)) {
foreach (var asm in model.Assemblies) {
writer.Write($"// Image {asm.Index}: {asm.FullName} - {asm.Definition.typeStart}\n");
}
writer.Write("\n");
foreach (var type in model.Assemblies.SelectMany(x => x.DefinedTypes)) {
@@ -35,13 +38,28 @@ namespace Il2CppInspector
if (ExcludedNamespaces?.Any(x => x == type.Namespace || type.Namespace.StartsWith(x + ".")) ?? false)
continue;
// Type declaration
writer.Write($"\n// Namespace: {type.Namespace}\n");
// Assembly.DefinedTypes returns nested types in the assembly by design - ignore them
if (!type.IsNested) {
writeType(type);
writer.Write("\n");
}
}
}
}
private void writeType(TypeInfo type, string prefix = "") {
// Only print namespace if we're not nested
if (!type.IsNested)
writer.Write($"{prefix}// Namespace: {type.Namespace}\n");
// Type declaration
if (type.IsImport)
writer.Write("[ComImport]");
writer.Write(prefix + "[ComImport]\n");
if (type.IsSerializable)
writer.Write("[Serializable]\n");
writer.Write(prefix + "[Serializable]\n");
writer.Write(prefix);
if (type.IsPublic || type.IsNestedPublic)
writer.Write("public ");
if (type.IsNestedPrivate)
@@ -72,7 +90,7 @@ namespace Il2CppInspector
writer.Write($"{param.ParameterType.CSharpName} {param.Name}");
}
writer.Write($"); // TypeDefIndex: {type.Index}; {formatAddress(del.VirtualAddress)}\n");
continue;
return;
}
// An abstract sealed class is a static class
@@ -100,17 +118,17 @@ namespace Il2CppInspector
@base.Insert(0, type.ElementType.CSharpName);
var baseText = @base.Count > 0 ? " : " + string.Join(", ", @base) : string.Empty;
writer.Write($"{type.Name}{baseText} // TypeDefIndex: {type.Index}\n{{\n");
writer.Write($"{type.Name}{baseText} // TypeDefIndex: {type.Index}\n" + prefix + "{\n");
// Fields
if (!type.IsEnum) {
if (type.DeclaredFields.Count > 0)
writer.Write("\t// Fields\n");
if (type.DeclaredFields.Any())
writer.Write(prefix + "\t// Fields\n");
foreach (var field in type.DeclaredFields) {
writer.Write("\t");
writer.Write(prefix + "\t");
if (field.IsNotSerialized)
writer.Write("[NonSerialized]\t");
writer.Write("[NonSerialized]\n" + prefix + "\t");
if (field.IsPrivate)
writer.Write("private ");
@@ -140,27 +158,27 @@ namespace Il2CppInspector
// Don't output field indices for const fields (they don't have any storage)
if (!field.IsLiteral)
writer.Write(" // 0x{0:X2}", (uint)field.Offset);
writer.WriteLine("");
writer.Write("\n");
}
if (type.DeclaredFields.Count > 0)
if (type.DeclaredFields.Any())
writer.Write("\n");
}
// Enumeration
else {
writer.Write(string.Join(",\n", type.GetEnumNames().Zip(type.GetEnumValues().OfType<object>(),
(k, v) => new { k, v }).OrderBy(x => x.v).Select(x => $"\t{x.k} = {x.v}")) + "\n");
(k, v) => new { k, v }).OrderBy(x => x.v).Select(x => $"{prefix}\t{x.k} = {x.v}")) + "\n");
}
var usedMethods = new List<Reflection.MethodInfo>();
var usedMethods = new List<MethodInfo>();
// Properties
if (type.DeclaredProperties.Count > 0)
writer.Write("\t// Properties\n");
if (type.DeclaredProperties.Any())
writer.Write(prefix + "\t// Properties\n");
foreach (var prop in type.DeclaredProperties) {
string modifiers = prop.GetMethod?.GetModifierString() ?? prop.SetMethod.GetModifierString();
writer.Write($"\t{modifiers}{prop.PropertyType.CSharpName} {prop.Name} {{ ");
writer.Write($"{prefix}\t{modifiers}{prop.PropertyType.CSharpName} {prop.Name} {{ ");
writer.Write((prop.GetMethod != null ? "get; " : "") + (prop.SetMethod != null ? "set; " : "") + "}");
if ((prop.GetMethod != null && prop.GetMethod.VirtualAddress != 0) || (prop.SetMethod != null && prop.SetMethod.VirtualAddress != 0))
writer.Write(" // ");
@@ -169,34 +187,43 @@ namespace Il2CppInspector
usedMethods.Add(prop.GetMethod);
usedMethods.Add(prop.SetMethod);
}
if (type.DeclaredProperties.Count > 0)
if (type.DeclaredProperties.Any())
writer.Write("\n");
// Events
if (type.DeclaredEvents.Count > 0)
writer.Write("\t// Events\n");
if (type.DeclaredEvents.Any())
writer.Write(prefix + "\t// Events\n");
foreach (var evt in type.DeclaredEvents) {
string modifiers = evt.AddMethod?.GetModifierString();
writer.Write($"\t{modifiers}event {evt.EventHandlerType.CSharpName} {evt.Name} {{\n");
writer.Write($"{prefix}\t{modifiers}event {evt.EventHandlerType.CSharpName} {evt.Name} {{\n");
var m = new Dictionary<string, ulong>();
if (evt.AddMethod != null) m.Add("add", evt.AddMethod.VirtualAddress);
if (evt.RemoveMethod != null) m.Add("remove", evt.RemoveMethod.VirtualAddress);
if (evt.RaiseMethod != null) m.Add("raise", evt.RaiseMethod.VirtualAddress);
writer.Write(string.Join("\n", m.Select(x => $"\t\t{x.Key}; // {formatAddress(x.Value)}")) + "\n\t}\n");
writer.Write(string.Join("\n", m.Select(x => $"{prefix}\t\t{x.Key}; // {formatAddress(x.Value)}")) + "\n" + prefix + "\t}\n");
usedMethods.Add(evt.AddMethod);
usedMethods.Add(evt.RemoveMethod);
usedMethods.Add(evt.RaiseMethod);
}
if (type.DeclaredEvents.Count > 0)
if (type.DeclaredEvents.Any())
writer.Write("\n");
// Nested types
if (type.DeclaredNestedTypes.Any())
writer.Write(prefix + "\t// Nested types\n");
foreach (var nestedType in type.DeclaredNestedTypes) {
writeType(nestedType, prefix + "\t");
writer.Write("\n");
}
// Constructors
if (type.DeclaredConstructors.Any())
writer.Write("\t// Constructors\n");
writer.Write(prefix + "\t// Constructors\n");
foreach (var method in type.DeclaredConstructors) {
writer.Write($"\t{method.GetModifierString()}{method.DeclaringType.Name}(");
writer.Write($"{prefix}\t{method.GetModifierString()}{method.DeclaringType.Name}(");
writer.Write(getParametersString(method.DeclaredParameters));
writer.Write(");" + (method.VirtualAddress != 0 ? $" // {formatAddress(method.VirtualAddress)}" : "") + "\n");
}
@@ -205,11 +232,11 @@ namespace Il2CppInspector
// Methods
if (type.DeclaredMethods.Except(usedMethods).Any())
writer.Write("\t// Methods\n");
writer.Write(prefix + "\t// Methods\n");
// Don't re-output methods for constructors, properties, events etc.
foreach (var method in type.DeclaredMethods.Except(usedMethods)) {
writer.Write($"\t{method.GetModifierString()}");
writer.Write($"{prefix}\t{method.GetModifierString()}");
if (method.Name != "op_Implicit" && method.Name != "op_Explicit")
writer.Write($"{method.ReturnType.CSharpName} {method.CSharpName}");
else
@@ -217,11 +244,10 @@ namespace Il2CppInspector
writer.Write("(" + getParametersString(method.DeclaredParameters));
writer.Write(");" + (method.VirtualAddress != 0 ? $" // {formatAddress(method.VirtualAddress)}" : "") + "\n");
}
writer.Write("}\n");
}
}
writer.Write(prefix + "}\n");
}
// TODO: Move this info ParameterInfo
private string getParametersString(List<ParameterInfo> @params) {
StringBuilder sb = new StringBuilder();