/// <summary>
/// Executes the provided command line as an external process.
/// </summary>
/// <param name="commandLine">The process command line</param>
/// <returns>true if the process terminated successfully (exit code = 0); false otherwise.</returns>
private static bool ExecuteCommand(CommandLine commandLine)
{
ProcessStartInfo info = new ProcessStartInfo
{
CreateNoWindow = false,
UseShellExecute = false,
WindowStyle = ProcessWindowStyle.Hidden,
WorkingDirectory = Path.GetTempPath(),
FileName = commandLine.FileName,
Arguments = commandLine.Arguments,
RedirectStandardError = false,
RedirectStandardInput = false
};
Process process = Process.Start(ProcessStartInfoEx.StartThroughCmdShell(info));
if (process != null)
{
process.WaitForExit(60000);
return(process.ExitCode == 0);
}
return(false);
}
public iProcess(ServiceController s, ProcessStartInfoEx pi)
{
Service = s;
StartInfo = pi;
StoreLocation64 = GetStoreLocation(RegistryView.Registry64);
StoreLocation32 = GetStoreLocation(RegistryView.Registry32);
}
public iProcess(Process p, ProcessStartInfoEx pi)
{
Process = p;
StartInfo = pi;
StoreLocation64 = GetStoreLocation(RegistryView.Registry64);
StoreLocation32 = GetStoreLocation(RegistryView.Registry32);
}
/// <summary>
/// Starts a Process using the provided command line arguments string.
/// </summary>
/// <param name="info">The process start info which will be used to launch the external test program.</param>
/// <returns>The newly spawned debug process.</returns>
private static Process Run(ProcessStartInfo info)
{
// Wrap the process inside cmd.exe in so that we can redirect stdout,
// stderr to file using a similar mechanism as available for Debug runs.
return(Process.Start(ProcessStartInfoEx.StartThroughCmdShell(info.Clone())));
}
/// <summary>
/// Run fuzzer
/// </summary>
/// <param name="action">Action</param>
/// <param name="args">Arguments</param>
public static void Run(Action <Stream> action, FuzzerRunArgs args = null)
{
CoverageHelper.CreateCoverageListener();
if (action == null)
{
throw new NullReferenceException(nameof(action));
}
// Check supervisor
Mutex mutex;
var supervisor = FuzzerRunArgs.SupervisorType.None;
if (args != null && args.Supervisor != FuzzerRunArgs.SupervisorType.None)
{
// Check if is the listener or the task with mutex
mutex = new Mutex(false, "TuringMachine.Supervisor." + Client.PublicName, out var isNew);
if (!isNew)
{
Client.PublicName += $".{Process.GetCurrentProcess().Id}:{args.TaskId}";
}
else
{
Client.PublicName += ".Supervisor";
supervisor = args.Supervisor;
}
}
else
{
mutex = null;
}
if (!Client.IsStarted)
{
// If you want other connection you must call this method before Run
Client.Start(CommandLineOptions.Parse().GetConnection());
}
// Send current files
Client.SendCurrentFiles(new OperationCanceledException(), null, true);
// Fuzz
var cancel = new CancelEventArgs();
var handler = new ConsoleCancelEventHandler((o, s) =>
{
cancel.Cancel = true;
s.Cancel = true;
});
Console.CancelKeyPress += handler;
// Ensure data
while (Client.GetInput() == null || Client.GetConfig() == null)
{
Thread.Sleep(50);
}
switch (supervisor)
{
case FuzzerRunArgs.SupervisorType.RegularSupervisor:
{
var pi = new ProcessStartInfoEx()
{
FileName = "dotnet",
Arguments = string.Join(" ", Environment.GetCommandLineArgs().Select(u => u)),
WindowStyle = ProcessWindowStyle.Normal,
RedirectStandardOutput = true,
RedirectStandardError = true,
UseShellExecute = false,
CreateNoWindow = false,
};
while (!cancel.Cancel)
{
using (var pr = new ProcessEx(pi))
{
pr.WaitForExit();
Exception exception;
switch (pr.ExitCode)
{
case StackOverflowExceptionCode:
{
exception = new StackOverflowException($"Unhandled exception: {pr.ExitCode}");
break;
}
default:
{
exception = new InvalidProgramException($"Unhandled exception: {pr.ExitCode}");
break;
}
}
if (Client.SendCurrentFiles(exception, pr.Output, true) == 0)
{
Client.SendLog(new FuzzerLog()
{
Coverage = CoverageHelper.CurrentCoverage,
InputId = Guid.Empty,
ConfigId = Guid.Empty,
});
}
}
}
break;
}
case FuzzerRunArgs.SupervisorType.None:
{
while (!cancel.Cancel)
{
var input = Client.GetInput();
var config = Client.GetConfig();
string currentStreamPath = null;
FileStream storeCurrentStream = null;
if (args?.StoreCurrent == true)
{
// Free current stream
currentStreamPath = $"{input.Id}.{config.Id}.{Process.GetCurrentProcess().Id}.{args.TaskId}.current";
storeCurrentStream = new FileStream(currentStreamPath, FileMode.Create, FileAccess.ReadWrite, FileShare.Read);
}
using (var stream = new FuzzingStream(config, input, storeCurrentStream)
{
ExtraLogInformation = "TaskId: " + args.TaskId
})
{
var log = Client.Execute(action, stream);
if (log != null)
{
Client.SendLog(log);
args?.OnLog?.Invoke(log, cancel);
}
}
if (storeCurrentStream != null)
{
// Delete current stream
storeCurrentStream.Close();
storeCurrentStream.Dispose();
File.Delete(currentStreamPath);
}
}
break;
}
}
Console.CancelKeyPress -= handler;
mutex?.Dispose();
}