public void OnAsyncMessage(PipeStream pipe, Byte[] data, Int32 bytes, Object state)
{
string scriptBlock = String.Empty;
try
{
scriptBlock = Encoding.UTF8.GetString(data, 0, bytes);
// EventLogHelper.WriteToEventLog(Config.ServiceName, 0, "IPC Script Block received: " + Environment.NewLine +
// "------------------------------------------" + Environment.NewLine +
// scriptBlock + Environment.NewLine +
// "------------------------------------------" + Environment.NewLine);
}
catch (Exception ex)
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 2,
"BluService: There was an error in reading script block as UTF8 string: " + Environment.NewLine + ex.Message);
}
// Execute and Write back results
try
{
string psResult = PowerShellRunspace.ExecuteScriptBlock(scriptBlock);
byte[] psResultBytes = Encoding.Default.GetBytes(psResult);
string result = Encoding.UTF8.GetString(psResultBytes, 0, Encoding.UTF8.GetByteCount(psResult));
data = Encoding.UTF8.GetBytes(result.ToCharArray());
pipe.BeginWrite(data, 0, Encoding.UTF8.GetByteCount(result), OnAsyncWriteComplete, pipe);
pipe.Close();
}
catch (Exception ex)
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 2,
"There is an error in executing script block UTF8 string: " + Environment.NewLine + ex.Message);
pipe.Close();
}
}
private void InitializeRunspace()
{
try
{
PowerShellRunspace.PsRunspace.Open();
}
catch (Exception ex)
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 2, "Error initializing runspace: " + ex.Message);
}
}
public static void DisposeRunspace()
{
PsRunspace.Dispose();
PsRunspace = null;
PsRunspace = RunspaceFactory.CreateRunspace(RunspaceConfiguration);
try
{
PsRunspace.Open();
}
catch (Exception ex)
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 2, "Error initializing runspace: " + ex.Message);
}
EventLogHelper.WriteToEventLog(Config.ServiceName, 1, "PowerShell Runspace is disposed." + Environment.NewLine + "All previously definied PS objects are garbage collected.");
}
static void Main(string[] args)
{
InputArguments inputArguments = new InputArguments(args);
string scriptBlock = String.Empty;
if (!string.IsNullOrEmpty(inputArguments["-Command"]))
{
scriptBlock = inputArguments["-Command"];
}
else if (!string.IsNullOrEmpty(inputArguments["-Define"]))
{
scriptBlock = inputArguments["-Define"].TransformForBlu();
}
else if (!string.IsNullOrEmpty(inputArguments["-Execute"]))
{
// TODO
}
else
{
Console.WriteLine("BluSheel needs to be executed with -Command or -Define or -Execute parameters.");
Environment.Exit(1);
}
try
{
IpcClient ipcClient = new IpcClient(".", "BluPowerShell");
PipeStream pipe = ipcClient.Connect(1);
Byte[] output = Encoding.UTF8.GetBytes(scriptBlock);
pipe.Write(output, 0, output.Length);
// Read the result
Byte[] data = new Byte[IpcClient.ClientInBufferSize];
Int32 bytesRead = pipe.Read(data, 0, data.Length);
ExitHandler(Encoding.UTF8.GetString(data, 0, bytesRead));
// Done with pipe
pipe.Close();
}
catch (Exception ex)
{
EventLogHelper.WriteToEventLog(Config.ShellName, 2, "Exception in send/recieve script block: " + Environment.NewLine + ex.Message);
Environment.Exit(1);
}
}
public static string ExecuteScriptBlock(string scriptBlock)
{
Pipeline pipeline;
string result = String.Empty;
string output;
string errors = String.Empty;
if (File.Exists(scriptBlock) && scriptBlock.EndsWith(".ps1"))
{
// Script block is actually a ps1 file, try to read it
string scriptFile = scriptBlock;
try
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 0, "Trying to read ps1 file: " + scriptFile);
scriptBlock = File.ReadAllText(scriptFile)
.TrimStart(' ')
.TrimStart(Environment.NewLine.ToCharArray())
.TrimStart('{')
.TrimEnd(' ')
.TrimEnd(Environment.NewLine.ToCharArray())
.TrimEnd('}');
EventLogHelper.WriteToEventLog(Config.ServiceName, 0, "File content is: " + scriptBlock);
}
catch (Exception)
{
return("Exit1:Cannot read: " + scriptFile);
}
}
try
{
pipeline = PsRunspace.CreatePipeline();
}
catch (Exception ex)
{
output = "Error creating pipeline: " + ex.Message;
EventLogHelper.WriteToEventLog(Config.ServiceName, 2, output);
return("Exit1:" + output);
}
// Dispose Runspace
if (scriptBlock == "DisposeRunspace")
{
DisposeRunspace();
return("Exit0:");
}
try
{
pipeline.Commands.AddScript(scriptBlock);
Collection <PSObject> psObjects = pipeline.Invoke();
if (pipeline.Error.Count > 0)
{
var error = pipeline.Error.Read() as Collection <ErrorRecord>;
if (error != null)
{
foreach (ErrorRecord er in error)
{
EventLogHelper.WriteToEventLog(Config.ServiceName, 1, "Collecting error messages...");
try
{
if (!string.IsNullOrEmpty(er.Exception.Message))
{
errors += "Message: " + er.Exception.Message + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.Source))
{
errors += "Source: " + er.Exception.Source + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.InnerException.ToString()))
{
errors += "InnerException: " + er.Exception.InnerException + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.StackTrace))
{
errors += "StackTrace: " + er.Exception.StackTrace + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.HelpLink))
{
errors += "HelpLink: " + er.Exception.HelpLink + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.TargetSite.ToString()))
{
errors += "TargetSite: " + er.Exception.TargetSite + Environment.NewLine;
}
if (!string.IsNullOrEmpty(er.Exception.Data.ToString()))
{
errors += "Exception Data: " + er.Exception.Data + Environment.NewLine;
}
errors += "--------------";
}
catch (Exception ex)
{
errors += "Error on collecting PowerShell exception messages: " + ex.Message;
}
output = "Script Block: " + scriptBlock.FormatForEventLog() + "Executed and returned the following errors:" +
Config.SeparatorLine + errors;
EventLogHelper.WriteToEventLog(Config.ServiceName, 2, output);
return("Exit1:" + errors);
}
}
}
output = "Execution of Script Block: " + scriptBlock.FormatForEventLog();
switch (psObjects.Count)
{
case 0:
// When there is no error, accept null output as a valid result, so return null
output += "Is completed successfully and returned null." + Environment.NewLine;
result = String.Empty;
break;
case 1:
// Accept null as a valid osObject and BaseObject, so return null
if (psObjects[0] == null || psObjects[0].BaseObject == null)
{
output += "Is completed successfully and returned null." + Environment.NewLine;
result = String.Empty;
}
else
{
// Else exit 0 with result
output += "Is completed successfully and returned:" + Environment.NewLine;
output += psObjects[0].BaseObject.ToString();
result += psObjects[0].BaseObject;
}
break;
default:
output += "Is completed successfully and returned:" + Environment.NewLine + Environment.NewLine;
foreach (var pso in psObjects)
{
output += pso.BaseObject + Environment.NewLine;
result += pso.BaseObject + Environment.NewLine;
}
break;
}
EventLogHelper.WriteToEventLog(Config.ServiceName, 0, "Output: " + output.FormatForEventLog());
return("Exit0:" + result.TrimEnd(Environment.NewLine.ToCharArray()).TrimEnd('\r', '\n'));
}
catch (Exception ex)
{
output = "Exception Invoking script block: " +
scriptBlock.FormatForEventLog() +
"Reason:" + Environment.NewLine +
ex.Message;
EventLogHelper.WriteToEventLog(Config.ServiceName, 2, output);
return("Exit1:" + output);
}
}