Remotely/Agent/Services/WindowsService.cs

75 lines
2.5 KiB
C#

using Remotely.Shared.Utilities;
using System;
using System.Diagnostics;
using System.ServiceProcess;
namespace Remotely.Agent.Services
{
partial class WindowsService : ServiceBase
{
public WindowsService()
{
CanHandleSessionChangeEvent = true;
InitializeComponent();
}
protected override void OnSessionChange(SessionChangeDescription changeDescription)
{
try
{
Logger.Write($"Session changed. Reason: {changeDescription.Reason}. Session: {changeDescription.SessionId}");
if (changeDescription.Reason == SessionChangeReason.ConsoleDisconnect ||
changeDescription.Reason == SessionChangeReason.RemoteDisconnect)
{
foreach (var screenCaster in Process.GetProcessesByName("Remotely_Desktop"))
{
if (screenCaster.SessionId == changeDescription.SessionId)
{
Logger.Write($"Session changed. Kill process ID {screenCaster.Id}.");
screenCaster.Kill();
}
}
}
base.OnSessionChange(changeDescription);
}
catch (Exception ex)
{
Logger.Write(ex);
}
}
protected override void OnStart(string[] args)
{
try
{
// Set Secure Attention Sequence policy to allow app to simulate Ctrl + Alt + Del.
var subkey = Microsoft.Win32.Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System", true);
subkey.SetValue("SoftwareSASGeneration", "3", Microsoft.Win32.RegistryValueKind.DWord);
}
catch (Exception ex)
{
Logger.Write(ex);
}
}
protected override void OnStop()
{
try
{
// Remove Secure Attention Sequence policy to allow app to simulate Ctrl + Alt + Del.
var subkey = Microsoft.Win32.Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System", true);
if (subkey.GetValue("SoftwareSASGeneration") != null)
{
subkey.DeleteValue("SoftwareSASGeneration");
}
}
catch (Exception ex)
{
Logger.Write(ex);
}
}
}
}