public static ExecutionResultWrapper Execute(Stream assemblyStream, RuntimeGuardToken guardToken)
{
try {
Console.SetOut(Output.Writer);
var assembly = Assembly.Load(ReadAllBytes(assemblyStream));
var main = assembly.EntryPoint;
using (guardToken.Scope(NewRuntimeGuardSettings())) {
var args = main.GetParameters().Length > 0 ? new object[] { new string[0] } : null;
var result = main.Invoke(null, args);
if (main.ReturnType != typeof(void))
{
result.Inspect("Return");
}
return(new ExecutionResultWrapper(new ExecutionResult(Output.Stream, Flow.Steps), null));
}
}
catch (Exception ex) {
if (ex is TargetInvocationException invocationEx)
{
ex = invocationEx.InnerException;
}
Flow.ReportException(ex);
ex.Inspect("Exception");
return(new ExecutionResultWrapper(new ExecutionResult(Output.Stream, Flow.Steps), ex));
}
}
public unsafe ExecutionResult Execute(Assembly assembly, RuntimeGuardToken guardToken, IWorkSession session)
{
try {
Output.Reset();
Flow.Reset();
Console.SetOut(Output.Writer);
var main = assembly.EntryPoint;
if (main == null)
{
throw new ArgumentException("Entry point not found in " + assembly, nameof(assembly));
}
using (guardToken.Scope(NewRuntimeGuardSettings())) {
var args = main.GetParameters().Length > 0 ? new object[] { new string[0] } : null;
PerformanceLog.Checkpoint("Executor.Invoke.Start");
var result = main.Invoke(null, args);
PerformanceLog.Checkpoint("Executor.Invoke.End");
if (main.ReturnType != typeof(void))
{
result.Inspect("Return");
}
return(new ExecutionResult(Output.Stream, Flow.Steps));
}
}
catch (Exception ex) {
PerformanceLog.Checkpoint("Executor.Invoke.Exception");
if (ex is TargetInvocationException invocationEx)
{
ex = invocationEx.InnerException ?? ex;
}
if (ex is RegexMatchTimeoutException)
{
ex = new TimeGuardException("Time limit reached while evaluating a Regex.\r\nNote that timeout was added by SharpLab — in real code this would not throw, but might run for a very long time.", ex);
}
if (ex is StackGuardException sgex)
{
throw new Exception($"{sgex.Message} {sgex.StackBaseline} {sgex.StackOffset} {sgex.StackLimit} {sgex.StackSize}");
}
Flow.ReportException(ex);
Output.Write(new SimpleInspection("Exception", ex.ToString()));
if (ShouldMonitorException(ex))
{
_monitor.Exception(ex !, session);
}
return(new ExecutionResult(Output.Stream, Flow.Steps));
}
}
private static object?RemoteRun(MemoryStream assemblyStream, RuntimeGuardToken token)
{
using (token.Scope()) {
var assembly = Assembly.Load(assemblyStream.ToArray());
var type = assembly.GetType("Program", true);
var method = type.GetMethod("Run", BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static);
if (method == null)
{
throw new NotSupportedException("Static method 'Run' not found on type 'Program'.");
}
var result = method.Invoke(null, null);
if (result?.GetType().Assembly == assembly || (result is MemberInfo m && m.Module.Assembly == assembly))
{
throw new Exception("Result returned by Program.Run must not belong to the user assembly.");
}
return(result);
}
}
public static unsafe ExecutionResultWrapper Execute(Stream assemblyStream, RuntimeGuardToken guardToken, int processId)
{
try {
Console.SetOut(Output.Writer);
InspectionSettings.CurrentProcessId = processId;
var assembly = Assembly.Load(ReadAllBytes(assemblyStream));
var main = assembly.EntryPoint;
using (guardToken.Scope(NewRuntimeGuardSettings())) {
var args = main.GetParameters().Length > 0 ? new object[] { new string[0] } : null;
byte *stackStart = stackalloc byte[1];
InspectionSettings.StackStart = (ulong)stackStart;
var result = main.Invoke(null, args);
if (main.ReturnType != typeof(void))
{
result.Inspect("Return");
}
return(new ExecutionResultWrapper(new ExecutionResult(Output.Stream, Flow.Steps), null));
}
}
catch (Exception ex) {
if (ex is TargetInvocationException invocationEx)
{
ex = invocationEx.InnerException;
}
if (ex is RegexMatchTimeoutException)
{
ex = new TimeGuardException("Time limit reached while evaluating a Regex.\r\nNote that timeout was added by SharpLab — in real code this would not throw, but might run for a very long time.", ex);
}
Flow.ReportException(ex);
ex.Inspect("Exception");
return(new ExecutionResultWrapper(new ExecutionResult(Output.Stream, Flow.Steps), ex));
}
}
