public void GetVariables_ViaWordlistType_Parse()
{
DataLine dataLine = new DataLine("good,day", wordlistType);
List <StringVariable> variables = dataLine.GetVariables();
Assert.Equal(2, variables.Count);
Assert.Equal("ONE", variables[0].Name);
Assert.Equal("TWO", variables[1].Name);
Assert.Equal("good", variables[0].AsString());
Assert.Equal("day", variables[1].AsString());
}
public async Task Run()
{
// Build the C# script if not in CSharp mode
if (config.Mode != ConfigMode.CSharp)
{
config.CSharpScript = config.Mode == ConfigMode.Stack
? Stack2CSharpTranspiler.Transpile(config.Stack, config.Settings)
: Loli2CSharpTranspiler.Transpile(config.LoliCodeScript, config.Settings);
}
if (options.UseProxy && !options.TestProxy.Contains(':'))
{
throw new InvalidProxyException(options.TestProxy);
}
if (!options.PersistLog)
{
logger.Clear();
}
// Close any previously opened browser
if (lastBrowser != null)
{
await lastBrowser.CloseAsync();
}
options.Variables.Clear();
IsRunning = true;
cts = new CancellationTokenSource();
var sw = new Stopwatch();
var wordlistType = RuriLibSettings.Environment.WordlistTypes.First(w => w.Name == options.WordlistType);
var dataLine = new DataLine(options.TestData, wordlistType);
var proxy = options.UseProxy ? Proxy.Parse(options.TestProxy, options.ProxyType) : null;
var providers = new Bots.Providers(RuriLibSettings)
{
RNG = RNGProvider
};
if (!RuriLibSettings.RuriLibSettings.GeneralSettings.UseCustomUserAgentsList)
{
providers.RandomUA = RandomUAProvider;
}
// Build the BotData
var data = new BotData(providers, config.Settings, logger, dataLine, proxy, options.UseProxy)
{
CancellationToken = cts.Token
};
using var httpClient = new HttpClient();
data.SetObject("httpClient", httpClient);
var runtime = Python.CreateRuntime();
var pyengine = runtime.GetEngine("py");
var pco = (PythonCompilerOptions)pyengine.GetCompilerOptions();
pco.Module &= ~ModuleOptions.Optimized;
data.SetObject("ironPyEngine", pyengine);
data.AsyncLocker = new();
dynamic globals = new ExpandoObject();
var script = new ScriptBuilder()
.Build(config.CSharpScript, config.Settings.ScriptSettings, PluginRepo);
logger.Log($"Sliced {dataLine.Data} into:");
foreach (var slice in dataLine.GetVariables())
{
var sliceValue = data.ConfigSettings.DataSettings.UrlEncodeDataAfterSlicing
? Uri.EscapeDataString(slice.AsString())
: slice.AsString();
logger.Log($"{slice.Name}: {sliceValue}");
}
// Initialize resources
Dictionary <string, ConfigResource> resources = new();
// Resources will need to be disposed of
foreach (var opt in config.Settings.DataSettings.Resources)
{
try
{
resources[opt.Name] = opt switch
{
LinesFromFileResourceOptions x => new LinesFromFileResource(x),
RandomLinesFromFileResourceOptions x => new RandomLinesFromFileResource(x),
_ => throw new NotImplementedException()
};
}
catch
{
logger.Log($"Could not create resource {opt.Name}", LogColors.Tomato);
}
}
// Add resources to global variables
globals.Resources = resources;
var scriptGlobals = new ScriptGlobals(data, globals);
// Set custom inputs
foreach (var input in config.Settings.InputSettings.CustomInputs)
{
(scriptGlobals.input as IDictionary <string, object>).Add(input.VariableName, input.DefaultAnswer);
}
try
{
sw.Start();
Started?.Invoke(this, EventArgs.Empty);
var state = await script.RunAsync(scriptGlobals, null, cts.Token);
foreach (var scriptVar in state.Variables)
{
try
{
var type = DescriptorsRepository.ToVariableType(scriptVar.Type);
if (type.HasValue && !scriptVar.Name.StartsWith("tmp_"))
{
var variable = DescriptorsRepository.ToVariable(scriptVar.Name, scriptVar.Type, scriptVar.Value);
variable.MarkedForCapture = data.MarkedForCapture.Contains(scriptVar.Name);
options.Variables.Add(variable);
}
}
catch
{
// The type is not supported, e.g. it was generated using custom C# code and not blocks
// so we just disregard it
}
}
}
catch (OperationCanceledException)
{
data.STATUS = "ERROR";
logger.Log($"Operation canceled", LogColors.Tomato);
}
catch (Exception ex)
{
data.STATUS = "ERROR";
var logErrorMessage = RuriLibSettings.RuriLibSettings.GeneralSettings.VerboseMode
? ex.ToString()
: ex.Message;
logger.Log($"[{data.ExecutionInfo}] {ex.GetType().Name}: {logErrorMessage}", LogColors.Tomato);
IsRunning = false;
throw;
}
finally
{
sw.Stop();
logger.Log($"BOT ENDED AFTER {sw.ElapsedMilliseconds} ms WITH STATUS: {data.STATUS}");
// Save the browser for later use
lastBrowser = data.TryGetObject <Browser>("puppeteer");
// Dispose stuff in data.Objects
data.DisposeObjectsExcept(new[] { "puppeteer", "puppeteerPage", "puppeteerFrame" });
// Dispose resources
foreach (var resource in resources.Where(r => r.Value is IDisposable)
.Select(r => r.Value).Cast <IDisposable>())
{
resource.Dispose();
}
data.AsyncLocker.Dispose();
}
IsRunning = false;
Stopped?.Invoke(this, EventArgs.Empty);
}
private async Task Run()
{
try
{
// Build the C# script if not in CSharp mode
if (Config.Mode != ConfigMode.CSharp)
{
Config.CSharpScript = Config.Mode == ConfigMode.Stack
? Stack2CSharpTranspiler.Transpile(Config.Stack, Config.Settings)
: Loli2CSharpTranspiler.Transpile(Config.LoliCodeScript, Config.Settings);
}
}
catch (Exception ex)
{
await js.AlertException(ex);
}
if (options.UseProxy && !options.TestProxy.Contains(':'))
{
await js.AlertError(Loc["InvalidProxy"], Loc["InvalidProxyMessage"]);
return;
}
if (!options.PersistLog)
{
logger.Clear();
}
// Close any previously opened browser
if (lastBrowser != null)
{
await lastBrowser.CloseAsync();
}
options.Variables.Clear();
isRunning = true;
cts = new CancellationTokenSource();
var sw = new Stopwatch();
var wordlistType = RuriLibSettings.Environment.WordlistTypes.First(w => w.Name == options.WordlistType);
var dataLine = new DataLine(options.TestData, wordlistType);
var proxy = options.UseProxy ? Proxy.Parse(options.TestProxy, options.ProxyType) : null;
var providers = new Providers(RuriLibSettings)
{
RandomUA = RandomUAProvider,
RNG = RNGProvider
};
// Build the BotData
var data = new BotData(providers, Config.Settings, logger, dataLine, proxy, options.UseProxy);
data.CancellationToken = cts.Token;
data.Objects.Add("httpClient", new HttpClient());
var runtime = Python.CreateRuntime();
var pyengine = runtime.GetEngine("py");
var pco = (PythonCompilerOptions)pyengine.GetCompilerOptions();
pco.Module &= ~ModuleOptions.Optimized;
data.Objects.Add("ironPyEngine", pyengine);
var script = new ScriptBuilder()
.Build(Config.CSharpScript, Config.Settings.ScriptSettings, PluginRepo);
logger.Log($"Sliced {dataLine.Data} into:");
foreach (var slice in dataLine.GetVariables())
{
var sliceValue = data.ConfigSettings.DataSettings.UrlEncodeDataAfterSlicing
? Uri.EscapeDataString(slice.AsString())
: slice.AsString();
logger.Log($"{slice.Name}: {sliceValue}");
}
logger.NewEntry += OnNewEntry;
try
{
var scriptGlobals = new ScriptGlobals(data, new ExpandoObject());
foreach (var input in Config.Settings.InputSettings.CustomInputs)
{
(scriptGlobals.input as IDictionary <string, object>).Add(input.VariableName, input.DefaultAnswer);
}
sw.Start();
var state = await script.RunAsync(scriptGlobals, null, cts.Token);
foreach (var scriptVar in state.Variables)
{
try
{
var type = DescriptorsRepository.ToVariableType(scriptVar.Type);
if (type.HasValue && !scriptVar.Name.StartsWith("tmp_"))
{
var variable = DescriptorsRepository.ToVariable(scriptVar.Name, scriptVar.Type, scriptVar.Value);
variable.MarkedForCapture = data.MarkedForCapture.Contains(scriptVar.Name);
options.Variables.Add(variable);
}
}
catch
{
// The type is not supported, e.g. it was generated using custom C# code and not blocks
// so we just disregard it
}
}
}
catch (OperationCanceledException)
{
data.STATUS = "ERROR";
logger.Log($"Operation canceled", LogColors.Tomato);
}
catch (Exception ex)
{
data.STATUS = "ERROR";
logger.Log($"[{data.ExecutionInfo}] {ex.GetType().Name}: {ex.Message}", LogColors.Tomato);
await js.AlertError(ex.GetType().Name, $"[{data.ExecutionInfo}] {ex.Message}");
}
finally
{
sw.Stop();
isRunning = false;
logger.Log($"BOT ENDED AFTER {sw.ElapsedMilliseconds} ms WITH STATUS: {data.STATUS}");
// Save the browser for later use
lastBrowser = data.Objects.ContainsKey("puppeteer") && data.Objects["puppeteer"] is Browser currentBrowser
? currentBrowser
: null;
// Dispose all disposable objects
foreach (var obj in data.Objects.Where(o => o.Value is IDisposable))
{
if (obj.Key.Contains("puppeteer"))
{
continue;
}
try
{
(obj.Value as IDisposable).Dispose();
}
catch
{
}
}
}
await loggerViewer?.Refresh();
variablesViewer?.Refresh();
await InvokeAsync(StateHasChanged);
}