static int Main(string[] args)
{
try
{
// Disable fatal error message popups by SetErrorMode(SEM_FAILCRITICALERRORS)
try
{
SetErrorMode(1);
}
catch (System.EntryPointNotFoundException)
{
}
AppDomainLoader.progress("Main: Starting");
AppDomain.CurrentDomain.UnhandledException += (a, b) => { AppDomainLoader.progress("Main: !!!" + b.IsTerminating); };
int exitCode = AppDomainLoader.Loader(args) ?? MainWithoutContext(args);
AppDomainLoader.progress("Main: Exiting with exit code " + exitCode);
return(exitCode);
}
catch (System.Security.SecurityException)
{
Console.Error.WriteLine("Code security exception. Please copy XSharper to a local hard drive and try again.");
return(-3);
}
}
static public int MainWithoutContext(string[] args)
{
AppDomainLoader.progress("MainWithoutContext: Entering");
ScriptContext context = new ScriptContext(AppDomainLoader.ResourceAssembly);
return(MainWithContext(context, args));
}
private static int restartAsAdmin(ScriptContext context, string[] args, bool hidden)
{
context.WriteLine(OutputType.Info, "** Administrative privileges are required to run this script.\n** Please confirm to continue.");
if (context.IsSet(xs.testElevation))
{
List <string> f = new List <string>();
foreach (string s in args)
{
if (string.Compare(s, xs.testElevation.Replace("xs.", "//"), StringComparison.OrdinalIgnoreCase) != 0)
{
f.Add(s);
}
}
args = f.ToArray();
}
AppDomainLoader.progress("MainWithContext: About to restart with CmdLine=" + string.Join(" ", args));
int n = AppDomainLoader.RunWithElevatedContext(
delegate(ScriptContext ctx)
{
ScriptContextScope.DefaultContext = ctx;
try
{
if (!ctx.IsAdministrator)
{
throw new ScriptRuntimeException("Administrator privileges are required");
}
return(MainWithContext(ctx, args));
}
finally
{
ScriptContextScope.DefaultContext = null;
}
}, hidden);
if (n == -1)
{
throw new ScriptRuntimeException("An error occured while granting administrative privileges.");
}
if (n != 0)
{
throw new ScriptRuntimeException("An error occured during script execution.");
}
return(0);
}
private static Script execGenerator(ScriptContext context, string c, string prefix, string suffix)
{
Script script = createEmptyScript(context, "generated");
if (context.Compiler.DefaultNETVersion >= new Version(3, 5))
{
// Yeah, I know it's deprecated, and it's best to use full assembly names, but
// but I don't really want to scan GAC myself
#pragma warning disable 618
context.AddAssembly(Assembly.LoadWithPartialName("System.Xml.Linq"), false);
context.AddAssembly(Assembly.LoadWithPartialName("System.Core"), false);
#pragma warning restore 618
}
script.Add(new Rem {
Text = "----- Generated from command line -------" + Environment.NewLine + "\t" + c + Environment.NewLine + "\t ------------"
});
script.Parameters.Add(new CommandLineParameter("arg", CommandLineValueCount.Multiple)
{
Last = true, Default = "", Required = false
});
script.UnknownSwitches = true;
c = c.Trim();
if (c.StartsWith("\"", StringComparison.Ordinal) && c.EndsWith("\"", StringComparison.Ordinal))
{
c = c.Substring(1, c.Length - 2);
}
if (c == "-")
{
c = Console.In.ReadToEnd();
}
if (c.StartsWith("="))
{
prefix = "=" + prefix;
c = "${=" + c.Substring(1) + "}";
}
if (prefix != null || suffix != null)
{
c = prefix + Environment.NewLine + c + Environment.NewLine + suffix;
}
c = c.Replace("`", "\"");
AppDomainLoader.progress("Read script begin");
using (new ScriptContextScope(context))
script.Load(c);
AppDomainLoader.progress("Read script end");
return(script);
}
private static Script loadScript(string location, ScriptContext context)
{
AppDomainLoader.progress("MainWithContext: Parsing XML");
try
{
byte[] data;
bool validate = context.GetBool(xs.validate, false);
if (location == ".")
{
location = loadFromConfig(out data);
return(context.LoadScript(new MemoryStream(data), location, validate));
}
else
{
return(context.LoadScript(location, validate));
}
}
finally
{
AppDomainLoader.progress("MainWithContext: Parsing completed");
}
}
static int MainWithContext(ScriptContext context, string[] args)
{
CommandLineParameters xsParams = getXsParams();
int exitCode = 0;
ConsoleRedirector redir = null;
AppDomainLoader.progress("MainWithContext: Entering --------------------");
bool utf8 = false;
foreach (string arg in args)
{
if (arg.Equals(xs.utf8.Replace("xs.", "//"), StringComparison.OrdinalIgnoreCase))
{
utf8 = true;
}
}
using (ConsoleWithColors cout = new ConsoleWithColors(Environment.GetEnvironmentVariable("XSH_COLORS"), utf8))
using (CtrlCInterceptor ctrl = new CtrlCInterceptor())
{
context.Output += cout.OnOutput;
ctrl.Output = context.Error;
ctrl.Abort += delegate { context.Abort(); };
Stopwatch w = Stopwatch.StartNew();
try
{
// Parse arguments
var usage = new UsageGenerator()
{
Options = UsageOptions.None
};
xsParams.Parse(context, args, false);
setOutputOptions(context, cout);
if (context.IsSet(xs.path))
{
context.ScriptPath = context.GetString(xs.path);
}
// Load references
List <IScriptAction> preScript = getCommandlineReferences(context);
// Print help
if (args.Length == 0 || context.IsSet(xs.help))
{
loadReferences(context, preScript);
exitCode = HelpHelper.Help(context, usage, xsParams);
goto end;
}
// Handle upgrade
if (context.GetBool(xs.upgrade, false))
{
return(upgrade(context));
}
if (context.IsSet(xs.updateStage))
{
return(updateStage(context, context.GetStringArray(xs.updateStage)));
}
AppDomainLoader.progress("MainWithContext: Processing options");
// Process the remaining options
context.Compiler.AddRequireAdmin(Utils.To <RequireAdminMode>(context.GetStr(xs.requireAdmin, RequireAdminMode.User.ToString())));
if (context.IsSet(xs.codeout))
{
context.CodeOutputDirectory = Path.GetFullPath(context.GetString(xs.codeout));
}
if (context.IsSet(xs.genconfig) || context.IsSet(xs.gensample))
{
genDemoConfig(cout, context);
}
if (context.IsSet(xs.forcenet20))
{
context.Compiler.DefaultNETVersion = new Version(2, 0);
}
AppDomainLoader.progress("MainWithContext: Processing options, continuing");
List <string> filteredArgs = new List <string>();
if (context.IsSet(xs.scriptargs))
{
filteredArgs.AddRange(context.GetStringArray(xs.scriptargs));
}
// Run utilities, like //download etc
Script script = getInlineScript(context, filteredArgs);
string scriptName = context.GetString(xs.script, null);
if (script == null)
{
// Load the script
if (scriptName == "/?" || scriptName == "-?")
{
for (int i = 0; i < args.Length; ++i)
{
if (args[i] == "/?" || args[i] == "-?")
{
if (i != args.Length - 1)
{
context[xs.help] = args[i + 1];
}
break;
}
}
loadReferences(context, preScript);
return(HelpHelper.Help(context, usage, xsParams));
}
if (scriptName != null)
{
AppDomainLoader.progress("MainWithContext: Loading script " + scriptName);
script = loadScript(scriptName, context);
AppDomainLoader.progress("MainWithContext: Loading completed");
}
}
AppDomainLoader.progress("MainWithContext: About to initialize");
// Attach script
if (script != null)
{
// Insert pre-script before the script body
int n = 0;
foreach (var list in preScript)
{
script.Items.Insert(n++, list);
}
AppDomainLoader.BaseDirectory = script.DirectoryName;
RequireAdminMode mode = context.Compiler.RequireAdmin;
if ((!context.IsAdministrator || context.GetBool(xs.testElevation, false)) && !isCodeGeneration(context) && mode != RequireAdminMode.User)
{
return(restartAsAdmin(context, args, mode == RequireAdminMode.Hidden && !(context.GetBool(xs.testElevation, false))));
}
AppDomainLoader.progress("MainWithContext: Before script initialization");
if (isCodeGeneration(context))
{
if (!context.EnableCodePrecompilation && (context.IsSet(xs.genexe) ||
context.IsSet(xs.genwinexe) ||
context.IsSet(xs.genlibrary) ||
context.IsSet(xs.gencs)))
{
throw new ParsingException("One of the loaded scripts has precompilation disabled. Executable cannot be generated from this script.");
}
context.EnableCodePrecompilation = false;
}
ctrl.IgnoreCtrlC = script.IgnoreCtrlC;
ctrl.AbortDelay = Utils.ToTimeSpan(script.AbortDelay) ?? ctrl.AbortDelay;
ctrl.ExitDelay = Utils.ToTimeSpan(script.ExitDelay) ?? ctrl.ExitDelay;
context.Initialize(script);
AppDomainLoader.progress("MainWithContext: Script initialization completed");
}
// After precompilation we're ready to write .exe, if requested
if (isCodeGeneration(context))
{
doCodeGeneration(context, script);
}
else if (script != null)
{
// Run the script
AppDomainLoader.progress("MainWithContext: Before script execution");
redir = new ConsoleRedirector(context);
try
{
object r = context.ExecuteScript(script, filteredArgs.ToArray(), CallIsolation.High);
if (r != null)
{
int.TryParse(r.ToString(), out exitCode);
}
}
finally
{
ctrl.KillAbortTimer();
redir.Dispose();
redir = null;
}
AppDomainLoader.progress("MainWithContext: Script execution completed");
}
}
catch (ThreadAbortException ae)
{
AppDomainLoader.progress("MainWithContext: ThreadAbortException is being aborted");
resetAbort(context);
context.WriteException(ae);
exitCode = -1;
}
catch (ScriptTerminateException te)
{
exitCode = te.ExitCode;
resetAbort(context);
if (te.InnerException != null)
{
context.WriteException(te.InnerException);
AppDomainLoader.progress("MainWithContext: " + te.InnerException);
}
AppDomainLoader.progress("MainWithContext: Terminating with exit code " + exitCode);
}
catch (Exception e)
{
exitCode = -1;
resetAbort(context);
context.WriteException(e);
AppDomainLoader.progress("MainWithContext: " + e);
}
finally
{
resetAbort(context);
AppDomainLoader.BaseDirectory = null;
}
end:
// Display how long did it take
w.Stop();
if (context.GetBool(xs.wait, false))
{
cout.WriteLine(OutputType.Info, string.Format("Completed in {0} with exit code={1}. Press Enter to close...", w.Elapsed, exitCode));
Console.ReadLine();
}
}
AppDomainLoader.progress("MainWithContext: Exiting with code " + exitCode);
return(exitCode);
}
