You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.4 KiB
116 lines
3.4 KiB
using Mono.Cecil;
|
|
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
|
|
namespace IPA.Patcher
|
|
{
|
|
class VirtualizedModule
|
|
{
|
|
private readonly FileInfo _file;
|
|
private ModuleDefinition _module;
|
|
|
|
public static VirtualizedModule Load(string engineFile)
|
|
{
|
|
return new VirtualizedModule(engineFile);
|
|
}
|
|
|
|
private VirtualizedModule(string assemblyFile)
|
|
{
|
|
_file = new FileInfo(assemblyFile);
|
|
|
|
LoadModules();
|
|
}
|
|
|
|
private void LoadModules()
|
|
{
|
|
var resolver = new DefaultAssemblyResolver();
|
|
resolver.AddSearchDirectory(_file.DirectoryName);
|
|
|
|
var parameters = new ReaderParameters
|
|
{
|
|
AssemblyResolver = resolver,
|
|
};
|
|
|
|
_module = ModuleDefinition.ReadModule(_file.FullName, parameters);
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
public void Virtualize()
|
|
{
|
|
|
|
foreach (var type in _module.Types)
|
|
{
|
|
VirtualizeType(type);
|
|
}
|
|
Console.WriteLine();
|
|
|
|
_module.Write(_file.FullName);
|
|
}
|
|
|
|
private void VirtualizeType(TypeDefinition type)
|
|
{
|
|
if(type.IsSealed)
|
|
{
|
|
// Unseal
|
|
type.IsSealed = false;
|
|
}
|
|
|
|
if (type.IsInterface) return;
|
|
if (type.IsAbstract) return;
|
|
|
|
// These two don't seem to work.
|
|
if (type.Name == "SceneControl" || type.Name == "ConfigUI") return;
|
|
|
|
//Console.CursorTop--;
|
|
Console.CursorLeft = 0;
|
|
Program.ClearLine();
|
|
Console.Write("Virtualizing {0}", type.Name);
|
|
// Take care of sub types
|
|
foreach (var subType in type.NestedTypes)
|
|
{
|
|
VirtualizeType(subType);
|
|
}
|
|
|
|
foreach (var method in type.Methods)
|
|
{
|
|
if (method.IsManaged
|
|
&& method.IsIL
|
|
&& !method.IsStatic
|
|
&& !method.IsVirtual
|
|
&& !method.IsAbstract
|
|
&& !method.IsAddOn
|
|
&& !method.IsConstructor
|
|
&& !method.IsSpecialName
|
|
&& !method.IsGenericInstance
|
|
&& !method.HasOverrides)
|
|
{
|
|
method.IsVirtual = true;
|
|
method.IsPublic = true;
|
|
method.IsPrivate = false;
|
|
method.IsNewSlot = true;
|
|
method.IsHideBySig = true;
|
|
}
|
|
}
|
|
|
|
foreach (var field in type.Fields)
|
|
{
|
|
if (field.IsPrivate) field.IsFamily = true;
|
|
}
|
|
}
|
|
|
|
public bool IsVirtualized
|
|
{
|
|
get
|
|
{
|
|
var awakeMethods = _module.GetTypes().SelectMany(t => t.Methods.Where(m => m.Name == "Awake"));
|
|
var methodDefinitions = awakeMethods as MethodDefinition[] ?? awakeMethods.ToArray();
|
|
if (!methodDefinitions.Any()) return false;
|
|
|
|
return ((float)methodDefinitions.Count(m => m.IsVirtual) / methodDefinitions.Count()) > 0.5f;
|
|
}
|
|
}
|
|
}
|
|
}
|