lancommander-ref/LANCommander
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix PowerShell scripts, add script debug mode

Browse source 
PowerShell scripts now execute correctly with the client. The built in PowerShell API's serializer will always return null on .NET 8 for some reason. The serializer/deserializer was switched to YAML, mostly due to convenience of already having a package for it.

Added a new settings for enabling script debugging mode. This ensures that when a script is executed, the PowerShell terminal window is not closed. This allows power users to get the same environment that the script executes in. It also allows for some scroll back to see what potentially went wrong in their script.
This commit is contained in:
Pat Hartl 2024-06-11 21:43:12 -05:00
parent 8e6c83d5a3
commit 6fec91e73e
7 changed files with 65 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

25
LANCommander.SDK/PowerShell/PowerShellScript.cs
View file

@ -7,6 +7,8 @@ using System.Linq;
using System.Reflection;
using System.Runtime.InteropServices;
using System.Text;
using YamlDotNet.Serialization;
using YamlDotNet.Serialization.NamingConventions;
namespace LANCommander.SDK.PowerShell
{
@ -17,6 +19,7 @@ namespace LANCommander.SDK.PowerShell
private bool AsAdmin { get; set; } = false;
private bool ShellExecute { get; set; } = false;
private bool IgnoreWow64 { get; set; } = false;
private bool Debug { get; set; } = false;
private ICollection<PowerShellVariable> Variables { get; set; }
private Dictionary<string, string> Arguments { get; set; }
private List<string> Modules { get; set; }
@ -122,6 +125,13 @@ namespace LANCommander.SDK.PowerShell
return this;
}
public PowerShellScript EnableDebug()
{
Debug = true;
return this;
}
public int Execute()
{
var scriptBuilder = new StringBuilder();
@ -143,6 +153,13 @@ namespace LANCommander.SDK.PowerShell
scriptBuilder.AppendLine(Contents);
if (Debug)
{
scriptBuilder.AppendLine("Read-Host");
Process.StartInfo.Arguments += " -NoExit";
}
var path = ScriptHelper.SaveTempScript(scriptBuilder.ToString());
AddArgument("File", path);
@ -181,8 +198,12 @@ namespace LANCommander.SDK.PowerShell
public static string Serialize<T>(T input)
{
// Use the PowerShell serializer to generate XML for our input. Then convert to base64 so we can put it on one line.
return Convert.ToBase64String(Encoding.UTF8.GetBytes(System.Management.Automation.PSSerializer.Serialize(input)));
var serializer = new SerializerBuilder()
.WithNamingConvention(new PascalCaseNamingConvention())
.Build();
// Use the YamlDotNet serializer to generate a string for our input. Then convert to base64 so we can put it on one line.
return Convert.ToBase64String(Encoding.UTF8.GetBytes(serializer.Serialize(input)));
}
[DllImport("kernel32.dll", SetLastError = true)]