1
0
mirror of synced 2024-12-14 08:42:54 +01:00
ImHex/plugins/script_loader/dotnet/AssemblyLoader/Program.cs

71 lines
2.0 KiB
C#
Raw Normal View History

using System.Reflection;
using System.Runtime.InteropServices;
using System.Runtime.Loader;
namespace ImHex
{
public class EntryPoint
{
public static int ExecuteScript(IntPtr arg, int argLength)
{
try
{
return ExecuteScript(Marshal.PtrToStringUTF8(arg, argLength)) ? 0 : 1;
}
catch (Exception e)
{
Console.WriteLine("[.NET Script] Exception in AssemblyLoader: " + e.Message);
return 1;
}
}
private static bool ExecuteScript(string path)
{
AssemblyLoadContext? context = new("ScriptDomain_" + Path.GetFileNameWithoutExtension(path), true);
try
{
var assembly = context.LoadFromStream(new MemoryStream(File.ReadAllBytes(path)));
var entryPointType = assembly.GetType("Script");
if (entryPointType == null)
{
Console.WriteLine("[.NET Script] Failed to find Script type");
return false;
}
var entryPointMethod = entryPointType.GetMethod("Main", BindingFlags.Static | BindingFlags.Public);
if (entryPointMethod == null)
{
Console.WriteLine("[.NET Script] Failed to find ScriptMain method");
return false;
}
entryPointMethod.Invoke(null, null);
}
catch (Exception e)
{
Console.WriteLine("[.NET Script] Exception in AssemblyLoader: " + e.Message);
return false;
}
finally
{
context.Unload();
context = null;
for (int i = 0; i < 10; i++)
{
GC.Collect();
GC.WaitForPendingFinalizers();
}
}
return true;
}
}
}